]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - STEER/AliRun.h
Possibility to define the magnetic field in the reconstruction (Yu.Belikov)
[u/mrichter/AliRoot.git] / STEER / AliRun.h
index 27964858751519eb96b34f58e053855c4fccaf2e..e3807990b3ac650488c317869acf2550fea096e3 100644 (file)
@@ -8,11 +8,16 @@
 class TBrowser;
 class TList;
 class TTree;
+class TBranch;
 class TGeometry;
 class TDatabasePDG;
+class TRandom;
+class TParticle;
 #include <TArrayI.h>
+#include "TClonesArray.h"
 #include <TArrayF.h>
 #include <TStopwatch.h>
+#include "TNamed.h"
 
 class AliDetector;
 class AliModule;
@@ -20,8 +25,12 @@ class AliMagF;
 class AliMC;
 class AliLego;
 class AliDisplay;
+class AliLegoGenerator;
 #include "AliHeader.h"
-#include "AliGenerator.h"
+class AliGenerator;
+class AliLegoGenerator;
+#include "AliMCProcess.h"
+class AliMCQA;
 
 enum {kKeepBit=1, kDaughtersBit=2, kDoneBit=4};
 
@@ -49,7 +58,8 @@ public:
    virtual  void  DumpPart (Int_t i) const;
    virtual  void  DumpPStack () const;
    virtual AliMagF *Field() const {return fField;}
-   virtual  void  FillTree();
+   virtual  void  PreTrack();
+   virtual  void  PostTrack();
    virtual  void  FinishPrimary();
    virtual  void  FinishEvent();
    virtual  void  FinishRun();
@@ -62,7 +72,10 @@ public:
    Int_t          GetDebug() const {return fDebug;}
    AliModule     *GetModule(const char *name) const;
    AliDetector   *GetDetector(const char *name) const;
+   AliMCQA       *GetMCQA() const {return fMCQA;}
    Int_t          GetModuleID(const char *name) const;
+   virtual  const char *GetBaseFile() const 
+    {return fBaseFileName.Data();}
    virtual  Int_t GetEvent(Int_t event);
    virtual  void  SetEvent(Int_t event) {fEvent=event;}
    virtual  void  SetConfigFunction(const char * config="Config();");
@@ -74,38 +87,59 @@ public:
                               Float_t &e, Float_t *vpos, Float_t *polar, 
                               Float_t &tof);
    Int_t          GetNtrack() const {return fNtrack;}
-   virtual  Int_t GetPrimary(Int_t track) const;
+   virtual  Int_t GetPrimary(Int_t track);
+   virtual  void  Hits2Digits(const char *detector=0); 
+   virtual  void  Hits2SDigits(const char *detector=0)   {Tree2Tree("S",detector);}
+   virtual  void  SDigits2Digits(const char *detector=0) {Tree2Tree("D",detector);}
+   virtual  void  Digits2Reco(const char *detector=0)    {Tree2Tree("R",detector);}
+   virtual  void  Tree2Tree(Option_t *option, const char *detector=0);
    virtual  void  InitMC(const char *setup="Config.C");
    virtual  void  Init(const char *setup="Config.C") {InitMC(setup);}
    Bool_t         IsFolder() const {return kTRUE;}
    virtual AliLego* Lego() const {return fLego;}
-   virtual  void  MakeTree(Option_t *option="KH");
-   TClonesArray  *Particles() {return fParticles;};
+   virtual  void  MakeTree(Option_t *option="KH", char *file = 0);
+   virtual TBranch* MakeBranchInTree(TTree *tree, const char* cname, void* address, Int_t size=32000, char *file=0);
+   virtual TBranch* MakeBranchInTree(TTree *tree, const char* cname, const char* name, void* address, Int_t size=32000, Int_t splitlevel=1, char *file=0);
+
+   TObjArray     *Particles() {return fParticleMap;};
+   TParticle     *Particle(Int_t i);
    virtual  void  PurifyKine();
    virtual  Int_t PurifyKine(Int_t lastSavedTrack, Int_t nofTracks);
    virtual  void  BeginEvent();
    virtual  void  ResetDigits();
+   virtual  void  ResetSDigits();
    virtual  void  ResetHits();
    virtual  void  ResetPoints();
    virtual  void  SetTransPar(char *filename="$(ALICE_ROOT)/data/galice.cuts");
