[u/mrichter/AliRoot.git] / PWGCF / FEMTOSCOPYAOD / AliAnalysisTaskPWG2ESDfilter.h

//-------------------------------------------------------------------------
//     Task for the Analysis Framework 
// Creates a PWG2 specific AOD, from the ESD, adding information required
// for PWG2 analysis. 
//  - Puts the per-track information into the AliPWG2AODTrack container, 
//    together with the link to the original AliAODTrack
//
//     Author: Adam Kisiel, OSU, Adam.Kisiel@cern.ch
//-------------------------------------------------------------------------
#ifndef ALIANALYSISTASKPWG2ESDFILTER_H
#define ALIANALYSISTASKPWG2ESDFILTER_H
 
#include <AliAnalysisTask.h>

class AliESDEvent;
class TChain;
class AliAODEvent;
class AliAnalysisFilter;
class TTree;
class TClonesArray;
class TList;

class AliAnalysisTaskPWG2ESDfilter : public AliAnalysisTask
{
 public:
    AliAnalysisTaskPWG2ESDfilter();
    AliAnalysisTaskPWG2ESDfilter(const char* name);
    AliAnalysisTaskPWG2ESDfilter(const AliAnalysisTaskPWG2ESDfilter &task); 
    virtual ~AliAnalysisTaskPWG2ESDfilter() {;}

    AliAnalysisTaskPWG2ESDfilter& operator=(const AliAnalysisTaskPWG2ESDfilter &task);
    // Implementation of interface methods
    virtual void ConnectInputData(Option_t *option = "");
    virtual void CreateOutputObjects();
    virtual void Init();
    virtual void LocalInit() {Init();}
    virtual void Exec(Option_t *option);
    virtual void Terminate(Option_t *option);
    // Setters
    virtual void SetTrackFilter(AliAnalysisFilter* trackF) {fTrackFilter = trackF;}
    virtual void SetKinkFilter (AliAnalysisFilter*  KinkF) {fKinkFilter  =  KinkF;}
    virtual void SetV0Filter   (AliAnalysisFilter*    V0F) {fV0Filter    =    V0F;}
    virtual void SetDebugLevel(Int_t level) {fDebug = level;}
    
 private:
    Int_t              fDebug;         //  Debug flag
    TTree*             fTree;          //! chained files
    AliESDEvent*       fESD;           //! ESD
    AliAODEvent*       fAOD;           //! AOD event 
    TTree*             fTreeA;         //! AOD tree
    AliAnalysisFilter* fTrackFilter;   //  Track Filter
    AliAnalysisFilter* fKinkFilter;    //  Kink  Filter
    AliAnalysisFilter* fV0Filter;      //  V0    Filter    
    TClonesArray*      fPWG2AODTracks; //! container for PWG2 specific information

    ClassDef(AliAnalysisTaskPWG2ESDfilter, 1); // Analysis task for standard ESD filtering
};
 
#endif
Commit	Line	Data
f9228736	1	//-------------------------------------------------------------------------
	2	// Task for the Analysis Framework
	3	// Creates a PWG2 specific AOD, from the ESD, adding information required
	4	// for PWG2 analysis.
	5	// - Puts the per-track information into the AliPWG2AODTrack container,
	6	// together with the link to the original AliAODTrack
	7	//
	8	// Author: Adam Kisiel, OSU, Adam.Kisiel@cern.ch
	9	//-------------------------------------------------------------------------
	10	#ifndef ALIANALYSISTASKPWG2ESDFILTER_H
	11	#define ALIANALYSISTASKPWG2ESDFILTER_H
	12
	13	#include <AliAnalysisTask.h>
	14
	15	class AliESDEvent;
	16	class TChain;
	17	class AliAODEvent;
	18	class AliAnalysisFilter;
	19	class TTree;
	20	class TClonesArray;
	21	class TList;
	22
	23	class AliAnalysisTaskPWG2ESDfilter : public AliAnalysisTask
	24	{
	25	public:
	26	AliAnalysisTaskPWG2ESDfilter();
	27	AliAnalysisTaskPWG2ESDfilter(const char* name);
	28	AliAnalysisTaskPWG2ESDfilter(const AliAnalysisTaskPWG2ESDfilter &task);
	29	virtual ~AliAnalysisTaskPWG2ESDfilter() {;}
	30
	31	AliAnalysisTaskPWG2ESDfilter& operator=(const AliAnalysisTaskPWG2ESDfilter &task);
	32	// Implementation of interface methods
	33	virtual void ConnectInputData(Option_t *option = "");
	34	virtual void CreateOutputObjects();
	35	virtual void Init();
	36	virtual void LocalInit() {Init();}
	37	virtual void Exec(Option_t *option);
	38	virtual void Terminate(Option_t *option);
	39	// Setters
	40	virtual void SetTrackFilter(AliAnalysisFilter* trackF) {fTrackFilter = trackF;}
	41	virtual void SetKinkFilter (AliAnalysisFilter* KinkF) {fKinkFilter = KinkF;}
	42	virtual void SetV0Filter (AliAnalysisFilter* V0F) {fV0Filter = V0F;}
	43	virtual void SetDebugLevel(Int_t level) {fDebug = level;}
	44
	45	private:
	46	Int_t fDebug; // Debug flag
	47	TTree* fTree; //! chained files
	48	AliESDEvent* fESD; //! ESD
	49	AliAODEvent* fAOD; //! AOD event
	50	TTree* fTreeA; //! AOD tree
	51	AliAnalysisFilter* fTrackFilter; // Track Filter
	52	AliAnalysisFilter* fKinkFilter; // Kink Filter
	53	AliAnalysisFilter* fV0Filter; // V0 Filter
	54	TClonesArray* fPWG2AODTracks; //! container for PWG2 specific information
	55
	56	ClassDef(AliAnalysisTaskPWG2ESDfilter, 1); // Analysis task for standard ESD filtering
	57	};
	58
	59	#endif