STEER/AliRun.h

   1 #ifndef AliRun_H
   2 #define AliRun_H
   3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 /* $Id$ */
   7
   8 #include <TROOT.h>
   9 #include <TBrowser.h>
  10 #include <TList.h>
  11 #include <TStopwatch.h>
  12 #include <TTree.h>
  13 #include <TGeometry.h>
  14 #include <TDatabasePDG.h>
  15
  16 #include "AliDetector.h"
  17 #include "AliHeader.h"
  18 #include "AliMagF.h"
  19 #include "AliMC.h"
  20 #include "AliGenerator.h"
  21 class AliLego;
  22 //#include "AliLego.h"
  23
  24 enum {Keep_Bit=1, Daughters_Bit=2, Done_Bit=4};
  25
  26 class AliDisplay;
  27
  28 class AliRun : public TNamed {
  29
  30 protected:
  31   Int_t         fRun;          //Current run number
  32   Int_t         fEvent;        //Current event number (from 1)
  33   Int_t         fNtrack;       //Number of tracks
  34   Int_t         fHgwmk;        //Last track purified
  35   Int_t         fCurrent;      //Last track returned from the stack
  36   Int_t         fDebug;        //Debug flag
  37   AliHeader     fHeader;       //Header information
  38   TTree        *fTreeD;        //Pointer to Tree for Digits
  39   TTree        *fTreeK;        //Pointer to Tree for Kinematics
  40   TTree        *fTreeH;        //Pointer to Tree for Hits
  41   TTree        *fTreeE;        //Pointer to Tree for Header
  42   TTree        *fTreeR;        //Pointer to Tree for Reconstructed Objects
  43   TObjArray    *fModules;      //List of Detectors
  44   TClonesArray *fParticles;    //Pointer to list of particles
  45   TGeometry    *fGeometry;     //Pointer to geometry
  46   AliDisplay   *fDisplay;      //Pointer to event display
  47   TStopwatch    fTimer;        //Timer object
  48   AliMagF      *fField;        //Magnetic Field Map
  49   AliMC        *fMC;           //pointer to MonteCarlo object
  50   TArrayI      *fImedia;       //Array of correspondence between media and detectors
  51   Int_t         fNdets;        //Number of detectors
  52   Float_t       fTrRmax;       //Maximum radius for tracking
  53   Float_t       fTrZmax;       //Maximu z for tracking
  54   AliGenerator *fGenerator;    //Generator used in the MC
  55   Bool_t        fInitDone;     //true when initialisation done
  56   AliLego      *fLego;         //pointer to aliLego object if it exists
  57   TDatabasePDG *fPDGDB;        //Particle factory object!
  58   TList        *fHitLists;     //Lists of hits to be remapped by PurifyKine
  59   TArrayF       fEventEnergy;  //Energy deposit for current event
  60   TArrayF       fSummEnergy;   //Energy per event in each volume
  61   TArrayF       fSum2Energy;   //Energy squared per event in each volume
  62   TString       fConfigFunction; //Configuration file to be executed
  63
  64
  65 public:
  66    // Creators - distructors
  67    AliRun();
  68    AliRun(const char *name, const char *title);
  69    virtual ~AliRun();
  70
  71    virtual  void  AddHit(Int_t id, Int_t track, Int_t *vol, Float_t *hits) const;
  72    virtual  void  AddDigit(Int_t id, Int_t *tracks, Int_t *digits) const;
  73    virtual  void  AddHitList(TCollection *hitList) {fHitLists->Add(hitList);}
  74    virtual  void  Browse(TBrowser *b);
  75    virtual  void  Build();
  76    virtual  void  BuildSimpleGeometry();
  77    virtual  void  CleanDetectors();
  78    virtual  void  CleanParents();
  79    TObjArray     *Detectors() const {return fModules;}
  80    TObjArray     *Modules() const {return fModules;}
  81    Int_t          CurrentTrack() const {return fCurrent;}
  82    AliDisplay    *Display() { return fDisplay;}
  83    virtual  Int_t DistancetoPrimitive(Int_t px, Int_t py);
  84    virtual  void  DumpPart (Int_t i);
  85    virtual  void  DumpPStack ();
  86    virtual AliMagF *Field() const {return fField;}
  87    virtual  void  FillTree();
  88    virtual  void  FinishPrimary();
  89    virtual  void  FinishEvent();
  90    virtual  void  FinishRun();
  91    virtual  void  FlagTrack(Int_t track);
  92    void           AddEnergyDeposit(Int_t id, Float_t edep)
  93                                        {fEventEnergy[id]+=edep;}
  94    Int_t          GetEvNumber() const {return fEvent;}
  95    Int_t          GetRunNumber() const {return fRun;}
  96    void           SetRunNumber(Int_t run) {fRun=run;}
  97    Int_t          GetDebug() const {return fDebug;}
  98    AliModule     *GetModule(const char *name);
  99    AliDetector   *GetDetector(const char *name);
 100    Int_t          GetModuleID(const char *name);
 101    virtual  Int_t GetEvent(Int_t event);
 102    virtual  void  SetEvent(Int_t event) {fEvent=event;}
 103    virtual  void  SetConfigFunction(const char * config="Config();")
 104     {fConfigFunction=config;}
 105    virtual  const char *GetConfigFunction() const
 106     {return fConfigFunction.Data();}
 107    TGeometry     *GetGeometry();
 108    AliHeader     *GetHeader() {return &fHeader;}
 109    virtual  void  GetNextTrack(Int_t &mtrack, Int_t &ipart, Float_t *pmom,
 110                                Float_t &e, Float_t *vpos, Float_t *polar,
 111                                Float_t &tof);
 112    Int_t          GetNtrack() {return fNtrack;}
 113    virtual  Int_t GetPrimary(Int_t track);
 114    virtual  void  InitMC(const char *setup="Config.C");
 115    virtual  void  Init(const char *setup="Config.C") {InitMC(setup);}
 116    Bool_t         IsFolder() {return kTRUE;}
 117    virtual  void  MakeTree(Option_t *option="KH");
 118    TClonesArray  *Particles() {return fParticles;};
 119    virtual  void  PurifyKine();
 120    virtual  Int_t PurifyKine(Int_t lastSavedTrack, Int_t nofTracks);
 121    virtual  void  BeginEvent();
 122    virtual  void  ResetDigits();
 123    virtual  void  ResetHits();
 124    virtual  void  ResetPoints();
 125    virtual  void  SetTransPar(char *filename="$(ALICE_ROOT)/data/galice.cuts");
 126    virtual  void  ResetStack() {fCurrent=-1;fHgwmk=0;fNtrack=0;fParticles->Clear();}
 127    virtual  void  RunMC(Int_t nevent=1, const char *setup="Config.C");
 128    virtual  void  Run(Int_t nevent=1, const char *setup="Config.C")
 129   {RunMC(nevent,setup);}
 130    virtual  void  RunLego(const char *setup="Config.C",Int_t ntheta=60,Float_t themin=2,Float_t themax=178,
 131                           Int_t nphi=60,Float_t phimin=0,Float_t phimax=360,Float_t rmin=0,
 132                           Float_t rmax=570,Float_t zmax=10000);
 133    virtual  Bool_t IsLegoRun() const {return (fLego!=0);}
 134    virtual  void  SetCurrentTrack(Int_t track);
 135    virtual  void  SetDebug(const Int_t level=1) {fDebug = level;}
 136    virtual  void  SetDisplay(AliDisplay *display) {fDisplay = display;}
 137    virtual  void  StepManager(Int_t id);
 138    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");
 139    virtual  void  SetTrack(Int_t done, Int_t parent, Int_t pdg,
 140                                Float_t *pmom, Float_t *vpos, Float_t *polar,
 141                                Float_t tof, const char *mecha, Int_t &ntr,
 142                                Float_t weight=1);
 143    virtual  void  KeepTrack(const Int_t);
 144    virtual  void  MediaTable();
 145    virtual  Float_t TrackingZmax() const {return fTrZmax;}
 146    virtual  Float_t TrackingRmax() const {return fTrRmax;}
 147    virtual  void    TrackingLimits( Float_t rmax=1.e10, Float_t zmax=1.e10) {fTrRmax=rmax; fTrZmax=zmax;}
 148    virtual  Int_t   DetFromMate(Int_t i) { return (*fImedia)[i];}
 149    virtual  AliGenerator* Generator() {return fGenerator;}
 150    virtual  void SetGenerator(AliGenerator *generator);
 151    virtual  void ResetGenerator(AliGenerator *generator);
 152    virtual  void EnergySummary();
 153    virtual  const TDatabasePDG* PDGDB() const {return fPDGDB;}
 154
 155
 156    TTree         *TreeD() {return fTreeD;}
 157    TTree         *TreeE() {return fTreeE;}
 158    TTree         *TreeH() {return fTreeH;}
 159    TTree         *TreeK() {return fTreeK;}
 160    TTree         *TreeR() {return fTreeR;}
 161
 162   // --------------------------- commons -------------------------------------
 163
 164    ClassDef(AliRun,3)      //Supervisor class for all Alice detectors
 165 };
 166
 167 R__EXTERN  AliRun *gAlice;
 168
 169 #endif