]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - STEER/AliESDtrack.cxx
Check for undecayed sigma0 in IsPhysicalPrimary
[u/mrichter/AliRoot.git] / STEER / AliESDtrack.cxx
index 60a44b9f850fb47a8f20c15627940bfa8f3d4694..9ed7979abe2d6508930d1d2aa03b2daf7895d695 100644 (file)
 
 #include "AliESDVertex.h"
 #include "AliESDtrack.h"
+#include "AliESDEvent.h"
 #include "AliKalmanTrack.h"
 #include "AliVTrack.h"
 #include "AliLog.h"
@@ -171,11 +172,11 @@ AliESDtrack::AliESDtrack() :
   fITSLabel(0),
   fTPCLabel(0),
   fTRDLabel(0),
-  fTOFCalChannel(0),
+  fTOFCalChannel(-1),
   fTOFindex(-1),
   fHMPIDqn(0),
   fHMPIDcluIdx(-1),
-  fEMCALindex(kEMCALNoMatch),
+  fCaloIndex(kEMCALNoMatch),
   fHMPIDtrkTheta(0),
   fHMPIDtrkPhi(0),
   fHMPIDsignal(0),
@@ -188,6 +189,7 @@ AliESDtrack::AliESDtrack() :
   fCchi2(0),
   fITSchi2(0),
   fTPCchi2(0),
+  fTPCchi2Iter1(0),
   fTRDchi2(0),
   fTOFchi2(0),
   fHMPIDchi2(0),
@@ -198,10 +200,15 @@ AliESDtrack::AliESDtrack() :
   fTRDsignal(0),
   fTRDQuality(0),
   fTRDBudget(0),
-  fTOFsignal(0),
-  fTOFsignalToT(0),
-  fTOFsignalRaw(0),
-  fTOFsignalDz(0),
+  fTOFsignal(99999),
+  fTOFsignalToT(99999),
+  fTOFsignalRaw(99999),
+  fTOFsignalDz(999),
+  fTOFsignalDx(999),
+  fTOFdeltaBC(999),
+  fTOFl0l1(999),
+  fCaloDx(0),
+  fCaloDz(0),
   fHMPIDtrkX(0),
   fHMPIDtrkY(0),
   fHMPIDmipX(0),
@@ -209,6 +216,8 @@ AliESDtrack::AliESDtrack() :
   fTPCncls(0),
   fTPCnclsF(0),
   fTPCsignalN(0),
+  fTPCnclsIter1(0),
+  fTPCnclsFIter1(0),
   fITSncls(0),
   fITSClusterMap(0),
   fTRDncls(0),
@@ -216,7 +225,8 @@ AliESDtrack::AliESDtrack() :
   fTRDntracklets(0),
   fTRDnSlices(0),
   fTRDslices(0x0),
-  fVertexID(-2) // -2 means an orphan track 
+  fVertexID(-2),// -2 means an orphan track 
+  fESDEvent(0)
 {
   //
   // The default ESD constructor 
@@ -239,7 +249,7 @@ AliESDtrack::AliESDtrack() :
   }
   for (i=0;i<4;i++) {fITSdEdxSamples[i]=0.;}
   for (i=0;i<4;i++) {fTPCPoints[i]=0;}
-  for (i=0;i<3;i++) {fTOFLabel[i]=0;}
+  for (i=0;i<3;i++) {fTOFLabel[i]=-1;}
   for (i=0;i<10;i++) {fTOFInfo[i]=0;}
   for (i=0;i<12;i++) {fITSModule[i]=-1;}
 }
@@ -265,7 +275,7 @@ AliESDtrack::AliESDtrack(const AliESDtrack& track):
   fTOFindex(track.fTOFindex),
   fHMPIDqn(track.fHMPIDqn),
   fHMPIDcluIdx(track.fHMPIDcluIdx),
-  fEMCALindex(track.fEMCALindex),
+  fCaloIndex(track.fCaloIndex),
   fHMPIDtrkTheta(track.fHMPIDtrkTheta),
   fHMPIDtrkPhi(track.fHMPIDtrkPhi),
   fHMPIDsignal(track.fHMPIDsignal),
@@ -278,6 +288,7 @@ AliESDtrack::AliESDtrack(const AliESDtrack& track):
   fCchi2(track.fCchi2),
   fITSchi2(track.fITSchi2),
   fTPCchi2(track.fTPCchi2),
