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 | |
143d1056 |
12 | #include <TObjArray.h> // used in inline function GetModule. |
cd77595e |
13 | #include <TBranch.h> // used in inline function SetHitsAddressBranch |
3bd79107 |
14 | |
9ad8b5dd |
15 | #include "AliRun.h" |
fe4da5cc |
16 | #include "AliDetector.h" |
54b8cf3b |
17 | #include "AliITSDetType.h" |
de82972f |
18 | |
d5da1ecf |
19 | class TString; |
20 | class TTree; |
55b5a0b3 |
21 | class TFile; |
fe4da5cc |
22 | |
54b8cf3b |
23 | //class AliITSDetType; |
3bd79107 |
24 | class AliITSsimulation; |
25 | class AliITSClusterFinder; |
de82972f |
26 | class AliITSsegmentation; |
27 | class AliITSresponse; |
28 | class AliITShit; |
29 | class AliITSgeom; |
2aea926d |
30 | class AliITSpListItem; |
de82972f |
31 | class AliITSdigit; |
32 | class AliITSRecPoint; |
33 | class AliITSRawCluster; |
34 | class AliITSmodule; |
27a043b1 |
35 | |
a8a6107b |
36 | const Int_t kNTYPES=3; |
de82972f |
37 | |
3bd79107 |
38 | class AliITS : public AliDetector { |
58005f18 |
39 | |
2aea926d |
40 | public: |
41 | //================= Standard Classes =============================== |
42 | AliITS(); // Default creator. |
43 | AliITS(const char *name, const char *title); // standard Creator |
44 | virtual ~AliITS(); // destructor |
45 | AliITS(AliITS &source); // copy constructor. Not to be used! |
46 | AliITS& operator=(AliITS &source); // = operator. Not to be used! |
47 | virtual Int_t IsVersion() const {return 1;} |
48 | virtual Int_t DistancetoPrimitive(Int_t px, Int_t py); |
49 | |
50 | //===================== Simulation Geometry ======================== |
e8189707 |
51 | // get geometry version - detailed (major) or coarse (minor) |
2aea926d |
52 | virtual Int_t GetMajorVersion(){return -1;} |
53 | virtual Int_t GetMinorVersion(){return -1;} |
54 | virtual void GetGeometryVersion(Int_t &a,Int_t &b) |
55 | {a = GetMajorVersion();b=GetMinorVersion();return;} |
56 | virtual void SetEUCLID(Bool_t euclid=1) {fEuclidOut = euclid;} |
57 | //-------------------- Geometry Transformations -------------------- |
58 | // ITS geometry functions |
59 | AliITSgeom *GetITSgeom() const {return fITSgeom;} |
f91c3e58 |
60 | // Sets ITS geometry ! be very careful using this function. |
61 | void SetITSgeom(AliITSgeom *geom) {fITSgeom = geom;} |
2aea926d |
62 | // return pointer to the array of modules |
63 | TObjArray *GetModules() const {return fITSmodules;} |
64 | // return pointer to a particular module |
65 | AliITSmodule *GetModule(Int_t index) {return (AliITSmodule *) |
66 | (fITSmodules->At(index));} |
3bd79107 |
67 | |
2aea926d |
68 | //================ Nessesary general Classes ======================= |
69 | virtual void Init(); |
70 | virtual void SetDefaults(); |
71 | virtual void SetDefaultSimulation(); |
72 | virtual void SetDefaultClusterFinders(); |
73 | virtual void MakeBranch(Option_t *opt=" ", const char *file=0); |
74 | virtual void SetTreeAddress(); |
cd77595e |
75 | // For a give branch from the treeH sets the TClonesArray address. |
9ad8b5dd |
76 | virtual void SetHitsAddressBranch(TBranch *b){b->SetAddress(&fHits);} |
2aea926d |
77 | // Return pointer to DetType #id |
78 | AliITSDetType *DetType(Int_t id); |
79 | //Int_t NDetTypes() {return fNDetTypes;} |
80 | //---------- Configuration Methods (per detector type) ------------- |
7d8046e8 |
81 | // Determines which ITS subdetectors will be processed. Effects |
82 | // digitization, and Reconstruction only. |
83 | void SetDetectors(Option_t *opt="All"){fOpt = opt;} |
84 | // Returns the list of ITS subdetectors that will be processed. |
85 | Option_t* GetDetectors(){return fOpt;} |
3bd79107 |
86 | // Set response |
2aea926d |
87 | virtual void SetResponseModel(Int_t id, AliITSresponse *response); |
3bd79107 |
88 | // Set segmentation |
2aea926d |
89 | virtual void SetSegmentationModel(Int_t id, AliITSsegmentation *seg); |
3bd79107 |
90 | // Set simulation - temporary |
2aea926d |
91 | virtual void SetSimulationModel(Int_t id, AliITSsimulation *sim); |
54b8cf3b |
92 | // Set simulation - temporary |
93 | virtual AliITSsimulation* GetSimulationModel(Int_t id){ |
94 | return ((AliITSDetType*)(fDetTypes->At(id)))->GetSimulationModel();} |
3bd79107 |
95 | // Set reconstruction |
2aea926d |
96 | virtual void SetReconstructionModel(Int_t id, AliITSClusterFinder *rec); |
3bd79107 |
97 | // Set class names for digit and rec point |
2aea926d |
98 | virtual void SetClasses(Int_t id, const char *digit, const char *cluster); |
99 | |
100 | //=================== Hits ========================================= |
101 | virtual void StepManager() {} // See Step Manager for specific geometry. |
102 | virtual void AddHit(Int_t track, Int_t *vol, Float_t *hits); |
103 | //------------ sort hits by module for Digitisation ---------------- |
104 | virtual void InitModules(Int_t size,Int_t &nmodules); |
105 | virtual void FillModules(Int_t evnt,Int_t bgrev,Int_t nmodules, |
106 | Option_t *opt,Text_t *filename); |
107 | virtual void ClearModules(); |
108 | |
109 | //===================== Digitisation =============================== |
110 | void MakeBranchS(const char *file); |
111 | void SetTreeAddressS(TTree *treeS); |
f91c3e58 |
112 | TClonesArray * GetSDigits() { return fSDigits; } |
9ad8b5dd |
113 | void MakeBranchInTreeD(TTree *treeD,const char *file=0); |
114 | void MakeBranchD(const char *file){ |
115 | MakeBranchInTreeD(gAlice->TreeD(),file);} |
2aea926d |
116 | void SetTreeAddressD(TTree *treeD); |
117 | void Hits2SDigits(); // Turn hits into SDigits |
118 | void Hits2PreDigits(); // Turn hits into SDigits |
e0fc0305 |
119 | void SDigits2Digits(){SDigitsToDigits("All");} // Turn SDigits to Digits |
120 | void SDigitsToDigits(Option_t *opt="All"); // Turn SDigits to Digits |
2aea926d |
121 | void Hits2Digits(); // Turn hits straight into Digits. |
122 | //------------------ Internal functions ---------------------------- |
123 | // Standard Hits To SDigits function |
124 | void HitsToSDigits(Int_t evNumber,Int_t bgrev,Int_t size, |
125 | Option_t *add, Option_t *det, Text_t *filename); |
126 | // Standard Hits To SDigits function |
127 | void HitsToPreDigits(Int_t evNumber,Int_t bgrev,Int_t size, |
128 | Option_t *add, Option_t *det, Text_t *filename); |
129 | // Standard Hits To Digits function |
130 | void HitsToDigits(Int_t evNumber,Int_t bgrev,Int_t size, |
131 | Option_t *add, Option_t *det, Text_t *filename); |
132 | void ResetSDigits(); // Resets the Summable digits. |
133 | void ResetDigits(); // depending on how the |
134 | void ResetDigits(Int_t branch); // tree will be filled only |
135 | void AddSumDigit(AliITSpListItem &sdig); |
136 | void AddRealDigit(Int_t branch, Int_t *digits); |
137 | void AddSimDigit(Int_t branch, AliITSdigit *d); |
138 | void AddSimDigit(Int_t branch,Float_t phys,Int_t* digits, |
139 | Int_t* tracks,Int_t *hits,Float_t* trkcharges); |
140 | // Return pointers to digits |
141 | TObjArray *Dtype() {return fDtype;} |
142 | Int_t *Ndtype() {return fNdtype;} |
143 | TClonesArray *DigitsAddress(Int_t id) |
144 | {return ((TClonesArray *) (*fDtype)[id]);} |
3bd79107 |
145 | |
2aea926d |
146 | //===================== Raw Data IO ================================ |
147 | // Write digits into raw data format |
148 | virtual void Digits2RawData() {} |
149 | // Decode raw data and store digits |
150 | virtual void RawData2Digits() {} |
58005f18 |
151 | |
2aea926d |
152 | //==================== Clusterization ============================== |
153 | // create separate tree for clusters - declustering refining |
154 | void MakeTreeC(Option_t *option="C"); |
155 | void GetTreeC(Int_t event); |
156 | void AddCluster(Int_t branch, AliITSRawCluster *c); |
157 | void ResetClusters(); // one of the methods in |
158 | void ResetClusters(Int_t branch); // the pair will be kept |
159 | // Return pointer to the tree of clusters |
160 | TTree *TreeC() {return fTreeC;} |
3bd79107 |
161 | // Return pointers to clusters |
2aea926d |
162 | TObjArray *Ctype() {return fCtype;} |
163 | Int_t *Nctype() {return fNctype;} |
164 | TClonesArray *ClustersAddress(Int_t id) |
3bd79107 |
165 | {return ((TClonesArray *) (*fCtype)[id]);} |
166 | |
2aea926d |
167 | //=================== Reconstruction =============================== |
ff0e455e |
168 | void MakeBranchR(const char *file, Option_t *opt=" "); |
169 | void MakeBranchRF(const char *file){MakeBranchR(file,"Fast");} |
2aea926d |
170 | void SetTreeAddressR(TTree *treeR); |
171 | void AddRecPoint(const AliITSRecPoint &p); |
172 | void HitsToFastRecPoints(Int_t evNumber,Int_t bgrev,Int_t size, |
173 | Option_t *add, Option_t *det, Text_t *filename); |
174 | void Digits2Reco(); |
175 | void DigitsToRecPoints(Int_t evNumber,Int_t lastEntry,Option_t *det); |
176 | void ResetRecPoints(); |
3bd79107 |
177 | // Return pointer to rec points |
178 | TClonesArray *RecPoints() {return fRecPoints;} |
3bd79107 |
179 | |
2aea926d |
180 | protected: |
181 | //================== Data Members ================================== |
182 | AliITSgeom *fITSgeom; // Pointer to ITS geometry |
e0fc0305 |
183 | Bool_t fEuclidOut; // Flag to write geometry in euclid format |
2aea926d |
184 | TObjArray *fITSmodules; //! Pointer to ITS modules |
7d8046e8 |
185 | Option_t *fOpt; //! Detector option ="All" unless changed. |
3bd79107 |
186 | |
e0fc0305 |
187 | Int_t fIdN; // the number of layers |
2aea926d |
188 | Int_t *fIdSens; //[fIdN] layer identifier |
189 | TString *fIdName; //[fIdN] layer identifier |
3bd79107 |
190 | |
e0fc0305 |
191 | Int_t fNDetTypes; // Number of detector types |
2aea926d |
192 | TObjArray *fDetTypes; // List of detector types |
e8189707 |
193 | |
e0fc0305 |
194 | TClonesArray *fSDigits; // List of Summable digits. |
2aea926d |
195 | Int_t fNSDigits; // Number of Summable Digits. |
55b5a0b3 |
196 | |
2aea926d |
197 | TObjArray *fDtype; // List of digits |
198 | Int_t *fNdtype; //[fNDetTypes] Num. of digits per type of det. |
3bd79107 |
199 | |
2aea926d |
200 | TObjArray *fCtype; // List of clusters |
201 | Int_t *fNctype; //[fNDetTypes] Num. of clust. per type of det. |
ed9afcd2 |
202 | TTree *fTreeC; //! Tree for raw clusters |
3bd79107 |
203 | |
2aea926d |
204 | TClonesArray *fRecPoints; // List of reconstructed points |
205 | Int_t fNRecPoints; // Number of rec points |
3bd79107 |
206 | |
ed9afcd2 |
207 | ClassDef(AliITS,2) // Base class for ITS |
fe4da5cc |
208 | }; |
3bd79107 |
209 | |
fe4da5cc |
210 | #endif |