]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - STEER/AliESDtrack.cxx
Possibility to propagate tracks to the DCA to the primary vertex at the AOD level...
[u/mrichter/AliRoot.git] / STEER / AliESDtrack.cxx
index bcdf46d871254e88a09cb6a409eabf7af9cd8138..9ed7979abe2d6508930d1d2aa03b2daf7895d695 100644 (file)
 
 #include <TMath.h>
 #include <TParticle.h>
+#include <TDatabasePDG.h>
 
 #include "AliESDVertex.h"
 #include "AliESDtrack.h"
+#include "AliESDEvent.h"
 #include "AliKalmanTrack.h"
 #include "AliVTrack.h"
 #include "AliLog.h"
@@ -160,6 +162,7 @@ AliESDtrack::AliESDtrack() :
   fIp(0),
   fTPCInner(0),
   fOp(0),
+  fHMPIDp(0),  
   fFriendTrack(new AliESDfriendTrack()),
   fTPCClusterMap(159),//number of padrows
   fTPCSharedMap(159),//number of padrows
@@ -169,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),
@@ -186,6 +189,7 @@ AliESDtrack::AliESDtrack() :
   fCchi2(0),
   fITSchi2(0),
   fTPCchi2(0),
+  fTPCchi2Iter1(0),
   fTRDchi2(0),
   fTOFchi2(0),
   fHMPIDchi2(0),
@@ -196,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),
@@ -207,6 +216,8 @@ AliESDtrack::AliESDtrack() :
   fTPCncls(0),
   fTPCnclsF(0),
   fTPCsignalN(0),
+  fTPCnclsIter1(0),
+  fTPCnclsFIter1(0),
   fITSncls(0),
   fITSClusterMap(0),
   fTRDncls(0),
@@ -214,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 
@@ -237,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;}
 }
@@ -249,6 +261,7 @@ AliESDtrack::AliESDtrack(const AliESDtrack& track):
   fIp(0),
   fTPCInner(0),
   fOp(0),
+  fHMPIDp(0),  
   fFriendTrack(0),
   fTPCClusterMap(track.fTPCClusterMap),
   fTPCSharedMap(track.fTPCSharedMap),
@@ -262,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),
@@ -275,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),
@@ -289,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),
@@ -296,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),
@@ -303,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
@@ -339,7 +361,8 @@ AliESDtrack::AliESDtrack(const AliESDtrack& track):
   if (track.fIp) fIp=new AliExternalTrackParam(*track.fIp);
   if (track.fTPCInner) fTPCInner=new AliExternalTrackParam(*track.fTPCInner);
   if (track.fOp) fOp=new AliExternalTrackParam(*track.fOp);
-
+  if (track.fHMPIDp) fHMPIDp=new AliExternalTrackParam(*track.fHMPIDp);
+  
   if (track.fFriendTrack) fFriendTrack=new AliESDfriendTrack(*(track.fFriendTrack));
 }
 
@@ -350,6 +373,7 @@ AliESDtrack::AliESDtrack(const AliVTrack *track) :
   fIp(0),
   fTPCInner(0),
   fOp(0),
+  fHMPIDp(0),  
   fFriendTrack(0),
   fTPCClusterMap(159),//number of padrows
   fTPCSharedMap(159),//number of padrows
@@ -359,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),
@@ -376,6 +400,7 @@ AliESDtrack::AliESDtrack(const AliVTrack *track) :
   fCchi2(0),
   fITSchi2(0),
   fTPCchi2(0),
+  fTPCchi2Iter1(0),
   fTRDchi2(0),
   fTOFchi2(0),
   fHMPIDchi2(0),
@@ -386,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),
@@ -397,6 +427,8 @@ AliESDtrack::AliESDtrack(const AliVTrack *track) :
   fTPCncls(0),
   fTPCnclsF(0),
   fTPCsignalN(0),
+  fTPCnclsIter1(0),
+  fTPCnclsFIter1(0),
   fITSncls(0),
   fITSClusterMap(0),
   fTRDncls(0),
@@ -404,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.
@@ -437,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;}
 
@@ -452,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){
@@ -470,6 +507,7 @@ AliESDtrack::AliESDtrack(TParticle * part) :
   fIp(0),
   fTPCInner(0),
   fOp(0),
+  fHMPIDp(0),  
   fFriendTrack(0),
   fTPCClusterMap(159),//number of padrows
   fTPCSharedMap(159),//number of padrows
@@ -479,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),
@@ -496,6 +534,7 @@ AliESDtrack::AliESDtrack(TParticle * part) :
   fCchi2(0),
   fITSchi2(0),
   fTPCchi2(0),
+  fTPCchi2Iter1(0),  
   fTRDchi2(0),
   fTOFchi2(0),
   fHMPIDchi2(0),
@@ -506,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),
@@ -517,6 +561,8 @@ AliESDtrack::AliESDtrack(TParticle * part) :
   fTPCncls(0),
   fTPCnclsF(0),
   fTPCsignalN(0),
+  fTPCnclsIter1(0),
+  fTPCnclsFIter1(0),
   fITSncls(0),
   fITSClusterMap(0),
   fTRDncls(0),
@@ -524,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
@@ -549,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;}
 
