]>
Commit | Line | Data |
---|---|---|
a65a7e70 | 1 | // now in options |
2 | ||
3 | //PID config | |
4 | Bool_t kUseNSigmaPID = kFALSE; | |
5 | Double_t nSigmaMax = 3.0; | |
6 | Bool_t kUseBayesianPID = kTRUE; | |
7 | Double_t gMinAcceptedProbability = 0.7; | |
8 | ||
9 | //_________________________________________________________// | |
10 | AliAnalysisTaskTriggeredBF *AddTaskBalanceTriggered(Double_t centrMin=0., | |
11 | Double_t centrMax=80., | |
12 | Bool_t gRunShuffling=kFALSE, | |
13 | Bool_t gRunMixing=kFALSE, | |
14 | Bool_t gRunV0=kFALSE, | |
15 | TString centralityEstimator="V0M", | |
16 | Double_t vertexZ=10., | |
17 | Double_t DCAxy=-1, | |
18 | Double_t DCAz=-1, | |
19 | Double_t ptMin=0.15, | |
20 | Double_t ptMax=20, | |
21 | Double_t etaMin=-0.8, | |
22 | Double_t etaMax=0.8, | |
23 | Double_t maxTPCchi2 = -1, | |
24 | Int_t minNClustersTPC = -1, | |
25 | Bool_t kUsePID = kFALSE, | |
26 | Int_t AODfilterBit = 128, | |
27 | Bool_t bCentralTrigger = kFALSE, | |
28 | TString fileNameBase="AnalysisResults") { | |
29 | ||
30 | // Creates a balance function analysis task and adds it to the analysis manager. | |
31 | // Get the pointer to the existing analysis manager via the static access method. | |
32 | TString centralityName(""); | |
33 | centralityName+=Form("%.0f",centrMin); | |
34 | centralityName+="-"; | |
35 | centralityName+=Form("%.0f",centrMax); | |
36 | centralityName+="_"; | |
37 | centralityName+=Form("%s",centralityEstimator.Data()); | |
38 | centralityName+="_"; | |
39 | centralityName+=Form("vZ%.1f",vertexZ); | |
40 | centralityName+="_"; | |
41 | centralityName+=Form("DCAxy%.1f",DCAxy); | |
42 | centralityName+="_"; | |
43 | centralityName+=Form("DCAz%.1f",DCAz); | |
44 | centralityName+="_Pt"; | |
45 | centralityName+=Form("%.1f",ptMin); | |
46 | centralityName+="-"; | |
47 | centralityName+=Form("%.1f",ptMax); | |
48 | centralityName+="_Eta"; | |
49 | centralityName+=Form("%.1f",etaMin); | |
50 | centralityName+="-"; | |
51 | centralityName+=Form("%.1f",etaMax); | |
52 | centralityName+="_Chi"; | |
53 | centralityName+=Form("%.1f",maxTPCchi2); | |
54 | centralityName+="_nClus"; | |
55 | centralityName+=Form("%d",minNClustersTPC); | |
56 | centralityName+="_Bit"; | |
57 | centralityName+=Form("%d",AODfilterBit); | |
58 | if(bCentralTrigger) centralityName+="_withCentralTrigger"; | |
59 | if(gRunV0) centralityName+="_V0"; | |
60 | ||
61 | ||
62 | ||
63 | TString outputFileName(fileNameBase); | |
64 | outputFileName.Append(".root"); | |
65 | ||
66 | //=========================================================================== | |
67 | AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager(); | |
68 | if (!mgr) { | |
69 | ::Error("AddTaskTriggeredBF", "No analysis manager to connect to."); | |
70 | return NULL; | |
71 | } | |
72 | ||
73 | // Check the analysis type using the event handlers connected to the analysis manager. | |
74 | //=========================================================================== | |
75 | if (!mgr->GetInputEventHandler()) { | |
76 | ::Error("AddTaskTriggeredBF", "This task requires an input event handler"); | |
77 | return NULL; | |
78 | } | |
79 | TString analysisType = mgr->GetInputEventHandler()->GetDataType(); // can be "ESD" or "AOD" | |
80 | if(dynamic_cast<AliMCEventHandler*> (AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler())) analysisType = "MC"; | |
81 | ||
82 | // setup the balance function objects | |
83 | AliBalanceTriggered *bf = 0; // Balance Function object | |
84 | AliBalanceTriggered *bfs = 0; // shuffled Balance function object | |
85 | AliBalanceTriggered *bfm = 0; // mixing Balance function object | |
86 | ||
87 | if (analysisType=="AOD"){ | |
88 | ||
89 | bf = new AliBalanceTriggered(); | |
90 | bf->SetAnalysisLevel(analysisType); | |
91 | bf->InitHistograms(); | |
92 | ||
93 | if(gRunShuffling){ | |
94 | bfs = new AliBalanceTriggered(); | |
95 | bfs->SetAnalysisLevel(analysisType); | |
96 | bfs->InitHistograms(); | |
97 | } | |
98 | if(gRunMixing){ | |
99 | bfm = new AliBalanceTriggered(); | |
100 | bfm->SetAnalysisLevel(analysisType); | |
101 | bfm->InitHistograms(); | |
102 | } | |
103 | } | |
104 | else{ | |
105 | ::Error("AddTaskTriggeredBF", "analysis type NOT supported."); | |
106 | return NULL; | |
107 | } | |
108 | ||
109 | // Create the task, add it to manager and configure it. | |
110 | //=========================================================================== | |
111 | AliAnalysisTaskTriggeredBF *taskTriggeredBF = new AliAnalysisTaskTriggeredBF("TaskTriggeredBF"); | |
112 | taskTriggeredBF->SetAnalysisObject(bf); | |
113 | if(gRunShuffling) taskTriggeredBF->SetShufflingObject(bfs); | |
114 | if(gRunMixing){ | |
115 | taskTriggeredBF->SetMixingObject(bfm); | |
116 | taskTriggeredBF->SetMixingTracks(50000); | |
117 | } | |
118 | if(gRunV0){ | |
119 | taskTriggeredBF->SetRunV0(kTRUE); | |
120 | } | |
121 | ||
122 | taskTriggeredBF->SetCentralityPercentileRange(centrMin,centrMax); | |
123 | if(analysisType == "AOD") { | |
124 | // pt and eta cut (pt_min, pt_max, eta_min, eta_max) | |
125 | taskTriggeredBF->SetAODtrackCutBit(AODfilterBit); | |
126 | taskTriggeredBF->SetKinematicsCutsAOD(ptMin,ptMax,etaMin,etaMax); | |
127 | ||
128 | // set extra DCA cuts (-1 no extra cut) | |
129 | taskTriggeredBF->SetExtraDCACutsAOD(DCAxy,DCAz); | |
130 | ||
131 | // set extra TPC chi2 / nr of clusters cut | |
132 | taskTriggeredBF->SetExtraTPCCutsAOD(maxTPCchi2, minNClustersTPC); | |
133 | ||
134 | } | |
135 | ||
136 | // offline trigger selection (AliVEvent.h) | |
137 | // taskTriggeredBF->UseOfflineTrigger(); // NOT used (selection is done with the AliAnalysisTaskSE::SelectCollisionCandidates()) | |
138 | // with this only selected events are analyzed (first 2 bins in event QA histogram are the same)) | |
139 | // documentation in https://twiki.cern.ch/twiki/bin/viewauth/ALICE/PWG1EvSelDocumentation | |
140 | if(bCentralTrigger) taskTriggeredBF->SelectCollisionCandidates(AliVEvent::kMB | AliVEvent::kCentral | AliVEvent::kSemiCentral); | |
141 | else taskTriggeredBF->SelectCollisionCandidates(AliVEvent::kMB); | |
142 | ||
143 | // centrality estimator (default = V0M) | |
144 | taskTriggeredBF->SetCentralityEstimator(centralityEstimator); | |
145 | ||
146 | // vertex cut (x,y,z) | |
147 | taskTriggeredBF->SetVertexDiamond(.3,.3,vertexZ); | |
148 | ||
149 | //bf->PrintAnalysisSettings(); | |
150 | mgr->AddTask(taskTriggeredBF); | |
151 | ||
152 | // Create ONLY the output containers for the data produced by the task. | |
153 | // Get and connect other common input/output containers via the manager as below | |
154 | //============================================================================== | |
155 | TString outputFileName = AliAnalysisManager::GetCommonFileName(); | |
156 | outputFileName += ":PWGCFEbyE.outputBalanceFunctionTriggeredAnalysis"; | |
157 | AliAnalysisDataContainer *coutQA = mgr->CreateContainer(Form("listQA_%s",centralityName.Data()), TList::Class(),AliAnalysisManager::kOutputContainer,outputFileName.Data()); | |
158 | AliAnalysisDataContainer *coutTriggeredBF = mgr->CreateContainer(Form("listTriggeredBF_%s",centralityName.Data()), TList::Class(),AliAnalysisManager::kOutputContainer,outputFileName.Data()); | |
159 | if(gRunShuffling) AliAnalysisDataContainer *coutTriggeredBFS = mgr->CreateContainer(Form("listTriggeredBFShuffled_%s",centralityName.Data()), TList::Class(),AliAnalysisManager::kOutputContainer,outputFileName.Data()); | |
160 | if(gRunMixing) AliAnalysisDataContainer *coutTriggeredBFM = mgr->CreateContainer(Form("listTriggeredBFMixed_%s",centralityName.Data()), TList::Class(),AliAnalysisManager::kOutputContainer,outputFileName.Data()); | |
161 | if(gRunV0) AliAnalysisDataContainer *coutQAV0 = mgr->CreateContainer(Form("listQAV0_%s",centralityName.Data()), TList::Class(),AliAnalysisManager::kOutputContainer,outputFileName.Data()); | |
162 | ||
163 | mgr->ConnectInput(taskTriggeredBF, 0, mgr->GetCommonInputContainer()); | |
164 | mgr->ConnectOutput(taskTriggeredBF, 1, coutQA); | |
165 | mgr->ConnectOutput(taskTriggeredBF, 2, coutTriggeredBF); | |
166 | if(gRunShuffling) mgr->ConnectOutput(taskTriggeredBF, 3, coutTriggeredBFS); | |
167 | if(gRunMixing) mgr->ConnectOutput(taskTriggeredBF, 4, coutTriggeredBFM); | |
168 | if(gRunV0) mgr->ConnectOutput(taskTriggeredBF, 5, coutQAV0); | |
169 | ||
170 | return taskTriggeredBF; | |
171 | } |