// --------------------------------------------------------------------------
//
// Macro for standard transport simulation using UrQMD input and GEANT3
// Standard CBM setup with STS, RICH, TRD and TOF
// Does not (yet) include ECAL
//
// author:   V. Friese   06/03/2006
// modified: T. Galatyuk 06/05/2006
// --------------------------------------------------------------------------

{
    // ========================================================================
    //          Adjust this part according to your requirements

    // Collision system
    TString system  = "auau";

    // Beam momentum
    TString beam    = "25gev";

    // Trigger (centrality)
    TString trigger = "centr";

    // Number of events
    Int_t nEvents   = 10;

    // Output file name
    TString outFile = system + "." + beam + "." + trigger + ".mc.root";
    outFile = "urqmd_omega_ee.auau.25gev.centr.1k.root";

    // Parameter file name
    TString parFile = "urqmd_omega_ee.auau.25gev.centr.1k.params.root";

    // Cave geometry
    TString caveGeom = "cave.geo";

    // Target geometry
    TString targetGeom = "target.geo";

    // Beam pipe geometry
    TString pipeGeom = "pipe_newvac.geo";

    // Magnet geometry and field map
    TString magnetGeom  = "magnet_active.geo";
    TString fieldMap    = "FieldActive";
    Double_t fieldZ     = 70.;     // z position of field centre
    Double_t fieldScale = 0.3;     // field scaling factor

    // STS geometry
    TString stsGeom = "sts-dilept-2maps-4strips.geo";

    // RICH geometry
    TString richGeom = "rich.geo";

    // TRD geometry
    TString trdGeom = "trd_9.geo";

    // TOF geometry
    TString tofGeom = "tof.geo";

    // In general, the following parts need not be touched
    // ========================================================================



    // -----   Input file name   ----------------------------------------------
    TString inUrqmdFile = "/d/cbm01/cbmsim/urqmd/" + system + "/" + beam + "/"
	+ trigger + "/urqmd." + system + "." + beam + "."
	+ trigger + ".0000.ftn14";

    TString inPlutoFile = "/d/cbm02/galatyuk/pluto/auau/25gev/omega/epem/pluto.auau.25gev.omega.epem.0000.root";
    // ------------------------------------------------------------------------



    // ----    Debug option   -------------------------------------------------
    gDebug = 0;
    // ------------------------------------------------------------------------



    // -----   Timer   --------------------------------------------------------
    TStopwatch timer;
    timer.Start();
    // ------------------------------------------------------------------------



    // ----  Load libraries   -------------------------------------------------
    gROOT->LoadMacro("$VMCWORKDIR/gconfig/basiclibs.C");
    basiclibs();
    gSystem->Load("libGeoBase");
    gSystem->Load("libParBase");
    gSystem->Load("libBase");
    gSystem->Load("libField");
    gSystem->Load("libGen");
    gSystem->Load("libPassive");
    gSystem->Load("libSts");
    gSystem->Load("libRich");
    gSystem->Load("libTrd");
    gSystem->Load("libTof");
    // ------------------------------------------------------------------------



    // -----   Create detectors and passive volumes   -------------------------
    FairModule* cave= new CbmCave("CAVE");
    cave->SetGeometryFileName(caveGeom);

    FairModule* pipe= new CbmPipe("PIPE");
    pipe->SetGeometryFileName(pipeGeom);

    FairModule* target= new CbmTarget("Target");
    target->SetGeometryFileName(targetGeom);

    FairModule* magnet= new CbmMagnet("MAGNET");
    magnet->SetGeometryFileName(magnetGeom);

    FairDetector* sts= new CbmSts("STS", kTRUE);
    sts->SetGeometryFileName(stsGeom);

    FairDetector* rich= new CbmRich("RICH", kTRUE);
    rich->SetGeometryFileName(richGeom);

    FairDetector* trd= new CbmTrd("TRD",kTRUE );
    trd->SetGeometryFileName(trdGeom);

    FairDetector* tof= new CbmTof("TOF", kTRUE );
    tof->SetGeometryFileName(tofGeom);

    // ------------------------------------------------------------------------



    // -----   Create magnetic field   ----------------------------------------
    if ( fieldMap == "FieldActive" || fieldMap == "FieldIron")
	magField = new CbmFieldMapSym3(fieldMap);
    else if ( fieldMap == "FieldAlligator" )
	magField = new CbmFieldMapSym2(fieldMap);
    else {
	cout << "===> ERROR: Field map " << fieldMap << " unknown! " << endl;
	exit;
    }
    magField->SetPosition(0., 0., fieldZ);
    magField->SetScale(fieldScale);
    // ------------------------------------------------------------------------



    // -----   Create PrimaryGenerator   --------------------------------------
    FairPrimaryGenerator* primGen = new FairPrimaryGenerator();

    FairUrqmdGenerator*  urqmdGen = new FairUrqmdGenerator(inUrqmdFile);
    primGen->AddGenerator(urqmdGen);

    FairPlutoGenerator*  plutoGen = new FairPlutoGenerator(inPlutoFile);
    primGen->AddGenerator(plutoGen);
    // ------------------------------------------------------------------------



    // -----   Create simulation run   ----------------------------------------
    FairRunSim* fRun = new FairRunSim();
    fRun->SetName("TGeant3");              // Transport engine
    fRun->SetOutputFile(outFile);          // Output file
    fRun->SetGenerator(primGen);           // PrimaryGenerator
    fRun->SetMaterials("media.geo");       // Materials
    fRun->AddModule(cave);
    fRun->AddModule(pipe);
    fRun->AddModule(target);
    fRun->AddModule(magnet);
    fRun->AddModule(sts);
    fRun->AddModule(rich);
    fRun->AddModule(trd);
    fRun->AddModule(tof);
    fRun->SetField(magField);
    fRun->Init();
    // ------------------------------------------------------------------------



    // -----   Fill parameter containers   ------------------------------------
    FairRuntimeDb* rtdb = fRun->GetRuntimeDb();
    Bool_t kParameterMerged = kTRUE;
    FairParRootFileIo* parOut = new FairParRootFileIo(kParameterMerged);
    parOut->open(parFile.Data());
    rtdb->setOutput(parOut);
    rtdb->saveOutput();
    rtdb->print();
    // ------------------------------------------------------------------------



    // -----   Start run   ----------------------------------------------------
    fRun->Run(nEvents);
    // ------------------------------------------------------------------------



    // -----   Finish   -------------------------------------------------------
    timer.Stop();
    Double_t rtime = timer.RealTime();
    Double_t ctime = timer.CpuTime();
    cout << endl << endl;
    cout << "Macro finished succesfully." << endl;
    cout << "Output file is "    << outFile << endl;
    cout << "Parameter file is " << parFile << endl;
    cout << "Real time " << rtime << " s, CPU time " << ctime
	<< "s" << endl << endl;
    // ------------------------------------------------------------------------
}