[u/mrichter/AliRoot.git] / STEER / 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$ */

class TBrowser;
class TList;
class TTree;
class TGeometry;
class TDatabasePDG;
#include <TArrayI.h>
#include <TArrayF.h>
#include <TStopwatch.h>

class AliDetector;
class AliModule;
class AliMagF;
class AliMC;
class AliLego;
class AliDisplay;
#include "AliHeader.h"
#include "AliGenerator.h"

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();
   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);
   virtual  void  DumpPart (Int_t i) const;
   virtual  void  DumpPStack () const;
   virtual AliMagF *Field() const {return fField;}
   virtual  void  FillTree();
   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;}
   Int_t          GetRunNumber() const {return fRun;}
   void           SetRunNumber(Int_t run) {fRun=run;}
   Int_t          GetDebug() const {return fDebug;}
   AliModule     *GetModule(const char *name) const;
   AliDetector   *GetDetector(const char *name) const;
   Int_t          GetModuleID(const char *name) const;
   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  Int_t GetPrimary(Int_t track) const;
   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  PurifyKine();
   virtual  Int_t PurifyKine(Int_t lastSavedTrack, Int_t nofTracks);
   virtual  void  BeginEvent();
   virtual  void  ResetDigits();
   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  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  Bool_t IsLegoRun() const {return (fLego!=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  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 weight=1);
  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;}


   TTree         *TreeD() {return fTreeD;}
   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

private:

   AliRun(const AliRun &) {}
   AliRun& operator = (const AliRun &) {return *this;}

   ClassDef(AliRun,3)      //Supervisor class for all Alice detectors
};
 
R__EXTERN  AliRun *gAlice;
 
#endif
Commit	Line	Data
aee8290b	1	#ifndef ALIRUN_H
aee8290b	2	#define ALIRUN_H
3da30618	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	/* $Id$ */
	7
94de3818	8	class TBrowser;
	9	class TList;
	10	class TTree;
	11	class TGeometry;
	12	class TDatabasePDG;
	13	#include <TArrayI.h>
	14	#include <TArrayF.h>
fe4da5cc	15	#include <TStopwatch.h>
1578254f	16
94de3818	17	class AliDetector;
	18	class AliModule;
	19	class AliMagF;
	20	class AliMC;
	21	class AliLego;
	22	class AliDisplay;
