]> git.uio.no Git - u/mrichter/AliRoot.git/blob - PWGCF/FLOW/macros/AddTaskFlowCentralityPIDSP.C
from nagmeh: addtask for pid config task and updates to pid config task
[u/mrichter/AliRoot.git] / PWGCF / FLOW / macros / AddTaskFlowCentralityPIDSP.C
1 /////////////////////////////////////////////////////////////////////////////////////////////
2 //
3 // AddTask* macro for flow analysis
4 // Creates a Flow Event task and adds it to the analysis manager.
5 // Sets the cuts using the correction framework (CORRFW) classes.
6 // Also creates Flow Analysis tasks and connects them to the output of the flow event task.
7 //
8 /////////////////////////////////////////////////////////////////////////////////////////////
9
10 void AddTaskFlowCentralityPIDSP(Int_t centralitysel,
11                                Float_t centrMin=10.,
12                                Float_t centrMax=20.,
13                                TString fileNameBase="output",
14                                Bool_t isPID = kTRUE,
15                                AliPID::EParticleType particleType=AliPID::kPion,
16                                AliFlowTrackCuts::PIDsource sourcePID = AliFlowTrackCuts::kTOFbayesian,
17                                Int_t charge=0,
18                                Int_t harmonic=2,
19                                Bool_t doQA=kFALSE,
20                                Float_t etamin=-0.8,
21                                Float_t etamax=0.8,       
22                                TString uniqueStr="",
23                                Int_t side      = 0,
24                                Int_t filterbit=1)
25 {
26   // Define the range for eta subevents (for SP method)
27   Double_t minA = -5;
28   Double_t maxA = -1.5;
29   Double_t minB = 1.5;
30   Double_t maxB = 5;
31
32   // AFTERBURNER
33   Bool_t useAfterBurner=kFALSE;
34   Double_t v1=0.0;
35   Double_t v2=0.0;
36   Double_t v3=0.0;
37   Double_t v4=0.0;
38   Int_t numberOfTrackClones=0; //non-flow
39
40   // Define a range of the detector to exclude
41   Bool_t ExcludeRegion = kFALSE;
42   Double_t excludeEtaMin = -0.;
43   Double_t excludeEtaMax = 0.;
44   Double_t excludePhiMin = 0.;
45   Double_t excludePhiMax = 0.;
46
47   // use physics selection class
48   Bool_t  UsePhysicsSelection = kTRUE;
49
50   // QA
51   Bool_t runQAtask=kFALSE;
52   Bool_t FillQAntuple=kFALSE;
53   Bool_t DoQAcorrelations=kFALSE;
54
55   // RUN SETTINGS
56   // Flow analysis method can be:(set to kTRUE or kFALSE)
57   Bool_t SP       = kTRUE;  // scalar product method (similar to eventplane method)
58   Bool_t QC       = kFALSE;  // cumulants using Q vectors
59   
60   //these are OBSOLETE, use at own peril
61   Bool_t GFC      = kFALSE;  // cumulants based on generating function
62   Bool_t MCEP     = kFALSE;  // correlation with Monte Carlo reaction plane
63   Bool_t FQD      = kFALSE;  // fit of the distribution of the Q vector (only integrated v)
64   Bool_t LYZ1SUM  = kFALSE;  // Lee Yang Zeroes using sum generating function (integrated v)
65   Bool_t LYZ1PROD = kFALSE;  // Lee Yang Zeroes using product generating function (integrated v)
66   Bool_t LYZ2SUM  = kFALSE; // Lee Yang Zeroes using sum generating function (second pass differential v)
67   Bool_t LYZ2PROD = kFALSE; // Lee Yang Zeroes using product generating function (second pass differential v)
68   Bool_t LYZEP    = kFALSE; // Lee Yang Zeroes Event plane using sum generating function (gives eventplane + weight)
69   Bool_t MH       = kFALSE;  // azimuthal correlators in mixed harmonics  
70   Bool_t NL       = kFALSE;  // nested loops (for instance distribution of phi1-phi2 for all distinct pairs)
71
72   Bool_t METHODS[] = {SP,LYZ1SUM,LYZ1PROD,LYZ2SUM,LYZ2PROD,LYZEP,GFC,QC,FQD,MCEP,MH,NL};
73
74   // Boolean to use/not use weights for the Q vector
75   Bool_t WEIGHTS[] = {kFALSE,kFALSE,kFALSE}; //Phi, v'(pt), v'(eta)
76
77   // SETTING THE CUTS
78
79   //---------Data selection----------
80   //kMC, kGlobal, kESD_TPConly, kESD_SPDtracklet
81   AliFlowTrackCuts::trackParameterType rptype = AliFlowTrackCuts::kTPCstandalone;
82   AliFlowTrackCuts::trackParameterType poitype = AliFlowTrackCuts::kTPCstandalone;
83
84   //---------Parameter mixing--------
85   //kPure - no mixing, kTrackWithMCkine, kTrackWithMCPID, kTrackWithMCpt
86   AliFlowTrackCuts::trackParameterMix rpmix = AliFlowTrackCuts::kPure;
87   AliFlowTrackCuts::trackParameterMix poimix = AliFlowTrackCuts::kPure;
88
89
90   const char* rptypestr = AliFlowTrackCuts::GetParamTypeName(rptype);
91   const char* poitypestr = AliFlowTrackCuts::GetParamTypeName(poitype);
92
93   //===========================================================================
94   // EVENTS CUTS:
95  AliFlowEventCuts* cutsEvent = new AliFlowEventCuts("event cuts");
96   cutsEvent->SetCentralityPercentileRange(centrMin,centrMax);
97   cutsEvent->SetCentralityPercentileMethod(AliFlowEventCuts::kV0);
98 //  cutsEvent->SetRefMultMethod(AliFlowEventCuts::kV0);
99   //cutsEvent->SetCentralityPercentileMethod(AliFlowEventCuts::kSPD1tracklets);
100 //  cutsEvent->SetNContributorsRange(2);
101   cutsEvent->SetPrimaryVertexZrange(-10.,10.);
102   cutsEvent->SetQA(doQA);
103   cutsEvent->SetCutTPCmultiplicityOutliers();
104   
105   // RP TRACK CUTS:
106   AliFlowTrackCuts* cutsRP2 = AliFlowTrackCuts::GetStandardVZEROOnlyTrackCuts();
107   AliFlowTrackCuts* cutsRP = new AliFlowTrackCuts("TPConlyRP");
108   cutsRP->SetParamType(rptype);
109   cutsRP->SetParamMix(rpmix);
110   cutsRP->SetPtRange(0.2,5.);
111   // side A
112   if(side < 0)
113     cutsRP->SetEtaRange(etamin,0.);
114
115   // side C
116   else if(side > 0)
117     cutsRP->SetEtaRange(0.,etamax);
118
119   else
120     cutsRP->SetEtaRange(etamin,etamax);
121
122   cutsRP->SetMinNClustersTPC(70);
123 //  cutsRP->SetMinChi2PerClusterTPC(0.1);
124 //  cutsRP->SetMaxChi2PerClusterTPC(4.0);
125   cutsRP->SetMaxDCAToVertexXY(3.0);
126   cutsRP->SetMaxDCAToVertexZ(3.0);
127   cutsRP->SetAcceptKinkDaughters(kFALSE);
128   cutsRP->SetMinimalTPCdedx(10.);
129   cutsRP->SetQA(doQA);
130
131   // POI TRACK CUTS:
132   AliFlowTrackCuts* cutsPOI = new AliFlowTrackCuts("TPConlyPOI");
133   cutsPOI->GetBayesianResponse()->ForceOldDedx(); // for 2010 data to use old TPC PID Response instead of the official one
134   cutsPOI->SetParamType(poitype);
135   cutsPOI->SetParamMix(poimix);
136   cutsPOI->SetPtRange(0.0,10.);
137   cutsPOI->SetEtaRange(etamin,etamax);
138   cutsPOI->SetMinNClustersTPC(70);
139 //  cutsPOI->SetMinChi2PerClusterTPC(0.1);
140 //  cutsPOI->SetMaxChi2PerClusterTPC(4.0);
141 //  cutsPOI->SetRequireITSRefit(kTRUE);
142 //  cutsPOI->SetRequireTPCRefit(kTRUE);
143 //  cutsPOI->SetMinNClustersITS(2);
144   //cutsPOI->SetMaxChi2PerClusterITS(1.e+09);
145   cutsPOI->SetMaxDCAToVertexXY(3.0);
146   cutsPOI->SetMaxDCAToVertexZ(3.0);
147   //cutsPOI->SetDCAToVertex2D(kTRUE);
148   //cutsPOI->SetMaxNsigmaToVertex(1.e+10);
149   //cutsPOI->SetRequireSigmaToVertex(kFALSE);
150   cutsPOI->SetAcceptKinkDaughters(kFALSE);
151   if(isPID) cutsPOI->SetPID(particleType, sourcePID);
152   if (charge!=0) cutsPOI->SetCharge(charge);
153   //cutsPOI->SetAllowTOFmismatch(kFALSE);
154   cutsPOI->SetRequireStrictTOFTPCagreement(kTRUE);
155   //iexample: francesco's tunig TPC Bethe Bloch for data:
156   //cutsPOI->GetESDpid().GetTPCResponse().SetBetheBlochParameters(4.36414e-02,1.75977e+01,1.14385e-08,2.27907e+00,3.36699e+00);
157   //cutsPOI->GetESDpid().GetTPCResponse().SetMip(49);
158   cutsPOI->SetMinimalTPCdedx(10.);
159   cutsPOI->SetAODfilterBit(filterbit);
160   cutsPOI->SetQA(doQA);
161   cutsPOI->SetPriors((centrMin+centrMax)*0.5); // set priors and PID as a function of the centrality
162
163   TString outputSlotName("");
164   outputSlotName+=uniqueStr;
165   outputSlotName+=Form("V%i ",harmonic);
166   outputSlotName+=cutsRP->GetName();
167   outputSlotName+=" ";
168   outputSlotName+=cutsPOI->GetName();
169   outputSlotName+=Form(" %.0f-",centrMin);
170   outputSlotName+=Form("%.0f ",centrMax);
171   if(isPID){
172      outputSlotName+=AliFlowTrackCuts::PIDsourceName(sourcePID);
173      outputSlotName+=" ";
174      outputSlotName+=AliPID::ParticleName(particleType);
175   }
176   else{
177      outputSlotName+="AllCharged";
178   }
179   if (charge<0) outputSlotName+="-";
180   if (charge>0) outputSlotName+="+";
181
182   TString fileName(fileNameBase);
183   fileName.Append(".root");
184
185   Bool_t useWeights  = WEIGHTS[0] || WEIGHTS[1] || WEIGHTS[2];
186   if (useWeights) cout<<"Weights are used"<<endl;
187   else cout<<"Weights are not used"<<endl;
188   
189   // Get the pointer to the existing analysis manager via the static access method.
190   //==============================================================================
191   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
192   if (!mgr) {
193     Error("AddTaskFlowEvent", "No analysis manager to connect to.");
194     return NULL;
195   }
196   
197   // Check the analysis type using the event handlers connected to the analysis
198   // manager. The availability of MC handler can also be checked here.
199   //==============================================================================
200   if (!mgr->GetInputEventHandler()) {
201     ::Error("AddTaskFlowEvent", "This task requires an input event handler");
202     return NULL;
203   }  
204
205   // Open external input files
206   //===========================================================================
207   //weights: 
208   TFile *weightsFile = NULL;
209   TList *weightsList = NULL;
210
211   if(useWeights) {
212     //open the file with the weights:
213     weightsFile = TFile::Open("weights.root","READ");
214     if(weightsFile) {
215       //access the list which holds the histos with weigths:
216       weightsList = (TList*)weightsFile->Get("weights");
217     }
218     else {
219       cout<<" WARNING: the file <weights.root> with weights from the previous run was not available."<<endl;
220       break;
221     } 
222   }
223   
224   //LYZ2
225   if (LYZ2SUM || LYZ2PROD) {
226     //read the outputfile of the first run
227     TString outputFileName = "AnalysisResults1.root";
228     TString pwd(gSystem->pwd());
229     pwd+="/";
230     pwd+=outputFileName.Data();
231     TFile *outputFile = NULL;
232     if(gSystem->AccessPathName(pwd.Data(),kFileExists)) {
233       cout<<"WARNING: You do not have an output file:"<<endl;
234       cout<<"         "<<pwd.Data()<<endl;
235       exit(0);
236     } else { outputFile = TFile::Open(pwd.Data(),"READ");}
237     
238     if (LYZ2SUM){  
239       // read the output directory from LYZ1SUM 
240       TString inputFileNameLYZ2SUM = "outputLYZ1SUManalysis" ;
241       inputFileNameLYZ2SUM += rptypestr;
242       cout<<"The input directory is "<<inputFileNameLYZ2SUM.Data()<<endl;
243       TFile* fInputFileLYZ2SUM = (TFile*)outputFile->FindObjectAny(inputFileNameLYZ2SUM.Data());
244       if(!fInputFileLYZ2SUM || fInputFileLYZ2SUM->IsZombie()) { 
245         cerr << " ERROR: To run LYZ2SUM you need the output file from LYZ1SUM. This file is not there! Please run LYZ1SUM first." << endl ; 
246         break;
247       }
248       else {
249         TList* fInputListLYZ2SUM = (TList*)fInputFileLYZ2SUM->Get("LYZ1SUM");
250         if (!fInputListLYZ2SUM) {cout<<"list is NULL pointer!"<<endl;}
251       }
252       cout<<"LYZ2SUM input file/list read..."<<endl;
253     }
254
255     if (LYZ2PROD){  
256       // read the output directory from LYZ1PROD 
257       TString inputFileNameLYZ2PROD = "outputLYZ1PRODanalysis" ;
258       inputFileNameLYZ2PROD += rptypestr;
259       cout<<"The input directory is "<<inputFileNameLYZ2PROD.Data()<<endl;
260       TFile* fInputFileLYZ2PROD = (TFile*)outputFile->FindObjectAny(inputFileNameLYZ2PROD.Data());
261       if(!fInputFileLYZ2PROD || fInputFileLYZ2PROD->IsZombie()) { 
262         cerr << " ERROR: To run LYZ2PROD you need the output file from LYZ1PROD. This file is not there! Please run LYZ1PROD first." << endl ; 
263         break;
264       }
265       else {
266         TList* fInputListLYZ2PROD = (TList*)fInputFileLYZ2PROD->Get("LYZ1PROD");
267         if (!fInputListLYZ2PROD) {cout<<"list is NULL pointer!"<<endl;}
268       }
269       cout<<"LYZ2PROD input file/list read..."<<endl;
270     }
271   }
272
273   if (LYZEP) {
274     //read the outputfile of the second run
275     TString outputFileName = "AnalysisResults2.root";
276     TString pwd(gSystem->pwd());
277     pwd+="/";
278     pwd+=outputFileName.Data();
279     TFile *outputFile = NULL;
280     if(gSystem->AccessPathName(pwd.Data(),kFileExists)) {
281       cout<<"WARNING: You do not have an output file:"<<endl;
282       cout<<"         "<<pwd.Data()<<endl;
283       exit(0);
284     } else {
285       outputFile = TFile::Open(pwd.Data(),"READ");
286     }
287     
288     // read the output file from LYZ2SUM
289     TString inputFileNameLYZEP = "outputLYZ2SUManalysis" ;
290     inputFileNameLYZEP += rptypestr;
291     cout<<"The input file is "<<inputFileNameLYZEP.Data()<<endl;
292     TFile* fInputFileLYZEP = (TFile*)outputFile->FindObjectAny(inputFileNameLYZEP.Data());
293     if(!fInputFileLYZEP || fInputFileLYZEP->IsZombie()) { 
294       cerr << " ERROR: To run LYZEP you need the output file from LYZ2SUM. This file is not there! Please run LYZ2SUM first." << endl ; 
295       break;
296     }
297     else {
298       TList* fInputListLYZEP = (TList*)fInputFileLYZEP->Get("LYZ2SUM");
299       if (!fInputListLYZEP) {cout<<"list is NULL pointer!"<<endl;}
300     }
301     cout<<"LYZEP input file/list read..."<<endl;
302   }
303   
304   
305   // Create the FMD task and add it to the manager
306   //===========================================================================
307   if (rptypestr == "FMD") {
308     AliFMDAnalysisTaskSE *taskfmd = NULL;
309     if (rptypestr == "FMD") {
310       taskfmd = new AliFMDAnalysisTaskSE("TaskFMD");
311       taskfmd->SelectCollisionCandidates(centralitysel);
312       mgr->AddTask(taskfmd);
313       
314       AliFMDAnaParameters* pars = AliFMDAnaParameters::Instance();
315       pars->Init();
316       pars->SetProcessPrimary(kTRUE); //for MC only
317       pars->SetProcessHits(kFALSE);
318       
319       //pars->SetRealData(kTRUE); //for real data
320       //pars->SetProcessPrimary(kFALSE); //for real data
321     }
322   }
323   
324   // Create the flow event task, add it to the manager.
325   //===========================================================================
326   AliAnalysisTaskFlowEvent *taskFE = NULL;
327
328   if(useAfterBurner)
329     { 
330       taskFE = new AliAnalysisTaskFlowEvent(Form("TaskFlowEvent %s",outputSlotName.Data()),"",doQA,1);
331       taskFE->SelectCollisionCandidates(centralitysel);
332       taskFE->SetFlow(v1,v2,v3,v4); 
333       taskFE->SetNonFlowNumberOfTrackClones(numberOfTrackClones);
334       taskFE->SetAfterburnerOn();
335     }
336   else {taskFE = new AliAnalysisTaskFlowEvent(Form("TaskFlowEvent %s",outputSlotName.Data()),"",doQA); 
337       taskFE->SelectCollisionCandidates(centralitysel);}
338   if (ExcludeRegion) {
339     taskFE->DefineDeadZone(excludeEtaMin, excludeEtaMax, excludePhiMin, excludePhiMax); 
340   }
341   taskFE->SetSubeventEtaRange(minA, maxA, minB, maxB);
342   if (UsePhysicsSelection) {
343     taskFE->SelectCollisionCandidates(AliVEvent::kMB);
344     cout<<"Using Physics Selection"<<endl;
345   }
346   mgr->AddTask(taskFE);
347   
348   // Pass cuts for RPs and POIs to the task:
349   taskFE->SetCutsEvent(cutsEvent);
350   taskFE->SetCutsRP(cutsRP2);
351   taskFE->SetCutsPOI(cutsPOI);
352   if (cutsRP->GetParamType()==AliFlowTrackCuts::kV0)
353   { 
354     //TODO: since this is set in a static object all analyses in an analysis train
355     //will be affected.
356     taskFE->SetHistWeightvsPhiMin(0.);
357     taskFE->SetHistWeightvsPhiMax(200.);
358   }
359
360   // Create the analysis tasks, add them to the manager.
361   //===========================================================================
362   if (SP){
363     AliAnalysisTaskScalarProduct *taskSP = new AliAnalysisTaskScalarProduct(Form("TaskScalarProduct %s",outputSlotName.Data()),WEIGHTS[0]);
364     taskSP->SelectCollisionCandidates(centralitysel);
365     if(side < 0)   taskSP->SetTotalQvector("Qb");
366     else     taskSP->SetTotalQvector("Qa");
367     taskSP->SetRelDiffMsub(1.0);
368     taskSP->SetApplyCorrectionForNUA(kTRUE);
369     mgr->AddTask(taskSP);
370   }
371   if (LYZ1SUM){
372     AliAnalysisTaskLeeYangZeros *taskLYZ1SUM = new AliAnalysisTaskLeeYangZeros(Form("TaskLeeYangZerosSUM %s",outputSlotName.Data()),kTRUE);
373     taskLYZ1SUM->SelectCollisionCandidates(centralitysel);
374     taskLYZ1SUM->SetFirstRunLYZ(kTRUE);
375     taskLYZ1SUM->SetUseSumLYZ(kTRUE);
376     mgr->AddTask(taskLYZ1SUM);
377   }
378   if (LYZ1PROD){
379     AliAnalysisTaskLeeYangZeros *taskLYZ1PROD = new AliAnalysisTaskLeeYangZeros(Form("TaskLeeYangZerosPROD %s",outputSlotName.Data()),kTRUE);
380     taskLYZ1PROD->SelectCollisionCandidates(centralitysel);
381     taskLYZ1PROD->SetFirstRunLYZ(kTRUE);
382     taskLYZ1PROD->SetUseSumLYZ(kFALSE);
383     mgr->AddTask(taskLYZ1PROD);
384   }
385   if (LYZ2SUM){
386     AliAnalysisTaskLeeYangZeros *taskLYZ2SUM = new AliAnalysisTaskLeeYangZeros(Form("TaskLeeYangZerosSUM %s",outputSlotName.Data()),kFALSE);
387     taskLYZ2SUM->SelectCollisionCandidates(centralitysel);
388     taskLYZ2SUM->SetFirstRunLYZ(kFALSE);
389     taskLYZ2SUM->SetUseSumLYZ(kTRUE);
390     mgr->AddTask(taskLYZ2SUM);
391   }
392   if (LYZ2PROD){
393     AliAnalysisTaskLeeYangZeros *taskLYZ2PROD = new AliAnalysisTaskLeeYangZeros(Form("TaskLeeYangZerosPROD %s",outputSlotName.Data()),kFALSE);
394     taskLYZ2PROD->SelectCollisionCandidates(centralitysel);
395     taskLYZ2PROD->SetFirstRunLYZ(kFALSE);
396     taskLYZ2PROD->SetUseSumLYZ(kFALSE);
397     mgr->AddTask(taskLYZ2PROD);
398   }
399   if (LYZEP){
400     AliAnalysisTaskLYZEventPlane *taskLYZEP = new AliAnalysisTaskLYZEventPlane(Form("TaskLYZEventPlane %s",outputSlotName.Data()));
401     taskLYZEP->SelectCollisionCandidates(centralitysel);
402     mgr->AddTask(taskLYZEP);
403   }
404   if (GFC){
405     AliAnalysisTaskCumulants *taskGFC = new AliAnalysisTaskCumulants(Form("TaskCumulants %s",outputSlotName.Data()),useWeights);
406     taskGFC->SelectCollisionCandidates(centralitysel);
407     taskGFC->SetUsePhiWeights(WEIGHTS[0]); 
408     taskGFC->SetUsePtWeights(WEIGHTS[1]);
409     taskGFC->SetUseEtaWeights(WEIGHTS[2]); 
410     mgr->AddTask(taskGFC);
411   }
412   if (QC){
413     AliAnalysisTaskQCumulants *taskQC = new AliAnalysisTaskQCumulants(Form("TaskQCumulants %s",outputSlotName.Data()),useWeights);
414     taskQC->SelectCollisionCandidates(centralitysel);
415     taskQC->SetUsePhiWeights(WEIGHTS[0]); 
416     taskQC->SetUsePtWeights(WEIGHTS[1]);
417     taskQC->SetUseEtaWeights(WEIGHTS[2]); 
418     taskQC->SetCalculateCumulantsVsM(kFALSE);
419     taskQC->SetnBinsMult(10000);
420     taskQC->SetMinMult(0.);
421     taskQC->SetMaxMult(10000.);
422     taskQC->SetHarmonic(harmonic);
423     taskQC->SetApplyCorrectionForNUA(kFALSE);
424     taskQC->SetFillMultipleControlHistograms(kFALSE);     
425     mgr->AddTask(taskQC);
426   }
427   if (FQD){
428     AliAnalysisTaskFittingQDistribution *taskFQD = new AliAnalysisTaskFittingQDistribution(Form("TaskFittingQDistribution %s",outputSlotName.Data()),kFALSE);
429     taskFQD->SelectCollisionCandidates(centralitysel);
430     taskFQD->SetUsePhiWeights(WEIGHTS[0]); 
431     taskFQD->SetqMin(0.);
432     taskFQD->SetqMax(1000.);
433     taskFQD->SetqNbins(10000);
434     mgr->AddTask(taskFQD);
435   }
436   if (MCEP){
437     AliAnalysisTaskMCEventPlane *taskMCEP = new AliAnalysisTaskMCEventPlane(Form("TaskMCEventPlane %s",outputSlotName.Data()));
438     taskMCEP->SelectCollisionCandidates(centralitysel);
439     mgr->AddTask(taskMCEP);
440   }
441   if (MH){
442     AliAnalysisTaskMixedHarmonics *taskMH = new AliAnalysisTaskMixedHarmonics(Form("TaskMixedHarmonics %s",outputSlotName.Data()),useWeights);
443     taskMH->SelectCollisionCandidates(centralitysel);
444     taskMH->SetHarmonic(1); // n in cos[n(phi1+phi2-2phi3)] and cos[n(psi1+psi2-2phi3)]
445     taskMH->SetNoOfMultipicityBins(10000);
446     taskMH->SetMultipicityBinWidth(1.);
447     taskMH->SetMinMultiplicity(1.);
448     taskMH->SetCorrectForDetectorEffects(kTRUE);
449     taskMH->SetEvaluateDifferential3pCorrelator(kFALSE); // evaluate <<cos[n(psi1+psi2-2phi3)]>> (Remark: two nested loops)    
450     taskMH->SetOppositeChargesPOI(kFALSE); // POIs psi1 and psi2 in cos[n(psi1+psi2-2phi3)] will have opposite charges  
451     mgr->AddTask(taskMH);
452   }  
453   if (NL){
454     AliAnalysisTaskNestedLoops *taskNL = new AliAnalysisTaskNestedLoops(Form("TaskNestedLoops %s",outputSlotName.Data()),useWeights);
455     taskNL->SelectCollisionCandidates(centralitysel);
456     taskNL->SetHarmonic(1); // n in cos[n(phi1+phi2-2phi3)] and cos[n(psi1+psi2-2phi3)]
457     taskNL->SetEvaluateNestedLoopsForRAD(kTRUE); // RAD = Relative Angle Distribution
458     taskNL->SetEvaluateNestedLoopsForMH(kTRUE); // evalaute <<cos[n(phi1+phi2-2phi3)]>> (Remark: three nested loops)   
459     taskNL->SetEvaluateDifferential3pCorrelator(kFALSE); // evaluate <<cos[n(psi1+psi2-2phi3)]>>  (Remark: three nested loops)   
460     taskNL->SetOppositeChargesPOI(kFALSE); // POIs psi1 and psi2 in cos[n(psi1+psi2-2phi3)] will have opposite charges  
461     mgr->AddTask(taskNL);
462   }
463
464   // Create the output container for the data produced by the task
465   // Connect to the input and output containers
466   //===========================================================================
467   AliAnalysisDataContainer *cinput1 = mgr->GetCommonInputContainer();
468   
469   if (rptypestr == "FMD") {
470     AliAnalysisDataContainer *coutputFMD = 
471       mgr->CreateContainer(Form("BackgroundCorrected %s",outputSlotName.Data()), TList::Class(), AliAnalysisManager::kExchangeContainer);
472     //input and output taskFMD     
473     mgr->ConnectInput(taskfmd, 0, cinput1);
474     mgr->ConnectOutput(taskfmd, 1, coutputFMD);
475     //input into taskFE
476     mgr->ConnectInput(taskFE,1,coutputFMD);
477   }
478   
479   AliAnalysisDataContainer *coutputFE = 
480   mgr->CreateContainer(Form("FlowEventSimple %s",outputSlotName.Data()),AliFlowEventSimple::Class(),AliAnalysisManager::kExchangeContainer);
481   mgr->ConnectInput(taskFE,0,cinput1); 
482   mgr->ConnectOutput(taskFE,1,coutputFE);
483  
484   if (taskFE->GetQAOn())
485   {
486     TString outputQA = fileName;
487     outputQA += ":QA";
488     AliAnalysisDataContainer* coutputFEQA = 
489     mgr->CreateContainer(Form("QA %s",outputSlotName.Data()), TList::Class(),AliAnalysisManager::kOutputContainer,outputQA);
490     mgr->ConnectOutput(taskFE,2,coutputFEQA);
491   }
492
493   // Create the output containers for the data produced by the analysis tasks
494   // Connect to the input and output containers
495   //===========================================================================
496   if (useWeights) {    
497     AliAnalysisDataContainer *cinputWeights = mgr->CreateContainer(Form("Weights %s",outputSlotName.Data()),
498                                                                    TList::Class(),AliAnalysisManager::kInputContainer); 
499   }
500
501   if(SP) {
502     TString outputSP = fileName;
503     outputSP += ":outputSPanalysis";
504     outputSP+= rptypestr;
505     AliAnalysisDataContainer *coutputSP = mgr->CreateContainer(Form("SP %s",outputSlotName.Data()), 
506                                                                TList::Class(),AliAnalysisManager::kOutputContainer,outputSP); 
507     mgr->ConnectInput(taskSP,0,coutputFE); 
508     mgr->ConnectOutput(taskSP,1,coutputSP); 
509     if (WEIGHTS[0]) {
510       mgr->ConnectInput(taskSP,1,cinputWeights);
511       cinputWeights->SetData(weightsList);
512     }
513   }
514   if(LYZ1SUM) {
515     TString outputLYZ1SUM = fileName;
516     outputLYZ1SUM += ":outputLYZ1SUManalysis";
517     outputLYZ1SUM+= rptypestr;
518     AliAnalysisDataContainer *coutputLYZ1SUM = mgr->CreateContainer(Form("LYZ1SUM %s",outputSlotName.Data()), 
519                                                                     TList::Class(),AliAnalysisManager::kOutputContainer,outputLYZ1SUM); 
520     mgr->ConnectInput(taskLYZ1SUM,0,coutputFE);
521     mgr->ConnectOutput(taskLYZ1SUM,1,coutputLYZ1SUM);
522   }
523   if(LYZ1PROD) {
524     TString outputLYZ1PROD = fileName;
525     outputLYZ1PROD += ":outputLYZ1PRODanalysis";
526     outputLYZ1PROD+= rptypestr;
527     AliAnalysisDataContainer *coutputLYZ1PROD = mgr->CreateContainer(Form("LYZ1PROD %s",outputSlotName.Data()), 
528                                                                      TList::Class(),AliAnalysisManager::kOutputContainer,outputLYZ1PROD); 
529     mgr->ConnectInput(taskLYZ1PROD,0,coutputFE); 
530     mgr->ConnectOutput(taskLYZ1PROD,1,coutputLYZ1PROD);
531   }
532   if(LYZ2SUM) {
533     AliAnalysisDataContainer *cinputLYZ2SUM = mgr->CreateContainer(Form("LYZ2SUMin %s",outputSlotName.Data()),
534                                                                    TList::Class(),AliAnalysisManager::kInputContainer);
535     TString outputLYZ2SUM = fileName;
536     outputLYZ2SUM += ":outputLYZ2SUManalysis";
537     outputLYZ2SUM+= rptypestr;
538     
539     AliAnalysisDataContainer *coutputLYZ2SUM = mgr->CreateContainer(Form("LYZ2SUM %s",outputSlotName.Data()), 
540                                                                     TList::Class(),AliAnalysisManager::kOutputContainer,outputLYZ2SUM); 
541     mgr->ConnectInput(taskLYZ2SUM,0,coutputFE); 
542     mgr->ConnectInput(taskLYZ2SUM,1,cinputLYZ2SUM);
543     mgr->ConnectOutput(taskLYZ2SUM,1,coutputLYZ2SUM); 
544     cinputLYZ2SUM->SetData(fInputListLYZ2SUM);
545   }
546   if(LYZ2PROD) {
547     AliAnalysisDataContainer *cinputLYZ2PROD = mgr->CreateContainer(Form("LYZ2PRODin %s",outputSlotName.Data()),
548                                                                     TList::Class(),AliAnalysisManager::kInputContainer);
549     TString outputLYZ2PROD = fileName;
550     outputLYZ2PROD += ":outputLYZ2PRODanalysis";
551     outputLYZ2PROD+= rptypestr;
552     
553     AliAnalysisDataContainer *coutputLYZ2PROD = mgr->CreateContainer(Form("LYZ2PROD %s",outputSlotName.Data()), 
554                                                                      TList::Class(),AliAnalysisManager::kOutputContainer,outputLYZ2PROD); 
555     mgr->ConnectInput(taskLYZ2PROD,0,coutputFE); 
556     mgr->ConnectInput(taskLYZ2PROD,1,cinputLYZ2PROD);
557     mgr->ConnectOutput(taskLYZ2PROD,1,coutputLYZ2PROD); 
558     cinputLYZ2PROD->SetData(fInputListLYZ2PROD);
559   }
560   if(LYZEP) {
561     AliAnalysisDataContainer *cinputLYZEP = mgr->CreateContainer(Form("LYZEPin %s",outputSlotName.Data()),
562                                                                  TList::Class(),AliAnalysisManager::kInputContainer);
563     TString outputLYZEP = fileName;
564     outputLYZEP += ":outputLYZEPanalysis";
565     outputLYZEP+= rptypestr;
566     
567     AliAnalysisDataContainer *coutputLYZEP = mgr->CreateContainer(Form("LYZEP %s",outputSlotName.Data()), 
568                                                                   TList::Class(),AliAnalysisManager::kOutputContainer,outputLYZEP); 
569     mgr->ConnectInput(taskLYZEP,0,coutputFE); 
570     mgr->ConnectInput(taskLYZEP,1,cinputLYZEP);
571     mgr->ConnectOutput(taskLYZEP,1,coutputLYZEP); 
572     cinputLYZEP->SetData(fInputListLYZEP);
573   }
574   if(GFC) {
575     TString outputGFC = fileName;
576     outputGFC += ":outputGFCanalysis";
577     outputGFC+= rptypestr;
578     
579     AliAnalysisDataContainer *coutputGFC = mgr->CreateContainer(Form("GFC %s",outputSlotName.Data()), 
580                                                                 TList::Class(),AliAnalysisManager::kOutputContainer,outputGFC); 
581     mgr->ConnectInput(taskGFC,0,coutputFE); 
582     mgr->ConnectOutput(taskGFC,1,coutputGFC);
583     if (useWeights) {
584       mgr->ConnectInput(taskGFC,1,cinputWeights);
585       cinputWeights->SetData(weightsList);
586     } 
587   }
588   if(QC) {
589     TString outputQC = fileName;
590     outputQC += ":outputQCanalysis";
591     outputQC+= rptypestr;
592
593     AliAnalysisDataContainer *coutputQC = mgr->CreateContainer(Form("QC %s",outputSlotName.Data()), 
594                                                                TList::Class(),AliAnalysisManager::kOutputContainer,outputQC); 
595     mgr->ConnectInput(taskQC,0,coutputFE); 
596     mgr->ConnectOutput(taskQC,1,coutputQC);
597     if (useWeights) {
598       mgr->ConnectInput(taskQC,1,cinputWeights);
599       cinputWeights->SetData(weightsList);
600     }
601   }
602   if(FQD) {
603     TString outputFQD = fileName;
604     outputFQD += ":outputFQDanalysis";
605     outputFQD+= rptypestr;
606     
607     AliAnalysisDataContainer *coutputFQD = mgr->CreateContainer(Form("FQD %s",outputSlotName.Data()), 
608                                                                 TList::Class(),AliAnalysisManager::kOutputContainer,outputFQD); 
609     mgr->ConnectInput(taskFQD,0,coutputFE); 
610     mgr->ConnectOutput(taskFQD,1,coutputFQD);
611     if(useWeights) {
612       mgr->ConnectInput(taskFQD,1,cinputWeights);
613       cinputWeights->SetData(weightsList);
614     } 
615   }
616   if(MCEP) {
617     TString outputMCEP = fileName;
618     outputMCEP += ":outputMCEPanalysis";
619     outputMCEP+= rptypestr;
620     
621     AliAnalysisDataContainer *coutputMCEP = mgr->CreateContainer(Form("MCEP %s",outputSlotName.Data()), 
622                                                                  TList::Class(),AliAnalysisManager::kOutputContainer,outputMCEP); 
623     mgr->ConnectInput(taskMCEP,0,coutputFE);
624     mgr->ConnectOutput(taskMCEP,1,coutputMCEP); 
625   }
626   if(MH) {
627     TString outputMH = fileName;
628     outputMH += ":outputMHanalysis";
629     outputMH += rptypestr;
630         
631     AliAnalysisDataContainer *coutputMH = mgr->CreateContainer(Form("MH %s",outputSlotName.Data()), 
632                                                                TList::Class(),AliAnalysisManager::kOutputContainer,outputMH); 
633     mgr->ConnectInput(taskMH,0,coutputFE); 
634     mgr->ConnectOutput(taskMH,1,coutputMH); 
635     //if (useWeights) {
636     //  mgr->ConnectInput(taskMH,1,cinputWeights);
637     //  cinputWeights->SetData(weightsList);
638     //} 
639   }
640   if(NL) {
641     TString outputNL = fileName;
642     outputNL += ":outputNLanalysis";
643     outputNL += rptypestr;
644
645     AliAnalysisDataContainer *coutputNL = mgr->CreateContainer(Form("NL %s",outputSlotName.Data()), 
646                                                                TList::Class(),AliAnalysisManager::kOutputContainer,outputNL); 
647     mgr->ConnectInput(taskNL,0,coutputFE);
648     mgr->ConnectOutput(taskNL,1,coutputNL);
649     //if (useWeights) {
650     //  mgr->ConnectInput(taskNL,1,cinputWeights);
651     //  cinputWeights->SetData(weightsList);
652     //} 
653   }
654
655   ///////////////////////////////////////////////////////////////////////////////////////////
656   if (runQAtask)
657   {
658     AliAnalysisTaskQAflow* taskQAflow = new AliAnalysisTaskQAflow(Form("TaskQAflow %s",outputSlotName.Data()));
659     taskQAflow->SelectCollisionCandidates(centralitysel);
660     taskQAflow->SetEventCuts(cutsEvent);
661     taskQAflow->SetTrackCuts(cutsRP);
662     taskQAflow->SetFillNTuple(FillQAntuple);
663     taskQAflow->SetDoCorrelations(DoQAcorrelations);
664     mgr->AddTask(taskQAflow);
665     
666     Printf("outputSlotName %s",outputSlotName.Data());
667     TString taskQAoutputFileName(fileNameBase);
668     taskQAoutputFileName.Append("_QA.root");
669     AliAnalysisDataContainer* coutputQAtask = mgr->CreateContainer(Form("flowQA %s",outputSlotName.Data()),
670                                               TObjArray::Class(),
671                                               AliAnalysisManager::kOutputContainer,
672                                               taskQAoutputFileName);
673     AliAnalysisDataContainer* coutputQAtaskTree = mgr->CreateContainer(Form("flowQAntuple %s",outputSlotName.Data()),
674                                               TNtuple::Class(),
675                                               AliAnalysisManager::kOutputContainer,
676                                               taskQAoutputFileName);
677     mgr->ConnectInput(taskQAflow,0,mgr->GetCommonInputContainer());
678     mgr->ConnectInput(taskQAflow,1,coutputFE);
679     mgr->ConnectOutput(taskQAflow,1,coutputQAtask);
680     if (FillQAntuple) mgr->ConnectOutput(taskQAflow,2,coutputQAtaskTree);
681   }
682 }
683
684
685
686
687