]> git.uio.no Git - u/mrichter/AliRoot.git/blame - STEER/AliMC.cxx
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
Coding Convention Violations
[u/mrichter/AliRoot.git] / STEER / AliMC.cxx
CommitLineData
5d12ce38 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/* $Id$ */
17
d0d4a6b3 18// This class is extracted from the AliRun class
19// and contains all the MC-related functionality
20// The number of dependencies has to be reduced...
21// Author: F.Carminati
22// Federico.Carminati@cern.ch
23
5d12ce38 24#include <TBrowser.h>
25#include <TStopwatch.h>
26#include <TSystem.h>
27#include <TVirtualMC.h>
28
29#include "AliDetector.h"
30#include "AliGenerator.h"
31#include "AliHeader.h"
32#include "AliLego.h"
33#include "AliMC.h"
34#include "AliMCQA.h"
35#include "AliRun.h"
36#include "AliStack.h"
37#include "AliTrackReference.h"
38
39
40ClassImp(AliMC)
41
42//_______________________________________________________________________
43AliMC::AliMC() :
44 fGenerator(0),
45 fEventEnergy(0),
46 fSummEnergy(0),
47 fSum2Energy(0),
48 fTrRmax(1.e10),
49 fTrZmax(1.e10),
c2eea27a 50 fDebug(0),
5d12ce38 51 fImedia(0),
52 fTransParName("\0"),
53 fMCQA(0),
54 fHitLists(0),
55 fTrackReferences(0)
56
57{
d0d4a6b3 58 //default constructor
5d12ce38 59}
60
61//_______________________________________________________________________
62AliMC::AliMC(const char *name, const char *title) :
63 TVirtualMCApplication(name, title),
64 fGenerator(0),
65 fEventEnergy(0),
66 fSummEnergy(0),
67 fSum2Energy(0),
68 fTrRmax(1.e10),
69 fTrZmax(1.e10),
c2eea27a 70 fDebug(0),
5d12ce38 71 fImedia(new TArrayI(1000)),
72 fTransParName("\0"),
73 fMCQA(0),
74 fHitLists(new TList()),
75 fTrackReferences(new TClonesArray("AliTrackReference", 100))
76{
d0d4a6b3 77 //constructor
5d12ce38 78 // Set transport parameters
79 SetTransPar();
80
81 // Prepare the tracking medium lists
82 for(Int_t i=0;i<1000;i++) (*fImedia)[i]=-99;
83
84}
85
86//_______________________________________________________________________
87AliMC::AliMC(const AliMC &mc) :
88 TVirtualMCApplication(mc),
89 fGenerator(0),
90 fEventEnergy(0),
91 fSummEnergy(0),
92 fSum2Energy(0),
93 fTrRmax(1.e10),
94 fTrZmax(1.e10),
c2eea27a 95 fDebug(0),
5d12ce38 96 fImedia(0),
97 fTransParName("\0"),
98 fMCQA(0),
99 fHitLists(0),
100 fTrackReferences(0)
101{
102 //
103 // Copy constructor for AliRun
104 //
105 mc.Copy(*this);
106}
107
108//_______________________________________________________________________
109AliMC::~AliMC()
110{
d0d4a6b3 111 //destructor
5d12ce38 112 delete fGenerator;
113 delete fImedia;
114 delete fMCQA;
115 delete fHitLists;
116 // Delete track references
117 if (fTrackReferences) {
118 fTrackReferences->Delete();
119 delete fTrackReferences;
120 fTrackReferences = 0;
121 }
122
123}
124
125//_______________________________________________________________________
6c4904c2 126void AliMC::Copy(TObject &) const
5d12ce38 127{
d0d4a6b3 128 //dummy Copy function
5d12ce38 129 Fatal("Copy","Not implemented!\n");
130}
131
132//_______________________________________________________________________
133void AliMC::ConstructGeometry()
134{
135 //
136 // Create modules, materials, geometry
137 //
138
139 TStopwatch stw;
140 TIter next(gAlice->Modules());
141 AliModule *detector;
142 if (GetDebug()) Info("ConstructGeometry","Geometry creation:");
143 while((detector = dynamic_cast<AliModule*>(next()))) {
144 stw.Start();
145 // Initialise detector materials and geometry
146 detector->CreateMaterials();
147 detector->CreateGeometry();
148 printf("%10s R:%.2fs C:%.2fs\n",
149 detector->GetName(),stw.RealTime(),stw.CpuTime());
150 }
151}
152
153//_______________________________________________________________________
154void AliMC::InitGeometry()
155{
156 //
157 // Initialize detectors and display geometry
158 //
159
160 printf("Initialisation:\n");
161 TStopwatch stw;
162 TIter next(gAlice->Modules());
163 AliModule *detector;
164 while((detector = dynamic_cast<AliModule*>(next()))) {
165 stw.Start();
166 // Initialise detector and display geometry
167 detector->Init();
168 detector->BuildGeometry();
169 printf("%10s R:%.2fs C:%.2fs\n",
170 detector->GetName(),stw.RealTime(),stw.CpuTime());
171 }
172
173}
174
175//_______________________________________________________________________
2057aecb 176void AliMC::GeneratePrimaries()
5d12ce38 177{
178 //
179 // Generate primary particles and fill them in the stack.
180 //
181
182 Generator()->Generate();
183}
184
185//_______________________________________________________________________
186void AliMC::SetGenerator(AliGenerator *generator)
187{
188 //
189 // Load the event generator
190 //
191 if(!fGenerator) fGenerator = generator;
192}
193
194//_______________________________________________________________________
195void AliMC::ResetGenerator(AliGenerator *generator)
196{
197 //
198 // Load the event generator
199 //
200 if(fGenerator)
201 if(generator)
202 Warning("ResetGenerator","Replacing generator %s with %s\n",
203 fGenerator->GetName(),generator->GetName());
204 else
205 Warning("ResetGenerator","Replacing generator %s with NULL\n",
206 fGenerator->GetName());
207 fGenerator = generator;
208}
209
210//_______________________________________________________________________
211void AliMC::FinishRun()
212{
213 // Clean generator information
214 if (GetDebug()) Info("FinishRun"," fGenerator->FinishRun()");
215 fGenerator->FinishRun();
216
217 //Output energy summary tables
32ac63db 218 if (GetDebug()) {
219 Info("FinishRun"," EnergySummary()");
220 EnergySummary();
221 }
5d12ce38 222
223}
224
225//_______________________________________________________________________
226void AliMC::BeginPrimary()
227{
228 //
229 // Called at the beginning of each primary track
230 //
231
232 // Reset Hits info
233 ResetHits();
234 ResetTrackReferences();
235
236}
237
238//_______________________________________________________________________
239void AliMC::PreTrack()
240{
d0d4a6b3 241 // Actions before the track's transport
5d12ce38 242 TObjArray &dets = *gAlice->Modules();
243 AliModule *module;
244
245 for(Int_t i=0; i<=gAlice->GetNdets(); i++)
246 if((module = dynamic_cast<AliModule*>(dets[i])))
247 module->PreTrack();
248
249 fMCQA->PreTrack();
250}
251
252//_______________________________________________________________________
253void AliMC::Stepping()
254{
255 //
256 // Called at every step during transport
257 //
258
259 Int_t id = DetFromMate(gMC->GetMedium());
260 if (id < 0) return;
261
262 //
263 // --- If lego option, do it and leave
264 if (gAlice->Lego())
265 gAlice->Lego()->StepManager();
266 else {
267 Int_t copy;
268 //Update energy deposition tables
269 AddEnergyDeposit(gMC->CurrentVolID(copy),gMC->Edep());
ab03c084 270 //
271 // write tracke reference for track which is dissapearing - MI
272 if (gMC->IsTrackDisappeared()) {
273 if (gMC->Etot()>0.05) AddTrackReference(GetCurrentTrackNumber());
274 }
5d12ce38 275
276 //Call the appropriate stepping routine;
277 AliModule *det = dynamic_cast<AliModule*>(gAlice->Modules()->At(id));
278 if(det && det->StepManagerIsEnabled()) {
279 fMCQA->StepManager(id);
280 det->StepManager();
281 }
282 }
283}
284
285//_______________________________________________________________________
286void AliMC::EnergySummary()
287{
288 //
289 // Print summary of deposited energy
290 //
291
292 Int_t ndep=0;
293 Float_t edtot=0;
294 Float_t ed, ed2;
295 Int_t kn, i, left, j, id;
296 const Float_t kzero=0;
297 Int_t ievent=gAlice->GetRunLoader()->GetHeader()->GetEvent()+1;
298 //
299 // Energy loss information
300 if(ievent) {
301 printf("***************** Energy Loss Information per event (GEV) *****************\n");
302 for(kn=1;kn<fEventEnergy.GetSize();kn++) {
303 ed=fSummEnergy[kn];
304 if(ed>0) {
305 fEventEnergy[ndep]=kn;
306 if(ievent>1) {
307 ed=ed/ievent;
308 ed2=fSum2Energy[kn];
309 ed2=ed2/ievent;
310 ed2=100*TMath::Sqrt(TMath::Max(ed2-ed*ed,kzero))/ed;
311 } else
312 ed2=99;
313 fSummEnergy[ndep]=ed;
314 fSum2Energy[ndep]=TMath::Min(static_cast<Float_t>(99.),TMath::Max(ed2,kzero));
315 edtot+=ed;
316 ndep++;
317 }
318 }
319 for(kn=0;kn<(ndep-1)/3+1;kn++) {
320 left=ndep-kn*3;
321 for(i=0;i<(3<left?3:left);i++) {
322 j=kn*3+i;
323 id=Int_t (fEventEnergy[j]+0.1);
324 printf(" %s %10.3f +- %10.3f%%;",gMC->VolName(id),fSummEnergy[j],fSum2Energy[j]);
325 }
326 printf("\n");
327 }
328 //
329 // Relative energy loss in different detectors
330 printf("******************** Relative Energy Loss per event ********************\n");
331 printf("Total energy loss per event %10.3f GeV\n",edtot);
332 for(kn=0;kn<(ndep-1)/5+1;kn++) {
333 left=ndep-kn*5;
334 for(i=0;i<(5<left?5:left);i++) {
335 j=kn*5+i;
336 id=Int_t (fEventEnergy[j]+0.1);
337 printf(" %s %10.3f%%;",gMC->VolName(id),100*fSummEnergy[j]/edtot);
338 }
339 printf("\n");
340 }
341 for(kn=0;kn<75;kn++) printf("*");
342 printf("\n");
343 }
344 //
345 // Reset the TArray's
346 // fEventEnergy.Set(0);
347 // fSummEnergy.Set(0);
348 // fSum2Energy.Set(0);
349}
350
351//_____________________________________________________________________________
352void AliMC::BeginEvent()
353{
354 //
355 // Clean-up previous event
356 // Energy scores
357 if (GetDebug())
358 {
359 Info("BeginEvent",">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>");
360 Info("BeginEvent",">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>");
361 Info("BeginEvent",">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>");
362 Info("BeginEvent"," BEGINNING EVENT ");
363 Info("BeginEvent",">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>");
364 Info("BeginEvent",">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>");
365 Info("BeginEvent",">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>");
366 }
367
368 AliRunLoader *runloader=gAlice->GetRunLoader();
369
370 /*******************************/
371 /* Clean after eventual */
372 /* previous event */
373 /*******************************/
374
375
376 //Set the next event in Run Loader -> Cleans trees (TreeK and all trees in detectors),
377 gAlice->SetEventNrInRun(gAlice->GetEventNrInRun()+1);
378 runloader->SetEventNumber(gAlice->GetEventNrInRun());// sets new files, cleans the previous event stuff, if necessary, etc.,
379 if (GetDebug()) Info("BeginEvent","EventNr is %d",gAlice->GetEventNrInRun());
380
381 fEventEnergy.Reset();
382 // Clean detector information
383
384 if (runloader->Stack())
385 runloader->Stack()->Reset();//clean stack -> tree is unloaded
386 else
387 runloader->MakeStack();//or make a new one
504b172d 388
389 if(gAlice->Lego() == 0x0)
390 {
391 if (GetDebug()) Info("BeginEvent"," fRunLoader->MakeTree(K)");
392 runloader->MakeTree("K");
393 }
394
5d12ce38 395 if (GetDebug()) Info("BeginEvent"," gMC->SetStack(fRunLoader->Stack())");
396 gMC->SetStack(gAlice->GetRunLoader()->Stack());//Was in InitMC - but was moved here
397 //because we don't have guarantee that
398 //stack pointer is not going to change from event to event
399 //since it bellobgs to header and is obtained via RunLoader
400 //
401 // Reset all Detectors & kinematics & make/reset trees
402 //
403
404 runloader->GetHeader()->Reset(gAlice->GetRunNumber(),gAlice->GetEvNumber(),
405 gAlice->GetEventNrInRun());
406// fRunLoader->WriteKinematics("OVERWRITE"); is there any reason to rewrite here since MakeTree does so
407
504b172d 408 if(gAlice->Lego())
409 {
410 gAlice->Lego()->BeginEvent();
411 return;
412 }
413
5d12ce38 414
415 if (GetDebug()) Info("BeginEvent"," ResetHits()");
416 ResetHits();
504b172d 417
5d12ce38 418 if (GetDebug()) Info("BeginEvent"," fRunLoader->MakeTree(H)");
419 runloader->MakeTree("H");
504b172d 420
421 if (GetDebug()) Info("BeginEvent"," fRunLoader->MakeTrackRefsContainer()");
422 runloader->MakeTrackRefsContainer();//for insurance
5d12ce38 423
5d12ce38 424
425 //create new branches and SetAdresses
426 TIter next(gAlice->Modules());
427 AliModule *detector;
428 while((detector = (AliModule*)next()))
429 {
430 if (GetDebug()) Info("BeginEvent"," %s->MakeBranch(H)",detector->GetName());
431 detector->MakeBranch("H");
432 if (GetDebug()) Info("BeginEvent"," %s->MakeBranchTR()",detector->GetName());
433 detector->MakeBranchTR();
434 if (GetDebug()) Info("BeginEvent"," %s->SetTreeAddress()",detector->GetName());
435 detector->SetTreeAddress();
436 }
ab03c084 437 // make branch for AliRun track References
438 TTree * treeTR = runloader->TreeTR();
439 if (treeTR){
440 // make branch for central track references
441 if (!fTrackReferences) fTrackReferences = new TClonesArray("AliTrackReference",0);
442 TBranch *branch;
443 branch = treeTR->Branch("AliRun",&fTrackReferences);
444 branch->SetAddress(&fTrackReferences);
445 }
446 //
5d12ce38 447}
448
449//_______________________________________________________________________
450void AliMC::ResetHits()
451{
452 //
453 // Reset all Detectors hits
454 //
455 TIter next(gAlice->Modules());
456 AliModule *detector;
457 while((detector = dynamic_cast<AliModule*>(next()))) {
458 detector->ResetHits();
459 }
460}
461
462//_______________________________________________________________________
463void AliMC::PostTrack()
464{
d0d4a6b3 465 // Posts tracks for each module
5d12ce38 466 TObjArray &dets = *gAlice->Modules();
467 AliModule *module;
468
469 for(Int_t i=0; i<=gAlice->GetNdets(); i++)
470 if((module = dynamic_cast<AliModule*>(dets[i])))
471 module->PostTrack();
472}
473
474//_______________________________________________________________________
475void AliMC::FinishPrimary()
476{
477 //
478 // Called at the end of each primary track
479 //
480 AliRunLoader *runloader=gAlice->GetRunLoader();
481 // static Int_t count=0;
482 // const Int_t times=10;
483 // This primary is finished, purify stack
484 runloader->Stack()->PurifyKine();
504b172d 485
5d12ce38 486 TIter next(gAlice->Modules());
487 AliModule *detector;
488 while((detector = dynamic_cast<AliModule*>(next()))) {
489 detector->FinishPrimary();
504b172d 490 AliLoader* loader = detector->GetLoader();
491 if(loader)
492 {
493 TTree* treeH = loader->TreeH();
494 if (treeH) treeH->Fill(); //can be Lego run and treeH can not exist
495 }
5d12ce38 496 }
497
498 // Write out track references if any
499 if (runloader->TreeTR()) runloader->TreeTR()->Fill();
500}
501
502//_______________________________________________________________________
503void AliMC::FinishEvent()
504{
505 //
506 // Called at the end of the event.
507 //
508
509 //
510 if(gAlice->Lego()) gAlice->Lego()->FinishEvent();
511
512 TIter next(gAlice->Modules());
513 AliModule *detector;
514 while((detector = dynamic_cast<AliModule*>(next()))) {
515 detector->FinishEvent();
516 }
517
518 //Update the energy deposit tables
519 Int_t i;
520 for(i=0;i<fEventEnergy.GetSize();i++)
521 {
522 fSummEnergy[i]+=fEventEnergy[i];
523 fSum2Energy[i]+=fEventEnergy[i]*fEventEnergy[i];
524 }
525
526 AliRunLoader *runloader=gAlice->GetRunLoader();
527
528 AliHeader* header = runloader->GetHeader();
529 AliStack* stack = runloader->Stack();
530 if ( (header == 0x0) || (stack == 0x0) )
531 {//check if we got header and stack. If not cry and exit aliroot
532 Fatal("AliRun","Can not get the stack or header from LOADER");
533 return;//never reached
534 }
535 // Update Header information
536 header->SetNprimary(stack->GetNprimary());
537 header->SetNtrack(stack->GetNtrack());
538
539
540 // Write out the kinematics
541 stack->FinishEvent();
542
543 // Write out the event Header information
544 TTree* treeE = runloader->TreeE();
545 if (treeE)
546 {
547 header->SetStack(stack);
548 treeE->Fill();
549 }
550 else
551 {
552 Error("FinishEvent","Can not get TreeE from RL");
553 }
554
504b172d 555 if(gAlice->Lego() == 0x0)
556 {
557 runloader->WriteKinematics("OVERWRITE");
558 runloader->WriteTrackRefs("OVERWRITE");
559 runloader->WriteHits("OVERWRITE");
560 }
561
5d12ce38 562 if (GetDebug())
563 {
564 Info("FinishEvent","<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<");
565 Info("FinishEvent","<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<");
566 Info("FinishEvent","<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<");
567 Info("FinishEvent"," FINISHING EVENT ");
568 Info("FinishEvent","<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<");
569 Info("FinishEvent","<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<");
570 Info("FinishEvent","<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<");
571 }
572}
573
574//_______________________________________________________________________
575void AliMC::Field(const Double_t* x, Double_t* b) const
576{
d0d4a6b3 577 // Calculates field "b" at point "x"
5d12ce38 578 gAlice->Field(x,b);
579}
580
581//_______________________________________________________________________
582void AliMC::Init()
583{
d0d4a6b3 584 // MC initialization
5d12ce38 585
586 //=================Create Materials and geometry
587 gMC->Init();
5d12ce38 588 //Read the cuts for all materials
589 ReadTransPar();
590 //Build the special IMEDIA table
591 MediaTable();
592
593 //Compute cross-sections
594 gMC->BuildPhysics();
595
596 //Write Geometry object to current file.
597 gAlice->GetRunLoader()->WriteGeometry();
598
599 //Initialise geometry deposition table
600 fEventEnergy.Set(gMC->NofVolumes()+1);
601 fSummEnergy.Set(gMC->NofVolumes()+1);
602 fSum2Energy.Set(gMC->NofVolumes()+1);
603
604 //
605 fMCQA = new AliMCQA(gAlice->GetNdets());
606
607 // Register MC in configuration
608 AliConfig::Instance()->Add(gMC);
609
610}
611
612//_______________________________________________________________________
613void AliMC::MediaTable()
614{
615 //
616 // Built media table to get from the media number to
617 // the detector id
618 //
619
620 Int_t kz, nz, idt, lz, i, k, ind;
621 // Int_t ibeg;
622 TObjArray &dets = *gAlice->Detectors();
623 AliModule *det;
624 Int_t ndets=gAlice->GetNdets();
625 //
626 // For all detectors
627 for (kz=0;kz<ndets;kz++) {
628 // If detector is defined
629 if((det=dynamic_cast<AliModule*>(dets[kz]))) {
630 TArrayI &idtmed = *(det->GetIdtmed());
631 for(nz=0;nz<100;nz++) {
632 // Find max and min material number
633 if((idt=idtmed[nz])) {
634 det->LoMedium() = det->LoMedium() < idt ? det->LoMedium() : idt;
635 det->HiMedium() = det->HiMedium() > idt ? det->HiMedium() : idt;
636 }
637 }
638 if(det->LoMedium() > det->HiMedium()) {
639 det->LoMedium() = 0;
640 det->HiMedium() = 0;
641 } else {
642 if(det->HiMedium() > fImedia->GetSize()) {
643 Error("MediaTable","Increase fImedia from %d to %d",
644 fImedia->GetSize(),det->HiMedium());
645 return;
646 }
647 // Tag all materials in rage as belonging to detector kz
648 for(lz=det->LoMedium(); lz<= det->HiMedium(); lz++) {
649 (*fImedia)[lz]=kz;
650 }
651 }
652 }
653 }
654 //
655 // Print summary table
656 printf(" Traking media ranges:\n");
657 for(i=0;i<(ndets-1)/6+1;i++) {
658 for(k=0;k< (6<ndets-i*6?6:ndets-i*6);k++) {
659 ind=i*6+k;
660 det=dynamic_cast<AliModule*>(dets[ind]);
661 if(det)
662 printf(" %6s: %3d -> %3d;",det->GetName(),det->LoMedium(),
663 det->HiMedium());
664 else
665 printf(" %6s: %3d -> %3d;","NULL",0,0);
666 }
667 printf("\n");
668 }
669}
670
671//_______________________________________________________________________
672void AliMC::ReadTransPar()
673{
674 //
675 // Read filename to set the transport parameters
676 //
677
678
679 const Int_t kncuts=10;
680 const Int_t knflags=11;
681 const Int_t knpars=kncuts+knflags;
682 const char kpars[knpars][7] = {"CUTGAM" ,"CUTELE","CUTNEU","CUTHAD","CUTMUO",
683 "BCUTE","BCUTM","DCUTE","DCUTM","PPCUTM","ANNI",
684 "BREM","COMP","DCAY","DRAY","HADR","LOSS",
685 "MULS","PAIR","PHOT","RAYL"};
686 char line[256];
687 char detName[7];
688 char* filtmp;
689 Float_t cut[kncuts];
690 Int_t flag[knflags];
691 Int_t i, itmed, iret, ktmed, kz;
692 FILE *lun;
693 //
694 // See whether the file is there
695 filtmp=gSystem->ExpandPathName(fTransParName.Data());
696 lun=fopen(filtmp,"r");
697 delete [] filtmp;
698 if(!lun) {
699 Warning("ReadTransPar","File %s does not exist!\n",fTransParName.Data());
700 return;
701 }
702 //
703 if(fDebug) {
704 printf(" "); for(i=0;i<60;i++) printf("*"); printf("\n");
705 printf(" *%59s\n","*");
706 printf(" * Please check carefully what you are doing!%10s\n","*");
707 printf(" *%59s\n","*");
708 }
709 //
710 while(1) {
711 // Initialise cuts and flags
712 for(i=0;i<kncuts;i++) cut[i]=-99;
713 for(i=0;i<knflags;i++) flag[i]=-99;
714 itmed=0;
715 for(i=0;i<256;i++) line[i]='\0';
716 // Read up to the end of line excluded
717 iret=fscanf(lun,"%[^\n]",line);
718 if(iret<0) {
719 //End of file
720 fclose(lun);
721 if(fDebug){
722 printf(" *%59s\n","*");
723 printf(" "); for(i=0;i<60;i++) printf("*"); printf("\n");
724 }
725 return;
726 }
727 // Read the end of line
728 fscanf(lun,"%*c");
729 if(!iret) continue;
730 if(line[0]=='*') continue;
731 // Read the numbers
732 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",
733 detName,&itmed,&cut[0],&cut[1],&cut[2],&cut[3],&cut[4],&cut[5],&cut[6],&cut[7],&cut[8],
734 &cut[9],&flag[0],&flag[1],&flag[2],&flag[3],&flag[4],&flag[5],&flag[6],&flag[7],
735 &flag[8],&flag[9],&flag[10]);
736 if(!iret) continue;
737 if(iret<0) {
738 //reading error
739 Warning("ReadTransPar","Error reading file %s\n",fTransParName.Data());
740 continue;
741 }
742 // Check that the module exist
743 AliModule *mod = gAlice->GetModule(detName);
744 if(mod) {
745 // Get the array of media numbers
746 TArrayI &idtmed = *mod->GetIdtmed();
747 // Check that the tracking medium code is valid
748 if(0<=itmed && itmed < 100) {
749 ktmed=idtmed[itmed];
750 if(!ktmed) {
751 Warning("ReadTransPar","Invalid tracking medium code %d for %s\n",itmed,mod->GetName());
752 continue;
753 }
754 // Set energy thresholds
755 for(kz=0;kz<kncuts;kz++) {
756 if(cut[kz]>=0) {
757 if(fDebug) printf(" * %-6s set to %10.3E for tracking medium code %4d for %s\n",
758 kpars[kz],cut[kz],itmed,mod->GetName());
759 gMC->Gstpar(ktmed,kpars[kz],cut[kz]);
760 }
761 }
762 // Set transport mechanisms
763 for(kz=0;kz<knflags;kz++) {
764 if(flag[kz]>=0) {
765 if(fDebug) printf(" * %-6s set to %10d for tracking medium code %4d for %s\n",
766 kpars[kncuts+kz],flag[kz],itmed,mod->GetName());
767 gMC->Gstpar(ktmed,kpars[kncuts+kz],Float_t(flag[kz]));
768 }
769 }
770 } else {
771 Warning("ReadTransPar","Invalid medium code %d *\n",itmed);
772 continue;
773 }
774 } else {
775 if(fDebug) printf("%s::ReadTransParModule: %s not present\n",ClassName(),detName);
776 continue;
777 }
778 }
779}
780
781//_______________________________________________________________________
782void AliMC::SetTransPar(const char *filename)
783{
784 //
785 // Sets the file name for transport parameters
786 //
787 fTransParName = filename;
788}
789
790//_______________________________________________________________________
2057aecb 791void AliMC::Browse(TBrowser *b) const
5d12ce38 792{
793 //
794 // Called when the item "Run" is clicked on the left pane
795 // of the Root browser.
796 // It displays the Root Trees and all detectors.
797 //
798 //detectors are in folders anyway
799 b->Add(fMCQA,"AliMCQA");
800}
801
802//PH
803//_______________________________________________________________________
804void AliMC::AddHit(Int_t id, Int_t track, Int_t *vol, Float_t *hits) const
805{
806 //
807 // Add a hit to detector id
808 //
809 TObjArray &dets = *gAlice->Modules();
810 if(dets[id]) dynamic_cast<AliModule*>(dets[id])->AddHit(track,vol,hits);
811}
812
813//_______________________________________________________________________
814void AliMC::AddDigit(Int_t id, Int_t *tracks, Int_t *digits) const
815{
816 //
817 // Add digit to detector id
818 //
819 TObjArray &dets = *gAlice->Modules();
820 if(dets[id]) dynamic_cast<AliModule*>(dets[id])->AddDigit(tracks,digits);
821}
822
823//_______________________________________________________________________
824Int_t AliMC::GetCurrentTrackNumber() const {
825 //
826 // Returns current track
827 //
828 return gAlice->GetRunLoader()->Stack()->GetCurrentTrackNumber();
829}
830
831//_______________________________________________________________________
832void AliMC::DumpPart (Int_t i) const
833{
834 //
835 // Dumps particle i in the stack
836 //
837 AliRunLoader * runloader = gAlice->GetRunLoader();
838 if (runloader->Stack())
839 runloader->Stack()->DumpPart(i);
840}
841
842//_______________________________________________________________________
843void AliMC::DumpPStack () const
844{
845 //
846 // Dumps the particle stack
847 //
848 AliRunLoader * runloader = gAlice->GetRunLoader();
849 if (runloader->Stack())
850 runloader->Stack()->DumpPStack();
851}
852
853//_______________________________________________________________________
854Int_t AliMC::GetNtrack() const {
855 //
856 // Returns number of tracks in stack
857 //
858 Int_t ntracks = -1;
859 AliRunLoader * runloader = gAlice->GetRunLoader();
860 if (runloader->Stack())
861 ntracks = runloader->Stack()->GetNtrack();
862 return ntracks;
863}
864
865//_______________________________________________________________________
866Int_t AliMC::GetPrimary(Int_t track) const
867{
868 //
869 // return number of primary that has generated track
870 //
871 Int_t nprimary = -999;
872 AliRunLoader * runloader = gAlice->GetRunLoader();
873 if (runloader->Stack())
874 nprimary = runloader->Stack()->GetPrimary(track);
875 return nprimary;
876}
877
878//_______________________________________________________________________
879TParticle* AliMC::Particle(Int_t i) const
880{
d0d4a6b3 881 // Returns the i-th particle from the stack taking into account
882 // the remaping done by PurifyKine
5d12ce38 883 AliRunLoader * runloader = gAlice->GetRunLoader();
884 if (runloader)
885 if (runloader->Stack())
886 return runloader->Stack()->Particle(i);
887 return 0x0;
888}
889
890//_______________________________________________________________________
891TObjArray* AliMC::Particles() const {
892 //
893 // Returns pointer to Particles array
894 //
895 AliRunLoader * runloader = gAlice->GetRunLoader();
896 if (runloader)
897 if (runloader->Stack())
898 return runloader->Stack()->Particles();
899 return 0x0;
900}
901
902//_______________________________________________________________________
903void AliMC::PushTrack(Int_t done, Int_t parent, Int_t pdg, Float_t *pmom,
904 Float_t *vpos, Float_t *polar, Float_t tof,
2057aecb 905 TMCProcess mech, Int_t &ntr, Float_t weight, Int_t is) const
5d12ce38 906{
907// Delegate to stack
908//
909 AliRunLoader * runloader = gAlice->GetRunLoader();
910 if (runloader)
911 if (runloader->Stack())
912 runloader->Stack()->PushTrack(done, parent, pdg, pmom, vpos, polar, tof,
913 mech, ntr, weight, is);
914}
915
916//_______________________________________________________________________
917void AliMC::PushTrack(Int_t done, Int_t parent, Int_t pdg,
918 Double_t px, Double_t py, Double_t pz, Double_t e,
919 Double_t vx, Double_t vy, Double_t vz, Double_t tof,
920 Double_t polx, Double_t poly, Double_t polz,
2057aecb 921 TMCProcess mech, Int_t &ntr, Float_t weight, Int_t is) const
5d12ce38 922{
923 // Delegate to stack
924 //
925 AliRunLoader * runloader = gAlice->GetRunLoader();
926 if (runloader)
927 if (runloader->Stack())
928 runloader->Stack()->PushTrack(done, parent, pdg, px, py, pz, e, vx, vy, vz, tof,
929 polx, poly, polz, mech, ntr, weight, is);
930}
931
932//_______________________________________________________________________
2057aecb 933void AliMC::SetHighWaterMark(Int_t nt) const
5d12ce38 934{
935 //
936 // Set high water mark for last track in event
937 AliRunLoader * runloader = gAlice->GetRunLoader();
938 if (runloader)
939 if (runloader->Stack())
940 runloader->Stack()->SetHighWaterMark(nt);
941}
942
943//_______________________________________________________________________
2057aecb 944void AliMC::KeepTrack(Int_t track) const
5d12ce38 945{
946 //
947 // Delegate to stack
948 //
949 AliRunLoader * runloader = gAlice->GetRunLoader();
950 if (runloader)
951 if (runloader->Stack())
952 runloader->Stack()->KeepTrack(track);
953}
954
955//_______________________________________________________________________
2057aecb 956void AliMC::FlagTrack(Int_t track) const
5d12ce38 957{
958 // Delegate to stack
959 //
960 AliRunLoader * runloader = gAlice->GetRunLoader();
961 if (runloader)
962 if (runloader->Stack())
963 runloader->Stack()->FlagTrack(track);
964}
965
966//_______________________________________________________________________
2057aecb 967void AliMC::SetCurrentTrack(Int_t track) const
5d12ce38 968{
969 //
970 // Set current track number
971 //
972 AliRunLoader * runloader = gAlice->GetRunLoader();
973 if (runloader)
974 if (runloader->Stack())
975 runloader->Stack()->SetCurrentTrack(track);
976}
977
978//_______________________________________________________________________
979void AliMC::AddTrackReference(Int_t label){
980 //
981 // add a trackrefernce to the list
982 if (!fTrackReferences) {
983 cerr<<"Container trackrefernce not active\n";
984 return;
985 }
986 Int_t nref = fTrackReferences->GetEntriesFast();
987 TClonesArray &lref = *fTrackReferences;
988 new(lref[nref]) AliTrackReference(label);
989}
990
991
992
993//_______________________________________________________________________
994void AliMC::ResetTrackReferences()
995{
996 //
997 // Reset all references
998 //
999 if (fTrackReferences) fTrackReferences->Clear();
1000
1001 TIter next(gAlice->Modules());
1002 AliModule *detector;
1003 while((detector = dynamic_cast<AliModule*>(next()))) {
1004 detector->ResetTrackReferences();
1005 }
1006}
1007
1008void AliMC::RemapTrackReferencesIDs(Int_t *map)
1009{
1010 //
1011 // Remapping track reference
1012 // Called at finish primary
1013 //
1014 if (!fTrackReferences) return;
1015 for (Int_t i=0;i<fTrackReferences->GetEntries();i++){
1016 AliTrackReference * ref = dynamic_cast<AliTrackReference*>(fTrackReferences->UncheckedAt(i));
1017 if (ref) {
1018 Int_t newID = map[ref->GetTrack()];
1019 if (newID>=0) ref->SetTrack(newID);
1020 else {
1021 //ref->SetTrack(-1);
1022 ref->SetBit(kNotDeleted,kFALSE);
1023 fTrackReferences->RemoveAt(i);
1024 }
1025 }
1026 }
1027 fTrackReferences->Compress();
1028}