e4f2f73d |
1 | /************************************************************************** |
29b87567 |
2 | * Copyright(c) 1998-1999, 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 | **************************************************************************/ |
e4f2f73d |
15 | |
16 | /* $Id$ */ |
17 | |
18 | //////////////////////////////////////////////////////////////////////////// |
19 | // // |
20 | // The TRD track seed // |
21 | // // |
22 | // Authors: // |
23 | // Alex Bercuci <A.Bercuci@gsi.de> // |
24 | // Markus Fasel <M.Fasel@gsi.de> // |
25 | // // |
26 | //////////////////////////////////////////////////////////////////////////// |
27 | |
28 | #include "TMath.h" |
29 | #include "TLinearFitter.h" |
eb38ed55 |
30 | #include "TClonesArray.h" // tmp |
31 | #include <TTreeStream.h> |
e4f2f73d |
32 | |
33 | #include "AliLog.h" |
34 | #include "AliMathBase.h" |
35 | |
03cef9b2 |
36 | #include "AliTRDpadPlane.h" |
e4f2f73d |
37 | #include "AliTRDcluster.h" |
f3d3af1b |
38 | #include "AliTRDseedV1.h" |
39 | #include "AliTRDtrackV1.h" |
e4f2f73d |
40 | #include "AliTRDcalibDB.h" |
eb38ed55 |
41 | #include "AliTRDchamberTimeBin.h" |
42 | #include "AliTRDtrackingChamber.h" |
43 | #include "AliTRDtrackerV1.h" |
44 | #include "AliTRDReconstructor.h" |
e4f2f73d |
45 | #include "AliTRDrecoParam.h" |
0906e73e |
46 | #include "Cal/AliTRDCalPID.h" |
e4f2f73d |
47 | |
e4f2f73d |
48 | ClassImp(AliTRDseedV1) |
49 | |
50 | //____________________________________________________________________ |
ae4e8b84 |
51 | AliTRDseedV1::AliTRDseedV1(Int_t det) |
e4f2f73d |
52 | :AliTRDseed() |
3a039a31 |
53 | ,fReconstructor(0x0) |
ae4e8b84 |
54 | ,fClusterIter(0x0) |
55 | ,fClusterIdx(0) |
56 | ,fDet(det) |
0906e73e |
57 | ,fMom(0.) |
bcb6fb78 |
58 | ,fSnp(0.) |
59 | ,fTgl(0.) |
60 | ,fdX(0.) |
e4f2f73d |
61 | { |
62 | // |
63 | // Constructor |
64 | // |
29b87567 |
65 | //printf("AliTRDseedV1::AliTRDseedV1()\n"); |
e44586fb |
66 | |
29b87567 |
67 | for(int islice=0; islice < knSlices; islice++) fdEdx[islice] = 0.; |
68 | for(int ispec=0; ispec<AliPID::kSPECIES; ispec++) fProb[ispec] = -1.; |
e4f2f73d |
69 | } |
70 | |
71 | //____________________________________________________________________ |
0906e73e |
72 | AliTRDseedV1::AliTRDseedV1(const AliTRDseedV1 &ref) |
e4f2f73d |
73 | :AliTRDseed((AliTRDseed&)ref) |
43d6ad34 |
74 | ,fReconstructor(ref.fReconstructor) |
ae4e8b84 |
75 | ,fClusterIter(0x0) |
76 | ,fClusterIdx(0) |
77 | ,fDet(ref.fDet) |
0906e73e |
78 | ,fMom(ref.fMom) |
bcb6fb78 |
79 | ,fSnp(ref.fSnp) |
80 | ,fTgl(ref.fTgl) |
81 | ,fdX(ref.fdX) |
e4f2f73d |
82 | { |
83 | // |
84 | // Copy Constructor performing a deep copy |
85 | // |
86 | |
29b87567 |
87 | //printf("AliTRDseedV1::AliTRDseedV1(const AliTRDseedV1 &)\n"); |
88 | SetBit(kOwner, kFALSE); |
89 | for(int islice=0; islice < knSlices; islice++) fdEdx[islice] = ref.fdEdx[islice]; |
90 | for(int ispec=0; ispec<AliPID::kSPECIES; ispec++) fProb[ispec] = ref.fProb[ispec]; |
fbb2ea06 |
91 | } |
d9950a5a |
92 | |
0906e73e |
93 | |
e4f2f73d |
94 | //____________________________________________________________________ |
95 | AliTRDseedV1& AliTRDseedV1::operator=(const AliTRDseedV1 &ref) |
96 | { |
97 | // |
98 | // Assignment Operator using the copy function |
99 | // |
100 | |
29b87567 |
101 | if(this != &ref){ |
102 | ref.Copy(*this); |
103 | } |
221ab7e0 |
104 | SetBit(kOwner, kFALSE); |
105 | |
29b87567 |
106 | return *this; |
e4f2f73d |
107 | |
108 | } |
109 | |
110 | //____________________________________________________________________ |
111 | AliTRDseedV1::~AliTRDseedV1() |
112 | { |
113 | // |
114 | // Destructor. The RecoParam object belongs to the underlying tracker. |
115 | // |
116 | |
29b87567 |
117 | //printf("I-AliTRDseedV1::~AliTRDseedV1() : Owner[%s]\n", IsOwner()?"YES":"NO"); |
e4f2f73d |
118 | |
29b87567 |
119 | if(IsOwner()) |
120 | for(int itb=0; itb<knTimebins; itb++){ |
121 | if(!fClusters[itb]) continue; |
122 | //AliInfo(Form("deleting c %p @ %d", fClusters[itb], itb)); |
123 | delete fClusters[itb]; |
124 | fClusters[itb] = 0x0; |
125 | } |
e4f2f73d |
126 | } |
127 | |
128 | //____________________________________________________________________ |
129 | void AliTRDseedV1::Copy(TObject &ref) const |
130 | { |
131 | // |
132 | // Copy function |
133 | // |
134 | |
29b87567 |
135 | //AliInfo(""); |
136 | AliTRDseedV1 &target = (AliTRDseedV1 &)ref; |
137 | |
ae4e8b84 |
138 | target.fClusterIter = 0x0; |
139 | target.fClusterIdx = 0; |
140 | target.fDet = fDet; |
29b87567 |
141 | target.fMom = fMom; |
142 | target.fSnp = fSnp; |
143 | target.fTgl = fTgl; |
144 | target.fdX = fdX; |
145 | target.fReconstructor = fReconstructor; |
146 | |
147 | for(int islice=0; islice < knSlices; islice++) target.fdEdx[islice] = fdEdx[islice]; |
148 | for(int ispec=0; ispec<AliPID::kSPECIES; ispec++) target.fProb[ispec] = fProb[ispec]; |
149 | |
150 | AliTRDseed::Copy(target); |
e4f2f73d |
151 | } |
152 | |
0906e73e |
153 | |
154 | //____________________________________________________________ |
f3d3af1b |
155 | Bool_t AliTRDseedV1::Init(AliTRDtrackV1 *track) |
0906e73e |
156 | { |
157 | // Initialize this tracklet using the track information |
158 | // |
159 | // Parameters: |
160 | // track - the TRD track used to initialize the tracklet |
161 | // |
162 | // Detailed description |
163 | // The function sets the starting point and direction of the |
164 | // tracklet according to the information from the TRD track. |
165 | // |
166 | // Caution |
167 | // The TRD track has to be propagated to the beginning of the |
168 | // chamber where the tracklet will be constructed |
169 | // |
170 | |
29b87567 |
171 | Double_t y, z; |
172 | if(!track->GetProlongation(fX0, y, z)) return kFALSE; |
173 | fYref[0] = y; |
174 | fYref[1] = track->GetSnp()/(1. - track->GetSnp()*track->GetSnp()); |
175 | fZref[0] = z; |
176 | fZref[1] = track->GetTgl(); |
0906e73e |
177 | |
29b87567 |
178 | //printf("Tracklet ref x[%7.3f] y[%7.3f] z[%7.3f], snp[%f] tgl[%f]\n", fX0, fYref[0], fZref[0], track->GetSnp(), track->GetTgl()); |
179 | return kTRUE; |
0906e73e |
180 | } |
181 | |
bcb6fb78 |
182 | |
183 | //____________________________________________________________________ |
184 | void AliTRDseedV1::CookdEdx(Int_t nslices) |
185 | { |
186 | // Calculates average dE/dx for all slices and store them in the internal array fdEdx. |
187 | // |
188 | // Parameters: |
189 | // nslices : number of slices for which dE/dx should be calculated |
190 | // Output: |
191 | // store results in the internal array fdEdx. This can be accessed with the method |
192 | // AliTRDseedV1::GetdEdx() |
193 | // |
194 | // Detailed description |
195 | // Calculates average dE/dx for all slices. Depending on the PID methode |
196 | // the number of slices can be 3 (LQ) or 8(NN). |
197 | // The calculation of dQ/dl are done using the tracklet fit results (see AliTRDseedV1::GetdQdl(Int_t)) i.e. |
198 | // |
199 | // dQ/dl = qc/(dx * sqrt(1 + dy/dx^2 + dz/dx^2)) |
200 | // |
201 | // The following effects are included in the calculation: |
202 | // 1. calibration values for t0 and vdrift (using x coordinate to calculate slice) |
203 | // 2. cluster sharing (optional see AliTRDrecoParam::SetClusterSharing()) |
204 | // 3. cluster size |
205 | // |
206 | |
29b87567 |
207 | Int_t nclusters[knSlices]; |
208 | for(int i=0; i<knSlices; i++){ |
209 | fdEdx[i] = 0.; |
210 | nclusters[i] = 0; |
211 | } |
212 | Float_t clength = (/*.5 * */AliTRDgeometry::AmThick() + AliTRDgeometry::DrThick()); |
213 | |
214 | AliTRDcluster *cluster = 0x0; |
215 | for(int ic=0; ic<AliTRDtrackerV1::GetNTimeBins(); ic++){ |
216 | if(!(cluster = fClusters[ic])) continue; |
217 | Float_t x = cluster->GetX(); |
218 | |
219 | // Filter clusters for dE/dx calculation |
220 | |
221 | // 1.consider calibration effects for slice determination |
222 | Int_t slice; |
223 | if(cluster->IsInChamber()) slice = Int_t(TMath::Abs(fX0 - x) * nslices / clength); |
224 | else slice = x < fX0 ? 0 : nslices-1; |
225 | |
226 | // 2. take sharing into account |
227 | Float_t w = cluster->IsShared() ? .5 : 1.; |
228 | |
229 | // 3. take into account large clusters TODO |
230 | //w *= c->GetNPads() > 3 ? .8 : 1.; |
231 | |
232 | //CHECK !!! |
233 | fdEdx[slice] += w * GetdQdl(ic); //fdQdl[ic]; |
234 | nclusters[slice]++; |
235 | } // End of loop over clusters |
236 | |
cd40b287 |
237 | //if(fReconstructor->GetPIDMethod() == AliTRDReconstructor::kLQPID){ |
238 | if(nslices == AliTRDReconstructor::kLQslices){ |
29b87567 |
239 | // calculate mean charge per slice (only LQ PID) |
240 | for(int is=0; is<nslices; is++){ |
241 | if(nclusters[is]) fdEdx[is] /= nclusters[is]; |
242 | } |
243 | } |
bcb6fb78 |
244 | } |
245 | |
b83573da |
246 | |
bcb6fb78 |
247 | //____________________________________________________________________ |
248 | Float_t AliTRDseedV1::GetdQdl(Int_t ic) const |
249 | { |
29b87567 |
250 | return fClusters[ic] ? TMath::Abs(fClusters[ic]->GetQ()) /fdX / TMath::Sqrt(1. + fYfit[1]*fYfit[1] + fZref[1]*fZref[1]) : 0.; |
bcb6fb78 |
251 | } |
252 | |
0906e73e |
253 | //____________________________________________________________________ |
254 | Double_t* AliTRDseedV1::GetProbability() |
255 | { |
256 | // Fill probability array for tracklet from the DB. |
257 | // |
258 | // Parameters |
259 | // |
260 | // Output |
261 | // returns pointer to the probability array and 0x0 if missing DB access |
262 | // |
263 | // Detailed description |
264 | |
29b87567 |
265 | |
266 | // retrive calibration db |
0906e73e |
267 | AliTRDcalibDB *calibration = AliTRDcalibDB::Instance(); |
268 | if (!calibration) { |
269 | AliError("No access to calibration data"); |
270 | return 0x0; |
271 | } |
272 | |
3a039a31 |
273 | if (!fReconstructor) { |
274 | AliError("Reconstructor not set."); |
4ba1d6ae |
275 | return 0x0; |
276 | } |
277 | |
0906e73e |
278 | // Retrieve the CDB container class with the parametric detector response |
3a039a31 |
279 | const AliTRDCalPID *pd = calibration->GetPIDObject(fReconstructor->GetPIDMethod()); |
0906e73e |
280 | if (!pd) { |
281 | AliError("No access to AliTRDCalPID object"); |
282 | return 0x0; |
283 | } |
29b87567 |
284 | //AliInfo(Form("Method[%d] : %s", fReconstructor->GetRecoParam() ->GetPIDMethod(), pd->IsA()->GetName())); |
10f75631 |
285 | |
29b87567 |
286 | // calculate tracklet length TO DO |
0906e73e |
287 | Float_t length = (AliTRDgeometry::AmThick() + AliTRDgeometry::DrThick()); |
288 | /// TMath::Sqrt((1.0 - fSnp[iPlane]*fSnp[iPlane]) / (1.0 + fTgl[iPlane]*fTgl[iPlane])); |
289 | |
290 | //calculate dE/dx |
3a039a31 |
291 | CookdEdx(fReconstructor->GetNdEdxSlices()); |
0906e73e |
292 | |
293 | // Sets the a priori probabilities |
294 | for(int ispec=0; ispec<AliPID::kSPECIES; ispec++) { |
ae4e8b84 |
295 | fProb[ispec] = pd->GetProbability(ispec, fMom, &fdEdx[0], length, GetPlane()); |
0906e73e |
296 | } |
297 | |
29b87567 |
298 | return &fProb[0]; |
0906e73e |
299 | } |
300 | |
e4f2f73d |
301 | //____________________________________________________________________ |
302 | Float_t AliTRDseedV1::GetQuality(Bool_t kZcorr) const |
303 | { |
304 | // |
305 | // Returns a quality measurement of the current seed |
306 | // |
307 | |
29b87567 |
308 | Float_t zcorr = kZcorr ? fTilt * (fZProb - fZref[0]) : 0.; |
309 | return |
310 | .5 * TMath::Abs(18.0 - fN2) |
311 | + 10.* TMath::Abs(fYfit[1] - fYref[1]) |
312 | + 5. * TMath::Abs(fYfit[0] - fYref[0] + zcorr) |
313 | + 2. * TMath::Abs(fMeanz - fZref[0]) / fPadLength; |
e4f2f73d |
314 | } |
315 | |
0906e73e |
316 | //____________________________________________________________________ |
317 | void AliTRDseedV1::GetCovAt(Double_t /*x*/, Double_t *cov) const |
318 | { |
319 | // Computes covariance in the y-z plane at radial point x |
320 | |
29b87567 |
321 | Int_t ic = 0; while (!fClusters[ic]) ic++; |
eb38ed55 |
322 | AliTRDcalibDB *fCalib = AliTRDcalibDB::Instance(); |
29b87567 |
323 | Double_t exB = fCalib->GetOmegaTau(fCalib->GetVdriftAverage(fClusters[ic]->GetDetector()), -AliTracker::GetBz()*0.1); |
eb38ed55 |
324 | |
29b87567 |
325 | Double_t sy2 = fSigmaY2*fSigmaY2 + .2*(fYfit[1]-exB)*(fYfit[1]-exB); |
326 | Double_t sz2 = fPadLength/12.; |
0906e73e |
327 | |
eb38ed55 |
328 | |
29b87567 |
329 | //printf("Yfit[1] %f sy20 %f SigmaY2 %f\n", fYfit[1], sy20, fSigmaY2); |
0906e73e |
330 | |
29b87567 |
331 | cov[0] = sy2; |
332 | cov[1] = fTilt*(sy2-sz2); |
333 | cov[2] = sz2; |
3a039a31 |
334 | |
335 | // insert systematic uncertainties calibration and misalignment |
336 | Double_t sys[15]; |
337 | fReconstructor->GetRecoParam()->GetSysCovMatrix(sys); |
338 | cov[0] += (sys[0]*sys[0]); |
339 | cov[2] += (sys[1]*sys[1]); |
0906e73e |
340 | } |
341 | |
0906e73e |
342 | |
343 | //____________________________________________________________________ |
29b87567 |
344 | void AliTRDseedV1::SetOwner() |
0906e73e |
345 | { |
29b87567 |
346 | //AliInfo(Form("own [%s] fOwner[%s]", own?"YES":"NO", fOwner?"YES":"NO")); |
347 | |
348 | if(TestBit(kOwner)) return; |
349 | for(int ic=0; ic<knTimebins; ic++){ |
350 | if(!fClusters[ic]) continue; |
351 | fClusters[ic] = new AliTRDcluster(*fClusters[ic]); |
352 | } |
353 | SetBit(kOwner); |
0906e73e |
354 | } |
355 | |
e4f2f73d |
356 | //____________________________________________________________________ |
eb38ed55 |
357 | Bool_t AliTRDseedV1::AttachClustersIter(AliTRDtrackingChamber *chamber, Float_t quality, Bool_t kZcorr, AliTRDcluster *c) |
e4f2f73d |
358 | { |
359 | // |
360 | // Iterative process to register clusters to the seed. |
361 | // In iteration 0 we try only one pad-row and if quality not |
362 | // sufficient we try 2 pad-rows (about 5% of tracks cross 2 pad-rows) |
363 | // |
29b87567 |
364 | // debug level 7 |
365 | // |
366 | |
367 | if(!fReconstructor->GetRecoParam() ){ |
368 | AliError("Seed can not be used without a valid RecoParam."); |
369 | return kFALSE; |
370 | } |
371 | |
372 | AliTRDchamberTimeBin *layer = 0x0; |
a8276d32 |
373 | if(fReconstructor->GetStreamLevel(AliTRDReconstructor::kTracker)>=7){ |
e8037fda |
374 | AliTRDtrackingChamber ch(*chamber); |
375 | ch.SetOwner(); |
a8276d32 |
376 | (*AliTRDtrackerV1::DebugStreamer()) << "AttachClustersIter" |
e8037fda |
377 | << "chamber.=" << &ch |
29b87567 |
378 | << "tracklet.=" << this |
29b87567 |
379 | << "\n"; |
380 | } |
381 | |
35c24814 |
382 | Float_t tquality; |
29b87567 |
383 | Double_t kroady = fReconstructor->GetRecoParam() ->GetRoad1y(); |
384 | Double_t kroadz = fPadLength * .5 + 1.; |
35c24814 |
385 | |
386 | // initialize configuration parameters |
387 | Float_t zcorr = kZcorr ? fTilt * (fZProb - fZref[0]) : 0.; |
388 | Int_t niter = kZcorr ? 1 : 2; |
389 | |
29b87567 |
390 | Double_t yexp, zexp; |
391 | Int_t ncl = 0; |
35c24814 |
392 | // start seed update |
393 | for (Int_t iter = 0; iter < niter; iter++) { |
29b87567 |
394 | ncl = 0; |
395 | for (Int_t iTime = 0; iTime < AliTRDtrackerV1::GetNTimeBins(); iTime++) { |
396 | if(!(layer = chamber->GetTB(iTime))) continue; |
397 | if(!Int_t(*layer)) continue; |
398 | |
399 | // define searching configuration |
400 | Double_t dxlayer = layer->GetX() - fX0; |
401 | if(c){ |
402 | zexp = c->GetZ(); |
403 | //Try 2 pad-rows in second iteration |
404 | if (iter > 0) { |
405 | zexp = fZref[0] + fZref[1] * dxlayer - zcorr; |
406 | if (zexp > c->GetZ()) zexp = c->GetZ() + fPadLength*0.5; |
407 | if (zexp < c->GetZ()) zexp = c->GetZ() - fPadLength*0.5; |
408 | } |
409 | } else zexp = fZref[0] + (kZcorr ? fZref[1] * dxlayer : 0.); |
35c24814 |
410 | yexp = fYref[0] + fYref[1] * dxlayer - zcorr; |
29b87567 |
411 | |
412 | // Get and register cluster |
413 | Int_t index = layer->SearchNearestCluster(yexp, zexp, kroady, kroadz); |
414 | if (index < 0) continue; |
415 | AliTRDcluster *cl = (*layer)[index]; |
35c24814 |
416 | |
29b87567 |
417 | fIndexes[iTime] = layer->GetGlobalIndex(index); |
418 | fClusters[iTime] = cl; |
419 | fY[iTime] = cl->GetY(); |
420 | fZ[iTime] = cl->GetZ(); |
421 | ncl++; |
422 | } |
35c24814 |
423 | if(fReconstructor->GetStreamLevel(AliTRDReconstructor::kTracker)>=7) AliInfo(Form("iter = %d ncl [%d] = %d", iter, fDet, ncl)); |
29b87567 |
424 | |
425 | if(ncl>1){ |
426 | // calculate length of the time bin (calibration aware) |
427 | Int_t irp = 0; Float_t x[2]; Int_t tb[2]; |
428 | for (Int_t iTime = 0; iTime < AliTRDtrackerV1::GetNTimeBins(); iTime++) { |
429 | if(!fClusters[iTime]) continue; |
430 | x[irp] = fClusters[iTime]->GetX(); |
431 | tb[irp] = iTime; |
432 | irp++; |
433 | if(irp==2) break; |
434 | } |
435 | fdX = (x[1] - x[0]) / (tb[0] - tb[1]); |
436 | |
437 | // update X0 from the clusters (calibration/alignment aware) |
438 | for (Int_t iTime = 0; iTime < AliTRDtrackerV1::GetNTimeBins(); iTime++) { |
439 | if(!(layer = chamber->GetTB(iTime))) continue; |
440 | if(!layer->IsT0()) continue; |
441 | if(fClusters[iTime]){ |
442 | fX0 = fClusters[iTime]->GetX(); |
443 | break; |
444 | } else { // we have to infere the position of the anode wire from the other clusters |
445 | for (Int_t jTime = iTime+1; jTime < AliTRDtrackerV1::GetNTimeBins(); jTime++) { |
446 | if(!fClusters[jTime]) continue; |
447 | fX0 = fClusters[jTime]->GetX() + fdX * (jTime - iTime); |
448 | } |
449 | break; |
450 | } |
451 | } |
452 | |
453 | // update YZ reference point |
454 | // TODO |
455 | |
456 | // update x reference positions (calibration/alignment aware) |
457 | for (Int_t iTime = 0; iTime < AliTRDtrackerV1::GetNTimeBins(); iTime++) { |
458 | if(!fClusters[iTime]) continue; |
459 | fX[iTime] = fClusters[iTime]->GetX() - fX0; |
460 | } |
461 | |
462 | AliTRDseed::Update(); |
463 | } |
35c24814 |
464 | if(fReconstructor->GetStreamLevel(AliTRDReconstructor::kTracker)>=7) AliInfo(Form("iter = %d nclFit [%d] = %d", iter, fDet, fN2)); |
29b87567 |
465 | |
466 | if(IsOK()){ |
467 | tquality = GetQuality(kZcorr); |
468 | if(tquality < quality) break; |
469 | else quality = tquality; |
470 | } |
471 | kroadz *= 2.; |
472 | } // Loop: iter |
473 | if (!IsOK()) return kFALSE; |
474 | |
804bb02e |
475 | if(fReconstructor->GetStreamLevel(AliTRDReconstructor::kTracker)>=1) CookLabels(); |
29b87567 |
476 | UpdateUsed(); |
477 | return kTRUE; |
e4f2f73d |
478 | } |
479 | |
480 | //____________________________________________________________________ |
eb38ed55 |
481 | Bool_t AliTRDseedV1::AttachClusters(AliTRDtrackingChamber *chamber |
29b87567 |
482 | ,Bool_t kZcorr) |
e4f2f73d |
483 | { |
484 | // |
485 | // Projective algorithm to attach clusters to seeding tracklets |
486 | // |
487 | // Parameters |
488 | // |
489 | // Output |
490 | // |
491 | // Detailed description |
492 | // 1. Collapse x coordinate for the full detector plane |
493 | // 2. truncated mean on y (r-phi) direction |
494 | // 3. purge clusters |
495 | // 4. truncated mean on z direction |
496 | // 5. purge clusters |
497 | // 6. fit tracklet |
498 | // |
499 | |
29b87567 |
500 | if(!fReconstructor->GetRecoParam() ){ |
501 | AliError("Seed can not be used without a valid RecoParam."); |
502 | return kFALSE; |
503 | } |
504 | |
505 | const Int_t kClusterCandidates = 2 * knTimebins; |
506 | |
507 | //define roads |
508 | Double_t kroady = fReconstructor->GetRecoParam() ->GetRoad1y(); |
509 | Double_t kroadz = fPadLength * 1.5 + 1.; |
510 | // correction to y for the tilting angle |
511 | Float_t zcorr = kZcorr ? fTilt * (fZProb - fZref[0]) : 0.; |
512 | |
513 | // working variables |
514 | AliTRDcluster *clusters[kClusterCandidates]; |
515 | Double_t cond[4], yexp[knTimebins], zexp[knTimebins], |
516 | yres[kClusterCandidates], zres[kClusterCandidates]; |
517 | Int_t ncl, *index = 0x0, tboundary[knTimebins]; |
518 | |
519 | // Do cluster projection |
520 | AliTRDchamberTimeBin *layer = 0x0; |
521 | Int_t nYclusters = 0; Bool_t kEXIT = kFALSE; |
522 | for (Int_t iTime = 0; iTime < AliTRDtrackerV1::GetNTimeBins(); iTime++) { |
523 | if(!(layer = chamber->GetTB(iTime))) continue; |
524 | if(!Int_t(*layer)) continue; |
525 | |
526 | fX[iTime] = layer->GetX() - fX0; |
527 | zexp[iTime] = fZref[0] + fZref[1] * fX[iTime]; |
528 | yexp[iTime] = fYref[0] + fYref[1] * fX[iTime] - zcorr; |
529 | |
530 | // build condition and process clusters |
531 | cond[0] = yexp[iTime] - kroady; cond[1] = yexp[iTime] + kroady; |
532 | cond[2] = zexp[iTime] - kroadz; cond[3] = zexp[iTime] + kroadz; |
533 | layer->GetClusters(cond, index, ncl); |
534 | for(Int_t ic = 0; ic<ncl; ic++){ |
535 | AliTRDcluster *c = layer->GetCluster(index[ic]); |
536 | clusters[nYclusters] = c; |
537 | yres[nYclusters++] = c->GetY() - yexp[iTime]; |
538 | if(nYclusters >= kClusterCandidates) { |
539 | AliWarning(Form("Cluster candidates reached limit %d. Some may be lost.", kClusterCandidates)); |
540 | kEXIT = kTRUE; |
541 | break; |
542 | } |
543 | } |
544 | tboundary[iTime] = nYclusters; |
545 | if(kEXIT) break; |
546 | } |
547 | |
548 | // Evaluate truncated mean on the y direction |
549 | Double_t mean, sigma; |
550 | AliMathBase::EvaluateUni(nYclusters, yres, mean, sigma, Int_t(nYclusters*.8)-2); |
551 | // purge cluster candidates |
552 | Int_t nZclusters = 0; |
553 | for(Int_t ic = 0; ic<nYclusters; ic++){ |
554 | if(yres[ic] - mean > 4. * sigma){ |
555 | clusters[ic] = 0x0; |
556 | continue; |
557 | } |
558 | zres[nZclusters++] = clusters[ic]->GetZ() - zexp[clusters[ic]->GetLocalTimeBin()]; |
559 | } |
560 | |
561 | // Evaluate truncated mean on the z direction |
562 | AliMathBase::EvaluateUni(nZclusters, zres, mean, sigma, Int_t(nZclusters*.8)-2); |
563 | // purge cluster candidates |
564 | for(Int_t ic = 0; ic<nZclusters; ic++){ |
565 | if(zres[ic] - mean > 4. * sigma){ |
566 | clusters[ic] = 0x0; |
567 | continue; |
568 | } |
569 | } |
570 | |
571 | |
572 | // Select only one cluster/TimeBin |
573 | Int_t lastCluster = 0; |
574 | fN2 = 0; |
575 | for (Int_t iTime = 0; iTime < AliTRDtrackerV1::GetNTimeBins(); iTime++) { |
576 | ncl = tboundary[iTime] - lastCluster; |
577 | if(!ncl) continue; |
578 | Int_t iptr = lastCluster; |
579 | if(ncl > 1){ |
580 | Float_t dold = 9999.; |
581 | for(int ic=lastCluster; ic<tboundary[iTime]; ic++){ |
582 | if(!clusters[ic]) continue; |
583 | Float_t y = yexp[iTime] - clusters[ic]->GetY(); |
584 | Float_t z = zexp[iTime] - clusters[ic]->GetZ(); |
585 | Float_t d = y * y + z * z; |
586 | if(d > dold) continue; |
587 | dold = d; |
588 | iptr = ic; |
589 | } |
590 | } |
591 | fIndexes[iTime] = chamber->GetTB(iTime)->GetGlobalIndex(iptr); |
592 | fClusters[iTime] = clusters[iptr]; |
593 | fY[iTime] = clusters[iptr]->GetY(); |
594 | fZ[iTime] = clusters[iptr]->GetZ(); |
595 | lastCluster = tboundary[iTime]; |
596 | fN2++; |
597 | } |
598 | |
599 | // number of minimum numbers of clusters expected for the tracklet |
600 | Int_t kClmin = Int_t(fReconstructor->GetRecoParam() ->GetFindableClusters()*AliTRDtrackerV1::GetNTimeBins()); |
e4f2f73d |
601 | if (fN2 < kClmin){ |
29b87567 |
602 | AliWarning(Form("Not enough clusters to fit the tracklet %d [%d].", fN2, kClmin)); |
e4f2f73d |
603 | fN2 = 0; |
604 | return kFALSE; |
605 | } |
0906e73e |
606 | |
29b87567 |
607 | // update used clusters |
608 | fNUsed = 0; |
609 | for (Int_t iTime = 0; iTime < AliTRDtrackerV1::GetNTimeBins(); iTime++) { |
610 | if(!fClusters[iTime]) continue; |
611 | if((fClusters[iTime]->IsUsed())) fNUsed++; |
612 | } |
0906e73e |
613 | |
614 | if (fN2-fNUsed < kClmin){ |
29b87567 |
615 | AliWarning(Form("Too many clusters already in use %d (from %d).", fNUsed, fN2)); |
0906e73e |
616 | fN2 = 0; |
617 | return kFALSE; |
618 | } |
29b87567 |
619 | |
620 | return kTRUE; |
e4f2f73d |
621 | } |
622 | |
03cef9b2 |
623 | //____________________________________________________________ |
624 | void AliTRDseedV1::Bootstrap(const AliTRDReconstructor *rec) |
625 | { |
626 | // Fill in all derived information. It has to be called after recovery from file or HLT. |
627 | // The primitive data are |
628 | // - list of clusters |
629 | // - detector (as the detector will be removed from clusters) |
630 | // - position of anode wire (fX0) - temporary |
631 | // - track reference position and direction |
632 | // - momentum of the track |
633 | // - time bin length [cm] |
634 | // |
635 | // A.Bercuci <A.Bercuci@gsi.de> Oct 30th 2008 |
636 | // |
637 | fReconstructor = rec; |
638 | AliTRDgeometry g; |
639 | AliTRDpadPlane *pp = g.GetPadPlane(fDet); |
640 | fTilt = TMath::Tan(TMath::DegToRad()*pp->GetTiltingAngle()); |
641 | fPadLength = pp->GetLengthIPad(); |
642 | fSnp = fYref[1]/TMath::Sqrt(1+fYref[1]*fYref[1]); |
643 | fTgl = fZref[1]; |
644 | fN = 0; fN2 = 0; fMPads = 0.; |
645 | AliTRDcluster **cit = &fClusters[0]; |
646 | for(Int_t ic = knTimebins; ic--; cit++){ |
647 | if(!(*cit)) return; |
648 | fN++; fN2++; |
649 | fX[ic] = (*cit)->GetX() - fX0; |
650 | fY[ic] = (*cit)->GetY(); |
651 | fZ[ic] = (*cit)->GetZ(); |
652 | } |
653 | Update(); // Fit(); |
654 | CookLabels(); |
655 | GetProbability(); |
656 | } |
657 | |
658 | |
e4f2f73d |
659 | //____________________________________________________________________ |
d2b9977a |
660 | Bool_t AliTRDseedV1::Fit(Bool_t tilt) |
e4f2f73d |
661 | { |
662 | // |
663 | // Linear fit of the tracklet |
664 | // |
665 | // Parameters : |
666 | // |
667 | // Output : |
668 | // True if successful |
669 | // |
670 | // Detailed description |
671 | // 2. Check if tracklet crosses pad row boundary |
672 | // 1. Calculate residuals in the y (r-phi) direction |
673 | // 3. Do a Least Square Fit to the data |
674 | // |
675 | |
29b87567 |
676 | const Int_t kClmin = 8; |
ae4e8b84 |
677 | const Float_t q0 = 100.; |
678 | const Float_t clSigma0 = 2.E-2; //[cm] |
679 | const Float_t clSlopeQ = -1.19E-2; //[1/cm] |
680 | |
2f7d6ac8 |
681 | // get track direction |
682 | Double_t y0 = fYref[0]; |
683 | Double_t dydx = fYref[1]; |
684 | Double_t z0 = fZref[0]; |
685 | Double_t dzdx = fZref[1]; |
686 | Double_t yt, zt; |
ae4e8b84 |
687 | |
29b87567 |
688 | const Int_t kNtb = AliTRDtrackerV1::GetNTimeBins(); |
689 | AliTRDtrackerV1::AliTRDLeastSquare fitterY, fitterZ; |
690 | |
691 | // convertion factor from square to gauss distribution for sigma |
692 | Double_t convert = 1./TMath::Sqrt(12.); |
ae4e8b84 |
693 | |
29b87567 |
694 | // book cluster information |
2f7d6ac8 |
695 | Double_t xc[knTimebins], yc[knTimebins], zc[knTimebins], sy[knTimebins], sz[knTimebins]; |
29b87567 |
696 | Int_t zRow[knTimebins]; |
2f7d6ac8 |
697 | |
698 | |
699 | fN = 0; |
9eb2d46c |
700 | AliTRDcluster *c=0x0, **jc = &fClusters[0]; |
9eb2d46c |
701 | for (Int_t ic=0; ic<kNtb; ic++, ++jc) { |
29b87567 |
702 | zRow[ic] = -1; |
703 | xc[ic] = -1.; |
704 | yc[ic] = 999.; |
705 | zc[ic] = 999.; |
706 | sy[ic] = 0.; |
707 | sz[ic] = 0.; |
9eb2d46c |
708 | if(!(c = (*jc))) continue; |
29b87567 |
709 | if(!c->IsInChamber()) continue; |
710 | Float_t w = 1.; |
711 | if(c->GetNPads()>4) w = .5; |
712 | if(c->GetNPads()>5) w = .2; |
2f7d6ac8 |
713 | zRow[fN] = c->GetPadRow(); |
714 | xc[fN] = fX0 - c->GetX(); |
715 | yc[fN] = c->GetY(); |
716 | zc[fN] = c->GetZ(); |
717 | |
718 | // extrapolated y value for the track |
719 | yt = y0 - xc[fN]*dydx; |
720 | // extrapolated z value for the track |
721 | zt = z0 - xc[fN]*dzdx; |
722 | // tilt correction |
723 | if(tilt) yc[fN] -= fTilt*(zc[fN] - zt); |
724 | |
725 | // elaborate cluster error |
ae4e8b84 |
726 | Float_t qr = c->GetQ() - q0; |
2f7d6ac8 |
727 | sy[fN] = qr < 0. ? clSigma0*TMath::Exp(clSlopeQ*qr) : clSigma0; |
728 | |
729 | fitterY.AddPoint(&xc[fN], yc[fN]-yt, sy[fN]); |
730 | |
731 | sz[fN] = fPadLength*convert; |
732 | fitterZ.AddPoint(&xc[fN], zc[fN], sz[fN]); |
733 | fN++; |
29b87567 |
734 | } |
47d5d320 |
735 | // to few clusters |
2f7d6ac8 |
736 | if (fN < kClmin) return kFALSE; |
737 | |
738 | // fit XY plane |
739 | fitterY.Eval(); |
740 | fYfit[0] = y0+fitterY.GetFunctionParameter(0); |
741 | fYfit[1] = dydx-fitterY.GetFunctionParameter(1); |
742 | |
743 | // check par row crossing |
744 | Int_t zN[2*AliTRDseed::knTimebins]; |
745 | Int_t nz = AliTRDtrackerV1::Freq(fN, zRow, zN, kFALSE); |
29b87567 |
746 | // more than one pad row crossing |
747 | if(nz>2) return kFALSE; |
9eb2d46c |
748 | |
29b87567 |
749 | |
750 | // determine z offset of the fit |
2f7d6ac8 |
751 | Float_t zslope = 0.; |
29b87567 |
752 | Int_t nchanges = 0, nCross = 0; |
753 | if(nz==2){ // tracklet is crossing pad row |
754 | // Find the break time allowing one chage on pad-rows |
755 | // with maximal number of accepted clusters |
756 | Int_t padRef = zRow[0]; |
2f7d6ac8 |
757 | for (Int_t ic=1; ic<fN; ic++) { |
29b87567 |
758 | if(zRow[ic] == padRef) continue; |
759 | |
760 | // debug |
761 | if(zRow[ic-1] == zRow[ic]){ |
762 | printf("ERROR in pad row change!!!\n"); |
763 | } |
764 | |
765 | // evaluate parameters of the crossing point |
766 | Float_t sx = (xc[ic-1] - xc[ic])*convert; |
2f7d6ac8 |
767 | fCross[0] = .5 * (xc[ic-1] + xc[ic]); |
768 | fCross[2] = .5 * (zc[ic-1] + zc[ic]); |
769 | fCross[3] = TMath::Max(dzdx * sx, .01); |
770 | zslope = zc[ic-1] > zc[ic] ? 1. : -1.; |
771 | padRef = zRow[ic]; |
772 | nCross = ic; |
29b87567 |
773 | nchanges++; |
774 | } |
775 | } |
776 | |
777 | // condition on nCross and reset nchanges TODO |
778 | |
779 | if(nchanges==1){ |
2f7d6ac8 |
780 | if(dzdx * zslope < 0.){ |
29b87567 |
781 | AliInfo("tracklet direction does not correspond to the track direction. TODO."); |
782 | } |
783 | SetBit(kRowCross, kTRUE); // mark pad row crossing |
2f7d6ac8 |
784 | fitterZ.AddPoint(&fCross[0], fCross[2], fCross[3]); |
29b87567 |
785 | fitterZ.Eval(); |
2f7d6ac8 |
786 | //zc[nc] = fitterZ.GetFunctionParameter(0); |
787 | fCross[1] = fYfit[0] - fCross[0] * fYfit[1]; |
788 | fCross[0] = fX0 - fCross[0]; |
29b87567 |
789 | } else if(nchanges > 1){ // debug |
2389e96f |
790 | AliError("N pad row crossing > 1."); |
29b87567 |
791 | return kFALSE; |
792 | } |
793 | |
2389e96f |
794 | UpdateUsed(); |
795 | |
29b87567 |
796 | return kTRUE; |
e4f2f73d |
797 | } |
798 | |
e4f2f73d |
799 | |
800 | //___________________________________________________________________ |
47d5d320 |
801 | void AliTRDseedV1::Print(Option_t*) const |
e4f2f73d |
802 | { |
803 | // |
804 | // Printing the seedstatus |
805 | // |
806 | |
4dc4dc2e |
807 | AliInfo(Form("Tracklet X0[%7.2f] Det[%d]", fX0, fDet)); |
808 | printf(" Tilt[%+6.2f] PadLength[%5.2f]\n", fTilt, fPadLength); |
809 | AliTRDcluster* const* jc = &fClusters[0]; |
810 | for(int ic=0; ic<AliTRDtrackerV1::GetNTimeBins(); ic++, jc++) { |
811 | if(!(*jc)) continue; |
812 | printf(" %2d X[%7.2f] Y[%7.2f] Z[%7.2f] Idx[%d] c[%p] usable[%s]\n", |
813 | ic, (*jc)->GetX(), (*jc)->GetY(), (*jc)->GetZ(), |
814 | fIndexes[ic], (void*)(*jc), fUsable[ic]?"y":"n"); |
815 | } |
e4f2f73d |
816 | |
29b87567 |
817 | printf(" fYref[0] =%f fYref[1] =%f\n", fYref[0], fYref[1]); |
818 | printf(" fZref[0] =%f fZref[1] =%f\n", fZref[0], fZref[1]); |
819 | printf(" fYfit[0] =%f fYfit[1] =%f\n", fYfit[0], fYfit[1]); |
29b87567 |
820 | printf(" fZfit[0] =%f fZfit[1] =%f\n", fZfit[0], fZfit[1]); |
29b87567 |
821 | printf(" fSigmaY =%f\n", fSigmaY); |
822 | printf(" fSigmaY2=%f\n", fSigmaY2); |
823 | printf(" fMeanz =%f\n", fMeanz); |
824 | printf(" fZProb =%f\n", fZProb); |
825 | printf(" fLabels[0]=%d fLabels[1]=%d\n", fLabels[0], fLabels[1]); |
826 | printf(" fN =%d\n", fN); |
4dc4dc2e |
827 | printf(" fN2 =%d (>4 isOK - to be redesigned)\n",fN2); |
29b87567 |
828 | printf(" fNUsed =%d\n", fNUsed); |
829 | printf(" fFreq =%d\n", fFreq); |
830 | printf(" fNChange=%d\n", fNChange); |
831 | printf(" fMPads =%f\n", fMPads); |
832 | |
833 | printf(" fC =%f\n", fC); |
834 | printf(" fCC =%f\n",fCC); |
835 | printf(" fChi2 =%f\n", fChi2); |
836 | printf(" fChi2Z =%f\n", fChi2Z); |
e4f2f73d |
837 | } |
47d5d320 |
838 | |