]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWGCF/FLOW/macros/AddTwoParticleResonanceFlowTask.C
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
Merge branch 'master' of https://git.cern.ch/reps/AliRoot
[u/mrichter/AliRoot.git] / PWGCF / FLOW / macros / AddTwoParticleResonanceFlowTask.C
CommitLineData
fbd0da9c 1/////////////////////////////////////////////////////////////////////////////////////////////
2//
3// AddTaskPhiFlow macro
4// Author: Redmer A. Bertens, Utrecht University, 2012
5// rbertens@cern.ch, r.a.bertens@uu.nl
6// Commented where necessary
7/////////////////////////////////////////////////////////////////////////////////////////////
8
9class AliAnalysisDataContainer;
10class AliFlowTrackCuts;
11class AliFlowTrackSimpleCuts;
12class AliFlowEventCuts;
13class AliFlowEventSimpleCuts;
14class AliAnalysisDataContainer;
15
16AliAnalysisTwoParticleResonanceFlowTask* AddTwoParticleResonanceFlowTask(Bool_t SP = 1, // select flow analysis methods
17 Bool_t SPSUB = 0,
18 Bool_t QC = 0,
19 Bool_t EP = 0,
20 Bool_t EP3sub = 0,
21 Bool_t VZERO_SP = 0, // use vzero sp method
22 Float_t centrMin = 20., // centrality selection
8da4b0c9 23 Float_t centrMax = 40.,
24 Float_t TPConlyA = 3.,
25 Float_t TPCTOFA = 5.,
26 Float_t TOFA = 3.,
27 Float_t TPConlyB = 3.,
28 Float_t TPCTOFB = 5.,
29 Float_t TOFB = 3.,
30 Float_t Bpurity = 0.9, // set to 0 to disable bayesian PID
fbd0da9c 31 TString suffixName = "UniqueID", // unique id for output objects
32 Bool_t bCentralTrigger = kTRUE, // trigger selection
33 Float_t EtaGap = 0., // eta gap for SPSUB
34 TString DCA = "pt", // dca mode (see task implementation)
35 Int_t harm = 2, // harmonic vn
36 UInt_t poi_filter = 1, // aod filterbits
37 UInt_t rp_filter = 1,
38 Float_t minMass = .71, // min mass for flow analysis
39 Float_t maxMaxx = 1.13, // max mass for flow analysis
40 Int_t speciesA = 3, // species a
41 Int_t speciesB = 2, // species b
8da4b0c9 42 Int_t chargeA = 1, // charge a
43 Int_t chargeB = -1, // charge b
fbd0da9c 44 Float_t massA = 4.93676999999999977e-01, // mass species a
45 Float_t massB = 1.39570e-01, // mass species b
8da4b0c9 46 Float_t minPtA = 0.15, // min pt for track a
47 Float_t maxPtA = 10., // max pt for track a
48 Float_t minPtB = 0.15, // min pt for track b
49 Float_t maxPtB = 10, // max pt for track b
fbd0da9c 50 Bool_t PhiMinusPsiMethod = 0, // use phi minus psi method
51 Bool_t event_mixing = kTRUE, // use event mixing
52 Bool_t highPtMode = kFALSE, // use with caution !!! disables invariant mass fit method
8da4b0c9 53 Float_t mass = 1.092, // mass for high pt analysis
54 Float_t deltaMass = 0.0003) // dM in which to look for resonance
fbd0da9c 55{
8da4b0c9 56 // removed because CINT cannot cope with more than 40 function arguments
57 Bool_t shrinkSP = kTRUE;
58 Bool_t debug = kFALSE;
fbd0da9c 59 /* some defaults that are used frequently:
60 * for phi:
61 * - minMass = 0.99
62 * - maxMass = 1.092
63 * for kstar
64 * - minMass = 0.71
65 * - maxMass = 1.13
66 * kaon mass = 4.93676999999999977e-01
67 * pion mass = 1.39570e-01
68 */
69 // some defaults that have been removed as function arguments (august 30 2012)
fbd0da9c 70 Float_t deltaDip = 0.;
71 Float_t deltaDipMaxPt = 0.;
fbd0da9c 72 Float_t vertexZ = 10.;
73 Float_t POIEtaMin = -0.8;
74 Float_t POIEtaMax = 0.8;
75 // start of macro
8da4b0c9 76 Float_t PIDconfig[] = {TPConlyA, TPCTOFA, TOFA, TPConlyB, TPCTOFB, TOFB, Bpurity};
fbd0da9c 77 // main function, create and add tasks
8da4b0c9 78 if(debug) cout << " === Adding Task TwoParticleResonanceFlow === " << endl;
fbd0da9c 79 // set up main output container's name
80 TString fileName = AliAnalysisManager::GetCommonFileName();
8da4b0c9 81 fileName += ":Reconstruction";
fbd0da9c 82 suffixName += Form("%.0f", centrMin);
83 suffixName += Form("%.0f", centrMax);
84 fileName+=suffixName;
85 if(debug) cout << " --> Reconstruction data container: " << fileName << endl;
1164b0ff 86 if(EP3sub) {
fbd0da9c 87 if(highPtMode) {
88 if(debug) cout << " --> Can't launch 3 subevent method for high pt analysis, exiting ... <--" << endl;
89 return 0x0;
90 }
91 if(debug) cout << " --> Starting 3 subevent plane method - try at your own risk !!! <-- " << endl;
92 if(!(harm!=2||harm!=3)) {
93 if(debug) cout << " --> Fatal error: can only return v2 and v3 with 3 subevent method! " << endl;
94 return 0x0;
95 }
96 }
97 if((EP3sub||PhiMinusPsiMethod)&&debug) {
98 gROOT->LoadMacro("$ALICE_ROOT/PWGCF/FLOW/macros/AddTaskVZERO.C");
99 AddTaskVZERO(0,0,0,0);
100 }
101 // get the manager and input event handler
102 AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
103 if (!mgr) {
104 if(debug) cout << " Fatal error: no analysis manager found! " << endl;
105 return 0x0;
106 }
107 if (!mgr->GetInputEventHandler()) {
108 if(debug) cout << " Fatal error: no imput event handler found!" << endl;
109 return 0x0;
110 }
111 if(EP3sub&&debug) { // don't use this on train ! this is why it's only enabled in macro debug mode
112 gROOT->LoadMacro("$ALICE_ROOT/PWGCF/FLOW/macros/AddTaskVZERO.C");
113 AddTaskVZERO(0,0,0,0);
114 }
115 // create the main task
8da4b0c9 116 AliAnalysisTwoParticleResonanceFlowTask *task = new AliAnalysisTwoParticleResonanceFlowTask(suffixName.Data());
117 if(debug) cout << " === AliAnalysisTwoParticleResonanceFlowTask === " << task << endl;
fbd0da9c 118 if(!task) {
119 if(debug) cout << " --> Unexpected error occurred: NO TASK WAS CREATED! (could be a library problem!) " << endl;
120 return 0x0;
121 }
122 if(task->UsePhiMinusPsiMethod(PhiMinusPsiMethod)) {
123 cout << " --> Using phi - Psi method <-- ... " << endl;
124 Float_t dphibins[] = {0., 0.63, 1.26, 1.89, 2.52, 3.15};
125 task->SetdPhiBins(dphibins, (Int_t)(sizeof(dphibins)/sizeof(dphibins[1])-1));
8da4b0c9 126 event_mixing = kTRUE;// only implemented for event mixing FIXME
fbd0da9c 127 }
8da4b0c9 128 task->SetupSpeciesA(speciesA, chargeA, massA, minPtA, maxPtA);
129 task->SetupSpeciesB(speciesB, chargeB, massB, minPtB, maxPtB);
fbd0da9c 130
131 if(task->SetVZEROSubEvents(EP3sub)) cout << " --> Setting up VZERO subevents method ... " << endl;
132 if(event_mixing) {
8da4b0c9 133 if(debug) cout << " --> Enabeling event mixing for reconstruction - try at your own risk !!! <-- " << endl;
fbd0da9c 134 // set vertex and mixing bins - arrays MUST have length 20!
135 Int_t c[] = {0, 2, 4, 6, 8, 10, 20, 30, 40, 50, 60, 70, 80, 90, 101, 0, 0, 0, 0, 0};
136 Int_t v[] = {-10, -8, -6, -4, -2, 0, 2, 4, 6, 8, 10, 0, 0, 0, 0, 0, 0, 0, 0, 0};
137 if(((Int_t)(sizeof(c)/sizeof(c[1]))!=20)||((Int_t)(sizeof(v)/sizeof(v[1]))!=20)) {
138 cout << " --> Fatal error: check mixing parameters ! <-- " << endl;
139 return 0x0;
140 }
141 else task->SetMixingBins(c, v);
142 }
143 // set triggers
144 if (bCentralTrigger) task->SelectCollisionCandidates(AliVEvent::kMB | AliVEvent::kCentral | AliVEvent::kSemiCentral);
145 else task->SelectCollisionCandidates(AliVEvent::kMB);
146 if(debug) cout << " --> Set trigger selection to ";
147 if(debug&&bCentralTrigger) cout << " kMB, kCentral, kSemiCentral " << endl;
148 if(debug&&(!bCentralTrigger)) cout << " kMB " << endl;
149 //set RP cuts for flow package analysis
150 AliFlowTrackCuts* cutsRP = new AliFlowTrackCuts("RFPcuts");
151 if(!cutsRP) {
152 if(debug) cout << " Fatal error: no RP cuts found, could be a library problem! " << endl;
153 return 0x0;
154 }
8da4b0c9 155 if(!VZERO_SP) {
fbd0da9c 156 AliFlowTrackCuts::trackParameterType rptype = AliFlowTrackCuts::kGlobal;
157 cutsRP->SetParamType(rptype);
158 cutsRP->SetPtRange(0.2, 5.0);
159 cutsRP->SetEtaRange(-0.8, 0.8);
160 cutsRP->SetMinNClustersTPC(70);
161 cutsRP->SetMinChi2PerClusterTPC(0.1);
162 cutsRP->SetMaxChi2PerClusterTPC(4.0);
163 cutsRP->SetRequireTPCRefit(kTRUE);
164 cutsRP->SetMaxDCAToVertexXY(0.3);
165 cutsRP->SetMaxDCAToVertexZ(0.3);
166 cutsRP->SetAcceptKinkDaughters(kFALSE);
167 cutsRP->SetMinimalTPCdedx(10.);
168 if(rp_filter < 9999 ) {
169 if(debug) cout << " > set RP filterbit " << rp_filter << endl;
170 cutsRP->SetAODfilterBit(rp_filter);
171 }
172 if(debug) cout << " --> kGlobal RP's " << cutsRP << endl;
173 }
174 if(VZERO_SP) { // use vzero sub analysis
175 cutsRP = cutsRP->GetStandardVZEROOnlyTrackCuts(); // select vzero tracks
176 SP = kFALSE; // disable other methods
177 SPSUB = kTRUE; // calculate sp_qa and sp_qb
178 QC = kFALSE;
179 EP = kFALSE;
180 EP3sub = kFALSE;
181 EtaGap = 0.; // no eta gap, full tpc poi's
182 if(debug) cout << " --> VZERO RP's " << cutsRP << endl;
183 }
184 task->SetRPCuts(cutsRP);
185 // set POI cuts for kaon selection
186 AliFlowTrackCuts* cutsPOI = new AliFlowTrackCuts("GlobalPOI");
187 if(!cutsPOI) {
188 if(debug) cout << " Fatal error: no POI cuts (could be a library problem)!" << endl;
189 return 0x0;
190 }
191 cutsPOI = cutsPOI->GetStandardGlobalTrackCuts2010();
fbd0da9c 192 cutsPOI->SetMaxDCAToVertexXY(0.3); // FIXME not implemented in aod086 aod095 see PassesDCACut() in implementation
193 cutsPOI->SetMaxDCAToVertexZ(0.3);
194 if(poi_filter < 9999 ) {
195 if(debug) cout << " > set POI filterbit " << poi_filter << endl;
196 cutsPOI->SetAODfilterBit(poi_filter);
197 }
198 if(debug) cout << " --> cutsPOI " << cutsPOI << endl;
199 task->SetPOICuts(cutsPOI);
200 //set POI cuts for aods XY Z - 3 distinct cases
201 Float_t POIDCA[] = {0., 0., 0., 0., 0.};
202 if(DCA == "none" ) { // 1 --- do nothing
203 if (debug) cout << " --> No DCA cut on POI's <-- " << endl;
204 for (Int_t i = 0; i < 5; i++) POIDCA[i] = 0.;
205 }
206 if(DCA == "fix" ) { // 2 --- use fixed values for xy z
207 if (debug) cout << " --> Fixed DCA cut on POI's <-- " << endl;
208 POIDCA[0] = -1.; POIDCA[1] = 0.3; POIDCA[2] = 0.3; POIDCA[3] = 0.; POIDCA[4] = 0.;
209 }
210 if(DCA == "pt" ) { // 3 --- use pt dependent cut
211 if (debug) cout << " --> Pt dependent DCA cut on POI's <-- " << endl;
212 POIDCA[0] = 1.; POIDCA[1] = 0.0105; POIDCA[2] = 0.0350; POIDCA[3] = 1.1; POIDCA[4] = 2.;
213 }
214 task->SetPOIDCAXYZ(POIDCA);
8da4b0c9 215 Float_t AddTaskMacroSummary[] = {(Float_t)0, (Float_t)SP, (Float_t)SPSUB, (Float_t)QC, (Float_t)EP, (Float_t)EP3sub, (Float_t)VZERO_SP, (Float_t)EtaGap, (Float_t)harm, (Float_t)highPtMode, (Float_t)deltaDip};
fbd0da9c 216 task->SetAddTaskMacroSummary(AddTaskMacroSummary);
217
218 if(highPtMode) { // high pt loop - loop will end macro
219 Float_t _pt[] = {0., 1., 2., 3., 4., 5., 6., 7., 8., 9., 10., 12., 15.};
220 task->SetPtBins(_pt, (Int_t)(sizeof(_pt)/sizeof(_pt[1]))-1);
221 // general approach: use kinematic filters which select kaon pairs with a certain mass window
222 AliFlowTrackSimpleCuts* HighPtSubEventFilterA = new AliFlowTrackSimpleCuts("HighPtSubEventFilterA");
223 HighPtSubEventFilterA->SetEtaMin(-0.8);
224 HighPtSubEventFilterA->SetEtaMax(0.0);
8da4b0c9 225 HighPtSubEventFilterA->SetMassMin(mass - deltaMass);
226 HighPtSubEventFilterA->SetMassMax(mass + deltaMass);
fbd0da9c 227 AliFlowTrackSimpleCuts* HighPtSubEventFilterB = new AliFlowTrackSimpleCuts("HighPtSubEventFilterB");
228 HighPtSubEventFilterB->SetEtaMin(0.0);
229 HighPtSubEventFilterB->SetEtaMax(+0.8);
8da4b0c9 230 HighPtSubEventFilterB->SetMassMin(mass - deltaMass);
231 HighPtSubEventFilterB->SetMassMax(mass + deltaMass);
fbd0da9c 232 AliFlowTrackSimpleCuts* HighPtGenericFilter = new AliFlowTrackSimpleCuts("HighPtGenericFilter");
233 HighPtGenericFilter->SetEtaMin(-0.8);
234 HighPtGenericFilter->SetEtaMax(+0.8);
8da4b0c9 235 HighPtGenericFilter->SetMassMin(mass - deltaMass);
236 HighPtGenericFilter->SetMassMax(mass + deltaMass);
fbd0da9c 237 if(debug) cout << " --> Created poi filters " << endl;
238 // set pair and event cuts
239 if((deltaDip>0.005)&&(deltaDipMaxPt>0.005)) task->SetMaxDeltaDipAngleAndPt(deltaDip, deltaDipMaxPt);
240 else cout << " --> Disabled Delta-Dip exclusion. <-- " << endl;
241 task->SetCandidateEtaAndPt(POIEtaMin, POIEtaMax, 0., 15.);
089d87ea 242 task->SetCentralityParameters(centrMin, centrMax, "TRK", "V0M", kTRUE, kFALSE);
fbd0da9c 243 task->SetVertexZ(vertexZ);
244 if(debug) cout << " --> Set pair cuts and event cuts" << endl;
245 // specify the PID procedure which will be used
246 task->SetPIDConfiguration(PIDconfig);
247 AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();
8da4b0c9 248 AliAnalysisDataContainer *coutHist = mgr->CreateContainer(Form("%s", TwoParticleResonanceFlow::OutputName(centrMin, centrMax,PIDconfig, suffixName, bCentralTrigger, EtaGap, POIEtaMin, POIEtaMax, deltaDip, deltaDipMaxPt, DCA, harm, vertexZ, debug).Data()), TList::Class(), AliAnalysisManager::kOutputContainer, fileName);
fbd0da9c 249 if(debug) cout << " --> Created IO containers " << cinput << ", " << coutHist << endl;
250 mgr->AddTask(task);
251 if(debug) cout << " === ADDING MAIN TASK == " << endl;
252 mgr->ConnectInput(task, 0, cinput);
253 mgr->ConnectOutput(task, 1, coutHist);
254 if(debug) cout << " --> Connected IO containers " << endl;
255 if (SP || EP || QC || SPSUB) // if flow analysis should be done after reconstruction
256 {
257 Int_t mb(999);
258 if(debug) cout << " === RECEIVED REQUEST FOR FLOW ANALYSIS === " << endl;
259 AliAnalysisDataContainer *flowEvent = mgr->CreateContainer(Form("FC%s", suffixName.Data()), AliFlowEventSimple::Class(), AliAnalysisManager::kExchangeContainer);
260 mgr->ConnectOutput(task, 2, flowEvent);
261 if(debug) cout << " --> Created IO containers " << flowEvent << endl;
262 if(debug) cout << " --> suffixName " << suffixName << endl;
263 if (QC) { // add qc tasks
8da4b0c9 264 TwoParticleResonanceFlow::AddQCmethod(Form("QCTPCMB_%d_%s", mb, suffixName.Data()), harm, flowEvent, HighPtGenericFilter, debug, 0x0, suffixName.Data());
fbd0da9c 265 if(debug) cout << " --> Hanging QC task ... " << mb << " succes! "<< endl;
266 }
267 if (SPSUB) { // add sp subevent tasks
8da4b0c9 268 TwoParticleResonanceFlow::AddSPmethod(Form("SPTPCMB_%d_%s", mb, suffixName.Data()), -0.8, -.5*EtaGap, .5*EtaGap, +0.8, "Qa", harm, flowEvent, HighPtSubEventFilterB, NULL, false, shrinkSP, debug, true, suffixName.Data());
fbd0da9c 269 if(debug) cout << " --> Hanging SP Qa task " << mb << " succes!" << endl;
8da4b0c9 270 TwoParticleResonanceFlow::AddSPmethod(Form("SPTPCMB_%d_%s", mb, suffixName.Data()), -0.8, -.5*EtaGap, .5*EtaGap, +0.8, "Qb", harm, flowEvent, HighPtSubEventFilterA, NULL, false, shrinkSP, debug, true, suffixName.Data());
fbd0da9c 271 if(debug) cout << " --> Hanging SP Qb task ..." << mb << " succes!"<< endl;
272 }
273 if (SP) { // add sp tasks
8da4b0c9 274 TwoParticleResonanceFlow::AddSPmethod(Form("SPTPCMB_%d_%s", mb, suffixName.Data()), -0.8, -0.0, +0.0, +0.8, "QaQb", harm, flowEvent, HighPtGenericFilter, NULL, false, shrinkSP, debug, 0x0, suffixName.Data());
fbd0da9c 275 if(debug) cout << " --> Hanging SP task ... " << mb << " succes!" << endl;
276 }
277 if (EP) { // add ep tasks
8da4b0c9 278 TwoParticleResonanceFlow::AddSPmethod(Form("EPTPCMBSP_%d_%s", mb, suffixName.Data()), -0.8, -0.0, +0.0, +0.8, "QaQb", harm, flowEvent, HighPtGenericFilter, NULL, true, shrinkSP, debug, 0x0, suffixName.Data());
fbd0da9c 279 if(debug) cout << " --> Hanging EP task ... " << mb << " succes!" << endl;
280 }
281 }
282 // print summary to screen
8da4b0c9 283 cout << endl << endl << " ==== AddTwoParticleResonanceFlowTask launched ===== " << endl;
fbd0da9c 284 cout << " ************ Configured PID routine ************ " << endl;
285 cout << " 0 < " << PIDconfig[1] << " p_t, ITS || TPC with s < " << PIDconfig[0] << endl;
286 if(PIDconfig[2] < 0.) cout << " --> TPC control disabled " << endl;
287 if(PIDconfig[2] > 0.) cout << " --> TPC control enabled " << endl;
288 cout << " " << PIDconfig[1] << " < " << PIDconfig[4] << " p_t, TPC || ITS with s < " << PIDconfig[3] << endl;
289 if(PIDconfig[5] < 0.) cout << " --> ITS control disabled " << endl;
290 if(PIDconfig[5] > 0.) cout << " --> ITS control enabled " << endl;
291 cout << " " << PIDconfig[4] << " < 7 p_t, TPC / TOF Bayesian with p < " << PIDconfig[6] << endl;
292 cout << " ************ Configured DCA setup ************** " << endl;
293 cout << " DCA type: " << DCA;
294 if (DCA == "") cout << "default";
295 cout << endl << " ************* Task statisti:q!cs ****************** " << endl;
8da4b0c9 296 cout << " -> Launched resonance reconstruction " << endl;
fbd0da9c 297 if(SP) cout << " --> Launched QaQb SP filters and corresponding SP task " << endl;
298 if(EP) cout << " --> Launched QaQb QC filters and corresponding EP task " << endl;
299 if(SPSUB) cout << " --> Launched Qa&&Qb SP filters and corresponding SP tasks " << endl;
300 if(QC) cout << " --> Launched QaQb QC filters and corresponding QC task " << endl;
301 if(EP3sub) cout << " --> Launched VZERO subevent analysis alongside reconstruction - USE WITH CAUTION!" << endl;
302 cout << " ************************************************ " << endl;
303 TString condit = "";
304 (task->SetQA(kFALSE)) ? condit+= " --> Enabled QA plots <-- " : condit+= " --> Disabled QA plots <-- ";
305 (task->SetIsMC(kFALSE)) ? condit+= " --> MC mode <-- " : condit+= " --> DATA mode <-- ";
306 (task->UseEventMixing(event_mixing)) ? condit+= " --> Using EVENT MIXING <--" : condit+= "--> Combinatorial background <--";
307 cout << condit << endl;
308 cout << " --> Now go for a coffee! <-- " << endl;
309 cout << " ************************************************ " << endl;
310 return task;
311 } // end of high pt loop - high-pt task is ready to go at this point
8da4b0c9 312 Float_t _pt[] = {0.0, 0.6, 1.2, 1.8, 2.4, 3.0, 3.5, 4.0, 4.5, 5.0, 5.5, 6.0, 6.5, 7.0};
fbd0da9c 313 task->SetPtBins(_pt, (Int_t)(sizeof(_pt)/sizeof(_pt[1]))-1);
314 // POI filter cuts, will filter invm mass bands and subevents
315 AliFlowTrackSimpleCuts* POIfilterQC[30];
316 AliFlowTrackSimpleCuts* POIfilterSP[30][2];
317 Double_t flowBands[2][30];
318 Double_t _inc = (maxMaxx-minMass)/30.;
319 for (Int_t mb = 0; mb < 30; mb++) {
320 flowBands[0][mb] = minMass + mb * _inc;
321 flowBands[1][mb] = minMass + (mb + 1) * _inc;
322 POIfilterSP[mb][0] = new AliFlowTrackSimpleCuts(Form("FilterPOISP_MB%d_ETANEG", mb));
323 if(!POIfilterSP[mb][0]) {
324 if(debug) cout << " FAILED to initialize POI filters, could be a library problem!" << endl;
325 return 0x0;
326 }
327 POIfilterSP[mb][0]->SetEtaMin(-0.8);
328 POIfilterSP[mb][0]->SetEtaMax(0.0);
329 POIfilterSP[mb][0]->SetMassMin(flowBands[0][mb]);
330 POIfilterSP[mb][0]->SetMassMax(flowBands[1][mb]);
331 POIfilterSP[mb][1] = new AliFlowTrackSimpleCuts(Form("FilterPOISP_MB%d_ETAPOS", mb));
332 POIfilterSP[mb][1]->SetEtaMin(0.0);
333 POIfilterSP[mb][1]->SetEtaMax(+0.8);
334 POIfilterSP[mb][1]->SetMassMin(flowBands[0][mb]);
335 POIfilterSP[mb][1]->SetMassMax(flowBands[1][mb]);
336 POIfilterQC[mb] = new AliFlowTrackSimpleCuts(Form("FilterPOIQC_MB%d", mb));
337 POIfilterQC[mb]->SetEtaMin(-0.8);
338 POIfilterQC[mb]->SetEtaMax(+0.8);
339 POIfilterQC[mb]->SetMassMin(flowBands[0][mb]);
340 POIfilterQC[mb]->SetMassMax(flowBands[1][mb]);
341 }
342 if(debug) cout << " --> Created poi filters " << endl;
343 task->SetCommonConstants(30, flowBands[0][0], flowBands[1][29]);
344 if(debug) cout << " --> Set common constants " << endl;
345 // set pair and event cuts
346 if((deltaDip>0.005)&&(deltaDipMaxPt>0.005)) task->SetMaxDeltaDipAngleAndPt(deltaDip, deltaDipMaxPt);
347 else cout << " --> Disabled Delta-Dip exclusion. <-- " << endl;
348 task->SetCandidateEtaAndPt(POIEtaMin, POIEtaMax, 0., 10.);
089d87ea 349 task->SetCentralityParameters(centrMin, centrMax, "TRK", "V0M", kTRUE, kFALSE);
fbd0da9c 350 task->SetVertexZ(vertexZ);
351 if(debug) cout << " --> Set pair cuts and event cuts" << endl;
352 // set the kaon cuts, and specify the PID procedure which will be used
353 task->SetPIDConfiguration(PIDconfig);
354 if(debug) {
355 cout << " ************ Configured PID routine ************ " << endl;
356 cout << " 0 < " << PIDconfig[1] << " p_t, ITS || TPC with s < " << PIDconfig[0] << endl;
357 if(PIDconfig[2] < 0.) cout << " --> TPC control disabled " << endl;
358 if(PIDconfig[2] > 0.) cout << " --> TPC control enabled " << endl;
359 cout << " " << PIDconfig[1] << " < " << PIDconfig[4] << " p_t, TPC || ITS with s < " << PIDconfig[3] << endl;
360 if(PIDconfig[5] < 0.) cout << " --> ITS control disabled " << endl;
361 if(PIDconfig[5] > 0.) cout << " --> ITS control enabled " << endl;
362 cout << " " << PIDconfig[4] << " < 7 p_t, TPC / TOF Bayesian with p < " << PIDconfig[6] << endl;
363 cout << " ************************************************ " << endl;
364 }
fbd0da9c 365 // create and configure IO containers
366 AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();
8da4b0c9 367 AliAnalysisDataContainer *coutHist = mgr->CreateContainer(Form("%s", TwoParticleResonanceFlow::OutputName(centrMin, centrMax,PIDconfig, suffixName, bCentralTrigger, EtaGap, POIEtaMin, POIEtaMax, deltaDip, deltaDipMaxPt, DCA, harm, vertexZ, debug).Data()), TList::Class(), AliAnalysisManager::kOutputContainer, fileName);
fbd0da9c 368 if(debug) cout << " --> Created IO containers " << cinput << ", " << coutHist << endl;
369 mgr->AddTask(task);
370 if(debug) cout << " === ADDING MAIN TASK == " << endl;
371 mgr->ConnectInput(task, 0, cinput);
372 mgr->ConnectOutput(task, 1, coutHist);
373 if(debug) cout << " --> Connected IO containers " << endl;
374 if (SP || EP || QC || SPSUB) // if flow analysis should be done after reconstruction
375 {
376 if(debug) cout << " === RECEIVED REQUEST FOR FLOW ANALYSIS === " << endl;
377 AliAnalysisDataContainer *flowEvent = mgr->CreateContainer(Form("FC%s", suffixName.Data()), AliFlowEventSimple::Class(), AliAnalysisManager::kExchangeContainer);
378 mgr->ConnectOutput(task, 2, flowEvent);
379 if(debug) cout << " --> Created IO containers " << flowEvent << endl;
380 if(debug) cout << " --> suffixName " << suffixName << endl;
381 for (int mb = 0; mb != 30; ++mb) {
382 if (QC) { // add qc tasks
8da4b0c9 383 TwoParticleResonanceFlow::AddQCmethod(Form("QCTPCMB_%d_%s", mb, suffixName.Data()), harm, flowEvent, POIfilterQC[mb], debug, 0x0, suffixName.Data());
fbd0da9c 384 if(debug) cout << " --> Hanging QC task ... " << mb << " succes! "<< endl;
385 }
386 if (SPSUB) { // add sp subevent tasks
8da4b0c9 387 TwoParticleResonanceFlow::AddSPmethod(Form("SPTPCMB_%d_%s", mb, suffixName.Data()), -0.8, -.5*EtaGap, .5*EtaGap, +0.8, "Qa", harm, flowEvent, POIfilterSP[mb][1], NULL, false, shrinkSP, debug, true, suffixName.Data(), VZERO_SP);
fbd0da9c 388 if(debug) cout << " --> Hanging SP Qa task " << mb << " succes!" << endl;
8da4b0c9 389 TwoParticleResonanceFlow::AddSPmethod(Form("SPTPCMB_%d_%s", mb, suffixName.Data()), -0.8, -.5*EtaGap, .5*EtaGap, +0.8, "Qb", harm, flowEvent, POIfilterSP[mb][0], NULL, false, shrinkSP, debug, true, suffixName.Data(), VZERO_SP);
fbd0da9c 390 if(debug) cout << " --> Hanging SP Qb task ..." << mb << " succes!"<< endl;
391 }
392 if (SP) { // add sp tasks
8da4b0c9 393 TwoParticleResonanceFlow::AddSPmethod(Form("SPTPCMB_%d_%s", mb, suffixName.Data()), -0.8, -0.0, +0.0, +0.8, "QaQb", harm, flowEvent, POIfilterQC[mb], NULL, false, shrinkSP, debug, 0x0, suffixName.Data());
fbd0da9c 394 if(debug) cout << " --> Hanging SP task ... " << mb << " succes!" << endl;
395 }
396 if (EP) { // add ep tasks
8da4b0c9 397 TwoParticleResonanceFlow::AddSPmethod(Form("EPTPCMB_%d_%s", mb, suffixName.Data()), -0.8, -0.0, +0.0, +0.8, "QaQb", harm, flowEvent, POIfilterQC[mb], NULL, true, shrinkSP, debug, 0x0, suffixName.Data());
fbd0da9c 398 if(debug) cout << " --> Hanging EP task ... " << mb << " succes!" << endl;
399 }
400 }
401 }
402 // print summary to screen
8da4b0c9 403 cout << endl << endl << " ==== AddTwoParticleResonanceFlowTask launched ===== " << endl;
fbd0da9c 404 cout << " ************ Configured PID routine ************ " << endl;
405 cout << " 0 < " << PIDconfig[1] << " p_t, ITS || TPC with s < " << PIDconfig[0] << endl;
406 if(PIDconfig[2] < 0.) cout << " --> TPC control disabled " << endl;
407 if(PIDconfig[2] > 0.) cout << " --> TPC control enabled " << endl;
408 cout << " " << PIDconfig[1] << " < " << PIDconfig[4] << " p_t, TPC || ITS with s < " << PIDconfig[3] << endl;
409 if(PIDconfig[5] < 0.) cout << " --> ITS control disabled " << endl;
410 if(PIDconfig[5] > 0.) cout << " --> ITS control enabled " << endl;
411 cout << " " << PIDconfig[4] << " < 7 p_t, TPC / TOF Bayesian with p < " << PIDconfig[6] << endl;
412 cout << " ************ Configured DCA setup ************** " << endl;
413 cout << " DCA type: " << DCA;
414 if (DCA == "") cout << "default";
415 cout << endl << " ************* Task statisti:q!cs ****************** " << endl;
8da4b0c9 416 cout << " -> Launched resonance reconstruction " << endl;
fbd0da9c 417 if(SP) cout << " --> Launched 30 QaQb SP filters and corresponding 30 SP tasks " << endl;
418 if(EP) cout << " --> Launched 30 QaQb QC filters and corresponding 30 EP tasks " << endl;
419 if(SPSUB) cout << " --> Launched 30+30 Qa&&Qb SP filters and corresponding 30+30 SP tasks " << endl;
420 if(QC) cout << " --> Launched 30 QaQb QC filters and corresponding 30 QC tasks " << endl;
421 if(EP3sub) cout << " --> Launched VZERO subevent analysis alongside reconstruction - USE WITH CAUTION!" << endl;
422 cout << " ************************************************ " << endl;
423 TString condit = "";
424 (task->SetQA(kFALSE)) ? condit+= " --> Enabled QA plots <-- " : condit+= " --> Disabled QA plots <-- ";
425 (task->SetIsMC(kFALSE)) ? condit+= " --> MC mode <-- " : condit+= " --> DATA mode <-- ";
426 (task->UseEventMixing(event_mixing)) ? condit+= " --> Using EVENT MIXING <--" : condit+= "--> Combinatorial background <--";
427 cout << condit << endl;
428 cout << " --> Now go for a coffee! <-- " << endl;
429 cout << " ************************************************ " << endl;
430 return task;
431}
8da4b0c9 432namespace TwoParticleResonanceFlow {
fbd0da9c 433//_____________________________________________________________________________
8da4b0c9 434 void AddSPmethod(char *name, double minEtaA, double maxEtaA, double minEtaB, double maxEtaB, char *Qvector, int harmonic, AliAnalysisDataContainer *flowEvent, AliFlowTrackSimpleCuts *cutsPOI = NULL, AliFlowTrackSimpleCuts *cutsRFP = NULL, bool bEP, bool shrink = false, bool debug, bool etagap = false, char *suffixName, Bool_t VZERO_SP = kFALSE)
435 {
436 // add sp task and invm filter tasks
437 if(debug) (bEP) ? cout << " ****** Reveived request for EP task ****** " << endl : cout << " ******* Switching to SP task ******* " << endl;
438 TString fileName = AliAnalysisManager::GetCommonFileName();
439 (bEP) ? fileName+=":EP" : fileName+=":SP";
440 fileName+=suffixName;
441 if(etagap) {
442 fileName+="_SUBEVENTS";
443 if(debug) cout << " --> Setting up subevent analysis <-- " << endl;
444 }
445 if(debug) cout << " --> fileName " << fileName << endl;
446 TString myNameSP;
447 (bEP) ? myNameSP = Form("%sSPasEPv%d%s", name, harmonic, Qvector): myNameSP = Form("%sSPv%d%s", name, harmonic, Qvector);
448 if(debug) cout << " Task and filter name: " << myNameSP << endl;
449 AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
450 AliAnalysisDataContainer *flowEvent2 = mgr->CreateContainer(Form("Filter_%s", myNameSP.Data()), AliFlowEventSimple::Class(), AliAnalysisManager::kExchangeContainer);
451 AliAnalysisTaskFilterFE *tskFilter = new AliAnalysisTaskFilterFE(Form("TaskFilter_%s", myNameSP.Data()), cutsRFP, cutsPOI);
452 tskFilter->SetSubeventEtaRange(minEtaA, maxEtaA, minEtaB, maxEtaB);
453 if(VZERO_SP) tskFilter->SetSubeventEtaRange(-10, 0, 0, 10);
454 mgr->AddTask(tskFilter);
455 mgr->ConnectInput(tskFilter, 0, flowEvent);
456 mgr->ConnectOutput(tskFilter, 1, flowEvent2);
457 AliAnalysisDataContainer *outSP = mgr->CreateContainer(myNameSP.Data(), TList::Class(), AliAnalysisManager::kOutputContainer, fileName);
458 AliAnalysisTaskScalarProduct *tskSP = new AliAnalysisTaskScalarProduct(Form("TaskScalarProduct_%s", myNameSP.Data()), kFALSE);
459 tskSP->SetApplyCorrectionForNUA(kTRUE);
460 tskSP->SetHarmonic(harmonic);
461 tskSP->SetTotalQvector(Qvector);
462 if (bEP) tskSP->SetBehaveAsEP();
463 if (shrink) tskSP->SetBookOnlyBasicCCH(kTRUE);
464 mgr->AddTask(tskSP);
465 mgr->ConnectInput(tskSP, 0, flowEvent2);
466 mgr->ConnectOutput(tskSP, 1, outSP);
467 }
fbd0da9c 468//_____________________________________________________________________________
8da4b0c9 469 void AddQCmethod(char *name, int harmonic, AliAnalysisDataContainer *flowEvent, AliFlowTrackSimpleCuts *cutsPOI = NULL, Bool_t debug, AliFlowTrackSimpleCuts *cutsRFP = NULL, char *suffixName)
470 {
471 // add qc task and invm filter tasks
472 if(debug) cout << " ****** Received request for QC v" << harmonic << " task " << name << ", POI " << cutsPOI << ", IO ****** " << flowEvent << endl;
473 TString fileName = AliAnalysisManager::GetCommonFileName();
474 fileName+=":QC";
475 fileName+=suffixName;
476 if(debug) cout << " --> Common filename: " << fileName << endl;
477 TString myName = Form("%s", name);
478 if(debug) cout << " --> myName: " << myName << endl;
479 AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
480 AliAnalysisDataContainer *flowEvent2 = mgr->CreateContainer(Form("Filter_%s", myName.Data()), AliFlowEventSimple::Class(),AliAnalysisManager::kExchangeContainer);
481 AliAnalysisTaskFilterFE *tskFilter = new AliAnalysisTaskFilterFE(Form("TaskFilter_%s", myName.Data()), cutsRFP, cutsPOI);
482 mgr->AddTask(tskFilter);
483 mgr->ConnectInput(tskFilter, 0, flowEvent);
484 mgr->ConnectOutput(tskFilter, 1, flowEvent2);
485 AliAnalysisDataContainer *outQC = mgr->CreateContainer(myName.Data(), TList::Class(), AliAnalysisManager::kOutputContainer, fileName);
486 AliAnalysisTaskQCumulants *tskQC = new AliAnalysisTaskQCumulants(Form("TaskQCumulants_%s", myName.Data()), kFALSE);
487 tskQC->SetApplyCorrectionForNUA(kTRUE);
488 tskQC->SetHarmonic(harmonic);
489 tskQC->SetBookOnlyBasicCCH(kTRUE);
490 mgr->AddTask(tskQC);
491 mgr->ConnectInput(tskQC, 0, flowEvent2);
492 mgr->ConnectOutput(tskQC, 1, outQC);
493 }
fbd0da9c 494//_____________________________________________________________________________
8da4b0c9 495 TString OutputName( Float_t centrMin,
496 Float_t centrMax,
497 Float_t PIDconfig[7],
498 TString suffixName,
499 Bool_t bCentralTrigger,
500 Float_t EtaGap,
501 Float_t POIEtaMin,
502 Float_t POIEtaMax,
503 Float_t deltaDip,
504 Float_t deltaDipMaxPt,
505 TString DCA,
506 Int_t harm,
507 Float_t vertexZ,
508 Bool_t debug)
509 {
510 // generate output name
511 TString centralityName = "";
512 centralityName += suffixName;
513 centralityName += "_DCA";
514 centralityName += DCA;
515 centralityName += Form("_vZ%.f", vertexZ);
516 centralityName += "_";
517 for(Int_t i = 0; i < 7; i++) centralityName += Form("%.1f_", PIDconfig[i]);
518 centralityName += "_POIEta";
519 centralityName += Form("%.1f", POIEtaMin);
520 centralityName += Form("%.1f", POIEtaMax);
521 centralityName += "_gap";
522 centralityName += Form("%.1f", -0.5*EtaGap);
523 centralityName += Form("%.1f", 0.5*EtaGap);
524 centralityName += "_";
525 centralityName += Form("dDip%.2f", deltaDip);
fbd0da9c 526 centralityName += "-";
8da4b0c9 527 centralityName += Form("dDipPt%.2f", deltaDipMaxPt);
528 if (bCentralTrigger) {
529 centralityName += "-";
530 centralityName += "kMBkCkSC";
531 }
532 if(debug) cout << " --> centralityName " << centralityName << endl;
533 return centralityName;
534 }
fbd0da9c 535//_____________________________________________________________________________
8da4b0c9 536} // end of namespace TwoParticleResonanceFlow