]> git.uio.no Git - u/mrichter/AliRoot.git/blame - STEER/AliRun.cxx
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
Corrected calculation of the cross talk in the outer sectors
[u/mrichter/AliRoot.git] / STEER / AliRun.cxx
CommitLineData
99d554c8 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$
3d8c4988 18Revision 1.85 2002/05/24 13:29:58 hristov
19AliTrackReference added, AliDisplay modified
20
aab9c8d5 21Revision 1.84 2002/05/21 16:26:07 hristov
22Find correctly TreeK in case CONFIG_SPLIT_FILE is set (Y.Schutz)
23
50a6540a 24Revision 1.83 2002/04/04 13:16:17 jchudoba
25add possibility to write sdigits, digits and rec. points into separate files
26
7a16e9cc 27Revision 1.82 2002/03/12 11:06:03 morsch
28Add particle status code to argument list of SetTrack(..).
29
47c8bcbe 30Revision 1.81 2001/12/19 14:46:26 morsch
31Add possibility to disable StepManager() for each module separately.
32
3f754d83 33Revision 1.80 2001/10/21 18:22:55 hristov
34BranchOld replaced by Branch. It works correctly with Root 2.02.xx
35
21e74529 36Revision 1.79 2001/10/09 18:00:35 hristov
37Temporary fix to provide unique event number in the simulation (J.Chudoba)
38
eef4b160 39Revision 1.78 2001/10/04 15:30:56 hristov
40Changes to accommodate the set of PHOS folders and tasks (Y.Schutz)
41
7e90ff59 42Revision 1.77 2001/09/04 15:09:11 hristov
43fTreeE->Branch replaced temporary by fTreeE->BranchOld to avoid data corruption in case of many events per file
44
62016442 45Revision 1.76 2001/08/03 14:38:35 morsch
46Use original method to access TreeH.
47
2057d00a 48Revision 1.75 2001/07/28 10:52:27 hristov
49Event number updated correctly (M.Ivanov)
50
fa85c972 51Revision 1.74 2001/07/28 10:39:16 morsch
52GetEventsPerRun() method needed by afterburners added to AliRun.h
53Corresponding setters and getters have been from AliGenerator.
54
6df200c3 55Revision 1.73 2001/07/27 12:34:30 jchudoba
56remove the dummy argument in fStack->GetEvent call
57
a2c28d96 58Revision 1.72 2001/07/03 08:10:57 hristov
59J.Chudoba's changes merged correctly with the HEAD
60
27f087a9 61Revision 1.70 2001/06/29 08:01:36 morsch
62Small correction to the previous.
63
64Revision 1.69 2001/06/28 16:27:50 morsch
65AliReco() with user control of event range.
66
67Revision 1.68 2001/06/11 13:14:40 morsch
68SetAliGenEventHeader() method added.
69
70Revision 1.67 2001/06/07 18:24:50 buncic
71Removed compilation warning in AliConfig initialisation.
72
484836ef 73Revision 1.66 2001/05/22 14:32:40 hristov
74Weird inline removed
75
b5c5eebc 76Revision 1.65 2001/05/21 17:22:51 buncic
77Fixed problem with missing AliConfig while reading galice.root
78
682a4a95 79Revision 1.64 2001/05/16 14:57:22 alibrary
80New files for folders and Stack
81
1e3fad37 82Revision 1.62 2001/04/06 11:12:33 morsch
83Clear fParticles after each event. (Ivana Hrivnacova)
84
ea1f7d5b 85Revision 1.61 2001/03/30 07:04:10 morsch
86Call fGenerator->FinishRun() for final print-outs, cross-section and weight calculations.
87
8e70f139 88Revision 1.60 2001/03/21 18:22:30 hristov
89fParticleFileMap fix (I.Hrivnacova)
90
02a02c36 91Revision 1.59 2001/03/12 17:47:03 hristov
92Changes needed on Sun with CC 5.0
93
5cf7bbad 94Revision 1.58 2001/03/09 14:27:26 morsch
95Fix for multiple events per file: inhibit decrease of size of fParticleFileMap.
96
6781433e 97Revision 1.57 2001/02/23 17:40:23 buncic
98All trees needed for simulation created in RunMC(). TreeR and its branches
99are now created in new RunReco() method.
100
d47c658f 101Revision 1.56 2001/02/14 15:45:20 hristov
102Algorithmic way of getting entry index in fParticleMap. Protection of fParticleFileMap (I.Hrivnacova)
103
a41d61a9 104Revision 1.55 2001/02/12 15:52:54 buncic
105Removed OpenBaseFile().
106
39de14fb 107Revision 1.54 2001/02/07 10:39:05 hristov
108Remove default value for argument
109
1d994b80 110Revision 1.53 2001/02/06 11:02:26 hristov
111New SetTrack interface added, added check for unfilled particles in FinishEvent (I.Hrivnacova)
112
89bbad6f 113Revision 1.52 2001/02/05 16:22:25 buncic
114Added TreeS to GetEvent().
115
82711e7a 116Revision 1.51 2001/02/02 15:16:20 morsch
117SetHighWaterMark method added to mark last particle in event.
118
4d69d91e 119Revision 1.50 2001/01/27 10:32:00 hristov
120Leave the loop when primaries are filled (I.Hrivnacova)
121
cb94ac2a 122Revision 1.49 2001/01/26 19:58:48 hristov
123Major upgrade of AliRoot code
124
2ab0c725 125Revision 1.48 2001/01/17 10:50:50 hristov
126Corrections to destructors
127
e460afec 128Revision 1.47 2000/12/18 10:44:01 morsch
129Possibility to set field map by passing pointer to objet of type AliMagF via
130SetField().
131Example:
132gAlice->SetField(new AliMagFCM("Map2", "$(ALICE_ROOT)/data/field01.dat",2,1.,10.));
133
d8408e76 134Revision 1.46 2000/12/14 19:29:27 fca
135galice.cuts was not read any more
136
327136d2 137Revision 1.45 2000/11/30 07:12:49 alibrary
138Introducing new Rndm and QA classes
139
65fb704d 140Revision 1.44 2000/10/26 13:58:59 morsch
141Add possibility to choose the lego generator (of type AliGeneratorLego or derived) when running
142RunLego(). Default is the base class AliGeneratorLego.
143
0a520a66 144Revision 1.43 2000/10/09 09:43:17 fca
145Special remapping of hits for TPC and TRD. End-of-primary action introduced
146
24de2263 147Revision 1.42 2000/10/02 21:28:14 fca
148Removal of useless dependecies via forward declarations
149
94de3818 150Revision 1.41 2000/07/13 16:19:09 fca
151Mainly coding conventions + some small bug fixes
152
ef42d733 153Revision 1.40 2000/07/12 08:56:25 fca
154Coding convention correction and warning removal
155
8918e700 156Revision 1.39 2000/07/11 18:24:59 fca
157Coding convention corrections + few minor bug fixes
158
aee8290b 159Revision 1.38 2000/06/20 13:05:45 fca
160Writing down the TREE headers before job starts
161
51e0e89d 162Revision 1.37 2000/06/09 20:05:11 morsch
163Introduce possibility to chose magnetic field version 3: AliMagFDM + field02.dat
164
f1b9d7c3 165Revision 1.36 2000/06/08 14:03:58 hristov
166Only one initializer for a default argument
167
d33c0226 168Revision 1.35 2000/06/07 10:13:14 hristov
169Delete only existent objects.
170
d2ecd553 171Revision 1.34 2000/05/18 10:45:38 fca
172Delete Particle Factory properly
173
c222d2b0 174Revision 1.33 2000/05/16 13:10:40 fca
175New method IsNewTrack and fix for a problem in Father-Daughter relations
176
a01a8b12 177Revision 1.32 2000/04/27 10:38:21 fca
178Correct termination of Lego Run and introduce Lego getter in AliRun
179
838edcaf 180Revision 1.31 2000/04/26 10:17:32 fca
181Changes in Lego for G4 compatibility
182
dffd31ef 183Revision 1.30 2000/04/18 19:11:40 fca
184Introduce variable Config.C function signature
185
45189757 186Revision 1.29 2000/04/07 11:12:34 fca
187G4 compatibility changes
188
875c717b 189Revision 1.28 2000/04/05 06:51:06 fca
190Workaround for an HP compiler problem
191
5eb58812 192Revision 1.27 2000/03/22 18:08:07 fca
193Rationalisation of the virtual MC interfaces
194
80762cb1 195Revision 1.26 2000/03/22 13:42:26 fca
196SetGenerator does not replace an existing generator, ResetGenerator does
197
ee1dd322 198Revision 1.25 2000/02/23 16:25:22 fca
199AliVMC and AliGeant3 classes introduced
200ReadEuclid moved from AliRun to AliModule
201
b13db077 202Revision 1.24 2000/01/19 17:17:20 fca
203Introducing a list of lists of hits -- more hits allowed for detector now
204
1cedd08a 205Revision 1.23 1999/12/03 11:14:31 fca
206Fixing previous wrong checking
207
00719c1b 208Revision 1.21 1999/11/25 10:40:08 fca
209Fixing daughters information also in primary tracks
210
ae23d366 211Revision 1.20 1999/10/04 18:08:49 fca
212Adding protection against inconsistent Euclid files
213
3fcc96a1 214Revision 1.19 1999/09/29 07:50:40 fca
215Introduction of the Copyright and cvs Log
216
99d554c8 217*/
218
fe4da5cc 219///////////////////////////////////////////////////////////////////////////////
220// //
221// Control class for Alice C++ //
222// Only one single instance of this class exists. //
223// The object is created in main program aliroot //
224// and is pointed by the global gAlice. //
225// //
8494b010 226// -Supports the list of all Alice Detectors (fModules). //
fe4da5cc 227// -Supports the list of particles (fParticles). //
228// -Supports the Trees. //
229// -Supports the geometry. //
230// -Supports the event display. //
231//Begin_Html
232/*
1439f98e 233<img src="picts/AliRunClass.gif">
fe4da5cc 234*/
235//End_Html
236//Begin_Html
237/*
1439f98e 238<img src="picts/alirun.gif">
fe4da5cc 239*/
240//End_Html
241// //
242///////////////////////////////////////////////////////////////////////////////
243
94de3818 244#include <stdlib.h>
245#include <stdio.h>
246#include <string.h>
2ab0c725 247#include <iostream.h>
94de3818 248
fe4da5cc 249#include <TFile.h>
250#include <TRandom.h>
251#include <TBRIK.h>
fe4da5cc 252#include <TCint.h>
253#include <TSystem.h>
a8f1fb7c 254#include <TObjectTable.h>
94de3818 255#include <TTree.h>
256#include <TGeometry.h>
257#include <TROOT.h>
9e1a0ddb 258#include <TBrowser.h>
259#include <TFolder.h>
3f754d83 260#include <TKey.h>
27f087a9 261#include <TNode.h>
1578254f 262#include "TParticle.h"
fe4da5cc 263#include "AliRun.h"
fe4da5cc 264#include "AliDisplay.h"
875c717b 265#include "AliMC.h"
dffd31ef 266#include "AliLego.h"
aee8290b 267#include "AliMagFC.h"
268#include "AliMagFCM.h"
269#include "AliMagFDM.h"
94de3818 270#include "AliHit.h"
65fb704d 271#include "TRandom3.h"
272#include "AliMCQA.h"
273#include "AliGenerator.h"
274#include "AliLegoGenerator.h"
9e1a0ddb 275#include "AliConfig.h"
276#include "AliStack.h"
277#include "AliGenEventHeader.h"
278#include "AliHeader.h"
94de3818 279
280#include "AliDetector.h"
fe4da5cc 281
fe4da5cc 282AliRun *gAlice;
283
fe4da5cc 284ClassImp(AliRun)
285
286//_____________________________________________________________________________
287AliRun::AliRun()
288{
289 //
290 // Default constructor for AliRun
291 //
9e1a0ddb 292 fHeader = 0;
fe4da5cc 293 fRun = 0;
294 fEvent = 0;
eef4b160 295 fEventNrInRun = 0;
9e1a0ddb 296 fStack = 0;
297 fModules = 0;
fe4da5cc 298 fGenerator = 0;
299 fTreeD = 0;
fe4da5cc 300 fTreeH = 0;
aab9c8d5 301 fTreeTR = 0;
fe4da5cc 302 fTreeE = 0;
303 fTreeR = 0;
2ab0c725 304 fTreeS = 0;
fe4da5cc 305 fGeometry = 0;
306 fDisplay = 0;
307 fField = 0;
9e1a0ddb 308 fMC = 0;
fe4da5cc 309 fNdets = 0;
310 fImedia = 0;
311 fTrRmax = 1.e10;
312 fTrZmax = 1.e10;
fe4da5cc 313 fInitDone = kFALSE;
314 fLego = 0;
1578254f 315 fPDGDB = 0; //Particle factory object!
1cedd08a 316 fHitLists = 0;
45189757 317 fConfigFunction = "\0";
65fb704d 318 fRandom = 0;
319 fMCQA = 0;
320 fTransParName = "\0";
39de14fb 321 fBaseFileName = ".\0";
9e1a0ddb 322 fDebug = 0;
7a16e9cc 323 fTreeDFile = 0;
324 fTreeSFileName = "";
325 fTreeSFile = 0;
326 fTreeSFileName = "";
327 fTreeRFile = 0;
328 fTreeRFileName = "";
fe4da5cc 329}
330
331//_____________________________________________________________________________
332AliRun::AliRun(const char *name, const char *title)
9e1a0ddb 333 : TNamed(name,title)
fe4da5cc 334{
335 //
336 // Constructor for the main processor.
337 // Creates the geometry
338 // Creates the list of Detectors.
339 // Creates the list of particles.
340 //
341 Int_t i;
342
343 gAlice = this;
344 fTreeD = 0;
fe4da5cc 345 fTreeH = 0;
aab9c8d5 346 fTreeTR = 0;
fe4da5cc 347 fTreeE = 0;
348 fTreeR = 0;
2ab0c725 349 fTreeS = 0;
fe4da5cc 350 fTrRmax = 1.e10;
351 fTrZmax = 1.e10;
1141f8e4 352 fGenerator = 0;
fe4da5cc 353 fInitDone = kFALSE;
354 fLego = 0;
355 fField = 0;
45189757 356 fConfigFunction = "Config();";
7a16e9cc 357 fTreeDFile = 0;
358 fTreeSFileName = "";
359 fTreeSFile = 0;
360 fTreeSFileName = "";
361 fTreeRFile = 0;
362 fTreeRFileName = "";
65fb704d 363
364 // Set random number generator
365 gRandom = fRandom = new TRandom3();
2ab0c725 366
367 if (gSystem->Getenv("CONFIG_SEED")) {
368 gRandom->SetSeed((UInt_t)atoi(gSystem->Getenv("CONFIG_SEED")));
369 }
fe4da5cc 370
371 gROOT->GetListOfBrowsables()->Add(this,name);
372 //
9e1a0ddb 373 // Particle stack
374 fStack = new AliStack(10000);
fe4da5cc 375 // create the support list for the various Detectors
8494b010 376 fModules = new TObjArray(77);
fe4da5cc 377 //
378 // Create the TNode geometry for the event display
379
380 BuildSimpleGeometry();
381
9e1a0ddb 382 fHeader = new AliHeader();
fe4da5cc 383 fRun = 0;
384 fEvent = 0;
eef4b160 385 fEventNrInRun = 0;
fe4da5cc 386 //
fe4da5cc 387 fDisplay = 0;
388 //
389 // Create default mag field
390 SetField();
391 //
875c717b 392 fMC = gMC;
fe4da5cc 393 //
394 // Prepare the tracking medium lists
395 fImedia = new TArrayI(1000);
396 for(i=0;i<1000;i++) (*fImedia)[i]=-99;
1578254f 397 //
398 // Make particles
399 fPDGDB = TDatabasePDG::Instance(); //Particle factory object!
9e1a0ddb 400
401 AliConfig::Instance()->Add(fPDGDB);
1cedd08a 402 //
403 // Create HitLists list
404 fHitLists = new TList();
65fb704d 405 //
406 SetTransPar();
39de14fb 407 fBaseFileName = ".\0";
9e1a0ddb 408 //
409 fDebug = 0;
fe4da5cc 410}
411
aee8290b 412
fe4da5cc 413//_____________________________________________________________________________
414AliRun::~AliRun()
415{
416 //
2ab0c725 417 // Default AliRun destructor
fe4da5cc 418 //
7a16e9cc 419 TFile *curfil =0;
420 if(fTreeE)curfil=fTreeE->GetCurrentFile();
fe4da5cc 421 delete fImedia;
422 delete fField;
875c717b 423 delete fMC;
fe4da5cc 424 delete fGeometry;
425 delete fDisplay;
426 delete fGenerator;
427 delete fLego;
428 delete fTreeD;
fe4da5cc 429 delete fTreeH;
aab9c8d5 430 delete fTreeTR;
fe4da5cc 431 delete fTreeE;
432 delete fTreeR;
2ab0c725 433 delete fTreeS;
8494b010 434 if (fModules) {
435 fModules->Delete();
436 delete fModules;
fe4da5cc 437 }
9e1a0ddb 438 delete fStack;
1cedd08a 439 delete fHitLists;
c222d2b0 440 delete fPDGDB;
e460afec 441 delete fMCQA;
9e1a0ddb 442 delete fHeader;
7a16e9cc 443 // avoid to delete TFile objects not owned by this object
444 // avoid multiple deletions
445 if(curfil == fTreeDFile) fTreeDFile=0;
446 if(curfil == fTreeSFile) fTreeSFile=0;
447 if(curfil == fTreeRFile) fTreeRFile=0;
448 if(fTreeSFile == fTreeDFile) fTreeSFile=0;
449 if(fTreeRFile == fTreeDFile) fTreeRFile=0;
450 if(fTreeRFile == fTreeSFile) fTreeRFile=0;
451 if(fTreeDFile){
452 if(fTreeDFile->IsOpen())fTreeDFile->Close();
453 delete fTreeDFile;
454 }
455 if(fTreeSFile){
456 if(fTreeSFile->IsOpen())fTreeSFile->Close();
457 delete fTreeSFile;
458 }
459 if(fTreeRFile){
460 if(fTreeRFile->IsOpen())fTreeRFile->Close();
461 delete fTreeRFile;
462 }
463 if (gROOT->GetListOfBrowsables())
464 gROOT->GetListOfBrowsables()->Remove(this);
fe4da5cc 465}
466
467//_____________________________________________________________________________
468void AliRun::AddHit(Int_t id, Int_t track, Int_t *vol, Float_t *hits) const
469{
470 //
471 // Add a hit to detector id
472 //
8494b010 473 TObjArray &dets = *fModules;
474 if(dets[id]) ((AliModule*) dets[id])->AddHit(track,vol,hits);
fe4da5cc 475}
476
477//_____________________________________________________________________________
478void AliRun::AddDigit(Int_t id, Int_t *tracks, Int_t *digits) const
479{
480 //
481 // Add digit to detector id
482 //
8494b010 483 TObjArray &dets = *fModules;
484 if(dets[id]) ((AliModule*) dets[id])->AddDigit(tracks,digits);
fe4da5cc 485}
486
487//_____________________________________________________________________________
488void AliRun::Browse(TBrowser *b)
489{
490 //
491 // Called when the item "Run" is clicked on the left pane
492 // of the Root browser.
493 // It displays the Root Trees and all detectors.
494 //
9e1a0ddb 495 if(!fStack) fStack=fHeader->Stack();
496 TTree* pTreeK = fStack->TreeK();
497
498 if (pTreeK) b->Add(pTreeK,pTreeK->GetName());
fe4da5cc 499 if (fTreeH) b->Add(fTreeH,fTreeH->GetName());
aab9c8d5 500 if (fTreeTR) b->Add(fTreeTR,fTreeH->GetName());
fe4da5cc 501 if (fTreeD) b->Add(fTreeD,fTreeD->GetName());
502 if (fTreeE) b->Add(fTreeE,fTreeE->GetName());
503 if (fTreeR) b->Add(fTreeR,fTreeR->GetName());
2ab0c725 504 if (fTreeS) b->Add(fTreeS,fTreeS->GetName());
fe4da5cc 505
8494b010 506 TIter next(fModules);
507 AliModule *detector;
508 while((detector = (AliModule*)next())) {
fe4da5cc 509 b->Add(detector,detector->GetName());
510 }
65fb704d 511 b->Add(fMCQA,"AliMCQA");
fe4da5cc 512}
513
514//_____________________________________________________________________________
515void AliRun::Build()
516{
517 //
518 // Initialize Alice geometry
519 // Dummy routine
520 //
521}
522
523//_____________________________________________________________________________
524void AliRun::BuildSimpleGeometry()
525{
526 //
527 // Create a simple TNode geometry used by Root display engine
528 //
529 // Initialise geometry
530 //
531 fGeometry = new TGeometry("AliceGeom","Galice Geometry for Hits");
532 new TMaterial("void","Vacuum",0,0,0); //Everything is void
533 TBRIK *brik = new TBRIK("S_alice","alice volume","void",2000,2000,3000);
534 brik->SetVisibility(0);
535 new TNode("alice","alice","S_alice");
536}
537
538//_____________________________________________________________________________
539void AliRun::CleanDetectors()
540{
541 //
542 // Clean Detectors at the end of event
543 //
8494b010 544 TIter next(fModules);
545 AliModule *detector;
546 while((detector = (AliModule*)next())) {
fe4da5cc 547 detector->FinishEvent();
548 }
549}
550
fe4da5cc 551//_____________________________________________________________________________
552Int_t AliRun::DistancetoPrimitive(Int_t, Int_t)
553{
554 //
555 // Return the distance from the mouse to the AliRun object
556 // Dummy routine
557 //
558 return 9999;
559}
560
561//_____________________________________________________________________________
94de3818 562void AliRun::DumpPart (Int_t i) const
fe4da5cc 563{
564 //
565 // Dumps particle i in the stack
566 //
9e1a0ddb 567 fStack->DumpPart(i);
fe4da5cc 568}
569
570//_____________________________________________________________________________
94de3818 571void AliRun::DumpPStack () const
fe4da5cc 572{
573 //
574 // Dumps the particle stack
575 //
9e1a0ddb 576 fStack->DumpPStack();
fe4da5cc 577}
578
9e1a0ddb 579//_____________________________________________________________________________
d8408e76 580void AliRun::SetField(AliMagF* magField)
581{
582 // Set Magnetic Field Map
583 fField = magField;
584 fField->ReadField();
585}
586
fe4da5cc 587//_____________________________________________________________________________
588void AliRun::SetField(Int_t type, Int_t version, Float_t scale,
589 Float_t maxField, char* filename)
590{
591 //
592 // Set magnetic field parameters
593 // type Magnetic field transport flag 0=no field, 2=helix, 3=Runge Kutta
594 // version Magnetic field map version (only 1 active now)
595 // scale Scale factor for the magnetic field
596 // maxField Maximum value for the magnetic field
597
598 //
599 // --- Sanity check on mag field flags
fe4da5cc 600 if(fField) delete fField;
601 if(version==1) {
d8408e76 602 fField = new AliMagFC("Map1"," ",type,scale,maxField);
f1b9d7c3 603 } else if(version<=2) {
d8408e76 604 fField = new AliMagFCM("Map2-3",filename,type,scale,maxField);
fe4da5cc 605 fField->ReadField();
f1b9d7c3 606 } else if(version==3) {
d8408e76 607 fField = new AliMagFDM("Map4",filename,type,scale,maxField);
f1b9d7c3 608 fField->ReadField();
fe4da5cc 609 } else {
23370b7a 610 Warning("SetField","Invalid map %d\n",version);
fe4da5cc 611 }
612}
fe4da5cc 613
65fb704d 614//_____________________________________________________________________________
615void AliRun::PreTrack()
616{
617 TObjArray &dets = *fModules;
618 AliModule *module;
619
620 for(Int_t i=0; i<=fNdets; i++)
621 if((module = (AliModule*)dets[i]))
622 module->PreTrack();
623
624 fMCQA->PreTrack();
625}
626
627//_____________________________________________________________________________
628void AliRun::PostTrack()
629{
630 TObjArray &dets = *fModules;
631 AliModule *module;
632
633 for(Int_t i=0; i<=fNdets; i++)
634 if((module = (AliModule*)dets[i]))
635 module->PostTrack();
636}
637
fe4da5cc 638//_____________________________________________________________________________
639void AliRun::FinishPrimary()
640{
641 //
642 // Called at the end of each primary track
643 //
644
6c9704e6 645 // static Int_t count=0;
646 // const Int_t times=10;
fe4da5cc 647 // This primary is finished, purify stack
9e1a0ddb 648 fStack->PurifyKine();
fe4da5cc 649
24de2263 650 TIter next(fModules);
651 AliModule *detector;
652 while((detector = (AliModule*)next())) {
653 detector->FinishPrimary();
654 }
655
fe4da5cc 656 // Write out hits if any
657 if (gAlice->TreeH()) {
658 gAlice->TreeH()->Fill();
659 }
aab9c8d5 660
661 // Write out hits if any
662 if (gAlice->TreeTR()) {
663 gAlice->TreeTR()->Fill();
664 }
665
fe4da5cc 666
9e1a0ddb 667 //
668 // if(++count%times==1) gObjectTable->Print();
669}
670
671//_____________________________________________________________________________
672void AliRun::BeginPrimary()
673{
674 //
675 // Called at the beginning of each primary track
676 //
677
fe4da5cc 678 // Reset Hits info
679 gAlice->ResetHits();
aab9c8d5 680 gAlice->ResetTrackReferences();
fe4da5cc 681}
682
683//_____________________________________________________________________________
684void AliRun::FinishEvent()
685{
686 //
687 // Called at the end of the event.
688 //
7fb01480 689
dffd31ef 690 //
691 if(fLego) fLego->FinishEvent();
692
fe4da5cc 693 //Update the energy deposit tables
694 Int_t i;
875c717b 695 for(i=0;i<fEventEnergy.GetSize();i++) {
696 fSummEnergy[i]+=fEventEnergy[i];
697 fSum2Energy[i]+=fEventEnergy[i]*fEventEnergy[i];
fe4da5cc 698 }
9e1a0ddb 699
700
fe4da5cc 701
9e1a0ddb 702 // Update Header information
703
704 fHeader->SetNprimary(fStack->GetNprimary());
705 fHeader->SetNtrack(fStack->GetNtrack());
706
fe4da5cc 707
708 // Write out the kinematics
9e1a0ddb 709 fStack->FinishEvent();
02a02c36 710
fe4da5cc 711 // Write out the event Header information
9e1a0ddb 712 if (fTreeE) {
713 fHeader->SetStack(fStack);
714 fTreeE->Fill();
715 }
fe4da5cc 716
fe4da5cc 717
718 // Write Tree headers
9e1a0ddb 719 TTree* pTreeK = fStack->TreeK();
720 if (pTreeK) pTreeK->Write(0,TObject::kOverwrite);
51e0e89d 721 if (fTreeH) fTreeH->Write(0,TObject::kOverwrite);
aab9c8d5 722 if (fTreeTR) fTreeTR->Write(0,TObject::kOverwrite);
2ab0c725 723
875c717b 724 ++fEvent;
eef4b160 725 ++fEventNrInRun;
fe4da5cc 726}
727
728//_____________________________________________________________________________
729void AliRun::FinishRun()
730{
731 //
732 // Called at the end of the run.
733 //
734
dffd31ef 735 //
736 if(fLego) fLego->FinishRun();
8e70f139 737
fe4da5cc 738 // Clean detector information
8494b010 739 TIter next(fModules);
740 AliModule *detector;
741 while((detector = (AliModule*)next())) {
fe4da5cc 742 detector->FinishRun();
743 }
744
745 //Output energy summary tables
746 EnergySummary();
2ab0c725 747
748 TFile *file = fTreeE->GetCurrentFile();
749
aee8290b 750 file->cd();
2ab0c725 751
51e0e89d 752 fTreeE->Write(0,TObject::kOverwrite);
fe4da5cc 753
2ab0c725 754 // Write AliRun info and all detectors parameters
d47c658f 755 Write(0,TObject::kOverwrite);
682a4a95 756
fe4da5cc 757 // Clean tree information
9e1a0ddb 758
759 fStack->FinishRun();
760
d2ecd553 761 if (fTreeH) {
762 delete fTreeH; fTreeH = 0;
763 }
aab9c8d5 764 if (fTreeTR) {
765 delete fTreeTR; fTreeTR = 0;
766 }
767
d2ecd553 768 if (fTreeD) {
769 delete fTreeD; fTreeD = 0;
770 }
771 if (fTreeR) {
772 delete fTreeR; fTreeR = 0;
773 }
7e90ff59 774// if (fTreeE) {
775// delete fTreeE; fTreeE = 0;
776// }
9e1a0ddb 777 if (fTreeS) {
778 delete fTreeS; fTreeS = 0;
779 }
6df200c3 780 fGenerator->FinishRun();
781
fe4da5cc 782 // Close output file
aee8290b 783 file->Write();
fe4da5cc 784}
785
786//_____________________________________________________________________________
787void AliRun::FlagTrack(Int_t track)
788{
9e1a0ddb 789 // Delegate to stack
fe4da5cc 790 //
9e1a0ddb 791 fStack->FlagTrack(track);
fe4da5cc 792}
793
794//_____________________________________________________________________________
795void AliRun::EnergySummary()
796{
797 //
798 // Print summary of deposited energy
799 //
800
fe4da5cc 801 Int_t ndep=0;
802 Float_t edtot=0;
803 Float_t ed, ed2;
804 Int_t kn, i, left, j, id;
aee8290b 805 const Float_t kzero=0;
9e1a0ddb 806 Int_t ievent=fHeader->GetEvent()+1;
fe4da5cc 807 //
808 // Energy loss information
809 if(ievent) {
810 printf("***************** Energy Loss Information per event (GEV) *****************\n");
875c717b 811 for(kn=1;kn<fEventEnergy.GetSize();kn++) {
812 ed=fSummEnergy[kn];
fe4da5cc 813 if(ed>0) {
875c717b 814 fEventEnergy[ndep]=kn;
fe4da5cc 815 if(ievent>1) {
816 ed=ed/ievent;
875c717b 817 ed2=fSum2Energy[kn];
fe4da5cc 818 ed2=ed2/ievent;
aee8290b 819 ed2=100*TMath::Sqrt(TMath::Max(ed2-ed*ed,kzero))/ed;
fe4da5cc 820 } else
821 ed2=99;
875c717b 822 fSummEnergy[ndep]=ed;
aee8290b 823 fSum2Energy[ndep]=TMath::Min((Float_t) 99.,TMath::Max(ed2,kzero));
fe4da5cc 824 edtot+=ed;
825 ndep++;
826 }
827 }
828 for(kn=0;kn<(ndep-1)/3+1;kn++) {
829 left=ndep-kn*3;
830 for(i=0;i<(3<left?3:left);i++) {
831 j=kn*3+i;
875c717b 832 id=Int_t (fEventEnergy[j]+0.1);
833 printf(" %s %10.3f +- %10.3f%%;",gMC->VolName(id),fSummEnergy[j],fSum2Energy[j]);
fe4da5cc 834 }
835 printf("\n");
836 }
837 //
838 // Relative energy loss in different detectors
839 printf("******************** Relative Energy Loss per event ********************\n");
840 printf("Total energy loss per event %10.3f GeV\n",edtot);
841 for(kn=0;kn<(ndep-1)/5+1;kn++) {
842 left=ndep-kn*5;
843 for(i=0;i<(5<left?5:left);i++) {
844 j=kn*5+i;
875c717b 845 id=Int_t (fEventEnergy[j]+0.1);
846 printf(" %s %10.3f%%;",gMC->VolName(id),100*fSummEnergy[j]/edtot);
fe4da5cc 847 }
848 printf("\n");
849 }
850 for(kn=0;kn<75;kn++) printf("*");
851 printf("\n");
852 }
853 //
854 // Reset the TArray's
875c717b 855 // fEventEnergy.Set(0);
856 // fSummEnergy.Set(0);
857 // fSum2Energy.Set(0);
fe4da5cc 858}
859
860//_____________________________________________________________________________
94de3818 861AliModule *AliRun::GetModule(const char *name) const
fe4da5cc 862{
863 //
864 // Return pointer to detector from name
865 //
8494b010 866 return (AliModule*)fModules->FindObject(name);
fe4da5cc 867}
868
a68348e9 869//_____________________________________________________________________________
94de3818 870AliDetector *AliRun::GetDetector(const char *name) const
a68348e9 871{
872 //
873 // Return pointer to detector from name
874 //
875 return (AliDetector*)fModules->FindObject(name);
876}
877
fe4da5cc 878//_____________________________________________________________________________
94de3818 879Int_t AliRun::GetModuleID(const char *name) const
fe4da5cc 880{
881 //
882 // Return galice internal detector identifier from name
883 //
23370b7a 884 Int_t i=-1;
885 TObject *mod=fModules->FindObject(name);
886 if(mod) i=fModules->IndexOf(mod);
887 return i;
fe4da5cc 888}
889
890//_____________________________________________________________________________
891Int_t AliRun::GetEvent(Int_t event)
892{
893 //
894 // Connect the Trees Kinematics and Hits for event # event
895 // Set branch addresses
896 //
fe4da5cc 897
898 // Reset existing structures
fe4da5cc 899 ResetHits();
aab9c8d5 900 ResetTrackReferences();
fe4da5cc 901 ResetDigits();
2ab0c725 902 ResetSDigits();
fe4da5cc 903
904 // Delete Trees already connected
ccf7a81f 905 if (fTreeH) { delete fTreeH; fTreeH = 0;}
aab9c8d5 906 if (fTreeTR) { delete fTreeTR; fTreeTR = 0;}
ccf7a81f 907 if (fTreeD) { delete fTreeD; fTreeD = 0;}
908 if (fTreeR) { delete fTreeR; fTreeR = 0;}
909 if (fTreeS) { delete fTreeS; fTreeS = 0;}
59fe9bd2 910
9e1a0ddb 911 // Create the particle stack
912 if (fHeader) delete fHeader;
913 fHeader = 0;
914
59fe9bd2 915 // Get header from file
9e1a0ddb 916 if(fTreeE) {
917 fTreeE->SetBranchAddress("Header", &fHeader);
ccf7a81f 918
919 if (!fTreeE->GetEntry(event)) {
920 Error("GetEvent","Cannot find event:%d\n",event);
921 return -1;
922 }
9e1a0ddb 923 }
ccf7a81f 924 else {
9e1a0ddb 925 Error("GetEvent","Cannot find Header Tree (TE)\n");
ccf7a81f 926 return -1;
927 }
9e1a0ddb 928
ccf7a81f 929 // Get the stack from the header, set fStack to 0 if it
930 // fails to get event
9e1a0ddb 931 //
2ab0c725 932 TFile *file = fTreeE->GetCurrentFile();
fe4da5cc 933 char treeName[20];
6df200c3 934
2ab0c725 935 file->cd();
936
50a6540a 937 if (fStack) delete fStack;
938 fStack = fHeader->Stack();
939 if (fStack) {
940 if (!fStack->GetEvent(event)) fStack = 0;
941 }
942
fe4da5cc 943 // Get Hits Tree header from file
944 sprintf(treeName,"TreeH%d",event);
2057d00a 945 fTreeH = (TTree*)gDirectory->Get(treeName);
946 if (!fTreeH) {
3d8c4988 947 //Error("GetEvent","cannot find Hits Tree for event:%d\n",event);
948 Warning("GetEvent","cannot find Hits Tree for event:%d\n",event);
2057d00a 949 }
950
aab9c8d5 951 // Get TracReferences Tree header from file
952 sprintf(treeName,"TreeTR%d",event);
953 fTreeTR = (TTree*)gDirectory->Get(treeName);
954 if (!fTreeTR) {
3d8c4988 955 //Error("GetEvent","cannot find TrackReferences Tree for event:%d\n",event);
956 Warning("GetEvent","cannot find TrackReferences Tree for event:%d\n",event);
aab9c8d5 957 }
958
7a16e9cc 959 // get current file name and compare with names containing trees S,D,R
960 TString curfilname=(TString)fTreeE->GetCurrentFile()->GetName();
961 if(fTreeDFileName==curfilname)fTreeDFileName="";
962 if(fTreeSFileName==curfilname)fTreeSFileName="";
963 if(fTreeRFileName==curfilname)fTreeRFileName="";
fe4da5cc 964 // Get Digits Tree header from file
965 sprintf(treeName,"TreeD%d",event);
7a16e9cc 966
967 if (!fTreeDFile && fTreeDFileName != "") {
968 InitTreeFile("D",fTreeDFileName);
969 }
970 if (fTreeDFile) {
971 fTreeD = (TTree*)fTreeDFile->Get(treeName);
972 } else {
973 fTreeD = (TTree*)file->Get(treeName);
974 }
fe4da5cc 975 if (!fTreeD) {
2ab0c725 976 // Warning("GetEvent","cannot find Digits Tree for event:%d\n",event);
fe4da5cc 977 }
7a16e9cc 978 if(fTreeDFileName != ""){
979 if(fTreeDFileName==fTreeSFileName) {
980 fTreeSFileName = "";
981 fTreeSFile = fTreeDFile;
982 }
983 if(fTreeDFileName==fTreeRFileName) {
984 fTreeRFileName = "";
985 fTreeRFile = fTreeDFile;
986 }
987 }
82711e7a 988
989 file->cd();
990
991 // Get SDigits Tree header from file
992 sprintf(treeName,"TreeS%d",event);
7a16e9cc 993 if (!fTreeSFile && fTreeSFileName != "") {
994 InitTreeFile("S",fTreeSFileName);
995 }
996 if (fTreeSFile) {
997 fTreeS = (TTree*)fTreeSFile->Get(treeName);
998 } else {
999 fTreeS = (TTree*)gDirectory->Get(treeName);
1000 }
82711e7a 1001 if (!fTreeS) {
1002 // Warning("GetEvent","cannot find SDigits Tree for event:%d\n",event);
1003 }
1004
7a16e9cc 1005 if(fTreeSFileName != ""){
1006 if(fTreeSFileName==fTreeRFileName){
1007 fTreeRFileName = "";
1008 fTreeRFile = fTreeSFile;
1009 }
1010 }
1011
2ab0c725 1012 file->cd();
fe4da5cc 1013
1014 // Get Reconstruct Tree header from file
1015 sprintf(treeName,"TreeR%d",event);
7a16e9cc 1016 if (!fTreeRFile && fTreeRFileName != "") {
1017 InitTreeFile("R",fTreeRFileName);
1018 }
1019 if(fTreeRFile) {
1020 fTreeR = (TTree*)fTreeRFile->Get(treeName);
1021 } else {
1022 fTreeR = (TTree*)gDirectory->Get(treeName);
1023 }
fe4da5cc 1024 if (!fTreeR) {
1025 // printf("WARNING: cannot find Reconstructed Tree for event:%d\n",event);
1026 }
2ab0c725 1027
1028 file->cd();
fe4da5cc 1029
1030 // Set Trees branch addresses
8494b010 1031 TIter next(fModules);
1032 AliModule *detector;
1033 while((detector = (AliModule*)next())) {
fe4da5cc 1034 detector->SetTreeAddress();
1035 }
9e1a0ddb 1036
fa85c972 1037 fEvent=event; //MI change
1038
eef4b160 1039 return fHeader->GetNtrack();
fe4da5cc 1040}
1041
1042//_____________________________________________________________________________
1043TGeometry *AliRun::GetGeometry()
1044{
1045 //
1046 // Import Alice geometry from current file
1047 // Return pointer to geometry object
1048 //
1049 if (!fGeometry) fGeometry = (TGeometry*)gDirectory->Get("AliceGeom");
1050 //
1051 // Unlink and relink nodes in detectors
1052 // This is bad and there must be a better way...
1053 //
fe4da5cc 1054
8494b010 1055 TIter next(fModules);
1056 AliModule *detector;
1057 while((detector = (AliModule*)next())) {
fe4da5cc 1058 TList *dnodes=detector->Nodes();
1059 Int_t j;
1060 TNode *node, *node1;
1061 for ( j=0; j<dnodes->GetSize(); j++) {
1062 node = (TNode*) dnodes->At(j);
52d0ab00 1063 node1 = fGeometry->GetNode(node->GetName());
fe4da5cc 1064 dnodes->Remove(node);
1065 dnodes->AddAt(node1,j);
1066 }
1067 }
1068 return fGeometry;
1069}
1070
1071//_____________________________________________________________________________
1072void AliRun::GetNextTrack(Int_t &mtrack, Int_t &ipart, Float_t *pmom,
1073 Float_t &e, Float_t *vpos, Float_t *polar,
1074 Float_t &tof)
1075{
9e1a0ddb 1076 // Delegate to stack
fe4da5cc 1077 //
9e1a0ddb 1078 fStack->GetNextTrack(mtrack, ipart, pmom, e, vpos, polar, tof);
fe4da5cc 1079}
1080
1081//_____________________________________________________________________________
9e1a0ddb 1082Int_t AliRun::GetPrimary(Int_t track) const
fe4da5cc 1083{
1084 //
1085 // return number of primary that has generated track
1086 //
9e1a0ddb 1087 return fStack->GetPrimary(track);
fe4da5cc 1088}
1089
1090//_____________________________________________________________________________
875c717b 1091void AliRun::InitMC(const char *setup)
fe4da5cc 1092{
1093 //
1094 // Initialize the Alice setup
1095 //
1096
51e0e89d 1097 if(fInitDone) {
1098 Warning("Init","Cannot initialise AliRun twice!\n");
1099 return;
1100 }
2ab0c725 1101
fe4da5cc 1102 gROOT->LoadMacro(setup);
45189757 1103 gInterpreter->ProcessLine(fConfigFunction.Data());
fe4da5cc 1104
2ab0c725 1105
cfce8870 1106 gMC->DefineParticles(); //Create standard MC particles
fe4da5cc 1107
1108 TObject *objfirst, *objlast;
1109
23370b7a 1110 fNdets = fModules->GetLast()+1;
1111
fe4da5cc 1112 //
875c717b 1113 //=================Create Materials and geometry
1114 gMC->Init();
1115
65fb704d 1116 // Added also after in case of interactive initialisation of modules
1117 fNdets = fModules->GetLast()+1;
1118
8494b010 1119 TIter next(fModules);
1120 AliModule *detector;
1121 while((detector = (AliModule*)next())) {
fe4da5cc 1122 detector->SetTreeAddress();
1123 objlast = gDirectory->GetList()->Last();
1124
fe4da5cc 1125 // Add Detector histograms in Detector list of histograms
1126 if (objlast) objfirst = gDirectory->GetList()->After(objlast);
1127 else objfirst = gDirectory->GetList()->First();
1128 while (objfirst) {
1129 detector->Histograms()->Add(objfirst);
1130 objfirst = gDirectory->GetList()->After(objfirst);
1131 }
1132 }
327136d2 1133 ReadTransPar(); //Read the cuts for all materials
fe4da5cc 1134
1135 MediaTable(); //Build the special IMEDIA table
1136
fe4da5cc 1137 //Initialise geometry deposition table
875c717b 1138 fEventEnergy.Set(gMC->NofVolumes()+1);
1139 fSummEnergy.Set(gMC->NofVolumes()+1);
1140 fSum2Energy.Set(gMC->NofVolumes()+1);
fe4da5cc 1141
fe4da5cc 1142 //Compute cross-sections
875c717b 1143 gMC->BuildPhysics();
fe4da5cc 1144
1145 //Write Geometry object to current file.
1146 fGeometry->Write();
1147
1148 fInitDone = kTRUE;
51e0e89d 1149
65fb704d 1150 fMCQA = new AliMCQA(fNdets);
1151
7a16e9cc 1152// JCH note: the following line is useless, AliConfig instance is already
1153// created in AliMC ctor. But it does not hurt.
27f087a9 1154 AliConfig::Instance();
51e0e89d 1155 //
1156 // Save stuff at the beginning of the file to avoid file corruption
1157 Write();
fe4da5cc 1158}
1159
1160//_____________________________________________________________________________
1161void AliRun::MediaTable()
1162{
1163 //
1164 // Built media table to get from the media number to
1165 // the detector id
1166 //
ad51aeb0 1167 Int_t kz, nz, idt, lz, i, k, ind;
1168 // Int_t ibeg;
fe4da5cc 1169 TObjArray &dets = *gAlice->Detectors();
8494b010 1170 AliModule *det;
fe4da5cc 1171 //
1172 // For all detectors
1173 for (kz=0;kz<fNdets;kz++) {
1174 // If detector is defined
8494b010 1175 if((det=(AliModule*) dets[kz])) {
ad51aeb0 1176 TArrayI &idtmed = *(det->GetIdtmed());
1177 for(nz=0;nz<100;nz++) {
fe4da5cc 1178 // Find max and min material number
ad51aeb0 1179 if((idt=idtmed[nz])) {
fe4da5cc 1180 det->LoMedium() = det->LoMedium() < idt ? det->LoMedium() : idt;
1181 det->HiMedium() = det->HiMedium() > idt ? det->HiMedium() : idt;
1182 }
1183 }
1184 if(det->LoMedium() > det->HiMedium()) {
1185 det->LoMedium() = 0;
1186 det->HiMedium() = 0;
1187 } else {
1188 if(det->HiMedium() > fImedia->GetSize()) {
ad51aeb0 1189 Error("MediaTable","Increase fImedia from %d to %d",
1190 fImedia->GetSize(),det->HiMedium());
fe4da5cc 1191 return;
1192 }
1193 // Tag all materials in rage as belonging to detector kz
1194 for(lz=det->LoMedium(); lz<= det->HiMedium(); lz++) {
1195 (*fImedia)[lz]=kz;
1196 }
1197 }
1198 }
1199 }
1200 //
1201 // Print summary table
1202 printf(" Traking media ranges:\n");
1203 for(i=0;i<(fNdets-1)/6+1;i++) {
1204 for(k=0;k< (6<fNdets-i*6?6:fNdets-i*6);k++) {
1205 ind=i*6+k;
8494b010 1206 det=(AliModule*)dets[ind];
fe4da5cc 1207 if(det)
1208 printf(" %6s: %3d -> %3d;",det->GetName(),det->LoMedium(),
1209 det->HiMedium());
1210 else
1211 printf(" %6s: %3d -> %3d;","NULL",0,0);
1212 }
1213 printf("\n");
1214 }
1215}
1216
1217//____________________________________________________________________________
1218void AliRun::SetGenerator(AliGenerator *generator)
ee1dd322 1219{
1220 //
1221 // Load the event generator
1222 //
1223 if(!fGenerator) fGenerator = generator;
1224}
1225
1226//____________________________________________________________________________
1227void AliRun::ResetGenerator(AliGenerator *generator)
fe4da5cc 1228{
1229 //
1230 // Load the event generator
1231 //
b13db077 1232 if(fGenerator)
838edcaf 1233 if(generator)
1234 Warning("ResetGenerator","Replacing generator %s with %s\n",
1235 fGenerator->GetName(),generator->GetName());
1236 else
1237 Warning("ResetGenerator","Replacing generator %s with NULL\n",
1238 fGenerator->GetName());
b13db077 1239 fGenerator = generator;
fe4da5cc 1240}
1241
1242//____________________________________________________________________________
65fb704d 1243void AliRun::SetTransPar(char *filename)
1244{
1245 fTransParName = filename;
1246}
1247
2ab0c725 1248//____________________________________________________________________________
1249void AliRun::SetBaseFile(char *filename)
1250{
39de14fb 1251 fBaseFileName = filename;
2ab0c725 1252}
1253
65fb704d 1254//____________________________________________________________________________
1255void AliRun::ReadTransPar()
fe4da5cc 1256{
1257 //
1258 // Read filename to set the transport parameters
1259 //
1260
fe4da5cc 1261
aee8290b 1262 const Int_t kncuts=10;
1263 const Int_t knflags=11;
1264 const Int_t knpars=kncuts+knflags;
1265 const char kpars[knpars][7] = {"CUTGAM" ,"CUTELE","CUTNEU","CUTHAD","CUTMUO",
fe4da5cc 1266 "BCUTE","BCUTM","DCUTE","DCUTM","PPCUTM","ANNI",
1267 "BREM","COMP","DCAY","DRAY","HADR","LOSS",
1268 "MULS","PAIR","PHOT","RAYL"};
1269 char line[256];
ad51aeb0 1270 char detName[7];
fe4da5cc 1271 char* filtmp;
aee8290b 1272 Float_t cut[kncuts];
1273 Int_t flag[knflags];
fe4da5cc 1274 Int_t i, itmed, iret, ktmed, kz;
1275 FILE *lun;
1276 //
1277 // See whether the file is there
65fb704d 1278 filtmp=gSystem->ExpandPathName(fTransParName.Data());
fe4da5cc 1279 lun=fopen(filtmp,"r");
1280 delete [] filtmp;
1281 if(!lun) {
65fb704d 1282 Warning("ReadTransPar","File %s does not exist!\n",fTransParName.Data());
fe4da5cc 1283 return;
1284 }
1285 //
9e1a0ddb 1286 if(fDebug) {
1287 printf(" "); for(i=0;i<60;i++) printf("*"); printf("\n");
1288 printf(" *%59s\n","*");
1289 printf(" * Please check carefully what you are doing!%10s\n","*");
1290 printf(" *%59s\n","*");
1291 }
fe4da5cc 1292 //
1293 while(1) {
1294 // Initialise cuts and flags
aee8290b 1295 for(i=0;i<kncuts;i++) cut[i]=-99;
1296 for(i=0;i<knflags;i++) flag[i]=-99;
fe4da5cc 1297 itmed=0;
1298 for(i=0;i<256;i++) line[i]='\0';
1299 // Read up to the end of line excluded
1300 iret=fscanf(lun,"%[^\n]",line);
1301 if(iret<0) {
1302 //End of file
1303 fclose(lun);
9e1a0ddb 1304 if(fDebug){
1305 printf(" *%59s\n","*");
1306 printf(" "); for(i=0;i<60;i++) printf("*"); printf("\n");
1307 }
fe4da5cc 1308 return;
1309 }
1310 // Read the end of line
1311 fscanf(lun,"%*c");
1312 if(!iret) continue;
1313 if(line[0]=='*') continue;
1314 // Read the numbers
ad51aeb0 1315 iret=sscanf(line,"%s %d %f %f %f %f %f %f %f %f %f %f %d %d %d %d %d %d %d %d %d %d %d",
1316 detName,&itmed,&cut[0],&cut[1],&cut[2],&cut[3],&cut[4],&cut[5],&cut[6],&cut[7],&cut[8],
1317 &cut[9],&flag[0],&flag[1],&flag[2],&flag[3],&flag[4],&flag[5],&flag[6],&flag[7],
1318 &flag[8],&flag[9],&flag[10]);
fe4da5cc 1319 if(!iret) continue;
1320 if(iret<0) {
1321 //reading error
65fb704d 1322 Warning("ReadTransPar","Error reading file %s\n",fTransParName.Data());
fe4da5cc 1323 continue;
1324 }
ad51aeb0 1325 // Check that the module exist
1326 AliModule *mod = GetModule(detName);
1327 if(mod) {
1328 // Get the array of media numbers
1329 TArrayI &idtmed = *mod->GetIdtmed();
1330 // Check that the tracking medium code is valid
1331 if(0<=itmed && itmed < 100) {
1332 ktmed=idtmed[itmed];
1333 if(!ktmed) {
65fb704d 1334 Warning("ReadTransPar","Invalid tracking medium code %d for %s\n",itmed,mod->GetName());
ad51aeb0 1335 continue;
fe4da5cc 1336 }
ad51aeb0 1337 // Set energy thresholds
aee8290b 1338 for(kz=0;kz<kncuts;kz++) {
ad51aeb0 1339 if(cut[kz]>=0) {
9e1a0ddb 1340 if(fDebug) printf(" * %-6s set to %10.3E for tracking medium code %4d for %s\n",
aee8290b 1341 kpars[kz],cut[kz],itmed,mod->GetName());
1342 gMC->Gstpar(ktmed,kpars[kz],cut[kz]);
ad51aeb0 1343 }
fe4da5cc 1344 }
ad51aeb0 1345 // Set transport mechanisms
aee8290b 1346 for(kz=0;kz<knflags;kz++) {
ad51aeb0 1347 if(flag[kz]>=0) {
9e1a0ddb 1348 if(fDebug) printf(" * %-6s set to %10d for tracking medium code %4d for %s\n",
aee8290b 1349 kpars[kncuts+kz],flag[kz],itmed,mod->GetName());
1350 gMC->Gstpar(ktmed,kpars[kncuts+kz],Float_t(flag[kz]));
ad51aeb0 1351 }
1352 }
1353 } else {
65fb704d 1354 Warning("ReadTransPar","Invalid medium code %d *\n",itmed);
ad51aeb0 1355 continue;
fe4da5cc 1356 }
1357 } else {
9e1a0ddb 1358 if(fDebug) printf("%s::ReadTransParModule: %s not present\n",ClassName(),detName);
fe4da5cc 1359 continue;
1360 }
1361 }
1362}
1363
2ab0c725 1364
1365//_____________________________________________________________________________
9e1a0ddb 1366void AliRun::MakeTree(Option_t *option, const char *file)
fe4da5cc 1367{
1368 //
1369 // Create the ROOT trees
1370 // Loop on all detectors to create the Root branch (if any)
1371 //
1372
b13db077 1373 char hname[30];
fe4da5cc 1374 //
1375 // Analyse options
5cf7bbad 1376 const char *oK = strstr(option,"K");
1377 const char *oH = strstr(option,"H");
aab9c8d5 1378 const char *oTR = strstr(option,"T");
5cf7bbad 1379 const char *oE = strstr(option,"E");
1380 const char *oD = strstr(option,"D");
1381 const char *oR = strstr(option,"R");
1382 const char *oS = strstr(option,"S");
fe4da5cc 1383 //
9e1a0ddb 1384
1385 TDirectory *cwd = gDirectory;
1386
1387 TBranch *branch = 0;
2ab0c725 1388
9e1a0ddb 1389 if (oK) fStack->MakeTree(fEvent, file);
1390
1391 if (oE && !fTreeE) {
1392 fTreeE = new TTree("TE","Header");
62016442 1393 // branch = fTreeE->Branch("Header", "AliHeader", &fHeader, 4000, 0);
21e74529 1394 branch = fTreeE->Branch("Header", "AliHeader", &fHeader, 4000, 0);
9e1a0ddb 1395 branch->SetAutoDelete(kFALSE);
1396 TFolder *folder = (TFolder *)gROOT->FindObjectAny("/Folders/RunMC/Event/Header");
1397 if (folder) folder->Add(fHeader);
1398// branch = fTreeE->Branch("Stack","AliStack", &fStack, 4000, 0);
1399// branch->SetAutoDelete(kFALSE);
1400// if (folder) folder->Add(fStack);
1401 fTreeE->Write(0,TObject::kOverwrite);
b13db077 1402 }
9e1a0ddb 1403
1404 if (file && branch) {
1405 char * outFile = new char[strlen(gAlice->GetBaseFile())+strlen(file)+2];
1406 sprintf(outFile,"%s/%s",GetBaseFile(),file);
1407 branch->SetFile(outFile);
1408 TIter next( branch->GetListOfBranches());
1409 while ((branch=(TBranch*)next())) {
1410 branch->SetFile(outFile);
1411 }
1412 if (GetDebug()>1)
1413 printf("* MakeBranch * Diverting Branch %s to file %s\n", branch->GetName(),file);
1414 cwd->cd();
1415 delete outFile;
1416 }
1417
aee8290b 1418 if (oH && !fTreeH) {
b13db077 1419 sprintf(hname,"TreeH%d",fEvent);
1420 fTreeH = new TTree(hname,"Hits");
1421 fTreeH->SetAutoSave(1000000000); //no autosave
9e1a0ddb 1422 fTreeH->Write(0,TObject::kOverwrite);
b13db077 1423 }
aab9c8d5 1424
1425 if (oTR && !fTreeTR) {
1426 sprintf(hname,"TreeTR%d",fEvent);
1427 fTreeTR = new TTree(hname,"TrackReferences");
1428 fTreeTR->SetAutoSave(1000000000); //no autosave
1429 fTreeTR->Write(0,TObject::kOverwrite);
1430 }
1431
aee8290b 1432 if (oD && !fTreeD) {
b13db077 1433 sprintf(hname,"TreeD%d",fEvent);
1434 fTreeD = new TTree(hname,"Digits");
9e1a0ddb 1435 fTreeD->Write(0,TObject::kOverwrite);
b13db077 1436 }
2ab0c725 1437 if (oS && !fTreeS) {
1438 sprintf(hname,"TreeS%d",fEvent);
1439 fTreeS = new TTree(hname,"SDigits");
9e1a0ddb 1440 fTreeS->Write(0,TObject::kOverwrite);
2ab0c725 1441 }
aee8290b 1442 if (oR && !fTreeR) {
b13db077 1443 sprintf(hname,"TreeR%d",fEvent);
1444 fTreeR = new TTree(hname,"Reconstruction");
9e1a0ddb 1445 fTreeR->Write(0,TObject::kOverwrite);
b13db077 1446 }
9e1a0ddb 1447
fe4da5cc 1448 //
1449 // Create a branch for hits/digits for each detector
1450 // Each branch is a TClonesArray. Each data member of the Hits classes
1451 // will be in turn a subbranch of the detector master branch
8494b010 1452 TIter next(fModules);
1453 AliModule *detector;
1454 while((detector = (AliModule*)next())) {
aab9c8d5 1455 if (oH) detector->MakeBranch(option,file);
1456 if (oTR) detector->MakeBranchTR(option,file);
fe4da5cc 1457 }
aab9c8d5 1458
1459
fe4da5cc 1460}
1461
1462//_____________________________________________________________________________
b5c5eebc 1463TParticle* AliRun::Particle(Int_t i)
2ab0c725 1464{
9e1a0ddb 1465 return fStack->Particle(i);
fe4da5cc 1466}
1467
1468//_____________________________________________________________________________
dffd31ef 1469void AliRun::BeginEvent()
fe4da5cc 1470{
9e1a0ddb 1471 // Clean-up previous event
1472 // Energy scores
1473 fEventEnergy.Reset();
1474 // Clean detector information
1475 CleanDetectors();
1476 // Reset stack info
1477 fStack->Reset();
1478
1479
fe4da5cc 1480 //
1481 // Reset all Detectors & kinematics & trees
1482 //
59fe9bd2 1483 char hname[30];
dffd31ef 1484 //
9e1a0ddb 1485 // Initialise event header
eef4b160 1486 fHeader->Reset(fRun,fEvent,fEventNrInRun);
9e1a0ddb 1487 //
1488 fStack->BeginEvent(fEvent);
dffd31ef 1489
1490 //
1491 if(fLego) {
1492 fLego->BeginEvent();
1493 return;
1494 }
1495
59fe9bd2 1496 //
9e1a0ddb 1497
fe4da5cc 1498 ResetHits();
aab9c8d5 1499 ResetTrackReferences();
fe4da5cc 1500 ResetDigits();
2ab0c725 1501 ResetSDigits();
fe4da5cc 1502
fe4da5cc 1503
59fe9bd2 1504 if(fTreeH) {
1505 fTreeH->Reset();
875c717b 1506 sprintf(hname,"TreeH%d",fEvent);
59fe9bd2 1507 fTreeH->SetName(hname);
1508 }
aab9c8d5 1509
1510 if(fTreeTR) {
1511 fTreeTR->Reset();
1512 sprintf(hname,"TreeTR%d",fEvent);
1513 fTreeTR->SetName(hname);
1514 }
1515
59fe9bd2 1516 if(fTreeD) {
1517 fTreeD->Reset();
875c717b 1518 sprintf(hname,"TreeD%d",fEvent);
59fe9bd2 1519 fTreeD->SetName(hname);
9e1a0ddb 1520 fTreeD->Write(0,TObject::kOverwrite);
59fe9bd2 1521 }
2ab0c725 1522 if(fTreeS) {
1523 fTreeS->Reset();
1524 sprintf(hname,"TreeS%d",fEvent);
1525 fTreeS->SetName(hname);
9e1a0ddb 1526 fTreeS->Write(0,TObject::kOverwrite);
2ab0c725 1527 }
59fe9bd2 1528 if(fTreeR) {
1529 fTreeR->Reset();
875c717b 1530 sprintf(hname,"TreeR%d",fEvent);
59fe9bd2 1531 fTreeR->SetName(hname);
9e1a0ddb 1532 fTreeR->Write(0,TObject::kOverwrite);
59fe9bd2 1533 }
fe4da5cc 1534}
fe4da5cc 1535//_____________________________________________________________________________
1536void AliRun::ResetDigits()
1537{
1538 //
1539 // Reset all Detectors digits
1540 //
8494b010 1541 TIter next(fModules);
1542 AliModule *detector;
1543 while((detector = (AliModule*)next())) {
fe4da5cc 1544 detector->ResetDigits();
1545 }
1546}
1547
2ab0c725 1548//_____________________________________________________________________________
1549void AliRun::ResetSDigits()
1550{
1551 //
1552 // Reset all Detectors digits
1553 //
1554 TIter next(fModules);
1555 AliModule *detector;
1556 while((detector = (AliModule*)next())) {
1557 detector->ResetSDigits();
1558 }
1559}
1560
fe4da5cc 1561//_____________________________________________________________________________
1562void AliRun::ResetHits()
1563{
1564 //
1565 // Reset all Detectors hits
1566 //
8494b010 1567 TIter next(fModules);
1568 AliModule *detector;
1569 while((detector = (AliModule*)next())) {
fe4da5cc 1570 detector->ResetHits();
1571 }
1572}
1573
aab9c8d5 1574//_____________________________________________________________________________
1575void AliRun::ResetTrackReferences()
1576{
1577 //
1578 // Reset all Detectors hits
1579 //
1580 TIter next(fModules);
1581 AliModule *detector;
1582 while((detector = (AliModule*)next())) {
1583 detector->ResetTrackReferences();
1584 }
1585}
1586
1587
1588
fe4da5cc 1589//_____________________________________________________________________________
1590void AliRun::ResetPoints()
1591{
1592 //
1593 // Reset all Detectors points
1594 //
8494b010 1595 TIter next(fModules);
1596 AliModule *detector;
1597 while((detector = (AliModule*)next())) {
fe4da5cc 1598 detector->ResetPoints();
1599 }
1600}
1601
1602//_____________________________________________________________________________
875c717b 1603void AliRun::RunMC(Int_t nevent, const char *setup)
fe4da5cc 1604{
1605 //
1606 // Main function to be called to process a galice run
1607 // example
1608 // Root > gAlice.Run();
1609 // a positive number of events will cause the finish routine
1610 // to be called
1611 //
6df200c3 1612 fEventsPerRun = nevent;
fe4da5cc 1613 // check if initialisation has been done
875c717b 1614 if (!fInitDone) InitMC(setup);
fe4da5cc 1615
fe4da5cc 1616 // Create the Root Tree with one branch per detector
2ab0c725 1617
aab9c8d5 1618 MakeTree("ESDRT"); // MI change
d47c658f 1619
2ab0c725 1620 if (gSystem->Getenv("CONFIG_SPLIT_FILE")) {
39de14fb 1621 MakeTree("K","Kine.root");
1622 MakeTree("H","Hits.root");
2ab0c725 1623 } else {
d47c658f 1624 MakeTree("KH");
2ab0c725 1625 }
fe4da5cc 1626
875c717b 1627 gMC->ProcessRun(nevent);
1628
fe4da5cc 1629 // End of this run, close files
80762cb1 1630 if(nevent>0) FinishRun();
fe4da5cc 1631}
1632
d47c658f 1633//_____________________________________________________________________________
27f087a9 1634void AliRun::RunReco(const char *selected, Int_t first, Int_t last)
d47c658f 1635{
1636 //
1637 // Main function to be called to reconstruct Alice event
27f087a9 1638 //
1639 cout << "Found "<< gAlice->TreeE()->GetEntries() << "events" << endl;
1640 Int_t nFirst = first;
1641 Int_t nLast = (last < 0)? (Int_t) gAlice->TreeE()->GetEntries() : last;
1642
1643 for (Int_t nevent = nFirst; nevent <= nLast; nevent++) {
1644 cout << "Processing event "<< nevent << endl;
9e1a0ddb 1645 GetEvent(nevent);
1646 // MakeTree("R");
1647 Digits2Reco(selected);
1648 }
d47c658f 1649}
1650
2ab0c725 1651//_____________________________________________________________________________
1652
1653void AliRun::Hits2Digits(const char *selected)
1654{
9e1a0ddb 1655
d47c658f 1656 // Convert Hits to sumable digits
1657 //
9e1a0ddb 1658 for (Int_t nevent=0; nevent<gAlice->TreeE()->GetEntries(); nevent++) {
1659 GetEvent(nevent);
1660 // MakeTree("D");
1661 Hits2SDigits(selected);
1662 SDigits2Digits(selected);
1663 }
2ab0c725 1664}
1665
d47c658f 1666
2ab0c725 1667//_____________________________________________________________________________
1668
d47c658f 1669void AliRun::Tree2Tree(Option_t *option, const char *selected)
2ab0c725 1670{
1671 //
d47c658f 1672 // Function to transform the content of
1673 //
1674 // - TreeH to TreeS (option "S")
1675 // - TreeS to TreeD (option "D")
1676 // - TreeD to TreeR (option "R")
1677 //
1678 // If multiple options are specified ("SDR"), transformation will be done in sequence for
1679 // selected detector and for all detectors if none is selected (detector string
1680 // can contain blank separated list of detector names).
2ab0c725 1681
2ab0c725 1682
5cf7bbad 1683 const char *oS = strstr(option,"S");
1684 const char *oD = strstr(option,"D");
1685 const char *oR = strstr(option,"R");
2ab0c725 1686
9e1a0ddb 1687 TObjArray *detectors = Detectors();
2ab0c725 1688
1689 TIter next(detectors);
1690
d47c658f 1691 AliDetector *detector = 0;
2ab0c725 1692
1693 TDirectory *cwd = gDirectory;
1694
d47c658f 1695 char outFile[32];
2ab0c725 1696
1697 while((detector = (AliDetector*)next())) {
d47c658f 1698 if (selected)
2ab0c725 1699 if (strcmp(detector->GetName(),selected)) continue;
2ab0c725 1700 if (detector->IsActive()){
d47c658f 1701 if (gSystem->Getenv("CONFIG_SPLIT_FILE")) {
1702 if (oS) {
1703 sprintf(outFile,"SDigits.%s.root",detector->GetName());
1704 detector->MakeBranch("S",outFile);
1705 }
1706 if (oD) {
1707 sprintf(outFile,"Digits.%s.root",detector->GetName());
1708 detector->MakeBranch("D",outFile);
1709 }
1710 if (oR) {
1711 sprintf(outFile,"Reco.%s.root",detector->GetName());
1712 detector->MakeBranch("R",outFile);
1713 }
2ab0c725 1714 } else {
d47c658f 1715 detector->MakeBranch(option);
1716 }
1717
1718 cwd->cd();
1719
9e1a0ddb 1720 if (oS) {
1721 cout << "Hits2SDigits: Processing " << detector->GetName() << "..." << endl;
d47c658f 1722 detector->Hits2SDigits();
9e1a0ddb 1723 }
1724 if (oD) {
1725 cout << "SDigits2Digits: Processing " << detector->GetName() << "..." << endl;
1726 detector->SDigits2Digits();
1727 }
1728 if (oR) {
1729 cout << "Digits2Reco: Processing " << detector->GetName() << "..." << endl;
d47c658f 1730 detector->Digits2Reco();
9e1a0ddb 1731 }
d47c658f 1732
9e1a0ddb 1733 cwd->cd();
1734 }
2ab0c725 1735 }
1736}
1737
d47c658f 1738
fe4da5cc 1739//_____________________________________________________________________________
0a520a66 1740void AliRun::RunLego(const char *setup, Int_t nc1, Float_t c1min,
1741 Float_t c1max,Int_t nc2,Float_t c2min,Float_t c2max,
1742 Float_t rmin,Float_t rmax,Float_t zmax, AliLegoGenerator* gener)
fe4da5cc 1743{
1744 //
1745 // Generates lego plots of:
1746 // - radiation length map phi vs theta
1747 // - radiation length map phi vs eta
1748 // - interaction length map
1749 // - g/cm2 length map
1750 //
1751 // ntheta bins in theta, eta
1752 // themin minimum angle in theta (degrees)
1753 // themax maximum angle in theta (degrees)
1754 // nphi bins in phi
1755 // phimin minimum angle in phi (degrees)
1756 // phimax maximum angle in phi (degrees)
1757 // rmin minimum radius
1758 // rmax maximum radius
1759 //
1760 //
1761 // The number of events generated = ntheta*nphi
1762 // run input parameters in macro setup (default="Config.C")
1763 //
1764 // Use macro "lego.C" to visualize the 3 lego plots in spherical coordinates
1765 //Begin_Html
1766 /*
1439f98e 1767 <img src="picts/AliRunLego1.gif">
fe4da5cc 1768 */
1769 //End_Html
1770 //Begin_Html
1771 /*
1439f98e 1772 <img src="picts/AliRunLego2.gif">
fe4da5cc 1773 */
1774 //End_Html
1775 //Begin_Html
1776 /*
1439f98e 1777 <img src="picts/AliRunLego3.gif">
fe4da5cc 1778 */
1779 //End_Html
1780 //
1781
1782 // check if initialisation has been done
875c717b 1783 if (!fInitDone) InitMC(setup);
838edcaf 1784 //Save current generator
1785 AliGenerator *gen=Generator();
1786
0a520a66 1787 // Set new generator
1788 if (!gener) gener = new AliLegoGenerator();
1789 ResetGenerator(gener);
1790 //
1791 // Configure Generator
1792 gener->SetRadiusRange(rmin, rmax);
1793 gener->SetZMax(zmax);
1794 gener->SetCoor1Range(nc1, c1min, c1max);
1795 gener->SetCoor2Range(nc2, c2min, c2max);
1796
1797
b13db077 1798 //Create Lego object
0a520a66 1799 fLego = new AliLego("lego",gener);
b13db077 1800
dffd31ef 1801 //Prepare MC for Lego Run
1802 gMC->InitLego();
1803
b13db077 1804 //Run Lego Object
0a520a66 1805
1806 gMC->ProcessRun(nc1*nc2+1);
fe4da5cc 1807
1808 // Create only the Root event Tree
80762cb1 1809 MakeTree("E");
fe4da5cc 1810
1811 // End of this run, close files
80762cb1 1812 FinishRun();
0a520a66 1813 // Restore current generator
1814 ResetGenerator(gen);
838edcaf 1815 // Delete Lego Object
1816 delete fLego; fLego=0;
fe4da5cc 1817}
1818
94de3818 1819//_____________________________________________________________________________
1820void AliRun::SetConfigFunction(const char * config)
1821{
1822 //
1823 // Set the signature of the function contained in Config.C to configure
1824 // the run
1825 //
1826 fConfigFunction=config;
1827}
1828
fe4da5cc 1829//_____________________________________________________________________________
1830void AliRun::SetCurrentTrack(Int_t track)
1831{
1832 //
1833 // Set current track number
1834 //
9e1a0ddb 1835 fStack->SetCurrentTrack(track);
fe4da5cc 1836}
1837
1838//_____________________________________________________________________________
1578254f 1839void AliRun::SetTrack(Int_t done, Int_t parent, Int_t pdg, Float_t *pmom,
fe4da5cc 1840 Float_t *vpos, Float_t *polar, Float_t tof,
47c8bcbe 1841 AliMCProcess mech, Int_t &ntr, Float_t weight, Int_t is)
fe4da5cc 1842{
9e1a0ddb 1843// Delegate to stack
1844//
1845 fStack->SetTrack(done, parent, pdg, pmom, vpos, polar, tof,
47c8bcbe 1846 mech, ntr, weight, is);
fe4da5cc 1847}
1848
89bbad6f 1849//_____________________________________________________________________________
1850void AliRun::SetTrack(Int_t done, Int_t parent, Int_t pdg,
1851 Double_t px, Double_t py, Double_t pz, Double_t e,
1852 Double_t vx, Double_t vy, Double_t vz, Double_t tof,
1853 Double_t polx, Double_t poly, Double_t polz,
47c8bcbe 1854 AliMCProcess mech, Int_t &ntr, Float_t weight, Int_t is)
89bbad6f 1855{
9e1a0ddb 1856 // Delegate to stack
89bbad6f 1857 //
9e1a0ddb 1858 fStack->SetTrack(done, parent, pdg, px, py, pz, e, vx, vy, vz, tof,
47c8bcbe 1859 polx, poly, polz, mech, ntr, weight, is);
9e1a0ddb 1860
89bbad6f 1861}
1862
1863//_____________________________________________________________________________
4d69d91e 1864void AliRun::SetHighWaterMark(const Int_t nt)
1865{
1866 //
1867 // Set high water mark for last track in event
9e1a0ddb 1868 fStack->SetHighWaterMark(nt);
4d69d91e 1869}
1870
fe4da5cc 1871//_____________________________________________________________________________
1872void AliRun::KeepTrack(const Int_t track)
1873{
1874 //
9e1a0ddb 1875 // Delegate to stack
fe4da5cc 1876 //
9e1a0ddb 1877 fStack->KeepTrack(track);
fe4da5cc 1878}
1879
1880//_____________________________________________________________________________
875c717b 1881void AliRun::StepManager(Int_t id)
fe4da5cc 1882{
1883 //
1884 // Called at every step during transport
1885 //
1886
fe4da5cc 1887 //
1888 // --- If lego option, do it and leave
b13db077 1889 if (fLego)
fe4da5cc 1890 fLego->StepManager();
b13db077 1891 else {
1892 Int_t copy;
1893 //Update energy deposition tables
875c717b 1894 AddEnergyDeposit(gMC->CurrentVolID(copy),gMC->Edep());
fe4da5cc 1895
b13db077 1896 //Call the appropriate stepping routine;
1897 AliModule *det = (AliModule*)fModules->At(id);
3f754d83 1898 if(det && det->StepManagerIsEnabled()) {
65fb704d 1899 fMCQA->StepManager(id);
1900 det->StepManager();
1901 }
fe4da5cc 1902 }
fe4da5cc 1903}
1904
fe4da5cc 1905//_____________________________________________________________________________
1906void AliRun::Streamer(TBuffer &R__b)
1907{
eef4b160 1908 // Stream an object of class AliRun.
2ab0c725 1909
7e90ff59 1910 if (R__b.IsReading()) {
1911 if (!gAlice) gAlice = this;
eef4b160 1912
7e90ff59 1913 AliRun::Class()->ReadBuffer(R__b, this);
1914 //
1915 gROOT->GetListOfBrowsables()->Add(this,"Run");
eef4b160 1916
7e90ff59 1917 fTreeE = (TTree*)gDirectory->Get("TE");
1918 if (fTreeE) {
eef4b160 1919 fTreeE->SetBranchAddress("Header", &fHeader);
7e90ff59 1920 }
eef4b160 1921
7e90ff59 1922 else Error("Streamer","cannot find Header Tree\n");
1923 fTreeE->GetEntry(0);
eef4b160 1924
7e90ff59 1925 gRandom = fRandom;
1926 } else {
eef4b160 1927 AliRun::Class()->WriteBuffer(R__b, this);
1928 }
2ab0c725 1929}
1930
9e1a0ddb 1931
1932//___________________________________________________________________________
1933Int_t AliRun::CurrentTrack() const {
1934 //
1935 // Returns current track
1936 //
1937 return fStack->CurrentTrack();
1938}
1939
1940//___________________________________________________________________________
1941Int_t AliRun::GetNtrack() const {
1942 //
1943 // Returns number of tracks in stack
1944 //
1945 return fStack->GetNtrack();
1946}
1947
1948//___________________________________________________________________________
1949TObjArray* AliRun::Particles() {
1950 //
1951 // Returns pointer to Particles array
1952 //
1953 return fStack->Particles();
1954}
1955
1956//___________________________________________________________________________
1957TTree* AliRun::TreeK() {
1958 //
1959 // Returns pointer to the TreeK array
1960 //
1961 return fStack->TreeK();
1962}
27f087a9 1963
7a16e9cc 1964////////////////////////////////////////////////////////////////////////
27f087a9 1965void AliRun::SetGenEventHeader(AliGenEventHeader* header)
1966{
1967 fHeader->SetGenEventHeader(header);
1968}
7a16e9cc 1969////////////////////////////////////////////////////////////////////////
1970TFile* AliRun::InitFile(TString fileName)
1971{
1972//
1973// create the file where the whole tree will be saved
1974//
1975 TDirectory *wd = gDirectory;
1976 TFile* file = TFile::Open(fileName,"update");
1977 gDirectory = wd;
1978 if (!file->IsOpen()) {
1979 Error("Cannot open file, %s\n",fileName);
1980 return 0;
1981 }
1982 return file;
1983}
1984
1985////////////////////////////////////////////////////////////////////////
1986TFile* AliRun::InitTreeFile(Option_t *option, TString fileName)
1987{
1988 //
1989 // create the file where one of the following trees will be saved
1990 // trees: S,D,R
1991 //
1992 Bool_t oS = (strstr(option,"S")!=0);
1993 Bool_t oR = (strstr(option,"R")!=0);
1994 Bool_t oD = (strstr(option,"D")!=0);
1995 Bool_t none = !(oS || oR || oD);
1996 TFile *ptr=0;
1997 if(none){
1998 Warning("InitTreeFile","wrong option - nothing done\n");
1999 } else {
2000 if (oS){
2001 fTreeSFileName=fileName;
2002 if (fTreeSFile) {
2003 if (fTreeSFile->IsOpen()) {
2004 Warning("InitTreeSFile","File already opened");
2005 }
2006 } else {
2007 fTreeSFile = InitFile(fileName);
2008 }
2009 ptr = fTreeSFile;
2010 }
2011 if (oD){
2012 fTreeDFileName=fileName;
2013 if(oS)fTreeDFile = fTreeSFile;
2014 if (fTreeDFile) {
2015 if (fTreeDFile->IsOpen()) {
2016 Warning("InitTreeDFile","File already opened");
2017 }
2018 } else {
2019 fTreeDFile = InitFile(fileName);
2020 }
2021 ptr = fTreeDFile;
2022 }
2023 if (oR){
2024 fTreeRFileName=fileName;
2025 if(oS)fTreeRFile = fTreeSFile;
2026 if(oD)fTreeRFile = fTreeDFile;
2027 if (fTreeRFile) {
2028 if (fTreeRFile->IsOpen()) {
2029 Warning("InitTreeRFile","File already opened");
2030 }
2031 } else {
2032 fTreeRFile = InitFile(fileName);
2033 }
2034 ptr = fTreeRFile;
2035 }
2036 }
2037 return ptr;
2038}
2039////////////////////////////////////////////////////////////////////////
2040void AliRun::MakeTree(Option_t *option, TFile *file)
2041{
2042 //
2043 // Create some trees in the separate file
2044 //
2045 const char *oD = strstr(option,"D");
2046 const char *oR = strstr(option,"R");
2047 const char *oS = strstr(option,"S");
2048
2049 TDirectory *cwd = gDirectory;
2050 char hname[30];
2051
2052 if (oD) {
2053 delete fTreeD;
2054 sprintf(hname,"TreeD%d",fEvent);
2055 file->cd();
2056 fTreeD = static_cast<TTree*>(file->Get("hname"));
2057 if (!fTreeD) {
2058 fTreeD = new TTree(hname,"Digits");
2059 fTreeD->Write(0,TObject::kOverwrite);
2060 }
2061 cwd->cd();
2062 }
2063 if (oS) {
2064 delete fTreeS;
2065 sprintf(hname,"TreeS%d",fEvent);
2066 file->cd();
2067 fTreeS = static_cast<TTree*>(file->Get("hname"));
2068 if (!fTreeS) {
2069 fTreeS = new TTree(hname,"SDigits");
2070 fTreeS->Write(0,TObject::kOverwrite);
2071 }
2072 cwd->cd();
2073 }
2074
2075 if (oR) {
2076 delete fTreeR;
2077 sprintf(hname,"TreeR%d",fEvent);
2078 file->cd();
2079 fTreeR = static_cast<TTree*>(file->Get("hname"));
2080 if (!fTreeR) {
2081 fTreeR = new TTree(hname,"RecPoint");
2082 fTreeR->Write(0,TObject::kOverwrite);
2083 }
2084 cwd->cd();
2085 }
2086}
2087////////////////////////////////////////////////////////////////////////
2088
2089
2090
2091
2092
2093
2094
2095