STEER/AliESDfriend.h

   1 #ifndef ALIESDFRIEND_H
   2 #define ALIESDFRIEND_H
   3
   4 //-------------------------------------------------------------------------
   5 //                     Class AliESDfriend
   6 //               This class contains ESD additions
   7 //       Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch
   8 //-------------------------------------------------------------------------
   9
  10 #include <TObject.h>
  11 #include <TClonesArray.h>
  12
  13 #include "AliESDfriendTrack.h"
  14 #include "AliESDVZEROfriend.h"
  15
  16 //_____________________________________________________________________________
  17 class AliESDfriend : public TObject {
  18 public:
  19   AliESDfriend();
  20   AliESDfriend(const AliESDfriend &);
  21   AliESDfriend& operator=(const AliESDfriend& esd);
  22   virtual ~AliESDfriend();
  23
  24   Int_t GetNumberOfTracks() const {return fTracks.GetEntriesFast();}
  25   AliESDfriendTrack *GetTrack(Int_t i) const {
  26      return (AliESDfriendTrack *)fTracks.UncheckedAt(i);
  27   }
  28   void AddTrack(const AliESDfriendTrack *t) {
  29      new(fTracks[fTracks.GetEntriesFast()]) AliESDfriendTrack(*t);
  30   }
  31
  32   void SetVZEROfriend(AliESDVZEROfriend * obj);
  33   AliESDVZEROfriend *GetVZEROfriend(){ return fESDVZEROfriend; }
  34
  35 protected:
  36   TClonesArray fTracks;    // ESD friend tracks
  37   AliESDVZEROfriend *fESDVZEROfriend; // VZERO object containing complete raw data
  38
  39   ClassDef(AliESDfriend,2) // ESD friend
  40 };
  41
  42 #endif
  43
  44