Fix for Savannah bug report 59287

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

diff --git a/STEER/AliESDfriend.h b/STEER/AliESDfriend.h
index b59da6a294f88c2db48c87f4efce66eff0704658..d2afe7000d39079fe53b5fb50990869a50789010 100644 (file)
--- a/STEER/AliESDfriend.h
+++ b/STEER/AliESDfriend.h
@@ -10,25 +10,33 @@
 #include <TObject.h>
 #include <TClonesArray.h>
 
-class AliESD;
-class AliESDfriendTrack;
+#include "AliESDfriendTrack.h"
+#include "AliESDVZEROfriend.h"
 
 //_____________________________________________________________________________
 class AliESDfriend : public TObject {
 public:
   AliESDfriend();
   AliESDfriend(const AliESDfriend &);
-  AliESDfriend(const AliESD &);
+  AliESDfriend& operator=(const AliESDfriend& esd);  
   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);
+  }
+
+  void SetVZEROfriend(AliESDVZEROfriend * obj);
+  AliESDVZEROfriend *GetVZEROfriend(){ return fESDVZEROfriend; }
 
 protected:
   TClonesArray fTracks;    // ESD friend tracks
-  ClassDef(AliESDfriend,1) // ESD friend
+  AliESDVZEROfriend *fESDVZEROfriend; // VZERO object containing complete raw data
+
+  ClassDef(AliESDfriend,2) // ESD friend
 };
 
 #endif