Bug fix

[u/mrichter/AliRoot.git] / STEER / AliMC.h

diff --git a/STEER/AliMC.h b/STEER/AliMC.h
index ab332362cdba30e0d69a0fc5aadb304d0f727938..23166f407d073bc1422e0e1c5ecf6dcdce9ed810 100644 (file)
--- a/STEER/AliMC.h
+++ b/STEER/AliMC.h
@@ -1,13 +1,135 @@
-#ifndef ALI_MC__H
-#define ALI_MC__H
+#ifndef ALIMC_H
+#define ALIMC_H
 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
  * See cxx source for full Copyright notice                               */
 
 /* $Id$ */
 
-#include "TVirtualMC.h"
+//
+// This is the ALICE implementation of TVirtualMCApplication
+// for simulation with different implementations
+// of the Virtual MonteCarlo
+//
 
-typedef TVirtualMC  AliMC;
+#include <TArrayF.h>
+#include <TArrayI.h>
+#include <TList.h>
+#include <TMCProcess.h>
+#include <TVirtualMCApplication.h>
 
-#endif //ALI_MC__H
+class TParticle;
+class AliGenerator;
+class AliMCQA;
 
+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  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  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
+    
+    ClassDef(AliMC,2)
+};
+
+ 
+#endif