runMQTrans.cxx

Go to the documentation of this file.

#include "runFairMQDevice.h"

// MQRunSim
#include "FairCave.h"
#include "FairMQTransportDevice.h"
#include "FairModule.h"
#include "FairOnlineSink.h"
#include "FairParAsciiFileIo.h"
#include "PixelDigitize.h"

#include <TObjArray.h>
#include <TRandom.h>
#include <TString.h>
#include <TSystem.h>
#include <cstdlib>
#include <string>
#include <vector>

namespace bpo = boost::program_options;

void addCustomOptions(bpo::options_description& options)
{
    // clang-format off
    options.add_options()
        ("random-seed",         bpo::value<int64_t>()->default_value(0),                   "Random seed number")
        ("transport-name",      bpo::value<std::string>()->default_value("TGeant3"),       "Transport name")
        ("fairroot-config-dir", bpo::value<std::string>()->default_value(""),              "FairRoot config dir")
        ("param-channel-name",  bpo::value<std::string>()->default_value("updateChannel"), "Parameter update channel name")
        ("running-mode",        bpo::value<std::string>()->default_value("pp"),            "pp to pull, rr to request")
        ("detector-library",    bpo::value<std::vector<std::string>>(),                    "detector library")
        ("run-digi-tasks",      bpo::value<bool>()->default_value(false),                  "Run digi tasks");
    // clang-format on
}

FairMQDevicePtr getDevice(const FairMQProgOptions& config)
{
    gRandom->SetSeed(config.GetValue<int64_t>("random-seed"));

    TString dir = getenv("VMCWORKDIR");
    TString tutdir = dir + "/MQ/pixelDetector";

    TString tut_geomdir = dir + "/common/geometry";
    gSystem->Setenv("GEOMPATH", tut_geomdir.Data());

    TString tut_configdir = config.GetValue<std::string>("fairroot-config-dir");
    if (tut_configdir.Length() < 1)
        tut_configdir = dir + "/common/gconfig";
    gSystem->Setenv("CONFIG_DIR", tut_configdir.Data());

    FairMQTransportDevice* run = new FairMQTransportDevice();
    run->RunInPullMode(true);
    if (config.GetValue<std::string>("running-mode") == "rr") {
        LOG(INFO) << "Going to request data.";
        run->RunInPullMode(false);
    } else {
        LOG(INFO) << "Going to pull data.";
    }

    //  TString outputfilename = Form("outputfile_%d.root",(int)(getpid()));
    //  FairRootFileSink* sink = new FairRootFileSink(outputfilename);
    FairOnlineSink* sink = new FairOnlineSink();
    sink->SetMQRunDevice(run);
    run->SetSink(sink);

    run->SetParamUpdateChannelName(config.GetValue<std::string>("param-channel-name"));

    run->SetTransportName(config.GetValue<std::string>("transport-name"));
    run->SetMaterials("media.geo");

    TObjArray* detArray = new TObjArray();
    FairModule* cave = new FairCave("CAVE");
    cave->SetGeometryFileName("cave_vacuum.geo");
    detArray->Add(cave);
    // Pixel*  det = new Pixel("PixelDetector", kTRUE);
    // det->SetGeometryFileName("pixel.geo");
    // detArray->Add(det);
    run->SetDetectorArray(detArray);

    run->SetStoreTraj(false);

    if ((config.GetValue<bool>("run-digi-tasks")) == true) {
        // Attach tasks if needed
        TString digParFile = tutdir + "/param/pixel_digi.par";
        FairParAsciiFileIo* parIo1 = new FairParAsciiFileIo();
        parIo1->open(digParFile.Data(), "in");
        run->SetFirstParameter(parIo1);

        TObjArray* taskArray = new TObjArray();
        PixelDigitize* digiTask = new PixelDigitize();
        taskArray->Add(digiTask);
        run->SetTaskArray(taskArray);
    }

    return run;
}