updates in cluster error parametrization + documentation
[u/mrichter/AliRoot.git] / TRD / AliTRDseedV1.cxx
CommitLineData
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 48ClassImp(AliTRDseedV1)
49
50//____________________________________________________________________
ae4e8b84 51AliTRDseedV1::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.)
6e4d4425 61 ,fXref(0.)
e4f2f73d 62{
63 //
64 // Constructor
65 //
29b87567 66 //printf("AliTRDseedV1::AliTRDseedV1()\n");
e44586fb 67
29b87567 68 for(int islice=0; islice < knSlices; islice++) fdEdx[islice] = 0.;
69 for(int ispec=0; ispec<AliPID::kSPECIES; ispec++) fProb[ispec] = -1.;
6e4d4425 70 fRefCov[0] = 1.; fRefCov[1] = 0.; fRefCov[2] = 1.;
e4f2f73d 71}
72
73//____________________________________________________________________
0906e73e 74AliTRDseedV1::AliTRDseedV1(const AliTRDseedV1 &ref)
e4f2f73d 75 :AliTRDseed((AliTRDseed&)ref)
43d6ad34 76 ,fReconstructor(ref.fReconstructor)
ae4e8b84 77 ,fClusterIter(0x0)
78 ,fClusterIdx(0)
79 ,fDet(ref.fDet)
0906e73e 80 ,fMom(ref.fMom)
bcb6fb78 81 ,fSnp(ref.fSnp)
82 ,fTgl(ref.fTgl)
83 ,fdX(ref.fdX)
6e4d4425 84 ,fXref(ref.fXref)
e4f2f73d 85{
86 //
87 // Copy Constructor performing a deep copy
88 //
89
29b87567 90 //printf("AliTRDseedV1::AliTRDseedV1(const AliTRDseedV1 &)\n");
91 SetBit(kOwner, kFALSE);
92 for(int islice=0; islice < knSlices; islice++) fdEdx[islice] = ref.fdEdx[islice];
93 for(int ispec=0; ispec<AliPID::kSPECIES; ispec++) fProb[ispec] = ref.fProb[ispec];
6e4d4425 94 memcpy(fRefCov, ref.fRefCov, 3*sizeof(Double_t));
fbb2ea06 95}
d9950a5a 96
0906e73e 97
e4f2f73d 98//____________________________________________________________________
99AliTRDseedV1& AliTRDseedV1::operator=(const AliTRDseedV1 &ref)
100{
101 //
102 // Assignment Operator using the copy function
103 //
104
29b87567 105 if(this != &ref){
106 ref.Copy(*this);
107 }
221ab7e0 108 SetBit(kOwner, kFALSE);
109
29b87567 110 return *this;
e4f2f73d 111
112}
113
114//____________________________________________________________________
115AliTRDseedV1::~AliTRDseedV1()
116{
117 //
118 // Destructor. The RecoParam object belongs to the underlying tracker.
119 //
120
29b87567 121 //printf("I-AliTRDseedV1::~AliTRDseedV1() : Owner[%s]\n", IsOwner()?"YES":"NO");
e4f2f73d 122
29b87567 123 if(IsOwner())
124 for(int itb=0; itb<knTimebins; itb++){
125 if(!fClusters[itb]) continue;
126 //AliInfo(Form("deleting c %p @ %d", fClusters[itb], itb));
127 delete fClusters[itb];
128 fClusters[itb] = 0x0;
129 }
e4f2f73d 130}
131
132//____________________________________________________________________
133void AliTRDseedV1::Copy(TObject &ref) const
134{
135 //
136 // Copy function
137 //
138
29b87567 139 //AliInfo("");
140 AliTRDseedV1 &target = (AliTRDseedV1 &)ref;
141
ae4e8b84 142 target.fClusterIter = 0x0;
143 target.fClusterIdx = 0;
144 target.fDet = fDet;
29b87567 145 target.fMom = fMom;
146 target.fSnp = fSnp;
147 target.fTgl = fTgl;
148 target.fdX = fdX;
6e4d4425 149 target.fXref = fXref;
29b87567 150 target.fReconstructor = fReconstructor;
151
152 for(int islice=0; islice < knSlices; islice++) target.fdEdx[islice] = fdEdx[islice];
153 for(int ispec=0; ispec<AliPID::kSPECIES; ispec++) target.fProb[ispec] = fProb[ispec];
6e4d4425 154 memcpy(target.fRefCov, fRefCov, 3*sizeof(Double_t));
29b87567 155
156 AliTRDseed::Copy(target);
e4f2f73d 157}
158
0906e73e 159
160//____________________________________________________________
f3d3af1b 161Bool_t AliTRDseedV1::Init(AliTRDtrackV1 *track)
0906e73e 162{
163// Initialize this tracklet using the track information
164//
165// Parameters:
166// track - the TRD track used to initialize the tracklet
167//
168// Detailed description
169// The function sets the starting point and direction of the
170// tracklet according to the information from the TRD track.
171//
172// Caution
173// The TRD track has to be propagated to the beginning of the
174// chamber where the tracklet will be constructed
175//
176
29b87567 177 Double_t y, z;
178 if(!track->GetProlongation(fX0, y, z)) return kFALSE;
179 fYref[0] = y;
180 fYref[1] = track->GetSnp()/(1. - track->GetSnp()*track->GetSnp());
181 fZref[0] = z;
182 fZref[1] = track->GetTgl();
6e4d4425 183
184 const Double_t *cov = track->GetCovariance();
185 fRefCov[0] = cov[0]; // Var(y)
186 fRefCov[1] = cov[1]; // Cov(yz)
187 fRefCov[2] = cov[5]; // Var(z)
0906e73e 188
29b87567 189 //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());
190 return kTRUE;
0906e73e 191}
192
bcb6fb78 193
194//____________________________________________________________________
195void AliTRDseedV1::CookdEdx(Int_t nslices)
196{
197// Calculates average dE/dx for all slices and store them in the internal array fdEdx.
198//
199// Parameters:
200// nslices : number of slices for which dE/dx should be calculated
201// Output:
202// store results in the internal array fdEdx. This can be accessed with the method
203// AliTRDseedV1::GetdEdx()
204//
205// Detailed description
206// Calculates average dE/dx for all slices. Depending on the PID methode
207// the number of slices can be 3 (LQ) or 8(NN).
208// The calculation of dQ/dl are done using the tracklet fit results (see AliTRDseedV1::GetdQdl(Int_t)) i.e.
209//
210// dQ/dl = qc/(dx * sqrt(1 + dy/dx^2 + dz/dx^2))
211//
212// The following effects are included in the calculation:
213// 1. calibration values for t0 and vdrift (using x coordinate to calculate slice)
214// 2. cluster sharing (optional see AliTRDrecoParam::SetClusterSharing())
215// 3. cluster size
216//
217
29b87567 218 Int_t nclusters[knSlices];
219 for(int i=0; i<knSlices; i++){
220 fdEdx[i] = 0.;
221 nclusters[i] = 0;
222 }
223 Float_t clength = (/*.5 * */AliTRDgeometry::AmThick() + AliTRDgeometry::DrThick());
224
225 AliTRDcluster *cluster = 0x0;
226 for(int ic=0; ic<AliTRDtrackerV1::GetNTimeBins(); ic++){
227 if(!(cluster = fClusters[ic])) continue;
228 Float_t x = cluster->GetX();
229
230 // Filter clusters for dE/dx calculation
231
232 // 1.consider calibration effects for slice determination
233 Int_t slice;
234 if(cluster->IsInChamber()) slice = Int_t(TMath::Abs(fX0 - x) * nslices / clength);
235 else slice = x < fX0 ? 0 : nslices-1;
236
237 // 2. take sharing into account
238 Float_t w = cluster->IsShared() ? .5 : 1.;
239
240 // 3. take into account large clusters TODO
241 //w *= c->GetNPads() > 3 ? .8 : 1.;
242
243 //CHECK !!!
244 fdEdx[slice] += w * GetdQdl(ic); //fdQdl[ic];
245 nclusters[slice]++;
246 } // End of loop over clusters
247
cd40b287 248 //if(fReconstructor->GetPIDMethod() == AliTRDReconstructor::kLQPID){
249 if(nslices == AliTRDReconstructor::kLQslices){
29b87567 250 // calculate mean charge per slice (only LQ PID)
251 for(int is=0; is<nslices; is++){
252 if(nclusters[is]) fdEdx[is] /= nclusters[is];
253 }
254 }
bcb6fb78 255}
256
b83573da 257
bcb6fb78 258//____________________________________________________________________
259Float_t AliTRDseedV1::GetdQdl(Int_t ic) const
260{
29b87567 261 return fClusters[ic] ? TMath::Abs(fClusters[ic]->GetQ()) /fdX / TMath::Sqrt(1. + fYfit[1]*fYfit[1] + fZref[1]*fZref[1]) : 0.;
bcb6fb78 262}
263
0906e73e 264//____________________________________________________________________
265Double_t* AliTRDseedV1::GetProbability()
266{
267// Fill probability array for tracklet from the DB.
268//
269// Parameters
270//
271// Output
272// returns pointer to the probability array and 0x0 if missing DB access
273//
274// Detailed description
275
29b87567 276
277 // retrive calibration db
0906e73e 278 AliTRDcalibDB *calibration = AliTRDcalibDB::Instance();
279 if (!calibration) {
280 AliError("No access to calibration data");
281 return 0x0;
282 }
283
3a039a31 284 if (!fReconstructor) {
285 AliError("Reconstructor not set.");
4ba1d6ae 286 return 0x0;
287 }
288
0906e73e 289 // Retrieve the CDB container class with the parametric detector response
3a039a31 290 const AliTRDCalPID *pd = calibration->GetPIDObject(fReconstructor->GetPIDMethod());
0906e73e 291 if (!pd) {
292 AliError("No access to AliTRDCalPID object");
293 return 0x0;
294 }
29b87567 295 //AliInfo(Form("Method[%d] : %s", fReconstructor->GetRecoParam() ->GetPIDMethod(), pd->IsA()->GetName()));
10f75631 296
29b87567 297 // calculate tracklet length TO DO
0906e73e 298 Float_t length = (AliTRDgeometry::AmThick() + AliTRDgeometry::DrThick());
299 /// TMath::Sqrt((1.0 - fSnp[iPlane]*fSnp[iPlane]) / (1.0 + fTgl[iPlane]*fTgl[iPlane]));
300
301 //calculate dE/dx
3a039a31 302 CookdEdx(fReconstructor->GetNdEdxSlices());
0906e73e 303
304 // Sets the a priori probabilities
305 for(int ispec=0; ispec<AliPID::kSPECIES; ispec++) {
ae4e8b84 306 fProb[ispec] = pd->GetProbability(ispec, fMom, &fdEdx[0], length, GetPlane());
0906e73e 307 }
308
29b87567 309 return &fProb[0];
0906e73e 310}
311
312//____________________________________________________________________
e4f2f73d 313Float_t AliTRDseedV1::GetQuality(Bool_t kZcorr) const
314{
315 //
316 // Returns a quality measurement of the current seed
317 //
318
29b87567 319 Float_t zcorr = kZcorr ? fTilt * (fZProb - fZref[0]) : 0.;
320 return
321 .5 * TMath::Abs(18.0 - fN2)
322 + 10.* TMath::Abs(fYfit[1] - fYref[1])
323 + 5. * TMath::Abs(fYfit[0] - fYref[0] + zcorr)
324 + 2. * TMath::Abs(fMeanz - fZref[0]) / fPadLength;
e4f2f73d 325}
326
327//____________________________________________________________________
0906e73e 328void AliTRDseedV1::GetCovAt(Double_t /*x*/, Double_t *cov) const
329{
330// Computes covariance in the y-z plane at radial point x
331
29b87567 332 Int_t ic = 0; while (!fClusters[ic]) ic++;
eb38ed55 333 AliTRDcalibDB *fCalib = AliTRDcalibDB::Instance();
29b87567 334 Double_t exB = fCalib->GetOmegaTau(fCalib->GetVdriftAverage(fClusters[ic]->GetDetector()), -AliTracker::GetBz()*0.1);
eb38ed55 335
29b87567 336 Double_t sy2 = fSigmaY2*fSigmaY2 + .2*(fYfit[1]-exB)*(fYfit[1]-exB);
337 Double_t sz2 = fPadLength/12.;
0906e73e 338
eb38ed55 339
29b87567 340 //printf("Yfit[1] %f sy20 %f SigmaY2 %f\n", fYfit[1], sy20, fSigmaY2);
0906e73e 341
29b87567 342 cov[0] = sy2;
343 cov[1] = fTilt*(sy2-sz2);
344 cov[2] = sz2;
3a039a31 345
346 // insert systematic uncertainties calibration and misalignment
347 Double_t sys[15];
348 fReconstructor->GetRecoParam()->GetSysCovMatrix(sys);
349 cov[0] += (sys[0]*sys[0]);
350 cov[2] += (sys[1]*sys[1]);
0906e73e 351}
352
0906e73e 353
354//____________________________________________________________________
29b87567 355void AliTRDseedV1::SetOwner()
0906e73e 356{
29b87567 357 //AliInfo(Form("own [%s] fOwner[%s]", own?"YES":"NO", fOwner?"YES":"NO"));
358
359 if(TestBit(kOwner)) return;
360 for(int ic=0; ic<knTimebins; ic++){
361 if(!fClusters[ic]) continue;
362 fClusters[ic] = new AliTRDcluster(*fClusters[ic]);
363 }
364 SetBit(kOwner);
0906e73e 365}
366
367//____________________________________________________________________
eb38ed55 368Bool_t AliTRDseedV1::AttachClustersIter(AliTRDtrackingChamber *chamber, Float_t quality, Bool_t kZcorr, AliTRDcluster *c)
e4f2f73d 369{
370 //
371 // Iterative process to register clusters to the seed.
372 // In iteration 0 we try only one pad-row and if quality not
373 // sufficient we try 2 pad-rows (about 5% of tracks cross 2 pad-rows)
374 //
29b87567 375 // debug level 7
376 //
377
378 if(!fReconstructor->GetRecoParam() ){
379 AliError("Seed can not be used without a valid RecoParam.");
380 return kFALSE;
381 }
382
383 AliTRDchamberTimeBin *layer = 0x0;
a8276d32 384 if(fReconstructor->GetStreamLevel(AliTRDReconstructor::kTracker)>=7){
e8037fda 385 AliTRDtrackingChamber ch(*chamber);
386 ch.SetOwner();
29f95561 387 TTreeSRedirector &cstreamer = *fReconstructor->GetDebugStream(AliTRDReconstructor::kTracker);
388 cstreamer << "AttachClustersIter"
e8037fda 389 << "chamber.=" << &ch
29b87567 390 << "tracklet.=" << this
29b87567 391 << "\n";
392 }
393
35c24814 394 Float_t tquality;
29b87567 395 Double_t kroady = fReconstructor->GetRecoParam() ->GetRoad1y();
396 Double_t kroadz = fPadLength * .5 + 1.;
35c24814 397
398 // initialize configuration parameters
399 Float_t zcorr = kZcorr ? fTilt * (fZProb - fZref[0]) : 0.;
400 Int_t niter = kZcorr ? 1 : 2;
401
29b87567 402 Double_t yexp, zexp;
403 Int_t ncl = 0;
35c24814 404 // start seed update
405 for (Int_t iter = 0; iter < niter; iter++) {
29b87567 406 ncl = 0;
407 for (Int_t iTime = 0; iTime < AliTRDtrackerV1::GetNTimeBins(); iTime++) {
408 if(!(layer = chamber->GetTB(iTime))) continue;
409 if(!Int_t(*layer)) continue;
410
411 // define searching configuration
412 Double_t dxlayer = layer->GetX() - fX0;
413 if(c){
414 zexp = c->GetZ();
415 //Try 2 pad-rows in second iteration
416 if (iter > 0) {
417 zexp = fZref[0] + fZref[1] * dxlayer - zcorr;
418 if (zexp > c->GetZ()) zexp = c->GetZ() + fPadLength*0.5;
419 if (zexp < c->GetZ()) zexp = c->GetZ() - fPadLength*0.5;
420 }
421 } else zexp = fZref[0] + (kZcorr ? fZref[1] * dxlayer : 0.);
35c24814 422 yexp = fYref[0] + fYref[1] * dxlayer - zcorr;
29b87567 423
424 // Get and register cluster
425 Int_t index = layer->SearchNearestCluster(yexp, zexp, kroady, kroadz);
426 if (index < 0) continue;
427 AliTRDcluster *cl = (*layer)[index];
35c24814 428
29b87567 429 fIndexes[iTime] = layer->GetGlobalIndex(index);
430 fClusters[iTime] = cl;
431 fY[iTime] = cl->GetY();
432 fZ[iTime] = cl->GetZ();
433 ncl++;
434 }
35c24814 435 if(fReconstructor->GetStreamLevel(AliTRDReconstructor::kTracker)>=7) AliInfo(Form("iter = %d ncl [%d] = %d", iter, fDet, ncl));
29b87567 436
437 if(ncl>1){
438 // calculate length of the time bin (calibration aware)
439 Int_t irp = 0; Float_t x[2]; Int_t tb[2];
440 for (Int_t iTime = 0; iTime < AliTRDtrackerV1::GetNTimeBins(); iTime++) {
441 if(!fClusters[iTime]) continue;
442 x[irp] = fClusters[iTime]->GetX();
443 tb[irp] = iTime;
444 irp++;
445 if(irp==2) break;
446 }
447 fdX = (x[1] - x[0]) / (tb[0] - tb[1]);
448
449 // update X0 from the clusters (calibration/alignment aware)
450 for (Int_t iTime = 0; iTime < AliTRDtrackerV1::GetNTimeBins(); iTime++) {
451 if(!(layer = chamber->GetTB(iTime))) continue;
452 if(!layer->IsT0()) continue;
453 if(fClusters[iTime]){
454 fX0 = fClusters[iTime]->GetX();
455 break;
456 } else { // we have to infere the position of the anode wire from the other clusters
457 for (Int_t jTime = iTime+1; jTime < AliTRDtrackerV1::GetNTimeBins(); jTime++) {
458 if(!fClusters[jTime]) continue;
459 fX0 = fClusters[jTime]->GetX() + fdX * (jTime - iTime);
f660dce9 460 break;
29b87567 461 }
29b87567 462 }
463 }
464
465 // update YZ reference point
466 // TODO
467
468 // update x reference positions (calibration/alignment aware)
469 for (Int_t iTime = 0; iTime < AliTRDtrackerV1::GetNTimeBins(); iTime++) {
470 if(!fClusters[iTime]) continue;
0849f128 471 fX[iTime] = fX0 - fClusters[iTime]->GetX();
29b87567 472 }
473
474 AliTRDseed::Update();
475 }
35c24814 476 if(fReconstructor->GetStreamLevel(AliTRDReconstructor::kTracker)>=7) AliInfo(Form("iter = %d nclFit [%d] = %d", iter, fDet, fN2));
29b87567 477
478 if(IsOK()){
479 tquality = GetQuality(kZcorr);
480 if(tquality < quality) break;
481 else quality = tquality;
482 }
483 kroadz *= 2.;
484 } // Loop: iter
485 if (!IsOK()) return kFALSE;
486
804bb02e 487 if(fReconstructor->GetStreamLevel(AliTRDReconstructor::kTracker)>=1) CookLabels();
29b87567 488 UpdateUsed();
489 return kTRUE;
e4f2f73d 490}
491
492//____________________________________________________________________
eb38ed55 493Bool_t AliTRDseedV1::AttachClusters(AliTRDtrackingChamber *chamber
29b87567 494 ,Bool_t kZcorr)
e4f2f73d 495{
496 //
497 // Projective algorithm to attach clusters to seeding tracklets
498 //
499 // Parameters
500 //
501 // Output
502 //
503 // Detailed description
504 // 1. Collapse x coordinate for the full detector plane
505 // 2. truncated mean on y (r-phi) direction
506 // 3. purge clusters
507 // 4. truncated mean on z direction
508 // 5. purge clusters
509 // 6. fit tracklet
510 //
511
29b87567 512 if(!fReconstructor->GetRecoParam() ){
513 AliError("Seed can not be used without a valid RecoParam.");
514 return kFALSE;
515 }
516
517 const Int_t kClusterCandidates = 2 * knTimebins;
518
519 //define roads
520 Double_t kroady = fReconstructor->GetRecoParam() ->GetRoad1y();
521 Double_t kroadz = fPadLength * 1.5 + 1.;
522 // correction to y for the tilting angle
523 Float_t zcorr = kZcorr ? fTilt * (fZProb - fZref[0]) : 0.;
524
525 // working variables
526 AliTRDcluster *clusters[kClusterCandidates];
527 Double_t cond[4], yexp[knTimebins], zexp[knTimebins],
528 yres[kClusterCandidates], zres[kClusterCandidates];
529 Int_t ncl, *index = 0x0, tboundary[knTimebins];
530
531 // Do cluster projection
532 AliTRDchamberTimeBin *layer = 0x0;
533 Int_t nYclusters = 0; Bool_t kEXIT = kFALSE;
534 for (Int_t iTime = 0; iTime < AliTRDtrackerV1::GetNTimeBins(); iTime++) {
535 if(!(layer = chamber->GetTB(iTime))) continue;
536 if(!Int_t(*layer)) continue;
537
538 fX[iTime] = layer->GetX() - fX0;
539 zexp[iTime] = fZref[0] + fZref[1] * fX[iTime];
540 yexp[iTime] = fYref[0] + fYref[1] * fX[iTime] - zcorr;
541
542 // build condition and process clusters
543 cond[0] = yexp[iTime] - kroady; cond[1] = yexp[iTime] + kroady;
544 cond[2] = zexp[iTime] - kroadz; cond[3] = zexp[iTime] + kroadz;
545 layer->GetClusters(cond, index, ncl);
546 for(Int_t ic = 0; ic<ncl; ic++){
547 AliTRDcluster *c = layer->GetCluster(index[ic]);
548 clusters[nYclusters] = c;
549 yres[nYclusters++] = c->GetY() - yexp[iTime];
550 if(nYclusters >= kClusterCandidates) {
551 AliWarning(Form("Cluster candidates reached limit %d. Some may be lost.", kClusterCandidates));
552 kEXIT = kTRUE;
553 break;
554 }
555 }
556 tboundary[iTime] = nYclusters;
557 if(kEXIT) break;
558 }
559
560 // Evaluate truncated mean on the y direction
561 Double_t mean, sigma;
562 AliMathBase::EvaluateUni(nYclusters, yres, mean, sigma, Int_t(nYclusters*.8)-2);
563 // purge cluster candidates
564 Int_t nZclusters = 0;
565 for(Int_t ic = 0; ic<nYclusters; ic++){
566 if(yres[ic] - mean > 4. * sigma){
567 clusters[ic] = 0x0;
568 continue;
569 }
570 zres[nZclusters++] = clusters[ic]->GetZ() - zexp[clusters[ic]->GetLocalTimeBin()];
571 }
572
573 // Evaluate truncated mean on the z direction
574 AliMathBase::EvaluateUni(nZclusters, zres, mean, sigma, Int_t(nZclusters*.8)-2);
575 // purge cluster candidates
576 for(Int_t ic = 0; ic<nZclusters; ic++){
577 if(zres[ic] - mean > 4. * sigma){
578 clusters[ic] = 0x0;
579 continue;
580 }
581 }
582
583
584 // Select only one cluster/TimeBin
585 Int_t lastCluster = 0;
586 fN2 = 0;
587 for (Int_t iTime = 0; iTime < AliTRDtrackerV1::GetNTimeBins(); iTime++) {
588 ncl = tboundary[iTime] - lastCluster;
589 if(!ncl) continue;
590 Int_t iptr = lastCluster;
591 if(ncl > 1){
592 Float_t dold = 9999.;
593 for(int ic=lastCluster; ic<tboundary[iTime]; ic++){
594 if(!clusters[ic]) continue;
595 Float_t y = yexp[iTime] - clusters[ic]->GetY();
596 Float_t z = zexp[iTime] - clusters[ic]->GetZ();
597 Float_t d = y * y + z * z;
598 if(d > dold) continue;
599 dold = d;
600 iptr = ic;
601 }
602 }
603 fIndexes[iTime] = chamber->GetTB(iTime)->GetGlobalIndex(iptr);
604 fClusters[iTime] = clusters[iptr];
605 fY[iTime] = clusters[iptr]->GetY();
606 fZ[iTime] = clusters[iptr]->GetZ();
607 lastCluster = tboundary[iTime];
608 fN2++;
609 }
610
611 // number of minimum numbers of clusters expected for the tracklet
612 Int_t kClmin = Int_t(fReconstructor->GetRecoParam() ->GetFindableClusters()*AliTRDtrackerV1::GetNTimeBins());
e4f2f73d 613 if (fN2 < kClmin){
29b87567 614 AliWarning(Form("Not enough clusters to fit the tracklet %d [%d].", fN2, kClmin));
e4f2f73d 615 fN2 = 0;
616 return kFALSE;
617 }
0906e73e 618
29b87567 619 // update used clusters
620 fNUsed = 0;
621 for (Int_t iTime = 0; iTime < AliTRDtrackerV1::GetNTimeBins(); iTime++) {
622 if(!fClusters[iTime]) continue;
623 if((fClusters[iTime]->IsUsed())) fNUsed++;
624 }
0906e73e 625
626 if (fN2-fNUsed < kClmin){
29b87567 627 AliWarning(Form("Too many clusters already in use %d (from %d).", fNUsed, fN2));
0906e73e 628 fN2 = 0;
629 return kFALSE;
630 }
29b87567 631
632 return kTRUE;
e4f2f73d 633}
634
03cef9b2 635//____________________________________________________________
636void AliTRDseedV1::Bootstrap(const AliTRDReconstructor *rec)
637{
638// Fill in all derived information. It has to be called after recovery from file or HLT.
639// The primitive data are
640// - list of clusters
641// - detector (as the detector will be removed from clusters)
642// - position of anode wire (fX0) - temporary
643// - track reference position and direction
644// - momentum of the track
645// - time bin length [cm]
646//
647// A.Bercuci <A.Bercuci@gsi.de> Oct 30th 2008
648//
649 fReconstructor = rec;
650 AliTRDgeometry g;
651 AliTRDpadPlane *pp = g.GetPadPlane(fDet);
652 fTilt = TMath::Tan(TMath::DegToRad()*pp->GetTiltingAngle());
653 fPadLength = pp->GetLengthIPad();
654 fSnp = fYref[1]/TMath::Sqrt(1+fYref[1]*fYref[1]);
655 fTgl = fZref[1];
656 fN = 0; fN2 = 0; fMPads = 0.;
657 AliTRDcluster **cit = &fClusters[0];
658 for(Int_t ic = knTimebins; ic--; cit++){
659 if(!(*cit)) return;
660 fN++; fN2++;
661 fX[ic] = (*cit)->GetX() - fX0;
662 fY[ic] = (*cit)->GetY();
663 fZ[ic] = (*cit)->GetZ();
664 }
665 Update(); // Fit();
666 CookLabels();
667 GetProbability();
668}
669
670
e4f2f73d 671//____________________________________________________________________
d2b9977a 672Bool_t AliTRDseedV1::Fit(Bool_t tilt)
e4f2f73d 673{
674 //
675 // Linear fit of the tracklet
676 //
677 // Parameters :
678 //
679 // Output :
680 // True if successful
681 //
682 // Detailed description
683 // 2. Check if tracklet crosses pad row boundary
684 // 1. Calculate residuals in the y (r-phi) direction
685 // 3. Do a Least Square Fit to the data
686 //
687
29b87567 688 const Int_t kClmin = 8;
9462866a 689 // drift velocity correction TODO to be moved to the clusterizer
690 const Float_t cx[] = {
6910.044168, 0.130812, -0.017411, -0.099284, -0.120416, -0.095457,
692-0.050021, -0.016758, 0.003570, 0.018618, 0.026380, 0.033786, 0.034889, 0.035264,
6930.035284, 0.036028, 0.035250, 0.034368, 0.032823, 0.031937, 0.032064, 0.022542,
694-0.025167, -0.120645, 0.};
010d62b0 695 // PRF correction TODO to be replaced by the gaussian
696 // approximation with full error parametrization and // moved to the clusterizer
697 const Float_t cy[AliTRDgeometry::kNlayer][3] = {
698 {0.000413, -0.008896, 6.858256},
699 {-0.000324, -0.009979, 6.649032},
700 {0.000886, -0.011371, 6.952959},
701 {-0.001558, -0.012692, 6.790299},
702 {0.002195, -0.014308, 6.855086},
703 {-0.001229, -0.015282, 6.612536}
704 };
705
9462866a 706
707 // cluster error parametrization parameters
010d62b0 708 // 1. sy total charge
9462866a 709 const Float_t sq0inv = 0.019962; // [1/q0]
710 const Float_t sqb = 1.0281564; //[cm]
010d62b0 711 // 2. sy for the PRF
712 const Float_t scy[AliTRDgeometry::kNlayer][4] = {
713 {2.813e-02, 1.879e-03, 4.331e-01, 2.267e-02},
714 {2.937e-02, 7.207e-04, 4.184e-01, 2.337e-02},
715 {3.076e-02, 1.890e-03, 4.050e-01, 2.399e-02},
716 {3.240e-02, -1.409e-05, 3.994e-01, 2.508e-02},
717 {3.417e-02, -5.888e-04, 3.924e-01, 2.621e-02},
718 {3.493e-02, 2.044e-03, 3.675e-01, 2.585e-02},
719 };
720 // 3. sy parallel to the track
9462866a 721 const Float_t sy0 = 2.60967e-01; // [mm] !!
722 const Float_t sya =-7.68941e+00; //
723 const Float_t syb =-3.41160e-01; //
010d62b0 724 // 4. sx parallel to the track
9462866a 725 const Float_t sxgc = 5.49018e-01;
726 const Float_t sxgm = 7.82999e-01;
727 const Float_t sxgs = 2.74451e-01;
728 const Float_t sxe0 = 2.53596e-01;
729 const Float_t sxe1 =-2.40078e-02;
010d62b0 730 // 5. sx perpendicular to the track
9462866a 731// const Float_t sxd0 = 0.190676;
732// const Float_t sxd1 =-3.9269;
733// const Float_t sxd2 =14.7851;
734
2f7d6ac8 735 // get track direction
736 Double_t y0 = fYref[0];
737 Double_t dydx = fYref[1];
738 Double_t z0 = fZref[0];
739 Double_t dzdx = fZref[1];
740 Double_t yt, zt;
ae4e8b84 741
29b87567 742 const Int_t kNtb = AliTRDtrackerV1::GetNTimeBins();
24d8660e 743 AliTRDtrackerV1::AliTRDLeastSquare fitterZ;
744 TLinearFitter fitterY(1, "pol1");
29b87567 745 // convertion factor from square to gauss distribution for sigma
746 Double_t convert = 1./TMath::Sqrt(12.);
ae4e8b84 747
29b87567 748 // book cluster information
9462866a 749 Double_t q, xc[knTimebins], yc[knTimebins], zc[knTimebins], sy[knTimebins], sz[knTimebins];
29b87567 750 Int_t zRow[knTimebins];
9462866a 751
752 // TODO move as data member of the tracklet
753 // TODO calculate for the exact position of the tracklet (det, col, row)
754 Double_t exb = -.16;
010d62b0 755
756 Int_t ily = AliTRDgeometry::GetLayer(fDet);
2f7d6ac8 757 fN = 0;
9eb2d46c 758 AliTRDcluster *c=0x0, **jc = &fClusters[0];
9eb2d46c 759 for (Int_t ic=0; ic<kNtb; ic++, ++jc) {
29b87567 760 zRow[ic] = -1;
761 xc[ic] = -1.;
762 yc[ic] = 999.;
763 zc[ic] = 999.;
764 sy[ic] = 0.;
765 sz[ic] = 0.;
9eb2d46c 766 if(!(c = (*jc))) continue;
29b87567 767 if(!c->IsInChamber()) continue;
9462866a 768
29b87567 769 Float_t w = 1.;
770 if(c->GetNPads()>4) w = .5;
771 if(c->GetNPads()>5) w = .2;
010d62b0 772
773 // correct cluster position for PRF and v drift
774 c->SetX(c->GetX() - cx[c->GetLocalTimeBin()]);
775 c->SetY(c->GetY() + cy[ily][0] + cy[ily][1] * TMath::Sin(cy[ily][2] * c->GetCenter()));
776
2f7d6ac8 777 zRow[fN] = c->GetPadRow();
010d62b0 778 xc[fN] = fX0 - c->GetX();
2f7d6ac8 779 yc[fN] = c->GetY();
780 zc[fN] = c->GetZ();
781
782 // extrapolated y value for the track
783 yt = y0 - xc[fN]*dydx;
784 // extrapolated z value for the track
785 zt = z0 - xc[fN]*dzdx;
786 // tilt correction
787 if(tilt) yc[fN] -= fTilt*(zc[fN] - zt);
788
010d62b0 789 // ELABORATE CLUSTER ERROR
790 // TODO to be moved to AliTRDcluster
9462866a 791 q = TMath::Abs(c->GetQ());
792 Double_t tgg = (dydx-exb)/(1.+dydx*exb);
010d62b0 793 // basic y error (|| to track).
794 sy[fN] = sy0 + TMath::Exp(sya*(xc[fN]+syb));
795 // y error due to total charge
796 sy[fN] += sqb*(1./q - sq0inv);
797 // y error due to PRF
798 sy[fN] += scy[ily][0]*TMath::Gaus(c->GetCenter(), scy[ily][1], scy[ily][2]) - scy[ily][3];
799 sy[fN] *= sy[fN];
800
801 // ADD ERROR ON x
9462866a 802 // error of drift length parallel to the track
803 Double_t sx = sxgc*TMath::Gaus(xc[fN], sxgm, sxgs) + sxe0*TMath::Exp(sxe1*xc[fN]); // [cm]
804 // error of drift length perpendicular to the track
805 //sx += sxd0 + sxd1*d + sxd2*d*d;
806 // global radial error due to misalignment/miscalibration
807 Double_t sx0 = 0.; // [cm]
9462866a 808 sy[fN] += tgg*tgg*(sx*sx+sx0*sx0);
809 // add error from ExB
810 sy[fN] += exb*exb*sx*sx;
811 sy[fN] = TMath::Sqrt(sy[fN]);
24d8660e 812 fitterY.AddPoint(&xc[fN], yc[fN]/*-yt*/, sy[fN]);
2f7d6ac8 813
814 sz[fN] = fPadLength*convert;
815 fitterZ.AddPoint(&xc[fN], zc[fN], sz[fN]);
816 fN++;
29b87567 817 }
47d5d320 818 // to few clusters
2f7d6ac8 819 if (fN < kClmin) return kFALSE;
820
821 // fit XY plane
822 fitterY.Eval();
24d8660e 823 fYfit[0] = /*y0+*/fitterY.GetParameter(0);
824 fYfit[1] = /*dydx-*/-fitterY.GetParameter(1);
2f7d6ac8 825
826 // check par row crossing
827 Int_t zN[2*AliTRDseed::knTimebins];
828 Int_t nz = AliTRDtrackerV1::Freq(fN, zRow, zN, kFALSE);
29b87567 829 // more than one pad row crossing
830 if(nz>2) return kFALSE;
9eb2d46c 831
29b87567 832
833 // determine z offset of the fit
2f7d6ac8 834 Float_t zslope = 0.;
29b87567 835 Int_t nchanges = 0, nCross = 0;
836 if(nz==2){ // tracklet is crossing pad row
837 // Find the break time allowing one chage on pad-rows
838 // with maximal number of accepted clusters
839 Int_t padRef = zRow[0];
2f7d6ac8 840 for (Int_t ic=1; ic<fN; ic++) {
29b87567 841 if(zRow[ic] == padRef) continue;
842
843 // debug
844 if(zRow[ic-1] == zRow[ic]){
845 printf("ERROR in pad row change!!!\n");
846 }
847
848 // evaluate parameters of the crossing point
849 Float_t sx = (xc[ic-1] - xc[ic])*convert;
2f7d6ac8 850 fCross[0] = .5 * (xc[ic-1] + xc[ic]);
851 fCross[2] = .5 * (zc[ic-1] + zc[ic]);
852 fCross[3] = TMath::Max(dzdx * sx, .01);
853 zslope = zc[ic-1] > zc[ic] ? 1. : -1.;
854 padRef = zRow[ic];
855 nCross = ic;
29b87567 856 nchanges++;
857 }
858 }
859
860 // condition on nCross and reset nchanges TODO
861
862 if(nchanges==1){
2f7d6ac8 863 if(dzdx * zslope < 0.){
29b87567 864 AliInfo("tracklet direction does not correspond to the track direction. TODO.");
865 }
866 SetBit(kRowCross, kTRUE); // mark pad row crossing
2f7d6ac8 867 fitterZ.AddPoint(&fCross[0], fCross[2], fCross[3]);
29b87567 868 fitterZ.Eval();
2f7d6ac8 869 //zc[nc] = fitterZ.GetFunctionParameter(0);
870 fCross[1] = fYfit[0] - fCross[0] * fYfit[1];
871 fCross[0] = fX0 - fCross[0];
29b87567 872 } else if(nchanges > 1){ // debug
2389e96f 873 AliError("N pad row crossing > 1.");
29b87567 874 return kFALSE;
875 }
876
2389e96f 877 UpdateUsed();
878
29b87567 879 return kTRUE;
e4f2f73d 880}
881
e4f2f73d 882
883//___________________________________________________________________
203967fc 884void AliTRDseedV1::Print(Option_t *o) const
e4f2f73d 885{
886 //
887 // Printing the seedstatus
888 //
889
203967fc 890 AliInfo(Form("Det[%3d] Tilt[%+6.2f] Pad[%5.2f]", fDet, fTilt, fPadLength));
891 AliInfo(Form("Nattach[%2d] Nfit[%2d] Nuse[%2d] pads[%f]", fN, fN2, fNUsed, fMPads));
892 AliInfo(Form("x[%7.2f] y[%7.2f] z[%7.2f] dydx[%5.2f] dzdx[%5.2f]", fX0, fYfit[0], fZfit[0], fYfit[1], fZfit[1]));
893 AliInfo(Form("Ref y[%7.2f] z[%7.2f] dydx[%5.2f] dzdx[%5.2f]", fYref[0], fZref[0], fYref[1], fZref[1]))
894
895
896 if(strcmp(o, "a")!=0) return;
897
4dc4dc2e 898 AliTRDcluster* const* jc = &fClusters[0];
899 for(int ic=0; ic<AliTRDtrackerV1::GetNTimeBins(); ic++, jc++) {
900 if(!(*jc)) continue;
203967fc 901 (*jc)->Print(o);
4dc4dc2e 902 }
e4f2f73d 903
203967fc 904/* printf(" fSigmaY =%f\n", fSigmaY);
29b87567 905 printf(" fSigmaY2=%f\n", fSigmaY2);
906 printf(" fMeanz =%f\n", fMeanz);
907 printf(" fZProb =%f\n", fZProb);
203967fc 908 printf(" fLabels[0]=%d fLabels[1]=%d\n", fLabels[0], fLabels[1]);*/
29b87567 909
203967fc 910/* printf(" fC =%f\n", fC);
29b87567 911 printf(" fCC =%f\n",fCC);
912 printf(" fChi2 =%f\n", fChi2);
203967fc 913 printf(" fChi2Z =%f\n", fChi2Z);*/
e4f2f73d 914}
47d5d320 915
203967fc 916
917//___________________________________________________________________
918Bool_t AliTRDseedV1::IsEqual(const TObject *o) const
919{
920 // Checks if current instance of the class has the same essential members
921 // as the given one
922
923 if(!o) return kFALSE;
924 const AliTRDseedV1 *inTracklet = dynamic_cast<const AliTRDseedV1*>(o);
925 if(!inTracklet) return kFALSE;
926
927 for (Int_t i = 0; i < 2; i++){
928 if ( fYref[i] != inTracklet->GetYref(i) ) return kFALSE;
929 if ( fZref[i] != inTracklet->GetZref(i) ) return kFALSE;
930 }
931
932 if ( fSigmaY != inTracklet->GetSigmaY() ) return kFALSE;
933 if ( fSigmaY2 != inTracklet->GetSigmaY2() ) return kFALSE;
934 if ( fTilt != inTracklet->GetTilt() ) return kFALSE;
935 if ( fPadLength != inTracklet->GetPadLength() ) return kFALSE;
936
937 for (Int_t i = 0; i < knTimebins; i++){
938 if ( fX[i] != inTracklet->GetX(i) ) return kFALSE;
939 if ( fY[i] != inTracklet->GetY(i) ) return kFALSE;
940 if ( fZ[i] != inTracklet->GetZ(i) ) return kFALSE;
941 if ( fIndexes[i] != inTracklet->GetIndexes(i) ) return kFALSE;
942 if ( fUsable[i] != inTracklet->IsUsable(i) ) return kFALSE;
943 }
944
945 for (Int_t i=0; i < 2; i++){
946 if ( fYfit[i] != inTracklet->GetYfit(i) ) return kFALSE;
947 if ( fZfit[i] != inTracklet->GetZfit(i) ) return kFALSE;
948 if ( fYfitR[i] != inTracklet->GetYfitR(i) ) return kFALSE;
949 if ( fZfitR[i] != inTracklet->GetZfitR(i) ) return kFALSE;
950 if ( fLabels[i] != inTracklet->GetLabels(i) ) return kFALSE;
951 }
952
953 if ( fMeanz != inTracklet->GetMeanz() ) return kFALSE;
954 if ( fZProb != inTracklet->GetZProb() ) return kFALSE;
955 if ( fN2 != inTracklet->GetN2() ) return kFALSE;
956 if ( fNUsed != inTracklet->GetNUsed() ) return kFALSE;
957 if ( fFreq != inTracklet->GetFreq() ) return kFALSE;
958 if ( fNChange != inTracklet->GetNChange() ) return kFALSE;
959 if ( fNChange != inTracklet->GetNChange() ) return kFALSE;
960
961 if ( fC != inTracklet->GetC() ) return kFALSE;
962 if ( fCC != inTracklet->GetCC() ) return kFALSE;
963 if ( fChi2 != inTracklet->GetChi2() ) return kFALSE;
964 // if ( fChi2Z != inTracklet->GetChi2Z() ) return kFALSE;
965
966 if ( fDet != inTracklet->GetDetector() ) return kFALSE;
967 if ( fMom != inTracklet->GetMomentum() ) return kFALSE;
968 if ( fdX != inTracklet->GetdX() ) return kFALSE;
969
970 for (Int_t iCluster = 0; iCluster < knTimebins; iCluster++){
971 AliTRDcluster *curCluster = fClusters[iCluster];
972 AliTRDcluster *inCluster = inTracklet->GetClusters(iCluster);
973 if (curCluster && inCluster){
974 if (! curCluster->IsEqual(inCluster) ) {
975 curCluster->Print();
976 inCluster->Print();
977 return kFALSE;
978 }
979 } else {
980 // if one cluster exists, and corresponding
981 // in other tracklet doesn't - return kFALSE
982 if(curCluster || inCluster) return kFALSE;
983 }
984 }
985 return kTRUE;
986}