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