+  fTPCchi2Iter1(track.fTPCchi2Iter1),
   fTRDchi2(track.fTRDchi2),
   fTOFchi2(track.fTOFchi2),
   fHMPIDchi2(track.fHMPIDchi2),
@@ -292,6 +303,11 @@ AliESDtrack::AliESDtrack(const AliESDtrack& track):
   fTOFsignalToT(track.fTOFsignalToT),
   fTOFsignalRaw(track.fTOFsignalRaw),
   fTOFsignalDz(track.fTOFsignalDz),
+  fTOFsignalDx(track.fTOFsignalDx),
+  fTOFdeltaBC(track.fTOFdeltaBC),
+  fTOFl0l1(track.fTOFl0l1),
+  fCaloDx(track.fCaloDx),
+  fCaloDz(track.fCaloDz),
   fHMPIDtrkX(track.fHMPIDtrkX),
   fHMPIDtrkY(track.fHMPIDtrkY),
   fHMPIDmipX(track.fHMPIDmipX),
@@ -299,6 +315,8 @@ AliESDtrack::AliESDtrack(const AliESDtrack& track):
   fTPCncls(track.fTPCncls),
   fTPCnclsF(track.fTPCnclsF),
   fTPCsignalN(track.fTPCsignalN),
+  fTPCnclsIter1(track.fTPCnclsIter1),
+  fTPCnclsFIter1(track.fTPCnclsIter1),
   fITSncls(track.fITSncls),
   fITSClusterMap(track.fITSClusterMap),
   fTRDncls(track.fTRDncls),
@@ -306,7 +324,8 @@ AliESDtrack::AliESDtrack(const AliESDtrack& track):
   fTRDntracklets(track.fTRDntracklets),
   fTRDnSlices(track.fTRDnSlices),
   fTRDslices(0x0),
-  fVertexID(track.fVertexID)
+  fVertexID(track.fVertexID),
+  fESDEvent(track.fESDEvent)
 {
   //
   //copy constructor
@@ -364,11 +383,11 @@ AliESDtrack::AliESDtrack(const AliVTrack *track) :
   fITSLabel(0),
   fTPCLabel(0),
   fTRDLabel(0),
-  fTOFCalChannel(0),
+  fTOFCalChannel(-1),
   fTOFindex(-1),
   fHMPIDqn(0),
   fHMPIDcluIdx(-1),
-  fEMCALindex(kEMCALNoMatch),
+  fCaloIndex(kEMCALNoMatch),
   fHMPIDtrkTheta(0),
   fHMPIDtrkPhi(0),
   fHMPIDsignal(0),
@@ -381,6 +400,7 @@ AliESDtrack::AliESDtrack(const AliVTrack *track) :
   fCchi2(0),
   fITSchi2(0),
   fTPCchi2(0),
+  fTPCchi2Iter1(0),
   fTRDchi2(0),
   fTOFchi2(0),
   fHMPIDchi2(0),
@@ -391,10 +411,15 @@ AliESDtrack::AliESDtrack(const AliVTrack *track) :
   fTRDsignal(0),
   fTRDQuality(0),
   fTRDBudget(0),
-  fTOFsignal(0),
-  fTOFsignalToT(0),
-  fTOFsignalRaw(0),
-  fTOFsignalDz(0),
+  fTOFsignal(99999),
+  fTOFsignalToT(99999),
+  fTOFsignalRaw(99999),
+  fTOFsignalDz(999),
+  fTOFsignalDx(999),
+  fTOFdeltaBC(999),
+  fTOFl0l1(999),
+  fCaloDx(0),
+  fCaloDz(0),
   fHMPIDtrkX(0),
   fHMPIDtrkY(0),
   fHMPIDmipX(0),
@@ -402,6 +427,8 @@ AliESDtrack::AliESDtrack(const AliVTrack *track) :
   fTPCncls(0),
   fTPCnclsF(0),
   fTPCsignalN(0),
+  fTPCnclsIter1(0),
+  fTPCnclsFIter1(0),
   fITSncls(0),
   fITSClusterMap(0),
   fTRDncls(0),
@@ -409,7 +436,8 @@ AliESDtrack::AliESDtrack(const AliVTrack *track) :
   fTRDntracklets(0),
   fTRDnSlices(0),
   fTRDslices(0x0),
-  fVertexID(-2)  // -2 means an orphan track
+  fVertexID(-2),  // -2 means an orphan track
+  fESDEvent(0)  
 {
   //
   // ESD track from AliVTrack.
@@ -442,7 +470,7 @@ AliESDtrack::AliESDtrack(const AliVTrack *track) :
   }
   for (i=0;i<4;i++) {fITSdEdxSamples[i]=0.;}
   for (i=0;i<4;i++) {fTPCPoints[i]=0;}
-  for (i=0;i<3;i++) {fTOFLabel[i]=0;}
+  for (i=0;i<3;i++) {fTOFLabel[i]=-1;}
   for (i=0;i<10;i++) {fTOFInfo[i]=0;}
   for (i=0;i<12;i++) {fITSModule[i]=-1;}
 
@@ -457,6 +485,10 @@ AliESDtrack::AliESDtrack(const AliVTrack *track) :
     if(HasPointOnITSLayer(i)) fITSncls++;
   }
 
