X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=STEER%2FAliRun.h;h=027097edbf6de85679ab5fd0f38f7b2c2e27f664;hb=7f6f9ab21e99b80c9d4e00a3b1811e8b80eff4f6;hp=a6b4cd984536a65e1ef7c11ef9a007ae7e6c5a9d;hpb=1cedd08ae5718ea98ac74257aa492d6b8956fba3;p=u%2Fmrichter%2FAliRoot.git diff --git a/STEER/AliRun.h b/STEER/AliRun.h index a6b4cd98453..027097edbf6 100644 --- a/STEER/AliRun.h +++ b/STEER/AliRun.h @@ -1,61 +1,44 @@ -#ifndef AliRun_H -#define AliRun_H +#ifndef ALIRUN_H +#define ALIRUN_H /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * * See cxx source for full Copyright notice */ /* $Id$ */ -#include -#include -#include +class TBrowser; +class TList; +class TTree; +class TBranch; +class TGeometry; +class TDatabasePDG; +class TRandom; +class TParticle; +#include +#include "TClonesArray.h" +#include #include -#include -#include -#include +#include "TNamed.h" -#include "AliDetector.h" -#include "AliHeader.h" -#include "AliMagF.h" -#include "AliMC.h" -#include "AliGenerator.h" -#include "AliLego.h" +class AliDetector; +class AliModule; +class AliMagF; +class AliMC; +class AliLego; +class AliDisplay; +class AliLegoGenerator; +class AliHeader; +class AliGenerator; +class AliLegoGenerator; +#include "AliMCProcess.h" +class AliMCQA; +class AliStack; +class AliGenEventHeader; -enum {Keep_Bit=1, Daughters_Bit=2, Done_Bit=4}; -class AliDisplay; +enum {kKeepBit=1, kDaughtersBit=2, kDoneBit=4}; -class AliRun : public TNamed { -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 - +class AliRun : public TNamed { public: // Creators - distructors AliRun(); @@ -69,84 +52,156 @@ 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;} + Int_t CurrentTrack() const; AliDisplay *Display() { return fDisplay;} virtual Int_t DistancetoPrimitive(Int_t px, Int_t py); - virtual void DumpPart (Int_t i); - virtual void DumpPStack (); + 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 BeginPrimary(); 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;} Int_t GetRunNumber() const {return fRun;} void SetRunNumber(Int_t run) {fRun=run;} Int_t GetDebug() const {return fDebug;} - AliModule *GetModule(const char *name); - AliDetector *GetDetector(const char *name); - Int_t GetModuleID(const char *name); + 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() {return fNtrack;} - virtual Int_t GetPrimary(Int_t track); - virtual void Init(const char *setup="Config.C"); - Bool_t IsFolder() {return kTRUE;} - virtual void MakeTree(Option_t *option="KH"); - TClonesArray *Particles() {return fParticles;}; - virtual void PurifyKine(); - virtual Int_t PurifyKine(Int_t lastSavedTrack, Int_t nofTracks); - virtual void Reset(Int_t run, Int_t idevent); + AliHeader* GetHeader() {return fHeader;} + virtual void SetGenEventHeader(AliGenEventHeader* header); + 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; + virtual Int_t GetPrimary(Int_t track) const; + 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", const char *file = 0); + + TObjArray *Particles(); + TParticle *Particle(Int_t i); + 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 Run(Int_t nevent=1, const char *setup="Config.C"); - 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=570,Float_t zmax=10000); + virtual void SetBaseFile(char *filename="galice.root"); + virtual void ReadTransPar(); + 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 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, 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) const; + 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); + 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;} virtual void TrackingLimits( Float_t rmax=1.e10, Float_t zmax=1.e10) {fTrRmax=rmax; fTrZmax=zmax;} - virtual Int_t DetFromMate(Int_t i) { return (*fImedia)[i];} - virtual AliGenerator* Generator() {return fGenerator;} + virtual Int_t DetFromMate(Int_t i) const { return (*fImedia)[i];} + virtual AliGenerator* Generator() const {return fGenerator;} virtual void SetGenerator(AliGenerator *generator); + virtual void ResetGenerator(AliGenerator *generator); virtual void EnergySummary(); - virtual const TDatabasePDG* PDGDB() const {return fPDGDB;} + virtual TDatabasePDG* PDGDB() const {return fPDGDB;} - // Functions from GEOCAD - //_______________________________________________________________________ - - virtual void ReadEuclid(const char*, const AliModule*, char*); - virtual void ReadEuclidMedia(const char*, const AliModule*); TTree *TreeD() {return fTreeD;} + TTree *TreeS() {return fTreeS;} TTree *TreeE() {return fTreeE;} TTree *TreeH() {return fTreeH;} - TTree *TreeK() {return fTreeK;} + TTree *TreeK() ; TTree *TreeR() {return fTreeR;} - // --------------------------- commons ------------------------------------- + AliStack *Stack() {return fStack;} + +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 fDebug; // Debug flag + AliHeader *fHeader; // Header information + TTree *fTreeD; //! Pointer to Tree for Digits + TTree *fTreeS; //! Pointer to Tree for SDigits + 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 + 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 + AliStack* fStack; // ! Particle Stack +private: - ClassDef(AliRun,2) //Supervisor class for all Alice detectors + AliRun(const AliRun &right) + {} + AliRun& operator = (const AliRun &) {return *this;} + ClassDef(AliRun,4) //Supervisor class for all Alice detectors }; R__EXTERN AliRun *gAlice;