//-------------------------------------------------------------------------
#include <TTree.h>
+#include <TList.h>
+#include <TNamed.h>
#include "AliAODInputHandler.h"
#include "AliAODEvent.h"
//______________________________________________________________________________
AliAODInputHandler::AliAODInputHandler() :
AliInputEventHandler(),
- fEvent(0)
+ fEvent(0),
+ fFriends(new TList())
{
- // default constructor
+ // Default constructor
}
//______________________________________________________________________________
-AliAODInputHandler::~AliAODInputHandler()
+AliAODInputHandler::AliAODInputHandler(const char* name, const char* title):
+ AliInputEventHandler(name, title),
+ fEvent(0),
+ fFriends(new TList())
{
-// destructor
+ // Constructor
}
//______________________________________________________________________________
-AliAODInputHandler::AliAODInputHandler(const char* name, const char* title):
- AliInputEventHandler(name, title)
+AliAODInputHandler::~AliAODInputHandler()
{
+// Destructor
+ fFriends->Delete();
}
+
Bool_t AliAODInputHandler::Init(TTree* tree, Option_t* /*opt*/)
{
// Initialisation necessary for each new tree
fTree = tree;
+ TIter next(fFriends);
+ TNamed* obj;
+
+ while(obj = (TNamed*)next()) {
+ if (fTree->GetTree()) {
+ (fTree->GetTree())->AddFriend("aodTree", obj->GetName());
+ } else {
+ fTree->AddFriend("aodTree", obj->GetName());
+ }
+ }
+
if (!fTree) return kFALSE;
// Get pointer to AOD event
if (fEvent) {
return kTRUE;
}
+Bool_t AliAODInputHandler::BeginEvent(Long64_t /*entry*/)
+{
+ //
+ //if (fTree) fTree->BranchRef();
+ return kTRUE;
+}
+
+void AliAODInputHandler::AddFriend(char* filename)
+{
+ // Add a friend tree
+ TNamed* obj = new TNamed(filename, filename);
+ fFriends->Add(obj);
+}