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