]>
Commit | Line | Data |
---|---|---|
4ae78bb1 | 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 | ||
803d1ab0 | 16 | /* $Id$ */ |
4ae78bb1 | 17 | |
18 | //_________________________________________________________________________ | |
19 | // A singleton. This class should be used in the analysis stage to get | |
20 | // reconstructed objects: Digits, RecPoints, TrackSegments and RecParticles, | |
21 | // instead of directly reading them from galice.root file. This container | |
22 | // ensures, that one reads Digits, made of these particular digits, RecPoints, | |
23 | // made of these particular RecPoints, TrackSegments and RecParticles. | |
24 | // This becomes non trivial if there are several identical branches, produced with | |
25 | // different set of parameters. | |
26 | // | |
27 | // An example of how to use (see also class AliPHOSAnalyser): | |
28 | // AliPHOSGetter * gime = AliPHOSGetter::GetInstance("galice.root","test") ; | |
29 | // for(Int_t irecp = 0; irecp < gime->NRecParticles() ; irecp++) | |
30 | // AliPHOSRecParticle * part = gime->RecParticle(1) ; | |
31 | // ................ | |
2a657981 | 32 | // gime->Event(event) ; // reads new event from galice.root |
4ae78bb1 | 33 | // |
2a657981 | 34 | //*-- Author: Yves Schutz (SUBATECH) & Dmitri Peressounko (RRC KI & SUBATECH) |
35 | //*-- Completely redesigned by Dmitri Peressounko March 2001 | |
4ae78bb1 | 36 | // |
37 | //*-- YS June 2001 : renamed the original AliPHOSIndexToObject and make | |
38 | //*-- systematic usage of TFolders without changing the interface | |
39 | ////////////////////////////////////////////////////////////////////////////// | |
40 | ||
4ae78bb1 | 41 | // --- ROOT system --- |
88cb7938 | 42 | |
024a7e64 | 43 | #include "TFile.h" |
4ae78bb1 | 44 | #include "TROOT.h" |
024a7e64 | 45 | #include "TSystem.h" |
88cb7938 | 46 | |
4ae78bb1 | 47 | |
48 | // --- Standard library --- | |
4ae78bb1 | 49 | |
50 | // --- AliRoot header files --- | |
024a7e64 | 51 | #include "AliESD.h" |
52 | #include "AliHeader.h" | |
53 | #include "AliMC.h" | |
e957fea8 | 54 | #include "AliPHOS.h" |
024a7e64 | 55 | #include "AliPHOSBeamTestEvent.h" |
56 | #include "AliPHOSGetter.h" | |
57 | #include "AliPHOSLoader.h" | |
88cb7938 | 58 | #include "AliRunLoader.h" |
59 | #include "AliStack.h" | |
88cb7938 | 60 | |
4ae78bb1 | 61 | ClassImp(AliPHOSGetter) |
62 | ||
88cb7938 | 63 | AliPHOSGetter * AliPHOSGetter::fgObjGetter = 0 ; |
64 | AliPHOSLoader * AliPHOSGetter::fgPhosLoader = 0; | |
65 | Int_t AliPHOSGetter::fgDebug = 0; | |
66 | ||
67 | // TFile * AliPHOSGetter::fgFile = 0 ; | |
4ae78bb1 | 68 | |
69 | //____________________________________________________________________________ | |
88cb7938 | 70 | AliPHOSGetter::AliPHOSGetter(const char* headerFile, const char* version, Option_t * openingOption) |
4ae78bb1 | 71 | { |
88cb7938 | 72 | // ctor only called by Instance() |
4ae78bb1 | 73 | |
88cb7938 | 74 | AliRunLoader* rl = AliRunLoader::GetRunLoader(version) ; |
75 | if (!rl) { | |
76 | rl = AliRunLoader::Open(headerFile, version, openingOption); | |
77 | if (!rl) { | |
78 | Fatal("AliPHOSGetter", "Could not find the Run Loader for %s - %s",headerFile, version) ; | |
79 | return ; | |
80 | } | |
81 | if (rl->GetAliRun() == 0x0) { | |
82 | rl->LoadgAlice(); | |
83 | gAlice = rl->GetAliRun(); // should be removed | |
84 | } | |
fbf811ec | 85 | } |
88cb7938 | 86 | fgPhosLoader = dynamic_cast<AliPHOSLoader*>(rl->GetLoader("PHOSLoader")); |
87 | if ( !fgPhosLoader ) | |
88 | Error("AliPHOSGetter", "Could not find PHOSLoader") ; | |
89 | else | |
90 | fgPhosLoader->SetTitle(version); | |
91 | ||
92 | ||
93 | // initialize data members | |
94 | SetDebug(0) ; | |
95 | fBTE = 0 ; | |
96 | fPrimaries = 0 ; | |
97 | fLoadingStatus = "" ; | |
d2713783 | 98 | |
99 | fESDFileName = rl->GetFileName() ; // this should be the galice.root file | |
100 | fESDFileName.ReplaceAll("galice.root", "AliESDs.root") ; | |
95635748 | 101 | fESDFile = 0 ; |
88cb7938 | 102 | } |
9bd3caba | 103 | |
88cb7938 | 104 | //____________________________________________________________________________ |
105 | AliPHOSGetter::~AliPHOSGetter() | |
106 | { | |
107 | // dtor | |
108 | delete fgPhosLoader ; | |
109 | fgPhosLoader = 0 ; | |
110 | delete fBTE ; | |
111 | fBTE = 0 ; | |
112 | fPrimaries->Delete() ; | |
113 | delete fPrimaries ; | |
0ef40383 | 114 | fgObjGetter = 0; |
115 | } | |
116 | ||
117 | //____________________________________________________________________________ | |
118 | void AliPHOSGetter::Reset() | |
119 | { | |
120 | // resets things in case the getter is called consecutively with different files | |
121 | // the PHOS Loader is already deleted by the Run Loader | |
122 | ||
123 | if (fPrimaries) { | |
124 | fPrimaries->Delete() ; | |
125 | delete fPrimaries ; | |
126 | } | |
127 | fgPhosLoader = 0; | |
128 | fgObjGetter = 0; | |
88cb7938 | 129 | } |
fbf811ec | 130 | |
88cb7938 | 131 | //____________________________________________________________________________ |
132 | AliPHOSClusterizer * AliPHOSGetter::Clusterizer() | |
133 | { | |
e957fea8 | 134 | // Returns pointer to the Clusterizer task |
88cb7938 | 135 | AliPHOSClusterizer * rv ; |
136 | rv = dynamic_cast<AliPHOSClusterizer *>(PhosLoader()->Reconstructioner()) ; | |
137 | if (!rv) { | |
138 | Event(0, "R") ; | |
139 | rv = dynamic_cast<AliPHOSClusterizer*>(PhosLoader()->Reconstructioner()) ; | |
140 | } | |
141 | return rv ; | |
142 | } | |
4ae78bb1 | 143 | |
88cb7938 | 144 | //____________________________________________________________________________ |
145 | TObjArray * AliPHOSGetter::CpvRecPoints() | |
146 | { | |
147 | // asks the Loader to return the CPV RecPoints container | |
4ae78bb1 | 148 | |
88cb7938 | 149 | TObjArray * rv = 0 ; |
150 | ||
151 | rv = PhosLoader()->CpvRecPoints() ; | |
152 | if (!rv) { | |
153 | PhosLoader()->MakeRecPointsArray() ; | |
154 | rv = PhosLoader()->CpvRecPoints() ; | |
155 | } | |
156 | return rv ; | |
157 | } | |
dca3a7c4 | 158 | |
88cb7938 | 159 | //____________________________________________________________________________ |
160 | TClonesArray * AliPHOSGetter::Digits() | |
161 | { | |
162 | // asks the Loader to return the Digits container | |
7bb289a7 | 163 | |
88cb7938 | 164 | TClonesArray * rv = 0 ; |
165 | rv = PhosLoader()->Digits() ; | |
7bb289a7 | 166 | |
88cb7938 | 167 | if( !rv ) { |
168 | PhosLoader()->MakeDigitsArray() ; | |
169 | rv = PhosLoader()->Digits() ; | |
4ae78bb1 | 170 | } |
88cb7938 | 171 | return rv ; |
172 | } | |
4b808d52 | 173 | |
88cb7938 | 174 | //____________________________________________________________________________ |
175 | AliPHOSDigitizer * AliPHOSGetter::Digitizer() | |
176 | { | |
e957fea8 | 177 | // Returns pointer to the Digitizer task |
88cb7938 | 178 | AliPHOSDigitizer * rv ; |
179 | rv = dynamic_cast<AliPHOSDigitizer *>(PhosLoader()->Digitizer()) ; | |
180 | if (!rv) { | |
181 | Event(0, "D") ; | |
182 | rv = dynamic_cast<AliPHOSDigitizer *>(PhosLoader()->Digitizer()) ; | |
183 | } | |
184 | return rv ; | |
4ae78bb1 | 185 | } |
fbf811ec | 186 | |
88cb7938 | 187 | |
4ae78bb1 | 188 | //____________________________________________________________________________ |
88cb7938 | 189 | TObjArray * AliPHOSGetter::EmcRecPoints() |
0bc3b8ed | 190 | { |
88cb7938 | 191 | // asks the Loader to return the EMC RecPoints container |
d489fb96 | 192 | |
88cb7938 | 193 | TObjArray * rv = 0 ; |
65549808 | 194 | |
88cb7938 | 195 | rv = PhosLoader()->EmcRecPoints() ; |
196 | if (!rv) { | |
197 | PhosLoader()->MakeRecPointsArray() ; | |
198 | rv = PhosLoader()->EmcRecPoints() ; | |
89165262 | 199 | } |
88cb7938 | 200 | return rv ; |
81bb1a45 | 201 | } |
7a9d98f9 | 202 | |
65549808 | 203 | //____________________________________________________________________________ |
88cb7938 | 204 | TClonesArray * AliPHOSGetter::TrackSegments() |
65549808 | 205 | { |
88cb7938 | 206 | // asks the Loader to return the TrackSegments container |
207 | ||
208 | TClonesArray * rv = 0 ; | |
209 | ||
210 | rv = PhosLoader()->TrackSegments() ; | |
211 | if (!rv) { | |
212 | PhosLoader()->MakeTrackSegmentsArray() ; | |
213 | rv = PhosLoader()->TrackSegments() ; | |
214 | } | |
215 | return rv ; | |
65549808 | 216 | } |
4ae78bb1 | 217 | |
0c87da39 | 218 | //____________________________________________________________________________ |
e957fea8 | 219 | AliPHOSTrackSegmentMaker * AliPHOSGetter::TrackSegmentMaker() |
88cb7938 | 220 | { |
e957fea8 | 221 | // Returns pointer to the TrackSegmentMaker task |
88cb7938 | 222 | AliPHOSTrackSegmentMaker * rv ; |
223 | rv = dynamic_cast<AliPHOSTrackSegmentMaker *>(PhosLoader()->TrackSegmentMaker()) ; | |
224 | if (!rv) { | |
225 | Event(0, "T") ; | |
226 | rv = dynamic_cast<AliPHOSTrackSegmentMaker *>(PhosLoader()->TrackSegmentMaker()) ; | |
0c87da39 | 227 | } |
88cb7938 | 228 | return rv ; |
0c87da39 | 229 | } |
230 | ||
4ae78bb1 | 231 | //____________________________________________________________________________ |
88cb7938 | 232 | TClonesArray * AliPHOSGetter::RecParticles() |
4ae78bb1 | 233 | { |
88cb7938 | 234 | // asks the Loader to return the TrackSegments container |
235 | ||
236 | TClonesArray * rv = 0 ; | |
4ae78bb1 | 237 | |
88cb7938 | 238 | rv = PhosLoader()->RecParticles() ; |
239 | if (!rv) { | |
240 | PhosLoader()->MakeRecParticlesArray() ; | |
241 | rv = PhosLoader()->RecParticles() ; | |
b134c32f | 242 | } |
88cb7938 | 243 | return rv ; |
4ae78bb1 | 244 | } |
4ae78bb1 | 245 | //____________________________________________________________________________ |
fc7e2f43 | 246 | void AliPHOSGetter::Event(Int_t event, const char* opt) |
4ae78bb1 | 247 | { |
88cb7938 | 248 | // Reads the content of all Tree's S, D and R |
548f0134 | 249 | |
88cb7938 | 250 | if ( event >= MaxEvent() ) { |
251 | Error("Event", "%d not found in TreeE !", event) ; | |
252 | return ; | |
fbf811ec | 253 | } |
4ae78bb1 | 254 | |
88cb7938 | 255 | AliRunLoader * rl = AliRunLoader::GetRunLoader(PhosLoader()->GetTitle()); |
df25f7dd | 256 | |
88cb7938 | 257 | // checks if we are dealing with test-beam data |
258 | TBranch * btb = rl->TreeE()->GetBranch("AliPHOSBeamTestEvent") ; | |
259 | if(btb){ | |
260 | if(!fBTE) | |
261 | fBTE = new AliPHOSBeamTestEvent() ; | |
262 | btb->SetAddress(&fBTE) ; | |
263 | btb->GetEntry(event) ; | |
fbf811ec | 264 | } |
88cb7938 | 265 | else{ |
266 | if(fBTE){ | |
267 | delete fBTE ; | |
268 | fBTE = 0 ; | |
fbf811ec | 269 | } |
fbf811ec | 270 | } |
88cb7938 | 271 | |
272 | // Loads the type of object(s) requested | |
b134c32f | 273 | |
88cb7938 | 274 | rl->GetEvent(event) ; |
275 | ||
276 | if( strstr(opt,"X") || (strcmp(opt,"")==0) ) | |
277 | ReadPrimaries() ; | |
278 | ||
279 | if(strstr(opt,"H") ) | |
280 | ReadTreeH(); | |
281 | ||
282 | if(strstr(opt,"S") ) | |
283 | ReadTreeS() ; | |
284 | ||
285 | if( strstr(opt,"D") ) | |
286 | ReadTreeD() ; | |
287 | ||
288 | if( strstr(opt,"R") ) | |
289 | ReadTreeR() ; | |
290 | ||
291 | if( strstr(opt,"T") ) | |
292 | ReadTreeT() ; | |
293 | ||
294 | if( strstr(opt,"P") ) | |
295 | ReadTreeP() ; | |
95635748 | 296 | |
88cb7938 | 297 | // if( strstr(opt,"Q") ) |
298 | // ReadTreeQA() ; | |
299 | ||
300 | } | |
301 | ||
302 | ||
303 | //____________________________________________________________________________ | |
304 | Int_t AliPHOSGetter::EventNumber() const | |
305 | { | |
306 | // return the current event number | |
307 | AliRunLoader * rl = AliRunLoader::GetRunLoader(PhosLoader()->GetTitle()); | |
308 | return static_cast<Int_t>(rl->GetEventNumber()) ; | |
fbf811ec | 309 | } |
d489fb96 | 310 | |
fbf811ec | 311 | //____________________________________________________________________________ |
88cb7938 | 312 | TClonesArray * AliPHOSGetter::Hits() |
fbf811ec | 313 | { |
88cb7938 | 314 | // asks the loader to return the Hits container |
315 | ||
316 | TClonesArray * rv = 0 ; | |
317 | ||
318 | rv = PhosLoader()->Hits() ; | |
319 | if ( !rv ) { | |
320 | PhosLoader()->LoadHits("read"); | |
321 | rv = PhosLoader()->Hits() ; | |
fbf811ec | 322 | } |
88cb7938 | 323 | return rv ; |
324 | } | |
fbf811ec | 325 | |
88cb7938 | 326 | //____________________________________________________________________________ |
327 | AliPHOSGetter * AliPHOSGetter::Instance(const char* alirunFileName, const char* version, Option_t * openingOption) | |
328 | { | |
329 | // Creates and returns the pointer of the unique instance | |
330 | // Must be called only when the environment has changed | |
d489fb96 | 331 | |
88cb7938 | 332 | //::Info("Instance","alirunFileName=%s version=%s openingOption=%s",alirunFileName,version,openingOption); |
333 | ||
334 | if(!fgObjGetter){ // first time the getter is called | |
335 | fgObjGetter = new AliPHOSGetter(alirunFileName, version, openingOption) ; | |
fbf811ec | 336 | } |
88cb7938 | 337 | else { // the getter has been called previously |
338 | AliRunLoader * rl = AliRunLoader::GetRunLoader(fgPhosLoader->GetTitle()); | |
339 | if ( rl->GetFileName() == alirunFileName ) {// the alirunFile has the same name | |
340 | // check if the file is already open | |
341 | TFile * galiceFile = dynamic_cast<TFile *>(gROOT->FindObject(rl->GetFileName()) ) ; | |
342 | ||
343 | if ( !galiceFile ) | |
344 | fgObjGetter = new AliPHOSGetter(alirunFileName, version, openingOption) ; | |
345 | ||
346 | else { // the file is already open check the version name | |
347 | TString currentVersionName = rl->GetEventFolder()->GetName() ; | |
348 | TString newVersionName(version) ; | |
349 | if (currentVersionName == newVersionName) | |
350 | if(fgDebug) | |
351 | ::Warning( "Instance", "Files with version %s already open", currentVersionName.Data() ) ; | |
352 | else { | |
353 | fgObjGetter = new AliPHOSGetter(alirunFileName, version, openingOption) ; | |
354 | } | |
355 | } | |
356 | } | |
7fba1747 | 357 | else { |
358 | AliRunLoader * rl = AliRunLoader::GetRunLoader(fgPhosLoader->GetTitle()) ; | |
e191bb57 | 359 | if ( strstr(version, AliConfig::GetDefaultEventFolderName()) ) // false in case of merging |
1c221c70 | 360 | delete rl ; |
88cb7938 | 361 | fgObjGetter = new AliPHOSGetter(alirunFileName, version, openingOption) ; |
7fba1747 | 362 | } |
48f12df6 | 363 | } |
88cb7938 | 364 | if (!fgObjGetter) |
95635748 | 365 | ::Error("AliPHOSGetter::Instance", "Failed to create the PHOS Getter object") ; |
88cb7938 | 366 | else |
367 | if (fgDebug) | |
368 | Print() ; | |
4ae78bb1 | 369 | |
88cb7938 | 370 | return fgObjGetter ; |
4ae78bb1 | 371 | } |
372 | ||
6ad0e528 | 373 | //____________________________________________________________________________ |
88cb7938 | 374 | AliPHOSGetter * AliPHOSGetter::Instance() |
6ad0e528 | 375 | { |
88cb7938 | 376 | // Returns the pointer of the unique instance already defined |
fbf811ec | 377 | |
95635748 | 378 | if(!fgObjGetter && fgDebug) |
379 | ::Warning("AliPHOSGetter::Instance", "Getter not initialized") ; | |
6ad0e528 | 380 | |
88cb7938 | 381 | return fgObjGetter ; |
382 | ||
6ad0e528 | 383 | } |
384 | ||
385 | //____________________________________________________________________________ | |
88cb7938 | 386 | Int_t AliPHOSGetter::MaxEvent() const |
6ad0e528 | 387 | { |
88cb7938 | 388 | // returns the number of events in the run (from TE) |
0bc3b8ed | 389 | |
88cb7938 | 390 | AliRunLoader * rl = AliRunLoader::GetRunLoader(PhosLoader()->GetTitle()); |
391 | return static_cast<Int_t>(rl->GetNumberOfEvents()) ; | |
6ad0e528 | 392 | } |
393 | ||
394 | //____________________________________________________________________________ | |
88cb7938 | 395 | TParticle * AliPHOSGetter::Primary(Int_t index) const |
6ad0e528 | 396 | { |
88cb7938 | 397 | AliRunLoader * rl = AliRunLoader::GetRunLoader(PhosLoader()->GetTitle()); |
398 | return rl->Stack()->Particle(index) ; | |
399 | } | |
6ad0e528 | 400 | |
4ae78bb1 | 401 | //____________________________________________________________________________ |
88cb7938 | 402 | AliPHOS * AliPHOSGetter:: PHOS() const |
4ae78bb1 | 403 | { |
404 | // returns the PHOS object | |
88cb7938 | 405 | AliPHOS * phos = dynamic_cast<AliPHOS*>(PhosLoader()->GetModulesFolder()->FindObject("PHOS")) ; |
7a9d98f9 | 406 | if (!phos) |
88cb7938 | 407 | if (fgDebug) |
408 | Warning("PHOS", "PHOS module not found in module folders: %s", PhosLoader()->GetModulesFolder()->GetName() ) ; | |
7a9d98f9 | 409 | return phos ; |
4ae78bb1 | 410 | } |
411 | ||
88cb7938 | 412 | |
413 | ||
414 | //____________________________________________________________________________ | |
e957fea8 | 415 | AliPHOSPID * AliPHOSGetter::PID() |
88cb7938 | 416 | { |
e957fea8 | 417 | // Returns pointer to the PID task |
88cb7938 | 418 | AliPHOSPID * rv ; |
419 | rv = dynamic_cast<AliPHOSPID *>(PhosLoader()->PIDTask()) ; | |
420 | if (!rv) { | |
421 | Event(0, "P") ; | |
422 | rv = dynamic_cast<AliPHOSPID *>(PhosLoader()->PIDTask()) ; | |
423 | } | |
424 | return rv ; | |
425 | } | |
426 | ||
4ae78bb1 | 427 | //____________________________________________________________________________ |
88cb7938 | 428 | AliPHOSGeometry * AliPHOSGetter::PHOSGeometry() const |
4ae78bb1 | 429 | { |
0bc3b8ed | 430 | // Returns PHOS geometry |
431 | ||
7a9d98f9 | 432 | AliPHOSGeometry * rv = 0 ; |
433 | if (PHOS() ) | |
434 | rv = PHOS()->GetGeometry() ; | |
435 | return rv ; | |
436 | } | |
4ae78bb1 | 437 | |
cb34a1fa | 438 | //____________________________________________________________________________ |
88cb7938 | 439 | TClonesArray * AliPHOSGetter::Primaries() |
440 | { | |
441 | // creates the Primaries container if needed | |
442 | if ( !fPrimaries ) { | |
443 | if (fgDebug) | |
444 | Info("Primaries", "Creating a new TClonesArray for primaries") ; | |
445 | fPrimaries = new TClonesArray("TParticle", 1000) ; | |
446 | } | |
447 | return fPrimaries ; | |
448 | } | |
449 | ||
450 | //____________________________________________________________________________ | |
451 | void AliPHOSGetter::Print() | |
452 | { | |
453 | // Print usefull information about the getter | |
454 | ||
455 | AliRunLoader * rl = AliRunLoader::GetRunLoader(fgPhosLoader->GetTitle()); | |
456 | ::Info( "Print", "gAlice file is %s -- version name is %s", (rl->GetFileName()).Data(), rl->GetEventFolder()->GetName() ) ; | |
457 | } | |
cb34a1fa | 458 | |
88cb7938 | 459 | //____________________________________________________________________________ |
460 | void AliPHOSGetter::ReadPrimaries() | |
461 | { | |
462 | // Read Primaries from Kinematics.root | |
cb34a1fa | 463 | |
88cb7938 | 464 | AliRunLoader * rl = AliRunLoader::GetRunLoader(PhosLoader()->GetTitle()); |
cb34a1fa | 465 | |
88cb7938 | 466 | // gets kine tree from the root file (Kinematics.root) |
7fba1747 | 467 | if ( ! rl->TreeK() ) { // load treeK the first time |
88cb7938 | 468 | rl->LoadKinematics() ; |
7fba1747 | 469 | } |
88cb7938 | 470 | |
7fba1747 | 471 | fNPrimaries = (rl->GetHeader())->GetNtrack(); |
88cb7938 | 472 | if (fgDebug) |
473 | Info( "ReadTreeK", "Found %d particles in event # %d", fNPrimaries, EventNumber() ) ; | |
cb34a1fa | 474 | |
cb34a1fa | 475 | |
88cb7938 | 476 | // first time creates the container |
477 | if ( Primaries() ) | |
478 | fPrimaries->Clear() ; | |
cb34a1fa | 479 | |
88cb7938 | 480 | Int_t index = 0 ; |
481 | for (index = 0 ; index < fNPrimaries; index++) { | |
482 | new ((*fPrimaries)[index]) TParticle(*(Primary(index))); | |
cb34a1fa | 483 | } |
cb34a1fa | 484 | } |
485 | ||
95635748 | 486 | //____________________________________________________________________________ |
487 | AliESD * AliPHOSGetter::ESD(Int_t event) | |
488 | { | |
489 | //Read the ESD | |
d2713783 | 490 | |
491 | AliESD * esd = 0 ; | |
95635748 | 492 | if (!fESDFile) |
d2713783 | 493 | if ( !OpenESDFile() ) |
494 | return esd ; | |
495 | ||
95635748 | 496 | TString esdEvent("ESD") ; |
497 | esdEvent+= event ; | |
d2713783 | 498 | esd = dynamic_cast<AliESD *>(fESDFile->Get(esdEvent)) ; |
95635748 | 499 | return esd ; |
500 | } | |
501 | ||
502 | //____________________________________________________________________________ | |
d2713783 | 503 | Bool_t AliPHOSGetter::OpenESDFile() |
95635748 | 504 | { |
d2713783 | 505 | //Open the ESD file |
95635748 | 506 | Bool_t rv = kTRUE ; |
d2713783 | 507 | if (!fESDFile) { |
508 | fESDFile = TFile::Open(fESDFileName) ; | |
509 | if (!fESDFile ) | |
510 | return kFALSE ; | |
511 | } | |
95635748 | 512 | else if (fESDFile->IsOpen()) { |
513 | fESDFile->Close() ; | |
514 | fESDFile = TFile::Open(fESDFileName) ; | |
515 | } | |
516 | if (!fESDFile->IsOpen()) | |
517 | rv = kFALSE ; | |
518 | return rv ; | |
519 | } | |
520 | ||
b0bba0af | 521 | //____________________________________________________________________________ |
88cb7938 | 522 | Int_t AliPHOSGetter::ReadTreeD() |
523 | { | |
524 | // Read the Digits | |
b0bba0af | 525 | |
7a9d98f9 | 526 | |
88cb7938 | 527 | // gets TreeD from the root file (PHOS.SDigits.root) |
528 | if ( !IsLoaded("D") ) { | |
529 | PhosLoader()->LoadDigits("UPDATE") ; | |
530 | PhosLoader()->LoadDigitizer("UPDATE") ; | |
531 | SetLoaded("D") ; | |
532 | } | |
533 | return Digits()->GetEntries() ; | |
534 | } | |
7a9d98f9 | 535 | |
b0bba0af | 536 | //____________________________________________________________________________ |
88cb7938 | 537 | Int_t AliPHOSGetter::ReadTreeH() |
538 | { | |
539 | // Read the Hits | |
540 | ||
541 | // gets TreeH from the root file (PHOS.Hit.root) | |
542 | if ( !IsLoaded("H") ) { | |
543 | PhosLoader()->LoadHits("UPDATE") ; | |
544 | SetLoaded("H") ; | |
545 | } | |
546 | return Hits()->GetEntries() ; | |
547 | } | |
4ae78bb1 | 548 | |
88cb7938 | 549 | //____________________________________________________________________________ |
550 | Int_t AliPHOSGetter::ReadTreeR() | |
551 | { | |
552 | // Read the RecPoints | |
b0bba0af | 553 | |
88cb7938 | 554 | |
555 | // gets TreeR from the root file (PHOS.RecPoints.root) | |
556 | if ( !IsLoaded("R") ) { | |
557 | PhosLoader()->LoadRecPoints("UPDATE") ; | |
558 | PhosLoader()->LoadClusterizer("UPDATE") ; | |
559 | SetLoaded("R") ; | |
7a9d98f9 | 560 | } |
f1611b7c | 561 | |
88cb7938 | 562 | return EmcRecPoints()->GetEntries() ; |
b0bba0af | 563 | } |
7a9d98f9 | 564 | |
b0bba0af | 565 | //____________________________________________________________________________ |
88cb7938 | 566 | Int_t AliPHOSGetter::ReadTreeT() |
567 | { | |
568 | // Read the TrackSegments | |
7a9d98f9 | 569 | |
407ae4df | 570 | |
88cb7938 | 571 | // gets TreeT from the root file (PHOS.TrackSegments.root) |
572 | if ( !IsLoaded("T") ) { | |
573 | PhosLoader()->LoadTracks("UPDATE") ; | |
574 | PhosLoader()->LoadTrackSegmentMaker("UPDATE") ; | |
575 | SetLoaded("T") ; | |
407ae4df | 576 | } |
fbf811ec | 577 | |
88cb7938 | 578 | return TrackSegments()->GetEntries() ; |
579 | } | |
b0bba0af | 580 | //____________________________________________________________________________ |
88cb7938 | 581 | Int_t AliPHOSGetter::ReadTreeP() |
582 | { | |
583 | // Read the TrackSegments | |
b0bba0af | 584 | |
88cb7938 | 585 | |
586 | // gets TreeT from the root file (PHOS.TrackSegments.root) | |
587 | if ( !IsLoaded("P") ) { | |
588 | PhosLoader()->LoadRecParticles("UPDATE") ; | |
589 | PhosLoader()->LoadPID("UPDATE") ; | |
590 | SetLoaded("P") ; | |
7a9d98f9 | 591 | } |
592 | ||
88cb7938 | 593 | return RecParticles()->GetEntries() ; |
594 | } | |
595 | //____________________________________________________________________________ | |
596 | Int_t AliPHOSGetter::ReadTreeS() | |
597 | { | |
598 | // Read the SDigits | |
599 | ||
7a9d98f9 | 600 | |
88cb7938 | 601 | // gets TreeS from the root file (PHOS.SDigits.root) |
602 | if ( !IsLoaded("S") ) { | |
7b4c1168 | 603 | PhosLoader()->LoadSDigits("READ") ; |
604 | PhosLoader()->LoadSDigitizer("READ") ; | |
88cb7938 | 605 | SetLoaded("S") ; |
7a9d98f9 | 606 | } |
b0bba0af | 607 | |
88cb7938 | 608 | return SDigits()->GetEntries() ; |
609 | } | |
f1611b7c | 610 | |
88cb7938 | 611 | //____________________________________________________________________________ |
612 | TClonesArray * AliPHOSGetter::SDigits() | |
613 | { | |
614 | // asks the Loader to return the Digits container | |
b0bba0af | 615 | |
88cb7938 | 616 | TClonesArray * rv = 0 ; |
617 | ||
618 | rv = PhosLoader()->SDigits() ; | |
619 | if (!rv) { | |
620 | PhosLoader()->MakeSDigitsArray() ; | |
621 | rv = PhosLoader()->SDigits() ; | |
622 | } | |
623 | return rv ; | |
b0bba0af | 624 | } |
625 | ||
626 | //____________________________________________________________________________ | |
e957fea8 | 627 | AliPHOSSDigitizer * AliPHOSGetter::SDigitizer() |
88cb7938 | 628 | { |
e957fea8 | 629 | // Returns pointer to the SDigitizer task |
88cb7938 | 630 | AliPHOSSDigitizer * rv ; |
631 | rv = dynamic_cast<AliPHOSSDigitizer *>(PhosLoader()->SDigitizer()) ; | |
632 | if (!rv) { | |
633 | Event(0, "S") ; | |
634 | rv = dynamic_cast<AliPHOSSDigitizer *>(PhosLoader()->SDigitizer()) ; | |
b0bba0af | 635 | } |
88cb7938 | 636 | return rv ; |
b0bba0af | 637 | } |
7a9d98f9 | 638 | |
b0bba0af | 639 | //____________________________________________________________________________ |
fc7e2f43 | 640 | TParticle * AliPHOSGetter::Secondary(const TParticle* p, Int_t index) const |
88cb7938 | 641 | { |
642 | // Return first (index=1) or second (index=2) secondary particle of primary particle p | |
643 | ||
644 | if(index <= 0) | |
645 | return 0 ; | |
646 | if(index > 2) | |
647 | return 0 ; | |
b0bba0af | 648 | |
88cb7938 | 649 | if(p) { |
650 | Int_t daughterIndex = p->GetDaughter(index-1) ; | |
651 | AliRunLoader * rl = AliRunLoader::GetRunLoader(PhosLoader()->GetTitle()); | |
5d12ce38 | 652 | return rl->GetAliRun()->GetMCApp()->Particle(daughterIndex) ; |
88cb7938 | 653 | } |
654 | else | |
655 | return 0 ; | |
656 | } | |
4ae78bb1 | 657 | |
88cb7938 | 658 | //____________________________________________________________________________ |
fc7e2f43 | 659 | void AliPHOSGetter::Track(Int_t itrack) |
88cb7938 | 660 | { |
661 | // Read the first entry of PHOS branch in hit tree gAlice->TreeH() | |
662 | ||
663 | AliRunLoader * rl = AliRunLoader::GetRunLoader(PhosLoader()->GetTitle()); | |
b0bba0af | 664 | |
88cb7938 | 665 | if( !TreeH() ) // load treeH the first time |
666 | rl->LoadHits() ; | |
b0bba0af | 667 | |
88cb7938 | 668 | // first time create the container |
669 | TClonesArray * hits = Hits() ; | |
670 | if ( hits ) | |
671 | hits->Clear() ; | |
b0bba0af | 672 | |
88cb7938 | 673 | TBranch * phosbranch = dynamic_cast<TBranch*>(TreeH()->GetBranch("PHOS")) ; |
674 | phosbranch->SetAddress(&hits) ; | |
675 | phosbranch->GetEntry(itrack) ; | |
b0bba0af | 676 | } |
677 | ||
b0bba0af | 678 | //____________________________________________________________________________ |
88cb7938 | 679 | TTree * AliPHOSGetter::TreeD() const |
680 | { | |
e957fea8 | 681 | // Returns pointer to the Digits Tree |
88cb7938 | 682 | TTree * rv = 0 ; |
683 | rv = PhosLoader()->TreeD() ; | |
684 | if ( !rv ) { | |
685 | PhosLoader()->MakeTree("D"); | |
686 | rv = PhosLoader()->TreeD() ; | |
b0bba0af | 687 | } |
b0bba0af | 688 | |
88cb7938 | 689 | return rv ; |
b0bba0af | 690 | } |
548f0134 | 691 | |
b0bba0af | 692 | //____________________________________________________________________________ |
88cb7938 | 693 | TTree * AliPHOSGetter::TreeH() const |
694 | { | |
e957fea8 | 695 | // Returns pointer to the Hits Tree |
88cb7938 | 696 | TTree * rv = 0 ; |
697 | rv = PhosLoader()->TreeH() ; | |
698 | if ( !rv ) { | |
699 | PhosLoader()->MakeTree("H"); | |
700 | rv = PhosLoader()->TreeH() ; | |
701 | } | |
702 | ||
703 | return rv ; | |
b0bba0af | 704 | } |
7a9d98f9 | 705 | |
b0bba0af | 706 | //____________________________________________________________________________ |
88cb7938 | 707 | TTree * AliPHOSGetter::TreeR() const |
0bc3b8ed | 708 | { |
e957fea8 | 709 | // Returns pointer to the RecPoints Tree |
88cb7938 | 710 | TTree * rv = 0 ; |
711 | rv = PhosLoader()->TreeR() ; | |
712 | if ( !rv ) { | |
713 | PhosLoader()->MakeTree("R"); | |
714 | rv = PhosLoader()->TreeR() ; | |
715 | } | |
b0bba0af | 716 | |
88cb7938 | 717 | return rv ; |
b0bba0af | 718 | } |
719 | ||
b0bba0af | 720 | //____________________________________________________________________________ |
88cb7938 | 721 | TTree * AliPHOSGetter::TreeT() const |
0bc3b8ed | 722 | { |
e957fea8 | 723 | // Returns pointer to the TrackSegments Tree |
88cb7938 | 724 | TTree * rv = 0 ; |
725 | rv = PhosLoader()->TreeT() ; | |
726 | if ( !rv ) { | |
727 | PhosLoader()->MakeTree("T"); | |
728 | rv = PhosLoader()->TreeT() ; | |
729 | } | |
b0bba0af | 730 | |
88cb7938 | 731 | return rv ; |
b0bba0af | 732 | } |
b0bba0af | 733 | //____________________________________________________________________________ |
88cb7938 | 734 | TTree * AliPHOSGetter::TreeP() const |
0bc3b8ed | 735 | { |
e957fea8 | 736 | // Returns pointer to the RecParticles Tree |
88cb7938 | 737 | TTree * rv = 0 ; |
738 | rv = PhosLoader()->TreeP() ; | |
739 | if ( !rv ) { | |
740 | PhosLoader()->MakeTree("P"); | |
741 | rv = PhosLoader()->TreeP() ; | |
742 | } | |
4ae78bb1 | 743 | |
88cb7938 | 744 | return rv ; |
b0bba0af | 745 | } |
746 | ||
747 | //____________________________________________________________________________ | |
88cb7938 | 748 | TTree * AliPHOSGetter::TreeS() const |
e957fea8 | 749 | { |
750 | // Returns pointer to the SDigits Tree | |
88cb7938 | 751 | TTree * rv = 0 ; |
752 | rv = PhosLoader()->TreeS() ; | |
753 | if ( !rv ) { | |
754 | PhosLoader()->MakeTree("S"); | |
755 | rv = PhosLoader()->TreeS() ; | |
b0bba0af | 756 | } |
b0bba0af | 757 | |
88cb7938 | 758 | return rv ; |
b0bba0af | 759 | } |
7a9d98f9 | 760 | |
b0bba0af | 761 | //____________________________________________________________________________ |
88cb7938 | 762 | Bool_t AliPHOSGetter::VersionExists(TString & opt) const |
763 | { | |
764 | // checks if the version with the present name already exists in the same directory | |
7a9d98f9 | 765 | |
88cb7938 | 766 | Bool_t rv = kFALSE ; |
767 | ||
768 | AliRunLoader * rl = AliRunLoader::GetRunLoader(PhosLoader()->GetTitle()); | |
769 | TString version( rl->GetEventFolder()->GetName() ) ; | |
7a9d98f9 | 770 | |
88cb7938 | 771 | opt.ToLower() ; |
b0bba0af | 772 | |
88cb7938 | 773 | if ( opt == "sdigits") { |
774 | // add the version name to the root file name | |
775 | TString fileName( PhosLoader()->GetSDigitsFileName() ) ; | |
e191bb57 | 776 | if (version != AliConfig::GetDefaultEventFolderName()) // only if not the default folder name |
88cb7938 | 777 | fileName = fileName.ReplaceAll(".root", "") + "_" + version + ".root" ; |
778 | if ( !(gSystem->AccessPathName(fileName)) ) { | |
779 | Warning("VersionExists", "The file %s already exists", fileName.Data()) ; | |
780 | rv = kTRUE ; | |
4ae78bb1 | 781 | } |
88cb7938 | 782 | PhosLoader()->SetSDigitsFileName(fileName) ; |
b0bba0af | 783 | } |
7a9d98f9 | 784 | |
88cb7938 | 785 | if ( opt == "digits") { |
786 | // add the version name to the root file name | |
787 | TString fileName( PhosLoader()->GetDigitsFileName() ) ; | |
e191bb57 | 788 | if (version != AliConfig::GetDefaultEventFolderName()) // only if not the default folder name |
88cb7938 | 789 | fileName = fileName.ReplaceAll(".root", "") + "_" + version + ".root" ; |
790 | if ( !(gSystem->AccessPathName(fileName)) ) { | |
791 | Warning("VersionExists", "The file %s already exists", fileName.Data()) ; | |
792 | rv = kTRUE ; | |
7a9d98f9 | 793 | } |
b0bba0af | 794 | } |
7a9d98f9 | 795 | |
88cb7938 | 796 | return rv ; |
7a9d98f9 | 797 | |
b0bba0af | 798 | } |
88cb7938 | 799 | |
7bb289a7 | 800 | //____________________________________________________________________________ |
88cb7938 | 801 | UShort_t AliPHOSGetter::EventPattern(void) const |
0bc3b8ed | 802 | { |
803 | // Return the pattern (trigger bit register) of the beam-test event | |
7bb289a7 | 804 | if(fBTE) |
805 | return fBTE->GetPattern() ; | |
806 | else | |
807 | return 0 ; | |
808 | } | |
809 | //____________________________________________________________________________ | |
88cb7938 | 810 | Float_t AliPHOSGetter::BeamEnergy(void) const |
0bc3b8ed | 811 | { |
812 | // Return the beam energy of the beam-test event | |
7bb289a7 | 813 | if(fBTE) |
814 | return fBTE->GetBeamEnergy() ; | |
815 | else | |
816 | return 0 ; | |
817 | } |