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