X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=TRD%2FAliTRDtrack.cxx;h=4df4749a96c677a1fffd8b2b275a94049c2e6bf5;hb=ac1dec3ba1bf78fccb97fd2c6d8b6519e491fd25;hp=3efe196aa324700ec2d4dfca0b172c362ffce017;hpb=73ae7b59de1b5494d1e7c09deaf4a243be997f80;p=u%2Fmrichter%2FAliRoot.git diff --git a/TRD/AliTRDtrack.cxx b/TRD/AliTRDtrack.cxx index 3efe196aa32..4df4749a96c 100644 --- a/TRD/AliTRDtrack.cxx +++ b/TRD/AliTRDtrack.cxx @@ -13,751 +13,1171 @@ * provided "as is" without express or implied warranty. * **************************************************************************/ -/* -$Log$ -Revision 1.23 2003/07/22 15:56:14 hristov -Implementing ESD functionality in the NewIO (Yu.Belikov) +/* $Id$ */ -Revision 1.20.2.1 2003/07/14 09:19:33 hristov -TOF included in the combined PID (Yu.Belikov) +#include -Revision 1.20 2003/05/27 17:46:13 hristov -TRD PID included in the ESD schema (T.Kuhr) +#include "AliTracker.h" -Revision 1.19 2003/05/22 10:46:46 hristov -Using access methods instead of data members - -Revision 1.18 2003/04/10 10:36:54 hristov -Code for unified TPC/TRD tracking (S.Radomski) - -Revision 1.17 2003/02/19 09:02:28 hristov -Track time measurement (S.Radomski) - -Revision 1.16 2003/02/10 14:06:10 cblume -Add tracking without tilted pads as option - -Revision 1.15 2003/01/27 16:34:49 cblume -Update of tracking by Sergei and Chuncheng - -Revision 1.14 2002/11/07 15:52:09 cblume -Update of tracking code for tilted pads +#include "AliTRDgeometry.h" +#include "AliTRDcluster.h" +#include "AliTRDtrack.h" +#include "AliTRDcalibDB.h" +#include "Cal/AliTRDCalPID.h" -Revision 1.13 2002/10/22 15:53:08 alibrary -Introducing Riostream.h +ClassImp(AliTRDtrack) -Revision 1.12 2002/10/14 14:57:44 hristov -Merging the VirtualMC branch to the main development branch (HEAD) +/////////////////////////////////////////////////////////////////////////////// +// // +// Represents a reconstructed TRD track // +// Local TRD Kalman track // +// // +/////////////////////////////////////////////////////////////////////////////// -Revision 1.8.10.2 2002/07/24 10:09:31 alibrary -Updating VirtualMC +//_____________________________________________________________________________ +AliTRDtrack::AliTRDtrack() + :AliKalmanTrack() + ,fSeedLab(-1) + ,fdEdx(0) + ,fDE(0) + ,fPIDquality(0) + ,fClusterOwner(kFALSE) + ,fPIDmethod(kLQ) + ,fStopped(kFALSE) + ,fLhElectron(0) + ,fNWrong(0) + ,fNRotate(0) + ,fNCross(0) + ,fNExpected(0) + ,fNLast(0) + ,fNExpectedLast(0) + ,fNdedx(0) + ,fChi2Last(1e10) + ,fBackupTrack(0x0) +{ + // + // AliTRDtrack default constructor + // -RRevision 1.11 2002/06/13 12:09:58 hristov -Minor corrections + for (Int_t i = 0; i < kNplane; i++) { + for (Int_t j = 0; j < kNslice; j++) { + fdEdxPlane[i][j] = 0.0; + } + fTimBinPlane[i] = -1; + fMom[i] = -1.; + fSnp[i] = 0.; + fTgl[i] = 0.; + fTrackletIndex[i] = -1; + } + for(int ispec=0; ispec -#include + Set(x,alpha,pp,cc); + SetNumberOfClusters(1); + fIndex[0] = index; + fClusters[0] = c; -#include "AliTRDgeometry.h" -#include "AliTRDcluster.h" -#include "AliTRDtrack.h" -#include "../TPC/AliTPCtrack.h" -#include "AliESDtrack.h" + for (Int_t i = 0; i < kNplane; i++) { + for (Int_t j = 0; j < kNslice; j++) { + fdEdxPlane[i][j] = 0.0; + } + fTimBinPlane[i] = -1; + fMom[i] = -1.; + fSnp[i] = 0.; + fTgl[i] = 0.; + fTrackletIndex[i] = -1; + } + for(int ispec=0; ispecGetQ()); + Double_t s = GetSnp(); + Double_t t = GetTgl(); + if (s*s < 1) { + q *= TMath::Sqrt((1-s*s)/(1+t*t)); + } -ClassImp(AliTRDtrack) + fdQdl[0] = q; + for (UInt_t i = 1; i < kMAXCLUSTERSPERTRACK; i++) { + fdQdl[i] = 0; + fIndex[i] = 0; + fIndexBackup[i] = 0; + fClusters[i] = 0x0; + } + for (Int_t i = 0; i < 3;i++) { + fBudget[i] = 0; + } +} + //_____________________________________________________________________________ +AliTRDtrack::AliTRDtrack(const AliTRDtrack &t/*, const Bool_t owner*/) + :AliKalmanTrack(t) + ,fSeedLab(t.GetSeedLabel()) + ,fdEdx(t.fdEdx) + ,fDE(t.fDE) + ,fPIDquality(t.fPIDquality) + ,fClusterOwner(kTRUE) + ,fPIDmethod(t.fPIDmethod) + ,fStopped(t.fStopped) + ,fLhElectron(0) + ,fNWrong(t.fNWrong) + ,fNRotate(t.fNRotate) + ,fNCross(t.fNCross) + ,fNExpected(t.fNExpected) + ,fNLast(t.fNLast) + ,fNExpectedLast(t.fNExpectedLast) + ,fNdedx(t.fNdedx) + ,fChi2Last(t.fChi2Last) + ,fBackupTrack(0x0) +{ + // + // Copy constructor. + // -AliTRDtrack::AliTRDtrack(const AliTRDcluster *c, UInt_t index, - const Double_t xx[5], const Double_t cc[15], - Double_t xref, Double_t alpha) : AliKalmanTrack() { - //----------------------------------------------------------------- - // This is the main track constructor. - //----------------------------------------------------------------- + for (Int_t i = 0; i < kNplane; i++) { + for (Int_t j = 0; j < kNslice; j++) { + fdEdxPlane[i][j] = t.fdEdxPlane[i][j]; + } + fTimBinPlane[i] = t.fTimBinPlane[i]; + fTracklets[i] = t.fTracklets[i]; + fMom[i] = t.fMom[i]; + fSnp[i] = t.fSnp[i]; + fTgl[i] = t.fTgl[i]; + fTrackletIndex[i] = t.fTrackletIndex[i]; + } - fSeedLab = -1; + Int_t n = t.GetNumberOfClusters(); + SetNumberOfClusters(n); - fAlpha=alpha; - if (fAlpha<-TMath::Pi()) fAlpha += 2*TMath::Pi(); - if (fAlpha>=TMath::Pi()) fAlpha -= 2*TMath::Pi(); + for (Int_t i = 0; i < n; i++) { + fIndex[i] = t.fIndex[i]; + fIndexBackup[i] = t.fIndex[i]; + fdQdl[i] = t.fdQdl[i]; + if (fClusterOwner && t.fClusters[i]) { + fClusters[i] = new AliTRDcluster(*(t.fClusters[i])); + } + else { + fClusters[i] = t.fClusters[i]; + } + } - fX=xref; + for (UInt_t i = n; i < kMAXCLUSTERSPERTRACK; i++) { + fdQdl[i] = 0; + fIndex[i] = 0; + fIndexBackup[i] = 0; + fClusters[i] = 0x0; + } - fY=xx[0]; fZ=xx[1]; fE=xx[2]; fT=xx[3]; fC=xx[4]; + for (Int_t i = 0; i < 3;i++) { + fBudget[i] = t.fBudget[i]; + } - fCyy=cc[0]; - fCzy=cc[1]; fCzz=cc[2]; - fCey=cc[3]; fCez=cc[4]; fCee=cc[5]; - fCty=cc[6]; fCtz=cc[7]; fCte=cc[8]; fCtt=cc[9]; - fCcy=cc[10]; fCcz=cc[11]; fCce=cc[12]; fCct=cc[13]; fCcc=cc[14]; - - fIndex[0]=index; - SetNumberOfClusters(1); + for(Int_t ispec = 0; ispecGetQ()); - Double_t s = fX*fC - fE, t=fT; - if(s*s < 1) q *= TMath::Sqrt((1-s*s)/(1+t*t)); + for (Int_t i = 0; i < kNplane; i++) { + for (Int_t j = 0; j < kNslice; j++) { + fdEdxPlane[i][j] = 0.0; + } + fTimBinPlane[i] = -1; + fMom[i] = -1.; + fSnp[i] = 0.; + fTgl[i] = 0.; + fTrackletIndex[i] = -1; + } + for(int ispec=0; ispecGetX() + ,par->GetAlpha() + ,par->GetParameter() + ,par->GetCovariance()); + + for (UInt_t i = 0; i < kMAXCLUSTERSPERTRACK; i++) { + fdQdl[i] = 0; + fClusters[i] = 0x0; + } - Int_t n=t.GetNumberOfClusters(); - SetNumberOfClusters(n); - for (Int_t i=0; i= TMath::Pi()) fAlpha -= 2*TMath::Pi(); +//____________________________________________________________________________ +Float_t AliTRDtrack::StatusForTOF() +{ + // + // Defines the status of the TOF extrapolation + // - Double_t x, p[5]; t.GetExternalParameters(x,p); + // Definition of res ???? + Float_t res = (0.2 + 0.8 * (fN / (fNExpected + 5.0))) + * (0.4 + 0.6 * fTracklets[5].GetN() / 20.0); + res *= (0.25 + 0.8 * 40.0 / (40.0 + fBudget[2])); + return res; - fX=x; +} - x = GetConvConst(); +//_____________________________________________________________________________ +Int_t AliTRDtrack::Compare(const TObject *o) const +{ + // + // Compares tracks according to their Y2 or curvature + // - fY=p[0]; - fZ=p[1]; - fT=p[3]; - fC=p[4]/x; - fE=fC*fX - p[2]; + AliTRDtrack *t = (AliTRDtrack *) o; - //Conversion of the covariance matrix - Double_t c[15]; t.GetExternalCovariance(c); + Double_t co = TMath::Abs(t->GetC()); + Double_t c = TMath::Abs(GetC()); - c[10]/=x; c[11]/=x; c[12]/=x; c[13]/=x; c[14]/=x*x; + if (c > co) { + return 1; + } + else if (c < co) { + return -1; + } - Double_t c22=fX*fX*c[14] - 2*fX*c[12] + c[5]; - Double_t c32=fX*c[13] - c[8]; - Double_t c20=fX*c[10] - c[3], c21=fX*c[11] - c[4], c42=fX*c[14] - c[12]; + return 0; - fCyy=c[0 ]; - fCzy=c[1 ]; fCzz=c[2 ]; - fCey=c20; fCez=c21; fCee=c22; - fCty=c[6 ]; fCtz=c[7 ]; fCte=c32; fCtt=c[9 ]; - fCcy=c[10]; fCcz=c[11]; fCce=c42; fCct=c[13]; fCcc=c[14]; +} - // Initialization [SR, GSI, 18.02.2003] - for(UInt_t i=0; i= TMath::Pi()) fAlpha -= 2*TMath::Pi(); +} - Double_t x, p[5]; t.GetExternalParameters(x,p); +//_____________________________________________________________________________ +void AliTRDtrack::CookdEdxTimBin(const Int_t/* tid*/) +{ + // + // Set fdEdxPlane and fTimBinPlane and also get the + // Time bin for Max. Cluster + // + // Authors: + // Prashant Shukla (shukla@physi.uni-heidelberg.de) + // Alexandru Bercuci (A.Bercuci@gsi.de) + // - fX=x; + // Max charge in chamber + Double_t maxcharge[kNplane]; + // Number of clusters attached to track per chamber and slice + Int_t nCluster[kNplane][kNslice]; + // Number of time bins in chamber + Int_t ntb = AliTRDcalibDB::Instance()->GetNumberOfTimeBins(); + Int_t plane; // Plane of current cluster + Int_t tb; // Time bin of current cluster + Int_t slice; // Current slice + AliTRDcluster *cluster = 0x0; // Pointer to current cluster + + // Reset class and local counters/variables + for (Int_t iPlane = 0; iPlane < kNplane; iPlane++) { + fTimBinPlane[iPlane] = -1; + maxcharge[iPlane] = 0.0; + for (Int_t iSlice = 0; iSlice < kNslice; iSlice++) { + fdEdxPlane[iPlane][iSlice] = 0.0; + nCluster[iPlane][iSlice] = 0; + } + } - x = GetConvConst(); + // Start looping over clusters attached to this track + for (Int_t iClus = 0; iClus < GetNumberOfClusters(); iClus++) { - fY=p[0]; - fZ=p[1]; - fT=p[3]; - fC=p[4]/x; - fE=fC*fX - p[2]; + cluster = fClusters[iClus]; //(AliTRDcluster*)tracker->GetCluster(fIndex[iClus]); + if (!cluster) continue; - //Conversion of the covariance matrix - Double_t c[15]; t.GetExternalCovariance(c); + // Read info from current cluster + plane = AliTRDgeometry::GetLayer(cluster->GetDetector()); + if (plane < 0 || plane >= kNplane) { + AliError(Form("Wrong plane %d", plane)); + continue; + } - c[10]/=x; c[11]/=x; c[12]/=x; c[13]/=x; c[14]/=x*x; + tb = cluster->GetLocalTimeBin(); + if ((tb < 0) || (tb >= ntb)) { + AliWarning(Form("time bin < 0 or > %d in cluster %d", ntb, iClus)); + AliInfo(Form("dQ/dl %f", fdQdl[iClus])); + continue; + } + + slice = tb * kNslice / ntb; - Double_t c22=fX*fX*c[14] - 2*fX*c[12] + c[5]; - Double_t c32=fX*c[13] - c[8]; - Double_t c20=fX*c[10] - c[3], c21=fX*c[11] - c[4], c42=fX*c[14] - c[12]; + fdEdxPlane[plane][slice] += fdQdl[iClus]; + if (fdQdl[iClus] > maxcharge[plane]) { + maxcharge[plane] = fdQdl[iClus]; + fTimBinPlane[plane] = tb; + } - fCyy=c[0 ]; - fCzy=c[1 ]; fCzz=c[2 ]; - fCey=c20; fCez=c21; fCee=c22; - fCty=c[6 ]; fCtz=c[7 ]; fCte=c32; fCtt=c[9 ]; - fCcy=c[10]; fCcz=c[11]; fCce=c42; fCct=c[13]; fCcc=c[14]; + nCluster[plane][slice]++; - // Initialization [SR, GSI, 18.02.2003] - for(UInt_t i=0; iSetLabel(GetLabel()); - track->SetX(fX, fAlpha); - track->SetNClusters(GetNumberOfClusters(), GetChi2()); - track->SetNWrongClusters(fNWrong); - track->SetNRotate(fNRotate); - Double_t times[10]; - GetIntegratedTimes(times); - track->SetTime(times, GetIntegratedLength()); - - track->SetMass(GetMass()); - track->SetdEdX(GetdEdx()); - - GetExternalParameters(xr, vec); - track->SetStateVector(vec); - - GetExternalCovariance(cov); - track->SetCovarianceMatrix(cov); + + //number of time bins in chamber + Int_t ntb = AliTRDcalibDB::Instance()->GetNumberOfTimeBins(); + + Int_t plane; // plane of current cluster + Int_t tb; // time bin of current cluster + Int_t slice; // curent slice + AliTRDcluster *cluster = 0x0; // pointer to current cluster + const Int_t kMLPscale = 16000; // scaling of the MLP input to be smaller than 1 + + // Reset class and local contors/variables + for (Int_t iPlane = 0; iPlane < kNplane; iPlane++){ + for (Int_t iSlice = 0; iSlice < kNMLPslice; iSlice++) { + *(dedx + (kNMLPslice * iPlane) + iSlice) = 0.0; + } + } + + // Start looping over clusters attached to this track + for (Int_t iClus = 0; iClus < GetNumberOfClusters(); iClus++) { + + cluster = fClusters[iClus]; //(AliTRDcluster*)tracker->GetCluster(fIndex[iClus]); + if (!cluster) { + continue; + } + + // Read info from current cluster + plane = AliTRDgeometry::GetLayer(cluster->GetDetector()); + if (plane < 0 || plane >= kNplane) { + AliError(Form("Wrong plane %d",plane)); + continue; + } + + tb = cluster->GetLocalTimeBin(); + if (tb == 0 || tb >= ntb) { + AliWarning(Form("time bin 0 or > %d in cluster %d",ntb,iClus)); + AliInfo(Form("dQ/dl %f",fdQdl[iClus])); + continue; + } + + slice = tb * kNMLPslice / ntb; + + *(dedx+(kNMLPslice * plane) + slice) += fdQdl[iClus]/kMLPscale; + + } // End of loop over cluster + } -//____________________________________________________________________________ -void AliTRDtrack::GetExternalParameters(Double_t& xr, Double_t x[5]) const { + +//_____________________________________________________________________________ +void AliTRDtrack::SetTrackSegmentDirMom(const Int_t plane) +{ // - // This function returns external TRD track representation + // Set the momenta for a track segment in a given plane // - xr=fX; - x[0]=GetY(); - x[1]=GetZ(); - x[2]=GetSnp(); - x[3]=GetTgl(); - x[4]=Get1Pt(); -} + + if ((plane < 0) || + (plane>= kNplane)) { + AliError(Form("Trying to access out of range plane (%d)", plane)); + return; + } + + fSnp[plane] = GetSnp(); + fTgl[plane] = GetTgl(); + Double_t p[3]; + GetPxPyPz(p); + fMom[plane] = TMath::Sqrt(p[0]*p[0] + p[1]*p[1] + p[2]*p[2]); + +} //_____________________________________________________________________________ -void AliTRDtrack::GetExternalCovariance(Double_t cc[15]) const { +Float_t AliTRDtrack::GetTrackLengthPlane(Int_t plane) const +{ // - // This function returns external representation of the covriance matrix. + // Calculate the track length of a track segment in a given plane // - Double_t a=GetConvConst(); - Double_t c22=fX*fX*fCcc-2*fX*fCce+fCee; - Double_t c32=fX*fCct-fCte; - Double_t c20=fX*fCcy-fCey, c21=fX*fCcz-fCez, c42=fX*fCcc-fCce; + if ((plane < 0) || + (plane >= kNplane)) { + return 0.0; + } - cc[0 ]=fCyy; - cc[1 ]=fCzy; cc[2 ]=fCzz; - cc[3 ]=c20; cc[4 ]=c21; cc[5 ]=c22; - cc[6 ]=fCty; cc[7 ]=fCtz; cc[8 ]=c32; cc[9 ]=fCtt; - cc[10]=fCcy*a; cc[11]=fCcz*a; cc[12]=c42*a; cc[13]=fCct*a; cc[14]=fCcc*a*a; - -} - + return (AliTRDgeometry::AmThick() + AliTRDgeometry::DrThick()) + / TMath::Sqrt((1.0 - fSnp[plane]*fSnp[plane]) + / (1.0 + fTgl[plane]*fTgl[plane])); + +} //_____________________________________________________________________________ -void AliTRDtrack::GetCovariance(Double_t cc[15]) const { +Bool_t AliTRDtrack::CookPID(Int_t &pidQuality) +{ + // + // This function calculates the PID probabilities based on TRD signals + // + // The method produces probabilities based on the charge + // and the position of the maximum time bin in each layer. + // The dE/dx information can be used as global charge or 2 to 3 + // slices. Check AliTRDCalPID and AliTRDCalPIDRefMaker for the actual + // implementation. + // + // Author + // Alex Bercuci (A.Bercuci@gsi.de) 2nd May 2007 + // - cc[0]=fCyy; - cc[1]=fCzy; cc[2]=fCzz; - cc[3]=fCey; cc[4]=fCez; cc[5]=fCee; - cc[6]=fCcy; cc[7]=fCcz; cc[8]=fCce; cc[9]=fCcc; - cc[10]=fCty; cc[11]=fCtz; cc[12]=fCte; cc[13]=fCct; cc[14]=fCtt; - -} + AliTRDcalibDB *calibration = AliTRDcalibDB::Instance(); + if (!calibration) { + AliError("No access to calibration data"); + return kFALSE; + } + + // Retrieve the CDB container class with the probability distributions + const AliTRDCalPID *pd = calibration->GetPIDObject(AliTRDReconstructor::kLQPID); + if (!pd) { + AliError("No access to AliTRDCalPID"); + return kFALSE; + } -//_____________________________________________________________________________ -Int_t AliTRDtrack::Compare(const TObject *o) const { + // Calculate the input for the NN if fPIDmethod is kNN + Float_t ldEdxNN[AliTRDgeometry::kNlayer * kNMLPslice], *dedx = 0x0; + if(fPIDmethod == kNN) { + CookdEdxNN(&ldEdxNN[0]); + } -// Compares tracks according to their Y2 or curvature + // Sets the a priori probabilities + for(int ispec=0; ispecGetSigmaY2(); - // Double_t c =GetSigmaY2(); + pidQuality = 0; + Float_t length; // track segment length in chamber + + // Skip tracks which have no TRD signal at all + if (fdEdx == 0.) return kFALSE; + + for (Int_t iPlane = 0; iPlane < AliTRDgeometry::kNlayer; iPlane++) { + + length = (AliTRDgeometry::AmThick() + AliTRDgeometry::DrThick()) + / TMath::Sqrt((1.0 - fSnp[iPlane]*fSnp[iPlane]) + / (1.0 + fTgl[iPlane]*fTgl[iPlane])); + + // check data + if((fdEdxPlane[iPlane][0] + fdEdxPlane[iPlane][1] + fdEdxPlane[iPlane][2]) <= 0. + || fTimBinPlane[iPlane] == -1.) continue; + + // this track segment has fulfilled all requierments + pidQuality++; + + // Get the probabilities for the different particle species + for (Int_t iSpecies = 0; iSpecies < AliPID::kSPECIES; iSpecies++) { + switch(fPIDmethod){ + case kLQ: + dedx = fdEdxPlane[iPlane]; + break; + case kNN: + dedx = &ldEdxNN[iPlane*kNMLPslice]; + break; + } + fPID[iSpecies] *= pd->GetProbability(iSpecies, fMom[iPlane], dedx, length, iPlane); + } - Double_t co=TMath::Abs(t->GetC()); - Double_t c =TMath::Abs(GetC()); + } - if (c>co) return 1; - else if (c 0.0001) { + // Make correction for curvature if neccesary + l2 = 0.5 * TMath::Sqrt((x-oldX)*(x-oldX) + + (y-oldY)*(y-oldY)); + l2 = 2.0 * TMath::ASin(l2 * crv) / crv; + l2 = TMath::Sqrt(l2*l2 + (z-oldZ)*(z-oldZ)); + } + AddTimeStep(l2); + } + } + + if (!AliExternalTrackParam::CorrectForMeanMaterial(xx0,xrho,GetMass())) { + return kFALSE; + } - do { - swap=0; - for (i=0; i 0.8*kinE) dE = 0.8 * kinE; // + if (dE < 0) dE = 0.0; // Not valid region for Bethe bloch + */ + + fDE += dE; + + /* + // 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); + fCcc += sigmac2; + fCee += fX*fX * sigmac2; + */ - SetdEdx(dedx); -} + } + return kTRUE; + +} //_____________________________________________________________________________ -Int_t AliTRDtrack::PropagateTo(Double_t xk,Double_t x0,Double_t rho) +Bool_t AliTRDtrack::Update(const AliTRDcluster *c, Double_t chisq + , Int_t index, Double_t h01) { - // Propagates a track of particle with mass=pm to a reference plane - // defined by x=xk through media of density=rho and radiationLength=x0 + // + // Assignes the found cluster to the track and updates + // track information. + // : predicted chi2 + // : ?? + // : Tilting factor + // - if (xk == fX) return 1; + Double_t p[2] = { c->GetY() + , c->GetZ() }; + Double_t sy2 = c->GetSigmaY2() * 4.0; + Double_t sz2 = c->GetSigmaZ2() * 4.0; + Double_t cov[3] = { sy2 + h01*h01*sz2 + , h01 * (sy2-sz2) + , sz2 + h01*h01*sy2 }; - if (TMath::Abs(fC*xk - fE) >= 0.99999) { - Int_t n=GetNumberOfClusters(); - if (n>4) cerr << n << " AliTRDtrack: Propagation failed, \tPt = " - << GetPt() << "\t" << GetLabel() << "\t" << GetMass() << endl; - return 0; + if (!AliExternalTrackParam::Update(p,cov)) { + return kFALSE; } - // track Length measurement [SR, GSI, 17.02.2003] - Double_t oldX = fX, oldY = fY, oldZ = fZ; - - Double_t x1=fX, x2=x1+(xk-x1), dx=x2-x1, y1=fY, z1=fZ; - Double_t c1=fC*x1 - fE; - if((c1*c1) > 1) return 0; - Double_t r1=sqrt(1.- c1*c1); - Double_t c2=fC*x2 - fE; - if((c2*c2) > 1) return 0; - Double_t r2=sqrt(1.- c2*c2); - - fY += dx*(c1+c2)/(r1+r2); - fZ += dx*(c1+c2)/(c1*r2 + c2*r1)*fT; - - //f = F - 1 - Double_t rr=r1+r2, cc=c1+c2, xx=x1+x2; - Double_t f02=-dx*(2*rr + cc*(c1/r1 + c2/r2))/(rr*rr); - Double_t f04= dx*(rr*xx + cc*(c1*x1/r1+c2*x2/r2))/(rr*rr); - Double_t cr=c1*r2+c2*r1; - Double_t f12=-dx*fT*(2*cr + cc*(c2*c1/r1-r1 + c1*c2/r2-r2))/(cr*cr); - Double_t f13= dx*cc/cr; - Double_t f14=dx*fT*(cr*xx-cc*(r1*x2-c2*c1*x1/r1+r2*x1-c1*c2*x2/r2))/(cr*cr); - - //b = C*ft - Double_t b00=f02*fCey + f04*fCcy, b01=f12*fCey + f14*fCcy + f13*fCty; - Double_t b10=f02*fCez + f04*fCcz, b11=f12*fCez + f14*fCcz + f13*fCtz; - Double_t b20=f02*fCee + f04*fCce, b21=f12*fCee + f14*fCce + f13*fCte; - Double_t b30=f02*fCte + f04*fCct, b31=f12*fCte + f14*fCct + f13*fCtt; - Double_t b40=f02*fCce + f04*fCcc, b41=f12*fCce + f14*fCcc + f13*fCct; - - //a = f*b = f*C*ft - Double_t a00=f02*b20+f04*b40,a01=f02*b21+f04*b41,a11=f12*b21+f14*b41+f13*b31; - - //F*C*Ft = C + (a + b + bt) - fCyy += a00 + 2*b00; - fCzy += a01 + b01 + b10; - fCey += b20; - fCty += b30; - fCcy += b40; - fCzz += a11 + 2*b11; - fCez += b21; - fCtz += b31; - fCcz += b41; - - fX=x2; - - //Multiple scattering ****************** - Double_t d=sqrt((x1-fX)*(x1-fX)+(y1-fY)*(y1-fY)+(z1-fZ)*(z1-fZ)); - Double_t p2=(1.+ GetTgl()*GetTgl())/(Get1Pt()*Get1Pt()); - Double_t beta2=p2/(p2 + GetMass()*GetMass()); - Double_t theta2=14.1*14.1/(beta2*p2*1e6)*d/x0*rho; - - Double_t ey=fC*fX - fE, ez=fT; - Double_t xz=fC*ez, zz1=ez*ez+1, xy=fE+ey; - - fCee += (2*ey*ez*ez*fE+1-ey*ey+ez*ez+fE*fE*ez*ez)*theta2; - fCte += ez*zz1*xy*theta2; - fCtt += zz1*zz1*theta2; - fCce += xz*ez*xy*theta2; - fCct += xz*zz1*theta2; - fCcc += xz*xz*theta2; - - //Energy losses************************ - if((5940*beta2/(1-beta2+1e-10) - beta2) < 0) return 0; - - Double_t dE=0.153e-3/beta2*(log(5940*beta2/(1-beta2+1e-10)) - beta2)*d*rho; - if (x1 < x2) dE=-dE; - cc=fC; - fC*=(1.- sqrt(p2+GetMass()*GetMass())/p2*dE); - fE+=fX*(fC-cc); - - // track time measurement [SR, GSI 17.02.2002] - if (IsStartedTimeIntegral()) { - Double_t l2 = (fX-oldX)*(fX-oldX) + (fY-oldY)*(fY-oldY) + (fZ-oldZ)*(fZ-oldZ); - AddTimeStep(TMath::Sqrt(l2)); - } - - return 1; -} + Int_t n = GetNumberOfClusters(); + fIndex[n] = index; + SetNumberOfClusters(n+1); + + SetChi2(GetChi2()+chisq); + return kTRUE; + +} //_____________________________________________________________________________ -Int_t AliTRDtrack::Update(const AliTRDcluster *c, Double_t chisq, UInt_t index, Double_t h01) +Int_t AliTRDtrack::UpdateMI(AliTRDcluster *c, Double_t chisq, Int_t index + , Double_t h01, Int_t /*plane*/, Int_t /*tid*/) { - // Assignes found cluster to the track and updates track information + // + // Assignes the found cluster to the track and + // updates track information + // : predicted chi2 + // : ?? + // : Tilting factor + // + // Difference to Update(AliTRDcluster *c): cluster is added to fClusters + // - Bool_t fNoTilt = kTRUE; - if(TMath::Abs(h01) > 0.003) fNoTilt = kFALSE; + Double_t p[2] = { c->GetY() + , c->GetZ() }; + Double_t sy2 = c->GetSigmaY2() * 4.0; + Double_t sz2 = c->GetSigmaZ2() * 4.0; + Double_t cov[3] = { sy2 + h01*h01*sz2 + , h01 * (sy2-sz2) + , sz2 + h01*h01*sy2 }; - Double_t r00=c->GetSigmaY2(), r01=0., r11=c->GetSigmaZ2(); - r00+=fCyy; r01+=fCzy; r11+=fCzz; - Double_t det=r00*r11 - r01*r01; - Double_t tmp=r00; r00=r11/det; r11=tmp/det; r01=-r01/det; + if (!AliExternalTrackParam::Update(p,cov)) { + return kFALSE; + } + + AliTracker::FillResiduals(this,p,cov,c->GetVolumeId()); - Double_t k00=fCyy*r00+fCzy*r01, k01=fCyy*r01+fCzy*r11; - Double_t k10=fCzy*r00+fCzz*r01, k11=fCzy*r01+fCzz*r11; - Double_t k20=fCey*r00+fCez*r01, k21=fCey*r01+fCez*r11; - Double_t k30=fCty*r00+fCtz*r01, k31=fCty*r01+fCtz*r11; - Double_t k40=fCcy*r00+fCcz*r01, k41=fCcy*r01+fCcz*r11; + // Register cluster to track + Int_t n = GetNumberOfClusters(); + fIndex[n] = index; + fClusters[n] = c; + SetNumberOfClusters(n+1); + SetChi2(GetChi2() + chisq); - Double_t dy=c->GetY() - fY, dz=c->GetZ() - fZ; - Double_t cur=fC + k40*dy + k41*dz, eta=fE + k20*dy + k21*dz; + return kTRUE; - Double_t c01=fCzy, c02=fCey, c03=fCty, c04=fCcy; - Double_t c12=fCez, c13=fCtz, c14=fCcz; +} - if(fNoTilt) { - if (TMath::Abs(cur*fX-eta) >= 0.99999) { - Int_t n=GetNumberOfClusters(); - if (n>4) cerr<GetSigmaY2(), r01=0., r11=c->GetSigmaZ2()*xu_factor; - r00+=(fCyy+2.0*h01*fCzy+h01*h01*fCzz); - r01+=(fCzy+h01*fCzz); - det=r00*r11 - r01*r01; - tmp=r00; r00=r11/det; r11=tmp/det; r01=-r01/det; - - k00=fCyy*r00+fCzy*(r01+h01*r00),k01=fCyy*r01+fCzy*(r11+h01*r01); - k10=fCzy*r00+fCzz*(r01+h01*r00),k11=fCzy*r01+fCzz*(r11+h01*r01); - k20=fCey*r00+fCez*(r01+h01*r00),k21=fCey*r01+fCez*(r11+h01*r01); - k30=fCty*r00+fCtz*(r01+h01*r00),k31=fCty*r01+fCtz*(r11+h01*r01); - k40=fCcy*r00+fCcz*(r01+h01*r00),k41=fCcy*r01+fCcz*(r11+h01*r01); - - dy=c->GetY() - fY; dz=c->GetZ() - fZ; - dy=dy+h01*dz; - - cur=fC + k40*dy + k41*dz; eta=fE + k20*dy + k21*dz; - if (TMath::Abs(cur*fX-eta) >= 0.99999) { - Int_t n=GetNumberOfClusters(); - if (n>4) cerr< to the track + // and updates track information + // : predicted chi2 + // : ?? + // - Int_t n=GetNumberOfClusters(); - fIndex[n]=index; - SetNumberOfClusters(n+1); + Double_t h01 = t.GetTilt(); // Is this correct???? + Double_t p[2] = { t.GetY(), t.GetZ() }; + Double_t sy2 = t.GetTrackletSigma2(); // Error pad-column + Double_t sz2 = 100000.0; // Error pad-row (????) + Double_t cov[3] = { sy2 + h01*h01*sz2 // Does this have any sense???? + , h01 * (sy2 - sz2) + , sz2 + h01*h01*sy2 }; + if (!AliExternalTrackParam::Update(p,cov)) { + return kFALSE; + } + Int_t n = GetNumberOfClusters(); + fIndex[n] = index; + SetNumberOfClusters(n+1); SetChi2(GetChi2()+chisq); - // cerr<<"in update: fIndex["<=TMath::Pi()) fAlpha -= 2*TMath::Pi(); + if (absolute) { + alpha -= GetAlpha(); + } + else{ + fNRotate++; + } - Double_t x1=fX, y1=fY; - Double_t ca=cos(alpha), sa=sin(alpha); - Double_t r1=fC*fX - fE; + return AliExternalTrackParam::Rotate(GetAlpha()+alpha); - fX = x1*ca + y1*sa; - fY =-x1*sa + y1*ca; - if((r1*r1) > 1) return 0; - fE=fE*ca + (fC*y1 + sqrt(1.- r1*r1))*sa; +} - Double_t r2=fC*fX - fE; - if (TMath::Abs(r2) >= 0.99999) { - Int_t n=GetNumberOfClusters(); - if (n>4) cerr<GetY() + , c->GetZ() }; + Double_t sy2 = c->GetSigmaY2() * 4.0; + Double_t sz2 = c->GetSigmaZ2() * 4.0; + Double_t cov[3] = { sy2 + h01*h01*sz2 + , h01*(sy2-sz2) + , sz2 + h01*h01*sy2 }; + + return AliExternalTrackParam::GetPredictedChi2(p,cov); + +} + +//_____________________________________________________________________________ +void AliTRDtrack::MakeBackupTrack() +{ + // + // Creates a backup track + // + + if (fBackupTrack) { + delete fBackupTrack; } + fBackupTrack = new AliTRDtrack(*this); + +} - if((r2*r2) > 1) return 0; - Double_t y0=fY + sqrt(1.- r2*r2)/fC; - if ((fY-y0)*fC >= 0.) { - Int_t n=GetNumberOfClusters(); - if (n>4) cerr< xr) ? -1.0 : 1.0; + + // Direction +- + for (Double_t x = GetX()+dir*step; dir*x < dir*xr; x += dir*step) { + + GetXYZ(xyz0); + GetProlongation(x,y,z); + xyz1[0] = x * TMath::Cos(GetAlpha()) + y * TMath::Sin(GetAlpha()); + xyz1[1] = x * TMath::Sin(GetAlpha()) - y * TMath::Cos(GetAlpha()); + xyz1[2] = z; + Double_t param[7]; + AliTracker::MeanMaterialBudget(xyz0,xyz1,param); + + if ((param[0] > 0) && + (param[1] > 0)) { + PropagateTo(x,param[1],param[0]*param[4]); + } - //b = C*ft - Double_t b00=fCyy*f00, b02=fCyy*f20+fCcy*f24+fCey*f22; - Double_t b10=fCzy*f00, b12=fCzy*f20+fCcz*f24+fCez*f22; - Double_t b20=fCey*f00, b22=fCey*f20+fCce*f24+fCee*f22; - Double_t b30=fCty*f00, b32=fCty*f20+fCct*f24+fCte*f22; - Double_t b40=fCcy*f00, b42=fCcy*f20+fCcc*f24+fCce*f22; + if (GetY() > GetX()*kTalphac) { + Rotate(-kAlphac); + } + if (GetY() < -GetX()*kTalphac) { + Rotate( kAlphac); + } - //a = f*b = f*C*ft - Double_t a00=f00*b00, a02=f00*b02, a22=f20*b02+f24*b42+f22*b22; + } - //F*C*Ft = C + (a + b + bt) - fCyy += a00 + 2*b00; - fCzy += b10; - fCey += a02+b20+b02; - fCty += b30; - fCcy += b40; - fCez += b12; - fCte += b32; - fCee += a22 + 2*b22; - fCce += b42; + PropagateTo(xr); - return 1; -} + return 0; +} //_____________________________________________________________________________ -Double_t AliTRDtrack::GetPredictedChi2(const AliTRDcluster *c, Double_t h01) const +Int_t AliTRDtrack::PropagateToR(Double_t r,Double_t step) { - - Bool_t fNoTilt = kTRUE; - if(TMath::Abs(h01) > 0.003) fNoTilt = kFALSE; - Double_t chi2, dy, r00, r01, r11; - - if(fNoTilt) { - dy=c->GetY() - fY; - r00=c->GetSigmaY2(); - chi2 = (dy*dy)/r00; - } - else { - r00=c->GetSigmaY2(); r01=0.; r11=c->GetSigmaZ2(); - r00+=fCyy; r01+=fCzy; r11+=fCzz; - - Double_t det=r00*r11 - r01*r01; - if (TMath::Abs(det) < 1.e-10) { - Int_t n=GetNumberOfClusters(); - if (n>4) cerr<GetY() - fY, dz=c->GetZ() - fZ; - dy=dy+h01*dz; + // + // Propagate track to the radial position + // Rotation always connected to the last track position + // - chi2 = (dy*r00*dy + 2*r01*dy*dz + dz*r11*dz)/det; + Double_t xyz0[3]; + Double_t xyz1[3]; + Double_t y; + Double_t z; + + Double_t radius = TMath::Sqrt(GetX()*GetX() + GetY()*GetY()); + // Direction +- + Double_t dir = (radius > r) ? -1.0 : 1.0; + + for (Double_t x = radius+dir*step; dir*x < dir*r; x += dir*step) { + + GetXYZ(xyz0); + Double_t alpha = TMath::ATan2(xyz0[1],xyz0[0]); + Rotate(alpha,kTRUE); + GetXYZ(xyz0); + GetProlongation(x,y,z); + xyz1[0] = x * TMath::Cos(alpha) + y * TMath::Sin(alpha); + xyz1[1] = x * TMath::Sin(alpha) - y * TMath::Cos(alpha); + xyz1[2] = z; + Double_t param[7]; + AliTracker::MeanMaterialBudget(xyz0,xyz1,param); + if (param[1] <= 0) { + param[1] = 100000000; + } + PropagateTo(x,param[1],param[0]*param[4]); + + } + + GetXYZ(xyz0); + Double_t alpha = TMath::ATan2(xyz0[1],xyz0[0]); + Rotate(alpha,kTRUE); + GetXYZ(xyz0); + GetProlongation(r,y,z); + xyz1[0] = r * TMath::Cos(alpha) + y * TMath::Sin(alpha); + xyz1[1] = r * TMath::Sin(alpha) - y * TMath::Cos(alpha); + xyz1[2] = z; + Double_t param[7]; + AliTracker::MeanMaterialBudget(xyz0,xyz1,param); + + if (param[1] <= 0) { + param[1] = 100000000; } - return chi2; -} + PropagateTo(r,param[1],param[0]*param[4]); + return 0; -//_________________________________________________________________________ -void AliTRDtrack::GetPxPyPz(Double_t& px, Double_t& py, Double_t& pz) const +} + +//_____________________________________________________________________________ +Int_t AliTRDtrack::GetSector() const { - // Returns reconstructed track momentum in the global system. + // + // Return the current sector + // - Double_t pt=TMath::Abs(GetPt()); // GeV/c - Double_t r=fC*fX-fE; + return Int_t(TVector2::Phi_0_2pi(GetAlpha()) / AliTRDgeometry::GetAlpha()) + % AliTRDgeometry::kNsector; - Double_t y0; - if(r > 1) { py = pt; px = 0; } - else if(r < -1) { py = -pt; px = 0; } - else { - y0=fY + sqrt(1.- r*r)/fC; - px=-pt*(fY-y0)*fC; //cos(phi); - py=-pt*(fE-fX*fC); //sin(phi); - } - pz=pt*fT; - Double_t tmp=px*TMath::Cos(fAlpha) - py*TMath::Sin(fAlpha); - py=px*TMath::Sin(fAlpha) + py*TMath::Cos(fAlpha); - px=tmp; +} + +//_____________________________________________________________________________ +void AliTRDtrack::SetSampledEdx(Float_t q, Int_t i) +{ + // + // The sampled energy loss + // -} + Double_t s = GetSnp(); + Double_t t = GetTgl(); + q *= TMath::Sqrt((1.0 - s*s) / (1.0 + t*t)); + fdQdl[i] = q; + +} -//_________________________________________________________________________ -void AliTRDtrack::GetGlobalXYZ(Double_t& x, Double_t& y, Double_t& z) const +//_____________________________________________________________________________ +void AliTRDtrack::SetSampledEdx(Float_t q) { - // Returns reconstructed track coordinates in the global system. + // + // The sampled energy loss + // - x = fX; y = fY; z = fZ; - Double_t tmp=x*TMath::Cos(fAlpha) - y*TMath::Sin(fAlpha); - y=x*TMath::Sin(fAlpha) + y*TMath::Cos(fAlpha); - x=tmp; + Double_t s = GetSnp(); + Double_t t = GetTgl(); + q *= TMath::Sqrt((1.0 - s*s) / (1.0 + t*t)); + fdQdl[fNdedx] = q; + fNdedx++; -} +} -//_________________________________________________________________________ -void AliTRDtrack::ResetCovariance() { +//_____________________________________________________________________________ +Double_t AliTRDtrack::GetBz() const +{ // - // Resets covariance matrix + // Returns Bz component of the magnetic field (kG) // - fCyy*=10.; - fCzy=0.; fCzz*=10.; - fCey=0.; fCez=0.; fCee*=10.; - fCty=0.; fCtz=0.; fCte=0.; fCtt*=10.; - fCcy=0.; fCcz=0.; fCce=0.; fCct=0.; fCcc*=10.; -} + if (AliTracker::UniformField()) { + return AliTracker::GetBz(); + } + Double_t r[3]; + GetXYZ(r); + + return AliTracker::GetBz(r); +}