X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=STEER%2FAliESDfriend.cxx;h=5978620f3a9ad4e28e509649cadd4469181e632d;hb=2894a0c933f304a64c12524138f980a805323c0f;hp=f8b9efa133b134ba1a67a9b69b684d6c64c3eb3f;hpb=1d99986f6f56caf1bc5004c841ce08dd56fc0c05;p=u%2Fmrichter%2FAliRoot.git

diff --git a/STEER/AliESDfriend.cxx b/STEER/AliESDfriend.cxx
index f8b9efa133b..5978620f3a9 100644
--- a/STEER/AliESDfriend.cxx
+++ b/STEER/AliESDfriend.cxx
@@ -21,41 +21,77 @@
 //-------------------------------------------------------------------------
 
 #include "AliESDfriend.h"
-#include "AliESDfriendTrack.h"
-#include "AliESD.h"
+#include "AliESDVZEROfriend.h"
+#include "AliESDTZEROfriend.h"
 
 ClassImp(AliESDfriend)
 
-AliESDfriend::AliESDfriend(): TObject(), fTracks("AliESDfriendTrack",15000)
+AliESDfriend::AliESDfriend(): TObject(), fTracks("AliESDfriendTrack",15000),
+  fESDVZEROfriend(NULL),
+  fESDTZEROfriend(NULL)
+
 {
  //
  // Default constructor
  //
 }
 
-AliESDfriend::AliESDfriend(const AliESDfriend &f):TObject(f),fTracks(f.fTracks)
+AliESDfriend::AliESDfriend(const AliESDfriend &f) :
+  TObject(f),
+  fTracks(f.fTracks),
+  fESDVZEROfriend(f.fESDVZEROfriend ? new AliESDVZEROfriend(*f.fESDVZEROfriend) : NULL),
+  fESDTZEROfriend(f.fESDTZEROfriend ? new AliESDTZEROfriend(*f.fESDTZEROfriend) : NULL)
 {
  //
  // Copy constructor
  //
 }
 
-AliESDfriend::AliESDfriend(const AliESD &event): TObject(event),
-fTracks("AliESDfriendTrack",event.GetNumberOfTracks()) {
-  //
-  // Extracts the additional info from the ESD
-  //
-  Int_t ntrk=event.GetNumberOfTracks();
+AliESDfriend& AliESDfriend::operator=(const AliESDfriend& esd)
+{
+    
+    // Assignment operator
+    if(&esd == this) return *this;
+    TObject::operator=(esd);
+    fTracks = esd.fTracks;
 
-  for (Int_t i=0; i<ntrk; i++) {
-    const AliESDtrack *t=event.GetTrack(i);
-    new (fTracks[fTracks.GetEntriesFast()]) AliESDfriendTrack(*t); 
-  }
+    delete fESDVZEROfriend; fESDVZEROfriend = NULL;
+    if (!esd.fESDVZEROfriend) fESDVZEROfriend = new AliESDVZEROfriend(*esd.fESDVZEROfriend);
+
+    delete fESDTZEROfriend; fESDTZEROfriend = NULL;
+    if (!esd.fESDTZEROfriend) fESDTZEROfriend = new AliESDTZEROfriend(*esd.fESDTZEROfriend);
+ 
+ 
+ 
+    return *this;
 }
 
+
+
 AliESDfriend::~AliESDfriend() {
   //
   // Destructor
   //
   fTracks.Delete();
+  delete fESDVZEROfriend;
+  delete fESDTZEROfriend;
+}
+
+
+
+void AliESDfriend::SetVZEROfriend(AliESDVZEROfriend * obj)
+{
+  //
+  // Set the VZERO friend data object
+  // (complete raw data)
+  if (!fESDVZEROfriend) fESDVZEROfriend = new AliESDVZEROfriend();
+  if (obj) *fESDVZEROfriend = *obj;
+}
+void AliESDfriend::SetTZEROfriend(AliESDTZEROfriend * obj)
+{
+  //
+  // Set the TZERO friend data object
+  // (complete raw data)
+  if (!fESDTZEROfriend) fESDTZEROfriend = new AliESDTZEROfriend();
+  if (obj) *fESDTZEROfriend = *obj;
 }