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
17 #include <TMCProcess.h>
18 #include <TVirtualMCApplication.h>
26 class AliTrackReference;
28 class AliMC : public TVirtualMCApplication {
31 AliMC(const char *name, const char *title);
32 AliMC(const AliMC &mc);
35 AliMC& operator= (const AliMC &mc) {
36 // Assignment operator
44 virtual void ConstructGeometry();
45 virtual Bool_t MisalignGeometry();
46 virtual void ConstructOpGeometry();
47 virtual void InitGeometry();
48 virtual void SetAllAlignableVolumes();
49 virtual void GeneratePrimaries();
50 virtual void BeginEvent();
51 virtual void BeginPrimary();
52 virtual void PreTrack();
53 virtual void Stepping();
54 virtual void PostTrack();
55 virtual void FinishPrimary();
56 virtual void FinishEvent();
57 virtual Double_t TrackingZmax() const {return fTrZmax;}
58 virtual Double_t TrackingRmax() const {return fTrRmax;}
59 virtual void Field(const Double_t* x, Double_t* b) const;
60 virtual Int_t DetFromMate(Int_t i) const { return (*fImedia)[i];}
63 virtual AliGenerator* Generator() const {return fGenerator;}
64 virtual void SetGenerator(AliGenerator *generator);
65 virtual void ResetGenerator(AliGenerator *generator);
68 virtual void ReadTransPar();
69 virtual void MediaTable();
70 virtual void EnergySummary();
71 virtual void FinishRun();
72 void AddEnergyDeposit(Int_t id, Float_t edep)
73 {fEventEnergy[id]+=edep;}
74 virtual void ResetHits();
75 virtual void TrackingLimits( Float_t rmax=1.e10, Float_t zmax=1.e10)
76 {fTrRmax=rmax; fTrZmax=zmax;}
77 virtual void DecayLimits( Float_t rmin = -1., Float_t rmax = -1., Int_t pdg = 0)
78 {fRDecayMin = rmin; fRDecayMax = rmax; fDecayPdg = pdg;}
81 virtual void SetTransPar(const char *filename="$(ALICE_ROOT)/data/galice.cuts");
82 virtual void Browse(TBrowser *b);
83 AliMCQA *GetMCQA() const {return fMCQA;}
85 virtual void AddHit(Int_t id, Int_t track, Int_t *vol, Float_t *hits) const;
86 virtual void AddDigit(Int_t id, Int_t *tracks, Int_t *digits) const;
87 virtual void AddHitList(TCollection *hitList) {if (hitList) fHitLists->Add(hitList);}
88 Int_t GetCurrentTrackNumber() const;
89 virtual void DumpPart (Int_t i) const;
90 virtual void DumpPStack () const;
91 TList* GetHitLists() const {return fHitLists ;}
92 Int_t GetNtrack() const;
93 virtual Int_t GetPrimary(Int_t track) const;
94 TObjArray *Particles() const;
95 TParticle *Particle(Int_t i) const;
96 virtual void PushTrack(Int_t done, Int_t parent, Int_t pdg,
97 Float_t *pmom, Float_t *vpos, Float_t *polar,
98 Float_t tof, TMCProcess mech, Int_t &ntr,
99 Float_t weight = 1, Int_t is = 0) const;
100 virtual void PushTrack(Int_t done, Int_t parent, Int_t pdg,
101 Double_t px, Double_t py, Double_t pz, Double_t e,
102 Double_t vx, Double_t vy, Double_t vz, Double_t tof,
103 Double_t polx, Double_t poly, Double_t polz,
104 TMCProcess mech, Int_t &ntr, Float_t weight=1,
106 virtual void SetHighWaterMark(Int_t nt) const;
108 virtual void KeepTrack(Int_t itra) const;
109 virtual void FlagTrack(Int_t track) const;
110 virtual void SetCurrentTrack(Int_t track) const;
111 // Track reference related
112 AliTrackReference* AddTrackReference(Int_t label, Int_t id = -999);
113 TClonesArray* TrackReferences() const {return fTrackReferences;}
114 virtual void RemapTrackReferencesIDs(Int_t *map); //remaping track references MI
115 virtual void ResetTrackReferences();
116 virtual void FixParticleDecaytime();
118 void MakeTmpTrackRefsTree();
119 void ReorderAndExpandTreeTR();
121 void Copy (TObject &mc) const;
123 AliGenerator *fGenerator; // Generator used in the MC
124 TArrayF fEventEnergy; //! Energy deposit for current event
125 TArrayF fSummEnergy; //! Energy per event in each volume
126 TArrayF fSum2Energy; //! Energy squared per event in each volume
127 Float_t fTrRmax; // Maximum radius for tracking
128 Float_t fTrZmax; // Maximu z for tracking
129 Float_t fRDecayMax; // Maximum radius for decay
130 Float_t fRDecayMin; // Minimum radius for decay
131 Int_t fDecayPdg; // PDG code of particle with forced decay length
132 TArrayI *fImedia; //! Array of correspondence between media and detectors
133 TString fTransParName; // Name of the transport parameters file
134 AliMCQA *fMCQA; // Pointer to MC Quality assurance class
135 TList *fHitLists; //! Lists of hits to be remapped by PurifyKine
136 //Temporary Track Reference tree related
137 TTree *fTmpTreeTR; //! Temporary track reference tree
138 TFile *fTmpFileTR; //! Temporary track reference file
139 TClonesArray *fTrackReferences; //! List of track references - for one primary track only
140 TClonesArray *fTmpTrackReferences; //! Temporary list of track references - for one primary track only