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