-   virtual  void  ResetStack() {fCurrent=-1;fHgwmk=0;fNtrack=0;fParticles->Clear();}
+   virtual  void  SetBaseFile(char *filename="galice.root");
+   virtual  void  ReadTransPar();
+   virtual  void  ResetStack() {fCurrent=-1;fHgwmk=fNtrack=fLoadPoint=0;fParticles->Clear();}
    virtual  void  RunMC(Int_t nevent=1, const char *setup="Config.C");
    virtual  void  Run(Int_t nevent=1, const char *setup="Config.C") 
   {RunMC(nevent,setup);}
-   virtual  void  RunLego(const char *setup="Config.C",Int_t ntheta=60,Float_t themin=2,Float_t themax=178,
-                         Int_t nphi=60,Float_t phimin=0,Float_t phimax=360,Float_t rmin=0,
-                         Float_t rmax=430,Float_t zmax=10000);
+   virtual  void  RunLego(const char *setup="Config.C",Int_t nc1=60,Float_t c1min=2,Float_t c1max=178,
+                         Int_t nc2=60,Float_t c2min=0,Float_t c2max=360,Float_t rmin=0,
+                         Float_t rmax=430,Float_t zmax=10000, AliLegoGenerator* gener=NULL);
    virtual  Bool_t IsLegoRun() const {return (fLego!=0);}
+   virtual  void  RunReco(const char *detector=0);
    virtual  void  SetCurrentTrack(Int_t track);                           
    virtual  void  SetDebug(const Int_t level=1) {fDebug = level;}
    virtual  void  SetDisplay(AliDisplay *display) {fDisplay = display;}
    virtual  void  StepManager(Int_t id);
    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");
+   virtual  void  SetField(AliMagF* magField);
    virtual  void  SetTrack(Int_t done, Int_t parent, Int_t pdg, 
                               Float_t *pmom, Float_t *vpos, Float_t *polar, 
-                               Float_t tof, const char *mecha, Int_t &ntr,
+                               Float_t tof, AliMCProcess mech, Int_t &ntr,
                                Float_t weight=1);
-  virtual  void  KeepTrack(const Int_t itra);
+   virtual  void  SetTrack(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,
+                     AliMCProcess mech, Int_t &ntr, Float_t weight=1);
+   virtual  void  SetHighWaterMark(const Int_t nt);
+   
+   virtual  void  KeepTrack(const Int_t itra);
    virtual  void  MediaTable();
    virtual  Float_t TrackingZmax() const {return fTrZmax;}
    virtual  Float_t TrackingRmax() const {return fTrRmax;}
@@ -119,51 +153,63 @@ public:
 
 
    TTree         *TreeD() {return fTreeD;}
+   TTree         *TreeS() {return fTreeS;}
    TTree         *TreeE() {return fTreeE;}
    TTree         *TreeH() {return fTreeH;}
    TTree         *TreeK() {return fTreeK;}
    TTree         *TreeR() {return fTreeR;}
 
 protected:
