]>
Commit | Line | Data |
---|---|---|
d89b8229 | 1 | /************************************************************************** |
2 | * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * | |
3 | * * | |
4 | * Author: The ALICE Off-line Project. * | |
5 | * Contributors are mentioned in the code where appropriate. * | |
6 | * * | |
7 | * Permission to use, copy, modify and distribute this software and its * | |
8 | * documentation strictly for non-commercial purposes is hereby granted * | |
9 | * without fee, provided that the above copyright notice appears in all * | |
10 | * copies and that both the copyright notice and this permission notice * | |
11 | * appear in the supporting documentation. The authors make no claims * | |
12 | * about the suitability of this software for any purpose. It is * | |
13 | * provided "as is" without express or implied warranty. * | |
14 | **************************************************************************/ | |
15 | ||
16 | //---------------------------------------------------------------- | |
17 | // Analysis task for interfacing the jet finders with the analysis framework | |
18 | // | |
19 | // Author: magali.estienne@subatech.in2p3.fr | |
20 | // alexandre.shabetai@cern.ch | |
21 | // | |
22 | //---------------------------------------------------------------- | |
23 | ||
24 | #include <Riostream.h> | |
25 | #include <TROOT.h> | |
26 | #include <TSystem.h> | |
27 | #include <TInterpreter.h> | |
28 | #include <TFile.h> | |
29 | #include <TList.h> | |
30 | #include <TH1.h> | |
31 | ||
32 | #include "AliAnalysisTaskJetsFinder.h" | |
33 | #include "AliAnalysisManager.h" | |
34 | #include "AliJetFinder.h" | |
35 | #include "AliJetHeader.h" | |
36 | #include "AliJetHistos.h" | |
37 | #include "AliAODEvent.h" | |
38 | #include "AliAODJetEventBackground.h" | |
39 | #include "AliAODHandler.h" | |
40 | #include "AliAODExtension.h" | |
41 | ||
42 | ClassImp(AliAnalysisTaskJetsFinder) | |
43 | ||
44 | //////////////////////////////////////////////////////////////////////// | |
45 | ||
46 | AliAnalysisTaskJetsFinder::AliAnalysisTaskJetsFinder(): | |
47 | AliAnalysisTaskSE(), | |
48 | fConfigFile("ConfigJetAnalysis.C"), | |
49 | fNonStdBranch(""), | |
50 | fNonStdFile(""), | |
51 | fJetFinder(0x0), | |
52 | fHistos(0x0), | |
53 | fAODExtension(0x0), | |
54 | fListOfHistos(0x0), | |
55 | fTreeI(0x0), | |
56 | fEvent(0x0), | |
57 | fUseAODBackground(kFALSE), | |
58 | fFilterPt(0.) | |
59 | { | |
60 | // Default constructor | |
61 | } | |
62 | ||
63 | //---------------------------------------------------------------- | |
64 | AliAnalysisTaskJetsFinder::AliAnalysisTaskJetsFinder(const char* name): | |
65 | AliAnalysisTaskSE(name), | |
66 | fConfigFile("ConfigJetAnalysis.C"), | |
67 | fNonStdBranch(""), | |
68 | fNonStdFile(""), | |
69 | fJetFinder(0x0), | |
70 | fHistos(0x0), | |
71 | fAODExtension(0x0), | |
72 | fListOfHistos(0x0), | |
73 | fTreeI(0x0), | |
74 | fEvent(0x0), | |
75 | fUseAODBackground(kFALSE), | |
76 | fFilterPt(0.) | |
77 | { | |
78 | // Constructor 2 | |
79 | DefineInput(1, TTree::Class()); | |
80 | DefineOutput(1, TList::Class()); | |
81 | ||
82 | } | |
83 | ||
84 | //---------------------------------------------------------------- | |
85 | AliAnalysisTaskJetsFinder::~AliAnalysisTaskJetsFinder() | |
86 | { | |
87 | // destructor | |
88 | if (fHistos && ! AliAnalysisManager::GetAnalysisManager()->IsProofMode()) delete fHistos; | |
89 | if (fListOfHistos && ! AliAnalysisManager::GetAnalysisManager()->IsProofMode()) delete fListOfHistos; | |
90 | ||
91 | } | |
92 | ||
93 | //---------------------------------------------------------------- | |
94 | void AliAnalysisTaskJetsFinder::ConnectInputData(Option_t *) | |
95 | { | |
96 | // Get the exchanged tree and event | |
97 | AliAnalysisTaskSE::ConnectInputData(); | |
98 | ||
99 | fTreeI = (TTree*)GetInputData(1); | |
100 | ||
101 | char **address =(char **)GetBranchAddress(1,"AliJetCalTrkEvent"); | |
102 | if (address) { | |
103 | fEvent = (AliJetCalTrkEvent*) (*address); | |
104 | } | |
105 | else { printf("AliJetCalTrkEvent address not found, pleae check containers"); } | |
106 | ||
107 | } | |
108 | ||
109 | //---------------------------------------------------------------- | |
110 | void AliAnalysisTaskJetsFinder::UserCreateOutputObjects() | |
111 | { | |
112 | // Create the output container containt | |
113 | // | |
114 | if (fDebug > 1) printf("AnalysisTaskJets::CreateOutPutData() \n"); | |
115 | ||
116 | AliJetHeader *fH = fJetFinder->GetJetHeader(); | |
117 | ||
118 | if(fNonStdBranch.Length()==0) | |
119 | { | |
120 | // Connect default AOD to jet finder | |
121 | // create a new branch for the background | |
122 | ||
123 | if(fUseAODBackground){ | |
124 | if(!AODEvent()->FindListObject(AliAODJetEventBackground::StdBranchName())){ | |
125 | AliAODJetEventBackground* evBkg = new AliAODJetEventBackground(); | |
126 | evBkg->SetName(AliAODJetEventBackground::StdBranchName()); | |
127 | AddAODBranch("AliAODJetEventBackground",&evBkg); | |
128 | } | |
129 | } | |
130 | fJetFinder->ConnectAOD(AODEvent()); | |
131 | } | |
132 | else | |
133 | { | |
134 | // Create a new branch for jets... | |
135 | // how is this reset? -> cleared in the UserExec.... | |
136 | // Can this be handled by the framework? | |
137 | // here we can also have the case that the brnaches are written to a separate file | |
138 | ||
139 | TClonesArray *tca = new TClonesArray("AliAODJet", 0); | |
140 | tca->SetName(fNonStdBranch.Data()); | |
141 | AddAODBranch("TClonesArray",&tca,fNonStdFile.Data()); | |
142 | if(fUseAODBackground){ | |
143 | if(!AODEvent() || !AODEvent()->FindListObject(Form("%s_%s",AliAODJetEventBackground::StdBranchName(),fNonStdBranch.Data()))){ | |
144 | AliAODJetEventBackground* evBkg = new AliAODJetEventBackground(); | |
145 | evBkg->SetName(Form("%s_%s",AliAODJetEventBackground::StdBranchName(),fNonStdBranch.Data())); | |
146 | AddAODBranch("AliAODJetEventBackground",&evBkg,fNonStdFile.Data()); | |
147 | } | |
148 | } | |
149 | if(fNonStdFile.Length()!=0){ | |
150 | // | |
151 | // case that we have an AOD extension we need to fetch the jets from the extended output | |
152 | // we identifay the extension aod event by looking for the branchname | |
153 | AliAODHandler *aodH = dynamic_cast<AliAODHandler*>(AliAnalysisManager::GetAnalysisManager()->GetOutputEventHandler()); | |
154 | fAODExtension = 0; | |
155 | if(aodH){ | |
156 | TObjArray* extArray = aodH->GetExtensions(); | |
157 | if (extArray) { | |
158 | TIter next(extArray); | |
159 | while ((fAODExtension=(AliAODExtension*)next())){ | |
160 | TObject *obj = fAODExtension->GetAOD()->FindListObject(fNonStdBranch.Data()); | |
161 | if(fDebug>10){ | |
162 | Printf("%s:%d Dumping..",(char*)__FILE__,__LINE__); | |
163 | fAODExtension->GetAOD()->Dump(); | |
164 | } | |
165 | if(obj){ | |
166 | if(fDebug>1)Printf("AODExtension found for %s",fNonStdBranch.Data()); | |
167 | break; | |
168 | } | |
169 | fAODExtension = 0; | |
170 | } | |
171 | } | |
172 | } | |
173 | if(fAODExtension)fJetFinder->ConnectAODNonStd(fAODExtension->GetAOD(), fNonStdBranch.Data()); | |
174 | } | |
175 | else{ | |
176 | if (fDebug > 1) printf("Connecting Non Std Branch AOD %p %s \n",AODEvent(),fNonStdBranch.Data()); | |
177 | fJetFinder->ConnectAODNonStd(AODEvent(), fNonStdBranch.Data()); | |
178 | } | |
179 | } | |
180 | ||
181 | // do not add the histograms in the directory | |
182 | // all handled by the list | |
183 | Bool_t oldStatus = TH1::AddDirectoryStatus(); | |
184 | TH1::AddDirectory(kFALSE); | |
185 | ||
186 | // Histograms | |
187 | fListOfHistos = new TList(); | |
188 | fListOfHistos->SetOwner(); | |
189 | fHistos = new AliJetHistos(); | |
190 | fHistos->CreateHistos(); | |
191 | fHistos->AddHistosToList(fListOfHistos); | |
192 | // CDF case | |
193 | if(fJetFinder->InheritsFrom("AliCdfJetFinder")) fJetFinder->CreateOutputObjects(fListOfHistos); | |
194 | ||
195 | // Add the JetFinderInformation to the Outputlist | |
196 | ||
197 | // Compose a characteristic output name | |
198 | // with the name of the output branch | |
199 | if(fH) { | |
200 | if(fNonStdBranch.Length()==0) { | |
201 | fH->SetName("AliJetHeader_jets"); | |
202 | } | |
203 | else { | |
204 | fH->SetName(Form("AliJetHeader_%s",fNonStdBranch.Data())); | |
205 | } | |
206 | } | |
207 | ||
208 | TH1::AddDirectory(oldStatus); | |
209 | ||
210 | if(!fAODExtension)OutputTree()->GetUserInfo()->Add(fH); | |
211 | else fAODExtension->GetTree()->GetUserInfo()->Add(fH); | |
212 | ||
213 | // post | |
214 | PostData(1, fListOfHistos); | |
215 | ||
216 | } | |
217 | ||
218 | //---------------------------------------------------------------- | |
219 | void AliAnalysisTaskJetsFinder::Init() | |
220 | { | |
221 | // Initialization | |
222 | if (fDebug > 1) printf("AnalysisTaskJetsFinder::Init() \n"); | |
223 | ||
224 | // Call configuration file | |
225 | if (fConfigFile.Length()) { | |
226 | gROOT->LoadMacro(fConfigFile); | |
227 | fJetFinder = (AliJetFinder*) gInterpreter->ProcessLine("ConfigJetAnalysis()"); | |
228 | } | |
229 | ||
230 | // Initialise Jet Analysis | |
231 | fJetFinder->Init(); | |
232 | ||
233 | } | |
234 | ||
235 | ||
236 | //---------------------------------------------------------------- | |
237 | Bool_t AliAnalysisTaskJetsFinder::Notify() | |
238 | { | |
239 | if (fDebug > 1) printf("AnalysisTaskJetsFinder::Notify() \n"); | |
240 | return kTRUE; | |
241 | ||
242 | } | |
243 | ||
244 | //---------------------------------------------------------------- | |
245 | void AliAnalysisTaskJetsFinder::UserExec(Option_t */*option*/) | |
246 | { | |
247 | ||
248 | // Execute analysis for current event | |
249 | ||
250 | if (fDebug > 1) printf("AnalysisTaskJetsFinder::UserExec() \n"); | |
251 | ||
252 | TClonesArray* jarray = 0; | |
253 | AliAODJetEventBackground* evBkg = 0; | |
254 | ||
255 | // only need this once | |
256 | static AliAODHandler *aodH = dynamic_cast<AliAODHandler*>(AliAnalysisManager::GetAnalysisManager()->GetOutputEventHandler()); | |
257 | ||
258 | if(fNonStdBranch.Length()==0) { | |
259 | jarray = AODEvent()->GetJets(); | |
260 | if(fUseAODBackground){ | |
261 | evBkg = (AliAODJetEventBackground*)(AODEvent()->FindListObject(AliAODJetEventBackground::StdBranchName())); | |
262 | evBkg->Reset(); | |
263 | } | |
264 | } | |
265 | else { | |
266 | if(AODEvent())jarray = (TClonesArray*)(AODEvent()->FindListObject(fNonStdBranch.Data())); | |
267 | if(!jarray)jarray = (TClonesArray*)(fAODExtension->GetAOD()->FindListObject(fNonStdBranch.Data())); | |
268 | if(jarray)jarray->Delete(); // this is our responsibility, clear before filling again | |
269 | if(fUseAODBackground){ | |
270 | if(AODEvent())evBkg = (AliAODJetEventBackground*)(AODEvent()->FindListObject(Form("%s_%s",AliAODJetEventBackground::StdBranchName(),fNonStdBranch.Data()))); | |
271 | if(!evBkg) evBkg = (AliAODJetEventBackground*)(fAODExtension->GetAOD()->FindListObject(Form("%s_%s",AliAODJetEventBackground::StdBranchName(),fNonStdBranch.Data()))); | |
272 | if(evBkg)evBkg->Reset(); | |
273 | } | |
274 | } | |
275 | ||
276 | fTreeI->GetEntry(0); | |
277 | ||
278 | fJetFinder->SetCalTrkEvent(*fEvent); | |
279 | ||
280 | fJetFinder->ProcessEvent(); | |
281 | ||
282 | // Fill control histos | |
283 | if(jarray)fHistos->FillHistos(jarray); | |
284 | ||
285 | // Store the jet branch in the AOD | |
286 | if(jarray&&aodH&&fFilterPt>0){ | |
287 | if(jarray->GetEntries()>0){ | |
288 | AliAODJet *jet = (AliAODJet*)jarray->At(0); | |
289 | if(jet->Pt()>fFilterPt){ | |
290 | aodH->SetFillAOD(kTRUE); | |
291 | } | |
292 | } | |
293 | } | |
294 | ||
295 | // Post the data | |
296 | PostData(1, fListOfHistos); | |
297 | return; | |
298 | ||
299 | } | |
300 | ||
301 | //---------------------------------------------------------------- | |
302 | void AliAnalysisTaskJetsFinder::Terminate(Option_t */*option*/) | |
303 | { | |
304 | // Terminate analysis | |
305 | if (fDebug > 1) printf("AnalysisJets: Terminate() \n"); | |
306 | ||
307 | } | |
308 |