@@ -651,6 +698,7 @@ AliESDtrack::~AliESDtrack(){
   delete fIp; 
   delete fTPCInner; 
   delete fOp;
+  delete fHMPIDp;
   delete fCp; 
   delete fFriendTrack;
   if(fTRDnSlices)
@@ -710,6 +758,19 @@ AliESDtrack &AliESDtrack::operator=(const AliESDtrack &source){
     fOp = 0;
   }
 
+  
+  if(source.fHMPIDp){
+    // we have the trackparam: assign or copy construct
+    if(fHMPIDp) *fHMPIDp = *source.fHMPIDp;
+    else fHMPIDp = new AliExternalTrackParam(*source.fHMPIDp);
+  }
+  else{
+    // no track param delete the old one
+    if(fHMPIDp)delete fHMPIDp;
+    fHMPIDp = 0;
+  }
+
+  
   // copy also the friend track 
   // use copy constructor
   if(source.fFriendTrack){
@@ -740,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]; 
@@ -779,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;      
@@ -813,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];    
   }
@@ -826,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; 
@@ -915,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];
@@ -922,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){
@@ -967,6 +1037,8 @@ void AliESDtrack::MakeMiniESDtrack(){
   delete fTPCInner;fTPCInner=0;
   // Reset track parameters at the inner wall of the TRD
   delete fOp;fOp = 0;
+  // Reset track parameters at the HMPID
+  delete fHMPIDp;fHMPIDp = 0;
 
 
   // Reset ITS track related information
@@ -980,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;      
@@ -1014,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
@@ -1035,7 +1113,7 @@ void AliESDtrack::MakeMiniESDtrack(){
   fHMPIDtrkY = 0;      
   fHMPIDmipX = 0;
   fHMPIDmipY = 0;
-  fEMCALindex = kEMCALNoMatch;
+  fCaloIndex = kEMCALNoMatch;
 
   // reset global track chi2
   fGlobalChi2 = 0;
@@ -1047,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
@@ -1068,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
 {
@@ -1099,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);
@@ -1119,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();
@@ -1138,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 
@@ -1162,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; 
 
@@ -1206,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);
@@ -1233,6 +1351,11 @@ Bool_t AliESDtrack::UpdateTrackParams(const AliKalmanTrack *t, ULong_t flags){
     break;
   case kTRDStop:
     break;
+  case kHMPIDout:
+  if (!fHMPIDp) fHMPIDp=new AliExternalTrackParam(*t);
+    else 
+      fHMPIDp->Set(t->GetX(),t->GetAlpha(),t->GetParameter(),t->GetCovariance());
+    break;
   default: 
     AliError("Wrong flag !");
     return kFALSE;
@@ -1317,6 +1440,16 @@ AliESDtrack::SetOuterParam(const AliExternalTrackParam *p, ULong_t flags) {
   fOp=new AliExternalTrackParam(*p);
 }
 
+void 
+AliESDtrack::SetOuterHmpParam(const AliExternalTrackParam *p, ULong_t flags) {
+  //
+  // This is a direct setter for the outer track parameters
+  //
+  SetStatus(flags);
+  if (fHMPIDp) delete fHMPIDp;
+  fHMPIDp=new AliExternalTrackParam(*p);
+}
+
 Bool_t 
 AliESDtrack::GetOuterExternalParameters
                  (Double_t &alpha, Double_t &x, Double_t p[5]) const {
@@ -1331,6 +1464,20 @@ AliESDtrack::GetOuterExternalParameters
   return kTRUE;
 }
 
+Bool_t 
+AliESDtrack::GetOuterHmpExternalParameters
+                 (Double_t &alpha, Double_t &x, Double_t p[5]) const {
+  //---------------------------------------------------------------------
+  // This function returns external representation of the track parameters 
+  // at the inner layer of TRD
+  //---------------------------------------------------------------------
+  if (!fHMPIDp) return kFALSE;
+  alpha=fHMPIDp->GetAlpha();
+  x=fHMPIDp->GetX();
+  for (Int_t i=0; i<5; i++) p[i]=fHMPIDp->GetParameter()[i];
+  return kTRUE;
+}
+
 Bool_t 
 AliESDtrack::GetOuterExternalCovariance(Double_t cov[15]) const {
  //---------------------------------------------------------------------
@@ -1342,6 +1489,17 @@ AliESDtrack::GetOuterExternalCovariance(Double_t cov[15]) const {
   return kTRUE;
 }
 
+Bool_t 
+AliESDtrack::GetOuterHmpExternalCovariance(Double_t cov[15]) const {
+ //---------------------------------------------------------------------
+ // This function returns external representation of the cov. matrix 
+ // at the inner layer of TRD
+ //---------------------------------------------------------------------
+  if (!fHMPIDp) return kFALSE;
+  for (Int_t i=0; i<15; i++) cov[i]=fHMPIDp->GetCovariance()[i];
+  return kTRUE;
+}
+
 Int_t AliESDtrack::GetNcls(Int_t idet) const
 {
   // Get number of clusters by subdetector index
@@ -1445,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;
 }
@@ -1517,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;
 }
@@ -1561,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;
 }
 
 //_______________________________________________________________________
@@ -1621,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.;
   }
 
@@ -1831,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) {
@@ -1886,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
@@ -1943,6 +2234,7 @@ void AliESDtrack::FillPolymarker(TPolyMarker3D *pol, Float_t magF, Float_t minR,
   arrayRef.AddLast(new AliExternalTrackParam(*this));
   if (fIp) arrayRef.AddLast(new AliExternalTrackParam(*fIp));
   if (fOp) arrayRef.AddLast(new AliExternalTrackParam(*fOp));
+  if (fHMPIDp) arrayRef.AddLast(new AliExternalTrackParam(*fHMPIDp));
   //
   Double_t mpos[3]={0,0,0};
   Int_t entries=arrayRef.GetEntries();
@@ -2004,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);
+}