emc_correction_data_production.C File Reference

Go to the source code of this file.

Functions
int	emc_correction_data_production (Int_t nEvents=10, TString part="gamma", TString TransportModel="TGeant3", Bool_t useFullPandaGeometry=true, TString emcModule="TS", TString energyRange="all", UInt_t seed=0)

Function Documentation

int emc_correction_data_production	(	Int_t	nEvents = 10,
		TString	part = "gamma",
		TString	TransportModel = "TGeant3",
		Bool_t	useFullPandaGeometry = true,
		TString	emcModule = "TS",
		TString	energyRange = "all",
		UInt_t	seed = 0
	)

Set the digitization parameters

Get the run time data base for this session and set the needed input

Parameters created for this simulation goes to the out put

Add Hit producer task to the simulation

Initialize the session

After initialization now we can save the field parameters

All parameters are initialized and ready to be saved

Definition at line 6 of file emc_correction_data_production.C.

References Bool_t, boxGen, Cave, ctime, digiFile, Dipole, Double_t, Drc, Dsk, Emc, emcDigiFile, emcHdrFiller, emcHitProd, emcHitsToWaveform, emcMakeBump, emcMakeCluster, emcWaveformToDigi, fField, fRun, Fts, gDebug, Gem, kParameterMerged, Magnet, Muo, Mvd, nEvents, output, Par, parIo1, Pipe, primGen, printf(), rtdb, rtime, seed, PndMdt::SetBarrel(), PndMdt::SetEndcap(), PndMdt::SetForward(), PndEmc::SetGeometryVersion(), PndMdt::SetMdtMagnet(), PndMdt::SetMdtMFIron(), PndMdt::SetMuonFilter(), PndMultiFieldPar::SetParameters(), PndDrc::SetRunCherenkov(), PndEmcMakeCluster::SetStorageOfData(), PndEmcMakeBump::SetStorageOfData(), PndEmcHitsToWaveform::SetStorageOfData(), PndEmcWaveformToDigi::SetStorageOfData(), PndEmcHitProducer::SetStorageOfData(), PndEmc::SetStorageOfData(), PndDsk::SetStoreCerenkovs(), PndDsk::SetStoreTrackPoints(), Stt, timer, and TString.