+  // Set TPC ncls 
+  fTPCncls=track->GetTPCNcls();
+
+
   // Set the combined PID
   const Double_t *pid = track->PID();
   if(pid){
@@ -485,11 +517,11 @@ AliESDtrack::AliESDtrack(TParticle * part) :
   fITSLabel(0),
   fTPCLabel(0),
   fTRDLabel(0),
-  fTOFCalChannel(0),
+  fTOFCalChannel(-1),
   fTOFindex(-1),
   fHMPIDqn(0),
   fHMPIDcluIdx(-1),
-  fEMCALindex(kEMCALNoMatch),
+  fCaloIndex(kEMCALNoMatch),
   fHMPIDtrkTheta(0),
   fHMPIDtrkPhi(0),
   fHMPIDsignal(0),
@@ -502,6 +534,7 @@ AliESDtrack::AliESDtrack(TParticle * part) :
   fCchi2(0),
   fITSchi2(0),
   fTPCchi2(0),
+  fTPCchi2Iter1(0),  
   fTRDchi2(0),
   fTOFchi2(0),
   fHMPIDchi2(0),
@@ -512,10 +545,15 @@ AliESDtrack::AliESDtrack(TParticle * part) :
   fTRDsignal(0),
   fTRDQuality(0),
   fTRDBudget(0),
-  fTOFsignal(0),
-  fTOFsignalToT(0),
-  fTOFsignalRaw(0),
-  fTOFsignalDz(0),
+  fTOFsignal(99999),
+  fTOFsignalToT(99999),
+  fTOFsignalRaw(99999),
+  fTOFsignalDz(999),
+  fTOFsignalDx(999),
+  fTOFdeltaBC(999),
+  fTOFl0l1(999),
+  fCaloDx(0),
+  fCaloDz(0),
   fHMPIDtrkX(0),
   fHMPIDtrkY(0),
   fHMPIDmipX(0),
@@ -523,6 +561,8 @@ AliESDtrack::AliESDtrack(TParticle * part) :
   fTPCncls(0),
   fTPCnclsF(0),
   fTPCsignalN(0),
+  fTPCnclsIter1(0),
+  fTPCnclsFIter1(0),
   fITSncls(0),
   fITSClusterMap(0),
   fTRDncls(0),
@@ -530,7 +570,8 @@ AliESDtrack::AliESDtrack(TParticle * part) :
   fTRDntracklets(0),
   fTRDnSlices(0),
   fTRDslices(0x0),
-  fVertexID(-2)  // -2 means an orphan track
+  fVertexID(-2),  // -2 means an orphan track
+  fESDEvent(0)
 {
   //
   // ESD track from TParticle
@@ -555,7 +596,7 @@ AliESDtrack::AliESDtrack(TParticle * part) :
   }
   for (i=0;i<4;i++) {fITSdEdxSamples[i]=0.;}
   for (i=0;i<4;i++) {fTPCPoints[i]=0;}
-  for (i=0;i<3;i++) {fTOFLabel[i]=0;}
+  for (i=0;i<3;i++) {fTOFLabel[i]=-1;}
   for (i=0;i<10;i++) {fTOFInfo[i]=0;}
   for (i=0;i<12;i++) {fITSModule[i]=-1;}
 
@@ -720,7 +761,7 @@ AliESDtrack &AliESDtrack::operator=(const AliESDtrack &source){
   
   if(source.fHMPIDp){
     // we have the trackparam: assign or copy construct
-    if(fOp) *fHMPIDp = *source.fHMPIDp;
+    if(fHMPIDp) *fHMPIDp = *source.fHMPIDp;
     else fHMPIDp = new AliExternalTrackParam(*source.fHMPIDp);
   }
   else{
@@ -760,7 +801,7 @@ AliESDtrack &AliESDtrack::operator=(const AliESDtrack &source){
   fTOFindex      = source.fTOFindex;
   fHMPIDqn       = source.fHMPIDqn;
   fHMPIDcluIdx   = source.fHMPIDcluIdx; 
-  fEMCALindex    = source.fEMCALindex;
+  fCaloIndex    = source.fCaloIndex;
 
   for(int i = 0; i< 3;++i){
     fKinkIndexes[i] = source.fKinkIndexes[i]; 
@@ -799,6 +840,7 @@ AliESDtrack &AliESDtrack::operator=(const AliESDtrack &source){
 
   fITSchi2   = source.fITSchi2;             
   fTPCchi2   = source.fTPCchi2;            
+  fTPCchi2Iter1   = source.fTPCchi2Iter1;            
   fTRDchi2   = source.fTRDchi2;      
   fTOFchi2   = source.fTOFchi2;      
   fHMPIDchi2 = source.fHMPIDchi2;      
@@ -833,7 +875,10 @@ AliESDtrack &AliESDtrack::operator=(const AliESDtrack &source){
   fTOFsignalToT = source.fTOFsignalToT;   
   fTOFsignalRaw = source.fTOFsignalRaw;  
   fTOFsignalDz  = source.fTOFsignalDz;      
-  
+  fTOFsignalDx  = source.fTOFsignalDx;      
+  fTOFdeltaBC   = source.fTOFdeltaBC;
+  fTOFl0l1      = source.fTOFl0l1;
   for(int i = 0;i<10;++i){
     fTOFInfo[i] = source.fTOFInfo[i];    
   }
@@ -846,6 +891,8 @@ AliESDtrack &AliESDtrack::operator=(const AliESDtrack &source){
   fTPCncls    = source.fTPCncls;      
   fTPCnclsF   = source.fTPCnclsF;     
   fTPCsignalN = source.fTPCsignalN;   
+  fTPCnclsIter1    = source.fTPCnclsIter1;      
+  fTPCnclsFIter1   = source.fTPCnclsFIter1;     
 
   fITSncls = source.fITSncls;       
   fITSClusterMap = source.fITSClusterMap; 
@@ -935,6 +982,7 @@ Bool_t AliESDtrack::FillTPCOnlyTrack(AliESDtrack &track){
   track.fTPCLabel = fTPCLabel;
 
   track.fTPCchi2 = fTPCchi2; 
+  track.fTPCchi2Iter1 = fTPCchi2Iter1; 
   track.fTPCsignal = fTPCsignal;
   track.fTPCsignalS = fTPCsignalS;
   for(int i = 0;i<4;++i)track.fTPCPoints[i] = fTPCPoints[i];
@@ -942,6 +990,8 @@ Bool_t AliESDtrack::FillTPCOnlyTrack(AliESDtrack &track){
   track.fTPCncls    = fTPCncls;     
   track.fTPCnclsF   = fTPCnclsF;     
   track.fTPCsignalN =  fTPCsignalN;
+  track.fTPCnclsIter1    = fTPCnclsIter1;     
+  track.fTPCnclsFIter1   = fTPCnclsFIter1;     
 
   // PID 
   for(int i=0;i<AliPID::kSPECIES;++i){
@@ -1002,8 +1052,11 @@ void AliESDtrack::MakeMiniESDtrack(){
 
   // Reset TPC related track information
   fTPCchi2 = 0;       
+  fTPCchi2Iter1 = 0;       
   fTPCncls = 0;       
   fTPCnclsF = 0;       
+  fTPCnclsIter1 = 0;       
+  fTPCnclsFIter1 = 0;       
   fTPCClusterMap = 0;  
   fTPCSharedMap = 0;  
   fTPCsignal= 0;      
@@ -1036,13 +1089,16 @@ void AliESDtrack::MakeMiniESDtrack(){
   // Reset TOF related track information
   fTOFchi2 = 0;        
   fTOFindex = -1;       
-  fTOFsignal = 0;      
-  fTOFCalChannel = 0;
-  fTOFsignalToT = 0;
-  fTOFsignalRaw = 0;
-  fTOFsignalDz = 0;
+  fTOFsignal = 99999;      
+  fTOFCalChannel = -1;
+  fTOFsignalToT = 99999;
+  fTOFsignalRaw = 99999;
+  fTOFsignalDz = 999;
+  fTOFsignalDx = 999;
+  fTOFdeltaBC = 999;
+  fTOFl0l1 = 999;
   for (Int_t i=0;i<AliPID::kSPECIES;i++) fTOFr[i] = 0;
-  for (Int_t i=0;i<3;i++) fTOFLabel[i] = 0;
+  for (Int_t i=0;i<3;i++) fTOFLabel[i] = -1;
   for (Int_t i=0;i<10;i++) fTOFInfo[i] = 0;
 
   // Reset HMPID related track information
@@ -1057,7 +1113,7 @@ void AliESDtrack::MakeMiniESDtrack(){
   fHMPIDtrkY = 0;      
   fHMPIDmipX = 0;
   fHMPIDmipY = 0;
-  fEMCALindex = kEMCALNoMatch;
+  fCaloIndex = kEMCALNoMatch;
 
   // reset global track chi2
   fGlobalChi2 = 0;
@@ -1069,9 +1125,17 @@ void AliESDtrack::MakeMiniESDtrack(){
 //_______________________________________________________________________
 Double_t AliESDtrack::GetMass() const {
   // Returns the mass of the most probable particle type
+
+  Int_t i;
+  for (i=0; i<AliPID::kSPECIES-1; i++) { 
+      if (fR[i] != fR[i+1]) break;
+  }
+  // If all the probabilities are equal, return the pion mass
+  if (i == AliPID::kSPECIES-1) return AliPID::ParticleMass(AliPID::kPion);
+
   Float_t max=0.;
   Int_t k=-1;
-  for (Int_t i=0; i<AliPID::kSPECIES; i++) {
+  for (i=0; i<AliPID::kSPECIES; i++) {
     if (fR[i]>max) {k=i; max=fR[i];}
   }
   if (k==0) { // dE/dx "crossing points" in the TPC
@@ -1090,6 +1154,16 @@ Double_t AliESDtrack::GetMass() const {
   return AliPID::ParticleMass(AliPID::kPion);
 }
 
+//______________________________________________________________________________
+Double_t AliESDtrack::M() const
+{
+  // Returns the assumed mass
+  // (the pion mass, if the particle can't be identified properly).
+
+  AliWarning("This is the ESD mass. Use it with care !"); 
+  return GetMass(); 
+}
+  
 //______________________________________________________________________________
 Double_t AliESDtrack::E() const
 {
@@ -1121,7 +1195,6 @@ Bool_t AliESDtrack::UpdateTrackParams(const AliKalmanTrack *t, ULong_t flags){
   //
   // This function updates track's running parameters 
   //
-  Int_t *index=0;
   Bool_t rc=kTRUE;
 
   SetStatus(flags);
@@ -1141,16 +1214,21 @@ Bool_t AliESDtrack::UpdateTrackParams(const AliKalmanTrack *t, ULong_t flags){
   switch (flags) {
     
   case kITSin: case kITSout: case kITSrefit:
+    {
     fITSClusterMap=0;
     fITSncls=t->GetNumberOfClusters();
-    index=fFriendTrack->GetITSindices(); 
+    Int_t* indexITS = new Int_t[AliESDfriendTrack::kMaxITScluster];
     for (Int_t i=0;i<AliESDfriendTrack::kMaxITScluster;i++) {
-        index[i]=t->GetClusterIndex(i);
+       indexITS[i]=t->GetClusterIndex(i);
+
        if (i<fITSncls) {
-           Int_t l=(index[i] & 0xf0000000) >> 28;
+         Int_t l=(indexITS[i] & 0xf0000000) >> 28;
            SETBIT(fITSClusterMap,l);                 
         }
     }
+    fFriendTrack->SetITSIndices(indexITS,AliESDfriendTrack::kMaxITScluster);
+    delete [] indexITS;
+
     fITSchi2=t->GetChi2();
     fITSsignal=t->GetPIDsignal();
     fITSLabel = t->GetLabel();
@@ -1160,16 +1238,24 @@ Bool_t AliESDtrack::UpdateTrackParams(const AliKalmanTrack *t, ULong_t flags){
       else 
         fOp->Set(t->GetX(),t->GetAlpha(),t->GetParameter(),t->GetCovariance());
     }   
+    }
     break;
     
   case kTPCin: case kTPCrefit:
+    {
     fTPCLabel = t->GetLabel();
-    if (flags==kTPCin)  fTPCInner=new AliExternalTrackParam(*t);
+    if (flags==kTPCin)  {
+      fTPCInner=new AliExternalTrackParam(*t); 
+      fTPCnclsIter1=t->GetNumberOfClusters();    
+      fTPCchi2Iter1=t->GetChi2();
+    }
     if (!fIp) fIp=new AliExternalTrackParam(*t);
     else 
       fIp->Set(t->GetX(),t->GetAlpha(),t->GetParameter(),t->GetCovariance());
+    }
   case kTPCout:
-    index=fFriendTrack->GetTPCindices(); 
+    {
+    Int_t* indexTPC = new Int_t[AliESDfriendTrack::kMaxTPCcluster];
     if (flags & kTPCout){
       if (!fOp) fOp=new AliExternalTrackParam(*t);
       else 
@@ -1184,8 +1270,8 @@ Bool_t AliESDtrack::UpdateTrackParams(const AliKalmanTrack *t, ULong_t flags){
        //       for (Int_t i=0;i<fTPCncls;i++) 
        for (Int_t i=0;i<AliESDfriendTrack::kMaxTPCcluster;i++) 
         {
-          index[i]=t->GetClusterIndex(i);
-          Int_t idx = index[i];
+         indexTPC[i]=t->GetClusterIndex(i);
+         Int_t idx = indexTPC[i];
 
          if (idx<0) continue; 
 
@@ -1228,20 +1314,30 @@ Bool_t AliESDtrack::UpdateTrackParams(const AliKalmanTrack *t, ULong_t flags){
            }
           // End Of Piotr's Cluster Map for HBT
         }
+       fFriendTrack->SetTPCIndices(indexTPC,AliESDfriendTrack::kMaxTPCcluster);
+       delete [] indexTPC;
+
      }
     fTPCsignal=t->GetPIDsignal();
+    }
     break;
 
   case kTRDin: case kTRDrefit:
     break;
   case kTRDout:
-    index     = fFriendTrack->GetTRDindices();
+    {
     fTRDLabel = t->GetLabel(); 
     fTRDchi2  = t->GetChi2();
     fTRDncls  = t->GetNumberOfClusters();
-    for (Int_t i=0;i<6;i++) index[i]=t->GetTrackletIndex(i);
+      Int_t* indexTRD = new Int_t[AliESDfriendTrack::kMaxTRDcluster];
+      for (Int_t i=0;i<AliESDfriendTrack::kMaxTRDcluster;i++) indexTRD[i]=-2;
+      for (Int_t i=0;i<6;i++) indexTRD[i]=t->GetTrackletIndex(i);
+      fFriendTrack->SetTRDIndices(indexTRD,AliESDfriendTrack::kMaxTRDcluster);
+      delete [] indexTRD;
+    
     
     fTRDsignal=t->GetPIDsignal();
+    }
     break;
   case kTRDbackup:
     if (!fOp) fOp=new AliExternalTrackParam(*t);
@@ -1507,12 +1603,17 @@ Char_t AliESDtrack::GetITSclusters(Int_t *idx) const {
   //---------------------------------------------------------------------
   // This function returns indices of the assgined ITS clusters 
   //---------------------------------------------------------------------
-  if (idx!=0) {
-     Int_t *index=fFriendTrack->GetITSindices();
-     for (Int_t i=0; i<AliESDfriendTrack::kMaxITScluster; i++) {
-         if ( (i>=fITSncls) && (i<6) ) idx[i]=-1;
-         else idx[i]=index[i];
-     }
+  if (idx) {
+    Int_t *index=fFriendTrack->GetITSindices();
+    for (Int_t i=0; i<AliESDfriendTrack::kMaxITScluster; i++) {
+      if ( (i>=fITSncls) && (i<6) ) idx[i]=-1;
+      else {
+       if (index) {
+         idx[i]=index[i];
+       }
+       else idx[i]= -2;
+      }
+    }
   }
   return fITSncls;
 }
@@ -1579,9 +1680,15 @@ UShort_t AliESDtrack::GetTPCclusters(Int_t *idx) const {
   //---------------------------------------------------------------------
   // This function returns indices of the assgined ITS clusters 
   //---------------------------------------------------------------------
-  if (idx!=0) {
+  if (idx) {
     Int_t *index=fFriendTrack->GetTPCindices();
-    for (Int_t i=0; i<AliESDfriendTrack::kMaxTPCcluster; i++) idx[i]=index[i];
+
+    if (index){
+      for (Int_t i=0; i<AliESDfriendTrack::kMaxTPCcluster; i++) idx[i]=index[i];
+    }
+    else {
+      for (Int_t i=0; i<AliESDfriendTrack::kMaxTPCcluster; i++) idx[i]=-2;
+    }
   }
   return fTPCncls;
 }
@@ -1623,23 +1730,42 @@ UChar_t AliESDtrack::GetTRDclusters(Int_t *idx) const {
   //---------------------------------------------------------------------
   // This function returns indices of the assgined TRD clusters 
   //---------------------------------------------------------------------
-  if (idx!=0) {
-     Int_t *index=fFriendTrack->GetTRDindices();
-     for (Int_t i=0; i<AliESDfriendTrack::kMaxTRDcluster; i++) idx[i]=index[i];
+  if (idx) {
+    Int_t *index=fFriendTrack->GetTRDindices();
+
+    if (index) {
+      for (Int_t i=0; i<AliESDfriendTrack::kMaxTRDcluster; i++) idx[i]=index[i];
+    }
+    else {
+      for (Int_t i=0; i<AliESDfriendTrack::kMaxTRDcluster; i++) idx[i]=-2;
+    }
   }
   return fTRDncls;
 }
 
 //_______________________________________________________________________
 UChar_t AliESDtrack::GetTRDtracklets(Int_t *idx) const {
-  //---------------------------------------------------------------------
-  // This function returns indices of the assigned TRD tracklets 
-  //---------------------------------------------------------------------
-  if (idx!=0) {
-     Int_t *index=fFriendTrack->GetTRDindices();
-     for (Int_t i=0; i<6/*AliESDfriendTrack::kMaxTRDcluster*/; i++) idx[i]=index[i];
+//
+// This function returns the number of TRD tracklets used in tracking
+// and it fills the indices of these tracklets in the array "idx" as they 
+// are registered in the TRD track list. 
+// 
+// Caution :
+//   1. The idx array has to be allocated with a size >= AliESDtrack::kTRDnPlanes
+//   2. The idx array store not only the index but also the layer of the tracklet. 
+//      Therefore tracks with TRD gaps contain default values for indices [-1] 
+
+  if (!idx) return GetTRDntracklets();
+  Int_t *index=fFriendTrack->GetTRDindices();
+  Int_t n = 0;
+  for (Int_t i=0; i<kTRDnPlanes; i++){ 
+    if (index){
+      if(index[i]>=0) n++;
+      idx[i]=index[i];
+    }
+    else idx[i] = -2;
   }
-  return fTRDncls;
+  return n;
 }
 
 //_______________________________________________________________________
@@ -1683,11 +1809,11 @@ Double_t AliESDtrack::GetTRDmomentum(Int_t plane, Double_t *sp) const
 // in TRD layer "plane".
 
   if (!fTRDnSlices) {
-    AliError("No TRD info allocated for this track !");
+    AliDebug(2, "No TRD info allocated for this track.");
     return -1.;
   }
   if ((plane<0) || (plane>=kTRDnPlanes)) {
-    AliError("Info for TRD plane not available!");
+    AliWarning(Form("Request for TRD plane[%d] outside range.", plane)); 
     return -1.;
   }
 
@@ -1893,6 +2019,54 @@ Double_t b, Double_t maxd, AliExternalTrackParam *cParam) {
   return kTRUE;
 }
 
+//_______________________________________________________________________
+Bool_t AliESDtrack::RelateToVertexTPCBxByBz(const AliESDVertex *vtx, 
+Double_t b[3], Double_t maxd, AliExternalTrackParam *cParam) {
+  //
+  // Try to relate the TPC-only track parameters to the vertex "vtx", 
+  // if the (rough) transverse impact parameter is not bigger then "maxd". 
+  //
+  // All three components of the magnetic field ,"b[3]" (kG), 
+  // are taken into account.
+  //
+  // a) The TPC-only paramters are extapolated to the DCA to the vertex.
+  // b) The impact parameters and their covariance matrix are calculated.
+  // c) An attempt to constrain the TPC-only params to the vertex is done.
+  //    The constrained params are returned via "cParam".
+  //
+  // In the case of success, the returned value is kTRUE
+  // otherwise, it's kFALSE)
+  // 
+
+  if (!fTPCInner) return kFALSE;
+  if (!vtx) return kFALSE;
+
+  Double_t dz[2],cov[3];
+  if (!fTPCInner->PropagateToDCABxByBz(vtx, b, maxd, dz, cov)) return kFALSE;
+
+  fdTPC = dz[0];
+  fzTPC = dz[1];  
+  fCddTPC = cov[0];
+  fCdzTPC = cov[1];
+  fCzzTPC = cov[2];
+  
+  Double_t covar[6]; vtx->GetCovMatrix(covar);
+  Double_t p[2]={GetParameter()[0]-dz[0],GetParameter()[1]-dz[1]};
+  Double_t c[3]={covar[2],0.,covar[5]};
+
+  Double_t chi2=GetPredictedChi2(p,c);
+  if (chi2>kVeryBig) return kFALSE;
+
+  fCchi2TPC=chi2;
+
+  if (!cParam) return kTRUE;
+
+  *cParam = *fTPCInner;
+  if (!cParam->Update(p,c)) return kFALSE;
+
+  return kTRUE;
+}
+
 //_______________________________________________________________________
 Bool_t AliESDtrack::RelateToVertex(const AliESDVertex *vtx, 
 Double_t b, Double_t maxd, AliExternalTrackParam *cParam) {
@@ -1948,6 +2122,61 @@ Double_t b, Double_t maxd, AliExternalTrackParam *cParam) {
   return kTRUE;
 }
 
+//_______________________________________________________________________
+Bool_t AliESDtrack::RelateToVertexBxByBz(const AliESDVertex *vtx, 
+Double_t b[3], Double_t maxd, AliExternalTrackParam *cParam) {
+  //
+  // Try to relate this track to the vertex "vtx", 
+  // if the (rough) transverse impact parameter is not bigger then "maxd". 
+  //            Magnetic field is "b" (kG).
+  //
+  // a) The track gets extapolated to the DCA to the vertex.
+  // b) The impact parameters and their covariance matrix are calculated.
+  // c) An attempt to constrain this track to the vertex is done.
+  //    The constrained params are returned via "cParam".
+  //
+  // In the case of success, the returned value is kTRUE
+  // (otherwise, it's kFALSE)
+  //  
+
+  if (!vtx) return kFALSE;
+
+  Double_t dz[2],cov[3];
+  if (!PropagateToDCABxByBz(vtx, b, maxd, dz, cov)) return kFALSE;
+
+  fD = dz[0];
+  fZ = dz[1];  
+  fCdd = cov[0];
+  fCdz = cov[1];
+  fCzz = cov[2];
+  
+  Double_t covar[6]; vtx->GetCovMatrix(covar);
+  Double_t p[2]={GetParameter()[0]-dz[0],GetParameter()[1]-dz[1]};
+  Double_t c[3]={covar[2],0.,covar[5]};
+
+  Double_t chi2=GetPredictedChi2(p,c);
+  if (chi2>kVeryBig) return kFALSE;
+
+  fCchi2=chi2;
+
+
+  //--- Could now these lines be removed ? ---
+  delete fCp;
+  fCp=new AliExternalTrackParam(*this);  
+
+  if (!fCp->Update(p,c)) {delete fCp; fCp=0; return kFALSE;}
+  //----------------------------------------
+
+  fVertexID = vtx->GetID();
+
+  if (!cParam) return kTRUE;
+
+  *cParam = *this;
+  if (!cParam->Update(p,c)) return kFALSE; 
+
+  return kTRUE;
+}
+
 //_______________________________________________________________________
 void AliESDtrack::Print(Option_t *) const {
   // Prints info on the track
@@ -2067,3 +2296,11 @@ void AliESDtrack::GetITSdEdxSamples(Double_t *s) const {
   //
   for (Int_t i=0; i<4; i++) s[i]=fITSdEdxSamples[i];
 }
+
+
+UShort_t   AliESDtrack::GetTPCnclsS(Int_t i0,Int_t i1) const{
+  //
+  // get number of shared clusters
+  //
+  return  fTPCSharedMap.CountBits(i0)-fTPCSharedMap.CountBits(i1);
+}