Protection added

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

diff --git a/STEER/AliESDfriend.cxx b/STEER/AliESDfriend.cxx
index 32158935b8343378b7f1ba4607e6ea675d992123..0bf592586862c5aedc4990a94f31925614ca8eae 100644 (file)
--- a/STEER/AliESDfriend.cxx
+++ b/STEER/AliESDfriend.cxx
@@ -32,14 +32,32 @@ AliESDfriend::AliESDfriend(): TObject(), fTracks("AliESDfriendTrack",15000),
  //
 }
 
-AliESDfriend::AliESDfriend(const AliESDfriend &f):TObject(f),fTracks(f.fTracks),
-                                                 fESDVZEROfriend(f.fESDVZEROfriend)
+AliESDfriend::AliESDfriend(const AliESDfriend &f) :
+  TObject(f),
+  fTracks(f.fTracks),
+  fESDVZEROfriend(f.fESDVZEROfriend ? new AliESDVZEROfriend(*f.fESDVZEROfriend) : NULL)
 {
  //
  // Copy constructor
  //
 }
 
+AliESDfriend& AliESDfriend::operator=(const AliESDfriend& esd)
+{
+    
+    // Assignment operator
+    if(&esd == this) return *this;
+    TObject::operator=(esd);
+    fTracks = esd.fTracks;
+
+    delete fESDVZEROfriend; fESDVZEROfriend = NULL;
+    if (!esd.fESDVZEROfriend) fESDVZEROfriend = new AliESDVZEROfriend(*esd.fESDVZEROfriend);
+
+    return *this;
+}
+
+
+
 AliESDfriend::~AliESDfriend() {
   //
   // Destructor