Introduction of the Copyright and cvs Log
[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$
18*/
19
fe4da5cc 20///////////////////////////////////////////////////////////////////////////////
21// //
22// Control class for Alice C++ //
23// Only one single instance of this class exists. //
24// The object is created in main program aliroot //
25// and is pointed by the global gAlice. //
26// //
8494b010 27// -Supports the list of all Alice Detectors (fModules). //
fe4da5cc 28// -Supports the list of particles (fParticles). //
29// -Supports the Trees. //
30// -Supports the geometry. //
31// -Supports the event display. //
32//Begin_Html
33/*
1439f98e 34<img src="picts/AliRunClass.gif">
fe4da5cc 35*/
36//End_Html
37//Begin_Html
38/*
1439f98e 39<img src="picts/alirun.gif">
fe4da5cc 40*/
41//End_Html
42// //
43///////////////////////////////////////////////////////////////////////////////
44
45#include <TFile.h>
46#include <TRandom.h>
47#include <TBRIK.h>
48#include <TNode.h>
fe4da5cc 49#include <TCint.h>
50#include <TSystem.h>
a8f1fb7c 51#include <TObjectTable.h>
fe4da5cc 52
1578254f 53#include "TParticle.h"
fe4da5cc 54#include "AliRun.h"
fe4da5cc 55#include "AliDisplay.h"
56
57#include "AliCallf77.h"
58
59#include <stdlib.h>
60#include <stdio.h>
61#include <string.h>
62
63AliRun *gAlice;
64
65static AliHeader *header;
66
67#ifndef WIN32
68
69# define rxgtrak rxgtrak_
70# define rxstrak rxstrak_
71# define rxkeep rxkeep_
72# define rxouth rxouth_
fe4da5cc 73#else
74
75# define rxgtrak RXGTRAK
76# define rxstrak RXSTRAK
77# define rxkeep RXKEEP
78# define rxouth RXOUTH
fe4da5cc 79#endif
80
81static TArrayF sEventEnergy;
82static TArrayF sSummEnergy;
83static TArrayF sSum2Energy;
84
fe4da5cc 85ClassImp(AliRun)
86
87//_____________________________________________________________________________
88AliRun::AliRun()
89{
90 //
91 // Default constructor for AliRun
92 //
93 header=&fHeader;
94 fRun = 0;
95 fEvent = 0;
96 fCurrent = -1;
8494b010 97 fModules = 0;
fe4da5cc 98 fGenerator = 0;
99 fTreeD = 0;
100 fTreeK = 0;
101 fTreeH = 0;
102 fTreeE = 0;
103 fTreeR = 0;
104 fParticles = 0;
105 fGeometry = 0;
106 fDisplay = 0;
107 fField = 0;
108 fMC = 0;
109 fNdets = 0;
110 fImedia = 0;
111 fTrRmax = 1.e10;
112 fTrZmax = 1.e10;
fe4da5cc 113 fInitDone = kFALSE;
114 fLego = 0;
1578254f 115 fPDGDB = 0; //Particle factory object!
fe4da5cc 116}
117
118//_____________________________________________________________________________
119AliRun::AliRun(const char *name, const char *title)
120 : TNamed(name,title)
121{
122 //
123 // Constructor for the main processor.
124 // Creates the geometry
125 // Creates the list of Detectors.
126 // Creates the list of particles.
127 //
128 Int_t i;
129
130 gAlice = this;
131 fTreeD = 0;
132 fTreeK = 0;
133 fTreeH = 0;
134 fTreeE = 0;
135 fTreeR = 0;
136 fTrRmax = 1.e10;
137 fTrZmax = 1.e10;
1141f8e4 138 fGenerator = 0;
fe4da5cc 139 fInitDone = kFALSE;
140 fLego = 0;
141 fField = 0;
142
143 gROOT->GetListOfBrowsables()->Add(this,name);
144 //
145 // create the support list for the various Detectors
8494b010 146 fModules = new TObjArray(77);
fe4da5cc 147 //
148 // Create the TNode geometry for the event display
149
150 BuildSimpleGeometry();
151
152
153 fNtrack=0;
154 fHgwmk=0;
155 fCurrent=-1;
156 header=&fHeader;
157 fRun = 0;
158 fEvent = 0;
159 //
160 // Create the particle stack
1578254f 161 fParticles = new TClonesArray("TParticle",100);
fe4da5cc 162
163 fDisplay = 0;
164 //
165 // Create default mag field
166 SetField();
167 //
cfce8870 168 fMC = gMC;
fe4da5cc 169 //
170 // Prepare the tracking medium lists
171 fImedia = new TArrayI(1000);
172 for(i=0;i<1000;i++) (*fImedia)[i]=-99;
1578254f 173 //
174 // Make particles
175 fPDGDB = TDatabasePDG::Instance(); //Particle factory object!
fe4da5cc 176}
177
178//_____________________________________________________________________________
179AliRun::~AliRun()
180{
181 //
182 // Defaullt AliRun destructor
183 //
fe4da5cc 184 delete fImedia;
185 delete fField;
186 delete fMC;
187 delete fGeometry;
188 delete fDisplay;
189 delete fGenerator;
190 delete fLego;
191 delete fTreeD;
192 delete fTreeK;
193 delete fTreeH;
194 delete fTreeE;
195 delete fTreeR;
8494b010 196 if (fModules) {
197 fModules->Delete();
198 delete fModules;
fe4da5cc 199 }
200 if (fParticles) {
201 fParticles->Delete();
202 delete fParticles;
203 }
204}
205
206//_____________________________________________________________________________
207void AliRun::AddHit(Int_t id, Int_t track, Int_t *vol, Float_t *hits) const
208{
209 //
210 // Add a hit to detector id
211 //
8494b010 212 TObjArray &dets = *fModules;
213 if(dets[id]) ((AliModule*) dets[id])->AddHit(track,vol,hits);
fe4da5cc 214}
215
216//_____________________________________________________________________________
217void AliRun::AddDigit(Int_t id, Int_t *tracks, Int_t *digits) const
218{
219 //
220 // Add digit to detector id
221 //
8494b010 222 TObjArray &dets = *fModules;
223 if(dets[id]) ((AliModule*) dets[id])->AddDigit(tracks,digits);
fe4da5cc 224}
225
226//_____________________________________________________________________________
227void AliRun::Browse(TBrowser *b)
228{
229 //
230 // Called when the item "Run" is clicked on the left pane
231 // of the Root browser.
232 // It displays the Root Trees and all detectors.
233 //
234 if (fTreeK) b->Add(fTreeK,fTreeK->GetName());
235 if (fTreeH) b->Add(fTreeH,fTreeH->GetName());
236 if (fTreeD) b->Add(fTreeD,fTreeD->GetName());
237 if (fTreeE) b->Add(fTreeE,fTreeE->GetName());
238 if (fTreeR) b->Add(fTreeR,fTreeR->GetName());
239
8494b010 240 TIter next(fModules);
241 AliModule *detector;
242 while((detector = (AliModule*)next())) {
fe4da5cc 243 b->Add(detector,detector->GetName());
244 }
245}
246
247//_____________________________________________________________________________
248void AliRun::Build()
249{
250 //
251 // Initialize Alice geometry
252 // Dummy routine
253 //
254}
255
256//_____________________________________________________________________________
257void AliRun::BuildSimpleGeometry()
258{
259 //
260 // Create a simple TNode geometry used by Root display engine
261 //
262 // Initialise geometry
263 //
264 fGeometry = new TGeometry("AliceGeom","Galice Geometry for Hits");
265 new TMaterial("void","Vacuum",0,0,0); //Everything is void
266 TBRIK *brik = new TBRIK("S_alice","alice volume","void",2000,2000,3000);
267 brik->SetVisibility(0);
268 new TNode("alice","alice","S_alice");
269}
270
271//_____________________________________________________________________________
272void AliRun::CleanDetectors()
273{
274 //
275 // Clean Detectors at the end of event
276 //
8494b010 277 TIter next(fModules);
278 AliModule *detector;
279 while((detector = (AliModule*)next())) {
fe4da5cc 280 detector->FinishEvent();
281 }
282}
283
284//_____________________________________________________________________________
285void AliRun::CleanParents()
286{
287 //
288 // Clean Particles stack.
1578254f 289 // Set parent/daughter relations
fe4da5cc 290 //
291 TClonesArray &particles = *(gAlice->Particles());
1578254f 292 TParticle *part;
fe4da5cc 293 int i;
294 for(i=0; i<fNtrack; i++) {
1578254f 295 part = (TParticle *)particles.UncheckedAt(i);
296 if(!part->TestBit(Daughters_Bit)) {
297 part->SetFirstDaughter(-1);
298 part->SetLastDaughter(-1);
fe4da5cc 299 }
300 }
301}
302
303//_____________________________________________________________________________
304Int_t AliRun::DistancetoPrimitive(Int_t, Int_t)
305{
306 //
307 // Return the distance from the mouse to the AliRun object
308 // Dummy routine
309 //
310 return 9999;
311}
312
313//_____________________________________________________________________________
314void AliRun::DumpPart (Int_t i)
315{
316 //
317 // Dumps particle i in the stack
318 //
319 TClonesArray &particles = *fParticles;
1578254f 320 ((TParticle*) particles[i])->Print();
fe4da5cc 321}
322
323//_____________________________________________________________________________
324void AliRun::DumpPStack ()
325{
326 //
327 // Dumps the particle stack
328 //
329 TClonesArray &particles = *fParticles;
330 printf(
331 "\n\n=======================================================================\n");
332 for (Int_t i=0;i<fNtrack;i++)
333 {
1578254f 334 printf("-> %d ",i); ((TParticle*) particles[i])->Print();
fe4da5cc 335 printf("--------------------------------------------------------------\n");
336 }
337 printf(
338 "\n=======================================================================\n\n");
339}
340
341//_____________________________________________________________________________
342void AliRun::SetField(Int_t type, Int_t version, Float_t scale,
343 Float_t maxField, char* filename)
344{
345 //
346 // Set magnetic field parameters
347 // type Magnetic field transport flag 0=no field, 2=helix, 3=Runge Kutta
348 // version Magnetic field map version (only 1 active now)
349 // scale Scale factor for the magnetic field
350 // maxField Maximum value for the magnetic field
351
352 //
353 // --- Sanity check on mag field flags
354 if(type<0 || type > 2) {
23370b7a 355 Warning("SetField",
356 "Invalid magnetic field flag: %5d; Helix tracking chosen instead\n"
fe4da5cc 357 ,type);
358 type=2;
359 }
360 if(fField) delete fField;
361 if(version==1) {
362 fField = new AliMagFC("Map1"," ",type,version,scale,maxField);
363 } else if(version<=3) {
364 fField = new AliMagFCM("Map2-3",filename,type,version,scale,maxField);
365 fField->ReadField();
366 } else {
23370b7a 367 Warning("SetField","Invalid map %d\n",version);
fe4da5cc 368 }
369}
370
371//_____________________________________________________________________________
372void AliRun::FillTree()
373{
374 //
375 // Fills all AliRun TTrees
376 //
377 if (fTreeK) fTreeK->Fill();
378 if (fTreeH) fTreeH->Fill();
379 if (fTreeD) fTreeD->Fill();
380 if (fTreeR) fTreeR->Fill();
381}
382
383//_____________________________________________________________________________
384void AliRun::FinishPrimary()
385{
386 //
387 // Called at the end of each primary track
388 //
389
6c9704e6 390 // static Int_t count=0;
391 // const Int_t times=10;
fe4da5cc 392 // This primary is finished, purify stack
393 gAlice->PurifyKine();
394
395 // Write out hits if any
396 if (gAlice->TreeH()) {
397 gAlice->TreeH()->Fill();
398 }
399
400 // Reset Hits info
401 gAlice->ResetHits();
a8f1fb7c 402
403 //
404 // if(++count%times==1) gObjectTable->Print();
fe4da5cc 405}
406
407//_____________________________________________________________________________
408void AliRun::FinishEvent()
409{
410 //
411 // Called at the end of the event.
412 //
413
414 //Update the energy deposit tables
415 Int_t i;
416 for(i=0;i<sEventEnergy.GetSize();i++) {
417 sSummEnergy[i]+=sEventEnergy[i];
418 sSum2Energy[i]+=sEventEnergy[i]*sEventEnergy[i];
419 }
420 sEventEnergy.Reset();
421
422 // Clean detector information
423 CleanDetectors();
424
425 // Write out the kinematics
426 if (fTreeK) {
427 CleanParents();
428 fTreeK->Fill();
429 }
430
431 // Write out the digits
432 if (fTreeD) {
433 fTreeD->Fill();
434 ResetDigits();
435 }
436
437 // Write out reconstructed clusters
438 if (fTreeR) {
439 fTreeR->Fill();
440 }
441
442 // Write out the event Header information
443 if (fTreeE) fTreeE->Fill();
444
445 // Reset stack info
446 ResetStack();
447
448 // Write Tree headers
59fe9bd2 449 // Int_t ievent = fHeader.GetEvent();
450 // char hname[30];
451 // sprintf(hname,"TreeK%d",ievent);
452 if (fTreeK) fTreeK->Write();
453 // sprintf(hname,"TreeH%d",ievent);
454 if (fTreeH) fTreeH->Write();
455 // sprintf(hname,"TreeD%d",ievent);
456 if (fTreeD) fTreeD->Write();
457 // sprintf(hname,"TreeR%d",ievent);
458 if (fTreeR) fTreeR->Write();
fe4da5cc 459}
460
461//_____________________________________________________________________________
462void AliRun::FinishRun()
463{
464 //
465 // Called at the end of the run.
466 //
467
468 // Clean detector information
8494b010 469 TIter next(fModules);
470 AliModule *detector;
471 while((detector = (AliModule*)next())) {
fe4da5cc 472 detector->FinishRun();
473 }
474
475 //Output energy summary tables
476 EnergySummary();
477
478 // file is retrieved from whatever tree
479 TFile *File = 0;
480 if (fTreeK) File = fTreeK->GetCurrentFile();
481 if ((!File) && (fTreeH)) File = fTreeH->GetCurrentFile();
482 if ((!File) && (fTreeD)) File = fTreeD->GetCurrentFile();
483 if ((!File) && (fTreeE)) File = fTreeE->GetCurrentFile();
484 if( NULL==File ) {
485 Error("FinishRun","There isn't root file!");
486 exit(1);
487 }
488 File->cd();
489 fTreeE->Write();
490
491 // Clean tree information
492 delete fTreeK; fTreeK = 0;
493 delete fTreeH; fTreeH = 0;
494 delete fTreeD; fTreeD = 0;
495 delete fTreeR; fTreeR = 0;
496 delete fTreeE; fTreeE = 0;
497
498 // Write AliRun info and all detectors parameters
499 Write();
500
501 // Close output file
502 File->Write();
503 File->Close();
504}
505
506//_____________________________________________________________________________
507void AliRun::FlagTrack(Int_t track)
508{
509 //
510 // Flags a track and all its family tree to be kept
511 //
512 int curr;
1578254f 513 TParticle *particle;
fe4da5cc 514
515 curr=track;
516 while(1) {
1578254f 517 particle=(TParticle*)fParticles->UncheckedAt(curr);
fe4da5cc 518
519 // If the particle is flagged the three from here upward is saved already
520 if(particle->TestBit(Keep_Bit)) return;
521
522 // Save this particle
523 particle->SetBit(Keep_Bit);
524
525 // Move to father if any
1578254f 526 if((curr=particle->GetFirstMother())==-1) return;
fe4da5cc 527 }
528}
529
530//_____________________________________________________________________________
531void AliRun::EnergySummary()
532{
533 //
534 // Print summary of deposited energy
535 //
536
fe4da5cc 537 Int_t ndep=0;
538 Float_t edtot=0;
539 Float_t ed, ed2;
540 Int_t kn, i, left, j, id;
541 const Float_t zero=0;
542 Int_t ievent=fHeader.GetEvent()+1;
543 //
544 // Energy loss information
545 if(ievent) {
546 printf("***************** Energy Loss Information per event (GEV) *****************\n");
547 for(kn=1;kn<sEventEnergy.GetSize();kn++) {
548 ed=sSummEnergy[kn];
549 if(ed>0) {
550 sEventEnergy[ndep]=kn;
551 if(ievent>1) {
552 ed=ed/ievent;
553 ed2=sSum2Energy[kn];
554 ed2=ed2/ievent;
555 ed2=100*TMath::Sqrt(TMath::Max(ed2-ed*ed,zero))/ed;
556 } else
557 ed2=99;
558 sSummEnergy[ndep]=ed;
559 sSum2Energy[ndep]=TMath::Min((Float_t) 99.,TMath::Max(ed2,zero));
560 edtot+=ed;
561 ndep++;
562 }
563 }
564 for(kn=0;kn<(ndep-1)/3+1;kn++) {
565 left=ndep-kn*3;
566 for(i=0;i<(3<left?3:left);i++) {
567 j=kn*3+i;
568 id=Int_t (sEventEnergy[j]+0.1);
cfce8870 569 printf(" %s %10.3f +- %10.3f%%;",gMC->VolName(id),sSummEnergy[j],sSum2Energy[j]);
fe4da5cc 570 }
571 printf("\n");
572 }
573 //
574 // Relative energy loss in different detectors
575 printf("******************** Relative Energy Loss per event ********************\n");
576 printf("Total energy loss per event %10.3f GeV\n",edtot);
577 for(kn=0;kn<(ndep-1)/5+1;kn++) {
578 left=ndep-kn*5;
579 for(i=0;i<(5<left?5:left);i++) {
580 j=kn*5+i;
581 id=Int_t (sEventEnergy[j]+0.1);
cfce8870 582 printf(" %s %10.3f%%;",gMC->VolName(id),100*sSummEnergy[j]/edtot);
fe4da5cc 583 }
584 printf("\n");
585 }
586 for(kn=0;kn<75;kn++) printf("*");
587 printf("\n");
588 }
589 //
590 // Reset the TArray's
591 sEventEnergy.Set(0);
592 sSummEnergy.Set(0);
593 sSum2Energy.Set(0);
594}
595
596//_____________________________________________________________________________
8494b010 597AliModule *AliRun::GetModule(const char *name)
fe4da5cc 598{
599 //
600 // Return pointer to detector from name
601 //
8494b010 602 return (AliModule*)fModules->FindObject(name);
fe4da5cc 603}
604
a68348e9 605//_____________________________________________________________________________
606AliDetector *AliRun::GetDetector(const char *name)
607{
608 //
609 // Return pointer to detector from name
610 //
611 return (AliDetector*)fModules->FindObject(name);
612}
613
fe4da5cc 614//_____________________________________________________________________________
8494b010 615Int_t AliRun::GetModuleID(const char *name)
fe4da5cc 616{
617 //
618 // Return galice internal detector identifier from name
619 //
23370b7a 620 Int_t i=-1;
621 TObject *mod=fModules->FindObject(name);
622 if(mod) i=fModules->IndexOf(mod);
623 return i;
fe4da5cc 624}
625
626//_____________________________________________________________________________
627Int_t AliRun::GetEvent(Int_t event)
628{
629 //
630 // Connect the Trees Kinematics and Hits for event # event
631 // Set branch addresses
632 //
fe4da5cc 633
634 // Reset existing structures
635 ResetStack();
636 ResetHits();
637 ResetDigits();
638
639 // Delete Trees already connected
640 if (fTreeK) delete fTreeK;
641 if (fTreeH) delete fTreeH;
642 if (fTreeD) delete fTreeD;
643 if (fTreeR) delete fTreeR;
59fe9bd2 644
645 // Get header from file
646 if(fTreeE) fTreeE->GetEntry(event);
647 else Error("GetEvent","Cannot file Header Tree\n");
fe4da5cc 648
649 // Get Kine Tree from file
650 char treeName[20];
651 sprintf(treeName,"TreeK%d",event);
652 fTreeK = (TTree*)gDirectory->Get(treeName);
653 if (fTreeK) fTreeK->SetBranchAddress("Particles", &fParticles);
23370b7a 654 else Error("GetEvent","cannot find Kine Tree for event:%d\n",event);
fe4da5cc 655
656 // Get Hits Tree header from file
657 sprintf(treeName,"TreeH%d",event);
658 fTreeH = (TTree*)gDirectory->Get(treeName);
659 if (!fTreeH) {
23370b7a 660 Error("GetEvent","cannot find Hits Tree for event:%d\n",event);
fe4da5cc 661 }
662
663 // Get Digits Tree header from file
664 sprintf(treeName,"TreeD%d",event);
665 fTreeD = (TTree*)gDirectory->Get(treeName);
666 if (!fTreeD) {
07a68c1d 667 Warning("GetEvent","cannot find Digits Tree for event:%d\n",event);
fe4da5cc 668 }
669
670
671 // Get Reconstruct Tree header from file
672 sprintf(treeName,"TreeR%d",event);
673 fTreeR = (TTree*)gDirectory->Get(treeName);
674 if (!fTreeR) {
675 // printf("WARNING: cannot find Reconstructed Tree for event:%d\n",event);
676 }
677
678 // Set Trees branch addresses
8494b010 679 TIter next(fModules);
680 AliModule *detector;
681 while((detector = (AliModule*)next())) {
fe4da5cc 682 detector->SetTreeAddress();
683 }
684
685 if (fTreeK) fTreeK->GetEvent(0);
686 fNtrack = Int_t (fParticles->GetEntries());
687 return fNtrack;
688}
689
690//_____________________________________________________________________________
691TGeometry *AliRun::GetGeometry()
692{
693 //
694 // Import Alice geometry from current file
695 // Return pointer to geometry object
696 //
697 if (!fGeometry) fGeometry = (TGeometry*)gDirectory->Get("AliceGeom");
698 //
699 // Unlink and relink nodes in detectors
700 // This is bad and there must be a better way...
701 //
fe4da5cc 702
8494b010 703 TIter next(fModules);
704 AliModule *detector;
705 while((detector = (AliModule*)next())) {
fe4da5cc 706 detector->SetTreeAddress();
707 TList *dnodes=detector->Nodes();
708 Int_t j;
709 TNode *node, *node1;
710 for ( j=0; j<dnodes->GetSize(); j++) {
711 node = (TNode*) dnodes->At(j);
52d0ab00 712 node1 = fGeometry->GetNode(node->GetName());
fe4da5cc 713 dnodes->Remove(node);
714 dnodes->AddAt(node1,j);
715 }
716 }
717 return fGeometry;
718}
719
720//_____________________________________________________________________________
721void AliRun::GetNextTrack(Int_t &mtrack, Int_t &ipart, Float_t *pmom,
722 Float_t &e, Float_t *vpos, Float_t *polar,
723 Float_t &tof)
724{
725 //
726 // Return next track from stack of particles
727 //
a8f1fb7c 728 TVector3 pol;
fe4da5cc 729 fCurrent=-1;
1578254f 730 TParticle *track;
fe4da5cc 731 for(Int_t i=fNtrack-1; i>=0; i--) {
1578254f 732 track=(TParticle*) fParticles->UncheckedAt(i);
fe4da5cc 733 if(!track->TestBit(Done_Bit)) {
734 //
735 // The track has not yet been processed
736 fCurrent=i;
1578254f 737 ipart=track->GetPdgCode();
738 pmom[0]=track->Px();
739 pmom[1]=track->Py();
740 pmom[2]=track->Pz();
741 e =track->Energy();
742 vpos[0]=track->Vx();
743 vpos[1]=track->Vy();
744 vpos[2]=track->Vz();
a8f1fb7c 745 track->GetPolarisation(pol);
746 polar[0]=pol.X();
747 polar[1]=pol.Y();
748 polar[2]=pol.Z();
1578254f 749 tof=track->T();
fe4da5cc 750 track->SetBit(Done_Bit);
751 break;
752 }
753 }
754 mtrack=fCurrent;
755 //
756 // stop and start timer when we start a primary track
757 Int_t nprimaries = fHeader.GetNprimary();
758 if (fCurrent >= nprimaries) return;
759 if (fCurrent < nprimaries-1) {
760 fTimer.Stop();
1578254f 761 track=(TParticle*) fParticles->UncheckedAt(fCurrent+1);
762 // track->SetProcessTime(fTimer.CpuTime());
fe4da5cc 763 }
764 fTimer.Start();
765}
766
767//_____________________________________________________________________________
768Int_t AliRun::GetPrimary(Int_t track)
769{
770 //
771 // return number of primary that has generated track
772 //
773 int current, parent;
1578254f 774 TParticle *part;
fe4da5cc 775 //
776 parent=track;
777 while (1) {
778 current=parent;
1578254f 779 part = (TParticle *)fParticles->UncheckedAt(current);
780 parent=part->GetFirstMother();
fe4da5cc 781 if(parent<0) return current;
782 }
783}
784
785//_____________________________________________________________________________
786void AliRun::Init(const char *setup)
787{
788 //
789 // Initialize the Alice setup
790 //
791
792 gROOT->LoadMacro(setup);
793 gInterpreter->ProcessLine("Config();");
794
cfce8870 795 gMC->DefineParticles(); //Create standard MC particles
fe4da5cc 796
797 TObject *objfirst, *objlast;
798
23370b7a 799 fNdets = fModules->GetLast()+1;
800
fe4da5cc 801 //
802 //=================Create Materials, geometry, histograms, etc
8494b010 803 TIter next(fModules);
804 AliModule *detector;
805 while((detector = (AliModule*)next())) {
fe4da5cc 806 detector->SetTreeAddress();
807 objlast = gDirectory->GetList()->Last();
808
809 // Initialise detector materials, geometry, histograms,etc
810 detector->CreateMaterials();
811 detector->CreateGeometry();
812 detector->BuildGeometry();
813 detector->Init();
814
815 // Add Detector histograms in Detector list of histograms
816 if (objlast) objfirst = gDirectory->GetList()->After(objlast);
817 else objfirst = gDirectory->GetList()->First();
818 while (objfirst) {
819 detector->Histograms()->Add(objfirst);
820 objfirst = gDirectory->GetList()->After(objfirst);
821 }
822 }
823 SetTransPar(); //Read the cuts for all materials
824
825 MediaTable(); //Build the special IMEDIA table
826
827 //Close the geometry structure
cfce8870 828 gMC->Ggclos();
fe4da5cc 829
830 //Initialise geometry deposition table
cfce8870 831 sEventEnergy.Set(gMC->NofVolumes()+1);
832 sSummEnergy.Set(gMC->NofVolumes()+1);
833 sSum2Energy.Set(gMC->NofVolumes()+1);
fe4da5cc 834
835 //Create the color table
cfce8870 836 gMC->SetColors();
fe4da5cc 837
838 //Compute cross-sections
cfce8870 839 gMC->Gphysi();
fe4da5cc 840
841 //Write Geometry object to current file.
842 fGeometry->Write();
843
844 fInitDone = kTRUE;
845}
846
847//_____________________________________________________________________________
848void AliRun::MediaTable()
849{
850 //
851 // Built media table to get from the media number to
852 // the detector id
853 //
ad51aeb0 854 Int_t kz, nz, idt, lz, i, k, ind;
855 // Int_t ibeg;
fe4da5cc 856 TObjArray &dets = *gAlice->Detectors();
8494b010 857 AliModule *det;
fe4da5cc 858 //
859 // For all detectors
860 for (kz=0;kz<fNdets;kz++) {
861 // If detector is defined
8494b010 862 if((det=(AliModule*) dets[kz])) {
ad51aeb0 863 TArrayI &idtmed = *(det->GetIdtmed());
864 for(nz=0;nz<100;nz++) {
fe4da5cc 865 // Find max and min material number
ad51aeb0 866 if((idt=idtmed[nz])) {
fe4da5cc 867 det->LoMedium() = det->LoMedium() < idt ? det->LoMedium() : idt;
868 det->HiMedium() = det->HiMedium() > idt ? det->HiMedium() : idt;
869 }
870 }
871 if(det->LoMedium() > det->HiMedium()) {
872 det->LoMedium() = 0;
873 det->HiMedium() = 0;
874 } else {
875 if(det->HiMedium() > fImedia->GetSize()) {
ad51aeb0 876 Error("MediaTable","Increase fImedia from %d to %d",
877 fImedia->GetSize(),det->HiMedium());
fe4da5cc 878 return;
879 }
880 // Tag all materials in rage as belonging to detector kz
881 for(lz=det->LoMedium(); lz<= det->HiMedium(); lz++) {
882 (*fImedia)[lz]=kz;
883 }
884 }
885 }
886 }
887 //
888 // Print summary table
889 printf(" Traking media ranges:\n");
890 for(i=0;i<(fNdets-1)/6+1;i++) {
891 for(k=0;k< (6<fNdets-i*6?6:fNdets-i*6);k++) {
892 ind=i*6+k;
8494b010 893 det=(AliModule*)dets[ind];
fe4da5cc 894 if(det)
895 printf(" %6s: %3d -> %3d;",det->GetName(),det->LoMedium(),
896 det->HiMedium());
897 else
898 printf(" %6s: %3d -> %3d;","NULL",0,0);
899 }
900 printf("\n");
901 }
902}
903
904//____________________________________________________________________________
905void AliRun::SetGenerator(AliGenerator *generator)
906{
907 //
908 // Load the event generator
909 //
910 if(!fGenerator) fGenerator = generator;
911}
912
913//____________________________________________________________________________
914void AliRun::SetTransPar(char* filename)
915{
916 //
917 // Read filename to set the transport parameters
918 //
919
fe4da5cc 920
921 const Int_t ncuts=10;
922 const Int_t nflags=11;
923 const Int_t npars=ncuts+nflags;
924 const char pars[npars][7] = {"CUTGAM" ,"CUTELE","CUTNEU","CUTHAD","CUTMUO",
925 "BCUTE","BCUTM","DCUTE","DCUTM","PPCUTM","ANNI",
926 "BREM","COMP","DCAY","DRAY","HADR","LOSS",
927 "MULS","PAIR","PHOT","RAYL"};
928 char line[256];
ad51aeb0 929 char detName[7];
fe4da5cc 930 char* filtmp;
931 Float_t cut[ncuts];
932 Int_t flag[nflags];
933 Int_t i, itmed, iret, ktmed, kz;
934 FILE *lun;
935 //
936 // See whether the file is there
937 filtmp=gSystem->ExpandPathName(filename);
938 lun=fopen(filtmp,"r");
939 delete [] filtmp;
940 if(!lun) {
ad51aeb0 941 Warning("SetTransPar","File %s does not exist!\n",filename);
fe4da5cc 942 return;
943 }
944 //
945 printf(" "); for(i=0;i<60;i++) printf("*"); printf("\n");
946 printf(" *%59s\n","*");
947 printf(" * Please check carefully what you are doing!%10s\n","*");
948 printf(" *%59s\n","*");
949 //
950 while(1) {
951 // Initialise cuts and flags
952 for(i=0;i<ncuts;i++) cut[i]=-99;
953 for(i=0;i<nflags;i++) flag[i]=-99;
954 itmed=0;
955 for(i=0;i<256;i++) line[i]='\0';
956 // Read up to the end of line excluded
957 iret=fscanf(lun,"%[^\n]",line);
958 if(iret<0) {
959 //End of file
960 fclose(lun);
961 printf(" *%59s\n","*");
962 printf(" "); for(i=0;i<60;i++) printf("*"); printf("\n");
963 return;
964 }
965 // Read the end of line
966 fscanf(lun,"%*c");
967 if(!iret) continue;
968 if(line[0]=='*') continue;
969 // Read the numbers
ad51aeb0 970 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",
971 detName,&itmed,&cut[0],&cut[1],&cut[2],&cut[3],&cut[4],&cut[5],&cut[6],&cut[7],&cut[8],
972 &cut[9],&flag[0],&flag[1],&flag[2],&flag[3],&flag[4],&flag[5],&flag[6],&flag[7],
973 &flag[8],&flag[9],&flag[10]);
fe4da5cc 974 if(!iret) continue;
975 if(iret<0) {
976 //reading error
ad51aeb0 977 Warning("SetTransPar","Error reading file %s\n",filename);
fe4da5cc 978 continue;
979 }
ad51aeb0 980 // Check that the module exist
981 AliModule *mod = GetModule(detName);
982 if(mod) {
983 // Get the array of media numbers
984 TArrayI &idtmed = *mod->GetIdtmed();
985 // Check that the tracking medium code is valid
986 if(0<=itmed && itmed < 100) {
987 ktmed=idtmed[itmed];
988 if(!ktmed) {
989 Warning("SetTransPar","Invalid tracking medium code %d for %s\n",itmed,mod->GetName());
990 continue;
fe4da5cc 991 }
ad51aeb0 992 // Set energy thresholds
993 for(kz=0;kz<ncuts;kz++) {
994 if(cut[kz]>=0) {
23370b7a 995 printf(" * %-6s set to %10.3E for tracking medium code %4d for %s\n",
ad51aeb0 996 pars[kz],cut[kz],itmed,mod->GetName());
cfce8870 997 gMC->Gstpar(ktmed,pars[kz],cut[kz]);
ad51aeb0 998 }
fe4da5cc 999 }
ad51aeb0 1000 // Set transport mechanisms
1001 for(kz=0;kz<nflags;kz++) {
1002 if(flag[kz]>=0) {
1003 printf(" * %-6s set to %10d for tracking medium code %4d for %s\n",
1004 pars[ncuts+kz],flag[kz],itmed,mod->GetName());
cfce8870 1005 gMC->Gstpar(ktmed,pars[ncuts+kz],Float_t(flag[kz]));
ad51aeb0 1006 }
1007 }
1008 } else {
1009 Warning("SetTransPar","Invalid medium code %d *\n",itmed);
1010 continue;
fe4da5cc 1011 }
1012 } else {
ad51aeb0 1013 Warning("SetTransPar","Module %s not present\n",detName);
fe4da5cc 1014 continue;
1015 }
1016 }
1017}
1018
1019//_____________________________________________________________________________
1020void AliRun::MakeTree(Option_t *option)
1021{
1022 //
1023 // Create the ROOT trees
1024 // Loop on all detectors to create the Root branch (if any)
1025 //
1026
1027 //
1028 // Analyse options
1029 char *K = strstr(option,"K");
1030 char *H = strstr(option,"H");
1031 char *E = strstr(option,"E");
1032 char *D = strstr(option,"D");
1033 char *R = strstr(option,"R");
1034 //
59fe9bd2 1035 if (K && !fTreeK) fTreeK = new TTree("TreeK0","Kinematics");
1036 if (H && !fTreeH) fTreeH = new TTree("TreeH0","Hits");
1037 if (D && !fTreeD) fTreeD = new TTree("TreeD0","Digits");
fe4da5cc 1038 if (E && !fTreeE) fTreeE = new TTree("TE","Header");
59fe9bd2 1039 if (R && !fTreeR) fTreeR = new TTree("TreeR0","Reconstruction");
fe4da5cc 1040 if (fTreeH) fTreeH->SetAutoSave(1000000000); //no autosave
1041 //
1042 // Create a branch for hits/digits for each detector
1043 // Each branch is a TClonesArray. Each data member of the Hits classes
1044 // will be in turn a subbranch of the detector master branch
8494b010 1045 TIter next(fModules);
1046 AliModule *detector;
1047 while((detector = (AliModule*)next())) {
fe4da5cc 1048 if (H || D || R) detector->MakeBranch(option);
1049 }
1050 // Create a branch for particles
1051 if (fTreeK && K) fTreeK->Branch("Particles",&fParticles,4000);
1052
1053 // Create a branch for Header
1054 if (fTreeE && E) fTreeE->Branch("Header","AliHeader",&header,4000);
1055}
1056
1057//_____________________________________________________________________________
1058Int_t AliRun::PurifyKine(Int_t lastSavedTrack, Int_t nofTracks)
1059{
1060 //
1061 // PurifyKine with external parameters
1062 //
1063 fHgwmk = lastSavedTrack;
1064 fNtrack = nofTracks;
1065 PurifyKine();
1066 return fHgwmk;
1067}
1068
1069//_____________________________________________________________________________
1070void AliRun::PurifyKine()
1071{
1072 //
1073 // Compress kinematic tree keeping only flagged particles
1074 // and renaming the particle id's in all the hits
1075 //
1076 TClonesArray &particles = *fParticles;
1077 int nkeep=fHgwmk+1, parent, i;
1578254f 1078 TParticle *part, *partnew, *father;
fe4da5cc 1079 AliHit *OneHit;
1080 int *map = new int[particles.GetEntries()];
1081
1082 // Save in Header total number of tracks before compression
1083 fHeader.SetNtrack(fHeader.GetNtrack()+fNtrack-fHgwmk);
1084
1085 // Preset map, to be removed later
1086 for(i=0; i<fNtrack; i++) {
1087 if(i<=fHgwmk) map[i]=i ; else map[i] = -99 ;}
1088 // Second pass, build map between old and new numbering
1089 for(i=fHgwmk+1; i<fNtrack; i++) {
1578254f 1090 part = (TParticle *)particles.UncheckedAt(i);
fe4da5cc 1091 if(part->TestBit(Keep_Bit)) {
1092
1093 // This particle has to be kept
1094 map[i]=nkeep;
1095 if(i!=nkeep) {
1096
1097 // Old and new are different, have to copy
1578254f 1098 partnew = (TParticle *)particles.UncheckedAt(nkeep);
fe4da5cc 1099 *partnew = *part;
1100 } else partnew = part;
1101
1102 // as the parent is always *before*, it must be already
1103 // in place. This is what we are checking anyway!
1578254f 1104 if((parent=partnew->GetFirstMother())>fHgwmk) {
fe4da5cc 1105 if(map[parent]==-99) printf("map[%d] = -99!\n",parent);
1578254f 1106 partnew->SetFirstMother(map[parent]);
fe4da5cc 1107 }
1108 nkeep++;
1109 }
1110 }
1111 fNtrack=nkeep;
1112
1578254f 1113 // Fix daughters information
fe4da5cc 1114 for (i=fHgwmk+1; i<fNtrack; i++) {
1578254f 1115 part = (TParticle *)particles.UncheckedAt(i);
1116 parent = part->GetFirstMother();
1117 father = (TParticle *)particles.UncheckedAt(parent);
1118 if(father->TestBit(Daughters_Bit)) {
fe4da5cc 1119
1578254f 1120 if(i<father->GetFirstDaughter()) father->SetFirstDaughter(i);
1121 if(i>father->GetLastDaughter()) father->SetLastDaughter(i);
fe4da5cc 1122 } else {
1578254f 1123 // Iitialise daughters info for first pass
1124 father->SetFirstDaughter(i);
1125 father->SetLastDaughter(i);
1126 father->SetBit(Daughters_Bit);
fe4da5cc 1127 }
1128 }
1129
1130 // Now loop on all detectors and reset the hits
8494b010 1131 TIter next(fModules);
1132 AliModule *detector;
1133 while((detector = (AliModule*)next())) {
fe4da5cc 1134 if (!detector->Hits()) continue;
1135 TClonesArray &vHits=*(detector->Hits());
1136 if(vHits.GetEntries() != detector->GetNhits())
1137 printf("vHits.GetEntries()!=detector->GetNhits(): %d != %d\n",
1138 vHits.GetEntries(),detector->GetNhits());
1139 for (i=0; i<detector->GetNhits(); i++) {
1140 OneHit = (AliHit *)vHits.UncheckedAt(i);
1141 OneHit->SetTrack(map[OneHit->GetTrack()]);
1142 }
1143 }
1144
1145 fHgwmk=nkeep-1;
1146 particles.SetLast(fHgwmk);
1147 delete [] map;
1148}
1149
1150//_____________________________________________________________________________
1151void AliRun::Reset(Int_t run, Int_t idevent)
1152{
1153 //
1154 // Reset all Detectors & kinematics & trees
1155 //
59fe9bd2 1156 char hname[30];
1157 //
fe4da5cc 1158 ResetStack();
1159 ResetHits();
1160 ResetDigits();
1161
1162 // Initialise event header
1163 fHeader.Reset(run,idevent);
1164
59fe9bd2 1165 if(fTreeK) {
1166 fTreeK->Reset();
1167 sprintf(hname,"TreeK%d",idevent);
1168 fTreeK->SetName(hname);
1169 }
1170 if(fTreeH) {
1171 fTreeH->Reset();
1172 sprintf(hname,"TreeH%d",idevent);
1173 fTreeH->SetName(hname);
1174 }
1175 if(fTreeD) {
1176 fTreeD->Reset();
1177 sprintf(hname,"TreeD%d",idevent);
1178 fTreeD->SetName(hname);
1179 }
1180 if(fTreeR) {
1181 fTreeR->Reset();
1182 sprintf(hname,"TreeR%d",idevent);
1183 fTreeR->SetName(hname);
1184 }
fe4da5cc 1185}
1186
1187//_____________________________________________________________________________
1188void AliRun::ResetDigits()
1189{
1190 //
1191 // Reset all Detectors digits
1192 //
8494b010 1193 TIter next(fModules);
1194 AliModule *detector;
1195 while((detector = (AliModule*)next())) {
fe4da5cc 1196 detector->ResetDigits();
1197 }
1198}
1199
1200//_____________________________________________________________________________
1201void AliRun::ResetHits()
1202{
1203 //
1204 // Reset all Detectors hits
1205 //
8494b010 1206 TIter next(fModules);
1207 AliModule *detector;
1208 while((detector = (AliModule*)next())) {
fe4da5cc 1209 detector->ResetHits();
1210 }
1211}
1212
1213//_____________________________________________________________________________
1214void AliRun::ResetPoints()
1215{
1216 //
1217 // Reset all Detectors points
1218 //
8494b010 1219 TIter next(fModules);
1220 AliModule *detector;
1221 while((detector = (AliModule*)next())) {
fe4da5cc 1222 detector->ResetPoints();
1223 }
1224}
1225
1226//_____________________________________________________________________________
1227void AliRun::Run(Int_t nevent, const char *setup)
1228{
1229 //
1230 // Main function to be called to process a galice run
1231 // example
1232 // Root > gAlice.Run();
1233 // a positive number of events will cause the finish routine
1234 // to be called
1235 //
1236
1237 Int_t i, todo;
1238 // check if initialisation has been done
1239 if (!fInitDone) Init(setup);
1240
fe4da5cc 1241 // Create the Root Tree with one branch per detector
1242 if(!fEvent) {
1243 gAlice->MakeTree("KHDER");
1244 }
1245
1246 todo = TMath::Abs(nevent);
1247 for (i=0; i<todo; i++) {
1248 // Process one run (one run = one event)
1249 gAlice->Reset(fRun, fEvent);
cfce8870 1250 gMC->Gtrigi();
1251 gMC->Gtrigc();
1252 gMC->Gtrig();
fe4da5cc 1253 gAlice->FinishEvent();
1254 fEvent++;
1255 }
1256
1257 // End of this run, close files
1258 if(nevent>0) gAlice->FinishRun();
1259}
1260
1261//_____________________________________________________________________________
1262void AliRun::RunLego(const char *setup,Int_t ntheta,Float_t themin,
1263 Float_t themax,Int_t nphi,Float_t phimin,Float_t phimax,
1264 Float_t rmin,Float_t rmax,Float_t zmax)
1265{
1266 //
1267 // Generates lego plots of:
1268 // - radiation length map phi vs theta
1269 // - radiation length map phi vs eta
1270 // - interaction length map
1271 // - g/cm2 length map
1272 //
1273 // ntheta bins in theta, eta
1274 // themin minimum angle in theta (degrees)
1275 // themax maximum angle in theta (degrees)
1276 // nphi bins in phi
1277 // phimin minimum angle in phi (degrees)
1278 // phimax maximum angle in phi (degrees)
1279 // rmin minimum radius
1280 // rmax maximum radius
1281 //
1282 //
1283 // The number of events generated = ntheta*nphi
1284 // run input parameters in macro setup (default="Config.C")
1285 //
1286 // Use macro "lego.C" to visualize the 3 lego plots in spherical coordinates
1287 //Begin_Html
1288 /*
1439f98e 1289 <img src="picts/AliRunLego1.gif">
fe4da5cc 1290 */
1291 //End_Html
1292 //Begin_Html
1293 /*
1439f98e 1294 <img src="picts/AliRunLego2.gif">
fe4da5cc 1295 */
1296 //End_Html
1297 //Begin_Html
1298 /*
1439f98e 1299 <img src="picts/AliRunLego3.gif">
fe4da5cc 1300 */
1301 //End_Html
1302 //
1303
1304 // check if initialisation has been done
1305 if (!fInitDone) Init(setup);
1306
1307 fLego = new AliLego("lego","lego");
1308 fLego->Init(ntheta,themin,themax,nphi,phimin,phimax,rmin,rmax,zmax);
1309 fLego->Run();
1310
1311 // Create only the Root event Tree
1312 gAlice->MakeTree("E");
1313
1314 // End of this run, close files
1315 gAlice->FinishRun();
1316}
1317
1318//_____________________________________________________________________________
1319void AliRun::SetCurrentTrack(Int_t track)
1320{
1321 //
1322 // Set current track number
1323 //
1324 fCurrent = track;
1325}
1326
1327//_____________________________________________________________________________
1578254f 1328void AliRun::SetTrack(Int_t done, Int_t parent, Int_t pdg, Float_t *pmom,
fe4da5cc 1329 Float_t *vpos, Float_t *polar, Float_t tof,
1330 const char *mecha, Int_t &ntr, Float_t weight)
1331{
1332 //
1333 // Load a track on the stack
1334 //
1335 // done 0 if the track has to be transported
1336 // 1 if not
1337 // parent identifier of the parent track. -1 for a primary
1578254f 1338 // pdg particle code
fe4da5cc 1339 // pmom momentum GeV/c
1340 // vpos position
1341 // polar polarisation
1342 // tof time of flight in seconds
1343 // mecha production mechanism
1344 // ntr on output the number of the track stored
1345 //
1346 TClonesArray &particles = *fParticles;
1578254f 1347 TParticle *particle;
fe4da5cc 1348 Float_t mass;
1578254f 1349 const Int_t firstdaughter=-1;
1350 const Int_t lastdaughter=-1;
fe4da5cc 1351 const Int_t KS=0;
1578254f 1352 // const Float_t tlife=0;
fe4da5cc 1353
1578254f 1354 //
1355 // Here we get the static mass
1356 // For MC is ok, but a more sophisticated method could be necessary
1357 // if the calculated mass is required
1358 // also, this method is potentially dangerous if the mass
1359 // used in the MC is not the same of the PDG database
1360 //
1361 mass = TDatabasePDG::Instance()->GetParticle(pdg)->Mass();
fe4da5cc 1362 Float_t e=TMath::Sqrt(mass*mass+pmom[0]*pmom[0]+
1363 pmom[1]*pmom[1]+pmom[2]*pmom[2]);
1364
1365 //printf("Loading particle %s mass %f ene %f No %d ip %d pos %f %f %f mom %f %f %f KS %d m %s\n",
1578254f 1366 //pname,mass,e,fNtrack,pdg,vpos[0],vpos[1],vpos[2],pmom[0],pmom[1],pmom[2],KS,mecha);
fe4da5cc 1367
1578254f 1368 particle=new(particles[fNtrack]) TParticle(pdg,KS,parent,-1,firstdaughter,
1369 lastdaughter,pmom[0],pmom[1],pmom[2],
1370 e,vpos[0],vpos[1],vpos[2],tof);
1371 // polar[0],polar[1],polar[2],tof,
1372 // mecha,weight);
1373 ((TParticle*)particles[fNtrack])->SetPolarisation(TVector3(polar[0],polar[1],polar[2]));
1374 ((TParticle*)particles[fNtrack])->SetWeight(weight);
fe4da5cc 1375 if(!done) particle->SetBit(Done_Bit);
1376
1377 if(parent>=0) {
1578254f 1378 particle=(TParticle*) fParticles->UncheckedAt(parent);
1379 particle->SetLastDaughter(fNtrack);
1380 if(particle->GetFirstDaughter()<0) particle->SetFirstDaughter(fNtrack);
fe4da5cc 1381 } else {
1382 //
1383 // This is a primary track. Set high water mark for this event
1384 fHgwmk=fNtrack;
1385 //
1386 // Set also number if primary tracks
1387 fHeader.SetNprimary(fHgwmk+1);
1388 fHeader.SetNtrack(fHgwmk+1);
1389 }
1390 ntr = fNtrack++;
1391}
1392
1393//_____________________________________________________________________________
1394void AliRun::KeepTrack(const Int_t track)
1395{
1396 //
1397 // flags a track to be kept
1398 //
1399 TClonesArray &particles = *fParticles;
1578254f 1400 ((TParticle*)particles[track])->SetBit(Keep_Bit);
fe4da5cc 1401}
1402
1403//_____________________________________________________________________________
1404void AliRun::StepManager(Int_t id) const
1405{
1406 //
1407 // Called at every step during transport
1408 //
1409
fe4da5cc 1410 Int_t copy;
1411 //
1412 // --- If lego option, do it and leave
1413 if (fLego) {
1414 fLego->StepManager();
1415 return;
1416 }
1417 //Update energy deposition tables
0a6d8768 1418 sEventEnergy[gMC->CurrentVolID(copy)]+=gMC->Edep();
fe4da5cc 1419
1420 //Call the appropriate stepping routine;
8494b010 1421 AliModule *det = (AliModule*)fModules->At(id);
fe4da5cc 1422 if(det) det->StepManager();
1423}
1424
1425//_____________________________________________________________________________
6c9704e6 1426void AliRun::ReadEuclid(const char* filnam, const AliModule *det, char* topvol)
fe4da5cc 1427{
1428 //
1429 // read in the geometry of the detector in euclid file format
1430 //
1431 // id_det : the detector identification (2=its,...)
1432 // topvol : return parameter describing the name of the top
1433 // volume of geometry.
1434 //
1435 // author : m. maire
1436 //
1437 // 28.07.98
1438 // several changes have been made by miroslav helbich
1439 // subroutine is rewrited to follow the new established way of memory
1440 // booking for tracking medias and rotation matrices.
1441 // all used tracking media have to be defined first, for this you can use
1442 // subroutine greutmed.
1443 // top volume is searched as only volume not positioned into another
1444 //
1445
fe4da5cc 1446 Int_t i, nvol, iret, itmed, irot, numed, npar, ndiv, iaxe;
1447 Int_t ndvmx, nr, flag;
1448 char key[5], card[77], natmed[21];
1449 char name[5], mother[5], shape[5], konly[5], volst[7000][5];
1450 char *filtmp;
1451 Float_t par[50];
1452 Float_t teta1, phi1, teta2, phi2, teta3, phi3, orig, step;
1453 Float_t xo, yo, zo;
1454 Int_t idrot[5000],istop[7000];
1455 FILE *lun;
fe4da5cc 1456 //
1457 // *** The input filnam name will be with extension '.euc'
1458 filtmp=gSystem->ExpandPathName(filnam);
1459 lun=fopen(filtmp,"r");
1460 delete [] filtmp;
1461 if(!lun) {
07a68c1d 1462 Error("ReadEuclid","Could not open file %s\n",filnam);
fe4da5cc 1463 return;
1464 }
ad51aeb0 1465 //* --- definition of rotation matrix 0 ---
1466 TArrayI &idtmed = *(det->GetIdtmed());
fe4da5cc 1467 idrot[0]=0;
1468 nvol=0;
1469 L10:
1470 for(i=0;i<77;i++) card[i]=0;
1471 iret=fscanf(lun,"%77[^\n]",card);
1472 if(iret<=0) goto L20;
1473 fscanf(lun,"%*c");
1474 //*
1475 strncpy(key,card,4);
1476 key[4]='\0';
1477 if (!strcmp(key,"TMED")) {
1478 sscanf(&card[5],"%d '%[^']'",&itmed,natmed);
1479 //Pad the string with blanks
1480 i=-1;
1481 while(natmed[++i]);
1482 while(i<20) natmed[i++]=' ';
1483 natmed[i]='\0';
1484 //
cfce8870 1485 gMC->Gckmat(idtmed[itmed],natmed);
fe4da5cc 1486 //*
1487 } else if (!strcmp(key,"ROTM")) {
1488 sscanf(&card[4],"%d %f %f %f %f %f %f",&irot,&teta1,&phi1,&teta2,&phi2,&teta3,&phi3);
1489 det->AliMatrix(idrot[irot],teta1,phi1,teta2,phi2,teta3,phi3);
1490 //*
1491 } else if (!strcmp(key,"VOLU")) {
1492 sscanf(&card[5],"'%[^']' '%[^']' %d %d", name, shape, &numed, &npar);
1493 if (npar>0) {
1494 for(i=0;i<npar;i++) fscanf(lun,"%f",&par[i]);
1495 fscanf(lun,"%*c");
1496 }
cfce8870 1497 gMC->Gsvolu( name, shape, idtmed[numed], par, npar);
fe4da5cc 1498 //* save the defined volumes
1499 strcpy(volst[++nvol],name);
1500 istop[nvol]=1;
1501 //*
1502 } else if (!strcmp(key,"DIVN")) {
1503 sscanf(&card[5],"'%[^']' '%[^']' %d %d", name, mother, &ndiv, &iaxe);
cfce8870 1504 gMC->Gsdvn ( name, mother, ndiv, iaxe );
fe4da5cc 1505 //*
1506 } else if (!strcmp(key,"DVN2")) {
1507 sscanf(&card[5],"'%[^']' '%[^']' %d %d %f %d",name, mother, &ndiv, &iaxe, &orig, &numed);
cfce8870 1508 gMC->Gsdvn2( name, mother, ndiv, iaxe, orig,idtmed[numed]);
fe4da5cc 1509 //*
1510 } else if (!strcmp(key,"DIVT")) {
1511 sscanf(&card[5],"'%[^']' '%[^']' %f %d %d %d", name, mother, &step, &iaxe, &numed, &ndvmx);
cfce8870 1512 gMC->Gsdvt ( name, mother, step, iaxe, idtmed[numed], ndvmx);
fe4da5cc 1513 //*
1514 } else if (!strcmp(key,"DVT2")) {
1515 sscanf(&card[5],"'%[^']' '%[^']' %f %d %f %d %d", name, mother, &step, &iaxe, &orig, &numed, &ndvmx);
cfce8870 1516 gMC->Gsdvt2 ( name, mother, step, iaxe, orig, idtmed[numed], ndvmx );
fe4da5cc 1517 //*
1518 } else if (!strcmp(key,"POSI")) {
1519 sscanf(&card[5],"'%[^']' %d '%[^']' %f %f %f %d '%[^']'", name, &nr, mother, &xo, &yo, &zo, &irot, konly);
1520 //*** volume name cannot be the top volume
1521 for(i=1;i<=nvol;i++) {
1522 if (!strcmp(volst[i],name)) istop[i]=0;
1523 }
1524 //*
cfce8870 1525 gMC->Gspos ( name, nr, mother, xo, yo, zo, idrot[irot], konly );
fe4da5cc 1526 //*
1527 } else if (!strcmp(key,"POSP")) {
1528 sscanf(&card[5],"'%[^']' %d '%[^']' %f %f %f %d '%[^']' %d", name, &nr, mother, &xo, &yo, &zo, &irot, konly, &npar);
1529 if (npar > 0) {
1530 for(i=0;i<npar;i++) fscanf(lun,"%f",&par[i]);
1531 fscanf(lun,"%*c");
1532 }
1533 //*** volume name cannot be the top volume
1534 for(i=1;i<=nvol;i++) {
1535 if (!strcmp(volst[i],name)) istop[i]=0;
1536 }
1537 //*
cfce8870 1538 gMC->Gsposp ( name, nr, mother, xo,yo,zo, idrot[irot], konly, par, npar);
fe4da5cc 1539 }
1540 //*
1541 if (strcmp(key,"END")) goto L10;
1542 //* find top volume in the geometry
1543 flag=0;
1544 for(i=1;i<=nvol;i++) {
1545 if (istop[i] && flag) {
07a68c1d 1546 Warning("ReadEuclid"," %s is another possible top volume\n",volst[i]);
fe4da5cc 1547 }
1548 if (istop[i] && !flag) {
dda5e866 1549 strcpy(topvol,volst[i]);
fe4da5cc 1550 printf(" *** GREUCL *** volume %s taken as a top volume\n",topvol);
1551 flag=1;
1552 }
1553 }
1554 if (!flag) {
07a68c1d 1555 Warning("ReadEuclid","top volume not found\n");
fe4da5cc 1556 }
1557 fclose (lun);
1558 //*
1559 //* commented out only for the not cernlib version
1560 printf(" *** GREUCL *** file: %s is now read in\n",filnam);
1561 //
1562 return;
1563 //*
1564 L20:
07a68c1d 1565 Error("ReadEuclid","reading error or premature end of file\n");
fe4da5cc 1566}
1567
1568//_____________________________________________________________________________
23370b7a 1569void AliRun::ReadEuclidMedia(const char* filnam, const AliModule *det)
fe4da5cc 1570{
1571 //
1572 // read in the materials and tracking media for the detector
1573 // in euclid file format
1574 //
1575 // filnam: name of the input file
1576 // id_det: id_det is the detector identification (2=its,...)
1577 //
1578 // author : miroslav helbich
1579 //
1580 Float_t sxmgmx = gAlice->Field()->Max();
1581 Int_t isxfld = gAlice->Field()->Integ();
1582 Int_t end, i, iret, itmed;
1583 char key[5], card[130], natmed[21], namate[21];
1584 Float_t ubuf[50];
1585 char* filtmp;
1586 FILE *lun;
1587 Int_t imate;
1588 Int_t nwbuf, isvol, ifield, nmat;
1589 Float_t a, z, dens, radl, absl, fieldm, tmaxfd, stemax, deemax, epsil, stmin;
23370b7a 1590 //
fe4da5cc 1591 end=strlen(filnam);
1592 for(i=0;i<end;i++) if(filnam[i]=='.') {
1593 end=i;
1594 break;
1595 }
1596 //
1597 // *** The input filnam name will be with extension '.euc'
1598 printf("The file name is %s\n",filnam); //Debug
1599 filtmp=gSystem->ExpandPathName(filnam);
1600 lun=fopen(filtmp,"r");
1601 delete [] filtmp;
1602 if(!lun) {
23370b7a 1603 Warning("ReadEuclidMedia","Could not open file %s\n",filnam);
fe4da5cc 1604 return;
1605 }
1606 //
1607 // Retrieve Mag Field parameters
1608 Int_t ISXFLD=gAlice->Field()->Integ();
1609 Float_t SXMGMX=gAlice->Field()->Max();
23370b7a 1610 // TArrayI &idtmed = *(det->GetIdtmed());
fe4da5cc 1611 //
1612 L10:
1613 for(i=0;i<130;i++) card[i]=0;
1614 iret=fscanf(lun,"%4s %[^\n]",key,card);
1615 if(iret<=0) goto L20;
1616 fscanf(lun,"%*c");
1617 //*
1618 //* read material
1619 if (!strcmp(key,"MATE")) {
1620 sscanf(card,"%d '%[^']' %f %f %f %f %f %d",&imate,namate,&a,&z,&dens,&radl,&absl,&nwbuf);
1621 if (nwbuf>0) for(i=0;i<nwbuf;i++) fscanf(lun,"%f",&ubuf[i]);
1622 //Pad the string with blanks
1623 i=-1;
1624 while(namate[++i]);
1625 while(i<20) namate[i++]=' ';
1626 namate[i]='\0';
1627 //
1628 det->AliMaterial(imate,namate,a,z,dens,radl,absl,ubuf,nwbuf);
1629 //* read tracking medium
1630 } else if (!strcmp(key,"TMED")) {
1631 sscanf(card,"%d '%[^']' %d %d %d %f %f %f %f %f %f %d",
1632 &itmed,natmed,&nmat,&isvol,&ifield,&fieldm,&tmaxfd,
1633 &stemax,&deemax,&epsil,&stmin,&nwbuf);
1634 if (nwbuf>0) for(i=0;i<nwbuf;i++) fscanf(lun,"%f",&ubuf[i]);
1635 if (ifield<0) ifield=isxfld;
1636 if (fieldm<0) fieldm=sxmgmx;
1637 //Pad the string with blanks
1638 i=-1;
1639 while(natmed[++i]);
1640 while(i<20) natmed[i++]=' ';
1641 natmed[i]='\0';
1642 //
ad51aeb0 1643 det->AliMedium(itmed,natmed,nmat,isvol,ISXFLD,SXMGMX,tmaxfd,
fe4da5cc 1644 stemax,deemax,epsil,stmin,ubuf,nwbuf);
23370b7a 1645 // (*fImedia)[idtmed[itmed]-1]=id_det;
fe4da5cc 1646 //*
1647 }
1648 //*
1649 if (strcmp(key,"END")) goto L10;
1650 fclose (lun);
1651 //*
1652 //* commented out only for the not cernlib version
23370b7a 1653 Warning("ReadEuclidMedia","file: %s is now read in\n",filnam);
fe4da5cc 1654 //*
1655 return;
1656 //*
1657 L20:
23370b7a 1658 Warning("ReadEuclidMedia","reading error or premature end of file\n");
fe4da5cc 1659}
1660
1661//_____________________________________________________________________________
1662void AliRun::Streamer(TBuffer &R__b)
1663{
1664 //
1665 // Stream an object of class AliRun.
1666 //
1667 if (R__b.IsReading()) {
1668 Version_t R__v = R__b.ReadVersion(); if (R__v) { }
1669 TNamed::Streamer(R__b);
1670 if (!gAlice) gAlice = this;
1671 gROOT->GetListOfBrowsables()->Add(this,"Run");
59fe9bd2 1672 fTreeE = (TTree*)gDirectory->Get("TE");
1673 if (fTreeE) fTreeE->SetBranchAddress("Header", &header);
1674 else Error("Streamer","cannot find Header Tree\n");
fe4da5cc 1675 R__b >> fNtrack;
1676 R__b >> fHgwmk;
1677 R__b >> fDebug;
1678 fHeader.Streamer(R__b);
8494b010 1679 R__b >> fModules;
fe4da5cc 1680 R__b >> fParticles;
1681 R__b >> fField;
1682 // R__b >> fMC;
1683 R__b >> fNdets;
1684 R__b >> fTrRmax;
1685 R__b >> fTrZmax;
1686 R__b >> fGenerator;
59fe9bd2 1687 if(R__v>1) {
1688 R__b >> fPDGDB; //Particle factory object!
1689 fTreeE->GetEntry(0);
1690 } else {
1691 fHeader.SetEvent(0);
1692 fPDGDB = TDatabasePDG::Instance(); //Particle factory object!
1693 }
fe4da5cc 1694 } else {
1695 R__b.WriteVersion(AliRun::IsA());
1696 TNamed::Streamer(R__b);
1697 R__b << fNtrack;
1698 R__b << fHgwmk;
1699 R__b << fDebug;
1700 fHeader.Streamer(R__b);
8494b010 1701 R__b << fModules;
fe4da5cc 1702 R__b << fParticles;
1703 R__b << fField;
1704 // R__b << fMC;
1705 R__b << fNdets;
1706 R__b << fTrRmax;
1707 R__b << fTrZmax;
1708 R__b << fGenerator;
1578254f 1709 R__b << fPDGDB; //Particle factory object!
fe4da5cc 1710 }
1711}
1712
1713
1714//_____________________________________________________________________________
1715//
1716// Interfaces to Fortran
1717//
1718//_____________________________________________________________________________
1719
1720extern "C" void type_of_call rxgtrak (Int_t &mtrack, Int_t &ipart, Float_t *pmom,
ad265d61 1721 Float_t &e, Float_t *vpos, Float_t *polar,
1722 Float_t &tof)
fe4da5cc 1723{
1724 //
1725 // Fetches next track from the ROOT stack for transport. Called by the
1726 // modified version of GTREVE.
1727 //
1728 // Track number in the ROOT stack. If MTRACK=0 no
1729 // mtrack more tracks are left in the stack to be
1730 // transported.
1731 // ipart Particle code in the GEANT conventions.
1732 // pmom[3] Particle momentum in GeV/c
1733 // e Particle energy in GeV
1734 // vpos[3] Particle position
1735 // tof Particle time of flight in seconds
1736 //
1578254f 1737 Int_t pdg;
1738 gAlice->GetNextTrack(mtrack, pdg, pmom, e, vpos, polar, tof);
1739 ipart = gMC->IdFromPDG(pdg);
fe4da5cc 1740 mtrack++;
1741}
1742
1743//_____________________________________________________________________________
1744extern "C" void type_of_call
1745#ifndef WIN32
1746rxstrak (Int_t &keep, Int_t &parent, Int_t &ipart, Float_t *pmom,
1747 Float_t *vpos, Float_t &tof, const char* cmech, Int_t &ntr, const int cmlen)
1748#else
1749rxstrak (Int_t &keep, Int_t &parent, Int_t &ipart, Float_t *pmom,
1750 Float_t *vpos, Float_t &tof, const char* cmech, const int cmlen,
1751 Int_t &ntr)
1752#endif
1753{
1754 //
1755 // Fetches next track from the ROOT stack for transport. Called by GUKINE
1756 // and GUSTEP.
1757 //
1758 // Status of the track. If keep=0 the track is put
1759 // keep on the ROOT stack but it is not fetched for
1760 // transport.
1761 // parent Parent track. If parent=0 the track is a primary.
1762 // In GUSTEP the routine is normally called to store
1763 // secondaries generated by the current track whose
1764 // ROOT stack number is MTRACK (common SCKINE.
1765 // ipart Particle code in the GEANT conventions.
1766 // pmom[3] Particle momentum in GeV/c
1767 // vpos[3] Particle position
1768 // tof Particle time of flight in seconds
1769 //
1770 // cmech (CHARACTER*10) Particle origin. This field is user
1771 // defined and it is not used inside the GALICE code.
1772 // ntr Number assigned to the particle in the ROOT stack.
1773 //
1774 char mecha[11];
1775 Float_t polar[3]={0.,0.,0.};
1776 for(int i=0; i<10 && i<cmlen; i++) mecha[i]=cmech[i];
1777 mecha[10]=0;
1578254f 1778 Int_t pdg=gMC->PDGFromId(ipart);
1779 gAlice->SetTrack(keep, parent-1, pdg, pmom, vpos, polar, tof, mecha, ntr);
fe4da5cc 1780 ntr++;
1781}
1782
1783//_____________________________________________________________________________
1784extern "C" void type_of_call rxkeep(const Int_t &n)
1785{
1786 if( NULL==gAlice ) exit(1);
1787
1788 if( n<=0 || n>gAlice->Particles()->GetEntries() )
1789 {
1790 printf(" Bad index n=%d must be 0<n<=%d\n",
1791 n,gAlice->Particles()->GetEntries());
1792 exit(1);
1793 }
1794
1578254f 1795 ((TParticle*)(gAlice->Particles()->UncheckedAt(n-1)))->SetBit(Keep_Bit);
fe4da5cc 1796}
1797
1798//_____________________________________________________________________________
1799extern "C" void type_of_call rxouth ()
1800{
1801 //
1802 // Called by Gtreve at the end of each primary track
1803 //
1804 gAlice->FinishPrimary();
1805}
1806