]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - STEER/AliRun.h
Transition to NewIO
[u/mrichter/AliRoot.git] / STEER / AliRun.h
index 69b22581782dd8ed52f20430fcdf9fea0bf743d6..21e1c09d6dbba29a3ed6612cb0ec0a68dabcafaf 100644 (file)
@@ -5,42 +5,53 @@
 
 /* $Id$ */
 
+#include <TArrayF.h>
+#include <TArrayI.h>
+#include <TClonesArray.h>
+#include <TMCProcess.h>
+#include <TStopwatch.h>
+#include <TVirtualMCApplication.h>
+#include <TVirtualMC.h>
+#include <TError.h>
+
+class TBranch;
 class TBrowser;
-class TList;
-class TTree;
-class TGeometry;
 class TDatabasePDG;
-class TRandom;
+class TFile;
+class TGeometry;
+class TList;
 class TParticle;
-#include <TArrayI.h>
-#include "TClonesArray.h"
-#include <TArrayF.h>
-#include <TStopwatch.h>
-#include "TNamed.h"
+class TRandom;
+class TTree;
 
+#include "AliRunLoader.h"
 class AliDetector;
-class AliModule;
-class AliMagF;
-class AliMC;
-class AliLego;
 class AliDisplay;
-class AliLegoGenerator;
-#include "AliHeader.h"
+class AliGenEventHeader;
 class AliGenerator;
+class AliHeader;
+class AliLego;
+class AliLegoGenerator;
 class AliLegoGenerator;
-#include "AliMCProcess.h"
 class AliMCQA;
+class AliMagF;
+class AliModule;
+class AliStack;
+
 
 enum {kKeepBit=1, kDaughtersBit=2, kDoneBit=4};
 
 
-class AliRun : public TNamed {
+class AliRun : public TVirtualMCApplication {
 public:
    // Creators - distructors
    AliRun();
    AliRun(const char *name, const char *title);
+   AliRun(const AliRun &arun);
    virtual ~AliRun();
 
+   AliRun& operator = (const AliRun &arun) 
+     {arun.Copy(*this); return (*this);}
    virtual  void  AddHit(Int_t id, Int_t track, Int_t *vol, Float_t *hits) const;
    virtual  void  AddDigit(Int_t id, Int_t *tracks, Int_t *digits) const;
    virtual  void  AddHitList(TCollection *hitList) {fHitLists->Add(hitList);}
@@ -48,89 +59,87 @@ public:
    virtual  void  Build();
    virtual  void  BuildSimpleGeometry();
    virtual  void  CleanDetectors();
-   virtual  void  CleanParents();
    TObjArray     *Detectors() const {return fModules;}
    TObjArray     *Modules() const {return fModules;}
-   Int_t          CurrentTrack() const {return fCurrent;}
-   AliDisplay    *Display() { return fDisplay;}
-   virtual  Int_t DistancetoPrimitive(Int_t px, Int_t py);
+   Int_t          CurrentTrack() const;
+   AliDisplay    *Display() const { return fDisplay;}
+   virtual  Int_t DistancetoPrimitive(Int_t px, Int_t py) const;
    virtual  void  DumpPart (Int_t i) const;
    virtual  void  DumpPStack () const;
    virtual AliMagF *Field() const {return fField;}
-   virtual  void  PreTrack();
-   virtual  void  PostTrack();
-   virtual  void  FinishPrimary();
-   virtual  void  FinishEvent();
    virtual  void  FinishRun();
    virtual  void  FlagTrack(Int_t track);
    void           AddEnergyDeposit(Int_t id, Float_t edep) 
                                        {fEventEnergy[id]+=edep;}
-   Int_t          GetEvNumber() const {return fEvent;}
+   void           AddModule(AliModule* mod);
+   Int_t          GetEvNumber() const;
    Int_t          GetRunNumber() const {return fRun;}
    void           SetRunNumber(Int_t run) {fRun=run;}
+   void           SetEventNrInRun(Int_t event) {fEventNrInRun=event;}
+   Int_t          GetEventNrInRun() const {return fEventNrInRun;}
+   Int_t          GetEventsPerRun() const {return fEventsPerRun;}
    Int_t          GetDebug() const {return fDebug;}
    AliModule     *GetModule(const char *name) const;
+   TList*         GetHitLists() const {return fHitLists ;}
    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();");
    virtual  const char *GetConfigFunction() const 
     {return fConfigFunction.Data();}
    TGeometry     *GetGeometry();
-   AliHeader     *GetHeader() {return &fHeader;}
-   virtual  void  GetNextTrack(Int_t &mtrack, Int_t &ipart, Float_t *pmom,
-                              Float_t &e, Float_t *vpos, Float_t *polar, 
-                              Float_t &tof);
-   Int_t          GetNtrack() const {return fNtrack;}
+   virtual  void  SetGenEventHeader(AliGenEventHeader* header);
+   Int_t          GetNtrack() const;
    virtual  Int_t GetPrimary(Int_t track) const;
-   virtual  void  Hits2Digits(const char *detector=0);
-   virtual  void  Hits2SDigits(const char *detector=0);
-   virtual  void  SDigits2Digits(const char *detector=0);
+   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  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", char *file = 0);
-   virtual  void  MakeBranchInTree(TTree *tree, const char* cname, void* address, Int_t size=32000, char *file=0);
-   virtual  void  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();
+   TObjArray     *Particles() const;
+   TParticle     *Particle(Int_t i) const;
    virtual  void  ResetDigits();
    virtual  void  ResetSDigits();
    virtual  void  ResetHits();
+   virtual  void  ResetTrackReferences();
    virtual  void  ResetPoints();
-   virtual  void  SetTransPar(char *filename="$(ALICE_ROOT)/data/galice.cuts");
-   virtual  void  SetBaseFile(char *filename="galice.root");
-   virtual  void  OpenBaseFile(const char *option);
+   virtual  void  SetTransPar(const char *filename="$(ALICE_ROOT)/data/galice.cuts");
+   virtual  void  SetBaseFile(const 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  Run(Int_t nevent=1, const char *setup="Config.C") {RunMC(nevent,setup);}
    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);
+                          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, Int_t first = 0, Int_t last = 0);
    virtual  void  SetCurrentTrack(Int_t track);                           
-   virtual  void  SetDebug(const Int_t level=1) {fDebug = level;}
+   virtual  void  SetDebug(const Int_t level=0) {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, AliMCProcess mech, Int_t &ntr,
-                               Float_t weight=1);
-  virtual  void  KeepTrack(const Int_t itra);
+                          Float_t *pmom, Float_t *vpos, Float_t *polar, 
+                          Float_t tof, TMCProcess mech, Int_t &ntr,
+                          Float_t weight = 1, Int_t is = 0);
+   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,
+                          TMCProcess mech, Int_t &ntr, Float_t weight=1,
+                          Int_t is = 0);
+   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;}
    virtual  void    TrackingLimits( Float_t rmax=1.e10, Float_t zmax=1.e10) {fTrRmax=rmax; fTrZmax=zmax;}
    virtual  Int_t   DetFromMate(Int_t i) const { return (*fImedia)[i];}
    virtual  AliGenerator* Generator() const {return fGenerator;}
