Possibility to add a mixing handler to the input.

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

diff --git a/STEER/AliAODInputHandler.cxx b/STEER/AliAODInputHandler.cxx
index 3f2a432b11d97ec402734ad0184591ec101301d5..85fc19482468686ee82b662bbcde46e113d9455f 100644 (file)
--- a/STEER/AliAODInputHandler.cxx
+++ b/STEER/AliAODInputHandler.cxx
@@ -70,7 +70,7 @@ AliAODInputHandler::~AliAODInputHandler()
 }
 
 
-Bool_t AliAODInputHandler::Init(TTree* tree, Option_t* /*opt*/)
+Bool_t AliAODInputHandler::Init(TTree* tree, Option_t* opt)
 {
     // Initialisation necessary for each new tree
     if (!fMergeEvents) {
@@ -118,6 +118,8 @@ Bool_t AliAODInputHandler::Init(TTree* tree, Option_t* /*opt*/)
     if (!fEvent) fEvent = new AliAODEvent();
 
     fEvent->ReadFromTree(fTree);
+    
+    if (fMixingHandler) fMixingHandler->Init(tree, opt);
 
     return kTRUE;
 }
@@ -130,10 +132,26 @@ Bool_t AliAODInputHandler::BeginEvent(Long64_t entry)
     if (fTreeToMerge) fTreeToMerge->GetEntry(entry + fMergeOffset);
     
     fIsSelectedResult = fEvent->GetHeader()->GetOfflineTrigger();
+
+    if (fMixingHandler) fMixingHandler->BeginEvent(entry);
     
     return kTRUE;
 }
 
+Bool_t AliAODInputHandler::Notify(const char* path)
+{
+  // Notifaction of directory change
+  if (fMixingHandler) fMixingHandler->Notify(path);
+  return kTRUE;
+}
+
+Bool_t AliAODInputHandler::FinishEvent()
+{
+  // Finish event
+  if (fMixingHandler) fMixingHandler->FinishEvent();
+  return kTRUE;
+}
+
 void AliAODInputHandler::AddFriend(char* filename)
 {
     // Add a friend tree