]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - STEER/AliMC.h
Getting properly the run number and adding a protection in case of run number inconsi...
[u/mrichter/AliRoot.git] / STEER / AliMC.h
index 65b727c23b3309c89adaedec06d836836e2226a2..c10e035825368e1971f73f8de32c1797065d8e77 100644 (file)
 
 /* $Id$ */
 
-///////////////////////////////////////////////////////////////////////////////
-//                                                                           //
-//                                                                           //
-//   Abstract Monte Carlo interface                                          //
-//                                                                           //
-//                                                                           //
-///////////////////////////////////////////////////////////////////////////////
-
-#include <TNamed.h>
-#include <AliRndm.h>
-#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 <TArrayF.h>
+#include <TArrayI.h>
+#include <TMCProcess.h>
+#include <TVirtualMCApplication.h>
+
+class TParticle;
+class AliGenerator;
+class AliMCQA;
+
+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;
-  
+    AliMC(const AliMC &mc);
+    virtual ~AliMC();
     
-    // 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;
+    AliMC& operator= (const AliMC &mc) {
+      // Assignment operator
+      mc.Copy(*this);
+      return *this;
+    }
+
+//
+//  MC Application
+//
+   virtual  void  ConstructGeometry();
+   virtual  void  InitGeometry();     
+   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  void Field(const Double_t* x, Double_t* b) const;
+   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  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");
+   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;
+   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;
+   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, 
+                          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 
+   virtual void   AddTrackReference(Int_t label);
+   TClonesArray   *TrackReferences()   const {return fTrackReferences;}
+   virtual void   RemapTrackReferencesIDs(Int_t *map); //remaping track references MI
+   virtual void   ResetTrackReferences();
+   virtual void   FixParticleDecaytime();
+
+private:
+   void Copy (TObject &mc) const;
+   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
+   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
     
-    //
-    // 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
+    ClassDef(AliMC,2)
 };
 
-#endif 
-
+#endif