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