@@ -138,37 +147,63 @@ public:
    virtual  void ResetGenerator(AliGenerator *generator);
    virtual  void EnergySummary();
    virtual  TDatabasePDG* PDGDB() const {return fPDGDB;}
+   
+   // MC Application
+   //
+   virtual  void  ConstructGeometry();
+   virtual  void  InitGeometry();     
+   virtual  void  GeneratePrimaries();
+   virtual  void  BeginEvent();
+   virtual  void  BeginPrimary();
+   virtual  void  PreTrack();
+   virtual  void  Stepping();         
+   virtual  void  PostTrack();
+   virtual  void  FinishPrimary();
+   virtual  void  FinishEvent();
+   virtual  Double_t  TrackingZmax() const {return fTrZmax;}
+   virtual  Double_t  TrackingRmax() const {return fTrRmax;}
+   virtual  void Field(const Double_t* x, Double_t* b) const;
+
+   //
+   // End of MC Application
 
+   TTree         *TreeE() {return (fRunLoader)?fRunLoader->TreeE():0x0;}
+   TTree         *TreeK() {return (fRunLoader)?fRunLoader->TreeK():0x0;}
+   AliStack      *Stack() {return (fRunLoader)?fRunLoader->Stack():0x0;}
+   AliHeader*     GetHeader() {return (fRunLoader)?fRunLoader->GetHeader():0x0;}
 
-   TTree         *TreeD() {return fTreeD;}
-   TTree         *TreeS() {return fTreeS;}
-   TTree         *TreeE() {return fTreeE;}
-   TTree         *TreeH() {return fTreeH;}
-   TTree         *TreeK() {return fTreeK;}
-   TTree         *TreeR() {return fTreeR;}
+   TTree         *TreeD() {MayNotUse("TreeD"); return 0x0;}
+   TTree         *TreeS() {MayNotUse("TreeS"); return 0x0;}
+   TTree         *TreeR() {MayNotUse("TreeR"); return 0x0;}
 
+   
+   void SetRunLoader(AliRunLoader* rloader);
+   AliRunLoader* GetRunLoader() const {return fRunLoader;}
+//   void SetEventFolderName(const char* eventfoldername);
+  virtual  void Announce() const;
+   
+  virtual  void  InitLoaders(); //prepares run (i.e. creates getters)
+  static void Deprecated(TObject *obj, const char *method,
+                        const char *replacement) {
+    if (obj)
+      ::Warning(Form("%s::%s", obj->ClassName(), method),
+               "method is depricated\nPlease use: %s", replacement);
+    else
+      ::Warning(method, "method is depricated\nPlease use: %s", replacement);
+  }
 protected:
+  virtual  void  Tree2Tree(Option_t *option, const char *detector=0);
   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          fEventNrInRun;      //! Current unique event number in run
+  Int_t          fEventsPerRun;      //  Number of events per run
   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
+  TVirtualMC    *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
@@ -186,16 +221,12 @@ protected:
   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
-
+  
+  AliRunLoader  *fRunLoader;         //!run getter - written as a separate object
 private:
+  void Copy(AliRun &arun) const;
 
-   AliRun(const AliRun &) {}
-   AliRun& operator = (const AliRun &) {return *this;}
-
-   ClassDef(AliRun,4)      //Supervisor class for all Alice detectors
+  ClassDef(AliRun,8)      //Supervisor class for all Alice detectors
 };
  
 R__EXTERN  AliRun *gAlice;