1 ///////////////////////////////////////////////////////////////////
4 // Author: Redmer A. Bertens, Utrecht University, 2013 //
6 ///////////////////////////////////////////////////////////////////
7 class AliAnalysisDataContainer;
8 class AliAnalysisTaskRhoVnModulation;
10 // AddTask macro for the jet flow analysis task
11 // this task uses an instance AliAnalysisTaskRhoVnModulation
12 // as a 'master' object for analysis flags and track and event cuts
14 AliAnalysisTaskRhoVnModulation* AddTaskJetFlow(
15 TString name = "name",
16 AliAnalysisTaskRhoVnModulation* t = 0x0,
18 Float_t CCMaxPt = 150,
19 Float_t CCBinsInPt = 100,
20 Bool_t VParticle = kFALSE,
21 TArrayD* ptArray = 0x0,
22 Bool_t VZEROEP = kTRUE,
25 Bool_t FlowPackageSP = kFALSE,
26 Bool_t FlowPackageQC = kTRUE,
27 Bool_t debug = kFALSE )
29 // first check the environment (common to all tasks)
30 if(debug) printf("\n\n >> AddTaskJetFlow <<\n");
31 TString fileName = AliAnalysisManager::GetCommonFileName();
32 fileName += ":JetFlow";
33 if(debug) printf(" - filename: %s \n",fileName.Data());
34 AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
36 if(debug) cout << " Fatal error: no analysis manager found! " << endl;
39 if (!mgr->GetInputEventHandler()) {
40 if(debug) cout << " Fatal error: no imput event handler found!" << endl;
43 AliAnalysisTaskRhoVnModulation* rhoTask = t;
44 // check the centrality setup
45 TArrayI* cent(rhoTask->GetCentralityClasses());
46 // add the tasks in a loop, one task for each centrality bin
47 for(Int_t i(0); i < cent->GetSize()-1; i++) {
48 TString tempName(Form("%s_%i_%i", name.Data(), cent->At(i), cent->At(i+1)));
50 AliAnalysisTaskJetFlow* task = new AliAnalysisTaskJetFlow(
59 task->SetCCMinPt(CCMinPt);
60 task->SetCCMaxPt(CCMaxPt);
61 task->SetCCBinsInPt(CCBinsInPt);
62 task->SetPtBins(ptArray); // if passed as NULL default a sane default is provided
63 (debug) ? task->SetDebugMode(1) : task->SetDebugMode(-1);
65 if(debug) cout << " --> Unexpected error occurred: NO TASK WAS CREATED! (could be a library problem!) " << endl;
68 else printf(" > added task with name %s and jet collection %s < \n", tempName.Data(), rhoTask->GetJetsName().Data());
69 // pass specific objects and settigns to the task
70 task->SetMinMaxCentrality(cent->At(i), cent->At(1+i));
72 mgr->ConnectInput(task,0,mgr->GetCommonInputContainer());
73 mgr->ConnectOutput(task,1,mgr->CreateContainer(Form("%s_histograms", tempName.Data()), TList::Class(), AliAnalysisManager::kOutputContainer, fileName.Data()));
74 // connect flow anaysis task
75 Bool_t slotTwoFilled(kFALSE);
77 AliAnalysisDataContainer *flowEvent_VZERO = mgr->CreateContainer(Form("flowEvent_VZERO_%s", tempName.Data()), AliFlowEventSimple::Class(), AliAnalysisManager::kExchangeContainer);
78 mgr->ConnectOutput(task, 2, flowEvent_VZERO);
79 slotTwoFilled = kTRUE;
82 AliAnalysisDataContainer *flowEvent_TPC = mgr->CreateContainer(Form("flowEvent_TPC_%s", tempName.Data()), AliFlowEventSimple::Class(), AliAnalysisManager::kExchangeContainer);
83 (slotTwoFilled) ? mgr->ConnectOutput(task, 3, flowEvent_TPC) : mgr->ConnectOutput(task, 2, flowEvent_TPC);
85 if(FlowPackageSP) TaskJetFlow::AddSPmethod(Form("SPVZERO_A_%s", tempName.Data()), "Qa", 2, flowEvent_VZERO);
86 if(FlowPackageSP) TaskJetFlow::AddSPmethod(Form("SPVZERO_B_%s", tempName.Data()), "Qb", 2, flowEvent_VZERO);
87 if(FlowPackageQC) TaskJetFlow::AddQCmethod(Form("QC_%s", tempName.Data()), 2, flowEvent_TPC);
91 //_____________________________________________________________________________
92 namespace TaskJetFlow{ // use unique namespace to avoid problems in TRAIN analysis
93 void AddSPmethod(char *name, char *Qvector, int harmonic, AliAnalysisDataContainer *flowEvent)
95 TString fileName = AliAnalysisManager::GetCommonFileName();
97 AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
98 AliAnalysisDataContainer *outSP = mgr->CreateContainer(name, TList::Class(), AliAnalysisManager::kOutputContainer, fileName);
99 AliAnalysisTaskScalarProduct *tskSP = new AliAnalysisTaskScalarProduct(Form("TaskScalarProduct_%s", name), kFALSE);
100 tskSP->SetApplyCorrectionForNUA(kTRUE);
101 tskSP->SetHarmonic(harmonic);
102 if(!strcmp("Qa", Qvector)) tskSP->SetTotalQvector("Qa");
103 if(!strcmp("Qb", Qvector)) tskSP->SetTotalQvector("Qb");
104 tskSP->SetBookOnlyBasicCCH(kFALSE);
106 mgr->ConnectInput(tskSP, 0, flowEvent);
107 mgr->ConnectOutput(tskSP, 1, outSP);
109 //_____________________________________________________________________________
110 void AddQCmethod(char *name, int harmonic, AliAnalysisDataContainer *flowEvent)
112 TString fileName = AliAnalysisManager::GetCommonFileName();
114 AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
115 AliAnalysisDataContainer *outQC = mgr->CreateContainer(name, TList::Class(), AliAnalysisManager::kOutputContainer, fileName);
116 AliAnalysisTaskQCumulants *tskQC = new AliAnalysisTaskQCumulants(Form("TaskQCumulants_%s", name), kFALSE);
117 tskQC->SetApplyCorrectionForNUA(kTRUE);
118 tskQC->SetHarmonic(harmonic);
119 tskQC->SetBookOnlyBasicCCH(kFALSE);
121 mgr->ConnectInput(tskQC, 0, flowEvent);
122 mgr->ConnectOutput(tskQC, 1, outQC);
124 //_____________________________________________________________________________
125 }// end of namespace TaskJetFlow