run_sim_bmn_zdc.C

Go to the documentation of this file.

#include <Rtypes.h>
 
// inFile - input file with generator data, default: auau.09gev.mbias.98k.ftn14
// nStartEvent - for compatibility, any number
// nEvents - number of events to transport, default: 1
// outFile - output file with MC data, default: evetest.root
// flag_store_FairRadLenPoint
void run_sim_bmn_zdc(TString inFile = "/nica/data4mpd1/QGSM/AuAu4mb_mer.r12.gz",
                     TString outFile = "evetest.root",
                     Int_t nStartEvent = 0,
                     Int_t nEvents = 1,
                     Bool_t flag_store_FairRadLenPoint = kFALSE,
                     Bool_t isFieldMap = kTRUE)
{
 
#define LAQGSM
    // #define URQMD
 
    TStopwatch timer;
    timer.Start();
    gDebug = 0;
 
    gROOT->LoadMacro("$VMCWORKDIR/macro/run/bmnloadlibs.C");
    bmnloadlibs();   // load libraries
 
    gROOT->LoadMacro("$VMCWORKDIR/macro/zdc/geometry_with_zdc.C");
 
    // -----   Create simulation run   ----------------------------------------
    FairRunSim* fRun = new FairRunSim();
 
    // Choose the Geant Navigation System
    //   fRun->SetName("TGeant3");
    fRun->SetName("TGeant4");
    // fRun->SetGeoModel("G3Native");
 
    //    geometry_run1(fRun); // load bmn geometry
    geometry(fRun);   // load bmn geometry
 
    // Use the experiment specific MC Event header instead of the default one
    // This one stores additional information about the reaction plane
    // MpdMCEventHeader* mcHeader = new MpdMCEventHeader();
    // fRun->SetMCEventHeader(mcHeader);
 
    // Create and Set Event Generator
    //-------------------------------
 
    // Use the CbmUrqmdGenrator which calculates a reaction plane and
    // rotate all particles accordingly
    // CbmUrqmdGenerator*  urqmdGen = new CbmUrqmdGenerator(inFile);
    // urqmdGen->SetEventPlane(0. , 360.);
    // primGen->AddGenerator(urqmdGen);
    // fRun->SetGenerator(primGen);
 
    FairPrimaryGenerator* primGen = new FairPrimaryGenerator();
    fRun->SetGenerator(primGen);
 
    // smearing of beam interaction point
    // primGen->SetBeam(0.0,0.0,0.1,0.1);
    // primGen->SetTarget(0.0,24.0);
    // primGen->SmearGausVertexZ(kTRUE);
    // primGen->SmearVertexXY(kTRUE);
 
#ifdef URQMD
    // ------- Urqmd  Generator
    TString hostname = gSystem->HostName(), dataFile;
 
    if (inFile.Contains("/"))
        dataFile = inFile;
    else {
        dataFile = find_path_to_URQMD_files();
        if ((hostname == "lxmpd-ui.jinr.ru") || (hostname == "lxmpd-ui"))
            dataFile += "auau.09gev.mbias.10k.f14";
        else
            dataFile += inFile;
    }
 
    if (!CheckFileExist(dataFile))
        return;
 
    MpdUrqmdGenerator* urqmdGen = new MpdUrqmdGenerator(dataFile);
    primGen->AddGenerator(urqmdGen);
    if (nStartEvent > 0)
        urqmdGen->SkipEvents(nStartEvent);
 
    // if nEvents is equal 0 then all events (start with nStartEvent) of the given file should be processed
    if (nEvents == 0)
        nEvents = MpdGetNumEvents::GetNumURQMDEvents(dataFile.Data()) - nStartEvent;
 
#else
#ifdef PART
    // ------- Particle Generator
    FairParticleGenerator* partGen = new FairParticleGenerator(211, 10, 1, 0, 3, 1, 0, 0);
    primGen->AddGenerator(partGen);
 
#else
#ifdef ION
    // ------- Ion Generator
    FairIonGenerator* fIongen =
        //                  new FairIonGenerator(79, 197, 79, 1, 0., 0., 25, 0., 0., -1.);
        new FairIonGenerator(6, 12, 6, 100, 0., 0., 4, 0., 0., -1.);
    primGen->AddGenerator(fIongen);
    //    fIongen = new FairIonGenerator(79, 197, 79, 5, 0., 0., 35, 0., 0., -1.);
 
#else
#ifdef BOX
    gRandom->SetSeed(0);
    // ------- Box Generator
    FairBoxGenerator* boxGen = new FairBoxGenerator(13, 50);   // 13 = muon; 1 = multipl.
    boxGen->SetPRange(0.2, 5.05);                              // GeV/c //setPRange vs setPtRange
    boxGen->SetPhiRange(0, 360);                               // Azimuth angle range [degree]
    boxGen->SetThetaRange(5, 20);                              // Polar angle in lab system range [degree]
    boxGen->SetXYZ(0., 0., 0.);                                // mm o cm ??
    primGen->AddGenerator(boxGen);
 
#else
#ifdef HSD
    // ------- HSD/PHSD Generator
    TString dataFile;
    if (inFile.Contains("/"))
        dataFile = inFile;
    else {
        dataFile = find_path_to_URQMD_files();
        dataFile += "/../../HSD/";   //  nc-farm
        dataFile += inFile;
    }
 
    if (!CheckFileExist(dataFile))
        return;
 
    MpdPHSDGenerator* hsdGen = new MpdPHSDGenerator(dataFile.Data());
    // hsdGen->SetPsiRP(0.); // set fixed Reaction Plane angle instead of random
    primGen->AddGenerator(hsdGen);
    if (nStartEvent > 0)
        hsdGen->SkipEvents(nStartEvent);
 
    // if nEvents is equal 0 then all events (start with nStartEvent) of the given file should be processed
    if (nEvents == 0)
        nEvents = MpdGetNumEvents::GetNumPHSDEvents(dataFile.Data()) - nStartEvent;
 
#else
#ifdef LAQGSM
    // ------- LAQGSM Generator
    TString dataFile;
    if (inFile.Contains("/"))
        dataFile = inFile;
    else {
        dataFile = find_path_to_URQMD_files();
        dataFile += "/../../QGSM/";   //  nc-farm
        dataFile += inFile;
    }
 
    if (!CheckFileExist(dataFile))
        return;
 
    MpdLAQGSMGenerator* guGen = new MpdLAQGSMGenerator(dataFile.Data(), kFALSE);
    primGen->AddGenerator(guGen);
    if (nStartEvent > 0)
        guGen->SkipEvents(nStartEvent);
 
    // if nEvents is equal 0 then all events (start with nStartEvent) of the given file should be processed
    if (nEvents == 0)
        nEvents = MpdGetNumEvents::GetNumQGSMEvents(dataFile.Data()) - nStartEvent;
 
#endif
#endif
#endif
#endif
#endif
#endif
 
    fRun->SetOutputFile(outFile.Data());
 
    // -----   Create magnetic field   ----------------------------------------
    if (isFieldMap) {
        Double_t fieldScale = 1.;
        // BmnFieldMap* magField = new BmnNewFieldMap("field_sp41v2_ascii_noExtrap.dat");
        BmnFieldMap* magField = new BmnNewFieldMap("field_sp41v3_ascii_Extrap.dat");
        // Double_t fieldZ = 124.5; // field centre z position
        // magField->SetPosition(0., 0., fieldZ);
        magField->SetScale(fieldScale);
        fRun->SetField(magField);
    } else {
        BmnFieldConst* magField = new BmnFieldConst();
        magField->SetFieldRegion(-300., 300., -300., 300., -300., 300);
        magField->SetField(0., -9. * 0.44, 0.);
        fRun->SetField(magField);
    }
 
    fRun->SetStoreTraj(kTRUE);
    fRun->SetRadLenRegister(flag_store_FairRadLenPoint);   // radiation length manager
 
    fRun->Init();
    if (isFieldMap)
        magField->Print();
 
    // Trajectories Visualization (TGeoManager only)
    //-------------------------------------------
    FairTrajFilter* trajFilter = FairTrajFilter::Instance();
    // Set cuts for storing the trajectories
    trajFilter->SetStepSizeCut(0.01);                                   // 1 cm
    trajFilter->SetVertexCut(-200., -200., -150., 200., 200., 1100.);   //
    trajFilter->SetMomentumCutP(10e-3);                                 // p_lab > 10 MeV
    trajFilter->SetEnergyCut(0., 4.);                                   // 0 < Etot < 1.04 GeV //
    trajFilter->SetStorePrimaries(kTRUE);
    trajFilter->SetStoreSecondaries(kTRUE);   // kFALSE
 
    // Fill the Parameter containers for this run
    //-------------------------------------------
    FairRuntimeDb* rtdb = fRun->GetRuntimeDb();
 
    BmnFieldPar* fieldPar = (BmnFieldPar*)rtdb->getContainer("BmnFieldPar");
    fieldPar->SetParameters(magField);
    fieldPar->setChanged();
    fieldPar->setInputVersion(fRun->GetRunId(), 1);
    Bool_t kParameterMerged = kTRUE;
    FairParRootFileIo* output = new FairParRootFileIo(kParameterMerged);
    // AZ output->open(parFile.Data());
    output->open(gFile);
    rtdb->setOutput(output);
 
    /*MpdMultiFieldPar* Par = (MpdMultiFieldPar*) rtdb->getContainer("MpdMultiFieldPar");
    if (fField)
      Par->SetParameters(fField);
    Par->setInputVersion(fRun->GetRunId(),1);
    Par->setChanged();
    // Par->printParams();
     */
 
    rtdb->saveOutput();
    rtdb->print();
 
    // Transport nEvents
    // -----------------
    fRun->Run(nEvents);
 
    fRun->CreateGeometryFile("geofile_full.root");
 
#ifdef LAQGSM
    TString Pdg_table_name =
        TString::Format("%s%s%c%s", gSystem->BaseName(dataFile.Data()), ".g", (fRun->GetName())[6], ".pdg_table.dat");
    (TDatabasePDG::Instance())->WritePDGTable(Pdg_table_name.Data());
#endif
 
    timer.Stop();
    Double_t rtime = timer.RealTime(), ctime = timer.CpuTime();
    printf("RealTime=%f seconds, CpuTime=%f seconds\n", rtime, ctime);
 
    cout << "Macro finished succesfully." << endl;
 
    gApplication->Terminate();
}