- Implementing Central task for filtering the ESD friends:

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

diff --git a/STEER/AliESDHandler.cxx b/STEER/AliESDHandler.cxx
index 72519f90fcc1a23744018941a3fb444456d9b82b..47cc2452bd7a4d979d53b81b2e4c1f880f95de0c 100644 (file)
--- a/STEER/AliESDHandler.cxx
+++ b/STEER/AliESDHandler.cxx
@@ -33,7 +33,6 @@
 #include "AliESDEvent.h"
 #include "AliESDfriend.h"
 
-
 ClassImp(AliESDHandler)
 
 //______________________________________________________________________________
@@ -42,8 +41,10 @@ AliESDHandler::AliESDHandler() :
        fesdf(NULL),
        fTreeEF(NULL),
        fFileEF(NULL),
-       fFileName("AliESDfriends_v2.root")
+       fFileName("AliESDfriends_v1.root"),
+       fIsEventSelectedForFriends(kFALSE)
 {
+
        // default constructor
 }
 
@@ -53,7 +54,8 @@ AliESDHandler::AliESDHandler(const char* name, const char* title):
        fesdf(NULL),
        fTreeEF(NULL),
        fFileEF(NULL),
-       fFileName("AliESDfriends_v2.root")
+       fFileName("AliESDfriends_v1.root"),
+       fIsEventSelectedForFriends(kFALSE)
 {
 
        // constructor with name and title
@@ -120,7 +122,7 @@ Bool_t AliESDHandler::FinishEvent()
        FillTree();
        
        // resetting
-       fesdf->~AliESDfriend();
+       if (fesdf) fesdf->~AliESDfriend();
        new(fesdf) AliESDfriend();  
        return kTRUE;
 }
@@ -159,10 +161,13 @@ void AliESDHandler::FillTree()
        //
        // Fill the ESD Tree
        //
-
-       AliDebug(2,Form("number of friend tracks = %d\n",fesdf->GetNumberOfTracks()));
-
+       if (fIsEventSelectedForFriends){
+               AliDebug(2,Form("number of friend tracks = %d\n",fesdf->GetNumberOfTracks()));
+       }
+       else {
+               fesdf->SetSkipBit(kTRUE);
+       }
+       AliDebug(2,Form("friend = %p",fesdf));
        fFileEF->cd();
        fTreeEF->Fill();
 }
-