Bug corrected.

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

diff --git a/STEER/AliESDfriend.cxx b/STEER/AliESDfriend.cxx
index 5f4209743e710879a91e4852702de433dbb10b86..5978620f3a9ad4e28e509649cadd4469181e632d 100644 (file)
--- a/STEER/AliESDfriend.cxx
+++ b/STEER/AliESDfriend.cxx
@@ -21,22 +21,26 @@
 //-------------------------------------------------------------------------
 
 #include "AliESDfriend.h"
+#include "AliESDVZEROfriend.h"
+#include "AliESDTZEROfriend.h"
 
 ClassImp(AliESDfriend)
 
 AliESDfriend::AliESDfriend(): TObject(), fTracks("AliESDfriendTrack",15000),
-                             fESDVZEROfriend(NULL)
+  fESDVZEROfriend(NULL),
+  fESDTZEROfriend(NULL)
+
 {
  //
  // Default constructor
  //
-       //fTracks.BypassStreamer(kFALSE);
 }
 
 AliESDfriend::AliESDfriend(const AliESDfriend &f) :
   TObject(f),
   fTracks(f.fTracks),
-  fESDVZEROfriend(f.fESDVZEROfriend ? new AliESDVZEROfriend(*f.fESDVZEROfriend) : NULL)
+  fESDVZEROfriend(f.fESDVZEROfriend ? new AliESDVZEROfriend(*f.fESDVZEROfriend) : NULL),
+  fESDTZEROfriend(f.fESDTZEROfriend ? new AliESDTZEROfriend(*f.fESDTZEROfriend) : NULL)
 {
  //
  // Copy constructor
@@ -54,6 +58,11 @@ AliESDfriend& AliESDfriend::operator=(const AliESDfriend& esd)
     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;
 }
 
@@ -65,8 +74,11 @@ AliESDfriend::~AliESDfriend() {
   //
   fTracks.Delete();
   delete fESDVZEROfriend;
+  delete fESDTZEROfriend;
 }
 
+
+
 void AliESDfriend::SetVZEROfriend(AliESDVZEROfriend * obj)
 {
   //
@@ -75,3 +87,11 @@ void AliESDfriend::SetVZEROfriend(AliESDVZEROfriend * obj)
   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;
+}