Commit 5b0cfdcf authored by Semen Lebedev's avatar Semen Lebedev
Browse files

Update in run_reco.C macro: Add CbmTrdSetTracksPidLike for particle ID in TRD

parent c29f97e9
......@@ -62,7 +62,7 @@
** @param sEvBuildRaw Option for raw event building
** @param setup Name of predefined geometry setup
** @param paramFile Parameter ROOT file (w/o extension .par.root)
** @param useMC Option to provide the trackfinder with MC information
** @param debugWithMC Option to provide the trackfinder with MC information
**
** This macro performs from the digis in a time-slice. It can be used
** for simulated data (result of run_digi.C) or real data after unpacking.
......@@ -88,7 +88,8 @@
**
**/
void run_reco(TString input = "", Int_t nTimeSlices = -1, Int_t firstTimeSlice = 0, TString output = "",
TString sEvBuildRaw = "", TString setup = "sis100_electron", TString paramFile = "", Bool_t useMC = false)
TString sEvBuildRaw = "", TString setup = "sis100_electron", TString paramFile = "",
Bool_t debugWithMC = false)
{
// ========================================================================
......@@ -176,7 +177,7 @@ void run_reco(TString input = "", Int_t nTimeSlices = -1, Int_t firstTimeSlice =
// ----- FairRunAna ---------------------------------------------------
FairRunAna* run = new FairRunAna();
FairFileSource* inputSource = new FairFileSource(rawFile);
if (useMC) { inputSource->AddFriend(traFile); }
if (debugWithMC) { inputSource->AddFriend(traFile); }
run->SetSource(inputSource);
run->SetOutputFile(outFile);
run->SetGenerateRunInfo(kTRUE);
......@@ -184,7 +185,7 @@ void run_reco(TString input = "", Int_t nTimeSlices = -1, Int_t firstTimeSlice =
// ------------------------------------------------------------------------
// ----- MCDataManager -----------------------------------
if (useMC) {
if (debugWithMC) {
CbmMCDataManager* mcManager = new CbmMCDataManager("MCDataManager", 0);
mcManager->AddFile(traFile);
run->AddTask(mcManager);
......@@ -254,7 +255,7 @@ void run_reco(TString input = "", Int_t nTimeSlices = -1, Int_t firstTimeSlice =
// ----------- QA for raw event builder -----------------------------------
if (eventBased && useMC) {
if (eventBased && debugWithMC) {
CbmBuildEventsQa* evBuildQA = new CbmBuildEventsQa();
run->AddTask(evBuildQA);
}
......@@ -359,7 +360,7 @@ void run_reco(TString input = "", Int_t nTimeSlices = -1, Int_t firstTimeSlice =
std::cout << "-I- " << myName << ": Added task " << psdHit->GetName() << std::endl;
}
// ------------------------------------------------------------------------
if (useMC) {
if (debugWithMC) {
CbmMatchRecoToMC* match1 = new CbmMatchRecoToMC();
run->AddTask(match1);
}
......@@ -369,7 +370,7 @@ void run_reco(TString input = "", Int_t nTimeSlices = -1, Int_t firstTimeSlice =
CbmKF* kalman = new CbmKF();
run->AddTask(kalman);
CbmL1* l1 = 0;
if (useMC) { l1 = new CbmL1("L1", 2, 3); }
if (debugWithMC) { l1 = new CbmL1("L1", 2, 3); }
else {
l1 = new CbmL1("L1", 0);
}
......@@ -431,6 +432,16 @@ void run_reco(TString input = "", Int_t nTimeSlices = -1, Int_t firstTimeSlice =
std::cout << "-I- : Added task " << finder->GetName() << std::endl;
// ----------------------------------------------------------------------
// --- Particle Id in TRD -----------------------------------------
if (useTrd) {
CbmTrdSetTracksPidLike* trdLI = new CbmTrdSetTracksPidLike("TRDLikelihood", "TRDLikelihood");
trdLI->SetUseMCInfo(kTRUE);
trdLI->SetUseMomDependence(kTRUE);
run->AddTask(trdLI);
std::cout << "-I- : Added task " << trdLI->GetName() << std::endl;
}
// ------------------------------------------------------------------------
// ----- RICH reconstruction ----------------------------------------
if (useRich) {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment