3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
9 // This is the ALICE implementation of TVirtualMCApplication
10 // for simulation with different implementations
11 // of the Virtual MonteCarlo
16 #include <TClonesArray.h>
18 #include <TMCProcess.h>
19 #include <TVirtualMCApplication.h>
26 class AliTrackReference;
27 class AliTransportMonitor;
29 class AliMC : public TVirtualMCApplication {
32 AliMC(const char *name, const char *title);
38 virtual void ConstructGeometry();
39 virtual Bool_t MisalignGeometry();
40 virtual void ConstructOpGeometry();
41 virtual void AddParticles();
42 virtual void InitGeometry();
43 virtual void SetAllAlignableVolumes();
44 virtual void GeneratePrimaries();
45 virtual void BeginEvent();
46 virtual void BeginPrimary();
47 virtual void PreTrack();
48 virtual void Stepping();
49 virtual void PostTrack();
50 virtual void FinishPrimary();
51 virtual void FinishEvent();
52 virtual Double_t TrackingZmax() const {return fTrZmax;}
53 virtual Double_t TrackingRmax() const {return fTrRmax;}
54 virtual Int_t DetFromMate(Int_t i) const { return (*fImedia)[i];}
57 virtual AliGenerator* Generator() const {return fGenerator;}
58 virtual void SetGenerator(AliGenerator *generator);
59 virtual void ResetGenerator(AliGenerator *generator);
62 virtual void ReadTransPar();
63 virtual void MediaTable();
64 virtual void EnergySummary();
65 virtual void FinishRun();
66 void AddEnergyDeposit(Int_t id, Float_t edep)
67 {fEventEnergy[id]+=edep;}
68 virtual void ResetHits();
69 virtual void ResetDigits();
70 virtual void ResetSDigits();
71 virtual void TrackingLimits( Float_t rmax=1.e10, Float_t zmax=1.e10)
72 {fTrRmax=rmax; fTrZmax=zmax;}
73 virtual void DecayLimits( Float_t rmin = -1., Float_t rmax = -1., Int_t pdg = 0)
74 {fRDecayMin = rmin; fRDecayMax = rmax; fDecayPdg = pdg;}
77 virtual void SetTransPar(const char *filename="$(ALICE_ROOT)/data/galice.cuts");
79 virtual void AddHit(Int_t id, Int_t track, Int_t *vol, Float_t *hits) const;
80 virtual void AddDigit(Int_t id, Int_t *tracks, Int_t *digits) const;
81 virtual void AddHitList(TCollection *hitList) {if (hitList) fHitLists->Add(hitList);}
82 Int_t GetCurrentTrackNumber() const;
83 virtual void DumpPart (Int_t i) const;
84 virtual void DumpPStack () const;
85 TList* GetHitLists() const {return fHitLists ;}
86 Int_t GetNtrack() const;
87 virtual Int_t GetPrimary(Int_t track) const;
88 const TObjArray* Particles() const;
89 TParticle *Particle(Int_t i) const;
90 virtual void PushTrack(Int_t done, Int_t parent, Int_t pdg,
91 const Float_t *pmom, const Float_t *vpos, const Float_t *polar,
92 Float_t tof, TMCProcess mech, Int_t &ntr,
93 Float_t weight = 1, Int_t is = 0) const;
94 virtual void PushTrack(Int_t done, Int_t parent, Int_t pdg,
95 Double_t px, Double_t py, Double_t pz, Double_t e,
96 Double_t vx, Double_t vy, Double_t vz, Double_t tof,
97 Double_t polx, Double_t poly, Double_t polz,
98 TMCProcess mech, Int_t &ntr, Float_t weight=1,
100 virtual void SetHighWaterMark(Int_t nt) const;
102 virtual void KeepTrack(Int_t itra) const;
103 virtual void FlagTrack(Int_t track) const;
104 virtual void SetCurrentTrack(Int_t track) const;
105 // Track reference related
106 AliTrackReference* AddTrackReference(Int_t label, Int_t id = -999);
107 // const TClonesArray* TrackReferences() const {return &fTrackReferences;}
108 virtual void RemapTrackReferencesIDs(const Int_t *map); //remaping track references MI
109 virtual void ResetTrackReferences();
110 virtual void FixParticleDecaytime();
112 void SetGeometryFromCDB();
113 Bool_t IsGeometryFromCDB() const;
115 void SetUseMonitoring(Bool_t flag=kTRUE) { fUseMonitoring = flag; }
116 AliTransportMonitor *GetTransportMonitor() const { return fMonitor; }
117 // Random number generator status
118 void SetSaveRndmStatus(Bool_t value) { fSaveRndmStatus = value; }
119 void SetSaveRndmStatusPerEvent(Bool_t value) { fSaveRndmEventStatus = value; }
120 void SetReadRndmStatus(Bool_t value) { fReadRndmStatus = value; };
121 void SetRndmFileName(const TString& fileName) { fRndmFileName = fileName; };
124 AliMC(const AliMC&); // Not implemented
125 AliMC& operator= (const AliMC&); // Not implemented
126 void MakeTmpTrackRefsTree();
127 void ReorderAndExpandTreeTR();
130 AliGenerator *fGenerator; // Generator used in the MC
131 Bool_t fSaveRndmStatus; //! Options to save random engine status
132 Bool_t fSaveRndmEventStatus; //! Options to save random engine status for each event
133 Bool_t fReadRndmStatus; //! Options to read random engine status
134 Bool_t fUseMonitoring; //! Activate monitoring
135 TString fRndmFileName; //! The file name of random engine status to be read in
136 TArrayF fEventEnergy; //! Energy deposit for current event
137 TArrayF fSummEnergy; //! Energy per event in each volume
138 TArrayF fSum2Energy; //! Energy squared per event in each volume
139 Float_t fTrRmax; // Maximum radius for tracking
140 Float_t fTrZmax; // Maximu z for tracking
141 Float_t fRDecayMax; // Maximum radius for decay
142 Float_t fRDecayMin; // Minimum radius for decay
143 Int_t fDecayPdg; // PDG code of particle with forced decay length
144 TArrayI *fImedia; //! Array of correspondence between media and detectors
145 TString fTransParName; // Name of the transport parameters file
146 AliTransportMonitor *fMonitor; //! Transport monitoring tool
147 TList *fHitLists; //! Lists of hits to be remapped by PurifyKine
148 //Temporary Track Reference tree related
149 TTree *fTmpTreeTR; //! Temporary track reference tree
150 TFile *fTmpFileTR; //! Temporary track reference file
151 TClonesArray fTrackReferences; //! List of track references - for one primary track only
152 TClonesArray fTmpTrackReferences; //! Temporary list of track references - for one primary track only