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 <TArrayF.h>
   9 #include <TArrayI.h>
  10 #include <TMCProcess.h>
  11 #include <TStopwatch.h>
  12 #include <TVirtualMC.h>
  13 #include <TVirtualMCApplication.h>
  14 #include <TError.h>
  15
  16 class TBranch;
  17 class TBrowser;
  18 class TDatabasePDG;
  19 class TFile;
  20 class TGeometry;
  21 class TList;
  22 class TParticle;
  23 class TRandom;
  24 class TTree;
  25
  26 class AliDetector;
  27 class AliDisplay;
  28 class AliGenEventHeader;
  29 class AliGenerator;
  30 class AliHeader;
  31 class AliLego;
  32 class AliLegoGenerator;
  33 class AliLegoGenerator;
  34 class AliMCQA;
  35 class AliMagF;
  36 class AliModule;
  37 class AliStack;
  38
  39 enum {kKeepBit=1, kDaughtersBit=2, kDoneBit=4};
  40
  41
  42 class AliRun : public TVirtualMCApplication {
  43 public:
  44    // Creators - distructors
  45    AliRun();
  46    AliRun(const char *name, const char *title);
  47    AliRun(const AliRun &arun);
  48    virtual ~AliRun();
  49
  50    AliRun& operator = (const AliRun &arun)
  51      {arun.Copy(*this); return (*this);}
  52    virtual  void  AddHit(Int_t id, Int_t track, Int_t *vol, Float_t *hits) const;
  53    virtual  void  AddDigit(Int_t id, Int_t *tracks, Int_t *digits) const;
  54    virtual  void  AddHitList(TCollection *hitList) {fHitLists->Add(hitList);}
  55    virtual  void  Browse(TBrowser *b);
  56    virtual  void  Build();
  57    virtual  void  BuildSimpleGeometry();
  58    virtual  void  CleanDetectors();
  59    TObjArray     *Detectors() const {return fModules;}
  60    TObjArray     *Modules() const {return fModules;}
  61    Int_t          CurrentTrack() const;
  62    AliDisplay    *Display() const { return fDisplay;}
  63    virtual  Int_t DistancetoPrimitive(Int_t px, Int_t py) const;
  64    virtual  void  DumpPart (Int_t i) const;
  65    virtual  void  DumpPStack () const;
  66    virtual AliMagF *Field() const {return fField;}
  67    virtual  void  FinishRun();
  68    virtual  void  FlagTrack(Int_t track);
  69    void           AddEnergyDeposit(Int_t id, Float_t edep)
  70                                        {fEventEnergy[id]+=edep;}
  71    Int_t          GetEvNumber() const {return fEvent;}
  72    Int_t          GetRunNumber() const {return fRun;}
  73    void           SetRunNumber(Int_t run) {fRun=run;}
  74    void           SetEventNrInRun(Int_t event) {fEventNrInRun=event;}
  75    Int_t          GetEventNrInRun() const {return fEventNrInRun;}
  76    Int_t          GetEventsPerRun() const {return fEventsPerRun;}
  77    Int_t          GetDebug() const {return fDebug;}
  78    AliModule     *GetModule(const char *name) const;
  79    TList*         GetHitLists() const {return fHitLists ;}
  80    AliDetector   *GetDetector(const char *name) const;
  81    AliMCQA       *GetMCQA() const {return fMCQA;}
  82    Int_t          GetModuleID(const char *name) const;
  83    virtual  const char *GetBaseFile() const
  84     {return fBaseFileName.Data();}
  85    virtual  Int_t GetEvent(Int_t event);
  86    virtual  void  SetEvent(Int_t event) {fEvent=event;}
  87    virtual  void  SetConfigFunction(const char * config="Config();");
  88    virtual  const char *GetConfigFunction() const
  89     {return fConfigFunction.Data();}
  90    TGeometry     *GetGeometry();
  91    AliHeader*     GetHeader() const {return fHeader;}
  92    virtual  void  SetGenEventHeader(AliGenEventHeader* header);
  93    Int_t          GetNtrack() const;
  94    virtual  Int_t GetPrimary(Int_t track) const;
  95    virtual  void  Hits2Digits(const char *detector=0);
  96    virtual  void  Hits2SDigits(const char *detector=0)   {Tree2Tree("S",detector);}
  97    virtual  void  SDigits2Digits(const char *detector=0) {Tree2Tree("D",detector);}
  98    virtual  void  Digits2Reco(const char *detector=0)    {Tree2Tree("R",detector);}
  99    virtual  void  InitMC(const char *setup="Config.C");
 100    virtual  void  Init(const char *setup="Config.C") {InitMC(setup);}
 101    Bool_t         IsFolder() const {return kTRUE;}
 102    virtual AliLego* Lego() const {return fLego;}
 103    virtual  void  MakeTree(Option_t *option="KH", const char *file = 0);
 104    void           MakeTree(Option_t *option, TFile *file);
 105
 106    TObjArray     *Particles() const;
 107    TParticle     *Particle(Int_t i) const;
 108    virtual  void  ResetDigits();
 109    virtual  void  ResetSDigits();
 110    virtual  void  ResetHits();
 111    virtual  void  ResetTrackReferences();
 112    virtual  void  ResetPoints();
 113    virtual  void  SetTransPar(const char *filename="$(ALICE_ROOT)/data/galice.cuts");
 114    virtual  void  SetBaseFile(const char *filename="galice.root");
 115    virtual  void  ReadTransPar();
 116    virtual  void  RunMC(Int_t nevent=1, const char *setup="Config.C");
 117    virtual  void  Run(Int_t nevent=1, const char *setup="Config.C")
 118   {RunMC(nevent,setup);}
 119    virtual  void  RunLego(const char *setup="Config.C",Int_t nc1=60,Float_t c1min=2,Float_t c1max=178,
 120                           Int_t nc2=60,Float_t c2min=0,Float_t c2max=360,Float_t rmin=0,
 121                           Float_t rmax=430,Float_t zmax=10000, AliLegoGenerator* gener=NULL);
 122    virtual  Bool_t IsLegoRun() const {return (fLego!=0);}
 123    virtual  void  RunReco(const char *detector=0, Int_t first = 0, Int_t last = 0);
 124    virtual  void  SetCurrentTrack(Int_t track);
 125    virtual  void  SetDebug(const Int_t level=0) {fDebug = level;}
 126    virtual  void  SetDisplay(AliDisplay *display) {fDisplay = display;}
 127    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");
 128    virtual  void  SetField(AliMagF* magField);
 129    virtual  void  SetTrack(Int_t done, Int_t parent, Int_t pdg,
 130                            Float_t *pmom, Float_t *vpos, Float_t *polar,
 131                            Float_t tof, TMCProcess mech, Int_t &ntr,
 132                            Float_t weight = 1, Int_t is = 0);
 133    virtual  void  SetTrack(Int_t done, Int_t parent, Int_t pdg,
 134                            Double_t px, Double_t py, Double_t pz, Double_t e,
 135                            Double_t vx, Double_t vy, Double_t vz, Double_t tof,
 136                            Double_t polx, Double_t poly, Double_t polz,
 137                            TMCProcess mech, Int_t &ntr, Float_t weight=1,
 138                            Int_t is = 0);
 139    virtual  void  SetHighWaterMark(const Int_t nt);
 140
 141    virtual  void  KeepTrack(const Int_t itra);
 142    virtual  void  MediaTable();
 143    virtual  void    TrackingLimits( Float_t rmax=1.e10, Float_t zmax=1.e10) {fTrRmax=rmax; fTrZmax=zmax;}
 144    virtual  Int_t   DetFromMate(Int_t i) const { return (*fImedia)[i];}
 145    virtual  AliGenerator* Generator() const {return fGenerator;}
 146    virtual  void SetGenerator(AliGenerator *generator);
 147    virtual  void ResetGenerator(AliGenerator *generator);
 148    virtual  void EnergySummary();
 149    virtual  TDatabasePDG* PDGDB() const {return fPDGDB;}
 150    virtual  void Announce() const;
 151
 152    // MC Application
 153    //
 154    virtual  void  ConstructGeometry();
 155    virtual  void  InitGeometry();
 156    virtual  void  GeneratePrimaries();
 157    virtual  void  BeginEvent();
 158    virtual  void  BeginPrimary();
 159    virtual  void  PreTrack();
 160    virtual  void  Stepping();
 161    virtual  void  PostTrack();
 162    virtual  void  FinishPrimary();
 163    virtual  void  FinishEvent();
 164    virtual  Double_t  TrackingZmax() const {return fTrZmax;}
 165    virtual  Double_t  TrackingRmax() const {return fTrRmax;}
 166    virtual  void Field(const Double_t* x, Double_t* b) const;
 167    //
 168    // End of MC Application
 169
 170    TFile* InitTreeFile(Option_t *option, TString fileName);
 171    void PrintTreeFile();
 172    void CloseTreeFile(Option_t *option);
 173    TString GetTreeDFileName() const {return fTreeDFileName;}
 174    TString GetTreeSFileName() const {return fTreeSFileName;}
 175    TString GetTreeRFileName() const {return fTreeRFileName;}
 176    void SetTreeDFileName(TString fileName){fTreeDFileName=fileName;}
 177    void SetTreeSFileName(TString fileName){fTreeSFileName=fileName;}
 178    void SetTreeRFileName(TString fileName){fTreeRFileName=fileName;}
 179    TFile* GetTreeDFile() const {return fTreeDFile;}
 180    TFile* GetTreeSFile() const {return fTreeSFile;}
 181    TFile* GetTreeRFile() const {return fTreeRFile;}
 182
 183
 184    TTree         *TreeD() const {return fTreeD;}
 185    TTree         *TreeS() const {return fTreeS;}
 186    TTree         *TreeE() const {return fTreeE;}
 187    TTree         *TreeH() const {return fTreeH;}
 188    TTree         *TreeTR() const {return fTreeTR;}
 189    TTree         *TreeK() const;
 190    TTree         *TreeR() const {return fTreeR;}
 191
 192    AliStack      *Stack() const {return fStack;}
 193
 194    static void Deprecated(TObject *obj, const char *method,
 195                           const char *replacement) {
 196      if (obj)
 197        ::Warning(Form("%s::%s", obj->ClassName(), method),
 198                  "method is depricated\nPlease use: %s", replacement);
 199      else
 200        ::Warning(method, "method is depricated\nPlease use: %s", replacement);
 201    }
 202
 203 protected:
 204   virtual  void  Tree2Tree(Option_t *option, const char *detector=0);
 205   TFile* InitFile(TString fileName="");
 206   Int_t          fRun;               //! Current run number
 207   Int_t          fEvent;             //! Current event number (from 1)
 208   Int_t          fEventNrInRun;      //! Current unique event number in run
 209   Int_t          fEventsPerRun;      //  Number of events per run
 210   Int_t          fDebug;             //  Debug flag
 211   AliHeader     *fHeader;            //  Header information
 212   TTree         *fTreeD;             //! Pointer to Tree for Digits
 213   TTree         *fTreeS;             //! Pointer to Tree for SDigits
 214   TTree         *fTreeH;             //! Pointer to Tree for Hits
 215   TTree         *fTreeTR;            //! Pointer to Tree for TrackRefernces
 216   TTree         *fTreeE;             //! Pointer to Tree for Header
 217   TTree         *fTreeR;             //! Pointer to Tree for Reconstructed Objects
 218   TObjArray     *fModules;           //  List of Detectors
 219   TGeometry     *fGeometry;          //  Pointer to geometry
 220   AliDisplay    *fDisplay;           //! Pointer to event display
 221   TStopwatch     fTimer;             //  Timer object
 222   AliMagF       *fField;             //  Magnetic Field Map
 223   TVirtualMC    *fMC;                //! Pointer to MonteCarlo object
 224   TArrayI       *fImedia;            //! Array of correspondence between media and detectors
 225   Int_t          fNdets;             //  Number of detectors
 226   Float_t        fTrRmax;            //  Maximum radius for tracking
 227   Float_t        fTrZmax;            //  Maximu z for tracking
 228   AliGenerator  *fGenerator;         //  Generator used in the MC
 229   Bool_t         fInitDone;          //! True when initialisation done
 230   AliLego       *fLego;              //! Pointer to aliLego object if it exists
 231   TDatabasePDG  *fPDGDB;             //  Particle factory object
 232   TList         *fHitLists;          //! Lists of hits to be remapped by PurifyKine
 233   TArrayF        fEventEnergy;       //! Energy deposit for current event
 234   TArrayF        fSummEnergy;        //! Energy per event in each volume
 235   TArrayF        fSum2Energy;        //! Energy squared per event in each volume
 236   TString        fConfigFunction;    //  Configuration file to be executed
 237   TRandom       *fRandom;            //  Pointer to the random number generator
 238   AliMCQA       *fMCQA;              //  Pointer to MC Quality assurance class
 239   TString        fTransParName;      //  Name of the transport parameters file
 240   TString        fBaseFileName;      //  Name of the base root file
 241   AliStack*      fStack;             //! Particle Stack
 242
 243   TString        fTreeDFileName;     //!  name of the file with TreeD
 244   TFile*         fTreeDFile;         //!  file with TreeD
 245   TString        fTreeSFileName;     //!  name of the file with TreeS
 246   TFile*         fTreeSFile;         //!  file with TreeS
 247   TString        fTreeRFileName;     //!  name of the file with TreeR
 248   TFile*         fTreeRFile;         //!  file with TreeR
 249
 250 private:
 251   void Copy(AliRun &arun) const;
 252
 253   ClassDef(AliRun,7)      //Supervisor class for all Alice detectors
 254 };
 255
 256 R__EXTERN  AliRun *gAlice;
 257
 258 #endif