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