Parametrise number of modules and length of names.
[u/mrichter/AliRoot.git] / STEER / AliRun.h
1 #ifndef AliRun_H
2 #define AliRun_H
3 #include <TROOT.h>
4 #include <TBrowser.h>
5 #include <TList.h>
6 #include <TStopwatch.h>
7 #include <TTree.h>
8 #include <TGeometry.h>
9 #include "AliModule.h"
10 #include "AliHeader.h"
11 #include "AliMagF.h"
12 #include "AliMC.h"
13 #include "AliGenerator.h"
14 #include "AliLego.h"
15
16 const enum {kMaxModules = 25, kLenModuleName=7};
17
18 class AliDisplay;
19
20 class AliRun : public TNamed {
21
22 protected:
23   Int_t         fRun;          //Current run number
24   Int_t         fEvent;        //Current event number (from 1)
25   Int_t         fNtrack;       //Number of tracks
26   Int_t         fHgwmk;        //Last track purified
27   Int_t         fCurrent;      //Last track returned from the stack
28   Int_t         fDebug;        //Debug flag
29   AliHeader     fHeader;       //Header information
30   TTree        *fTreeD;        //Pointer to Tree for Digits
31   TTree        *fTreeK;        //Pointer to Tree for Kinematics
32   TTree        *fTreeH;        //Pointer to Tree for Hits
33   TTree        *fTreeE;        //Pointer to Tree for Header
34   TTree        *fTreeR;        //Pointer to Tree for Reconstructed Objects
35   TObjArray    *fModules;    //List of Detectors
36   TClonesArray *fParticles;    //Pointer to list of particles
37   TGeometry    *fGeometry;     //Pointer to geometry
38   AliDisplay   *fDisplay;      //Pointer to event display
39   TStopwatch    fTimer;        //Timer object
40   AliMagF      *fField;        //Magnetic Field Map
41   AliMC        *fMC;           //pointer to MonteCarlo object
42   char          fDnames[kMaxModules][kLenModuleName];
43                                //Array of detector names
44   TArrayI      *fImedia;       //Array of correspondence between media and detectors
45   Int_t         fNdets;        //Number of detectors
46   Float_t       fTrRmax;       //Maximum radius for tracking
47   Float_t       fTrZmax;       //Maximu z for tracking
48   AliGenerator *fGenerator;    //Generator used in the MC
49   Int_t        *fIdtmed;       //Array to contain media numbers
50   Bool_t        fInitDone;     //true when initialisation done
51   AliLego      *fLego;         //pointer to aliLego object if it exists
52   
53 public:
54    // Creators - distructors
55    AliRun();
56    AliRun(const char *name, const char *title);
57    virtual ~AliRun();
58
59    virtual  void  AddHit(Int_t id, Int_t track, Int_t *vol, Float_t *hits) const;
60    virtual  void  AddDigit(Int_t id, Int_t *tracks, Int_t *digits) const;
61    virtual  void  Browse(TBrowser *b);
62    virtual  void  Build();
63    virtual  void  BuildSimpleGeometry();
64    virtual  void  CleanDetectors();
65    virtual  void  CleanParents();
66    TObjArray     *Detectors() const {return fModules;}
67    TObjArray     *Modules() const {return fModules;}
68    Int_t          CurrentTrack() const {return fCurrent;}
69    AliDisplay    *Display() { return fDisplay;}
70    virtual  Int_t DistancetoPrimitive(Int_t px, Int_t py);
71    virtual  void  DumpPart (Int_t i);
72    virtual  void  DumpPStack ();
73    virtual AliMagF *Field() const {return fField;}
74    virtual  void  FillTree();
75    virtual  void  FinishPrimary();
76    virtual  void  FinishEvent();
77    virtual  void  FinishRun();
78    virtual  void  FlagTrack(Int_t track);
79    Int_t          GetEvNumber() const {return fEvent;}
80    Int_t          GetRunNumber() const {return fRun;}
81    void           SetRunNumber(Int_t run) {fRun=run;}
82    Int_t          GetDebug() const {return fDebug;}
83    AliModule     *GetModule(const char *name);
84    Int_t          GetModuleID(const char *name);
85    virtual  Int_t GetEvent(Int_t event);
86    TGeometry     *GetGeometry();
87    AliHeader     *GetHeader() {return &fHeader;}
88    virtual  void  GetNextTrack(Int_t &mtrack, Int_t &ipart, Float_t *pmom, Float_t &e, Float_t *vpos, Float_t *polar, Float_t &tof);
89    Int_t          GetNtrack() {return fNtrack;}
90    virtual  Int_t GetPrimary(Int_t track);
91    virtual  void  Init(const char *setup="Config.C");
92    Bool_t         IsFolder() {return kTRUE;}
93    virtual  void  MakeTree(Option_t *option="KH");
94    TClonesArray  *Particles() {return fParticles;};
95    virtual  void  PurifyKine();
96    virtual  Int_t PurifyKine(Int_t lastSavedTrack, Int_t nofTracks);
97    virtual  void  Reset(Int_t run, Int_t idevent);
98    virtual  void  ResetDigits();
99    virtual  void  ResetHits();
100    virtual  void  ResetPoints();
101    virtual  void  SetTransPar(char *filename="$(ALICE_ROOT)/data/galice.cuts");
102    virtual  void  ResetStack() {fCurrent=-1;fHgwmk=0;fNtrack=0;fParticles->Clear();}
103    virtual  void  Run(Int_t nevent=1, const char *setup="Config.C");
104    virtual  void  RunLego(const char *setup="Config.C",Int_t ntheta=60,Float_t themin=2,Float_t themax=178,
105                           Int_t nphi=60,Float_t phimin=0,Float_t phimax=360,Float_t rmin=0,
106                           Float_t rmax=570,Float_t zmax=10000);
107    virtual  void  SetCurrentTrack(Int_t track);                           
108    virtual  void  SetDebug(const Int_t level=1) {fDebug = level;}
109    virtual  void  SetDisplay(AliDisplay *display) {fDisplay = display;}
110    virtual  void  StepManager(Int_t id) const;
111    virtual  void  SetField(Int_t type=2, Int_t version=1, Float_t scale=1, Float_t maxField=10, char*filename="$(ALICE_ROOT)/data/field01.dat");
112    virtual  void  SetTrack(Int_t done, Int_t parent, Int_t ipart, 
113                                Float_t *pmom, Float_t *vpos, Float_t *polar, 
114                                Float_t tof, const char *mecha, Int_t &ntr,
115                                Float_t weight=1);
116    virtual  void  KeepTrack(const Int_t);
117    virtual  void  MediaTable();
118    virtual  Float_t TrackingZmax() const {return fTrZmax;}
119    virtual  Float_t TrackingRmax() const {return fTrRmax;}
120    virtual  void    TrackingLimits( Float_t rmax=1.e10, Float_t zmax=1.e10) {fTrRmax=rmax; fTrZmax=zmax;}
121    virtual  Int_t   DetFromMate(Int_t i) { return (*fImedia)[i];}
122    virtual  AliGenerator* Generator() {return fGenerator;}
123    virtual  void SetGenerator(AliGenerator *generator);
124    virtual  void EnergySummary();
125    virtual  Int_t* Idtmed() {return fIdtmed;}
126
127   // Functions from GEOCAD
128   //_______________________________________________________________________
129   
130    virtual void ReadEuclid(const char*, Int_t, const char*);
131    virtual void ReadEuclidMedia(const char*, Int_t);
132
133    TTree         *TreeD() {return fTreeD;}
134    TTree         *TreeE() {return fTreeE;}
135    TTree         *TreeH() {return fTreeH;}
136    TTree         *TreeK() {return fTreeK;}
137    TTree         *TreeR() {return fTreeR;}
138
139   // --------------------------- commons -------------------------------------
140
141    ClassDef(AliRun,1)      //Supervisor class for all Alice detectors
142 };
143  
144 EXTERN  AliRun *gAlice;
145  
146 #endif