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