make friend access work in online and offline
authormkrzewic <mikolaj.krzewicki@cern.ch>
Fri, 19 Sep 2014 14:57:32 +0000 (16:57 +0200)
committermkrzewic <mikolaj.krzewicki@cern.ch>
Fri, 19 Sep 2014 14:57:32 +0000 (16:57 +0200)
STEER/ESD/AliESDEvent.cxx
STEER/ESD/AliESDEvent.h
STEER/ESD/AliESDInputHandler.cxx
STEER/ESD/AliESDInputHandler.h

index f1d8bd8..3103e7a 100644 (file)
@@ -1467,7 +1467,7 @@ void AliESDEvent::GetESDfriend(AliESDfriend *ev) const
 
   }
 
-  AliESDfriend *fr = (AliESDfriend*)(const_cast<AliESDEvent*>(this)->FindListObject("AliESDfriend"));
+  AliESDfriend *fr = (AliESDfriend*)(const_cast<AliESDEvent*>(this)->FindFriend());
   if (fr) ev->SetVZEROfriend(fr->GetVZEROfriend());
 }
 
@@ -2429,3 +2429,9 @@ void AliESDEvent::ConnectTracks() {
   //
 }
 
+//______________________________________________________________________________
+AliESDfriend* AliESDEvent::FindFriend() const 
+{ 
+  return static_cast<AliESDfriend*>(FindListObject("AliESDfriend")); 
+}
+
index 0dea219..793bea8 100644 (file)
@@ -274,7 +274,7 @@ public:
 
   void SetESDfriend(const AliESDfriend *f) const;
   void GetESDfriend(AliESDfriend *f) const;
-  AliESDfriend* FindFriend() const { return static_cast<AliESDfriend*>(FindListObject("AliESDfriend")); }
+  virtual AliESDfriend* FindFriend() const;
 
   void SetPrimaryVertexTPC(const AliESDVertex *vertex); 
   const AliESDVertex *GetPrimaryVertexTPC() const {return fTPCVertex;}
index f970399..76ed08e 100644 (file)
@@ -188,7 +188,7 @@ void AliESDInputHandler::ConnectFriends()
 
     cTree->AddFriend("esdFriendTree", esdFriendTreeFName.Data());
     cTree->SetBranchStatus("ESDfriend.", 1);
-    fFriend = (AliESDfriend*)(fEvent->FindListObject("AliESDfriend"));
+    fFriend = fEvent->FindFriend();
     if (fFriend) cTree->SetBranchAddress("ESDfriend.", &fFriend);
   }
 }
index 3b7706b..8a603ef 100644 (file)
@@ -67,6 +67,9 @@ class AliESDInputHandler : public AliInputEventHandler {
     virtual void CreatePIDResponse(Bool_t isMC=kFALSE);
     AliESDpid           *GetESDpid()       const {return fESDpid;}
     void                 SetESDpid(AliESDpid* pid)     {Changed(); fESDpid = pid;}
+
+    //HLT
+    virtual AliVfriendEvent*   GetVFriendEvent() const {return fFriend;};
   
  private:
     AliESDInputHandler(const AliESDInputHandler& handler);