Problems with the HP compiler fixed
[u/mrichter/AliRoot.git] / ITS / AliITS.h
1 #ifndef ALIITS_H
2 #define ALIITS_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 ////////////////////////////////////////////////////////////////////////
9 //           Manager class for set: ITS                               //
10 ////////////////////////////////////////////////////////////////////////
11
12 #include <TObjArray.h>
13 #include <TBranch.h>
14 #include <TTree.h>
15 #include <TFile.h>
16 #include <TObjectTable.h>
17
18 #include "AliDetector.h"
19 #include "AliITSgeom.h"
20 #include "AliITSsegmentation.h"
21 #include "AliITSresponse.h"
22 #include "AliITSDetType.h"
23 #include "AliITShit.h"
24 #include "AliITSdigit.h"
25 #include "AliITSmodule.h"
26 #include "AliITSRecPoint.h"
27 #include "AliITSRawCluster.h"
28
29
30 class AliITSsimulation;
31 class AliITSClusterFinder;
32 //class AliITStrack;
33
34 class AliITS : public AliDetector {
35
36   public:
37                     AliITS();
38                     AliITS(const char *name, const char *title);
39     virtual        ~AliITS();
40     AliITS(AliITS &source);
41     AliITS & operator=(AliITS &source);
42
43     virtual void   AddHit(Int_t track, Int_t *vol, Float_t *hits);
44     virtual void   AddRealDigit(Int_t branch, Int_t *digits);
45     virtual void   AddDigit(Int_t branch, AliITSdigit *d);
46     virtual void   AddDigit(Int_t branch,Float_t phys,Int_t* digits, Int_t* tracks, Float_t* trkcharges); 
47     //
48     virtual void   AddCluster(Int_t branch, AliITSRawCluster *c);
49     virtual void   AddRecPoint(const AliITSRecPoint &p);
50     virtual void   AddTrack(Float_t* tracks) {}
51 //    virtual void   AddTrack(const AliITStrack&) {} 
52
53     virtual void   ResetDigits();                   // depending on how the
54     virtual void   ResetDigits(Int_t branch);       // tree will be filled only
55     virtual void   ResetClusters();                 // one of the methods in 
56     virtual void   ResetClusters(Int_t branch);     // the pair will be kept
57     virtual void   ResetRecPoints();
58
59     virtual void   BuildGeometry() {}
60     virtual void   CreateGeometry() {}
61     virtual void   CreateMaterials() {}
62
63     void GetGeometryVersion(Int_t &a,Int_t &b) const 
64                    {a = fMajorVersion;b=fMinorVersion;return;}
65     virtual Int_t  IsVersion() const {return 1;}
66             Int_t  DistancetoPrimitive(Int_t px, Int_t py);
67     virtual void   Init();
68     virtual void   SetDefaults();
69     // create separate tree for clusters - declustering refining
70     virtual  void  MakeTreeC(Option_t *option="C");
71     void           GetTreeC(Int_t event);
72     virtual void   MakeBranch(Option_t *opt=" ");
73     void           SetTreeAddress();
74     virtual void   SetEUCLID(Bool_t euclid=1) {fEuclidOut = euclid;}
75     virtual void   StepManager() {}
76     // sort hits by module
77     virtual void   InitModules(Int_t size,Int_t &nmodules);  
78     virtual void   FillModules(Int_t evnt,Int_t bgrev,Int_t lastev,
79                        Int_t nmodules,Option_t *opt,Text_t *filename);
80     virtual void   ClearModules();
81     // Digitisation
82     void HitsToDigits(Int_t evNumber,Int_t bgrev,Int_t lastev,Int_t size,
83                  Option_t *add, Option_t *det, Text_t *filename);
84     // Reconstruct hits
85     void DigitsToRecPoints(Int_t evNumber,Int_t lastEntry,Option_t *det);
86     // Fast simulation of space points from hits
87     void HitsToFastRecPoints(Int_t evNumber,Int_t bgrev,Int_t lastev,Int_t size,
88                  Option_t *add, Option_t *det, Text_t *filename);
89
90
91
92     // Write digits into raw data format
93     virtual void Digits2RawData() {}
94
95     // Configuration Methods (per detector type )
96     // Set number of detector types
97     virtual void   SetNDetTypes(Int_t ndtypes=3) {fNDetTypes=ndtypes;}
98     // Set response 
99     virtual void   SetResponseModel(Int_t id, AliITSresponse *response);
100     // Set segmentation 
101     virtual void   SetSegmentationModel(Int_t id, AliITSsegmentation *seg);
102     // Set simulation - temporary 
103     virtual void   SetSimulationModel(Int_t id, AliITSsimulation *sim);
104     // Set reconstruction 
105     virtual void   SetReconstructionModel(Int_t id, AliITSClusterFinder *rec);
106     // Set class names for digit and rec point 
107     virtual void   SetClasses(Int_t id, char* digit, char* cluster);
108
109
110     // Getters
111     // ITS geometry functions
112     virtual AliITSgeom  *GetITSgeom() const {return fITSgeom;}
113     virtual TObjArray   *GetModules() const {return fITSmodules;}
114     AliITSmodule *GetModule(Int_t index) {return (AliITSmodule *)
115                                               (fITSmodules->At(index));}
116
117     // Return pointers to digits 
118     TObjArray            *Dtype() {return fDtype;}
119     Int_t                *Ndtype() {return fNdtype;}
120     virtual TClonesArray *DigitsAddress(Int_t id) 
121                    {return ((TClonesArray *) (*fDtype)[id]);}
122     // Return pointers to clusters 
123     TObjArray            *Ctype() {return fCtype;}
124     Int_t                *Nctype() {return fNctype;}
125     virtual TClonesArray *ClustersAddress(Int_t id) 
126                    {return ((TClonesArray *) (*fCtype)[id]);}
127
128     // Return pointer to rec points 
129     TClonesArray  *RecPoints()   {return fRecPoints;}
130     // Return pointer to tracks 
131     TClonesArray  *Tracks()   {return fTracks;}
132
133     // Return pointer to DetType #id
134     AliITSDetType  *DetType(Int_t id);
135     Int_t           NDetTypes() {return fNDetTypes;}
136
137     // Return pointer to the tree of clusters
138     TTree          *TreeC() {return fTreeC;}
139
140  protected:
141
142     AliITSgeom  *fITSgeom;                   // Pointer to ITS geometry
143     TObjArray   *fITSmodules;                // Pointer to ITS modules
144
145     Bool_t fEuclidOut;                       // Flag to write out geometry 
146                                              //in euclid format
147     Int_t  fIdN;                             // the number of layers
148     Int_t  *fIdSens; char **fIdName;         //layer identifier
149     // Geometry and Stepmanager version numbers used.
150     Int_t fMajorVersion,fMinorVersion;       //detailed and coarse(minor) versions
151     //
152     //
153     TObjArray            *fDetTypes;           // List of detector types
154     Int_t                 fNDetTypes;          // Number of Detector types
155
156     TObjArray            *fDtype;              // List of digits
157     Int_t                *fNdtype;             // Number of digits per type of
158                                                // detector 
159     TObjArray            *fCtype;              // List of clusters
160     Int_t                *fNctype;             // Number of clusters per type
161                                                // of detector
162
163
164     TClonesArray         *fRecPoints;          // List of reconstructed points
165     Int_t                 fNRecPoints;         // Number of rec points
166     TClonesArray         *fTracks;             // List of reconstructed tracks
167     TTree                *fTreeC;              // Tree for raw clusters
168
169
170     ClassDef(AliITS,1) // Base class for ITS
171 };
172
173 #endif