Changes needed by the new TRD tracker (Alex)

[u/mrichter/AliRoot.git] / STEER / AliESDtrack.cxx

diff --git a/STEER/AliESDtrack.cxx b/STEER/AliESDtrack.cxx
index 2cc0903ee1d6b915aae95ddf6562628f1da94477..b048d0dc4ec305b172365f2895dc40a5bd70dac6 100644 (file)
--- a/STEER/AliESDtrack.cxx
+++ b/STEER/AliESDtrack.cxx
@@ -690,9 +690,11 @@ Bool_t AliESDtrack::UpdateTrackParams(const AliKalmanTrack *t, ULong_t flags){
   case kTRDout: case kTRDin: case kTRDrefit:
     index=fFriendTrack->GetTRDindices();
     fTRDLabel = t->GetLabel(); 
-    fTRDncls=t->GetNumberOfClusters();
     fTRDchi2=t->GetChi2();
-    for (Int_t i=0;i<fTRDncls;i++) index[i]=t->GetClusterIndex(i);
+    fTRDncls=6;//t->GetNumberOfTracklets(); //t->GetNumberOfClusters();
+    //for (Int_t i=0;i<fTRDncls;i++) index[i]=t->GetClusterIndex(i);
+    for (Int_t i=0;i<6;i++) index[i]=t->GetTrackletIndex(i);
+    
     fTRDsignal=t->GetPIDsignal();
     break;
   case kTRDbackup:
@@ -1009,6 +1011,18 @@ UChar_t AliESDtrack::GetTRDclusters(Int_t *idx) const {
   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];
+  }
+  return fTRDncls;
+}
+
 //_______________________________________________________________________
 void AliESDtrack::SetTRDpid(const Double_t *p) {  
   // Sets values for the probability of each particle type (in TRD)