Bug fix (Gustavo)

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

diff --git a/STEER/AliESDfriend.h b/STEER/AliESDfriend.h
index b59da6a294f88c2db48c87f4efce66eff0704658..def5faabec7ad55497b8aec1d05ecb696ed168c3 100644 (file)
--- a/STEER/AliESDfriend.h
+++ b/STEER/AliESDfriend.h
@@ -10,20 +10,21 @@
 #include <TObject.h>
 #include <TClonesArray.h>
 
-class AliESD;
-class AliESDfriendTrack;
+#include "AliESDfriendTrack.h"
 
 //_____________________________________________________________________________
 class AliESDfriend : public TObject {
 public:
   AliESDfriend();
   AliESDfriend(const AliESDfriend &);
-  AliESDfriend(const AliESD &);
   virtual ~AliESDfriend();
 
   Int_t GetNumberOfTracks() const {return fTracks.GetEntriesFast();}
   AliESDfriendTrack *GetTrack(Int_t i) const {
-    return (AliESDfriendTrack *)fTracks.UncheckedAt(i);
+     return (AliESDfriendTrack *)fTracks.UncheckedAt(i);
+  }
+  void AddTrack(const AliESDfriendTrack *t) {
+     new(fTracks[fTracks.GetEntriesFast()]) AliESDfriendTrack(*t);
   }
 
 protected: