]>
Commit | Line | Data |
---|---|---|
1db1733e | 1 | |
2 | /************************************************************************************************* | |
3 | *** Add Fragmentation Function Task *** | |
4 | ************************************************************************************************** | |
44b235ce | 5 | The fragmentation function task expects an ESD filter and jet finder running before this task. |
1db1733e | 6 | Or it runs on delta-AODs filled with filtered tracks and jets before. |
7 | ||
8 | ** Parameters ** | |
9 | (char) recJetsBranch: branch in AOD for (reconstructed) jets | |
10 | (char) genJetsBranch: branch in AOD for (generated) jets | |
11 | (char) jetType: "AOD" jets from recJetsBranch | |
12 | "AODMC" jets from genJetsBranch | |
13 | "KINE" jets from PYCELL | |
14 | +"b" (e.g. "AODb") jets with acceptance cuts | |
15 | (char) trackType: "AOD" reconstructed tracks from AOD filled by ESD filter (choose filter mask!) | |
16 | "AODMC" MC tracks from AOD filled by kine filter | |
17 | "KINE" kine particles from MC event | |
18 | +"2" (e.g. "AOD2") charged tracks only | |
19 | +"b" (e.g. "AOD2b") with acceptance cuts | |
20 | (UInt_t) filterMask: select filter bit of ESD filter task | |
21 | ||
22 | ***************************************************************************************************/ | |
23 | ||
44b235ce | 24 | AliAnalysisTaskFragmentationFunction *AddTaskFragmentationFunction(UInt_t iFlag=1, UInt_t filterMask=32, Int_t eventClass=0){ |
1db1733e | 25 | |
26 | AliAnalysisTaskFragmentationFunction *ff=0; | |
27 | ||
c2658535 | 28 | // UA1 Jets |
1db1733e | 29 | // only reconstructed (default) |
44b235ce | 30 | if(iFlag&(1<<0)) ff = AddTaskFragmentationFunction("jetsAOD_UA1", "", "", "", "", filterMask, 0.4,0,1000., eventClass); |
ac017ff3 | 31 | // charged MC tracks and jets |
44b235ce | 32 | if(iFlag&(1<<1)) ff = AddTaskFragmentationFunction("jetsAOD_UA1", "", "jetsAODMC2_UA1", "AODMC", "AODMC2", filterMask, 0.4,0,1000., eventClass); |
ac017ff3 | 33 | // charged MC tracks and jets with acceptance cuts |
44b235ce | 34 | if(iFlag&(1<<2)) ff = AddTaskFragmentationFunction("jetsAOD_UA1", "", "jetsAODMC2_UA1", "AODMCb", "AODMC2b", filterMask, 0.4,0,1000., eventClass); |
1db1733e | 35 | // kine tracks in acceptance, pythia jets in acceptance |
44b235ce | 36 | if(iFlag&(1<<3)) ff = AddTaskFragmentationFunction("jetsAOD_UA1", "", "", "KINEb", "KINEb", filterMask, 0.4,0,1000., eventClass); |
1db1733e | 37 | // reconstructed charged tracks after cuts, MC jets in acceptance |
44b235ce | 38 | if(iFlag&(1<<4)) ff = AddTaskFragmentationFunction("jetsAOD_UA1", "", "jetsMC2b", "AODMCb", "AOD2b", filterMask, 0.4,0,1000., eventClass); |
cfd02305 | 39 | // reconstruction efficiency: pointing with rec jet axis into gen tracks |
44b235ce | 40 | if(iFlag&(1<<5)) ff = AddTaskFragmentationFunction("jetsAOD_UA1", "", "jetsAODMC2_UA1", "AODb", "AODMC2b", filterMask, 0.4,0,1000., eventClass); |
c2658535 | 41 | |
c2658535 | 42 | |
c2658535 | 43 | |
a5f1b959 | 44 | // Jet background subtracted |
45 | // anti-kt, pt cut 1 GeV | |
46 | if(iFlag&(1<<20)) ff = AddTaskFragmentationFunction("clustersAOD_ANTIKT", "", "", "", "", filterMask, 0.4,2,1000.,eventClass, "_Skip00"); | |
47 | // anti-kt, pt cut 2 GeV | |
48 | if(iFlag&(1<<21)) ff = AddTaskFragmentationFunction("clustersAOD_ANTIKT", "", "", "", "", filterMask, 0.4,2,2000.,eventClass, "_Skip00"); | |
49 | // anti-kt, pt cut 150 MeV | |
50 | if(iFlag&(1<<22)) ff = AddTaskFragmentationFunction("clustersAOD_ANTIKT", "", "", "", "", filterMask, 0.2,2,150.,eventClass, "_Skip00"); | |
51 | ||
52 | ||
44b235ce | 53 | // Jet background subtracted |
a5f1b959 | 54 | if(iFlag&(1<<23)) ff = AddTaskFragmentationFunction("clustersAOD_ANTIKT", "", "", "", "", filterMask, 0.4,2,150.,eventClass, "_Skip00"); |
44b235ce | 55 | // charged MC tracks and jets |
a5f1b959 | 56 | if(iFlag&(1<<24)) ff = AddTaskFragmentationFunction("clustersAOD_ANTIKT", "", "jetsAODMC2_FASTJET", "AODMC", "AODMC2", filterMask, 0.4,2,150.,eventClass, "_Skip00"); |
44b235ce | 57 | // charged MC tracks and jets with acceptance cuts |
a5f1b959 | 58 | if(iFlag&(1<<25)) ff = AddTaskFragmentationFunction("clustersAOD_ANTIKT", "", "jetsAODMC2_FASTJET", "AODMCb", "AODMC2b", filterMask, 0.4,2,150., eventClass, "_Skip00"); |
c2658535 | 59 | |
a5f1b959 | 60 | if(iFlag&(1<<26)) ff = AddTaskFragmentationFunction("clustersAOD_ANTIKT", "", "", "", "", filterMask, 0.4,1,150.,eventClass, "_Skip00"); |
b101fa2e | 61 | |
34af6526 | 62 | if(iFlag&(1<<27)) ff = AddTaskFragmentationFunction("clustersAOD_ANTIKT", "", "", "", "", filterMask, 0.4,3,150.,eventClass, "_Skip00"); |
b101fa2e | 63 | |
64 | // SISCONE | |
65 | if(iFlag&(1<<28)) ff = AddTaskFragmentationFunction("jetsAOD_SISCONE", "", "", "", "", filterMask, 0.4,1,150.,eventClass); | |
66 | if(iFlag&(1<<29)) ff = AddTaskFragmentationFunction("jetsAOD_SISCONE", "", "", "", "", filterMask, 0.4,2,150.,eventClass); | |
67 | if(iFlag&(1<<30)) ff = AddTaskFragmentationFunction("jetsAOD_SISCONE", "", "", "", "", filterMask, 0.4,3,150.,eventClass); | |
68 | ||
1db1733e | 69 | return ff; |
70 | } | |
71 | ||
72 | // _______________________________________________________________________________________ | |
73 | ||
1aa4f09f | 74 | AliAnalysisTaskFragmentationFunction *AddTaskFragmentationFunctionAllCent( |
75 | const char* recJetsBranch, | |
76 | const char* recJetsBackBranch, | |
77 | const char* genJetsBranch, | |
78 | const char* jetType, | |
79 | const char* trackType, | |
80 | UInt_t filterMask, | |
81 | Float_t radius, | |
82 | int kBackgroundMode, | |
83 | float PtTrackMin, | |
84 | TString BrOpt="", | |
85 | TString BrOpt2="", | |
86 | Float_t radiusBckg=0.4) | |
87 | { | |
88 | ||
89 | // adds task with given configuration for all centralities | |
90 | ||
91 | AliAnalysisTaskFragmentationFunction *ff=0; | |
92 | ||
93 | for(Int_t eventClass=1; eventClass<=4; eventClass++){ | |
94 | ||
95 | ff = AddTaskFragmentationFunction(recJetsBranch, | |
96 | recJetsBackBranch, | |
97 | genJetsBranch, | |
98 | jetType, | |
99 | trackType, | |
100 | filterMask, | |
101 | radius, | |
102 | kBackgroundMode, | |
103 | PtTrackMin, | |
104 | eventClass, | |
105 | BrOpt, | |
106 | BrOpt2, | |
107 | radiusBckg); | |
108 | } | |
109 | ||
110 | return ff; | |
111 | } | |
112 | ||
113 | // _______________________________________________________________________________________ | |
114 | ||
1db1733e | 115 | AliAnalysisTaskFragmentationFunction *AddTaskFragmentationFunction( |
116 | const char* recJetsBranch, | |
44b235ce | 117 | const char* recJetsBackBranch, |
1db1733e | 118 | const char* genJetsBranch, |
119 | const char* jetType, | |
120 | const char* trackType, | |
44b235ce | 121 | UInt_t filterMask, |
122 | Float_t radius, | |
123 | int kBackgroundMode, | |
124 | float PtTrackMin, | |
125 | Int_t eventClass=0, | |
126 | TString BrOpt="", | |
127 | TString BrOpt2="", | |
128 | Float_t radiusBckg=0.4) | |
6bec8236 | 129 | { |
1db1733e | 130 | // Creates a fragmentation function task, |
131 | // configures it and adds it to the analysis manager. | |
132 | ||
133 | //****************************************************************************** | |
134 | //*** Configuration Parameter ************************************************** | |
135 | //****************************************************************************** | |
136 | ||
137 | // space for configuration parameter: histo bin, cuts, ... | |
138 | // so far only default parameter used | |
139 | ||
140 | Int_t debug = -1; // debug level, -1: not set here | |
6bec8236 | 141 | |
1db1733e | 142 | //****************************************************************************** |
143 | ||
144 | ||
145 | ||
6bec8236 | 146 | // Get the pointer to the existing analysis manager via the static access method. |
147 | //============================================================================== | |
148 | AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager(); | |
149 | if (!mgr) { | |
1db1733e | 150 | ::Error("AddTaskFragmentationFunction", "No analysis manager to connect to."); |
151 | return NULL; | |
152 | } | |
6bec8236 | 153 | |
154 | // Check the analysis type using the event handlers connected to the analysis manager. | |
155 | //============================================================================== | |
156 | if (!mgr->GetInputEventHandler()) { | |
1db1733e | 157 | ::Error("AddTaskFragmentationFunction", "This task requires an input event handler"); |
158 | return NULL; | |
6bec8236 | 159 | } |
160 | ||
1db1733e | 161 | TString type = mgr->GetInputEventHandler()->GetDataType(); // can be "ESD" or "AOD" |
162 | Printf("Data Type: %s", type.Data()); | |
163 | ||
44b235ce | 164 | TString branchRecBackJets(recJetsBackBranch); |
1db1733e | 165 | TString branchRecJets(recJetsBranch); |
166 | TString branchGenJets(genJetsBranch); | |
167 | TString typeJets(jetType); | |
168 | TString typeTracks(trackType); | |
169 | ||
44b235ce | 170 | if(branchRecBackJets.Length()==0) branchRecBackJets = "noRecBackJets"; |
1db1733e | 171 | if(branchRecJets.Length()==0) branchRecJets = "noRecJets"; |
172 | if(branchGenJets.Length()==0) branchGenJets = "noGenJets"; | |
173 | if(typeTracks.Length()==0) typeTracks = "trackTypeUndef"; | |
174 | if(typeJets.Length()==0) typeJets = "jetTypeUndef"; | |
175 | ||
6bec8236 | 176 | // Create the task and configure it. |
177 | //=========================================================================== | |
1db1733e | 178 | |
179 | AliAnalysisTaskFragmentationFunction *task = new AliAnalysisTaskFragmentationFunction( | |
44b235ce | 180 | Form("Fragmentation Function %s %s %s %s", branchRecJets.Data(), branchGenJets.Data(), typeJets.Data(), typeTracks.Data())); |
1db1733e | 181 | |
182 | if(debug>=0) task->SetDebugLevel(debug); | |
6bec8236 | 183 | |
1db1733e | 184 | Printf("Rec Jets %s", branchRecJets.Data()); |
44b235ce | 185 | Printf("Back Rec Jets %s", branchRecBackJets.Data()); |
1db1733e | 186 | Printf("Gen Jets %s", branchGenJets.Data()); |
187 | Printf("Jet Type %s", typeJets.Data()); | |
188 | Printf("Track Type %s", typeTracks.Data()); | |
6bec8236 | 189 | |
44b235ce | 190 | // attach the filter mask and options |
191 | TString cAdd = ""; | |
192 | cAdd += Form("%02d",(int)((radius+0.01)*10.)); | |
193 | cAdd += Form("_B%d",(int)((kBackgroundMode))); | |
194 | cAdd += Form("_Filter%05d",filterMask); | |
195 | cAdd += Form("_Cut%05d",(int)((PtTrackMin))); | |
196 | cAdd += Form("%s",BrOpt.Data()); | |
197 | cAdd += Form("%s",BrOpt2.Data()); | |
198 | Printf("%s",cAdd.Data()); | |
199 | ||
200 | TString cAddb = ""; | |
201 | cAddb += Form("%02d",(int)((radiusBckg+0.01)*10.)); | |
202 | cAddb += Form("_B%d",(int)((kBackgroundMode))); | |
203 | cAddb += Form("_Filter%05d",filterMask); | |
204 | cAddb += Form("_Cut%05d",(int)((PtTrackMin))); | |
205 | cAddb += Form("%s",BrOpt.Data()); | |
206 | cAddb += Form("%s",BrOpt2.Data()); | |
207 | Printf("%s",cAddb.Data()); | |
208 | ||
209 | TString cAddmc = ""; | |
210 | cAddmc += Form("%02d",(int)((radius+0.01)*10.)); | |
211 | cAddmc += Form("_B%d",(int)((kBackgroundMode))); | |
212 | cAddmc += Form("_Filter%05d",filterMask); | |
213 | cAddmc += Form("_Cut%05d",(int)((PtTrackMin))); | |
214 | Printf("%s",cAddmc.Data()); | |
215 | ||
216 | ||
217 | if(branchRecJets.Contains("AOD")&&branchRecJets.Contains("jets")&&!branchRecJets.Contains("MC"))branchRecJets = branchRecJets + cAdd; | |
218 | if(branchRecJets.Contains("AOD")&&branchRecJets.Contains("cluster")&&!branchRecJets.Contains("MC"))branchRecJets = branchRecJets + cAdd; | |
219 | ||
220 | if(branchRecBackJets.Contains("back")&&branchRecBackJets.Contains("cluster")&&!branchRecBackJets.Contains("MC"))branchRecBackJets = branchRecBackJets + cAddb; | |
221 | ||
222 | if(branchGenJets.Contains("AOD")&&branchGenJets.Contains("MC"))branchGenJets = branchGenJets + cAddmc; | |
223 | ||
224 | Printf("Gen jets branch %s: ", branchGenJets.Data()); | |
225 | Printf("Rec jets branch %s: ", branchRecJets.Data()); | |
226 | Printf("Jet backg branch %s: ", branchRecBackJets.Data()); | |
f4132e7d | 227 | |
1db1733e | 228 | if(!branchRecJets.Contains("noRecJets")) task->SetBranchRecJets(branchRecJets); |
44b235ce | 229 | if(!branchRecBackJets.Contains("noRecBackJets")) task->SetBranchRecBackJets(branchRecBackJets); |
1db1733e | 230 | if(!branchGenJets.Contains("noGenJets")) task->SetBranchGenJets(branchGenJets); |
6bec8236 | 231 | |
232 | ||
1db1733e | 233 | if(typeTracks.Contains("AODMC2b")) task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackAODMCChargedAcceptance); |
234 | else if(typeTracks.Contains("AODMC2")) task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackAODMCCharged); | |
235 | else if(typeTracks.Contains("AODMC")) task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackAODMCAll); | |
236 | else if(typeTracks.Contains("KINE2b")) task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackKineChargedAcceptance); | |
237 | else if(typeTracks.Contains("KINE2")) task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackKineCharged); | |
238 | else if(typeTracks.Contains("KINE")) task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackKineAll); | |
239 | else if(typeTracks.Contains("AODb")) task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackAODCuts); | |
240 | else if(typeTracks.Contains("AOD")) task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackAOD); | |
241 | else if(typeTracks.Contains("trackTypeUndef")) task->SetTrackTypeGen(0); // undefined | |
242 | else Printf("trackType %s not found", typeTracks.Data()); | |
243 | ||
244 | if(typeJets.Contains("AODMCb")) task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsGenAcceptance); | |
245 | else if(typeJets.Contains("AODMC")) task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsGen); | |
246 | else if(typeJets.Contains("KINEb")) task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsKineAcceptance); | |
247 | else if(typeJets.Contains("KINE")) task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsKine); | |
248 | else if(typeJets.Contains("AODb")) task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsRecAcceptance); | |
249 | else if(typeJets.Contains("AOD")) task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsRec); | |
250 | else if(typeJets.Contains("jetTypeUndef")) task->SetJetTypeGen(0); // undefined | |
251 | else Printf("jetType %s not found", typeJets.Data()); | |
252 | ||
7d9e8b32 | 253 | if(typeJets.Contains("AODMCb")) task->SetJetTypeRecEff(AliAnalysisTaskFragmentationFunction::kJetsGenAcceptance); // kJetsRecAcceptance |
254 | else if(typeJets.Contains("AODb")) task->SetJetTypeRecEff(AliAnalysisTaskFragmentationFunction::kJetsRecAcceptance); | |
255 | else task->SetJetTypeRecEff(0); | |
256 | ||
1db1733e | 257 | task->SetFilterMask(filterMask); |
44b235ce | 258 | task->SetEventClass(eventClass); |
1db1733e | 259 | |
f3564520 | 260 | // Set default parameters |
261 | // Cut selection | |
a5f1b959 | 262 | |
263 | if((int)PtTrackMin == 2000) task->SetTrackCuts(2.0, -0.9, 0.9, 0., 2*TMath::Pi()); | |
264 | else if((int)PtTrackMin == 1000) task->SetTrackCuts(1.0, -0.9, 0.9, 0., 2*TMath::Pi()); | |
265 | else task->SetTrackCuts(); // default : pt > 0.150 GeV, |eta|<0.9, full phi acc | |
266 | ||
267 | ||
1db1733e | 268 | task->SetJetCuts(); // default: jet pt > 5 GeV, |eta|<0.5, full phi acc |
f3564520 | 269 | task->SetDiJetCuts(); // default: type of cut = 1 (cut in deltaPhi), deltaPhi = 0., cdf = 0.5, fraction of pt = 0.6 |
270 | task->SetKindSlices(); // default: kindSlice = 1 (inv mass) | |
2e06aa40 | 271 | //if(radius <= 0.2) task->SetFFRadius(0.2); // R = 0.2 |
272 | //else task->SetFFRadius(); // default: R = 0.4 | |
633ed180 | 273 | task->SetFFRadius(radius); |
1da9498b | 274 | task->SetFFBckgRadius(); // default: R = 0.7 |
44b235ce | 275 | task->SetBckgSubMethod(); // default: subMethod = O, 1 = leading jet removed for rho extraction, 2 = 2 leading jets removed |
a5f1b959 | 276 | task->SetIJMode(0); // default: ijMode = 1 |
44b235ce | 277 | task->SetQAMode(); // default: qaMode = 3 |
278 | task->SetFFMode(); // default: ffMode = 1 | |
b101fa2e | 279 | task->SetDJMode(0); // default: djMode = 1 |
280 | task->SetEffMode(0); // default: effMode = 1 | |
281 | task->SetPhiCorrMode(0); // default: phiCorrMode = 1 | |
a5f1b959 | 282 | task->SetHighPtThreshold(); // default: pt > 5 Gev |
cb76764e | 283 | task->UseRecEffRecJetPtBins(); // efficiency in bins of rec/gen jet pt - default: kTRUE |
284 | ||
c08c3ad2 | 285 | task->SetBckgMode(1); // default: bgMode = 1 |
3170a3f8 | 286 | task->SetBckgType(); |
d5e8640d | 287 | task->SetBranchRecBackClusters(Form("clustersAOD_KT04_B0_Filter%05d_Cut00150_Skip00",filterMask)); |
c08c3ad2 | 288 | |
f3564520 | 289 | // Define histo bins |
a5f1b959 | 290 | task->SetFFHistoBins(23, 5, 120, 480, 0., 120.,70, 0., 7.,22, 0., 1.1); |
291 | ||
1db1733e | 292 | task->SetQAJetHistoBins(); |
293 | task->SetQATrackHistoBins(); | |
f3564520 | 294 | task->SetIJHistoBins(); |
295 | task->SetDiJetHistoBins(); | |
296 | task->SetQADiJetHistoBins(); | |
1db1733e | 297 | |
298 | mgr->AddTask(task); | |
6bec8236 | 299 | |
6bec8236 | 300 | // Create ONLY the output containers for the data produced by the task. |
301 | // Get and connect other common input/output containers via the manager as below | |
302 | //============================================================================== | |
6bec8236 | 303 | |
1db1733e | 304 | AliAnalysisDataContainer *coutput_FragFunc = mgr->CreateContainer( |
44b235ce | 305 | Form("fracfunc_%s_%s_%s_%s_cl%d", branchRecJets.Data(), branchGenJets.Data(), typeTracks.Data(), typeJets.Data(), eventClass), |
1db1733e | 306 | TList::Class(), |
307 | AliAnalysisManager::kOutputContainer, | |
44b235ce | 308 | Form("%s:PWG4_FragmentationFunction_%s_%s_%s_%s_cl%d", |
309 | AliAnalysisManager::GetCommonFileName(), branchRecJets.Data(), branchGenJets. Data(), typeTracks.Data(), typeJets.Data(), eventClass)); | |
1db1733e | 310 | |
311 | mgr->ConnectInput (task, 0, mgr->GetCommonInputContainer()); | |
312 | mgr->ConnectOutput (task, 0, mgr->GetCommonOutputContainer()); | |
313 | mgr->ConnectOutput (task, 1, coutput_FragFunc); | |
6bec8236 | 314 | |
1db1733e | 315 | return task; |
6bec8236 | 316 | } |