fe4da5cc	23	#include "AliHeader.h"
fe4da5cc	24	#include "AliGenerator.h"
fe4da5cc	25
aee8290b	26	enum {kKeepBit=1, kDaughtersBit=2, kDoneBit=4};
1578254f	27
fe4da5cc	28
fe4da5cc	29	class AliRun : public TNamed {
fe4da5cc	30	public:
	31	// Creators - distructors
	32	AliRun();
	33	AliRun(const char name, const char title);
	34	virtual ~AliRun();
	35
	36	virtual void AddHit(Int_t id, Int_t track, Int_t vol, Float_t hits) const;
	37	virtual void AddDigit(Int_t id, Int_t tracks, Int_t digits) const;
1cedd08a	38	virtual void AddHitList(TCollection *hitList) {fHitLists->Add(hitList);}
fe4da5cc	39	virtual void Browse(TBrowser *b);
	40	virtual void Build();
	41	virtual void BuildSimpleGeometry();
	42	virtual void CleanDetectors();
	43	virtual void CleanParents();
8494b010	44	TObjArray *Detectors() const {return fModules;}
8494b010	45	TObjArray *Modules() const {return fModules;}
fe4da5cc	46	Int_t CurrentTrack() const {return fCurrent;}
	47	AliDisplay *Display() { return fDisplay;}
	48	virtual Int_t DistancetoPrimitive(Int_t px, Int_t py);
94de3818	49	virtual void DumpPart (Int_t i) const;
94de3818	50	virtual void DumpPStack () const;
fe4da5cc	51	virtual AliMagF *Field() const {return fField;}
	52	virtual void FillTree();
	53	virtual void FinishPrimary();
	54	virtual void FinishEvent();
	55	virtual void FinishRun();
	56	virtual void FlagTrack(Int_t track);
875c717b	57	void AddEnergyDeposit(Int_t id, Float_t edep)
875c717b	58	{fEventEnergy[id]+=edep;}
fe4da5cc	59	Int_t GetEvNumber() const {return fEvent;}
fe4da5cc	60	Int_t GetRunNumber() const {return fRun;}
8494b010	61	void SetRunNumber(Int_t run) {fRun=run;}
fe4da5cc	62	Int_t GetDebug() const {return fDebug;}
94de3818	63	AliModule GetModule(const char name) const;
	64	AliDetector GetDetector(const char name) const;
	65	Int_t GetModuleID(const char *name) const;
fe4da5cc	66	virtual Int_t GetEvent(Int_t event);
45189757	67	virtual void SetEvent(Int_t event) {fEvent=event;}
94de3818	68	virtual void SetConfigFunction(const char * config="Config();");
45189757	69	virtual const char *GetConfigFunction() const
45189757	70	{return fConfigFunction.Data();}
fe4da5cc	71	TGeometry *GetGeometry();
fe4da5cc	72	AliHeader *GetHeader() {return &fHeader;}
45189757	73	virtual void GetNextTrack(Int_t &mtrack, Int_t &ipart, Float_t *pmom,
	74	Float_t &e, Float_t vpos, Float_t polar,
	75	Float_t &tof);
94de3818	76	Int_t GetNtrack() const {return fNtrack;}
94de3818	77	virtual Int_t GetPrimary(Int_t track) const;
875c717b	78	virtual void InitMC(const char *setup="Config.C");
875c717b	79	virtual void Init(const char *setup="Config.C") {InitMC(setup);}
b21a3d1a	80	Bool_t IsFolder() const {return kTRUE;}
838edcaf	81	virtual AliLego* Lego() const {return fLego;}
fe4da5cc	82	virtual void MakeTree(Option_t *option="KH");
	83	TClonesArray *Particles() {return fParticles;};
	84	virtual void PurifyKine();
	85	virtual Int_t PurifyKine(Int_t lastSavedTrack, Int_t nofTracks);
dffd31ef	86	virtual void BeginEvent();
fe4da5cc	87	virtual void ResetDigits();
	88	virtual void ResetHits();
	89	virtual void ResetPoints();
	90	virtual void SetTransPar(char *filename="$(ALICE_ROOT)/data/galice.cuts");
	91	virtual void ResetStack() {fCurrent=-1;fHgwmk=0;fNtrack=0;fParticles->Clear();}
875c717b	92	virtual void RunMC(Int_t nevent=1, const char *setup="Config.C");
	93	virtual void Run(Int_t nevent=1, const char *setup="Config.C")
	94	{RunMC(nevent,setup);}
fe4da5cc	95	virtual void RunLego(const char *setup="Config.C",Int_t ntheta=60,Float_t themin=2,Float_t themax=178,
fe4da5cc	96	Int_t nphi=60,Float_t phimin=0,Float_t phimax=360,Float_t rmin=0,
838edcaf	97	Float_t rmax=430,Float_t zmax=10000);
dffd31ef	98	virtual Bool_t IsLegoRun() const {return (fLego!=0);}
fe4da5cc	99	virtual void SetCurrentTrack(Int_t track);
	100	virtual void SetDebug(const Int_t level=1) {fDebug = level;}
	101	virtual void SetDisplay(AliDisplay *display) {fDisplay = display;}
875c717b	102	virtual void StepManager(Int_t id);
fe4da5cc	103	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");
1578254f	104	virtual void SetTrack(Int_t done, Int_t parent, Int_t pdg,
fe4da5cc	105	Float_t pmom, Float_t vpos, Float_t *polar,
	106	Float_t tof, const char *mecha, Int_t &ntr,
	107	Float_t weight=1);
aee8290b	108	virtual void KeepTrack(const Int_t itra);
fe4da5cc	109	virtual void MediaTable();
	110	virtual Float_t TrackingZmax() const {return fTrZmax;}
	111	virtual Float_t TrackingRmax() const {return fTrRmax;}
	112	virtual void TrackingLimits( Float_t rmax=1.e10, Float_t zmax=1.e10) {fTrRmax=rmax; fTrZmax=zmax;}
94de3818	113	virtual Int_t DetFromMate(Int_t i) const { return (*fImedia)[i];}
94de3818	114	virtual AliGenerator* Generator() const {return fGenerator;}
fe4da5cc	115	virtual void SetGenerator(AliGenerator *generator);
ee1dd322	116	virtual void ResetGenerator(AliGenerator *generator);
fe4da5cc	117	virtual void EnergySummary();
94de3818	118	virtual TDatabasePDG* PDGDB() const {return fPDGDB;}
fe4da5cc	119
875c717b	120
fe4da5cc	121	TTree *TreeD() {return fTreeD;}
	122	TTree *TreeE() {return fTreeE;}
	123	TTree *TreeH() {return fTreeH;}
	124	TTree *TreeK() {return fTreeK;}
	125	TTree *TreeR() {return fTreeR;}
	126
aee8290b	127	protected:
	128	Int_t fRun; //Current run number
	129	Int_t fEvent; //Current event number (from 1)
	130	Int_t fNtrack; //Number of tracks
	131	Int_t fHgwmk; //Last track purified
	132	Int_t fCurrent; //Last track returned from the stack
	133	Int_t fDebug; //Debug flag
	134	AliHeader fHeader; //Header information
	135	TTree *fTreeD; //Pointer to Tree for Digits
	136	TTree *fTreeK; //Pointer to Tree for Kinematics
	137	TTree *fTreeH; //Pointer to Tree for Hits
	138	TTree *fTreeE; //Pointer to Tree for Header
	139	TTree *fTreeR; //Pointer to Tree for Reconstructed Objects
	140	TObjArray *fModules; //List of Detectors
	141	TClonesArray *fParticles; //Pointer to list of particles
	142	TGeometry *fGeometry; //Pointer to geometry
	143	AliDisplay *fDisplay; //Pointer to event display
	144	TStopwatch fTimer; //Timer object
	145	AliMagF *fField; //Magnetic Field Map
	146	AliMC *fMC; //pointer to MonteCarlo object
	147	TArrayI *fImedia; //Array of correspondence between media and detectors
	148	Int_t fNdets; //Number of detectors
	149	Float_t fTrRmax; //Maximum radius for tracking
	150	Float_t fTrZmax; //Maximu z for tracking
	151	AliGenerator *fGenerator; //Generator used in the MC
	152	Bool_t fInitDone; //true when initialisation done
	153	AliLego *fLego; //pointer to aliLego object if it exists
	154	TDatabasePDG *fPDGDB; //Particle factory object!
	155	TList *fHitLists; //Lists of hits to be remapped by PurifyKine
	156	TArrayF fEventEnergy; //Energy deposit for current event
	157	TArrayF fSummEnergy; //Energy per event in each volume
	158	TArrayF fSum2Energy; //Energy squared per event in each volume
	159	TString fConfigFunction; //Configuration file to be executed
fe4da5cc	160
ef42d733	161	private:
	162
	163	AliRun(const AliRun &) {}
	164	AliRun& operator = (const AliRun &) {return *this;}
	165
45189757	166	ClassDef(AliRun,3) //Supervisor class for all Alice detectors
fe4da5cc	167	};
fe4da5cc	168
75a25b1b	169	R__EXTERN AliRun *gAlice;
fe4da5cc	170
fe4da5cc	171	#endif