Added SetLabel() method

[u/mrichter/AliRoot.git] / STEER / AliAODTracklets.h

diff --git a/STEER/AliAODTracklets.h b/STEER/AliAODTracklets.h
index 4ad5068d67634efa68e4927ab90294d68792ea50..09c54119e2c97b68096e690b01ffa1515d744ab8 100644 (file)
--- a/STEER/AliAODTracklets.h
+++ b/STEER/AliAODTracklets.h
@@ -34,6 +34,7 @@ class AliAODTracklets : public TNamed
   inline Double32_t GetPhi(Int_t i) const;
   inline Double32_t GetDeltaPhi(Int_t i) const;
   inline Int_t   GetLabel(Int_t i, Int_t layer) const;
+  inline void    SetLabel(Int_t i, Int_t layer,Int_t label);
 
  protected:
   Int_t      fNTracks;       // Number of tracklets
@@ -87,4 +88,15 @@ Int_t AliAODTracklets::GetLabel(Int_t i, Int_t layer) const
     Error("GetLabel","Invalid track number %d",i); return -9999;
 }
 
+
+void AliAODTracklets::SetLabel(Int_t i, Int_t layer,Int_t label)  
+{
+  if (i>=0 && i<fNTracks) 
+  {
+    if(layer == 0) fLabels[i] = label;
+    else fLabelsL2[i] = label;
+  }
+}
+
+
 #endif