X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=STEER%2FAliMC.h;h=bef8127d5f56264a6332de841e3b371e496ee73c;hb=8386cd34e0d4e7f268d6d586a1f62dcadad46d7a;hp=65b727c23b3309c89adaedec06d836836e2226a2;hpb=541d9fb65fa8c0c82d95a2838d99b44f147a159d;p=u%2Fmrichter%2FAliRoot.git diff --git a/STEER/AliMC.h b/STEER/AliMC.h index 65b727c23b3..bef8127d5f5 100644 --- a/STEER/AliMC.h +++ b/STEER/AliMC.h @@ -5,218 +5,137 @@ /* $Id$ */ -/////////////////////////////////////////////////////////////////////////////// -// // -// // -// Abstract Monte Carlo interface // -// // -// // -/////////////////////////////////////////////////////////////////////////////// - -#include -#include -#include "AliMCProcess.h" - -class TLorentzVector; -class AliMC; -class AliDecayer; -class TArrayI; - -R__EXTERN AliMC *gMC; - -enum AliMCGeomType { kGeant3, kGeant4 , kFluka}; - -class AliMC : public TNamed, public AliRndm -{ - public: +// +// 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 AliTrackReference; + +class AliMC : public TVirtualMCApplication { +public: + AliMC(); AliMC(const char *name, const char *title); - AliMC() {fRandom=0;} - virtual ~AliMC() {fgMC=gMC=0;fRandom=0;} - - // static access method - static AliMC* GetMC() { return fgMC; } - - // - // methods for building/management of geometry - // ------------------------------------------------ - // - - // functions from GCONS - virtual void Gfmate(Int_t imat, char *name, Float_t &a, Float_t &z, - Float_t &dens, Float_t &radl, Float_t &absl, - Float_t* ubuf, Int_t& nbuf) = 0; - - // detector composition - virtual void Material(Int_t& kmat, const char* name, Float_t a, - Float_t z, Float_t dens, Float_t radl, Float_t absl, - Float_t* buf, Int_t nwbuf) = 0; - virtual void Mixture(Int_t& kmat, const char *name, Float_t *a, - Float_t *z, Float_t dens, Int_t nlmat, Float_t *wmat) = 0; - virtual void Medium(Int_t& kmed, const char *name, Int_t nmat, - Int_t isvol, Int_t ifield, Float_t fieldm, Float_t tmaxfd, - Float_t stemax, Float_t deemax, Float_t epsil, - Float_t stmin, Float_t* ubuf, Int_t nbuf) = 0; - virtual void Matrix(Int_t& krot, Float_t thetaX, Float_t phiX, - Float_t thetaY, Float_t phiY, Float_t thetaZ, - Float_t phiZ) = 0; - virtual void Gstpar(Int_t itmed, const char *param, Float_t parval) = 0; - - // functions from GGEOM - virtual Int_t Gsvolu(const char *name, const char *shape, Int_t nmed, - Float_t *upar, Int_t np) = 0; - virtual void Gsdvn(const char *name, const char *mother, Int_t ndiv, - Int_t iaxis) = 0; - virtual void Gsdvn2(const char *name, const char *mother, Int_t ndiv, - Int_t iaxis, Float_t c0i, Int_t numed) = 0; - virtual void Gsdvt(const char *name, const char *mother, Float_t step, - Int_t iaxis, Int_t numed, Int_t ndvmx) = 0; - virtual void Gsdvt2(const char *name, const char *mother, Float_t step, - Int_t iaxis, Float_t c0, Int_t numed, Int_t ndvmx) = 0; - virtual void Gsord(const char *name, Int_t iax) = 0; - virtual void Gspos(const char *name, Int_t nr, const char *mother, - Float_t x, Float_t y, Float_t z, Int_t irot, - const char *konly="ONLY") = 0; - virtual void Gsposp(const char *name, Int_t nr, const char *mother, - Float_t x, Float_t y, Float_t z, Int_t irot, - const char *konly, Float_t *upar, Int_t np) = 0; - virtual void Gsbool(const char* onlyVolName, const char* manyVolName) = 0; - - virtual void SetCerenkov(Int_t itmed, Int_t npckov, Float_t *ppckov, - Float_t *absco, Float_t *effic, Float_t *rindex) = 0; - - - // functions for drawing - virtual void DrawOneSpec(const char* name) = 0; - virtual void Gsatt(const char* name, const char* att, Int_t val) = 0; - virtual void Gdraw(const char*,Float_t theta = 30, Float_t phi = 30, - Float_t psi = 0, Float_t u0 = 10, Float_t v0 = 10, - Float_t ul = 0.01, Float_t vl = 0.01) = 0; - - // Euclid - virtual void WriteEuclid(const char*, const char*, Int_t, Int_t) = 0; - - // get methods - virtual Int_t VolId(const Text_t* volName) const = 0; - virtual const char* VolName(Int_t id) const = 0; - virtual Int_t NofVolumes() const = 0; - virtual Int_t VolId2Mate(Int_t id) const = 0; - - // - // methods for physics management - // ------------------------------------------------ - // - - // set methods - virtual void SetCut(const char* cutName, Float_t cutValue) = 0; - virtual void SetProcess(const char* flagName, Int_t flagValue) = 0; - virtual Float_t Xsec(char*, Float_t, Int_t, Int_t) = 0; - - // particle table usage - virtual Int_t IdFromPDG(Int_t id) const =0; - virtual Int_t PDGFromId(Int_t pdg) const =0; - virtual void DefineParticles() = 0; - - // - // methods for step management - // ------------------------------------------------ - // - - // action methods - virtual void StopTrack() =0; - virtual void StopEvent() =0; - - // set methods - virtual void SetMaxStep(Float_t) = 0; - virtual void SetMaxNStep(Int_t) = 0; - virtual void SetUserDecay(Int_t) =0; - - // get methods - // tracking volume(s) - virtual Int_t CurrentVolID(Int_t& copyNo) const =0; - virtual Int_t CurrentVolOffID(Int_t off, Int_t& copyNo) const =0; - virtual const char* CurrentVolName() const =0; - virtual const char* CurrentVolOffName(Int_t off) const =0; - virtual Int_t CurrentMaterial(Float_t &a, Float_t &z, - Float_t &dens, Float_t &radl, Float_t &absl) const =0; - virtual Int_t CurrentEvent() const =0; - virtual void Gmtod(Float_t* xm, Float_t* xd, Int_t iflag) = 0; - virtual void Gdtom(Float_t* xd, Float_t* xm, Int_t iflag)= 0 ; - virtual Float_t MaxStep() const =0; - virtual Int_t GetMaxNStep() const = 0; - virtual Int_t GetMedium() const =0; - - // tracking particle - // dynamic properties - virtual void TrackPosition(TLorentzVector& position) const =0; - virtual void TrackMomentum(TLorentzVector& momentum) const =0; - virtual Float_t TrackStep() const =0; - virtual Float_t TrackLength() const =0; - virtual Float_t TrackTime() const =0; - virtual Float_t Edep() const =0; - // static properties - virtual Int_t TrackPid() const =0; - virtual Float_t TrackCharge() const =0; - virtual Float_t TrackMass() const =0; - virtual Float_t Etot() const =0; - - // track status - virtual Bool_t IsNewTrack() const =0; - virtual Bool_t IsTrackInside() const =0; - virtual Bool_t IsTrackEntering() const =0; - virtual Bool_t IsTrackExiting() const =0; - virtual Bool_t IsTrackOut() const =0; - virtual Bool_t IsTrackDisappeared() const =0; - virtual Bool_t IsTrackStop() const =0; - virtual Bool_t IsTrackAlive() const=0; - - // secondaries - virtual Int_t NSecondaries() const=0; - virtual void GetSecondary(Int_t isec, Int_t& particleId, - TLorentzVector& position, TLorentzVector& momentum) =0; - virtual AliMCProcess ProdProcess(Int_t isec) const =0; - virtual Int_t StepProcesses(TArrayI &proc) const = 0; - - // - // other (then geometry/step/run management) methods - // ---------------------------------------------- - // - virtual AliMCGeomType GetMCGeomType() const = 0; + virtual ~AliMC(); - // - // Geant3 specific methods - // !!! need to be transformed to common interface - // - virtual void Gdopt(const char*,const char*) = 0; - virtual void SetClipBox(const char*,Float_t=-9999,Float_t=0, Float_t=-9999,Float_t=0,Float_t=-9999,Float_t=0) = 0; - virtual void DefaultRange() = 0; - virtual void Gdhead(Int_t, const char*, Float_t=0) = 0; - virtual void Gdman(Float_t, Float_t, const char*) = 0; - virtual void SetColors() = 0; - virtual void Gtreve() = 0; - virtual void GtreveRoot() = 0; - virtual void Gckmat(Int_t, char*) = 0; - virtual void InitLego() = 0; - virtual void Gfpart(Int_t, char*, Int_t&, Float_t&, Float_t&, Float_t&) = 0; - virtual void Gspart(Int_t, const char*, Int_t, Float_t, Float_t, Float_t) = 0; - - // Control Methods - - virtual void Init() = 0; - virtual void FinishGeometry() = 0; - virtual void BuildPhysics() = 0; - virtual void ProcessEvent() = 0; - virtual void ProcessRun(Int_t nevent) = 0; - // External Decayer - virtual void SetExternalDecayer(AliDecayer* decayer) =0; - virtual AliDecayer* Decayer() const =0; - - private: - static AliMC* fgMC; // Pointer to the virtual MonteCarlo object - AliMC(const AliMC &) {} - AliMC & operator=(const AliMC &) {return (*this);} - - ClassDef(AliMC,1) //Virtual MonteCarlo Interface +// +// 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(); + virtual void PreTrack(); + virtual void Stepping(); + virtual void PostTrack(); + virtual void FinishPrimary(); + virtual void FinishEvent(); + virtual Double_t TrackingZmax() const {return fTrZmax;} + virtual Double_t TrackingRmax() const {return fTrRmax;} + virtual Int_t DetFromMate(Int_t i) const { return (*fImedia)[i];} +// + + virtual AliGenerator* Generator() const {return fGenerator;} + virtual void SetGenerator(AliGenerator *generator); + virtual void ResetGenerator(AliGenerator *generator); + +// + virtual void ReadTransPar(); + virtual void MediaTable(); + virtual void EnergySummary(); + virtual void FinishRun(); + 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;} + 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"); + //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; + virtual void AddHitList(TCollection *hitList) {if (hitList) fHitLists->Add(hitList);} + Int_t GetCurrentTrackNumber() const; + virtual void DumpPart (Int_t i) const; + virtual void DumpPStack () const; + TList* GetHitLists() const {return fHitLists ;} + Int_t GetNtrack() const; + virtual Int_t GetPrimary(Int_t track) const; + const TObjArray* Particles() const; + TParticle *Particle(Int_t i) const; + virtual void PushTrack(Int_t done, Int_t parent, Int_t pdg, + 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) 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) 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(const Int_t *map); //remaping track references MI + virtual void ResetTrackReferences(); + virtual void FixParticleDecaytime(); +// Geometry related + void SetGeometryFromCDB(); + Bool_t IsGeometryFromCDB() 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 + 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 + TList *fHitLists; //! Lists of hits to be remapped by PurifyKine + //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) }; -#endif - + +#endif