X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=TRD%2FAliTRDtrackV1.cxx;h=4921d25bf9caf2268b70c5848f4fbfad72e80733;hb=0224bae45b3fa90bf541c790956567ddea1cc520;hp=0c27e1773074b96a1f36cd58990030667e6e1911;hpb=0b433f724fda289449c24d74213b993693e592d8;p=u%2Fmrichter%2FAliRoot.git diff --git a/TRD/AliTRDtrackV1.cxx b/TRD/AliTRDtrackV1.cxx index 0c27e177307..4921d25bf9c 100644 --- a/TRD/AliTRDtrackV1.cxx +++ b/TRD/AliTRDtrackV1.cxx @@ -1,3 +1,4 @@ + /************************************************************************** * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * * * @@ -15,6 +16,7 @@ /* $Id$ */ +#include "TVectorT.h" #include "AliLog.h" #include "AliESDtrack.h" #include "AliTracker.h" @@ -23,7 +25,12 @@ #include "AliTRDcluster.h" #include "AliTRDcalibDB.h" #include "AliTRDReconstructor.h" +#include "AliTRDPIDResponse.h" #include "AliTRDrecoParam.h" +#include "AliTRDdEdxBaseUtils.h" +#include "AliTRDdEdxCalibHistArray.h" +#include "AliTRDdEdxCalibUtils.h" +#include "AliTRDdEdxReconUtils.h" ClassImp(AliTRDtrackV1) @@ -41,11 +48,16 @@ ClassImp(AliTRDtrackV1) //_______________________________________________________________ AliTRDtrackV1::AliTRDtrackV1() : AliKalmanTrack() ,fStatus(0) + ,fESDid(0) ,fDE(0.) - ,fReconstructor(0x0) - ,fBackupTrack(0x0) - ,fTrackLow(0x0) - ,fTrackHigh(0x0) + ,fTruncatedMean(0) + ,fNchamberdEdx(0) + ,fNclusterdEdx(0) + ,fNdEdxSlices(0) + ,fkReconstructor(NULL) + ,fBackupTrack(NULL) + ,fTrackLow(NULL) + ,fTrackHigh(NULL) { // // Default constructor @@ -58,19 +70,25 @@ AliTRDtrackV1::AliTRDtrackV1() : AliKalmanTrack() for(int is =0; isGetX() ,par->GetAlpha() ,par->GetParameter() @@ -135,22 +161,27 @@ AliTRDtrackV1::AliTRDtrackV1(const AliESDtrack &t) : AliKalmanTrack() if(t.GetStatus() & AliESDtrack::kTIME) { StartTimeIntegral(); - Double_t times[10]; - t.GetIntegratedTimes(times); + Double_t times[AliPID::kSPECIESC]; + t.GetIntegratedTimes(times,AliPID::kSPECIESC); SetIntegratedTimes(times); SetIntegratedLength(t.GetIntegratedLength()); } } //_______________________________________________________________ -AliTRDtrackV1::AliTRDtrackV1(AliTRDseedV1 *trklts, const Double_t p[5], const Double_t cov[15] +AliTRDtrackV1::AliTRDtrackV1(AliTRDseedV1 * const trklts, const Double_t p[5], const Double_t cov[15] , Double_t x, Double_t alpha) : AliKalmanTrack() ,fStatus(0) + ,fESDid(0) ,fDE(0.) - ,fReconstructor(0x0) - ,fBackupTrack(0x0) - ,fTrackLow(0x0) - ,fTrackHigh(0x0) + ,fTruncatedMean(0) + ,fNchamberdEdx(0) + ,fNclusterdEdx(0) + ,fNdEdxSlices(0) + ,fkReconstructor(NULL) + ,fBackupTrack(NULL) + ,fTrackLow(NULL) + ,fTrackHigh(NULL) { // // The stand alone tracking constructor @@ -160,150 +191,232 @@ AliTRDtrackV1::AliTRDtrackV1(AliTRDseedV1 *trklts, const Double_t p[5], const Do // 2. dQdl calculation // - Double_t cnv = GetBz() < 1.e-5 ? 1.e5 : 1.0 / (GetBz() * kB2C); - // Double_t cnv = 1.0 / (GetBz() * kB2C); - + Double_t b(GetBz()); + Double_t cnv = (TMath::Abs(b) < 1.e-5) ? 1.e5 : 1./GetBz()/kB2C; + Double_t pp[5] = { p[0] - , p[1] - , p[2] - , p[3] - , p[4]*cnv }; - + , p[1] + , p[2] + , p[3] + , p[4]*cnv }; + Double_t c22 = x*x*cov[14] - 2*x*cov[12] + cov[ 5]; Double_t c32 = x*cov[13] - cov[ 8]; Double_t c20 = x*cov[10] - cov[ 3]; Double_t c21 = x*cov[11] - cov[ 4]; Double_t c42 = x*cov[14] - cov[12]; - + Double_t cc[15] = { cov[ 0] , cov[ 1], cov[ 2] , c20, c21, c22 , cov[ 6], cov[ 7], c32, cov[ 9] , cov[10]*cnv, cov[11]*cnv, c42*cnv, cov[13]*cnv, cov[14]*cnv*cnv }; - + Double_t mostProbablePt=AliExternalTrackParam::GetMostProbablePt(); Double_t p0=TMath::Sign(1/mostProbablePt,pp[4]); Double_t w0=cc[14]/(cc[14] + p0*p0), w1=p0*p0/(cc[14] + p0*p0); + AliDebug(3, Form("Pt mixing : w0[%4.2f] pt0[%5.3f] w1[%4.2f] pt[%5.3f]", w0, 1./p0, w1, 1./pp[4])); + pp[4] = w0*p0 + w1*pp[4]; cc[10]*=w1; cc[11]*=w1; cc[12]*=w1; cc[13]*=w1; cc[14]*=w1; - - Set(x,alpha,pp,cc); + Set(x,alpha,pp,cc); + AliDebug(2, Form("Init @ x[%6.2f] pt[%5.3f]", x, 1./pp[4])); Int_t ncls = 0; - for(int iplane=0; iplaneGetN(); } - } + } AliKalmanTrack::SetNumberOfClusters(ncls); for(int i =0; i<3; i++) fBudget[i] = 0.; Float_t pid = 1./AliPID::kSPECIES; for(int is =0; isGetClusters(ic))) continue; - for (Int_t k = 0; k < 3; k++) { - label = c->GetLabel(k); - labelAdded = kFALSE; - Int_t j = 0; - if (label >= 0) { - while ((!labelAdded) && (j < kMAXCLUSTERSPERTRACK)) { - if ((s[j][0] == label) || - (s[j][1] == 0)) { - s[j][0] = label; - s[j][1]++; - labelAdded = kTRUE; - } - j++; - } + for (Int_t k(0); k < 3; k++) { + if ((label = c->GetLabel(k)) < 0) continue; + Int_t j(0); + while(j < kMAXCLUSTERSPERTRACK){ + if(s[0][j]!=label && s[1][j]!=0){j++; continue;} + if(!s[1][j]) nlabels++; + s[0][j] = label; s[1][j]++; + break; } } } } - - Int_t max = 0; - label = -123456789; - for (Int_t i = 0; i < kMAXCLUSTERSPERTRACK; i++) { - if (s[i][1] <= max) continue; - max = s[i][1]; - label = s[i][0]; + //printf(" Found %4d labels\n", nlabels); + Float_t prob(1.); + if(!nlabels){ + AliError(Form("No MC labels found for track %d.", fESDid)); + return 0; + } else if(nlabels==1) { + label = s[0][0]; + if(labs && freq){labs[0]=label; freq[0]=1.;} + } else { + Int_t idx[kMAXCLUSTERSPERTRACK]; + TMath::Sort(nlabels, s[1], idx); + label = s[0][idx[0]]; prob = s[1][idx[0]]/Float_t(ncl); + if(labs && freq){ + for (Int_t i(0); i wrong) label = -label; - - SetLabel(label); - - return kTRUE; + SetLabel((1.-prob > wrong)?-label:label); + return nlabels; } //_______________________________________________________________ Bool_t AliTRDtrackV1::CookPID() { - // - // Cook the PID information - // - - // Reset the a priori probabilities - Double_t pid = 1. / AliPID::kSPECIES; - for(int ispec=0; ispec +//End_Html +// + const AliTRDPIDResponse *pidResponse = AliTRDcalibDB::Instance()->GetPIDResponse(fkReconstructor->GetRecoParam()->GetPIDmethod()); + if(!pidResponse){ + AliError("PID Response not available"); return kFALSE; } - - for (Int_t iSpecies = 0; iSpecies < AliPID::kSPECIES; iSpecies++) fPID[iSpecies] /= probTotal; - + Double_t dEdx[kNplane * (Int_t)AliTRDseedV1::kNdEdxSlices] = {0.}; + Float_t trackletP[kNplane] = {0.}; + + fNdEdxSlices = pidResponse->GetNumberOfSlices(); + for(Int_t iseed = 0; iseed < kNplane; iseed++){ + if(!fTracklet[iseed]) continue; + trackletP[iseed] = fTracklet[iseed]->GetMomentum(); +// if(pidResponse->GetPIDmethod() == AliTRDPIDResponse::kLQ1D){ + dEdx[iseed] = fTracklet[iseed]->GetdQdl(); +/* } else { + fTracklet[iseed]->CookdEdx(fNdEdxSlices); + const Float_t *trackletdEdx = fTracklet[iseed]->GetdEdx(); + for(Int_t islice = 0; islice < fNdEdxSlices; islice++){ + dEdx[iseed*fNdEdxSlices + islice] = trackletdEdx[islice]; + } + }*/ + } + pidResponse->GetResponse(fNdEdxSlices, dEdx, trackletP, fPID); + + static Int_t nprint = 0; + if(!nprint){ + AliTRDdEdxBaseUtils::PrintControl(); + nprint++; + } + + //do truncated mean + AliTRDdEdxCalibUtils::SetObjArray(AliTRDcalibDB::Instance()->GetPHQ()); + const Double_t mag = AliTRDdEdxBaseUtils::IsExBOn() ? GetBz() : -1; + const Double_t charge = AliTRDdEdxBaseUtils::IsExBOn() ? Charge() : -1; + fTruncatedMean = CookTruncatedMean(0, mag, charge, kTRUE, fNchamberdEdx, fNclusterdEdx); + return kTRUE; } @@ -312,55 +425,20 @@ UChar_t AliTRDtrackV1::GetNumberOfTrackletsPID() const { // Retrieve number of tracklets used for PID calculation. - UChar_t fPIDquality = 0; - Float_t *prob = 0x0; - for(int ip=0; ipIsOK()) continue; - if(!(prob = fTracklet[ip]->GetProbability(kFALSE))) return 0; - - Int_t nspec = 0; // quality check of tracklet dEdx - for(int ispec=0; ispecIsOK()) continue; - if(!(prob = fTracklet[ip]->GetProbability(recalc))) return 0; - Int_t nspec = 0; // quality check of tracklet dEdx - for(int ispec=0; ispecGetN(); continue; } - AliTRDcluster *c = 0x0; + AliTRDcluster *c = NULL; for(Int_t ic=AliTRDseedV1::kNclusters; ic--;){ if(!(c = fTracklet[ip]->GetClusters(ic))) continue; @@ -376,12 +454,13 @@ AliTRDcluster* AliTRDtrackV1::GetCluster(Int_t id) return c; } } - return 0x0; + return NULL; } //_______________________________________________________________ Int_t AliTRDtrackV1::GetClusterIndex(Int_t id) const { + // Get the cluster index at a certain position in the track Int_t n = 0; for(Int_t ip=0; ipGetN(); continue; } - AliTRDcluster *c = 0x0; for(Int_t ic=AliTRDseedV1::kNclusters; ic--;){ - if(!(c = fTracklet[ip]->GetClusters(ic))) continue; - + if(!(fTracklet[ip]->GetClusters(ic))) continue; if(nGetIndexes(ic); } @@ -427,39 +504,30 @@ Int_t AliTRDtrackV1::GetSector() const //_______________________________________________________________ Bool_t AliTRDtrackV1::IsEqual(const TObject *o) const { + // Checks whether two tracks are equal if (!o) return kFALSE; const AliTRDtrackV1 *inTrack = dynamic_cast(o); if (!inTrack) return kFALSE; //if ( fPIDquality != inTrack->GetPIDquality() ) return kFALSE; - for(Int_t i = 0; i < AliPID::kSPECIES; i++){ - if ( fPID[i] != inTrack->GetPID(i) ) return kFALSE; - } - - for (Int_t i = 0; i < 3; i++){ - if ( fBudget[i] != inTrack->GetBudget(i) ) return kFALSE; - } - if ( fDE != inTrack->GetEdep() ) return kFALSE; - if ( fFakeRatio != inTrack->GetFakeRatio() ) return kFALSE; - if ( fChi2 != inTrack->GetChi2() ) return kFALSE; - if ( fMass != inTrack->GetMass() ) return kFALSE; - if ( fLab != inTrack->GetLabel() ) return kFALSE; - if ( fN != inTrack->GetNumberOfClusters() ) return kFALSE; - if ( AliKalmanTrack::GetIntegratedLength() != inTrack->GetIntegratedLength() ) return kFALSE; - - if ( GetX() != inTrack->GetX() ) return kFALSE; - if ( GetAlpha() != inTrack->GetAlpha() ) return kFALSE; - const Double_t *inP = inTrack->GetParameter(); - const Double_t *curP = GetParameter(); - for (Int_t i = 0; i < 5; i++){ - if ( curP[i] != inP[i]) return kFALSE; - } - const Double_t *inC = inTrack->GetCovariance(); - const Double_t *curC = GetCovariance(); - for (Int_t i = 0; i < 15; i++){ - if ( curC[i] != inC[i]) return kFALSE; - } + if(memcmp(fPID, inTrack->fPID, AliPID::kSPECIES*sizeof(Double32_t))) return kFALSE; + if(memcmp(fBudget, inTrack->fBudget, 3*sizeof(Double32_t))) return kFALSE; + if(memcmp(&fDE, &inTrack->fDE, sizeof(Double32_t))) return kFALSE; + if(memcmp(&fFakeRatio, &inTrack->fFakeRatio, sizeof(Double32_t))) return kFALSE; + if(memcmp(&fChi2, &inTrack->fChi2, sizeof(Double32_t))) return kFALSE; + if(memcmp(&fMass, &inTrack->fMass, sizeof(Double32_t))) return kFALSE; + if( fLab != inTrack->fLab ) return kFALSE; + if( fN != inTrack->fN ) return kFALSE; + Double32_t l(0.), in(0.); + l = GetIntegratedLength(); in = inTrack->GetIntegratedLength(); + if(memcmp(&l, &in, sizeof(Double32_t))) return kFALSE; + l=GetX(); in=inTrack->GetX(); + if(memcmp(&l, &in, sizeof(Double32_t))) return kFALSE; + l = GetAlpha(); in = inTrack->GetAlpha(); + if(memcmp(&l, &in, sizeof(Double32_t))) return kFALSE; + if(memcmp(GetParameter(), inTrack->GetParameter(), 5*sizeof(Double32_t))) return kFALSE; + if(memcmp(GetCovariance(), inTrack->GetCovariance(), 15*sizeof(Double32_t))) return kFALSE; for (Int_t iTracklet = 0; iTracklet < kNplane; iTracklet++){ AliTRDseedV1 *curTracklet = fTracklet[iTracklet]; @@ -483,37 +551,76 @@ Bool_t AliTRDtrackV1::IsEqual(const TObject *o) const //_______________________________________________________________ Bool_t AliTRDtrackV1::IsElectron() const { - if(GetPID(0) > fReconstructor->GetRecoParam()->GetPIDThreshold(GetP())) return kTRUE; + if(GetPID(0) > fkReconstructor->GetRecoParam()->GetPIDThreshold(GetP())) return kTRUE; return kFALSE; } //_____________________________________________________________________________ -void AliTRDtrackV1::MakeBackupTrack() +Int_t AliTRDtrackV1::MakeBackupTrack() { - // - // Creates a backup track - // +// +// Creates a backup track +// TO DO update quality check of the track. +// + + Float_t occupancy(0.); Int_t n(0), ncls(0); + for(Int_t il(AliTRDgeometry::kNlayer); il--;){ + if(!fTracklet[il]) continue; + n++; + occupancy+=fTracklet[il]->GetTBoccupancy()/AliTRDseedV1::kNtb; + ncls += fTracklet[il]->GetN(); + } + if(!n) return -1; + occupancy/=n; + + //Float_t ratio1 = Float_t(t.GetNumberOfClusters()+1) / Float_t(t.GetNExpected()+1); + + Int_t failedCutId(0); + if(GetChi2()/n > 5.0) failedCutId=1; + if(occupancy < 0.7) failedCutId=2; + //if(ratio1 > 0.6) && + //if(ratio0+ratio1 > 1.5) && + if(GetNCross() != 0) failedCutId=3; + if(TMath::Abs(GetSnp()) > 0.85) failedCutId=4; + if(ncls < 20) failedCutId=5; + + if(failedCutId){ + AliDebug(2, Form("\n" + "chi2/tracklet < 5.0 [%c] %5.2f\n" + "occupancy > 0.7 [%c] %4.2f\n" + "NCross == 0 [%c] %d\n" + "Abs(snp) < 0.85 [%c] %4.2f\n" + "NClusters > 20 [%c] %d" + ,(GetChi2()/n<5.0)?'y':'n', GetChi2()/n + ,(occupancy>0.7)?'y':'n', occupancy + ,(GetNCross()==0)?'y':'n', GetNCross() + ,(TMath::Abs(GetSnp())<0.85)?'y':'n', TMath::Abs(GetSnp()) + ,(ncls>20)?'y':'n', ncls + )); + return failedCutId; + } if(fBackupTrack) { fBackupTrack->~AliTRDtrackV1(); new(fBackupTrack) AliTRDtrackV1((AliTRDtrackV1&)(*this)); - return; + return 0; } fBackupTrack = new AliTRDtrackV1((AliTRDtrackV1&)(*this)); + return 0; } //_____________________________________________________________________________ -Int_t AliTRDtrackV1::GetProlongation(Double_t xk, Double_t &y, Double_t &z) +Int_t AliTRDtrackV1::GetProlongation(Double_t xk, Double_t &y, Double_t &z) const { // // Find a prolongation at given x - // Return 0 if it does not exist + // Return -1 if it does not exist // Double_t bz = GetBz(); - if (!AliExternalTrackParam::GetYAt(xk,bz,y)) return 0; - if (!AliExternalTrackParam::GetZAt(xk,bz,z)) return 0; + if (!AliExternalTrackParam::GetYAt(xk,bz,y)) return -1; + if (!AliExternalTrackParam::GetZAt(xk,bz,z)) return -1; return 1; @@ -530,50 +637,42 @@ Bool_t AliTRDtrackV1::PropagateTo(Double_t xk, Double_t xx0, Double_t xrho) // "xrho" - thickness*density [g/cm^2] // - if (xk == GetX()) { - return kTRUE; - } - - Double_t oldX = GetX(); - Double_t oldY = GetY(); - Double_t oldZ = GetZ(); - - Double_t bz = GetBz(); - - if (!AliExternalTrackParam::PropagateTo(xk,bz)) { - return kFALSE; - } + if (TMath::Abs(xk - GetX()) x1[%6.2f] dx[%6.2f] rho[%f]\n", xyz0[0], xyz1[0], xyz0[0]-xk, xrho/TMath::Abs(xyz0[0]-xk)); + if(xyz0[0] < xk) { xrho = -xrho; if (IsStartedTimeIntegral()) { - Double_t l2 = TMath::Sqrt((x-oldX)*(x-oldX) - + (y-oldY)*(y-oldY) - + (z-oldZ)*(z-oldZ)); - Double_t crv = AliExternalTrackParam::GetC(bz); + Double_t l2 = TMath::Sqrt((xyz1[0]-xyz0[0])*(xyz1[0]-xyz0[0]) + + (xyz1[1]-xyz0[1])*(xyz1[1]-xyz0[1]) + + (xyz1[2]-xyz0[2])*(xyz1[2]-xyz0[2])); + Double_t crv = AliExternalTrackParam::GetC(b[2]); if (TMath::Abs(l2*crv) > 0.0001) { // Make correction for curvature if neccesary - l2 = 0.5 * TMath::Sqrt((x-oldX)*(x-oldX) - + (y-oldY)*(y-oldY)); + l2 = 0.5 * TMath::Sqrt((xyz1[0]-xyz0[0])*(xyz1[0]-xyz0[0]) + + (xyz1[1]-xyz0[1])*(xyz1[1]-xyz0[1])); l2 = 2.0 * TMath::ASin(l2 * crv) / crv; - l2 = TMath::Sqrt(l2*l2 + (z-oldZ)*(z-oldZ)); + l2 = TMath::Sqrt(l2*l2 + (xyz1[2]-xyz0[2])*(xyz1[2]-xyz0[2])); } AddTimeStep(l2); } } - - if (!AliExternalTrackParam::CorrectForMeanMaterial(xx0, xrho, GetMass())) return kFALSE; + if (!AliExternalTrackParam::CorrectForMeanMaterial(xx0, xrho, fMass)) return kFALSE; { // Energy losses Double_t p2 = (1.0 + GetTgl()*GetTgl()) / (GetSigned1Pt()*GetSigned1Pt()); - Double_t beta2 = p2 / (p2 + GetMass()*GetMass()); + if (fMass<0) p2 *= 4; // q=2 + Double_t beta2 = p2 / (p2 + fMass*fMass); if ((beta2 < 1.0e-10) || ((5940.0 * beta2/(1.0 - beta2 + 1.0e-10) - beta2) < 0.0)) { return kFALSE; @@ -582,6 +681,7 @@ Bool_t AliTRDtrackV1::PropagateTo(Double_t xk, Double_t xx0, Double_t xrho) Double_t dE = 0.153e-3 / beta2 * (TMath::Log(5940.0 * beta2/(1.0 - beta2 + 1.0e-10)) - beta2) * xrho; + if (fMass<0) dE *= 4; // q=2 fBudget[0] += xrho; /* @@ -596,7 +696,7 @@ Bool_t AliTRDtrackV1::PropagateTo(Double_t xk, Double_t xx0, Double_t xrho) /* // Suspicious ! I.B. Double_t sigmade = 0.07 * TMath::Sqrt(TMath::Abs(dE)); // Energy loss fluctuation - Double_t sigmac2 = sigmade*sigmade*fC*fC*(p2+GetMass()*GetMass())/(p2*p2); + Double_t sigmac2 = sigmade*sigmade*fC*fC*(p2+fMass*fMass)/(p2*p2); fCcc += sigmac2; fCee += fX*fX * sigmac2; */ @@ -629,7 +729,7 @@ Int_t AliTRDtrackV1::PropagateToR(Double_t r,Double_t step) Double_t alpha = TMath::ATan2(xyz0[1],xyz0[0]); Rotate(alpha,kTRUE); GetXYZ(xyz0); - GetProlongation(x,y,z); + if(GetProlongation(x,y,z)<0) return -1; xyz1[0] = x * TMath::Cos(alpha) + y * TMath::Sin(alpha); xyz1[1] = x * TMath::Sin(alpha) - y * TMath::Cos(alpha); xyz1[2] = z; @@ -646,7 +746,7 @@ Int_t AliTRDtrackV1::PropagateToR(Double_t r,Double_t step) Double_t alpha = TMath::ATan2(xyz0[1],xyz0[0]); Rotate(alpha,kTRUE); GetXYZ(xyz0); - GetProlongation(r,y,z); + if(GetProlongation(r,y,z)<0) return -1; xyz1[0] = r * TMath::Cos(alpha) + y * TMath::Sin(alpha); xyz1[1] = r * TMath::Sin(alpha) - y * TMath::Cos(alpha); xyz1[2] = z; @@ -665,13 +765,13 @@ Int_t AliTRDtrackV1::PropagateToR(Double_t r,Double_t step) //_____________________________________________________________________________ void AliTRDtrackV1::Print(Option_t *o) const { + // Print track status AliInfo(Form("PID [%4.1f %4.1f %4.1f %4.1f %4.1f]", 1.E2*fPID[0], 1.E2*fPID[1], 1.E2*fPID[2], 1.E2*fPID[3], 1.E2*fPID[4])); AliInfo(Form("Material[%5.2f %5.2f %5.2f]", fBudget[0], fBudget[1], fBudget[2])); AliInfo(Form("x[%7.2f] t[%7.4f] alpha[%f] mass[%f]", GetX(), GetIntegratedLength(), GetAlpha(), fMass)); AliInfo(Form("Ntr[%1d] NtrPID[%1d] Ncl[%3d] lab[%3d]", GetNumberOfTracklets(), GetNumberOfTrackletsPID(), fN, fLab)); - if(strcmp(o, "a")!=0) return; printf("|X| = ("); const Double_t *curP = GetParameter(); for (Int_t i = 0; i < 5; i++) printf("%7.2f ", curP[i]); @@ -686,6 +786,7 @@ void AliTRDtrackV1::Print(Option_t *o) const k=-1; j--; } } + if(strcmp(o, "a")!=0) return; for(Int_t ip=0; ipGetN(); + if(fTracklet[ip] && fTrackletIndex[ip] >= 0) ncls += fTracklet[ip]->GetN(); } AliKalmanTrack::SetNumberOfClusters(ncls); } @@ -731,7 +832,7 @@ void AliTRDtrackV1::SetOwner() if(TestBit(kOwner)) return; for (Int_t ip = 0; ip < kNplane; ip++) { - if(fTrackletIndex[ip] == 0xffff) continue; + if(fTrackletIndex[ip]<0 || !fTracklet[ip]) continue; fTracklet[ip] = new AliTRDseedV1(*fTracklet[ip]); fTracklet[ip]->SetOwner(); } @@ -739,7 +840,7 @@ void AliTRDtrackV1::SetOwner() } //_______________________________________________________________ -void AliTRDtrackV1::SetTracklet(AliTRDseedV1 *trklt, Int_t index) +void AliTRDtrackV1::SetTracklet(AliTRDseedV1 *const trklt, Int_t index) { // // Set the tracklets @@ -751,40 +852,47 @@ void AliTRDtrackV1::SetTracklet(AliTRDseedV1 *trklt, Int_t index) } //_______________________________________________________________ -void AliTRDtrackV1::SetTrackLow() +void AliTRDtrackV1::SetTrackIn() { +// Save location of birth for the TRD track +// If the pointer is not valid allocate memory +// const AliExternalTrackParam *op = dynamic_cast(this); - fTrackLow = fTrackLow ? new(fTrackLow) AliExternalTrackParam(*op) : new AliExternalTrackParam(*op); + + //printf("SetTrackIn() : fTrackLow[%p]\n", (void*)fTrackLow); + if(fTrackLow){ + fTrackLow->~AliExternalTrackParam(); + new(fTrackLow) AliExternalTrackParam(*op); + } else fTrackLow = new AliExternalTrackParam(*op); } //_______________________________________________________________ -void AliTRDtrackV1::SetTrackHigh(const AliExternalTrackParam *op) +void AliTRDtrackV1::SetTrackOut(const AliExternalTrackParam *op) { +// Save location of death for the TRD track +// If the pointer is not valid allocate memory +// if(!op) op = dynamic_cast(this); - fTrackHigh = fTrackHigh ? new(fTrackHigh) AliExternalTrackParam(*op) : new AliExternalTrackParam(*op); + if(fTrackHigh){ + fTrackHigh->~AliExternalTrackParam(); + new(fTrackHigh) AliExternalTrackParam(*op); + } else fTrackHigh = new AliExternalTrackParam(*op); } //_______________________________________________________________ void AliTRDtrackV1::UnsetTracklet(Int_t plane) { - if(plane<0 && plane >= kNplane) return; - fTrackletIndex[plane] = 0xffff; - fTracklet[plane] = 0x0; + if(plane<0) return; + fTrackletIndex[plane] = -1; + fTracklet[plane] = NULL; } //_______________________________________________________________ -Bool_t AliTRDtrackV1::Update(Double_t *p, Double_t *cov, Double_t chi2) +void AliTRDtrackV1::UpdateChi2(Float_t chi2) { - // - // Update track - // - if(!AliExternalTrackParam::Update(p, cov)) return kFALSE; - - // Register info to track - SetNumberOfClusters(); +// Update chi2/track with one tracklet contribution SetChi2(GetChi2() + chi2); - return kTRUE; } //_______________________________________________________________ @@ -794,24 +902,101 @@ void AliTRDtrackV1::UpdateESDtrack(AliESDtrack *track) // Update the TRD PID information in the ESD track // - Int_t nslices = fReconstructor->IsEightSlices() ? (Int_t)AliTRDpidUtil::kNNslices : (Int_t)AliTRDpidUtil::kLQslices; - track->SetNumberOfTRDslices(nslices); - - Int_t n = 0; +// Int_t nslices = AliTRDcalibDB::Instance()->GetPIDResponse(fkReconstructor->GetRecoParam()->GetPIDmethod())->GetNumberOfSlices(); + // number of tracklets used for PID calculation + UChar_t nPID = GetNumberOfTrackletsPID(); + // number of tracklets attached to the track + UChar_t nTrk = GetNumberOfTracklets(); + // pack the two numbers together and store them in the ESD + track->SetTRDntracklets(nPID | (nTrk<<3)); + // allocate space to store raw PID signals dEdx & momentum + // independent of the method used to calculate PID (see below AliTRDPIDResponse) + track->SetNumberOfTRDslices((AliTRDseedV1::kNdEdxSlices+2)*AliTRDgeometry::kNlayer); + // store raw signals + Float_t p, sp; Double_t spd; for (Int_t ip = 0; ip < kNplane; ip++) { - if(fTrackletIndex[ip] == 0xffff) continue; - n++; - if(!fTracklet[ip]->IsOK()) continue; - fTracklet[ip]->CookdEdx(nslices); - Float_t *dedx = fTracklet[ip]->GetdEdx(); - for (Int_t js = 0; js < nslices; js++, dedx++) track->SetTRDslice(*dedx, ip, js); + if(fTrackletIndex[ip]<0 || !fTracklet[ip]) continue; + // Fill TRD dEdx info into ESD track + // a. Set Summed dEdx into the first slice + track->SetTRDslice(fTracklet[ip]->GetdQdl(), ip, 0); + // b. Set NN dEdx slices + fTracklet[ip]->CookdEdx(AliTRDPIDResponse::kNslicesNN); + const Float_t *dedx = fTracklet[ip]->GetdEdx(); + for (Int_t js(0), ks(1); js < AliTRDPIDResponse::kNslicesNN; js++, ks++, dedx++){ + if(ks>=AliTRDseedV1::kNdEdxSlices){ + AliError(Form("Exceed allocated space for dEdx slices.")); + break; + } + track->SetTRDslice(*dedx, ip, ks); + } + // fill TRD momentum info into ESD track + p = fTracklet[ip]->GetMomentum(&sp); spd = sp; + track->SetTRDmomentum(p, ip, &spd); + // store global quality per tracklet instead of momentum error + // 26.09.11 A.Bercuci + // first implementation store no. of time bins filled in tracklet (5bits see "y" bits) and + // no. of double clusters in case of pad row cross (4bits see "x" bits) + // bit map for tracklet quality xxxxyyyyy + // 27.10.11 A.Bercuci + // add chamber status bit "z" bit + // bit map for tracklet quality zxxxxyyyyy + // 12.11.11 A.Bercuci + // fit tracklet quality into the field fTRDTimeBin [Char_t] + // bit map for tracklet quality zxxyyyyy + // The information should be retrieved by the following functions of AliESDtrack for each TRD layer + // GetTRDtrkltOccupancy(layer) -> no of TB filled in tracklet + // GetTRDtrkltClCross(layer) -> no of TB filled in crossing pad rows + // IsTRDtrkltChmbGood(layer) -> status of the chamber from which the tracklet is found + Int_t nCross(fTracklet[ip]->IsRowCross()?fTracklet[ip]->GetTBcross():0); if(nCross>3) nCross = 3; + Char_t trackletQ = Char_t(fTracklet[ip]->GetTBoccupancy() | (nCross<<5) | (fTracklet[ip]->IsChmbGood()<<7)); + track->SetTRDTimBin(trackletQ, ip); } + // store PID probabilities + track->SetTRDpid(fPID); - UChar_t nPID = GetNumberOfTrackletsPID(); - if(!nPID) track->SetTRDntracklets(n); - else { - track->SetTRDpid(fPID); - nPID |= (n<<3); - track->SetTRDntracklets(nPID); + //store truncated mean + track->SetTRDsignal(fTruncatedMean); + track->SetTRDNchamberdEdx(fNchamberdEdx); + track->SetTRDNclusterdEdx(fNclusterdEdx); +} + +//_______________________________________________________________ +Double_t AliTRDtrackV1::CookTruncatedMean(const Bool_t kinvq, const Double_t mag, const Int_t charge, const Int_t kcalib, Int_t &nch, Int_t &ncls, TVectorD *Qs, TVectorD *Xs, Int_t timeBin0, Int_t timeBin1, Int_t tstep) const +{ + // + //Origin: Xianguo Lu , Marian Ivanov + // + + TVectorD arrayQ(200), arrayX(200); + ncls = AliTRDdEdxReconUtils::GetArrayClusterQ(kinvq, &arrayQ, &arrayX, this, timeBin0, timeBin1, tstep); + + const TObjArray *cobj = kcalib ? AliTRDdEdxCalibUtils::GetObj(kinvq, mag, charge) : NULL; + + const Double_t tmean = AliTRDdEdxReconUtils::ToyCook(kinvq, ncls, &arrayQ, &arrayX, cobj); + + nch = AliTRDdEdxReconUtils::UpdateArrayX(ncls, &arrayX); + + if(Qs && Xs){ + (*Qs)=arrayQ; + (*Xs)=arrayX; + } + + //printf("\ntest %.10f %d %d\n", tmean, nch, ncls); + + return tmean; +} + +//_______________________________________________________________ +TObject* AliTRDtrackV1::Clone(const char* newname) const +{ + // temporary override TObject::Clone to avoid crashes in reco + AliTRDtrackV1* src = (AliTRDtrackV1*)this; + Bool_t isown = src->IsOwner(); + AliInfo(Form("src_owner %d",isown)); + AliTRDtrackV1* dst = new AliTRDtrackV1(*src); + if (isown) { + src->SetBit(kOwner); + dst->SetOwner(); } + return dst; }