04366a57 |
1 | /************************************************************************** |
2 | * Copyright(c) 1998-2003, ALICE Experiment at CERN, All rights reserved. * |
3 | * * |
4 | * Author: The ALICE Off-line Project. * |
5 | * Contributors are mentioned in the code where appropriate. * |
6 | * * |
7 | * Permission to use, copy, modify and distribute this software and its * |
8 | * documentation strictly for non-commercial purposes is hereby granted * |
9 | * without fee, provided that the above copyright notice appears in all * |
10 | * copies and that both the copyright notice and this permission notice * |
11 | * appear in the supporting documentation. The authors make no claims * |
12 | * about the suitability of this software for any purpose. It is * |
13 | * provided "as is" without express or implied warranty. * |
14 | **************************************************************************/ |
15 | //////////////////////////////////////////////////////////////////////////// |
16 | // Implementation of the ITS clusterer V2 class // |
17 | // // |
18 | // Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch // |
19 | // // |
20 | /////////////////////////////////////////////////////////////////////////// |
21 | |
22 | |
23 | #include "AliRun.h" |
24 | |
25 | #include "AliITSClusterFinderV2SDD.h" |
26 | #include "AliITSclusterV2.h" |
27 | #include "AliRawReader.h" |
28 | #include "AliITSRawStreamSDD.h" |
29 | |
30 | #include <TClonesArray.h> |
31 | #include "AliITS.h" |
32 | #include "AliITSgeom.h" |
33 | #include "AliITSdigitSDD.h" |
34 | |
35 | ClassImp(AliITSClusterFinderV2SDD) |
36 | |
37 | extern AliRun *gAlice; |
38 | |
39 | AliITSClusterFinderV2SDD::AliITSClusterFinderV2SDD():AliITSClusterFinderV2(){ |
40 | |
41 | //Default constructor |
42 | |
43 | fNySDD=256; fNzSDD=256; |
44 | fYpitchSDD=0.01825; |
45 | fZpitchSDD=0.02940; |
46 | fHwSDD=3.5085; fHlSDD=3.7632; |
47 | fYoffSDD=0.0425; |
48 | |
49 | |
50 | |
51 | } |
52 | |
53 | |
54 | void AliITSClusterFinderV2SDD::FindRawClusters(Int_t mod){ |
55 | |
56 | //Find clusters V2 |
57 | SetModule(mod); |
58 | FindClustersSDD(fDigits); |
59 | |
60 | } |
61 | |
62 | void AliITSClusterFinderV2SDD::FindClustersSDD(TClonesArray *digits) { |
63 | //------------------------------------------------------------ |
64 | // Actual SDD cluster finder |
65 | //------------------------------------------------------------ |
66 | Int_t kNzBins = fNzSDD + 2; |
67 | const Int_t kMAXBIN=kNzBins*(fNySDD+2); |
68 | |
69 | AliBin *bins[2]; |
70 | bins[0]=new AliBin[kMAXBIN]; |
71 | bins[1]=new AliBin[kMAXBIN]; |
72 | |
73 | AliITSdigitSDD *d=0; |
74 | Int_t i, ndigits=digits->GetEntriesFast(); |
75 | for (i=0; i<ndigits; i++) { |
76 | d=(AliITSdigitSDD*)digits->UncheckedAt(i); |
77 | Int_t y=d->GetCoord2()+1; //y |
78 | Int_t z=d->GetCoord1()+1; //z |
79 | Int_t q=d->GetSignal(); |
80 | if (q<3) continue; |
81 | |
82 | if (z <= fNzSDD) { |
83 | bins[0][y*kNzBins+z].SetQ(q); |
84 | bins[0][y*kNzBins+z].SetMask(1); |
85 | bins[0][y*kNzBins+z].SetIndex(i); |
86 | } else { |
87 | z-=fNzSDD; |
88 | bins[1][y*kNzBins+z].SetQ(q); |
89 | bins[1][y*kNzBins+z].SetMask(1); |
90 | bins[1][y*kNzBins+z].SetIndex(i); |
91 | } |
92 | } |
93 | |
94 | FindClustersSDD(bins, kMAXBIN, kNzBins, digits); |
95 | |
96 | delete[] bins[0]; |
97 | delete[] bins[1]; |
98 | |
99 | } |
100 | |
101 | void AliITSClusterFinderV2SDD:: |
102 | FindClustersSDD(AliBin* bins[2], Int_t nMaxBin, Int_t nzBins, |
103 | TClonesArray *digits, TClonesArray *clusters) { |
104 | //------------------------------------------------------------ |
105 | // Actual SDD cluster finder |
106 | //------------------------------------------------------------ |
107 | Int_t ncl=0; |
108 | TClonesArray &cl=*clusters; |
109 | for (Int_t s=0; s<2; s++) |
110 | for (Int_t i=0; i<nMaxBin; i++) { |
111 | if (bins[s][i].IsUsed()) continue; |
112 | Int_t idx[32]; UInt_t msk[32]; Int_t npeaks=0; |
113 | FindPeaks(i, nzBins, bins[s], idx, msk, npeaks); |
114 | |
115 | if (npeaks>30) continue; |
116 | if (npeaks==0) continue; |
117 | |
118 | Int_t k,l; |
119 | for (k=0; k<npeaks-1; k++){//mark adjacent peaks |
120 | if (idx[k] < 0) continue; //this peak is already removed |
121 | for (l=k+1; l<npeaks; l++) { |
122 | if (idx[l] < 0) continue; //this peak is already removed |
123 | Int_t ki=idx[k]/nzBins, kj=idx[k] - ki*nzBins; |
124 | Int_t li=idx[l]/nzBins, lj=idx[l] - li*nzBins; |
125 | Int_t di=TMath::Abs(ki - li); |
126 | Int_t dj=TMath::Abs(kj - lj); |
127 | if (di>1 || dj>1) continue; |
128 | if (bins[s][idx[k]].GetQ() > bins[s][idx[l]].GetQ()) { |
129 | msk[l]=msk[k]; |
130 | idx[l]*=-1; |
131 | } else { |
132 | msk[k]=msk[l]; |
133 | idx[k]*=-1; |
134 | break; |
135 | } |
136 | } |
137 | } |
138 | |
139 | for (k=0; k<npeaks; k++) { |
140 | MarkPeak(TMath::Abs(idx[k]), nzBins, bins[s], msk[k]); |
141 | } |
142 | |
143 | for (k=0; k<npeaks; k++) { |
144 | if (idx[k] < 0) continue; //removed peak |
145 | AliITSclusterV2 c; |
146 | MakeCluster(idx[k], nzBins, bins[s], msk[k], c); |
147 | //mi change |
148 | Int_t milab[10]; |
149 | for (Int_t ilab=0;ilab<10;ilab++){ |
150 | milab[ilab]=-2; |
151 | } |
152 | Int_t maxi=0,mini=0,maxj=0,minj=0; |
153 | //AliBin *bmax=&bins[s][idx[k]]; |
154 | //Float_t max = TMath::Max(TMath::Abs(bmax->GetQ())/5.,3.); |
155 | Float_t max=3; |
156 | for (Int_t di=-2; di<=2;di++) |
157 | for (Int_t dj=-3;dj<=3;dj++){ |
158 | Int_t index = idx[k]+di+dj*nzBins; |
159 | if (index<0) continue; |
160 | if (index>=nMaxBin) continue; |
161 | AliBin *b=&bins[s][index]; |
162 | if (TMath::Abs(b->GetQ())>max){ |
163 | if (di>maxi) maxi=di; |
164 | if (di<mini) mini=di; |
165 | if (dj>maxj) maxj=dj; |
166 | if (dj<minj) minj=dj; |
167 | // |
168 | if(digits) { |
169 | if (TMath::Abs(di)<2&&TMath::Abs(dj)<2){ |
170 | AliITSdigitSDD* d=(AliITSdigitSDD*)digits->UncheckedAt(b->GetIndex()); |
171 | for (Int_t itrack=0;itrack<10;itrack++){ |
172 | Int_t track = (d->GetTracks())[itrack]; |
173 | if (track>=0) { |
174 | AddLabel(milab, track); |
175 | } |
176 | } |
177 | } |
178 | } |
179 | } |
180 | } |
181 | |
182 | /* |
183 | Float_t s2 = c.GetSigmaY2()/c.GetQ() - c.GetY()*c.GetY(); |
184 | Float_t w=par->GetPadPitchWidth(sec); |
185 | c.SetSigmaY2(s2); |
186 | if (s2 != 0.) { |
187 | c.SetSigmaY2(c.GetSigmaY2()*0.108); |
188 | if (sec<par->GetNInnerSector()) c.SetSigmaY2(c.GetSigmaY2()*2.07); |
189 | } |
190 | s2 = c.GetSigmaZ2()/c.GetQ() - c.GetZ()*c.GetZ(); |
191 | w=par->GetZWidth(); |
192 | c.SetSigmaZ2(s2); |
193 | |
194 | if (s2 != 0.) { |
195 | c.SetSigmaZ2(c.GetSigmaZ2()*0.169); |
196 | if (sec<par->GetNInnerSector()) c.SetSigmaZ2(c.GetSigmaZ2()*1.77); |
197 | } |
198 | */ |
199 | |
200 | c.SetSigmaY2(0.0030*0.0030); |
201 | c.SetSigmaZ2(0.0020*0.0020); |
202 | c.SetDetectorIndex(fNdet[fModule]); |
203 | |
204 | Float_t y=c.GetY(),z=c.GetZ(), q=c.GetQ(); |
205 | y/=q; z/=q; |
206 | // |
207 | //Float_t s2 = c.GetSigmaY2()/c.GetQ() - y*y; |
208 | // c.SetSigmaY2(s2); |
209 | //s2 = c.GetSigmaZ2()/c.GetQ() - z*z; |
210 | //c.SetSigmaZ2(s2); |
211 | // |
212 | y=(y-0.5)*fYpitchSDD; |
213 | y-=fHwSDD; |
214 | y-=fYoffSDD; //delay ? |
215 | if (s) y=-y; |
216 | |
217 | z=(z-0.5)*fZpitchSDD; |
218 | z-=fHlSDD; |
219 | |
220 | y=-(-y+fYshift[fModule]); |
221 | z= -z+fZshift[fModule]; |
222 | c.SetY(y); |
223 | c.SetZ(z); |
224 | c.SetNy(maxj-minj+1); |
225 | c.SetNz(maxi-mini+1); |
226 | c.SetType(npeaks); |
227 | c.SetQ(q/12.7); //to be consistent with the SSD charges |
228 | |
229 | if (c.GetQ() < 20.) continue; //noise cluster |
230 | |
231 | if (digits) { |
232 | // AliBin *b=&bins[s][idx[k]]; |
233 | // AliITSdigitSDD* d=(AliITSdigitSDD*)digits->UncheckedAt(b->GetIndex()); |
234 | { |
235 | //Int_t lab[3]; |
236 | //lab[0]=(d->GetTracks())[0]; |
237 | //lab[1]=(d->GetTracks())[1]; |
238 | //lab[2]=(d->GetTracks())[2]; |
239 | //CheckLabels(lab); |
240 | CheckLabels2(milab); |
241 | c.SetLabel(milab[0],0); |
242 | c.SetLabel(milab[1],1); |
243 | c.SetLabel(milab[2],2); |
244 | c.SetLayer(fNlayer[fModule]); |
245 | } |
246 | } |
247 | if(clusters) new (cl[ncl]) AliITSclusterV2(c); |
248 | else { |
249 | fITS->AddClusterV2(c); |
250 | } |
251 | ncl++; |
252 | } |
253 | } |
254 | } |
255 | |
256 | |
257 | |
258 | void AliITSClusterFinderV2SDD::RawdataToClusters(AliRawReader* rawReader,TClonesArray** clusters){ |
259 | //------------------------------------------------------------ |
260 | // This function creates ITS clusters from raw data |
261 | //------------------------------------------------------------ |
262 | rawReader->Reset(); |
263 | AliITSRawStreamSDD inputSDD(rawReader); |
264 | FindClustersSDD(&inputSDD,clusters); |
265 | |
266 | } |
267 | |
268 | void AliITSClusterFinderV2SDD::FindClustersSDD(AliITSRawStream* input, |
269 | TClonesArray** clusters) |
270 | { |
271 | //------------------------------------------------------------ |
272 | // Actual SDD cluster finder for raw data |
273 | //------------------------------------------------------------ |
274 | Int_t nClustersSDD = 0; |
275 | Int_t kNzBins = fNzSDD + 2; |
276 | Int_t kMaxBin = kNzBins * (fNySDD+2); |
277 | AliBin* bins[2] = {NULL, NULL}; |
278 | |
279 | // read raw data input stream |
280 | while (kTRUE) { |
281 | Bool_t next = input->Next(); |
282 | if (!next || input->IsNewModule()) { |
283 | Int_t iModule = input->GetPrevModuleID(); |
284 | |
285 | // when all data from a module was read, search for clusters |
286 | if (bins[0]) { |
287 | clusters[iModule] = new TClonesArray("AliITSclusterV2"); |
288 | fModule = iModule; |
289 | FindClustersSDD(bins, kMaxBin, kNzBins, NULL, clusters[iModule]); |
290 | Int_t nClusters = clusters[iModule]->GetEntriesFast(); |
291 | nClustersSDD += nClusters; |
292 | delete[] bins[0]; |
293 | delete[] bins[1]; |
294 | } |
295 | |
296 | if (!next) break; |
297 | bins[0] = new AliBin[kMaxBin]; |
298 | bins[1] = new AliBin[kMaxBin]; |
299 | } |
300 | |
301 | // fill the current digit into the bins array |
302 | if(input->GetSignal()>=3) { |
303 | Int_t iz = input->GetCoord1()+1; |
304 | Int_t side = ((iz <= fNzSDD) ? 0 : 1); |
305 | iz -= side*fNzSDD; |
306 | Int_t index = (input->GetCoord2()+1) * kNzBins + iz; |
307 | bins[side][index].SetQ(input->GetSignal()); |
308 | bins[side][index].SetMask(1); |
309 | bins[side][index].SetIndex(index); |
310 | } |
311 | } |
312 | |
313 | Info("FindClustersSDD", "found clusters in ITS SDD: %d", nClustersSDD); |
314 | |
315 | } |
316 | |
317 | |