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

/**************************************************************************
 * Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

/* $Id$ */

//-------------------------------------------------------------------------
//     Event handler for AOD input 
//     Author: Andreas Morsch, CERN
//-------------------------------------------------------------------------

#include <TTree.h>
#include <TList.h>
#include <TNamed.h>

#include "AliAODInputHandler.h"
#include "AliAODEvent.h"

ClassImp(AliAODInputHandler)

//______________________________________________________________________________
AliAODInputHandler::AliAODInputHandler() :
    AliInputEventHandler(),
    fEvent(0),
    fFriends(new TList())
{
  // Default constructor
}

//______________________________________________________________________________
AliAODInputHandler::AliAODInputHandler(const char* name, const char* title):
  AliInputEventHandler(name, title),
  fEvent(0),
  fFriends(new TList())
{
    // Constructor
}

//______________________________________________________________________________
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) {
      delete fEvent;
      fEvent = 0;
    }
    fEvent = new AliAODEvent();

    fEvent->ReadFromTree(fTree);
    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);
}
Commit	Line	Data
	1	/**************************************************************************
	2	* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
	15
	16	/* $Id$ */
	17
	18	//-------------------------------------------------------------------------
	19	// Event handler for AOD input
	20	// Author: Andreas Morsch, CERN
	21	//-------------------------------------------------------------------------
	22
	23	#include <TTree.h>
	24	#include <TList.h>
	25	#include <TNamed.h>
	26
	27	#include "AliAODInputHandler.h"
	28	#include "AliAODEvent.h"
	29
	30	ClassImp(AliAODInputHandler)
	31
	32	//______________________________________________________________________________
	33	AliAODInputHandler::AliAODInputHandler() :
	34	AliInputEventHandler(),
	35	fEvent(0),
	36	fFriends(new TList())
	37	{
	38	// Default constructor
	39	}
	40
	41	//______________________________________________________________________________
	42	AliAODInputHandler::AliAODInputHandler(const char* name, const char* title):
	43	AliInputEventHandler(name, title),
	44	fEvent(0),
	45	fFriends(new TList())
	46	{
	47	// Constructor
	48	}
	49
	50	//______________________________________________________________________________
	51	AliAODInputHandler::~AliAODInputHandler()
	52	{
	53	// Destructor
	54	fFriends->Delete();
	55	}
	56
	57
	58	Bool_t AliAODInputHandler::Init(TTree* tree, Option_t* /opt/)
	59	{
	60	// Initialisation necessary for each new tree
	61	fTree = tree;
	62	TIter next(fFriends);
	63	TNamed* obj;
	64
	65	while((obj = (TNamed*)next())) {
	66	if (fTree->GetTree()) {
	67	(fTree->GetTree())->AddFriend("aodTree", obj->GetName());
	68	} else {
	69	fTree->AddFriend("aodTree", obj->GetName());
	70	}
	71	}
	72
	73	if (!fTree) return kFALSE;
	74	// Get pointer to AOD event
	75	if (fEvent) {
	76	delete fEvent;
	77	fEvent = 0;
	78	}
	79	fEvent = new AliAODEvent();
	80
	81	fEvent->ReadFromTree(fTree);
	82	return kTRUE;
	83	}
	84
	85	Bool_t AliAODInputHandler::BeginEvent(Long64_t /entry/)
	86	{
	87	//
	88	//if (fTree) fTree->BranchRef();
	89	return kTRUE;
	90	}
	91
	92	void AliAODInputHandler::AddFriend(char* filename)
	93	{
	94	// Add a friend tree
	95	TNamed* obj = new TNamed(filename, filename);
	96	fFriends->Add(obj);
	97	}