X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=STEER%2FAliMC.h;h=ebe244c82a1ac2fc5d53358321d4d50f1c1ec2c5;hb=d2aa6df0faa31b02d774c80b2c84bd90e8e16ecf;hp=b6ee0015bd71646ed3b21b4031359af3097575a3;hpb=024a7e6436f5c6bf580ecbde53395c968bb1ecc8;p=u%2Fmrichter%2FAliRoot.git diff --git a/STEER/AliMC.h b/STEER/AliMC.h index b6ee0015bd7..ebe244c82a1 100644 --- a/STEER/AliMC.h +++ b/STEER/AliMC.h @@ -5,32 +5,40 @@ /* $Id$ */ +// +// This is the ALICE implementation of TVirtualMCApplication +// for simulation with different implementations +// of the Virtual MonteCarlo +// + #include #include +#include +#include #include #include +class TParticle; +class TFile; +class TTree; + class AliGenerator; -class AliMCQA; +class AliTrackReference; class AliMC : public TVirtualMCApplication { public: AliMC(); AliMC(const char *name, const char *title); - AliMC(const AliMC &mc); virtual ~AliMC(); - AliMC& operator= (const AliMC &mc) { - // Assignment operator - mc.Copy(*this); - return *this; - } - // // MC Application // virtual void ConstructGeometry(); + virtual Bool_t MisalignGeometry(); + virtual void ConstructOpGeometry(); virtual void InitGeometry(); + virtual void SetAllAlignableVolumes(); virtual void GeneratePrimaries(); virtual void BeginEvent(); virtual void BeginPrimary(); @@ -41,7 +49,6 @@ public: virtual void FinishEvent(); virtual Double_t TrackingZmax() const {return fTrZmax;} virtual Double_t TrackingRmax() const {return fTrRmax;} - virtual void Field(const Double_t* x, Double_t* b) const; virtual Int_t DetFromMate(Int_t i) const { return (*fImedia)[i];} // @@ -57,14 +64,15 @@ public: void AddEnergyDeposit(Int_t id, Float_t edep) {fEventEnergy[id]+=edep;} virtual void ResetHits(); + virtual void ResetDigits(); + virtual void ResetSDigits(); virtual void TrackingLimits( Float_t rmax=1.e10, Float_t zmax=1.e10) - {fTrRmax=rmax; fTrZmax=zmax;} - Int_t GetDebug() const {return fDebug;} - virtual void SetDebug(Int_t level=0) {fDebug = level;} + {fTrRmax=rmax; fTrZmax=zmax;} + virtual void DecayLimits( Float_t rmin = -1., Float_t rmax = -1., Int_t pdg = 0) + {fRDecayMin = rmin; fRDecayMax = rmax; fDecayPdg = pdg;} + virtual void Init(); virtual void SetTransPar(const char *filename="$(ALICE_ROOT)/data/galice.cuts"); - virtual void Browse(TBrowser *b); - AliMCQA *GetMCQA() const {return fMCQA;} //PH virtual void AddHit(Int_t id, Int_t track, Int_t *vol, Float_t *hits) const; virtual void AddDigit(Int_t id, Int_t *tracks, Int_t *digits) const; @@ -75,48 +83,58 @@ public: TList* GetHitLists() const {return fHitLists ;} Int_t GetNtrack() const; virtual Int_t GetPrimary(Int_t track) const; - TObjArray *Particles() const; + const TObjArray* Particles() const; TParticle *Particle(Int_t i) const; virtual void PushTrack(Int_t done, Int_t parent, Int_t pdg, - Float_t *pmom, Float_t *vpos, Float_t *polar, + const Float_t *pmom, const Float_t *vpos, const Float_t *polar, Float_t tof, TMCProcess mech, Int_t &ntr, - Float_t weight = 1, Int_t is = 0); + Float_t weight = 1, Int_t is = 0) const; virtual void PushTrack(Int_t done, Int_t parent, Int_t pdg, Double_t px, Double_t py, Double_t pz, Double_t e, Double_t vx, Double_t vy, Double_t vz, Double_t tof, Double_t polx, Double_t poly, Double_t polz, TMCProcess mech, Int_t &ntr, Float_t weight=1, - Int_t is = 0); - virtual void SetHighWaterMark(Int_t nt); - - virtual void KeepTrack(Int_t itra); - virtual void FlagTrack(Int_t track); - virtual void SetCurrentTrack(Int_t track); -// Track reference related - virtual void AddTrackReference(Int_t label); - TClonesArray *TrackReferences() const {return fTrackReferences;} - virtual void RemapTrackReferencesIDs(Int_t *map); //remaping track references MI - virtual void ResetTrackReferences(); + Int_t is = 0) const; + virtual void SetHighWaterMark(Int_t nt) const; + virtual void KeepTrack(Int_t itra) const; + virtual void FlagTrack(Int_t track) const; + virtual void SetCurrentTrack(Int_t track) const; +// Track reference related + AliTrackReference* AddTrackReference(Int_t label, Int_t id = -999); + // const TClonesArray* TrackReferences() const {return &fTrackReferences;} + virtual void RemapTrackReferencesIDs(Int_t *map); //remaping track references MI + virtual void ResetTrackReferences(); + virtual void FixParticleDecaytime(); +// Geometry related + void SetGeometryFromCDB(); + Bool_t IsGeometryFromCDB() const; - -private: - void Copy (TObject &mc) const; + private: + AliMC(const AliMC&); // Not implemented + AliMC& operator= (const AliMC&); // Not implemented + void MakeTmpTrackRefsTree(); + void ReorderAndExpandTreeTR(); + private: + void RemapHits(); AliGenerator *fGenerator; // Generator used in the MC TArrayF fEventEnergy; //! Energy deposit for current event TArrayF fSummEnergy; //! Energy per event in each volume TArrayF fSum2Energy; //! Energy squared per event in each volume Float_t fTrRmax; // Maximum radius for tracking Float_t fTrZmax; // Maximu z for tracking - Int_t fDebug; // Debug flag + Float_t fRDecayMax; // Maximum radius for decay + Float_t fRDecayMin; // Minimum radius for decay + Int_t fDecayPdg; // PDG code of particle with forced decay length TArrayI *fImedia; //! Array of correspondence between media and detectors TString fTransParName; // Name of the transport parameters file - AliMCQA *fMCQA; // Pointer to MC Quality assurance class - //PH TList *fHitLists; //! Lists of hits to be remapped by PurifyKine - TClonesArray *fTrackReferences; //!list of track references - for one primary track only -MI - - ClassDef(AliMC,1) + //Temporary Track Reference tree related + TTree *fTmpTreeTR; //! Temporary track reference tree + TFile *fTmpFileTR; //! Temporary track reference file + TClonesArray fTrackReferences; //! List of track references - for one primary track only + TClonesArray fTmpTrackReferences; //! Temporary list of track references - for one primary track only + ClassDef(AliMC, 4) };