X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=ITS%2FAliITStrackV2.cxx;h=a4d86770e9bf142c443f69aa886ca6869fda3ff4;hb=c15ec07ef7f47feec905016b23322738ae39c91f;hp=eebf9da91623bd97130eb39779c9e13aa95721d7;hpb=5aa865deb2f46cf83e77e811340a302d3238b7bd;p=u%2Fmrichter%2FAliRoot.git diff --git a/ITS/AliITStrackV2.cxx b/ITS/AliITStrackV2.cxx index eebf9da9162..a4d86770e9b 100644 --- a/ITS/AliITStrackV2.cxx +++ b/ITS/AliITStrackV2.cxx @@ -13,146 +13,126 @@ * provided "as is" without express or implied warranty. * **************************************************************************/ -//------------------------------------------------------------------------- +/* $Id$ */ + +/////////////////////////////////////////////////////////////////////////// // Implementation of the ITS track class // // Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch // dEdx analysis by: Boris Batyunya, JINR, Boris.Batiounia@cern.ch -//------------------------------------------------------------------------- - -#include - +/////////////////////////////////////////////////////////////////////////// #include #include "AliCluster.h" -#include "AliESDtrack.h" +#include "AliESDVertex.h" +#include "AliITSReconstructor.h" #include "AliITStrackV2.h" +#include "AliTracker.h" +#include "AliLog.h" + +const Int_t AliITStrackV2::fgkWARN = 5; ClassImp(AliITStrackV2) -const Int_t kWARN=5; //____________________________________________________________________________ -AliITStrackV2::AliITStrackV2():AliKalmanTrack(), - fX(0), - fAlpha(0), +AliITStrackV2::AliITStrackV2() : AliKalmanTrack(), + fCheckInvariant(kTRUE), fdEdx(0), - fP0(0), - fP1(0), - fP2(0), - fP3(0), - fP4(0), - fC00(0), - fC10(0), - fC11(0), - fC20(0), - fC21(0), - fC22(0), - fC30(0), - fC31(0), - fC32(0), - fC33(0), - fC40(0), - fC41(0), - fC42(0), - fC43(0), - fC44(0), - fNUsed(0), - fNSkipped(0), - fReconstructed(kFALSE), fESDtrack(0) - { - for(Int_t i=0; i ITS track. // If c==kTRUE, create the ITS track out of the constrained params. //------------------------------------------------------------------ - SetNumberOfClusters(t.GetITSclusters(fIndex)); + const AliExternalTrackParam *par=&t; + if (c) { + par=t.GetConstrainedParam(); + if (!par) AliError("AliITStrackV2: conversion failed !\n"); + } + Set(par->GetX(),par->GetAlpha(),par->GetParameter(),par->GetCovariance()); + SetLabel(t.GetLabel()); SetMass(t.GetMass()); - - fdEdx=t.GetITSsignal(); - fAlpha = t.GetAlpha(); - if (fAlpha < -TMath::Pi()) fAlpha += 2*TMath::Pi(); - else if (fAlpha >= TMath::Pi()) fAlpha -= 2*TMath::Pi(); - - //Conversion of the track parameters - Double_t x,p[5]; - if (c) t.GetConstrainedExternalParameters(x,p); - else t.GetExternalParameters(x,p); - fX=x; x=GetConvConst(); - fP0=p[0]; - fP1=p[1]; - fP2=p[2]; - fP3=p[3]; - fP4=p[4]/x; - - //Conversion of the covariance matrix - Double_t cv[15]; - if (c) t.GetConstrainedExternalCovariance(cv); - else t.GetExternalCovariance(cv); - fC00=cv[0 ]; - fC10=cv[1 ]; fC11=cv[2 ]; - fC20=cv[3 ]; fC21=cv[4 ]; fC22=cv[5 ]; - fC30=cv[6 ]; fC31=cv[7 ]; fC32=cv[8 ]; fC33=cv[9 ]; - fC40=cv[10]/x; fC41=cv[11]/x; fC42=cv[12]/x; fC43=cv[13]/x; fC44=cv[14]/x/x; + SetNumberOfClusters(t.GetITSclusters(fIndex)); if (t.GetStatus()&AliESDtrack::kTIME) { StartTimeIntegral(); Double_t times[10]; t.GetIntegratedTimes(times); SetIntegratedTimes(times); SetIntegratedLength(t.GetIntegratedLength()); } - fESDtrack=&t; - fNUsed = 0; - fReconstructed = kFALSE; - fNSkipped =0; - for(Int_t i=0; i<6; i++) {fDy[i]=0; fDz[i]=0; fSigmaY[i]=0; fSigmaZ[i]=0;; fChi2MIP[i]=0;} - //if (!Invariant()) throw "AliITStrackV2: conversion failed !\n"; - SetFakeRatio(t.GetITSFakeRatio()); -} -void AliITStrackV2::UpdateESDtrack(ULong_t flags) { - fESDtrack->UpdateTrackParams(this,flags); - if (flags == AliESDtrack::kITSin) fESDtrack->SetITSChi2MIP(fChi2MIP); -} -void AliITStrackV2::SetConstrainedESDtrack(Double_t chi2) { - fESDtrack->SetConstrainedTrackParams(this,chi2); + for(Int_t i=0; iUpdateTrackParams(this,flags); + // copy the module indices + Int_t i; + for(i=0;i<2*AliITSgeomTGeo::kNLayers;i++) { + // printf(" %d\n",GetModuleIndex(i)); + fESDtrack->SetITSModuleIndex(i,GetModuleIndex(i)); + } + // copy the map of shared clusters + if(flags==AliESDtrack::kITSin) { + UChar_t itsSharedMap=0; + for(i=0;i0) SETBIT(itsSharedMap,i); + + } + fESDtrack->SetITSSharedMap(itsSharedMap); + } - fC00=t.fC00; - fC10=t.fC10; fC11=t.fC11; - fC20=t.fC20; fC21=t.fC21; fC22=t.fC22; - fC30=t.fC30; fC31=t.fC31; fC32=t.fC32; fC33=t.fC33; - fC40=t.fC40; fC41=t.fC41; fC42=t.fC42; fC43=t.fC43; fC44=t.fC44; + // copy the 4 dedx samples + Double_t sdedx[4]={0.,0.,0.,0.}; + for(i=0; i<4; i++) sdedx[i]=fdEdxSample[i]; + fESDtrack->SetITSdEdxSamples(sdedx); +} - Int_t n=GetNumberOfClusters(); - for (Int_t i=0; iGet1Pt()); - //Double_t c =TMath::Abs(Get1Pt()); - Double_t co=t->GetSigmaY2()*t->GetSigmaZ2()*TMath::Sqrt(TMath::Abs(fP4)); - Double_t c =GetSigmaY2()*GetSigmaZ2()*TMath::Sqrt(TMath::Abs(fP4)); + //Double_t co=OneOverPt(); + //Double_t c =OneOverPt(); + Double_t co=t->GetSigmaY2()*t->GetSigmaZ2(); + Double_t c =GetSigmaY2()*GetSigmaZ2(); if (c>co) return 1; else if (c= 0.9999) { - Int_t n=GetNumberOfClusters(); - if (n>kWARN) - Warning("GetGlobalXYZat","Propagation failed (%d) !\n",n); - return 0; - } - - Double_t r1=sqrt(1.- f1*f1), r2=sqrt(1.- f2*f2); - - Double_t yk = fP0 + dx*(f1+f2)/(r1+r2); - Double_t zk = fP1 + dx*(f1+f2)/(f1*r2 + f2*r1)*fP3; - - Double_t cs=TMath::Cos(fAlpha), sn=TMath::Sin(fAlpha); - x = xk*cs - yk*sn; - y = xk*sn + yk*cs; - z = zk; - - return 1; + Double_t r[3]; + Bool_t rc=GetXYZAt(xloc, GetBz(), r); + x=r[0]; y=r[1]; z=r[2]; + return rc; } //_____________________________________________________________________________ -Double_t AliITStrackV2::GetPredictedChi2(const AliCluster *c) const -{ +Double_t AliITStrackV2::GetPredictedChi2(const AliCluster *c) const { //----------------------------------------------------------------- // This function calculates a predicted chi2 increment. //----------------------------------------------------------------- - Double_t r00=c->GetSigmaY2(), r01=0., r11=c->GetSigmaZ2(); - r00+=fC00; r01+=fC10; r11+=fC11; - // - Double_t det=r00*r11 - r01*r01; - if (TMath::Abs(det) < 1.e-30) { - Int_t n=GetNumberOfClusters(); - if (n>kWARN) - Warning("GetPredictedChi2","Singular matrix (%d) !\n",n); - return 1e10; - } - Double_t tmp=r00; r00=r11; r11=tmp; r01=-r01; - - Double_t dy=c->GetY() - fP0, dz=c->GetZ() - fP1; - - return (dy*r00*dy + 2*r01*dy*dz + dz*r11*dz)/det; + Double_t p[2]={c->GetY(), c->GetZ()}; + Double_t cov[3]={c->GetSigmaY2(), 0., c->GetSigmaZ2()}; + return AliExternalTrackParam::GetPredictedChi2(p,cov); } //____________________________________________________________________________ -Int_t AliITStrackV2::CorrectForMaterial(Double_t d, Double_t x0) { +Bool_t AliITStrackV2::PropagateTo(Double_t xk, Double_t d, Double_t x0) { //------------------------------------------------------------------ - //This function corrects the track parameters for crossed material + //This function propagates a track //------------------------------------------------------------------ - Double_t p2=(1.+ GetTgl()*GetTgl())/(Get1Pt()*Get1Pt()); - Double_t beta2=p2/(p2 + GetMass()*GetMass()); - d*=TMath::Sqrt((1.+ fP3*fP3)/(1.- fP2*fP2)); - //Multiple scattering****************** - if (d!=0) { - Double_t theta2=14.1*14.1/(beta2*p2*1e6)*TMath::Abs(d); - //Double_t theta2=1.0259e-6*14*14/28/(beta2*p2)*TMath::Abs(d)*9.36*2.33; - fC22 += theta2*(1.- fP2*fP2)*(1. + fP3*fP3); - fC33 += theta2*(1. + fP3*fP3)*(1. + fP3*fP3); - fC43 += theta2*fP3*fP4*(1. + fP3*fP3); - fC44 += theta2*fP3*fP4*fP3*fP4; - } - - //Energy losses************************ - if (x0!=0.) { - d*=x0; - Double_t dE=0.153e-3/beta2*(log(5940*beta2/(1-beta2)) - beta2)*d; - if (beta2/(1-beta2)>3.5*3.5) - dE=0.153e-3/beta2*(log(3.5*5940)+0.5*log(beta2/(1-beta2)) - beta2)*d; - fP4*=(1.- sqrt(p2+GetMass()*GetMass())/p2*dE); + Double_t oldX=GetX(), oldY=GetY(), oldZ=GetZ(); + + //Double_t bz=GetBz(); + //if (!AliExternalTrackParam::PropagateTo(xk,bz)) return kFALSE; + Double_t b[3]; GetBxByBz(b); + if (!AliExternalTrackParam::PropagateToBxByBz(xk,b)) return kFALSE; + Double_t xOverX0,xTimesRho; + xOverX0 = d; xTimesRho = d*x0; + if (!CorrectForMeanMaterial(xOverX0,xTimesRho,kTRUE)) return kFALSE; + + Double_t x=GetX(), y=GetY(), z=GetZ(); + if (IsStartedTimeIntegral() && x>oldX) { + Double_t l2 = (x-oldX)*(x-oldX) + (y-oldY)*(y-oldY) + (z-oldZ)*(z-oldZ); + AddTimeStep(TMath::Sqrt(l2)); } - if (!Invariant()) return 0; - - return 1; + return kTRUE; } //____________________________________________________________________________ -Int_t AliITStrackV2::PropagateTo(Double_t xk, Double_t d, Double_t x0) { - //------------------------------------------------------------------ - //This function propagates a track - //------------------------------------------------------------------ - Double_t x1=fX, x2=xk, dx=x2-x1; - Double_t f1=fP2, f2=f1 + fP4*dx; - if (TMath::Abs(f2) >= 0.9999) { - Int_t n=GetNumberOfClusters(); - if (n>kWARN) - Warning("PropagateTo","Propagation failed !\n",n); - return 0; - } +Bool_t AliITStrackV2::PropagateToTGeo(Double_t xToGo, Int_t nstep, Double_t &xOverX0, Double_t &xTimesRho, Bool_t addTime) { + //------------------------------------------------------------------- + // Propagates the track to a reference plane x=xToGo in n steps. + // These n steps are only used to take into account the curvature. + // The material is calculated with TGeo. (L.Gaudichet) + //------------------------------------------------------------------- + + Double_t startx = GetX(), starty = GetY(), startz = GetZ(); + Double_t sign = (startxoldX) { - Double_t l2 = (fX-oldX)*(fX-oldX)+(fP0-oldY)*(fP0-oldY)+ - (fP1-oldZ)*(fP1-oldZ); + if (addTime && IsStartedTimeIntegral() && GetX()>startx) { + Double_t l2 = ( (GetX()-startx)*(GetX()-startx) + + (GetY()-starty)*(GetY()-starty) + + (GetZ()-startz)*(GetZ()-startz) ); AddTimeStep(TMath::Sqrt(l2)); } - // - return 1; + return kTRUE; } //____________________________________________________________________________ -Int_t AliITStrackV2::Update(const AliCluster* c, Double_t chi2, UInt_t index) { +Bool_t AliITStrackV2::Update(const AliCluster* c, Double_t chi2, Int_t index) +{ //------------------------------------------------------------------ //This function updates track parameters //------------------------------------------------------------------ - Double_t p0=fP0,p1=fP1,p2=fP2,p3=fP3,p4=fP4; - Double_t c00=fC00; - Double_t c10=fC10, c11=fC11; - Double_t c20=fC20, c21=fC21, c22=fC22; - Double_t c30=fC30, c31=fC31, c32=fC32, c33=fC33; - Double_t c40=fC40, c41=fC41, c42=fC42, c43=fC43, c44=fC44; - - - Double_t r00=c->GetSigmaY2(), r01=0., r11=c->GetSigmaZ2(); - r00+=fC00; r01+=fC10; r11+=fC11; - Double_t det=r00*r11 - r01*r01; - Double_t tmp=r00; r00=r11/det; r11=tmp/det; r01=-r01/det; - - - Double_t k00=fC00*r00+fC10*r01, k01=fC00*r01+fC10*r11; - Double_t k10=fC10*r00+fC11*r01, k11=fC10*r01+fC11*r11; - Double_t k20=fC20*r00+fC21*r01, k21=fC20*r01+fC21*r11; - Double_t k30=fC30*r00+fC31*r01, k31=fC30*r01+fC31*r11; - Double_t k40=fC40*r00+fC41*r01, k41=fC40*r01+fC41*r11; - - Double_t dy=c->GetY() - fP0, dz=c->GetZ() - fP1; - Int_t layer = (index & 0xf0000000) >> 28; - fDy[layer] = dy; - fDz[layer] = dz; - fSigmaY[layer] = TMath::Sqrt(c->GetSigmaY2()+fC00); - fSigmaZ[layer] = TMath::Sqrt(c->GetSigmaZ2()+fC11); - - Double_t sf=fP2 + k20*dy + k21*dz; - - fP0 += k00*dy + k01*dz; - fP1 += k10*dy + k11*dz; - fP2 = sf; - fP3 += k30*dy + k31*dz; - fP4 += k40*dy + k41*dz; - - Double_t c01=fC10, c02=fC20, c03=fC30, c04=fC40; - Double_t c12=fC21, c13=fC31, c14=fC41; - - fC00-=k00*fC00+k01*fC10; fC10-=k00*c01+k01*fC11; - fC20-=k00*c02+k01*c12; fC30-=k00*c03+k01*c13; - fC40-=k00*c04+k01*c14; + Double_t p[2]={c->GetY(), c->GetZ()}; + Double_t cov[3]={c->GetSigmaY2(), c->GetSigmaYZ(), c->GetSigmaZ2()}; - fC11-=k10*c01+k11*fC11; - fC21-=k10*c02+k11*c12; fC31-=k10*c03+k11*c13; - fC41-=k10*c04+k11*c14; - - fC22-=k20*c02+k21*c12; fC32-=k20*c03+k21*c13; - fC42-=k20*c04+k21*c14; - - fC33-=k30*c03+k31*c13; - fC43-=k30*c04+k31*c14; - - fC44-=k40*c04+k41*c14; + if (!AliExternalTrackParam::Update(p,cov)) return kFALSE; + Int_t n=GetNumberOfClusters(); if (!Invariant()) { - fP0=p0; fP1=p1; fP2=p2; fP3=p3; fP4=p4; - fC00=c00; - fC10=c10; fC11=c11; - fC20=c20; fC21=c21; fC22=c22; - fC30=c30; fC31=c31; fC32=c32; fC33=c33; - fC40=c40; fC41=c41; fC42=c42; fC43=c43; fC44=c44; - return 0; + if (n>fgkWARN) AliDebug(1,"Wrong invariant !"); + return kFALSE; } - if (chi2<0) return 1; + if (chi2<0) return kTRUE; + + // fill residuals for ITS+TPC tracks + if (fESDtrack) { + if (fESDtrack->GetStatus()&AliESDtrack::kTPCin) { + AliTracker::FillResiduals(this,p,cov,c->GetVolumeId()); + } + } - Int_t n=GetNumberOfClusters(); fIndex[n]=index; SetNumberOfClusters(n+1); SetChi2(GetChi2()+chi2); - return 1; + return kTRUE; } -Int_t AliITStrackV2::Invariant() const { +Bool_t AliITStrackV2::Invariant() const { //------------------------------------------------------------------ // This function is for debugging purpose only //------------------------------------------------------------------ + if(!fCheckInvariant) return kTRUE; + Int_t n=GetNumberOfClusters(); - - if (TMath::Abs(fP2)>=0.9999){ - if (n>kWARN) Warning("Invariant","fP2=%f\n",fP2); - return 0; + static Float_t bz = GetBz(); + // take into account the misalignment error + Float_t maxMisalErrY2=0,maxMisalErrZ2=0; + //RS + const AliITSRecoParam* recopar = AliITSReconstructor::GetRecoParam(); + if (!recopar) recopar = AliITSRecoParam::GetHighFluxParam(); + + for (Int_t lay=0; layGetClusterMisalErrorY(lay,bz)); + maxMisalErrZ2 = TMath::Max(maxMisalErrZ2,recopar->GetClusterMisalErrorZ(lay,bz)); } - if (fC00<=0 || fC00>9.) { - if (n>kWARN) Warning("Invariant","fC00=%f\n",fC00); - return 0; + maxMisalErrY2 *= maxMisalErrY2; + maxMisalErrZ2 *= maxMisalErrZ2; + // this is because when we reset before refitting, we multiply the + // matrix by 10 + maxMisalErrY2 *= 10.; + maxMisalErrZ2 *= 10.; + + Double_t sP2=GetParameter()[2]; + if (TMath::Abs(sP2) >= kAlmost1){ + if (n>fgkWARN) AliDebug(1,Form("fP2=%f\n",sP2)); + return kFALSE; } - if (fC11<=0 || fC11>9.) { - if (n>kWARN) Warning("Invariant","fC11=%f\n",fC11); - return 0; + Double_t sC00=GetCovariance()[0]; + if (sC00<=0 || sC00>(9.+maxMisalErrY2)) { + if (n>fgkWARN) AliDebug(1,Form("fC00=%f\n",sC00)); + return kFALSE; } - if (fC22<=0 || fC22>1.) { - if (n>kWARN) Warning("Invariant","fC22=%f\n",fC22); - return 0; + Double_t sC11=GetCovariance()[2]; + if (sC11<=0 || sC11>(9.+maxMisalErrZ2)) { + if (n>fgkWARN) AliDebug(1,Form("fC11=%f\n",sC11)); + return kFALSE; } - if (fC33<=0 || fC33>1.) { - if (n>kWARN) Warning("Invariant","fC33=%f\n",fC33); - return 0; + Double_t sC22=GetCovariance()[5]; + if (sC22<=0 || sC22>1.) { + if (n>fgkWARN) AliDebug(1,Form("fC22=%f\n",sC22)); + return kFALSE; } - if (fC44<=0 || fC44>6e-5) { - if (n>kWARN) Warning("Invariant","fC44=%f\n",fC44); - return 0; + Double_t sC33=GetCovariance()[9]; + if (sC33<=0 || sC33>1.) { + if (n>fgkWARN) AliDebug(1,Form("fC33=%f\n",sC33)); + return kFALSE; } - return 1; + Double_t sC44=GetCovariance()[14]; + if (sC44<=0 /*|| sC44>6e-5*/) { + if (n>fgkWARN) AliDebug(1,Form("fC44=%f\n",sC44)); + return kFALSE; + } + + return kTRUE; } //____________________________________________________________________________ -Int_t AliITStrackV2::Propagate(Double_t alp,Double_t xk) { +Bool_t AliITStrackV2::Propagate(Double_t alp,Double_t xk) { //------------------------------------------------------------------ //This function propagates a track //------------------------------------------------------------------ - Double_t alpha=fAlpha, x=fX; - Double_t p0=fP0,p1=fP1,p2=fP2,p3=fP3,p4=fP4; - Double_t c00=fC00; - Double_t c10=fC10, c11=fC11; - Double_t c20=fC20, c21=fC21, c22=fC22; - Double_t c30=fC30, c31=fC31, c32=fC32, c33=fC33; - Double_t c40=fC40, c41=fC41, c42=fC42, c43=fC43, c44=fC44; - - if (alp < -TMath::Pi()) alp += 2*TMath::Pi(); - else if (alp >= TMath::Pi()) alp -= 2*TMath::Pi(); - Double_t ca=TMath::Cos(alp-fAlpha), sa=TMath::Sin(alp-fAlpha); - Double_t sf=fP2, cf=TMath::Sqrt(1.- fP2*fP2); - - TMatrixD *tT=0; - // **** rotation ********************** - { - fAlpha = alp; - fX = x*ca + p0*sa; - fP0= -x*sa + p0*ca; - fP2= sf*ca - cf*sa; - - TMatrixD cC(5,5); - cC(0,0)=c00; - cC(1,0)=c10; cC(1,1)=c11; - cC(2,0)=c20; cC(2,1)=c21; cC(2,2)=c22; - cC(3,0)=c30; cC(3,1)=c31; cC(3,2)=c32; cC(3,3)=c33; - cC(4,0)=c40; cC(4,1)=c41; cC(4,2)=c42; cC(4,3)=c43; cC(4,4)=c44; - cC(0,1)=cC(1,0); - cC(0,2)=cC(2,0); cC(1,2)=cC(2,1); - cC(0,3)=cC(3,0); cC(1,3)=cC(3,1); cC(2,3)=cC(3,2); - cC(0,4)=cC(4,0); cC(1,4)=cC(4,1); cC(2,4)=cC(4,2); cC(3,4)=cC(4,3); - - TMatrixD mF(6,5); - mF(0,0)=sa; - mF(1,0)=ca; - mF(2,1)=mF(4,3)=mF(5,4)=1; - mF(3,2)=ca + sf/cf*sa; - - TMatrixD tmp(cC,TMatrixD::kMult,TMatrixD(TMatrixD::kTransposed, mF)); - tT=new TMatrixD(mF,TMatrixD::kMult,tmp); - } - - // **** translation ****************** - { - Double_t dx=xk-fX; - Double_t f1=fP2, f2=f1 + fP4*dx; - if (TMath::Abs(f2) >= 0.9999) { - Int_t n=GetNumberOfClusters(); - if (n>kWARN) - Warning("Propagate","Propagation failed (%d) !\n",n); - return 0; - } - Double_t r1=TMath::Sqrt(1.- f1*f1), r2=TMath::Sqrt(1.- f2*f2); - - fX=xk; - fP0 += dx*(f1+f2)/(r1+r2); - fP1 += dx*(f1+f2)/(f1*r2 + f2*r1)*fP3; - fP2 += dx*fP4; - - TMatrixD mF(5,6); - mF(0,1)=mF(1,2)=mF(2,3)=mF(3,4)=mF(4,5)=1; - mF(0,3)=dx/(r1+r2)*(2+(f1+f2)*(f2/r2+f1/r1)/(r1+r2)); - mF(0,5)=dx*dx/(r1+r2)*(1+(f1+f2)*f2/(r1+r2)); - mF(1,3)=dx*fP3/(f1*r2 + f2*r1)*(2-(f1+f2)*(r2-f1*f2/r2+r1-f2*f1/r1)/(f1*r2 + f2*r1)); - mF(1,4)=dx*(f1+f2)/(f1*r2 + f2*r1); - mF(1,5)=dx*dx*fP3/(f1*r2 + f2*r1)*(1-(f1+f2)*(-f1*f2/r2+r1)/(f1*r2 + f2*r1)); - mF(2,5)=dx; - mF(0,0)=-1/(r1+r2)*((f1+f2)+dx*fP4*(1+(f1+f2)/(r1+r2)*f2/r2)); - mF(1,0)=-fP3/(f1*r2 + f2*r1)*((f1+f2)+dx*fP4*(1+(f1+f2)/(f1*r2 + f2*r1)*(f1*f2/r2-r1))); - mF(2,0)=-fP4; - - TMatrixD tmp(*tT,TMatrixD::kMult,TMatrixD(TMatrixD::kTransposed, mF)); - delete tT; - TMatrixD cC(mF,TMatrixD::kMult,tmp); - - fC00=cC(0,0); - fC10=cC(1,0); fC11=cC(1,1); - fC20=cC(2,0); fC21=cC(2,1); fC22=cC(2,2); - fC30=cC(3,0); fC31=cC(3,1); fC32=cC(3,2); fC33=cC(3,3); - fC40=cC(4,0); fC41=cC(4,1); fC42=cC(4,2); fC43=cC(4,3); fC44=cC(4,4); + //Double_t bz=GetBz(); + //if (!AliExternalTrackParam::Propagate(alp,xk,bz)) return kFALSE; + Double_t b[3]; GetBxByBz(b); + if (!AliExternalTrackParam::PropagateBxByBz(alp,xk,b)) return kFALSE; if (!Invariant()) { - fAlpha=alpha; - fX=x; - fP0=p0; fP1=p1; fP2=p2; fP3=p3; fP4=p4; - fC00=c00; - fC10=c10; fC11=c11; - fC20=c20; fC21=c21; fC22=c22; - fC30=c30; fC31=c31; fC32=c32; fC33=c33; - fC40=c40; fC41=c41; fC42=c42; fC43=c43; fC44=c44; - return 0; - } + Int_t n=GetNumberOfClusters(); + if (n>fgkWARN) AliDebug(1,"Wrong invariant !"); + return kFALSE; } - return 1; + return kTRUE; } -Double_t AliITStrackV2::GetD(Double_t x, Double_t y) const { - //------------------------------------------------------------------ - // This function calculates the transverse impact parameter - // with respect to a point with global coordinates (x,y) - //------------------------------------------------------------------ - Double_t xt=fX, yt=fP0; +Bool_t AliITStrackV2::MeanBudgetToPrimVertex(Double_t xyz[3], Double_t step, Double_t &d) const { - Double_t sn=TMath::Sin(fAlpha), cs=TMath::Cos(fAlpha); - Double_t a = x*cs + y*sn; - y = -x*sn + y*cs; x=a; - xt-=x; yt-=y; + //------------------------------------------------------------------- + // Get the mean material budget between the actual point and the + // primary vertex. (L.Gaudichet) + //------------------------------------------------------------------- - sn=fP4*xt - fP2; cs=fP4*yt + TMath::Sqrt(1.- fP2*fP2); - a=2*(xt*fP2 - yt*TMath::Sqrt(1.- fP2*fP2))-fP4*(xt*xt + yt*yt); - if (fP4<0) a=-a; - return a/(1 + TMath::Sqrt(sn*sn + cs*cs)); -} + Double_t cs=TMath::Cos(GetAlpha()), sn=TMath::Sin(GetAlpha()); + Double_t vertexX = xyz[0]*cs + xyz[1]*sn; -Double_t AliITStrackV2::GetZat(Double_t x) const { - //------------------------------------------------------------------ - // This function calculates the z at given x point - in current coordinate system - //------------------------------------------------------------------ - Double_t x1=fX, x2=x, dx=x2-x1; - // - Double_t f1=fP2, f2=f1 + fP4*dx; - if (TMath::Abs(f2) >= 0.9999) { - return 10000000; - } + Int_t nstep = Int_t((GetX()-vertexX)/step); + if (nstep<1) nstep = 1; + step = (GetX()-vertexX)/nstep; - Double_t r1=sqrt(1.- f1*f1), r2=sqrt(1.- f2*f2); - Double_t z = fP1 + dx*(f1+f2)/(f1*r2 + f2*r1)*fP3; - return z; -} + // Double_t mparam[7], densMean=0, radLength=0, length=0; + Double_t mparam[7]; + Double_t p1[3], p2[3], x = GetX(), bz = GetBz(); + GetXYZ(p1); + d=0.; + for (Int_t i=0; i900000) return kFALSE; + d += mparam[1]; + p1[0] = p2[0]; + p1[1] = p2[1]; + p1[2] = p2[2]; + } + + return kTRUE; +} -Int_t AliITStrackV2::Improve(Double_t x0,Double_t xyz[3],Double_t ers[3]) { +Bool_t AliITStrackV2::Improve(Double_t x0,Double_t xyz[3],Double_t ers[3]) { //------------------------------------------------------------------ - //This function improves angular track parameters + //This function improves angular track parameters //------------------------------------------------------------------ - Double_t cs=TMath::Cos(fAlpha), sn=TMath::Sin(fAlpha); - //Double_t xv = xyz[0]*cs + xyz[1]*sn; // vertex - Double_t yv =-xyz[0]*sn + xyz[1]*cs; // in the - Double_t zv = xyz[2]; // local frame - Double_t dy=fP0-yv, dz=fP1-zv; - Double_t r2=fX*fX+dy*dy; - Double_t p2=(1.+ GetTgl()*GetTgl())/(Get1Pt()*Get1Pt()); + //Store the initail track parameters + + Double_t x = GetX(); + Double_t alpha = GetAlpha(); + Double_t par[] = {GetY(),GetZ(),GetSnp(),GetTgl(),GetSigned1Pt()}; + Double_t cov[] = { + GetSigmaY2(), + GetSigmaZY(), + GetSigmaZ2(), + GetSigmaSnpY(), + GetSigmaSnpZ(), + GetSigmaSnp2(), + GetSigmaTglY(), + GetSigmaTglZ(), + GetSigmaTglSnp(), + GetSigmaTgl2(), + GetSigma1PtY(), + GetSigma1PtZ(), + GetSigma1PtSnp(), + GetSigma1PtTgl(), + GetSigma1Pt2() + }; + + + Double_t cs=TMath::Cos(GetAlpha()), sn=TMath::Sin(GetAlpha()); + Double_t xv = xyz[0]*cs + xyz[1]*sn; // vertex + Double_t yv =-xyz[0]*sn + xyz[1]*cs; // in the + Double_t zv = xyz[2]; // local frame + + Double_t dx = x - xv, dy = par[0] - yv, dz = par[1] - zv; + Double_t r2=dx*dx + dy*dy; + Double_t p2=(1.+ GetTgl()*GetTgl())/(GetSigned1Pt()*GetSigned1Pt()); Double_t beta2=p2/(p2 + GetMass()*GetMass()); x0*=TMath::Sqrt((1.+ GetTgl()*GetTgl())/(1.- GetSnp()*GetSnp())); Double_t theta2=14.1*14.1/(beta2*p2*1e6)*x0; //Double_t theta2=1.0259e-6*14*14/28/(beta2*p2)*x0*9.36*2.33; + + Double_t bz=GetBz(); + Double_t cnv=bz*kB2C; + Double_t curv=GetC(bz); { - Double_t parp=0.5*(fP4*fX + dy*TMath::Sqrt(4/r2-fP4*fP4)); - Double_t sigma2p = theta2*(1.- GetSnp()*GetSnp())*(1. + GetTgl()*GetTgl()); - sigma2p += fC00/r2*(1.- dy*dy/r2)*(1.- dy*dy/r2); - sigma2p += ers[1]*ers[1]/r2; - sigma2p += 0.25*fC44*fX*fX; - Double_t eps2p=sigma2p/(fC22+sigma2p); - fP0 += fC20/(fC22+sigma2p)*(parp-fP2); - fP2 = eps2p*fP2 + (1-eps2p)*parp; - fC22 *= eps2p; - fC20 *= eps2p; + Double_t dummy = 4/r2 - curv*curv; + if (dummy < 0) return kFALSE; + Double_t parp = 0.5*(curv*dx + dy*TMath::Sqrt(dummy)); + Double_t sigma2p = theta2*(1.-GetSnp())*(1.+GetSnp())*(1. + GetTgl()*GetTgl()); + Double_t ovSqr2 = 1./TMath::Sqrt(r2); + Double_t tfact = ovSqr2*(1.-dy*ovSqr2)*(1.+dy*ovSqr2); + sigma2p += cov[0]*tfact*tfact; + sigma2p += ers[1]*ers[1]/r2; + sigma2p += 0.25*cov[14]*cnv*cnv*dx*dx; + Double_t eps2p=sigma2p/(cov[5] + sigma2p); + par[0] += cov[3]/(cov[5] + sigma2p)*(parp - GetSnp()); + par[2] = eps2p*GetSnp() + (1 - eps2p)*parp; + cov[5] *= eps2p; + cov[3] *= eps2p; } { - Double_t parl=0.5*fP4*dz/TMath::ASin(0.5*fP4*TMath::Sqrt(r2)); - Double_t sigma2l=theta2; - sigma2l += fC11/r2+fC00*dy*dy*dz*dz/(r2*r2*r2); - sigma2l += ers[2]*ers[2]/r2; - Double_t eps2l=sigma2l/(fC33+sigma2l); - fP1 += fC31/(fC33+sigma2l)*(parl-fP3); - fP4 += fC43/(fC33+sigma2l)*(parl-fP3); - fP3 = eps2l*fP3 + (1-eps2l)*parl; - fC33 *= eps2l; fC43 *= eps2l; - fC31 *= eps2l; - } - if (!Invariant()) return 0; - return 1; -} + Double_t parl=0.5*curv*dz/TMath::ASin(0.5*curv*TMath::Sqrt(r2)); + Double_t sigma2l=theta2; + sigma2l += cov[2]/r2 + cov[0]*dy*dy*dz*dz/(r2*r2*r2); + sigma2l += ers[2]*ers[2]/r2; + Double_t eps2l = sigma2l/(cov[9] + sigma2l); + par[1] += cov[7 ]/(cov[9] + sigma2l)*(parl - par[3]); + par[4] += cov[13]/(cov[9] + sigma2l)*(parl - par[3]); + par[3] = eps2l*par[3] + (1-eps2l)*parl; + cov[9] *= eps2l; + cov[13]*= eps2l; + cov[7] *= eps2l; + } -/* -Int_t AliITStrackV2::Improve(Double_t x0,Double_t yv,Double_t zv) { - //------------------------------------------------------------------ - //This function improves angular track parameters - //------------------------------------------------------------------ - Double_t dy=fP0-yv, dz=fP1-zv; - Double_t r2=fX*fX+dy*dy; - Double_t p2=(1.+ GetTgl()*GetTgl())/(Get1Pt()*Get1Pt()); - Double_t beta2=p2/(p2 + GetMass()*GetMass()); - x0*=TMath::Sqrt((1.+ GetTgl()*GetTgl())/(1.- GetSnp()*GetSnp())); - //Double_t theta2=14.1*14.1/(beta2*p2*1e6)*x0; - Double_t theta2=1.0259e-6*14*14/28/(beta2*p2)*x0*9.36*2.33; - - Double_t par=0.5*(fP4*fX + dy*TMath::Sqrt(4/r2-fP4*fP4)); - Double_t sigma2 = theta2*(1.- GetSnp()*GetSnp())*(1. + GetTgl()*GetTgl()); - sigma2 += fC00/r2*(1.- dy*dy/r2)*(1.- dy*dy/r2); - sigma2 += kSigmaYV*kSigmaYV/r2; - sigma2 += 0.25*fC44*fX*fX; - Double_t eps2=sigma2/(fC22+sigma2), eps=TMath::Sqrt(eps2); - if (10*r2*fC44>28; - if (idx>1) nc++; // Take only SSD and SDD + Float_t dedx[4]; + for (Int_t il=0; il<4; il++) { // count good (>0) dE/dx values + if(fdEdxSample[il]>0.){ + dedx[nc]= fdEdxSample[il]; + nc++; + } + } + if(nc<1){ + SetdEdx(0.); + return; } - Int_t swap;//stupid sorting + Int_t swap; // sort in ascending order do { swap=0; - for (i=0; i0) dedx /= (nu-nl); - SetdEdx(dedx); + Double_t sumamp=0,sumweight=0; + Double_t weight[4]={1.,1.,0.,0.}; + if(nc==3) weight[1]=0.5; + else if(nc<3) weight[1]=0.; + for (Int_t i=0; i r) { + if (r>1e-1) return kFALSE; + r = TMath::Abs(d); + } + + Double_t rcurr=TMath::Sqrt(GetX()*GetX() + GetY()*GetY()); + if (TMath::Abs(d) > rcurr) return kFALSE; + Double_t globXYZcurr[3]; GetXYZ(globXYZcurr); + Double_t phicurr=TMath::ATan2(globXYZcurr[1],globXYZcurr[0]); + + if (GetX()>=0.) { + phi=phicurr+TMath::ASin(d/r)-TMath::ASin(d/rcurr); + } else { + phi=phicurr+TMath::ASin(d/r)+TMath::ASin(d/rcurr)-TMath::Pi(); + } + + // return a phi in [0,2pi[ + if (phi<0.) phi+=2.*TMath::Pi(); + else if (phi>=2.*TMath::Pi()) phi-=2.*TMath::Pi(); + z=GetZ()+GetTgl()*(TMath::Sqrt((r-d)*(r+d))-TMath::Sqrt((rcurr-d)*(rcurr+d))); + return kTRUE; +} +//____________________________________________________________________________ +Bool_t AliITStrackV2:: +GetLocalXat(Double_t r,Double_t &xloc) const { + //------------------------------------------------------------------ + // This function returns the local x of the track + // position estimated at the radius r. + // The track curvature is neglected. + //------------------------------------------------------------------ + Double_t d=GetD(0.,0.); + if (TMath::Abs(d) > r) { + if (r>1e-1) return kFALSE; + r = TMath::Abs(d); + } + + Double_t rcurr=TMath::Sqrt(GetX()*GetX() + GetY()*GetY()); + Double_t globXYZcurr[3]; GetXYZ(globXYZcurr); + Double_t phicurr=TMath::ATan2(globXYZcurr[1],globXYZcurr[0]); + Double_t phi; + if (GetX()>=0.) { + phi=phicurr+TMath::ASin(d/r)-TMath::ASin(d/rcurr); + } else { + phi=phicurr+TMath::ASin(d/r)+TMath::ASin(d/rcurr)-TMath::Pi(); + } + + xloc=r*(TMath::Cos(phi)*TMath::Cos(GetAlpha()) + +TMath::Sin(phi)*TMath::Sin(GetAlpha())); + + return kTRUE; +} + +//____________________________________________________________________________ +Bool_t AliITStrackV2::ImproveKalman(Double_t xyz[3],Double_t ers[3], const Double_t* xlMS, const Double_t* x2X0MS, Int_t nMS) +{ + // Substitute the state of the track (p_{k|k},C_{k|k}) at the k-th measumerent by its + // smoothed value from the k-th measurement + measurement at the vertex. + // Account for the MS on nMS layers at x-postions xlMS with x/x0 = x2X0MS + // p_{k|kv} = p_{k|k} + C_{k|k}*D^Tr_{k+1} B^{-1}_{k+1} ( vtx - D_{k+1}*p_{k|k}) + // C_{k|kv} = C_{k|k}*( I - D^Tr_{k+1} B^{-1}_{k+1} D_{k+1} C^Tr_{k|k}) + // + // where D_{k} = H_{k} F_{k} with H being the matrix converting the tracks parameters + // to measurements m_{k} = H_{k} p_{k} and F_{k} the matrix propagating the track between the + // the point k-1 and k: p_{k|k-1} = F_{k} p_{k-1|k-1} + // + // B_{k+1} = V_{k+1} + H_{k+1} C_{k+1|k} H^Tr_{k+1} with V_{k+1} being the error of the measurment + // at point k+1 (i.e. vertex), and C_{k+1|k} - error matrix extrapolated from k-th measurement to + // k+1 (vtx) and accounting for the MS inbetween + // + // H = {{1,0,0,0,0},{0,1,0,0,0}} + // + double covc[15], *cori = (double*) GetCovariance(),par[5] = {GetY(),GetZ(),GetSnp(),GetTgl(),GetSigned1Pt()}, + &c00=cori[0], + &c01=cori[1],&c11=cori[2], + &c02=cori[3],&c12=cori[4],&c22=cori[5], + &c03=cori[6],&c13=cori[7],&c23=cori[8],&c33=cori[9], + &c04=cori[10],&c14=cori[11],&c24=cori[12],&c34=cori[13],&c44=cori[14], + // for smoothed cov matrix + &cov00=covc[0], + &cov01=covc[1],&cov11=covc[2], + &cov02=covc[3],&cov12=covc[4],&cov22=covc[5], + &cov03=covc[6],&cov13=covc[7],&cov23=covc[8],&cov33=covc[9], + &cov04=covc[10],&cov14=covc[11],&cov24=covc[12],&cov34=covc[13],&cov44=covc[14]; + // + double x = GetX(), alpha = GetAlpha(); + // vertex in the track frame + double cs=TMath::Cos(alpha), sn=TMath::Sin(alpha); + double xv = xyz[0]*cs + xyz[1]*sn, yv =-xyz[0]*sn + xyz[1]*cs, zv = xyz[2]; + double dx = xv - GetX(); + if (TMath::Abs(dx)<=kAlmost0) return kTRUE; + // + double cnv=GetBz()*kB2C, x2r=cnv*par[4]*dx, f1=par[2], f2=f1+x2r; + if (TMath::Abs(f1) >= kAlmost1 || TMath::Abs(f2) >= kAlmost1) { + AliInfo(Form("Fail: %+e %+e",f1,f2)); + return kFALSE; + } + double r1=TMath::Sqrt((1.-f1)*(1.+f1)), r2=TMath::Sqrt((1.-f2)*(1.+f2)), dx2r=dx/(r1+r2); + // elements of matrix F_{k+1} (1s on diagonal) + double f02 = 2*dx2r, f04 = cnv*dx*dx2r, f13/*, f24 = cnv*dx*/; + if (TMath::Abs(x2r)<0.05) f13 = dx*r2+f2*(f1+f2)*dx2r; // see AliExternalTrackParam::PropagateTo + else { + double dy2dx = (f1+f2)/(r1+r2); + f13 = 2*TMath::ASin(0.5*TMath::Sqrt(1+dy2dx*dy2dx)*x2r)/(cnv*par[4]); + } + // elements of matrix D_{k+1} = H_{k+1} * F_{k+1} + // double d00 = 1., d11 = 1.; + double &d02 = f02, &d04 = f04, &d13 = f13; + // + // elements of matrix DC = D_{k+1}*C_{kk}^T + double dc00 = c00+c02*d02+c04*d04, dc10 = c01+c03*d13; + double dc01 = c01+c12*d02+c14*d04, dc11 = c11+c13*d13; + double dc02 = c02+c22*d02+c24*d04, dc12 = c12+c23*d13; + double dc03 = c03+c23*d02+c34*d04, dc13 = c13+c33*d13; + double dc04 = c04+c24*d02+c44*d04, dc14 = c14+c34*d13; + // + // difference between the vertex and the the track extrapolated to vertex + yv -= par[0] + par[2]*d02 + par[4]*d04; + zv -= par[1] + par[3]*d13; + // + // y,z part of the cov.matrix extrapolated to vtx (w/o MS contribution) + // C_{k+1,k} = H F_{k+1} C_{k,k} F^Tr_{k+1} H^Tr = D C D^Tr + double cv00 = dc00+dc02*d02+dc04*d04, cv01 = dc01+dc03*d13, cv11 = dc11+dc13*d13; + // + // add MS contribution layer by layer + double xCurr = x; + double p2Curr = par[2]; + // + // precalculated factors of MS contribution matrix: + double ms22t = (1. + par[3]*par[3]); + double ms33t = ms22t*ms22t; + double p34 = par[3]*par[4]; + double ms34t = p34*ms22t; + double ms44t = p34*p34; + // + double p2=(1.+ par[3]*par[3])/(par[4]*par[4]); + double beta2 = p2/(p2+GetMass()*GetMass()); + double theta2t = 14.1*14.1/(beta2*p2*1e6) * (1. + par[3]*par[3]); + // + // account for the MS in the layers between the last measurement and the vertex + for (int il=0;il= kAlmost1 || TMath::Abs(f2L) >= kAlmost1) { + AliInfo(Form("FailMS at step %d of %d: dfx:%e dxL:%e %e %e",il,nMS,dfx,dxL,f1L,f2L)); + return kFALSE; + } + double r1L=TMath::Sqrt((1.-f1L)*(1.+f1L)), r2L=TMath::Sqrt((1.-f2L)*(1.+f2L)), dx2rL=dxL/(r1L+r2L); + // elements of matrix for propagation from scatering layer to vertex + double f02L = 2*dx2rL, f04L = cnv*dxL*dx2rL, f13L/*, f24L = cnv*dxL*/; + if (TMath::Abs(x2rL)<0.05) f13L = dxL*r2L+f2L*(f1L+f2L)*dx2rL; // see AliExternalTrackParam::PropagateTo + else { + double dy2dxL = (f1L+f2L)/(r1L+r2L); + f13L = 2*TMath::ASin(0.5*TMath::Sqrt(1+dy2dxL*dy2dxL)*x2rL)/(cnv*par[4]); + } + // MS contribution matrix: + double theta2 = theta2t*TMath::Abs(x2X0MS[il]); + double ms22 = theta2*(1.-p2Curr)*(1.+p2Curr)*ms22t; + double ms33 = theta2*ms33t; + double ms34 = theta2*ms34t; + double ms44 = theta2*ms44t; + // + // add H F MS F^Tr H^Tr to cv + cv00 += f02L*f02L*ms22 + f04L*f04L*ms44; + cv01 += f04L*f13L*ms34; + cv11 += f13L*f13L*ms33; + } + // + // inverse of matrix B + double b11 = ers[1]*ers[1] + cv00; + double b00 = ers[2]*ers[2] + cv11; + double det = b11*b00 - cv01*cv01; + if (TMath::Abs(det)