run_tracking_sttcombi_timebased.C File Reference

Go to the source code of this file.

Functions
int	run_tracking_sttcombi_timebased ()

Function Documentation

int run_tracking_sttcombi_timebased

(

)

Definition at line 1 of file run_tracking_sttcombi_timebased.C.

References PndSttFindTracks::AddHitCollectionName(), allDigiFile, creator, ctime, DigiFile, Double_t, fRun, Geane, PndEmcMapper::Init(), iVerbose, MCFile, mvdTrackFinder, nEvents, outFile, parFile, parInput1, parIo1, RecoFile, recoKalman, recoKalmanFwd, rtdb, rtime, PndRecoKalmanTask::SetBusyCut(), PndMvdRiemannTrackFinderTask::SetMaxDist(), PndMvdRiemannTrackFinderTask::SetPersistence(), PndFtsTrackerIdeal::SetRelativeMomentumSmearing(), PndRecoKalmanTask::SetTrackInBranchName(), PndFtsTrackerIdeal::SetTrackingEfficiency(), PndRecoKalmanTask::SetTrackOutBranchName(), PndFtsTrackerIdeal::SetTrackOutput(), PndMvdRiemannTrackFinderTask::SetVerbose(), PndFtsTrackerIdeal::SetVertexSmearing(), sttFindTracks, SttMvdGemTracking, SttMvdTracking, sttTrackFinder, sysFile, timer, trackFts, trackMC, trackMC2, trackMC3, and TString.

                                     {
  // ========================================================================
  // Verbosity level (0=quiet, 1=event level, 2=track level, 3=debug)
  Int_t iVerbose = 0;

  // Input file
  TString MCFile = "Mvd_Sim.root";

  // Number of events to process
  Int_t nEvents = 0;

  PndFileNameCreator creator(MCFile.Data());
  TString DigiFile = creator.GetDigiFileName("timebased").c_str();
  TString RecoFile = creator.GetRecoFileName("timebased").c_str();
  TString parFile = creator.GetParFileName().c_str();
  TString outFile = creator.GetTrackFindingFileName("timebased").c_str();

  TString sysFile = gSystem->Getenv("VMCWORKDIR");
  // ------------------------------------------------------------------------
  // In general, the following parts need not be touched
  // ========================================================================

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

  // -----   Digitization run   -------------------------------------------
  FairRunAna *fRun= new FairRunAna();
  fRun->SetInputFile(RecoFile);
  fRun->AddFriend(MCFile);
  fRun->AddFriend(DigiFile);
  fRun->RunWithTimeStamps();
  fRun->SetOutputFile(outFile);
  FairGeane *Geane = new FairGeane();
  fRun->AddTask(Geane);
  // ------------------------------------------------------------------------

  // -----  Parameter database   --------------------------------------------
   TString allDigiFile = sysFile+"/macro/params/all.par";

  FairRuntimeDb* rtdb = fRun->GetRuntimeDb();
  FairParRootFileIo* parInput1 = new FairParRootFileIo();
  parInput1->open(parFile.Data());
        
  FairParAsciiFileIo* parIo1 = new FairParAsciiFileIo();
  parIo1->open(allDigiFile.Data(),"in");
        
  rtdb->setFirstInput(parInput1);
  rtdb->setSecondInput(parIo1);
  // ------------------------------------------------------------------------
 
  PndMvdRiemannTrackFinderTask* mvdTrackFinder = new PndMvdRiemannTrackFinderTask();
  mvdTrackFinder->SetVerbose(1);
  mvdTrackFinder->SetMaxDist(0.05);
  mvdTrackFinder->SetPersistence(kTRUE);
  fRun->AddTask(mvdTrackFinder);

  //  PndSttTrackFinderIdeal* sttTrackFinder = new PndSttTrackFinderIdeal(iVerbose);
  PndSttTrackFinderReal* sttTrackFinder = new PndSttTrackFinderReal(0);
  PndSttFindTracks* sttFindTracks = new PndSttFindTracks("Track Finder", "FairTask", sttTrackFinder, iVerbose);
  sttFindTracks->AddHitCollectionName("STTHit", "STTPoint");
  //sttFindTracks->SetPersistence(kFALSE);
//  fRun->AddTask(sttFindTracks);
  
  PndSttMvdTracking *  SttMvdTracking = new PndSttMvdTracking(0, false, false);
  //SttMvdTracking->Cleanup();
  SttMvdTracking->SetPersistence(kFALSE);
//  fRun->AddTask(SttMvdTracking);
  
  //PndMCTrackAssociator* trackMC0 = new PndMCTrackAssociator();
  //trackMC0->SetTrackInBranchName("SttMvdTrack");
  //trackMC0->SetTrackOutBranchName("SttMvdTrackID");
  //trackMC0->SetPersistence(kFALSE);
  //fRun->AddTask(trackMC0);

  PndSttMvdGemTracking * SttMvdGemTracking = new PndSttMvdGemTracking(0);
  //SttMvdGemTracking->SetPdgFromMC();
//  fRun->AddTask(SttMvdGemTracking);

  PndMCTrackAssociator* trackMC = new PndMCTrackAssociator();
  trackMC->SetTrackInBranchName("SttMvdGemTrack");
  trackMC->SetTrackOutBranchName("SttMvdGemTrackID");
//  fRun->AddTask(trackMC);

  PndRecoKalmanTask* recoKalman = new PndRecoKalmanTask();
  recoKalman->SetTrackInBranchName("SttMvdGemTrack");
//  recoKalman->SetTrackInIDBranchName("SttMvdGemTrackID");
  recoKalman->SetTrackOutBranchName("SttMvdGemGenTrack");
  recoKalman->SetBusyCut(50); // CHECK to be tuned
  //recoKalman->SetIdealHyp(kTRUE);
  //recoKalman->SetNumIterations(3);
//  fRun->AddTask(recoKalman);

  PndMCTrackAssociator* trackMC2 = new PndMCTrackAssociator();
  trackMC2->SetTrackInBranchName("SttMvdGemGenTrack"); 
  trackMC2->SetTrackOutBranchName("SttMvdGemGenTrackID");
//  fRun->AddTask(trackMC2);
 
  PndFtsTrackerIdeal* trackFts = new PndFtsTrackerIdeal();
  trackFts->SetRelativeMomentumSmearing(0.02);
  trackFts->SetVertexSmearing(0.02, 0.02, 0.02);
  trackFts->SetTrackingEfficiency(1.);
  trackFts->SetTrackOutput("FtsIdealTrack");
//  fRun->AddTask(trackFts);

  PndRecoKalmanTask* recoKalmanFwd = new PndRecoKalmanTask();
  recoKalmanFwd->SetTrackInBranchName("FtsIdealTrack");
  //recoKalmanFwd->SetTrackInIDBranchName("FtsIdealTrackID");
  recoKalmanFwd->SetTrackOutBranchName("FtsIdealGenTrack");
  recoKalmanFwd->SetBusyCut(50); // CHECK to be tuned
  //recoKalmanFwd->SetIdealHyp(kTRUE);
  //recoKalmanFwd->SetNumIterations(3);
//  fRun->AddTask(recoKalmanFwd);

  PndMCTrackAssociator* trackMC3 = new PndMCTrackAssociator();
  trackMC3->SetTrackInBranchName("FtsIdealGenTrack");
  trackMC3->SetTrackOutBranchName("FtsIdealGenTrackID");
//  fRun->AddTask(trackMC3);
 
  // -----   Intialise and run   --------------------------------------------
  PndEmcMapper::Init(1);
  fRun->Init();
  fRun->Run(0, nEvents);

  rtdb->saveOutput();
  rtdb->print();

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

  // -----   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;
  cout << endl;
  // ------------------------------------------------------------------------

  return 0;

}