]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - STEER/AliMC.h
- Three classes by MinJung Kweon AliHFEpriVtx, AliHFEsecVtx and AliHFEmcQA for primar...
[u/mrichter/AliRoot.git] / STEER / AliMC.h
index b6ee0015bd71646ed3b21b4031359af3097575a3..f5de1698fa46f57a165e5d91179689e6e136843c 100644 (file)
@@ -5,32 +5,40 @@
 
 /* $Id$ */
 
+//
+// This is the ALICE implementation of TVirtualMCApplication
+// for simulation with different implementations
+// of the Virtual MonteCarlo
+//
+
 #include <TArrayF.h>
 #include <TArrayI.h>
+#include <TClonesArray.h>
+#include <TList.h>
 #include <TMCProcess.h>
 #include <TVirtualMCApplication.h>
 
+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,54 @@ 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);
+                          Int_t is = 0) const;
+   virtual  void  SetHighWaterMark(Int_t nt) const;
    
-   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();
-   
-
-
-private:
-   void Copy (TObject &mc) 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();
+ 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)
 };