#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$ */ class TBrowser; class TList; class TTree; class TBranch; class TGeometry; class TDatabasePDG; class TRandom; class TParticle; class TFile; #include #include "TClonesArray.h" #include #include #include "TNamed.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 {kKeepBit=1, kDaughtersBit=2, kDoneBit=4}; class AliRun : public TNamed { public: // Creators - distructors AliRun(); AliRun(const char *name, const char *title); virtual ~AliRun(); 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);} virtual void Browse(TBrowser *b); virtual void Build(); virtual void BuildSimpleGeometry(); virtual void CleanDetectors(); TObjArray *Detectors() const {return fModules;} TObjArray *Modules() const {return fModules;} Int_t CurrentTrack() const; AliDisplay *Display() { return fDisplay;} virtual Int_t DistancetoPrimitive(Int_t px, Int_t py); 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 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;} 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 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); void MakeTree(Option_t *option, TFile *file); TObjArray *Particles(); TParticle *Particle(Int_t i); virtual void BeginEvent(); 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 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=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, 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, AliMCProcess 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;} virtual void SetGenerator(AliGenerator *generator); virtual void ResetGenerator(AliGenerator *generator); virtual void EnergySummary(); virtual TDatabasePDG* PDGDB() const {return fPDGDB;} TFile* InitTreeFile(Option_t *option, TString fileName); TString GetTreeDFileName() const {return fTreeDFileName;} TString GetTreeSFileName() const {return fTreeSFileName;} TString GetTreeRFileName() const {return fTreeRFileName;} void SetTreeDFileName(TString fileName){fTreeDFileName=fileName;} void SetTreeSFileName(TString fileName){fTreeSFileName=fileName;} void SetTreeRFileName(TString fileName){fTreeRFileName=fileName;} TFile* GetTreeDFile() const {return fTreeDFile;} TFile* GetTreeSFile() const {return fTreeSFile;} TFile* GetTreeRFile() const {return fTreeRFile;} void PrintTreeFile(); TTree *TreeD() {return fTreeD;} TTree *TreeS() {return fTreeS;} TTree *TreeE() {return fTreeE;} TTree *TreeH() {return fTreeH;} TTree *TreeTR() {return fTreeTR;} TTree *TreeK() ; TTree *TreeR() {return fTreeR;} AliStack *Stack() {return fStack;} protected: void CloseTreeFile(Option_t *option); virtual void Tree2Tree(Option_t *option, const char *detector=0); TFile* InitFile(TString fileName=""); Int_t fRun; //! Current run number Int_t fEvent; //! Current event number (from 1) 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 *fTreeH; //! Pointer to Tree for Hits TTree *fTreeTR; //! Pointer to Tree for TrackRefernces 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 TString fTreeDFileName; //! name of the file with TreeD TFile* fTreeDFile; //! file with TreeD TString fTreeSFileName; //! name of the file with TreeS TFile* fTreeSFile; //! file with TreeS TString fTreeRFileName; //! name of the file with TreeR TFile* fTreeRFile; //! file with TreeR private: AliRun(const AliRun &right) {} AliRun& operator = (const AliRun &) {return *this;} ClassDef(AliRun,6) //Supervisor class for all Alice detectors }; R__EXTERN AliRun *gAlice; #endif