Updated macros for lego train (M. Vala)

[u/mrichter/AliRoot.git] / PWGLF / RESONANCES / macros / lego_train / AddRsnDaughterCutsRho.C

#ifndef __CINT__\r
#include <AliRsnCutPIDNSigma.h>\r
#include <PWGLF/RESONANCES/AliRsnCutTrackQuality.h>\r
#endif\r
Int_t AddRsnDaughterCutsRho(AliPID::EParticleType type1,AliPID::EParticleType type2,TString opt,Bool_t isRsnMini=kFALSE,AliRsnInputHandler *rsnIH=0,AliAnalysisTaskSE *task=0)\r
{\r
\r
   if (!rsnIH) return 0;\r
\r
   Bool_t valid = kTRUE;\r
   Int_t isPP = AliAnalysisManager::GetGlobalInt("rsnIsPP",valid);\r
\r
   Bool_t usePPCut = kFALSE;\r
\r
   if (isPP && (opt.Contains("usePP"))) usePPCut = kTRUE;\r
\r
\r
   // === USER HAS TO SET CORRECT NUMBER OF CUTS SETS =====\r
   Int_t numberOfCuts = 1;\r
\r
   //---------------------------------------------\r
   //  Define single cuts\r
   //---------------------------------------------\r
\r
   Printf("AliRsnCutPIDNSigma Option : %s",opt.Data());\r
\r
   Double_t nSigmaTPC=3.0;\r
   Double_t nSigmaTOF=3.0;\r
   Double_t etaRange=0.8;\r
\r
   if (opt.Contains("TPCsigma1")) nSigmaTPC = 1.0;\r
   if (opt.Contains("TPCsigma2")) nSigmaTPC = 2.0;\r
   if (opt.Contains("TPCsigma3")) nSigmaTPC = 3.0;\r
\r
   if (opt.Contains("TOFsigma1")) nSigmaTOF = 1.0;\r
   if (opt.Contains("TOFsigma2")) nSigmaTOF = 2.0;\r
   if (opt.Contains("TOFsigma3")) nSigmaTOF = 3.0;\r
\r
   Bool_t usePDG=kFALSE;\r
   if (opt.Contains("pdg")) {\r
      Printf("Using PDG");\r
      usePDG = kTRUE;\r
   }\r
\r
   Bool_t useEta = kFALSE;\r
   if (opt.Contains("eta")) {\r
      Printf("Using ETA range (%.2f,%.2f)",-etaRange,etaRange);\r
      useEta = kTRUE;\r
   }\r
\r
   Bool_t usetof = kFALSE;\r
   if(opt.Contains("tof")) {\r
      Printf("Using tof PID range (%.2f,%.2f)",0.0,1E+20);\r
      usetof = kTRUE;\r
   }\r
\r
   Bool_t usetpc = kFALSE;\r
   if(opt.Contains("tpc")) {\r
      Printf("Using tpc PID range (%.2f,%.2f)",0.0,1E+20);\r
      usetpc = kTRUE;\r
   }\r
\r
//---------------------------------------------\r
//  Combine cuts\r
//---------------------------------------------\r
\r
   TString cutname = "pionRho";\r
   if (!opt.IsNull()) cutname += Form("_%s",opt.Data());\r
   AliRsnCutSet *cuts = new AliRsnCutSet(cutname.Data(), AliRsnTarget::kDaughter);\r
\r
   TString scheme="";\r
\r
   AliRsnCutTrackQuality *qualityCut = new AliRsnCutTrackQuality("cutQuatity");\r
   //qualityCut->SetDefaults2010();\r
   //qualityCut->SetDCAZmax(0.3);\r
   //qualityCut->SetDCARmax(0.05);\r
\r
   qualityCut->SetDCAZmax(0.2);\r
   qualityCut->SetDCARmax(0.02);\r
   qualityCut->AddStatusFlag(AliESDtrack::kTPCin   , kTRUE);\r
   qualityCut->AddStatusFlag(AliESDtrack::kTPCrefit, kTRUE);\r
   qualityCut->AddStatusFlag(AliESDtrack::kITSrefit, kTRUE);\r
\r
\r
   qualityCut->SetPtRange(0.15, 1E+20);\r
   qualityCut->SetEtaRange(-0.8, 0.8);\r
   qualityCut->SetSPDminNClusters(0);\r
   qualityCut->SetITSminNClusters(0);\r
   qualityCut->SetITSmaxChi2(1E+20);\r
   qualityCut->SetTPCminNClusters(80);\r
   qualityCut->SetTPCmaxChi2(4.0);\r
   qualityCut->SetRejectKinkDaughters();\r
   cuts->AddCut(qualityCut);\r
   if (!scheme.IsNull()) scheme += "&";\r
   scheme += qualityCut->GetName();\r
\r
   if (usetpc) {\r
      AliRsnCutPIDNSigma *cutPiTPC = new AliRsnCutPIDNSigma("cutPIDNSigmaTPC",AliPID::kPion,AliRsnCutPIDNSigma::kTPC);\r
      cutPiTPC->SinglePIDRange(nSigmaTPC);\r
      //cutPiTPC->AddPIDRange(nSigmaTPC,0.0,0.7);\r
      cuts->AddCut(cutPiTPC);\r
      if (!scheme.IsNull()) scheme += "&";\r
      scheme += cutPiTPC->GetName();\r
   }\r
\r
   if(usetof) {\r
      AliRsnCutPIDNSigma *cutPiTOF = new AliRsnCutPIDNSigma("cutPIDNSigmaTOF",AliPID::kPion,AliRsnCutPIDNSigma::kTOF);\r
      cutPiTOF->SinglePIDRange(nSigmaTOF);\r
      //cutPiTOF->AddPIDRange(nSigmaTOF,0.7,1e20);\r
      cuts->AddCut(cutPiTOF);\r
      if (!scheme.IsNull()) scheme += "&";\r
      scheme += cutPiTOF->GetName();\r
   }\r
\r
   if (useEta) {\r
      Printf("Adding ETA ...");\r
      AliRsnValueDaughter *valEta = new AliRsnValueDaughter(Form("val%sETA%s",AliPID::ParticleName(type1),opt.Data()),AliRsnValueDaughter::kEta);\r
      AliRsnCutValue *cutEta = new AliRsnCutValue(Form("cut%sETA%s",AliPID::ParticleName(type1),opt.Data()),-etaRange,etaRange);\r
      cutEta->SetTargetType(AliRsnTarget::kDaughter);\r
      cutEta->SetValueObj(valEta);\r
      cuts->AddCut(cutEta);\r
      if (!scheme.IsNull()) scheme += "&";\r
      scheme += cutEta->GetName();\r
   }\r
   if (usePDG) {\r
      Printf("Adding PDG ...");\r
      AliRsnCutPID *cutPDG = new AliRsnCutPID(Form("cut%sPDG%s",AliPID::ParticleName(type1),opt.Data()),type1,0.0,kTRUE);\r
      cuts->AddCut(cutPDG);\r
      if (!scheme.IsNull()) scheme += "&";\r
      scheme += cutPDG->GetName();\r
   }\r
\r
   Printf ("CUT Scheme is '%s'",scheme.Data());\r
   cuts->SetCutScheme(scheme.Data());\r
\r
   if (opt.Contains("mon")) {\r
      AddMonitorOutput(cuts->GetMonitorOutput(),opt);\r
   }\r
   if (isRsnMini) {\r
      AliRsnMiniAnalysisTask *taskRsnMini = dynamic_cast<AliRsnMiniAnalysisTask *>(task);\r
      if (taskRsnMini) {\r
         taskRsnMini->AddTrackCuts(cuts);\r
      }\r
   } else {\r
      AliRsnDaughterSelector *sel = rsnIH->GetSelector();\r
//       sel->SetLabelCheck(kFALSE);\r
      sel->Add(cuts, kTRUE);\r
   }\r
   return numberOfCuts;\r
\r
}\r
\r