{
        gRandom->SetSeed(seed);
        Double_t phi_min = 0; Double_t phi_max = 360;
        
        if (emcModule=="TS")
        {
                Double_t theta_min = 5; Double_t theta_max = 172;
        } else if (emcModule=="shashlyk")
        {
                Double_t theta_min = 0; Double_t theta_max = 10;
        } else 
        {
                std::cout<<"Incorrect emcModule parameter: "<<emcModule<<std::endl;
                abort();
        }
        
        // energyRange= "low", "high", "all"
        if (energyRange=="low")
        {
                Double_t momentum_min = 0.0; Double_t momentum_max = 1.5;
        } else if (energyRange=="high")
        {
                Double_t momentum_min = 1.0; Double_t momentum_max = 10.0;
        } else if (energyRange=="all")
        {
                Double_t momentum_min = 0.0; Double_t momentum_max = 10.0;
        } else
        {
                std::cout<<"Incorrect parameter energyRange: "<<energyRange<<std::endl;
                abort();
        }
        
        TString pandaGeometry;
        if (useFullPandaGeometry==0)
                pandaGeometry="emc";
        else
                pandaGeometry="full";
        
        TString s1; s1+=seed;
        
        TString OutputSimFile = "emc_complete";
        OutputSimFile = OutputSimFile+"_"+part+"_"+TransportModel+
        "_"+pandaGeometry+"_"+emcModule+"_"+s1+".root";
        TString OutputDatabaseFile = "simparams";
        OutputDatabaseFile = OutputDatabaseFile+"_"+part+"_"+TransportModel+
        "_"+pandaGeometry+"_"+emcModule+"_"+s1+".root";

        TStopwatch timer;
        timer.Start();
        gDebug=0;

        FairLogger::GetLogger()->SetLogToFile(kFALSE);          
  FairRunSim *fRun = new FairRunSim();
        
  TString digiFile = "emc.par";
        
  // set the MC version used
  // ------------------------
        
   fRun->SetName(TransportModel);
        
  
  fRun->SetOutputFile(OutputSimFile);
        

  TString emcDigiFile = gSystem->Getenv("VMCWORKDIR");
  emcDigiFile += "/macro/params/";
  emcDigiFile += digiFile;

  FairRuntimeDb* rtdb = fRun->GetRuntimeDb();
  FairParAsciiFileIo* parIo1 = new FairParAsciiFileIo();
  parIo1->open(emcDigiFile.Data(),"in");
  rtdb->setFirstInput(parIo1);        
  Bool_t kParameterMerged=kTRUE;

  FairParRootFileIo* output=new FairParRootFileIo(kParameterMerged);
  output->open(OutputDatabaseFile);
  rtdb->setOutput(output);



  // Set Material file Name
  //-----------------------
  fRun->SetMaterials("media_pnd.geo");
        
  // Create and add detectors
  //-------------------------
  FairModule *Cave= new PndCave("CAVE");
  Cave->SetGeometryFileName("pndcave.geo");
  fRun->AddModule(Cave); 
        
  PndEmc *Emc = new PndEmc("EMC",kTRUE);
  Emc->SetGeometryVersion(1);
  Emc->SetStorageOfData(kFALSE);
  fRun->AddModule(Emc);
  
  if (useFullPandaGeometry)
  {
        FairModule *Magnet= new PndMagnet("MAGNET");
        Magnet->SetGeometryFileName("FullSuperconductingSolenoid_v831.root");
        fRun->AddModule(Magnet);

        FairModule *Dipole= new PndMagnet("MAGNET");
        Dipole->SetGeometryFileName("dipole.geo");
        fRun->AddModule(Dipole);

  FairModule *Pipe= new PndPipe("PIPE");
  Pipe->SetGeometryFileName("beampipe_201309.root");
  fRun->AddModule(Pipe);

        FairDetector *Stt= new PndStt("STT", kTRUE);
        Stt->SetGeometryFileName("straws_skewed_blocks_35cm_pipe.geo");
        fRun->AddModule(Stt);

        FairDetector *Mvd = new PndMvdDetector("MVD", kTRUE);
        Mvd->SetGeometryFileName("Mvd-2.1_FullVersion.root");
        fRun->AddModule(Mvd);

  //-------------------------  MDT       -----------------
  PndMdt *Muo = new PndMdt("MDT",kTRUE);
  Muo->SetBarrel("fast");
  Muo->SetEndcap("fast");
  Muo->SetMuonFilter("fast");
  Muo->SetForward("fast");
  Muo->SetMdtMagnet(kTRUE);
  Muo->SetMdtMFIron(kTRUE);
  fRun->AddModule(Muo);

        FairDetector *Gem = new PndGemDetector("GEM", kTRUE);
        Gem->SetGeometryFileName("gem_3Stations.root");
        fRun->AddModule(Gem);
  //-------------------------  SCITIL    -----------------
  FairDetector *SciT = new PndSciT("SCIT",kTRUE);
  SciT->SetGeometryFileName("barrel-SciTil_07022013.root");
  fRun->AddModule(SciT);
  //-------------------------  DRC       -----------------
  PndDrc *Drc = new PndDrc("DIRC", kTRUE);
  Drc->SetGeometryFileName("dirc_l0_p0_updated.root"); 
  Drc->SetRunCherenkov(kFALSE);
  fRun->AddModule(Drc); 

        PndDsk* Dsk = new PndDsk("DSK", kTRUE);
        Dsk->SetGeometryFileName("dsk.root");
        Dsk->SetStoreCerenkovs(kFALSE);
        Dsk->SetStoreTrackPoints(kFALSE);
        fRun->AddModule(Dsk);

        FairDetector *Fts= new PndFts("FTS", kTRUE);
        Fts->SetGeometryFileName("fts.geo");
        fRun->AddModule(Fts);
  }
        
  // Create and Set Event Generator
  //-------------------------------
        
  FairPrimaryGenerator* primGen = new FairPrimaryGenerator();
  fRun->SetGenerator(primGen);
  
  TDatabasePDG *pdg = new TDatabasePDG();
  Int_t part_pid = pdg->GetParticle(part)->PdgCode();
  
  // Box Generator: 
  FairBoxGenerator* boxGen = new FairBoxGenerator(part_pid, 1); // 13 = muon; 1 = multipl. // 211 = pi+
  // first number: PDG particle code: 2nd number: particle multiplicity per event
        
  boxGen->SetPRange(momentum_min,momentum_max); // GeV/c
  // boxGen->SetPtRange(1.,1.); // GeV/c
  boxGen->SetPhiRange(phi_min, phi_max); // Azimuth angle range [degree]
  boxGen->SetThetaRange(theta_min, theta_max); // Polar angle in lab system range [degree]
  boxGen->SetXYZ(0., 0., 0.); // vertex coordinates [mm]
  primGen->AddGenerator(boxGen);  
        
  // fRun->SetStoreTraj(kTRUE); // to store particle trajectories 
  fRun->SetBeamMom(15); 
        //---------------------Create and Set the Field(s)---------- 
  PndMultiField *fField= new PndMultiField("FULL");
  fRun->SetField(fField);
        
  PndEmcHitProducer* emcHitProd = new PndEmcHitProducer();
  emcHitProd->SetStorageOfData(kFALSE);//FALSE);
  fRun->AddTask(emcHitProd);

  PndEmcHitsToWaveform* emcHitsToWaveform= new PndEmcHitsToWaveform(0);
  PndEmcWaveformToDigi* emcWaveformToDigi=new PndEmcWaveformToDigi(0);
  emcHitsToWaveform->SetStorageOfData(kFALSE);
  emcWaveformToDigi->SetStorageOfData(kTRUE);
  fRun->AddTask(emcHitsToWaveform);  // full digitization
  fRun->AddTask(emcWaveformToDigi);  // full digitization
 
  PndEmcMakeCluster* emcMakeCluster= new PndEmcMakeCluster(0);
  emcMakeCluster->SetStorageOfData(kTRUE);
  fRun->AddTask(emcMakeCluster);

  PndEmcHdrFiller* emcHdrFiller = new PndEmcHdrFiller();
  fRun->AddTask(emcHdrFiller); // ECM header

  PndEmcMakeBump* emcMakeBump= new PndEmcMakeBump();
  emcMakeBump->SetStorageOfData(kTRUE);
  fRun->AddTask(emcMakeBump);
   
  fRun->Init();
  PndMultiFieldPar* Par = (PndMultiFieldPar*) rtdb->getContainer("PndMultiFieldPar");
  if (fField) {  Par->SetParameters(fField); }
  Par->setInputVersion(fRun->GetRunId(),1);
  Par->setChanged();
        
  rtdb->saveOutput();
  //    rtdb->print();
                
  fRun->Run(nEvents);
  timer.Stop();
        
  Double_t rtime = timer.RealTime();
  Double_t ctime = timer.CpuTime();
  printf("RealTime=%f seconds, CpuTime=%f seconds\n",rtime,ctime);
        
  return 0;
}