]> git.uio.no Git - u/mrichter/AliRoot.git/blob - STEER/AliModule.h
Code for simulation, sdigitization and digitization moved from macros to compiled...
[u/mrichter/AliRoot.git] / STEER / AliModule.h
1 #ifndef ALIMODULE_H
2 #define ALIMODULE_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6 /* $Id$ */
7
8 #include "Riostream.h"
9
10 #include "TAttLine.h"
11 #include "TAttMarker.h"
12 #include "TNamed.h"
13
14 #include "AliRndm.h"
15
16 class TClonesArray;
17 class TBrowser;
18 class TArrayI;
19 class TFile;
20 class TTree;
21 class AliLoader;
22 class AliTrackReference;
23 class AliDigitizer;
24 class AliRunDigitizer;
25
26
27 class AliModule : public TNamed , public TAttLine, public TAttMarker,
28                   public AliRndm {
29 public:
30
31   // Creators - distructors
32   AliModule(const char* name, const char *title);
33   AliModule();
34   AliModule(const AliModule &mod);
35   virtual ~AliModule();
36
37   // Inline functions
38   virtual  int           GetNdigits() const {return 0;}
39   virtual  int           GetNhits()  const {return 0;}
40   virtual  TArrayI      *GetIdtmed()   const {return fIdtmed;}
41   virtual  TList        *Histograms() const {return fHistograms;}
42   virtual  TList        *Nodes()  const {return fNodes;}
43   virtual  TClonesArray *Digits() const {return 0;}
44   virtual  TClonesArray *Hits()   const {return 0;}
45   virtual  TObjArray    *Points() const {return 0;}
46   virtual  Int_t         GetIshunt() const {return 0;}
47   virtual  void          SetIshunt(Int_t) {}
48   virtual  Bool_t        IsActive() const {return fActive;}
49   virtual  Bool_t        IsFolder() const {return kTRUE;}
50   virtual  Int_t&        LoMedium() {return fLoMedium;}
51   virtual  Int_t&        HiMedium() {return fHiMedium;}
52   virtual  Int_t         GetDebug() const {return fDebug;}
53   virtual  void          SetDebug(Int_t deb=0) {fDebug=deb;}
54
55   // Module composition
56   virtual void AliMaterial(Int_t imat, const char* name, Float_t a, 
57                            Float_t z, Float_t dens, Float_t radl,
58                            Float_t absl, Float_t *buf=0, Int_t nwbuf=0) const;
59   virtual void AliGetMaterial(Int_t imat, char* name, Float_t &a, 
60                               Float_t &z, Float_t &dens, Float_t &radl,
61                               Float_t &absl) const;
62   virtual void AliMixture(Int_t imat, const char *name, Float_t *a,
63                           Float_t *z, Float_t dens, Int_t nlmat,
64                           Float_t *wmat) const;
65   virtual void AliMedium(Int_t numed, const char *name, Int_t nmat,
66                           Int_t isvol, Int_t ifield, Float_t fieldm,
67                           Float_t tmaxfd, Float_t stemax, Float_t deemax,
68                           Float_t epsil, Float_t stmin, Float_t *ubuf=0,
69                           Int_t nbuf=0) const;
70   virtual void AliMatrix(Int_t &nmat, Float_t theta1, Float_t phi1,
71                           Float_t theta2, Float_t phi2, Float_t theta3,
72                           Float_t phi3) const;
73   
74   // Virtual methods
75   virtual void  BuildGeometry() {};
76   virtual Int_t IsVersion() const =0;
77   
78   
79   // Other methods
80   virtual void        AddDigit(Int_t*, Int_t*){
81   Error("AddDigit","Digits cannot be added to module %s\n",fName.Data());}
82   virtual void        AddHit(Int_t, Int_t*, Float_t *) {
83   Error("AddDigit","Hits cannot be added to module %s\n",fName.Data());}
84   virtual void        Hits2SDigits() {}
85   virtual AliDigitizer* CreateDigitizer(AliRunDigitizer* /*manager*/) 
86     {return NULL;}
87   virtual void        SDigits2Digits() {}
88   virtual void        Hits2Digits() {}
89   virtual void        Digits2Reco() {}
90   virtual void        Digits2Raw() {}
91   virtual void        Raw2Digits() {}
92   virtual void        Browse(TBrowser *) {}
93   virtual void        CreateGeometry() {}
94   virtual void        CreateMaterials() {}
95   virtual void        Disable();
96   virtual Int_t       DistancetoPrimitive(Int_t px, Int_t py) const;
97   virtual void        Enable();
98   virtual void        PreTrack(){}
99   virtual void        PostTrack(){}
100   virtual void        FinishEvent() {}
101   virtual void        FinishRun() {}
102   virtual void        FinishPrimary() {}
103   virtual void        Init() {}
104   virtual void        LoadPoints(Int_t ) {}
105
106
107
108   virtual void        MakeBranch(Option_t * /*opt =" "*/) {}
109   virtual void        MakeTree(Option_t *) {}//skowron 
110
111   virtual AliLoader*  MakeLoader(const char* topfoldername);//skowron   
112   virtual AliLoader*  GetLoader() const {return 0x0;} //skowron
113   
114
115   virtual void        Paint(Option_t *) {}
116   virtual void        ResetDigits() {}
117   virtual void        ResetSDigits() {}
118   virtual void        ResetHits() {}
119   virtual void        ResetPoints() {}
120   virtual void        SetTreeAddress();
121   virtual void        SetTimeGate(Float_t) {}
122   virtual Float_t     GetTimeGate() const {return 1.e10;}
123   virtual void        StepManager() {}
124   virtual void        DisableStepManager() {fEnable = kFALSE;}
125   virtual Bool_t      StepManagerIsEnabled() const {return fEnable;}
126   virtual void        SetBufferSize(Int_t) {}  
127   virtual Float_t     ZMin() const;
128   virtual Float_t     ZMax() const;
129   virtual void        SetEuclidFile(char *material,char *geometry=0);
130   virtual void        ReadEuclid(const char *filnam, char *topvol);
131   virtual void        ReadEuclidMedia(const char *filnam);
132 // Track reference related
133   TClonesArray *TrackReferences()   const {return fTrackReferences;}
134   virtual void        RemapTrackHitIDs(Int_t *) {}
135   virtual void        RemapTrackReferencesIDs(Int_t *map); //remaping track references MI
136   virtual void        ResetTrackReferences();
137   virtual void        AddTrackReference(Int_t label);
138   virtual  AliTrackReference * FirstTrackReference(Int_t track);
139   virtual  AliTrackReference * NextTrackReference();
140   virtual void        MakeBranchTR(Option_t *opt=" ");
141   TTree* TreeTR();  //shorcut method for accessing treeTR from folder
142   
143 //
144   AliModule& operator=(const AliModule &mod)
145     {mod.Copy(*this); return (*this);}
146  
147 protected:      
148   void Copy(AliModule &mod) const;
149
150   // Data members
151   
152   TString       fEuclidMaterial;  //!Name of the Euclid file for materials (if any)
153   TString       fEuclidGeometry;  //!Name of the Euclid file for geometry (if any)
154   
155   TArrayI      *fIdtmed;      //!List of tracking medium numbers
156   TArrayI      *fIdmate;      //!List of material numbers
157   Int_t         fLoMedium;   //!Minimum tracking medium ID for this Module
158   Int_t         fHiMedium;   //!Maximum tracking medium ID for this Module
159
160   Bool_t        fActive;      //Detector activity flag
161   TList        *fHistograms;  //List of histograms
162   TList        *fNodes;       //List of geometry nodes
163   Int_t         fDebug;       //Debug flag
164   Bool_t        fEnable;      //StepManager enabling flag
165   TClonesArray *fTrackReferences;     //!list of track references - for one primary track only -MI
166   Int_t         fMaxIterTrackRef;     //!for track refernce iterator routines
167   Int_t         fCurrentIterTrackRef; //!for track refernce iterator routines
168   ClassDef(AliModule,5)  //Base class for ALICE Modules
169 };
170 #endif