]>
Commit | Line | Data |
---|---|---|
5d12ce38 | 1 | #ifndef ALIMC_H |
2 | #define ALIMC_H | |
3 | /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * | |
4 | * See cxx source for full Copyright notice */ | |
5 | ||
6 | /* $Id$ */ | |
7 | ||
8 | #include <TArrayF.h> | |
9 | #include <TArrayI.h> | |
10 | #include <TVirtualMCApplication.h> | |
11 | ||
12 | class AliGenerator; | |
13 | class AliMCQA; | |
14 | ||
15 | class AliMC : public TVirtualMCApplication { | |
16 | public: | |
17 | AliMC(); | |
18 | AliMC(const char *name, const char *title); | |
19 | AliMC(const AliMC &mc); | |
20 | virtual ~AliMC(); | |
21 | ||
d0d4a6b3 | 22 | AliMC& operator= (const AliMC &mc) { |
23 | // Assignment operator | |
24 | mc.Copy(*this); | |
25 | return *this; | |
26 | } | |
5d12ce38 | 27 | |
28 | // | |
29 | // MC Application | |
30 | // | |
31 | virtual void ConstructGeometry(); | |
32 | virtual void InitGeometry(); | |
33 | virtual void GeneratePrimaries(); | |
34 | virtual void BeginEvent(); | |
35 | virtual void BeginPrimary(); | |
36 | virtual void PreTrack(); | |
37 | virtual void Stepping(); | |
38 | virtual void PostTrack(); | |
39 | virtual void FinishPrimary(); | |
40 | virtual void FinishEvent(); | |
41 | virtual Double_t TrackingZmax() const {return fTrZmax;} | |
42 | virtual Double_t TrackingRmax() const {return fTrRmax;} | |
43 | virtual void Field(const Double_t* x, Double_t* b) const; | |
44 | virtual Int_t DetFromMate(Int_t i) const { return (*fImedia)[i];} | |
45 | // | |
46 | ||
47 | virtual AliGenerator* Generator() const {return fGenerator;} | |
48 | virtual void SetGenerator(AliGenerator *generator); | |
49 | virtual void ResetGenerator(AliGenerator *generator); | |
50 | ||
51 | // | |
52 | virtual void ReadTransPar(); | |
53 | virtual void MediaTable(); | |
54 | virtual void EnergySummary(); | |
55 | virtual void FinishRun(); | |
56 | void AddEnergyDeposit(Int_t id, Float_t edep) | |
57 | {fEventEnergy[id]+=edep;} | |
58 | virtual void ResetHits(); | |
59 | virtual void TrackingLimits( Float_t rmax=1.e10, Float_t zmax=1.e10) | |
60 | {fTrRmax=rmax; fTrZmax=zmax;} | |
61 | Int_t GetDebug() const {return fDebug;} | |
d0f1ee3b | 62 | virtual void SetDebug(Int_t level=0) {fDebug = level;} |
5d12ce38 | 63 | virtual void Init(); |
64 | virtual void SetTransPar(const char *filename="$(ALICE_ROOT)/data/galice.cuts"); | |
65 | virtual void Browse(TBrowser *b); | |
66 | AliMCQA *GetMCQA() const {return fMCQA;} | |
67 | //PH | |
68 | virtual void AddHit(Int_t id, Int_t track, Int_t *vol, Float_t *hits) const; | |
69 | virtual void AddDigit(Int_t id, Int_t *tracks, Int_t *digits) const; | |
70 | virtual void AddHitList(TCollection *hitList) {fHitLists->Add(hitList);} | |
71 | Int_t GetCurrentTrackNumber() const; | |
72 | virtual void DumpPart (Int_t i) const; | |
73 | virtual void DumpPStack () const; | |
74 | TList* GetHitLists() const {return fHitLists ;} | |
75 | Int_t GetNtrack() const; | |
76 | virtual Int_t GetPrimary(Int_t track) const; | |
77 | TObjArray *Particles() const; | |
78 | TParticle *Particle(Int_t i) const; | |
79 | virtual void PushTrack(Int_t done, Int_t parent, Int_t pdg, | |
80 | Float_t *pmom, Float_t *vpos, Float_t *polar, | |
81 | Float_t tof, TMCProcess mech, Int_t &ntr, | |
82 | Float_t weight = 1, Int_t is = 0); | |
83 | virtual void PushTrack(Int_t done, Int_t parent, Int_t pdg, | |
84 | Double_t px, Double_t py, Double_t pz, Double_t e, | |
85 | Double_t vx, Double_t vy, Double_t vz, Double_t tof, | |
86 | Double_t polx, Double_t poly, Double_t polz, | |
87 | TMCProcess mech, Int_t &ntr, Float_t weight=1, | |
88 | Int_t is = 0); | |
d0f1ee3b | 89 | virtual void SetHighWaterMark(Int_t nt); |
5d12ce38 | 90 | |
d0f1ee3b | 91 | virtual void KeepTrack(Int_t itra); |
5d12ce38 | 92 | virtual void FlagTrack(Int_t track); |
93 | virtual void SetCurrentTrack(Int_t track); | |
94 | // Track reference related | |
95 | virtual void AddTrackReference(Int_t label); | |
96 | TClonesArray *TrackReferences() const {return fTrackReferences;} | |
97 | virtual void RemapTrackReferencesIDs(Int_t *map); //remaping track references MI | |
98 | virtual void ResetTrackReferences(); | |
99 | ||
100 | ||
101 | ||
102 | private: | |
6c4904c2 | 103 | void Copy (TObject &mc) const; |
5d12ce38 | 104 | AliGenerator *fGenerator; // Generator used in the MC |
105 | TArrayF fEventEnergy; //! Energy deposit for current event | |
106 | TArrayF fSummEnergy; //! Energy per event in each volume | |
107 | TArrayF fSum2Energy; //! Energy squared per event in each volume | |
108 | Float_t fTrRmax; // Maximum radius for tracking | |
109 | Float_t fTrZmax; // Maximu z for tracking | |
110 | Int_t fDebug; // Debug flag | |
111 | TArrayI *fImedia; //! Array of correspondence between media and detectors | |
112 | TString fTransParName; // Name of the transport parameters file | |
113 | AliMCQA *fMCQA; // Pointer to MC Quality assurance class | |
114 | //PH | |
115 | TList *fHitLists; //! Lists of hits to be remapped by PurifyKine | |
116 | TClonesArray *fTrackReferences; //!list of track references - for one primary track only -MI | |
117 | ||
118 | ClassDef(AliMC,1) | |
119 | }; | |
120 | ||
121 | ||
122 | #endif |