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