Do not reset a zero pointer to MC info

[u/mrichter/AliRoot.git] / STEER / AliESDInputHandler.h

diff --git a/STEER/AliESDInputHandler.h b/STEER/AliESDInputHandler.h
index 07db470a1381b9c8110fba1b5843e7f6bf72b996..62bd6bcddeea4173ac5df61c45eadebdc01e887c 100644 (file)
--- a/STEER/AliESDInputHandler.h
+++ b/STEER/AliESDInputHandler.h
@@ -16,6 +16,9 @@ class TChain;
 class TTree;
 class AliRunTag;
 class TMap;
+class AliESDfriend;
+class AliESDpid;
+
 
 class AliESDInputHandler : public AliInputEventHandler {
 
@@ -35,34 +38,49 @@ class AliESDInputHandler : public AliInputEventHandler {
     // Tag cut summary analysis
     Int_t                GetNEventAcceptedInFile();
     Int_t                GetNEventRejectedInFile();
-    // HLT analysis
+    Bool_t               GetCutSummaryForChain(Int_t *aTotal, Int_t *aAccepted, Int_t *aRejected);
+    Int_t                GetNFilesEmpty();
+    // HLT  analysis
     AliESDEvent         *GetHLTEvent()     const {return fHLTEvent;}
     TTree               *GetHLTTree()      const {return fHLTTree;}    
     void                SetReadHLT()             {fUseHLT = kTRUE;}
+    // Friends&Co
+    AliESDfriend        *GetESDfriend()    const {return fFriend;}
+    AliESDpid           *GetESDpid()       const {return fESDpid;}
+    void                 SetESDpid(AliESDpid* pid)     {fESDpid = pid;}
+    void                 SetReadFriends(Bool_t flag)   {fReadFriends = flag;}
+    void                 SetFriendFileName(const char *fname)  {fFriendFileName = fname;}
     // Tag analysis
     void SetReadTags() {fUseTags = kTRUE;}
     AliRunTag           *GetRunTag() const {return fRunTag;}
-           
+    
  private:
     AliESDInputHandler(const AliESDInputHandler& handler);             
     AliESDInputHandler& operator=(const AliESDInputHandler& handler);  
  protected:
     // ESD event
-    AliESDEvent    *fEvent;        //! Pointer to the event
-    Option_t       *fAnalysisType; //! local, proof, grid
-    Int_t           fNEvents;      //! Number of events in the current tree
+    AliESDEvent    *fEvent;         //! Pointer to the event
+    AliESDfriend   *fFriend;        //! Pointer to the esd friend
+    AliESDpid      *fESDpid;        //! Pointer to PID information
+    Option_t       *fAnalysisType;  //! local, proof, grid
+    Int_t           fNEvents;       //! Number of events in the current tree
     // HLT event
-    AliESDEvent    *fHLTEvent;     //! Pointer to the HLT Event (if present)
-    TTree          *fHLTTree;      //! Pointer to the HLT Event (if present)
-    Bool_t          fUseHLT;       //  Flag to access HLT Events
+    AliESDEvent    *fHLTEvent;      //! Pointer to the HLT Event (if present)
+    TTree          *fHLTTree;       //! Pointer to the HLT Event (if present)
+    Bool_t          fUseHLT;        //  Flag to access HLT Events
     // ESD Tag Cut Summary
-    TMap           *fTagCutSumm;   //! Tag cut summary map
+    TMap           *fTagCutSumm;    //! Tag cut summary map
     // ESD Tags (optional)
-    Bool_t          fUseTags;    //  Flag to use tags
-    TChain         *fChainT;     //! File with event tags
-    TTree          *fTreeT;      //! Tree of tags
-    AliRunTag      *fRunTag;     //! Pointer to the run tag
-    ClassDef(AliESDInputHandler, 4);
+    Bool_t          fUseTags;       //  Flag to use tags
+    TChain         *fChainT;        //! File with event tags
+    TTree          *fTreeT;         //! Tree of tags
+    AliRunTag      *fRunTag;        //! Pointer to the run tag
+    // Friends
+    Bool_t          fReadFriends;   //  Flag for friends reading 
+    TString         fFriendFileName;//  Name of the file containing the frien tree (branch)
+    
+
+    ClassDef(AliESDInputHandler, 6);
 };
 
 #endif