]>
Commit | Line | Data |
---|---|---|
9ce40367 | 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 | $Log$ | |
314558d7 | 18 | Revision 1.10 2001/11/15 11:07:25 jchudoba |
19 | Set to zero new pointers to TPC and TRD special trees in the default ctor. Add const to all Get functions. Remove unused constant, rename constant according coding rules. | |
20 | ||
9ae76683 | 21 | Revision 1.9 2001/11/15 09:00:11 jchudoba |
22 | Add special treatment for TPC and TRD, they use different trees than other detectors | |
23 | ||
f34e67f9 | 24 | Revision 1.8 2001/10/21 18:38:43 hristov |
25 | Several pointers were set to zero in the default constructors to avoid memory management problems | |
26 | ||
2685bf00 | 27 | Revision 1.7 2001/10/04 15:56:07 jchudoba |
28 | TTask inheritance | |
29 | ||
bd7a3098 | 30 | Revision 1.4 2001/09/19 06:23:50 jchudoba |
31 | Move some tasks to AliStream and AliMergeCombi classes | |
32 | ||
a90ecf3c | 33 | Revision 1.3 2001/07/30 14:04:18 jchudoba |
34 | correct bug in the initialization | |
35 | ||
af20391f | 36 | Revision 1.2 2001/07/28 10:44:32 hristov |
37 | Loop variable declared once; typos corrected | |
38 | ||
aee2ebe9 | 39 | Revision 1.1 2001/07/27 12:59:00 jchudoba |
40 | Manager class for merging/digitization | |
41 | ||
9ce40367 | 42 | */ |
43 | ||
44 | //////////////////////////////////////////////////////////////////////// | |
45 | // | |
46 | // AliRunDigitizer.cxx | |
47 | // | |
48 | // Manager object for merging/digitization | |
49 | // | |
50 | // Instance of this class manages the digitization and/or merging of | |
51 | // Sdigits into Digits. | |
52 | // | |
53 | // Only one instance of this class is created in the macro: | |
a90ecf3c | 54 | // AliRunDigitizer * manager = |
55 | // new AliRunDigitizer(nInputStreams,SPERB); | |
56 | // where nInputStreams is number of input streams and SPERB is | |
57 | // signals per background variable, which determines how combinations | |
58 | // of signal and background events are generated. | |
9ce40367 | 59 | // Then instances of specific detector digitizers are created: |
60 | // AliMUONDigitizer *dMUON = new AliMUONDigitizer(manager) | |
a90ecf3c | 61 | // and the I/O configured (you have to specify input files |
62 | // and an output file). The manager connects appropriate trees from | |
63 | // the input files according a combination returned by AliMergeCombi | |
9ae76683 | 64 | // class. It creates TreeD in the output and runs once per |
a90ecf3c | 65 | // event Digitize method of all existing AliDetDigitizers |
66 | // (without any option). AliDetDigitizers ask manager | |
67 | // for a TTree with input (manager->GetInputTreeS(Int_t i), | |
9ce40367 | 68 | // merge all inputs, digitize it, and save it in the TreeD |
69 | // obtained by manager->GetTreeD(). Output events are stored with | |
70 | // numbers from 0, this default can be changed by | |
71 | // manager->SetFirstOutputEventNr(Int_t) method. The particle numbers | |
72 | // in the output are shifted by MASK, which is taken from manager. | |
73 | // | |
8d5e6345 | 74 | // The default output is to the signal file (stream 0). This can be |
75 | // changed with the SetOutputFile(TString fn) method. | |
76 | // | |
9ae76683 | 77 | // Single input file is permitted. Maximum kMaxStreamsToMerge can be merged. |
9ce40367 | 78 | // Input from the memory (on-the-fly merging) is not yet |
79 | // supported, as well as access to the input data by invoking methods | |
80 | // on the output data. | |
81 | // | |
a90ecf3c | 82 | // Access to the some data is via gAlice for now (supposing the |
83 | // same geometry in all input files), gAlice is taken from the first | |
84 | // input file on the first stream. | |
9ce40367 | 85 | // |
314558d7 | 86 | // Example with MUON digitizer, no merging, just digitization |
87 | // | |
88 | // AliRunDigitizer * manager = new AliRunDigitizer(1,1); | |
89 | // manager->SetInputStream(0,"galice.root"); | |
90 | // AliMUONDigitizer *dMUON = new AliMUONDigitizer(manager); | |
91 | // manager->Exec(""); | |
92 | // | |
93 | // Example with MUON digitizer, merge all events from | |
94 | // galice.root (signal) file with events from bgr.root | |
95 | // (background) file. Number of merged events is | |
96 | // min(number of events in galice.root, number of events in bgr.root) | |
97 | // | |
98 | // AliRunDigitizer * manager = new AliRunDigitizer(2,1); | |
99 | // manager->SetInputStream(0,"galice.root"); | |
100 | // manager->SetInputStream(1,"bgr.root"); | |
101 | // AliMUONDigitizer *dMUON = new AliMUONDigitizer(manager); | |
102 | // manager->Exec(""); | |
103 | // | |
104 | // Example with MUON digitizer, save digits in a new file digits.root, | |
105 | // process only 1 event | |
9ce40367 | 106 | // |
a90ecf3c | 107 | // AliRunDigitizer * manager = new AliRunDigitizer(2,1); |
314558d7 | 108 | // manager->SetInputStream(0,"galice.root"); |
109 | // manager->SetInputStream(1,"bgr.root"); | |
a90ecf3c | 110 | // manager->SetOutputFile("digits.root"); |
111 | // AliMUONDigitizer *dMUON = new AliMUONDigitizer(manager); | |
112 | // manager->SetNrOfEventsToWrite(1); | |
8d5e6345 | 113 | // manager->Exec(""); |
9ce40367 | 114 | // |
115 | //////////////////////////////////////////////////////////////////////// | |
116 | ||
117 | // system includes | |
118 | ||
119 | #include <iostream.h> | |
120 | ||
121 | // ROOT includes | |
122 | ||
123 | #include "TFile.h" | |
124 | #include "TTree.h" | |
8d5e6345 | 125 | #include "TList.h" |
9ce40367 | 126 | |
127 | // AliROOT includes | |
128 | ||
129 | #include "AliRunDigitizer.h" | |
130 | #include "AliDigitizer.h" | |
131 | #include "AliRun.h" | |
132 | #include "AliHeader.h" | |
133 | #include "TParticle.h" | |
a90ecf3c | 134 | #include "AliStream.h" |
135 | #include "AliMergeCombi.h" | |
9ce40367 | 136 | |
137 | ClassImp(AliRunDigitizer) | |
138 | ||
314558d7 | 139 | //////////////////////////////////////////////////////////////////////// |
140 | AliRunDigitizer::AliRunDigitizer() | |
141 | { | |
142 | // root requires default ctor, where no new objects can be created | |
143 | // do not use this ctor, it is supplied only for root needs | |
144 | ||
145 | // just set all pointers - data members to 0 | |
146 | fOutput = 0; | |
147 | fTreeD = 0; | |
148 | fTreeDTPC = 0; | |
149 | fTreeDTRD = 0; | |
150 | fInputStreams = 0; | |
151 | for (Int_t i=0;i<kMaxStreamsToMerge;i++) { | |
152 | fArrayTreeS[i]=fArrayTreeH[i]=fArrayTreeTPCS[i]=fArrayTreeTRDS[i]=NULL; | |
153 | fInputFiles[i]=0; | |
154 | } | |
155 | fCombi = 0; | |
156 | ||
157 | } | |
158 | ||
9ce40367 | 159 | //////////////////////////////////////////////////////////////////////// |
8d5e6345 | 160 | AliRunDigitizer::AliRunDigitizer(Int_t nInputStreams, Int_t sperb) : TTask("AliRunDigitizer","The manager for Merging") |
a90ecf3c | 161 | { |
314558d7 | 162 | // ctor which should be used to create a manager for merging/digitization |
a90ecf3c | 163 | if (nInputStreams == 0) { |
164 | Error("AliRunDigitizer","Specify nr of input streams"); | |
165 | return; | |
166 | } | |
aee2ebe9 | 167 | Int_t i; |
a90ecf3c | 168 | fNinputs = nInputStreams; |
8d5e6345 | 169 | fOutputFileName = ""; |
170 | fOutputDirName = "."; | |
9ae76683 | 171 | fCombination.Set(kMaxStreamsToMerge); |
172 | for (i=0;i<kMaxStreamsToMerge;i++) { | |
314558d7 | 173 | fArrayTreeS[i]=fArrayTreeH[i]=fArrayTreeTPCS[i]=fArrayTreeTRDS[i]=NULL; |
9ce40367 | 174 | fCombination[i]=-1; |
175 | } | |
176 | fkMASKSTEP = 10000000; | |
177 | fkMASK[0] = 0; | |
9ae76683 | 178 | for (i=1;i<kMaxStreamsToMerge;i++) { |
9ce40367 | 179 | fkMASK[i] = fkMASK[i-1] + fkMASKSTEP; |
180 | } | |
a90ecf3c | 181 | fInputStreams = new TClonesArray("AliStream",nInputStreams); |
182 | TClonesArray &lInputStreams = *fInputStreams; | |
8d5e6345 | 183 | // the first Input is open RW to be output as well |
184 | new(lInputStreams[0]) AliStream("UPDATE"); | |
185 | for (i=1;i<nInputStreams;i++) { | |
314558d7 | 186 | new(lInputStreams[i]) AliStream("READ"); |
a90ecf3c | 187 | } |
9ce40367 | 188 | fOutput = 0; |
189 | fEvent = 0; | |
8d5e6345 | 190 | fNrOfEventsToWrite = -1; |
9ce40367 | 191 | fNrOfEventsWritten = 0; |
192 | fCopyTreesFromInput = -1; | |
a90ecf3c | 193 | fCombi = new AliMergeCombi(nInputStreams,sperb); |
8d5e6345 | 194 | fDebug = 0; |
2685bf00 | 195 | fTreeD = 0; |
9ae76683 | 196 | fTreeDTPC = 0; |
197 | fTreeDTRD = 0; | |
198 | ||
199 | for (i=0; i<kMaxStreamsToMerge; i++) fInputFiles[i]=0; | |
9ce40367 | 200 | } |
201 | ||
202 | //////////////////////////////////////////////////////////////////////// | |
203 | ||
204 | AliRunDigitizer::~AliRunDigitizer() { | |
205 | // dtor | |
206 | ||
a90ecf3c | 207 | if (fInputStreams) { |
208 | delete fInputStreams; | |
209 | fInputStreams = 0; | |
210 | } | |
211 | if (fCombi) { | |
212 | delete fCombi; | |
213 | fCombi = 0; | |
214 | } | |
215 | ||
9ce40367 | 216 | } |
9ce40367 | 217 | //////////////////////////////////////////////////////////////////////// |
218 | void AliRunDigitizer::AddDigitizer(AliDigitizer *digitizer) | |
219 | { | |
220 | // add digitizer to the list of active digitizers | |
8d5e6345 | 221 | this->Add(digitizer); |
9ce40367 | 222 | } |
9ce40367 | 223 | //////////////////////////////////////////////////////////////////////// |
a90ecf3c | 224 | void AliRunDigitizer::SetInputStream(Int_t i, char *inputFile) |
9ce40367 | 225 | { |
a90ecf3c | 226 | if (i > fInputStreams->GetLast()) { |
227 | Error("SetInputStream","Input stream number too high"); | |
228 | return; | |
9ce40367 | 229 | } |
a90ecf3c | 230 | static_cast<AliStream*>(fInputStreams->At(i))->AddFile(inputFile); |
9ce40367 | 231 | } |
232 | ||
233 | //////////////////////////////////////////////////////////////////////// | |
8d5e6345 | 234 | void AliRunDigitizer::Digitize(Option_t* option) |
9ce40367 | 235 | { |
236 | // get a new combination of inputs, connect input trees and loop | |
237 | // over all digitizers | |
238 | ||
239 | if (!InitGlobal()) { | |
240 | cerr<<"False from InitGlobal"<<endl; | |
241 | return; | |
242 | } | |
243 | // take gAlice from the first input file. It is needed to access | |
244 | // geometry data | |
a90ecf3c | 245 | if (!static_cast<AliStream*>(fInputStreams->At(0))->ImportgAlice()) { |
246 | cerr<<"gAlice object not found in the first file of " | |
247 | <<"the 1st stream"<<endl; | |
248 | return; | |
249 | } | |
250 | Int_t eventsCreated = 0; | |
8d5e6345 | 251 | // loop until there is anything on the input in case fNrOfEventsToWrite < 0 |
252 | while ((eventsCreated++ < fNrOfEventsToWrite) || (fNrOfEventsToWrite < 0)) { | |
253 | if (!ConnectInputTrees()) break; | |
a90ecf3c | 254 | InitEvent(); |
9ce40367 | 255 | // loop over all registered digitizers and let them do the work |
8d5e6345 | 256 | ExecuteTasks(""); |
257 | CleanTasks(); | |
a90ecf3c | 258 | FinishEvent(); |
9ce40367 | 259 | } |
260 | FinishGlobal(); | |
261 | } | |
262 | ||
263 | //////////////////////////////////////////////////////////////////////// | |
a90ecf3c | 264 | Bool_t AliRunDigitizer::ConnectInputTrees() |
9ce40367 | 265 | { |
266 | // fill arrays fArrayTreeS, fArrayTreeH and fArrayTreeTPCS with | |
267 | // pointers to the correct events according fCombination values | |
268 | // null pointers can be in the output, AliDigitizer has to check it | |
269 | ||
9ce40367 | 270 | TTree *tree; |
8d5e6345 | 271 | char treeName[50]; |
a90ecf3c | 272 | Int_t serialNr; |
9ae76683 | 273 | Int_t eventNr[kMaxStreamsToMerge], delta[kMaxStreamsToMerge]; |
a90ecf3c | 274 | fCombi->Combination(eventNr, delta); |
9ce40367 | 275 | for (Int_t i=0;i<fNinputs;i++) { |
a90ecf3c | 276 | if (delta[i] == 1) { |
277 | AliStream *iStream = static_cast<AliStream*>(fInputStreams->At(i)); | |
278 | if (!iStream->NextEventInStream(serialNr)) return kFALSE; | |
8d5e6345 | 279 | fInputFiles[i]=iStream->CurrentFile(); |
a90ecf3c | 280 | sprintf(treeName,"TreeS%d",serialNr); |
281 | tree = static_cast<TTree*>(iStream->CurrentFile()->Get(treeName)); | |
282 | fArrayTreeS[i] = tree; | |
283 | sprintf(treeName,"TreeH%d",serialNr); | |
284 | tree = static_cast<TTree*>(iStream->CurrentFile()->Get(treeName)); | |
285 | fArrayTreeH[i] = tree; | |
286 | sprintf(treeName,"TreeS_75x40_100x60_%d",serialNr); | |
287 | tree = static_cast<TTree*>(iStream->CurrentFile()->Get(treeName)); | |
288 | fArrayTreeTPCS[i] = tree; | |
f34e67f9 | 289 | sprintf(treeName,"TreeS%d_TRD",serialNr); |
290 | tree = static_cast<TTree*>(iStream->CurrentFile()->Get(treeName)); | |
291 | fArrayTreeTRDS[i] = tree; | |
a90ecf3c | 292 | } else if (delta[i] != 0) { |
293 | Error("ConnectInputTrees","Only delta 0 or 1 is implemented"); | |
294 | return kFALSE; | |
295 | } | |
296 | } | |
297 | return kTRUE; | |
9ce40367 | 298 | } |
299 | ||
300 | //////////////////////////////////////////////////////////////////////// | |
301 | Bool_t AliRunDigitizer::InitGlobal() | |
302 | { | |
303 | // called once before Digitize() is called, initialize digitizers and output | |
304 | ||
8d5e6345 | 305 | TList* subTasks = this->GetListOfTasks(); |
306 | if (subTasks) { | |
307 | subTasks->ForEach(AliDigitizer,Init)(); | |
308 | } | |
9ce40367 | 309 | return kTRUE; |
310 | } | |
311 | ||
8d5e6345 | 312 | //////////////////////////////////////////////////////////////////////// |
313 | ||
314 | void AliRunDigitizer::SetOutputFile(TString fn) | |
315 | // the output will be to separate file, not to the signal file | |
316 | { | |
317 | fOutputFileName = fn; | |
318 | (static_cast<AliStream*>(fInputStreams->At(0)))->ChangeMode("READ"); | |
319 | InitOutputGlobal(); | |
320 | } | |
321 | ||
9ce40367 | 322 | //////////////////////////////////////////////////////////////////////// |
323 | Bool_t AliRunDigitizer::InitOutputGlobal() | |
324 | { | |
8d5e6345 | 325 | // Creates the output file, called by InitEvent() |
9ce40367 | 326 | |
327 | TString fn; | |
328 | fn = fOutputDirName + '/' + fOutputFileName; | |
329 | fOutput = new TFile(fn,"recreate"); | |
330 | if (GetDebug()>2) { | |
8d5e6345 | 331 | cerr<<"AliRunDigitizer::InitOutputGlobal(): file "<<fn.Data()<<" was opened"<<endl; |
9ce40367 | 332 | } |
333 | if (fOutput) return kTRUE; | |
334 | Error("InitOutputGlobal","Could not create output file."); | |
335 | return kFALSE; | |
336 | } | |
337 | ||
338 | ||
339 | //////////////////////////////////////////////////////////////////////// | |
a90ecf3c | 340 | void AliRunDigitizer::InitEvent() |
9ce40367 | 341 | { |
342 | // Creates TreeDxx in the output file, called from Digitize() once for | |
343 | // each event. xx = fEvent | |
344 | ||
345 | if (GetDebug()>2) | |
a90ecf3c | 346 | cerr<<"AliRunDigitizer::InitEvent: fEvent = "<<fEvent<<endl; |
8d5e6345 | 347 | |
348 | // if fOutputFileName was not given, write output to signal file | |
349 | if (fOutputFileName == "") { | |
350 | fOutput = (static_cast<AliStream*>(fInputStreams->At(0)))->CurrentFile(); | |
351 | } | |
9ce40367 | 352 | fOutput->cd(); |
8d5e6345 | 353 | char treeName[30]; |
354 | sprintf(treeName,"TreeD%d",fEvent); | |
355 | fTreeD = static_cast<TTree*>(fOutput->Get(treeName)); | |
356 | if (!fTreeD) { | |
357 | fTreeD = new TTree(treeName,"Digits"); | |
358 | fTreeD->Write(0,TObject::kOverwrite); | |
359 | } | |
f34e67f9 | 360 | |
361 | // special tree for TPC | |
362 | sprintf(treeName,"TreeD_75x40_100x60_%d",fEvent); | |
363 | fTreeDTPC = static_cast<TTree*>(fOutput->Get(treeName)); | |
364 | if (!fTreeDTPC) { | |
365 | fTreeDTPC = new TTree(treeName,"TPC_Digits"); | |
366 | fTreeDTPC->Write(0,TObject::kOverwrite); | |
367 | } | |
368 | ||
369 | // special tree for TRD | |
370 | sprintf(treeName,"TreeD%d_TRD",fEvent); | |
371 | fTreeDTRD = static_cast<TTree*>(fOutput->Get(treeName)); | |
372 | if (!fTreeDTRD) { | |
373 | fTreeDTRD = new TTree(treeName,"TRD_Digits"); | |
374 | fTreeDTRD->Write(0,TObject::kOverwrite); | |
375 | } | |
376 | ||
9ce40367 | 377 | } |
378 | ||
379 | //////////////////////////////////////////////////////////////////////// | |
a90ecf3c | 380 | void AliRunDigitizer::FinishEvent() |
9ce40367 | 381 | { |
382 | // called at the end of loop over digitizers | |
383 | ||
384 | fOutput->cd(); | |
385 | if (fCopyTreesFromInput > -1) { | |
386 | char treeName[20]; | |
387 | Int_t i = fCopyTreesFromInput; | |
388 | sprintf(treeName,"TreeK%d",fCombination[i]); | |
8d5e6345 | 389 | fInputFiles[i]->Get(treeName)->Clone()->Write(); |
9ce40367 | 390 | sprintf(treeName,"TreeH%d",fCombination[i]); |
8d5e6345 | 391 | fInputFiles[i]->Get(treeName)->Clone()->Write(); |
9ce40367 | 392 | } |
393 | fEvent++; | |
394 | fNrOfEventsWritten++; | |
395 | if (fTreeD) { | |
396 | delete fTreeD; | |
397 | fTreeD = 0; | |
398 | } | |
f34e67f9 | 399 | if (fTreeDTPC) { |
400 | delete fTreeDTPC; | |
401 | fTreeDTPC = 0; | |
402 | } | |
403 | if (fTreeDTRD) { | |
404 | delete fTreeDTRD; | |
405 | fTreeDTRD = 0; | |
406 | } | |
9ce40367 | 407 | } |
408 | //////////////////////////////////////////////////////////////////////// | |
409 | void AliRunDigitizer::FinishGlobal() | |
410 | { | |
411 | // called at the end of Exec | |
412 | // save unique objects to the output file | |
413 | ||
414 | fOutput->cd(); | |
415 | this->Write(); | |
416 | if (fCopyTreesFromInput > -1) { | |
8d5e6345 | 417 | fInputFiles[fCopyTreesFromInput]->Get("TE")->Clone()->Write(); |
9ce40367 | 418 | gAlice->Write(); |
419 | } | |
420 | fOutput->Close(); | |
421 | } | |
422 | ||
423 | ||
424 | //////////////////////////////////////////////////////////////////////// | |
9ae76683 | 425 | Int_t AliRunDigitizer::GetNParticles(Int_t event) const |
9ce40367 | 426 | { |
427 | // return number of particles in all input files for a given | |
428 | // event (as numbered in the output file) | |
429 | // return -1 if some file cannot be accessed | |
430 | ||
431 | Int_t sum = 0; | |
432 | Int_t sumI; | |
433 | for (Int_t i = 0; i < fNinputs; i++) { | |
434 | sumI = GetNParticles(GetInputEventNumber(event,i), i); | |
435 | if (sumI < 0) return -1; | |
436 | sum += sumI; | |
437 | } | |
438 | return sum; | |
439 | } | |
440 | ||
441 | //////////////////////////////////////////////////////////////////////// | |
9ae76683 | 442 | Int_t AliRunDigitizer::GetNParticles(Int_t event, Int_t input) const |
9ce40367 | 443 | { |
444 | // return number of particles in input file input for a given | |
445 | // event (as numbered in this input file) | |
446 | // return -1 if some error | |
447 | ||
bd7a3098 | 448 | // Must be revised in the version with AliStream |
449 | ||
450 | return -1; | |
451 | ||
452 | /* | |
9ce40367 | 453 | TFile *file = ConnectInputFile(input); |
454 | if (!file) { | |
455 | Error("GetNParticles","Cannot open input file"); | |
456 | return -1; | |
457 | } | |
458 | ||
459 | // find the header and get Nprimaries and Nsecondaries | |
460 | TTree* tE = (TTree *)file->Get("TE") ; | |
461 | if (!tE) { | |
462 | Error("GetNParticles","input file does not contain TE"); | |
463 | return -1; | |
464 | } | |
465 | AliHeader* header; | |
466 | header = 0; | |
467 | tE->SetBranchAddress("Header", &header); | |
468 | if (!tE->GetEntry(event)) { | |
469 | Error("GetNParticles","event %d not found",event); | |
470 | return -1; | |
471 | } | |
472 | if (GetDebug()>2) { | |
473 | cerr<<"Nprimary: "<< header->GetNprimary()<<endl; | |
474 | cerr<<"Nsecondary: "<<header->GetNsecondary()<<endl; | |
475 | } | |
476 | return header->GetNprimary() + header->GetNsecondary(); | |
bd7a3098 | 477 | */ |
9ce40367 | 478 | } |
479 | ||
480 | //////////////////////////////////////////////////////////////////////// | |
9ae76683 | 481 | Int_t* AliRunDigitizer::GetInputEventNumbers(Int_t event) const |
9ce40367 | 482 | { |
483 | // return pointer to an int array with input event numbers which were | |
484 | // merged in the output event event | |
485 | ||
486 | // simplified for now, implement later | |
9ae76683 | 487 | Int_t * a = new Int_t[kMaxStreamsToMerge]; |
9ce40367 | 488 | for (Int_t i = 0; i < fNinputs; i++) { |
489 | a[i] = event; | |
490 | } | |
491 | return a; | |
492 | } | |
493 | //////////////////////////////////////////////////////////////////////// | |
9ae76683 | 494 | Int_t AliRunDigitizer::GetInputEventNumber(Int_t event, Int_t input) const |
9ce40367 | 495 | { |
496 | // return an event number of an eventInput from input file input | |
497 | // which was merged to create output event event | |
498 | ||
499 | // simplified for now, implement later | |
500 | return event; | |
501 | } | |
9ce40367 | 502 | //////////////////////////////////////////////////////////////////////// |
9ae76683 | 503 | TParticle* AliRunDigitizer::GetParticle(Int_t i, Int_t event) const |
9ce40367 | 504 | { |
505 | // return pointer to particle with index i (index with mask) | |
506 | ||
507 | // decode the MASK | |
508 | Int_t input = i/fkMASKSTEP; | |
509 | return GetParticle(i,input,GetInputEventNumber(event,input)); | |
510 | } | |
511 | ||
512 | //////////////////////////////////////////////////////////////////////// | |
9ae76683 | 513 | TParticle* AliRunDigitizer::GetParticle(Int_t i, Int_t input, Int_t event) const |
9ce40367 | 514 | { |
515 | // return pointer to particle with index i in the input file input | |
516 | // (index without mask) | |
517 | // event is the event number in the file input | |
8d5e6345 | 518 | // return 0 i fit does not exist |
bd7a3098 | 519 | |
520 | // Must be revised in the version with AliStream | |
9ce40367 | 521 | |
bd7a3098 | 522 | return 0; |
523 | /* | |
9ce40367 | 524 | TFile *file = ConnectInputFile(input); |
525 | if (!file) { | |
526 | Error("GetParticle","Cannot open input file"); | |
527 | return 0; | |
528 | } | |
529 | ||
530 | // find the header and get Nprimaries and Nsecondaries | |
531 | TTree* tE = (TTree *)file->Get("TE") ; | |
532 | if (!tE) { | |
533 | Error("GetParticle","input file does not contain TE"); | |
534 | return 0; | |
535 | } | |
536 | AliHeader* header; | |
537 | header = 0; | |
538 | tE->SetBranchAddress("Header", &header); | |
539 | if (!tE->GetEntry(event)) { | |
540 | Error("GetParticle","event %d not found",event); | |
541 | return 0; | |
542 | } | |
543 | ||
544 | // connect TreeK | |
545 | char treeName[30]; | |
546 | sprintf(treeName,"TreeK%d",event); | |
547 | TTree* tK = static_cast<TTree*>(file->Get(treeName)); | |
548 | if (!tK) { | |
549 | Error("GetParticle","input file does not contain TreeK%d",event); | |
550 | return 0; | |
551 | } | |
552 | TParticle *particleBuffer; | |
553 | particleBuffer = 0; | |
554 | tK->SetBranchAddress("Particles", &particleBuffer); | |
555 | ||
556 | ||
557 | // algorithmic way of getting entry index | |
558 | // (primary particles are filled after secondaries) | |
559 | Int_t entry; | |
560 | if (i<header->GetNprimary()) | |
561 | entry = i+header->GetNsecondary(); | |
562 | else | |
563 | entry = i-header->GetNprimary(); | |
9ce40367 | 564 | Int_t bytesRead = tK->GetEntry(entry); |
565 | // new ((*fParticles)[nentries]) TParticle(*fParticleBuffer); | |
566 | if (bytesRead) | |
567 | return particleBuffer; | |
568 | return 0; | |
bd7a3098 | 569 | */ |
9ce40367 | 570 | } |
8d5e6345 | 571 | |
9ce40367 | 572 | //////////////////////////////////////////////////////////////////////// |
8d5e6345 | 573 | void AliRunDigitizer::ExecuteTask(Option_t* option) |
574 | { | |
575 | // overwrite ExecuteTask to do Digitize only | |
9ce40367 | 576 | |
8d5e6345 | 577 | if (!IsActive()) return; |
578 | Digitize(option); | |
579 | fHasExecuted = kTRUE; | |
580 | return; | |
581 | } |