runLumiPixel5cCombFilt.C

Go to the documentation of this file.

int runLumiPixel5cCombFilt(const int nEvents=10, const int startEvent=0, TString storePath="tmpOutput", const int verboseLevel=0, const double mom=15, const double dX=0, const double dY=0)
{
  // ========================================================================
  // Input file (MC events)
  TString MCFile = storePath+"/Lumi_MC_";
  MCFile += startEvent;
  MCFile += ".root";
  // TString DigiFile = storePath+"/Lumi_digi_";
  // DigiFile += startEvent;
  // DigiFile += ".root";
  // // Digi file
  // TString RecoFile = storePath;
  // if(mergedHits)RecoFile+="/Lumi_recoMerged_";
  // else RecoFile+="/Lumi_reco_";
  // RecoFile += startEvent;
  // RecoFile += ".root";
  // // TCand file
  // TString CandFile = storePath+"/Lumi_TCand_";
  // CandFile += startEvent;
  // CandFile += ".root";
  // Parameter file
  TString parFile = storePath+"/Lumi_Params_";
  parFile += startEvent;
  parFile += ".root";
  // Track file
  TString TrkFile = storePath+"/Lumi_Track_";
  TrkFile += startEvent;
  TrkFile += ".root";
  // Geane output file
  TString GeaneFile = storePath+"/Lumi_Geane_";
  GeaneFile += startEvent;
  GeaneFile += ".root";

  //Output file
  TString outFile = storePath+"/Lumi_GeaneFiltered_";
  outFile += startEvent;
  outFile += ".root";


  // // ----  Load libraries   -------------------------------------------------
  // gROOT->Macro("$VMCWORKDIR/gconfig/rootlogon.C");
  // gSystem->Load("libSds");
  // gSystem->Load("libSdsReco");
  // gSystem->Load("libLmd");
  // gSystem->Load("libLmdReco");
  // gSystem->Load("libLmdTrk");
  // // ------------------------------------------------------------------------
    
    std::cout << "MCFile  : " << MCFile.Data()<< std::endl;
    // std::cout << "DigiFile: " << DigiFile.Data()<< std::endl;
    // std::cout << "RecoFile: " << RecoFile.Data()<< std::endl;
    // std::cout << "TCandFile: " << CandFile.Data()<< std::endl;
    std::cout << "TrackFile: " << TrkFile.Data()<< std::endl;
    std::cout << "GeaneFile: " << GeaneFile.Data()<< std::endl;
    std::cout << "OutputFile: " << outFile.Data()<< std::endl;
  // ---  Now choose concrete engines for the different tasks   -------------
  // -----------------------------------------------------------------------------------


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




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


  // -----   Reconstruction run   -------------------------------------------
  FairRunAna *fRun= new FairRunAna();
  fRun->SetInputFile(MCFile);
  // fRun->AddFriend(DigiFile);
  // fRun->AddFriend(RecoFile);
  // fRun->AddFriend(CandFile);
  fRun->AddFriend(TrkFile);
  fRun->AddFriend(GeaneFile);
  fRun->SetOutputFile(outFile);
  // ------------------------------------------------------------------------


  // -----  Parameter database   --------------------------------------------
  FairRuntimeDb* rtdb = fRun->GetRuntimeDb();
  FairParRootFileIo* parInput1 = new FairParRootFileIo(kTRUE);
  parInput1->open(parFile.Data());
  rtdb->setFirstInput(parInput1);


  // =========================================================================
  // ======                 Cleaning                                                                                                                  ======
  // =========================================================================
  Double_t fpBeam = mom;
  TString dir = "TMVAweights/";
  PndLmdSigCleanTask* lmdclean = new PndLmdSigCleanTask(fpBeam, dir);
  lmdclean->SetXY(true);
  lmdclean->SetDX(dX);
  lmdclean->SetDY(dY);
  lmdclean->SetVerbose(verboseLevel);
  fRun->AddTask(lmdclean);
  rtdb->setOutput(parInput1);
  rtdb->print();
  // =========================================================================
  
  // -----   Intialise and run   --------------------------------------------
  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;
}