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