3bd79107 |
1 | #ifndef ALIITS_H |
2 | #define ALIITS_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 | |
58005f18 |
8 | //////////////////////////////////////////////////////////////////////// |
3bd79107 |
9 | // Manager class for set: ITS // |
58005f18 |
10 | //////////////////////////////////////////////////////////////////////// |
fe4da5cc |
11 | |
3bd79107 |
12 | #include <TObjArray.h> |
13 | #include <TBranch.h> |
14 | #include <TTree.h> |
15 | #include <TFile.h> |
16 | #include <TObjectTable.h> |
17 | |
fe4da5cc |
18 | #include "AliDetector.h" |
58005f18 |
19 | #include "AliITSgeom.h" |
3bd79107 |
20 | #include "AliITSsegmentation.h" |
21 | #include "AliITSresponse.h" |
22 | #include "AliITSDetType.h" |
23 | #include "AliITShit.h" |
58005f18 |
24 | #include "AliITSdigit.h" |
25 | #include "AliITSmodule.h" |
3bd79107 |
26 | #include "AliITSRecPoint.h" |
27 | #include "AliITSRawCluster.h" |
fe4da5cc |
28 | |
c9a71be1 |
29 | |
3bd79107 |
30 | class AliITSsimulation; |
31 | class AliITSClusterFinder; |
32 | //class AliITStrack; |
33 | |
34 | class AliITS : public AliDetector { |
58005f18 |
35 | |
36 | public: |
3bd79107 |
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); |
58005f18 |
47 | // |
3bd79107 |
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 |
be33dccb |
107 | virtual void SetClasses(Int_t id, char* digit, char* cluster); |
3bd79107 |
108 | |
109 | |
110 | // Getters |
58005f18 |
111 | // ITS geometry functions |
3bd79107 |
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));} |
58005f18 |
116 | |
3bd79107 |
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 |
fe4da5cc |
171 | }; |
3bd79107 |
172 | |
fe4da5cc |
173 | #endif |