Introducing the ESD friend classes for storing complementary to ESD information in...

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

diff --git a/STEER/AliESDfriend.h b/STEER/AliESDfriend.h
new file mode 100644 (file)
index 0000000..b59da6a
--- /dev/null
+++ b/STEER/AliESDfriend.h
@@ -0,0 +1,36 @@
+#ifndef ALIESDFRIEND_H
+#define ALIESDFRIEND_H
+
+//-------------------------------------------------------------------------
+//                     Class AliESDfriend
+//               This class contains ESD additions
+//       Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch 
+//-------------------------------------------------------------------------
+
+#include <TObject.h>
+#include <TClonesArray.h>
+
+class AliESD;
+class AliESDfriendTrack;
+
+//_____________________________________________________________________________
+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);
+  }
+
+protected:
+  TClonesArray fTracks;    // ESD friend tracks
+  ClassDef(AliESDfriend,1) // ESD friend
+};
+
+#endif
+
+