Possibility to filter the ESD friends and add objects to AliESDfriends.root. Major...

[u/mrichter/AliRoot.git] / ANALYSIS / AliAnalysisTaskFilter.cxx

/**************************************************************************
 * Copyright(c) 1998-1999, 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$ */

//////////////////////////////////////////////////////////////////////////
//
//  Base class for filtering friends
//
//////////////////////////////////////////////////////////////////////////
 
#include <TChain.h>
#include <TFile.h>
#include <TList.h>

#include "AliAnalysisTaskFilter.h"
#include "AliAnalysisManager.h"
#include "AliAnalysisDataSlot.h"
#include "AliESDEvent.h"
#include "AliESD.h"
#include "AliVEvent.h"
#include "AliESDHandler.h"
#include "AliInputEventHandler.h"
#include "AliLog.h"
#include "AliESDfriend.h"
#include "AliESDfriendTrack.h"


ClassImp(AliAnalysisTaskFilter)

////////////////////////////////////////////////////////////////////////

AliAnalysisTaskFilter::AliAnalysisTaskFilter():
	AliAnalysisTask(),
	fDebug(0),
	fEntry(0),
	fInputEvent(0x0),
	fInputHandler(0x0),
	fOutputESD(0x0),
	fOutputESDfriend(0x0),
	fTreeE(0x0)
{
	//
	// Default constructor
	//
}

//______________________________________________________________________

AliAnalysisTaskFilter::AliAnalysisTaskFilter(const char* name):
	AliAnalysisTask(name, "AnalysisTaskFilter"),
	fDebug(0),
	fEntry(0),
	fInputEvent(0x0),
	fInputHandler(0x0),
	fOutputESD(0x0),
	fOutputESDfriend(0x0),
	fTreeE(0x0)
{
	//
	// Default constructor
	//

	DefineInput (0, TChain::Class());
	DefineOutput(0,  TTree::Class());
}

//______________________________________________________________________

AliAnalysisTaskFilter::AliAnalysisTaskFilter(const AliAnalysisTaskFilter& obj):
	AliAnalysisTask(obj),
	fDebug(0),
	fEntry(0),
	fInputEvent(0x0),
	fInputHandler(0x0),
	fOutputESD(0x0),
	fOutputESDfriend(0x0),
	fTreeE(0x0)
{
	//
	// Copy constructor
	//

	fDebug        = obj.fDebug;
	fEntry        = obj.fEntry;
	fInputEvent   = obj.fInputEvent;
	fInputHandler = obj.fInputHandler;
	fOutputESD    = obj.fOutputESD;
	fOutputESDfriend = obj.fOutputESDfriend;
	fTreeE        = obj.fTreeE;    
}


//______________________________________________________________________

AliAnalysisTaskFilter& AliAnalysisTaskFilter::operator=(const AliAnalysisTaskFilter& other)
{
	//
	// Assignment
	//

	AliAnalysisTask::operator=(other);
	fDebug        = other.fDebug;
	fEntry        = other.fEntry;
	fInputEvent   = other.fInputEvent;
	fInputHandler = other.fInputHandler;
	fOutputESD    = other.fOutputESD;
	fOutputESDfriend = other.fOutputESDfriend;
	fTreeE        = other.fTreeE;    
	return *this;
}


//______________________________________________________________________

void AliAnalysisTaskFilter::ConnectInputData(Option_t* /*option*/)
{
	//
	// Connect the input data
	//

	if (fDebug > 1) printf("AnalysisTaskFilter::ConnectInputData() \n");
	fInputHandler = (AliInputEventHandler*) 
		((AliAnalysisManager::GetAnalysisManager())->GetInputEventHandler());
	if (fInputHandler) {
		fInputEvent = fInputHandler->GetEvent();
	} else {
		AliError("No Input Event Handler connected") ; 
		return ; 
	}
}

//______________________________________________________________________

void AliAnalysisTaskFilter::CreateOutputObjects()
{
	//
	// Create the output container
	//

	if (fDebug > 1) printf("AnalysisTaskFilter::CreateOutPutData() \n");

	AliESDHandler* handler = (AliESDHandler*) ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler());
    
	if (handler) {
		fOutputESD   = handler->GetESD();
		fTreeE = handler->GetTree();
		fOutputESD->GetStdContent();
	}
	else {
		AliWarning("No AOD Event Handler connected.") ; 
	}

	UserCreateOutputObjects();
}

//______________________________________________________________________

void AliAnalysisTaskFilter::Exec(Option_t* option)
{
	//
	// Exec analysis of one event
	//

	if (fDebug > 1) AliInfo("AliAnalysisTaskFilter::Exec() \n");

	if( fInputHandler ) {
		fEntry = fInputHandler->GetReadEntry();
	}
    
	   
	if ( !((Entry()-1)%100) && fDebug > 0) {
		AliInfo(Form("%s ----> Processing event # %lld", CurrentFileName(), Entry()));
	}
	AliESDHandler* handler = (AliESDHandler*)((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler());
	
	if (UserSelectESDfriendForCurrentEvent()){
		// Call the user analysis only if the event was selected   
		fOutputESDfriend   = handler->GetESDfriend();
		UserExec(option);
	}

	// Added protection in case the derived task is not an AOD producer.
	AliAnalysisDataSlot *out0 = GetOutputSlot(0);
	if (out0 && out0->IsConnected()) PostData(0, fTreeE);    
}

//______________________________________________________________________

const char* AliAnalysisTaskFilter::CurrentFileName()
{
	// Returns the current file name    
	if( fInputHandler ){
		return fInputHandler->GetTree()->GetCurrentFile()->GetName();
	}
	else return "";
}

//______________________________________________________________________

void AliAnalysisTaskFilter::AddFriendTrackAt(AliESDfriendTrack* t, Int_t index)
{
	//
	// Adds the friend track at the i-th position in the TClonesArray
	// of the ESD friend tracks
	//

	AliESDfriendTrack* currentTrack = (AliESDfriendTrack*)fOutputESDfriend->GetTrack(index);
	if(currentTrack && currentTrack->GetCalibObject(0)){
		AliWarning("Friend already there");
	}
	else {
		//		AliInfo("Adding track");
		fOutputESDfriend->AddTrackAt(t,index);
	}
}