[u/mrichter/AliRoot.git] / PWGLF / RESONANCES / macros / train / AddRsnInputHandlerNew.C

//
// This macro adds the specific RSN input handler, with all single particle cuts
//
void AddRsnInputHandler(Bool_t isMC, AliMultiInputEventHandler *multi)
{
   void myError  (const char *msg) {::Error  ("AddRsnInputHandler", msg);}
   void myWarning(const char *msg) {::Warning("AddRsnInputHandler", msg);}
   void myInfo   (const char *msg) {::Info   ("AddRsnInputHandler", msg);}
   
   if (!multi) {
      ::Error("AddRsnInputHandler", "Required a WELL INITIALIZED AliMultiInputEventHandler object");
      return;
   }
   
   //---------------------------------------------
   //  Define single cuts
   //---------------------------------------------

   // Track quality for ITS standalone:
   // this cut is used to select tracks of good quality, irrespective of the PID.
   // When adding status flags, the second argument tells if each considered flag
   // must be active or not in the track status, since the ITS-SA tracks need that
   // some of them are OFF (e.g.: kTPCin)
   AliRsnCutTrackQuality *cutQualityITS = new AliRsnCutTrackQuality("cutQualityITS");
   cutQualityITS->AddStatusFlag(AliESDtrack::kITSin    , kTRUE);
   cutQualityITS->AddStatusFlag(AliESDtrack::kTPCin    , kFALSE);
   cutQualityITS->AddStatusFlag(AliESDtrack::kITSrefit , kTRUE);
   cutQualityITS->AddStatusFlag(AliESDtrack::kTPCrefit , kFALSE);
   cutQualityITS->AddStatusFlag(AliESDtrack::kITSpureSA, kFALSE);
   cutQualityITS->AddStatusFlag(AliESDtrack::kITSpid   , kTRUE);
   cutQualityITS->SetPtRange(0.15, 1E+20);
   cutQualityITS->SetEtaRange(-0.8, 0.8);
   cutQualityITS->SetDCARPtFormula("0.0595+0.0182/pt^1.55");
   cutQualityITS->SetDCAZmax(2.0);
   cutQualityITS->SetSPDminNClusters(1);
   cutQualityITS->SetITSminNClusters(4);
   cutQualityITS->SetITSmaxChi2(2.0);
   cutQualityITS->SetTPCminNClusters(0);
   cutQualityITS->SetTPCmaxChi2(1E+10);
   cutQualityITS->SetRejectKinkDaughters();
      
   // Track quality for TPC+ITS:
   // works exactly like the one above, but has settings for selecting TPC+ITS tracks
   // in this case, the flags required are all necessary, so here the procedure is simpler
   AliRsnCutTrackQuality *cutQualityTPC = new AliRsnCutTrackQuality("cutQualityTPC");
   cutQualityTPC->AddStatusFlag(AliESDtrack::kTPCin   , kTRUE);
   cutQualityTPC->AddStatusFlag(AliESDtrack::kTPCrefit, kTRUE);
   cutQualityTPC->AddStatusFlag(AliESDtrack::kITSrefit, kTRUE);
   cutQualityTPC->SetPtRange(0.15, 1E+20);
   cutQualityTPC->SetEtaRange(-0.8, 0.8);
   cutQualityTPC->SetDCARPtFormula("0.0182+0.0350/pt^1.01");
   cutQualityTPC->SetDCAZmax(2.0);
   cutQualityTPC->SetSPDminNClusters(1);
   cutQualityTPC->SetITSminNClusters(0);
   cutQualityTPC->SetITSmaxChi2(1E+20);
   cutQualityTPC->SetTPCminNClusters(70);
   cutQualityTPC->SetTPCmaxChi2(4.0);
   cutQualityTPC->SetRejectKinkDaughters();
   
   // PID cuts for all needed detectors:
   // use new implementation based on AliPIDResponse and its related task
   // requires that AliAnalysisTaskPIDResponse is added otherwise it will raise errors
   AliRsnCutPIDNSigma *cutPIDITSpion   = new AliRsnCutPIDNSigma("cutPIDITSpion"  , AliPID::kPion, AliRsnCutPIDNSigma::kITS, 3.0);
   AliRsnCutPIDNSigma *cutPIDTPCpion   = new AliRsnCutPIDNSigma("cutPIDTPCpion"  , AliPID::kPion, AliRsnCutPIDNSigma::kTPC, 3.0);
   AliRsnCutPIDNSigma *cutPIDTOFpion   = new AliRsnCutPIDNSigma("cutPIDITSpion"  , AliPID::kPion, AliRsnCutPIDNSigma::kTOF, 3.0);
   AliRsnCutPIDNSigma *cutPIDITSkaon   = new AliRsnCutPIDNSigma("cutPIDITSkaon"  , AliPID::kKaon, AliRsnCutPIDNSigma::kITS, 3.0);
   AliRsnCutPIDNSigma *cutPIDTPCkaonLo = new AliRsnCutPIDNSigma("cutPIDTPCkaonLo", AliPID::kKaon, AliRsnCutPIDNSigma::kTPC, 5.0);
   AliRsnCutPIDNSigma *cutPIDTPCkaonHi = new AliRsnCutPIDNSigma("cutPIDTPCkaonHi", AliPID::kKaon, AliRsnCutPIDNSigma::kTPC, 3.0);
   AliRsnCutPIDNSigma *cutPIDTOFkaon   = new AliRsnCutPIDNSigma("cutPIDITSkaon"  , AliPID::kKaon, AliRsnCutPIDNSigma::kTOF, 3.0);
   
   // ITS PID:
   // - reject unmatched tracks
   cutPIDITSpion->SetRejectUnmatched();
   cutPIDITSkaon->SetRejectUnmatched();
   
   // TPC PID:
   // - pions               --> 3 sigma cut
   // - kaons below 350 MeV --> 5 sigma cut
   // - kaons above 350 MeV --> 3 sigma cut
   // - reject unmatched tracks
   cutPIDTPCkaonLo->SetMomentumRange(0.00, 0.35);
   cutPIDTPCkaonHi->SetMomentumRange(0.35, 1E20);
   cutPIDTPCkaonLo->SetRejectOutside();
   cutPIDTPCkaonHi->SetRejectOutside();
   cutPIDTPCpion  ->SetRejectUnmatched();
   cutPIDTPCkaonLo->SetRejectUnmatched();
   cutPIDTPCkaonHi->SetRejectUnmatched();
   
   // TOF PID:
   // must specify that unmatched tracks must be accepted
   cutPIDTOFpion->SetRejectUnmatched(kFALSE);
   cutPIDTOFkaon->SetRejectUnmatched(kFALSE);
   
   //---------------------------------------------
   //  Combine cuts
   //---------------------------------------------
   
   // make several combinations of cuts:
   // ITS and TPC standard
   //AliRsnCutSet *cutsQualityITS = new AliRsnCutSet("qualityITS", AliRsnTarget::kDaughter);
   AliRsnCutSet *cutsQualityTPC = new AliRsnCutSet("qualityTPC", AliRsnTarget::kDaughter);
   //AliRsnCutSet *cutsPionITS    = new AliRsnCutSet("pionITS"   , AliRsnTarget::kDaughter);
   AliRsnCutSet *cutsPionTPC    = new AliRsnCutSet("pionTPC"   , AliRsnTarget::kDaughter);
   //AliRsnCutSet *cutsKaonITS    = new AliRsnCutSet("kaonITS"   , AliRsnTarget::kDaughter);
   AliRsnCutSet *cutsKaonTPC    = new AliRsnCutSet("kaonTPC"   , AliRsnTarget::kDaughter);
   
   // ITS standalone: quality only
   //cutsQualityITS->AddCut(cutQualityITS);
   //cutsQualityITS->SetCutScheme(cutQualityITS->GetName());
   
   // TPC+ITS: quality only
   cutsQualityTPC->AddCut(cutQualityTPC);
   cutsQualityTPC->SetCutScheme(cutQualityTPC->GetName());
   
   // ITS standalone: quality and ITS PID
   //cutsPionITS->AddCut(cutQualityITS);
   //cutsPionITS->AddCut(cutPIDITSpion);
   //cutsPionITS->SetCutScheme(Form("%s&%s", cutQualityITS->GetName(), cutPIDITSpion->GetName()));
   //cutsKaonITS->AddCut(cutQualityITS);
   //cutsKaonITS->AddCut(cutPIDITSkaon);
   //cutsKaonITS->SetCutScheme(Form("%s&%s", cutQualityITS->GetName(), cutPIDITSkaon->GetName()));
   
   // TPC standalone: quality and TPC PID
   cutsPionTPC->AddCut(cutQualityTPC);
   cutsPionTPC->AddCut(cutPIDTPCpion);
   cutsPionTPC->AddCut(cutPIDTOFpion);
   cutsPionTPC->SetCutScheme(Form("%s&%s&%s", cutQualityTPC->GetName(), cutPIDTPCpion->GetName(), cutPIDTOFpion->GetName()));
   
   cutsKaonTPC->AddCut(cutQualityTPC);
   cutsKaonTPC->AddCut(cutPIDTPCkaonLo);
   cutsKaonTPC->AddCut(cutPIDTPCkaonHi);
   cutsKaonTPC->AddCut(cutPIDTOFkaon);
   cutsKaonTPC->SetCutScheme(Form("%s&(%s|%s)&%s", cutQualityTPC->GetName(), cutPIDTPCkaonLo->GetName(), cutPIDTPCkaonHi->GetName(), cutPIDTOFkaon->GetName()));
   
   // setup selector in the handler and add RSN input handler
   AliRsnInputHandler *rsnIH = new AliRsnInputHandler();
   AliRsnDaughterSelector *sel = rsnIH->GetSelector();
   //sel->Add(cutsQualityITS, kTRUE);
   sel->Add(cutsQualityTPC, kTRUE);
   //sel->Add(cutsPionITS, kTRUE);
   sel->Add(cutsPionTPC, kTRUE);
   //sel->Add(cutsKaonITS, kTRUE);
   sel->Add(cutsKaonTPC, kTRUE);
   //sel->Add(cutsKaonAll, kTRUE);
   sel->Init();
   multi->AddInputEventHandler(rsnIH);
}
Commit	Line	Data
00163883	1	//
	2	// This macro adds the specific RSN input handler, with all single particle cuts
	3	//
	4	void AddRsnInputHandler(Bool_t isMC, AliMultiInputEventHandler *multi)
	5	{
	6	void myError (const char *msg) {::Error ("AddRsnInputHandler", msg);}
	7	void myWarning(const char *msg) {::Warning("AddRsnInputHandler", msg);}
	8	void myInfo (const char *msg) {::Info ("AddRsnInputHandler", msg);}
	9
	10	if (!multi) {
	11	::Error("AddRsnInputHandler", "Required a WELL INITIALIZED AliMultiInputEventHandler object");
	12	return;
	13	}
	14
	15	//---------------------------------------------
	16	// Define single cuts
	17	//---------------------------------------------
	18
	19	// Track quality for ITS standalone:
	20	// this cut is used to select tracks of good quality, irrespective of the PID.
	21	// When adding status flags, the second argument tells if each considered flag
	22	// must be active or not in the track status, since the ITS-SA tracks need that
	23	// some of them are OFF (e.g.: kTPCin)
	24	AliRsnCutTrackQuality *cutQualityITS = new AliRsnCutTrackQuality("cutQualityITS");
	25	cutQualityITS->AddStatusFlag(AliESDtrack::kITSin , kTRUE);
	26	cutQualityITS->AddStatusFlag(AliESDtrack::kTPCin , kFALSE);
	27	cutQualityITS->AddStatusFlag(AliESDtrack::kITSrefit , kTRUE);
	28	cutQualityITS->AddStatusFlag(AliESDtrack::kTPCrefit , kFALSE);
	29	cutQualityITS->AddStatusFlag(AliESDtrack::kITSpureSA, kFALSE);
	30	cutQualityITS->AddStatusFlag(AliESDtrack::kITSpid , kTRUE);
	31	cutQualityITS->SetPtRange(0.15, 1E+20);
	32	cutQualityITS->SetEtaRange(-0.8, 0.8);
	33	cutQualityITS->SetDCARPtFormula("0.0595+0.0182/pt^1.55");
	34	cutQualityITS->SetDCAZmax(2.0);
	35	cutQualityITS->SetSPDminNClusters(1);
	36	cutQualityITS->SetITSminNClusters(4);
	37	cutQualityITS->SetITSmaxChi2(2.0);
	38	cutQualityITS->SetTPCminNClusters(0);
	39	cutQualityITS->SetTPCmaxChi2(1E+10);
	40	cutQualityITS->SetRejectKinkDaughters();
	41
	42	// Track quality for TPC+ITS:
	43	// works exactly like the one above, but has settings for selecting TPC+ITS tracks
	44	// in this case, the flags required are all necessary, so here the procedure is simpler
	45	AliRsnCutTrackQuality *cutQualityTPC = new AliRsnCutTrackQuality("cutQualityTPC");
	46	cutQualityTPC->AddStatusFlag(AliESDtrack::kTPCin , kTRUE);
	47	cutQualityTPC->AddStatusFlag(AliESDtrack::kTPCrefit, kTRUE);
	48	cutQualityTPC->AddStatusFlag(AliESDtrack::kITSrefit, kTRUE);
	49	cutQualityTPC->SetPtRange(0.15, 1E+20);
	50	cutQualityTPC->SetEtaRange(-0.8, 0.8);
	51	cutQualityTPC->SetDCARPtFormula("0.0182+0.0350/pt^1.01");
	52	cutQualityTPC->SetDCAZmax(2.0);
	53	cutQualityTPC->SetSPDminNClusters(1);
	54	cutQualityTPC->SetITSminNClusters(0);
	55	cutQualityTPC->SetITSmaxChi2(1E+20);
	56	cutQualityTPC->SetTPCminNClusters(70);
	57	cutQualityTPC->SetTPCmaxChi2(4.0);
	58	cutQualityTPC->SetRejectKinkDaughters();
	59
	60	// PID cuts for all needed detectors:
	61	// use new implementation based on AliPIDResponse and its related task
	62	// requires that AliAnalysisTaskPIDResponse is added otherwise it will raise errors
	63	AliRsnCutPIDNSigma *cutPIDITSpion = new AliRsnCutPIDNSigma("cutPIDITSpion" , AliPID::kPion, AliRsnCutPIDNSigma::kITS, 3.0);
	64	AliRsnCutPIDNSigma *cutPIDTPCpion = new AliRsnCutPIDNSigma("cutPIDTPCpion" , AliPID::kPion, AliRsnCutPIDNSigma::kTPC, 3.0);
65	AliRsnCutPIDNSigma *cutPIDTOFpion = new AliRsnCutPIDNSigma("cutPIDITSpion" , AliPID::kPion, AliRsnCutPIDNSigma::kTOF, 3.0);
66	AliRsnCutPIDNSigma *cutPIDITSkaon = new AliRsnCutPIDNSigma("cutPIDITSkaon" , AliPID::kKaon, AliRsnCutPIDNSigma::kITS, 3.0);
67	AliRsnCutPIDNSigma *cutPIDTPCkaonLo = new AliRsnCutPIDNSigma("cutPIDTPCkaonLo", AliPID::kKaon, AliRsnCutPIDNSigma::kTPC, 5.0);
68	AliRsnCutPIDNSigma *cutPIDTPCkaonHi = new AliRsnCutPIDNSigma("cutPIDTPCkaonHi", AliPID::kKaon, AliRsnCutPIDNSigma::kTPC, 3.0);
69	AliRsnCutPIDNSigma *cutPIDTOFkaon = new AliRsnCutPIDNSigma("cutPIDITSkaon" , AliPID::kKaon, AliRsnCutPIDNSigma::kTOF, 3.0);
70
71	// ITS PID:
72	// - reject unmatched tracks
73	cutPIDITSpion->SetRejectUnmatched();
74	cutPIDITSkaon->SetRejectUnmatched();
75
76	// TPC PID:
77	// - pions --> 3 sigma cut
78	// - kaons below 350 MeV --> 5 sigma cut
79	// - kaons above 350 MeV --> 3 sigma cut
80	// - reject unmatched tracks
81	cutPIDTPCkaonLo->SetMomentumRange(0.00, 0.35);
82	cutPIDTPCkaonHi->SetMomentumRange(0.35, 1E20);
83	cutPIDTPCkaonLo->SetRejectOutside();
84	cutPIDTPCkaonHi->SetRejectOutside();
85	cutPIDTPCpion ->SetRejectUnmatched();
86	cutPIDTPCkaonLo->SetRejectUnmatched();
87	cutPIDTPCkaonHi->SetRejectUnmatched();
88
89	// TOF PID:
90	// must specify that unmatched tracks must be accepted
91	cutPIDTOFpion->SetRejectUnmatched(kFALSE);
92	cutPIDTOFkaon->SetRejectUnmatched(kFALSE);
93
94	//---------------------------------------------
95	// Combine cuts
96	//---------------------------------------------
97
98	// make several combinations of cuts:
99	// ITS and TPC standard
100	//AliRsnCutSet *cutsQualityITS = new AliRsnCutSet("qualityITS", AliRsnTarget::kDaughter);
101	AliRsnCutSet *cutsQualityTPC = new AliRsnCutSet("qualityTPC", AliRsnTarget::kDaughter);
102	//AliRsnCutSet *cutsPionITS = new AliRsnCutSet("pionITS" , AliRsnTarget::kDaughter);
103	AliRsnCutSet *cutsPionTPC = new AliRsnCutSet("pionTPC" , AliRsnTarget::kDaughter);
104	//AliRsnCutSet *cutsKaonITS = new AliRsnCutSet("kaonITS" , AliRsnTarget::kDaughter);
105	AliRsnCutSet *cutsKaonTPC = new AliRsnCutSet("kaonTPC" , AliRsnTarget::kDaughter);
106
107	// ITS standalone: quality only
108	//cutsQualityITS->AddCut(cutQualityITS);
109	//cutsQualityITS->SetCutScheme(cutQualityITS->GetName());
110
111	// TPC+ITS: quality only
112	cutsQualityTPC->AddCut(cutQualityTPC);
113	cutsQualityTPC->SetCutScheme(cutQualityTPC->GetName());
114
115	// ITS standalone: quality and ITS PID
116	//cutsPionITS->AddCut(cutQualityITS);
117	//cutsPionITS->AddCut(cutPIDITSpion);
118	//cutsPionITS->SetCutScheme(Form("%s&%s", cutQualityITS->GetName(), cutPIDITSpion->GetName()));
119	//cutsKaonITS->AddCut(cutQualityITS);
120	//cutsKaonITS->AddCut(cutPIDITSkaon);
121	//cutsKaonITS->SetCutScheme(Form("%s&%s", cutQualityITS->GetName(), cutPIDITSkaon->GetName()));
122
123	// TPC standalone: quality and TPC PID
124	cutsPionTPC->AddCut(cutQualityTPC);
125	cutsPionTPC->AddCut(cutPIDTPCpion);
126	cutsPionTPC->AddCut(cutPIDTOFpion);
127	cutsPionTPC->SetCutScheme(Form("%s&%s&%s", cutQualityTPC->GetName(), cutPIDTPCpion->GetName(), cutPIDTOFpion->GetName()));
128
129	cutsKaonTPC->AddCut(cutQualityTPC);
130	cutsKaonTPC->AddCut(cutPIDTPCkaonLo);
131	cutsKaonTPC->AddCut(cutPIDTPCkaonHi);
132	cutsKaonTPC->AddCut(cutPIDTOFkaon);
133	cutsKaonTPC->SetCutScheme(Form("%s&(%s\|%s)&%s", cutQualityTPC->GetName(), cutPIDTPCkaonLo->GetName(), cutPIDTPCkaonHi->GetName(), cutPIDTOFkaon->GetName()));
134
135	// setup selector in the handler and add RSN input handler
136	AliRsnInputHandler *rsnIH = new AliRsnInputHandler();
137	AliRsnDaughterSelector *sel = rsnIH->GetSelector();
138	//sel->Add(cutsQualityITS, kTRUE);
139	sel->Add(cutsQualityTPC, kTRUE);
140	//sel->Add(cutsPionITS, kTRUE);
141	sel->Add(cutsPionTPC, kTRUE);
142	//sel->Add(cutsKaonITS, kTRUE);
143	sel->Add(cutsKaonTPC, kTRUE);
144	//sel->Add(cutsKaonAll, kTRUE);
145	sel->Init();
146	multi->AddInputEventHandler(rsnIH);
147	}