Updated macros for lego train (mvala) + added scripts folder

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

#ifndef __CINT__\r
#include <AliRsnCutPIDNSigma.h>\r
#include <PWGLF/RESONANCES/AliRsnCutTrackQuality.h>\r
#endif\r
Int_t AddRsnDaughterCutsRhoNsigma(AliPID::EParticleType type1,AliPID::EParticleType type2,TString opt,AliRsnInputHandler *rsnIH=0,AliAnalysisTaskSE *task=0)\r
{\r
\r
   if (!rsnIH) return 0;\r
\r
   Bool_t valid;\r
   Int_t isRsnMini = AliRsnTrainManager::GetGlobalInt("IsRsnMini",valid);\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("AddRsnDaughterCutsRho Option : %s",opt.Data());\r
\r
   Double_t nSigmaTPC=3.0;\r
   Double_t nSigmaTOF=3.0;\r
   Double_t etaRange=0.8;\r
   Bool_t useTPC_Pi=kFALSE;\r
   Bool_t useTOF_Pi=kFALSE;\r
\r
   if (opt.Contains("qualityonly")) {\r
      useTPC_Pi=kFALSE;\r
      useTOF_Pi=kFALSE;\r
   } else if (!opt.Contains("nsig")) {\r
      useTPC_Pi=kTRUE;\r
      useTOF_Pi=kTRUE;\r
   }\r
\r
   if (opt.Contains("PiTPCnsig"))  useTPC_Pi=kTRUE;\r
   if (opt.Contains("PiTOFnsig"))  useTOF_Pi=kTRUE;\r
\r
   if (opt.Contains("PiTPCnsig10")) nSigmaTPC = 1.0;\r
   if (opt.Contains("PiTPCnsig15")) nSigmaTPC = 1.5;\r
   if (opt.Contains("PiTPCnsig20")) nSigmaTPC = 2.0;\r
   if (opt.Contains("PiTPCnsig25")) nSigmaTPC = 2.5;\r
   if (opt.Contains("PiTPCnsig30")) nSigmaTPC = 3.0;\r
\r
   if (opt.Contains("PiTOFnsig10")) nSigmaTOF = 1.0;\r
   if (opt.Contains("PiTOFnsig15")) nSigmaTOF = 1.5;\r
   if (opt.Contains("PiTOFnsig20")) nSigmaTOF = 2.0;\r
   if (opt.Contains("PiTOFnsig25")) nSigmaTOF = 2.5;\r
   if (opt.Contains("PiTOFnsig30")) 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
//---------------------------------------------\r
//  Combine cuts\r
//---------------------------------------------\r
\r
   TString cutname = "Pi_Rho";\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("cutQuatityPi");\r
   //qualityCut->SetDefaults2010();\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
	qualityCut->SetPtRange(0.2, 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(70);\r
	qualityCut->SetTPCmaxChi2(4.0);\r
	qualityCut->SetRejectKinkDaughters();\r
\r
   cuts->AddCut(qualityCut);\r
   if (!scheme.IsNull()) scheme += "&";\r
   scheme += qualityCut->GetName();\r
\r
   if (useTPC_Pi) {\r
      AliRsnCutPIDNSigma *cutPiTPC = new AliRsnCutPIDNSigma("cutPIDNSigmaTPCPi",AliPID::kPion,AliRsnCutPIDNSigma::kTPC);\r
      cutPiTPC->SinglePIDRange(nSigmaTPC);\r
      cuts->AddCut(cutPiTPC);\r
      if (!scheme.IsNull()) scheme += "&";\r
      scheme += cutPiTPC->GetName();\r
   }\r
   if (useTOF_Pi) {\r
      AliRsnCutPIDNSigma *cutPiTOF = new AliRsnCutPIDNSigma("cutPIDNSigmaTOFPi",AliPID::kPion,AliRsnCutPIDNSigma::kTOF);\r
      cutPiTOF->SinglePIDRange(nSigmaTOF);\r
      cuts->AddCut(cutPiTOF);\r
      if (!scheme.IsNull()) scheme += "&";\r
      scheme += cutPiTOF->GetName();\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