-  Int_t         fRun;          //Current run number
-  Int_t         fEvent;        //Current event number (from 1)
-  Int_t         fNtrack;       //Number of tracks
-  Int_t         fHgwmk;        //Last track purified
-  Int_t         fCurrent;      //Last track returned from the stack
-  Int_t         fDebug;        //Debug flag
-  AliHeader     fHeader;       //Header information
-  TTree        *fTreeD;        //Pointer to Tree for Digits
-  TTree        *fTreeK;        //Pointer to Tree for Kinematics
-  TTree        *fTreeH;        //Pointer to Tree for Hits
-  TTree        *fTreeE;        //Pointer to Tree for Header
-  TTree        *fTreeR;        //Pointer to Tree for Reconstructed Objects
-  TObjArray    *fModules;      //List of Detectors
-  TClonesArray *fParticles;    //Pointer to list of particles
-  TGeometry    *fGeometry;     //Pointer to geometry
-  AliDisplay   *fDisplay;      //Pointer to event display
-  TStopwatch    fTimer;        //Timer object
-  AliMagF      *fField;        //Magnetic Field Map
-  AliMC        *fMC;           //pointer to MonteCarlo object
-  TArrayI      *fImedia;       //Array of correspondence between media and detectors
-  Int_t         fNdets;        //Number of detectors
-  Float_t       fTrRmax;       //Maximum radius for tracking
-  Float_t       fTrZmax;       //Maximu z for tracking
-  AliGenerator *fGenerator;    //Generator used in the MC
-  Bool_t        fInitDone;     //true when initialisation done
-  AliLego      *fLego;         //pointer to aliLego object if it exists
-  TDatabasePDG *fPDGDB;        //Particle factory object!
-  TList        *fHitLists;     //Lists of hits to be remapped by PurifyKine
-  TArrayF       fEventEnergy;  //Energy deposit for current event
-  TArrayF       fSummEnergy;   //Energy per event in each volume
-  TArrayF       fSum2Energy;   //Energy squared per event in each volume
-  TString       fConfigFunction; //Configuration file to be executed
+  Int_t          fRun;               //! Current run number
+  Int_t          fEvent;             //! Current event number (from 1)
+  Int_t          fNtrack;            //  Number of tracks
+  Int_t          fHgwmk;             //! Last track purified
+  Int_t          fLoadPoint;         //! Next free position in the particle buffer
+  Int_t          fCurrent;           //! Last track returned from the stack
+  Int_t          fDebug;             //  Debug flag
+  AliHeader      fHeader;            //  Header information
+  TTree         *fTreeD;             //! Pointer to Tree for Digits
+  TTree         *fTreeS;             //! Pointer to Tree for SDigits
+  TTree         *fTreeK;             //! Pointer to Tree for Kinematics
+  TTree         *fTreeH;             //! Pointer to Tree for Hits
+  TTree         *fTreeE;             //! Pointer to Tree for Header
+  TTree         *fTreeR;             //! Pointer to Tree for Reconstructed Objects
+  TObjArray     *fModules;           //  List of Detectors
+  TClonesArray  *fParticles;         //! Pointer to list of particles
+  TGeometry     *fGeometry;          //  Pointer to geometry
+  AliDisplay    *fDisplay;           //! Pointer to event display
+  TStopwatch     fTimer;             //  Timer object
+  AliMagF       *fField;             //  Magnetic Field Map
+  AliMC         *fMC;                //! Pointer to MonteCarlo object
+  TArrayI       *fImedia;            //! Array of correspondence between media and detectors
+  Int_t          fNdets;             //  Number of detectors
+  Float_t        fTrRmax;            //  Maximum radius for tracking
+  Float_t        fTrZmax;            //  Maximu z for tracking
+  AliGenerator  *fGenerator;         //  Generator used in the MC
+  Bool_t         fInitDone;          //! True when initialisation done
+  AliLego       *fLego;              //! Pointer to aliLego object if it exists
+  TDatabasePDG  *fPDGDB;             //  Particle factory object
+  TList         *fHitLists;          //! Lists of hits to be remapped by PurifyKine
+  TArrayF        fEventEnergy;       //! Energy deposit for current event
+  TArrayF        fSummEnergy;        //! Energy per event in each volume
+  TArrayF        fSum2Energy;        //! Energy squared per event in each volume
+  TString        fConfigFunction;    //  Configuration file to be executed
+  TRandom       *fRandom;            //  Pointer to the random number generator
+  AliMCQA       *fMCQA;              //  Pointer to MC Quality assurance class
+  TString        fTransParName;      //  Name of the transport parameters file
+  TString        fBaseFileName;      //  Name of the base root file
+  TParticle     *fParticleBuffer;    //! Pointer to current particle for writing
+  TObjArray     *fParticleMap;       //! Map of particles in the supporting TClonesArray
+  TArrayI&       fParticleFileMap;   //! Map of particles in the file
 
 private:
 
-   AliRun(const AliRun &) {}
-   AliRun& operator = (const AliRun &) {return *this;}
+   AliRun(const AliRun &right) 
+     : fParticleFileMap(right.fParticleFileMap) {}  
+   AliRun& operator = (const AliRun &right) 
+      { fParticleFileMap= right.fParticleFileMap; return *this; }
 
-   ClassDef(AliRun,3)      //Supervisor class for all Alice detectors
+   ClassDef(AliRun,4)      //Supervisor class for all Alice detectors
 };
  
 R__EXTERN  AliRun *gAlice;