[u/mrichter/AliRoot.git] / PWGLF / SPECTRA / AntiprotonToProton / AliProtonCorrectionAnalysisTask.h

#ifndef ALIPROTONCORRECTIONTASK_H
#define ALIPROTONCORRECTIONTASK_H
#include "AliAnalysisTask.h"

class TList;
class AliESDEvent;
class AliAODEvent;
class AliMCEvent;
class AliProtonFeedDownAnalysis;
class AliProtonAbsorptionCorrection;
class AliProtonSpectraCorrection; 
class AliProtonAnalysisBase;


class AliProtonCorrectionAnalysisTask : public AliAnalysisTask {
 public:
	AliProtonCorrectionAnalysisTask();
	AliProtonCorrectionAnalysisTask(const char *name);
	virtual ~AliProtonCorrectionAnalysisTask() {}
	
	virtual void   ConnectInputData(Option_t *);
	virtual void   CreateOutputObjects();
	virtual void   Exec(Option_t *option);
	virtual void   Terminate(Option_t *);
	
	void SetAnalysisObjectAbsorptionCorrection(AliProtonAbsorptionCorrection* const analysis) ;
	void SetAnalysisObjectFeedDown(AliProtonFeedDownAnalysis* const analysis); 
	void SetAnalysisObjectSpectraCorrection(AliProtonSpectraCorrection* const analysis);
    	void SetBaseAnalysis(AliProtonAnalysisBase* const baseAnalysis) { fProtonAnalysisBase = baseAnalysis;}
 private:
	AliESDEvent *fESD;    //ESD object 
	AliAODEvent *fAOD;    //AOD object
	AliMCEvent  *fMC;     //MC object 
	
	TList  *fList; //TList output object 
	
	AliProtonAnalysisBase *fProtonAnalysisBase; //base analysis object
	AliProtonAbsorptionCorrection *fProtonAbsorptionCorrection;//analysis object 
	AliProtonFeedDownAnalysis *fProtonFeedDownAnalysis; //analysis object 
	AliProtonSpectraCorrection *fProtonSpectraCorrection;//analysis object 
	
	TH1F *fStatHist;
	
	
	Bool_t fIsOn_AliProtonAbsorptionCorrection;
	Bool_t fIsOn_AliProtonFeedDownAnalysis;
	Bool_t fIsOn_AliProtonSpectraCorrection;
	
	AliProtonCorrectionAnalysisTask(const AliProtonCorrectionAnalysisTask&); // not implemented
	AliProtonCorrectionAnalysisTask& operator=(const AliProtonCorrectionAnalysisTask&); // not implemented
	
ClassDef(AliProtonCorrectionAnalysisTask, 1); // example of analysis
};

#endif
Commit	Line	Data
a65a7e70	1	#ifndef ALIPROTONCORRECTIONTASK_H
	2	#define ALIPROTONCORRECTIONTASK_H
	3	#include "AliAnalysisTask.h"
	4
	5	class TList;
	6	class AliESDEvent;
	7	class AliAODEvent;
	8	class AliMCEvent;
	9	class AliProtonFeedDownAnalysis;
	10	class AliProtonAbsorptionCorrection;
	11	class AliProtonSpectraCorrection;
	12	class AliProtonAnalysisBase;
	13
	14
	15
	16	class AliProtonCorrectionAnalysisTask : public AliAnalysisTask {
	17	public:
	18	AliProtonCorrectionAnalysisTask();
	19	AliProtonCorrectionAnalysisTask(const char *name);
	20	virtual ~AliProtonCorrectionAnalysisTask() {}
	21
	22	virtual void ConnectInputData(Option_t *);
	23	virtual void CreateOutputObjects();
	24	virtual void Exec(Option_t *option);
	25	virtual void Terminate(Option_t *);
	26
	27	void SetAnalysisObjectAbsorptionCorrection(AliProtonAbsorptionCorrection* const analysis) ;
	28	void SetAnalysisObjectFeedDown(AliProtonFeedDownAnalysis* const analysis);
	29	void SetAnalysisObjectSpectraCorrection(AliProtonSpectraCorrection* const analysis);
	30	void SetBaseAnalysis(AliProtonAnalysisBase* const baseAnalysis) { fProtonAnalysisBase = baseAnalysis;}
	31	private:
	32	AliESDEvent *fESD; //ESD object
	33	AliAODEvent *fAOD; //AOD object
	34	AliMCEvent *fMC; //MC object
	35
	36	TList *fList; //TList output object
	37
	38	AliProtonAnalysisBase *fProtonAnalysisBase; //base analysis object
	39	AliProtonAbsorptionCorrection *fProtonAbsorptionCorrection;//analysis object
	40	AliProtonFeedDownAnalysis *fProtonFeedDownAnalysis; //analysis object
	41	AliProtonSpectraCorrection *fProtonSpectraCorrection;//analysis object
	42
	43	TH1F *fStatHist;
	44
	45
	46	Bool_t fIsOn_AliProtonAbsorptionCorrection;
	47	Bool_t fIsOn_AliProtonFeedDownAnalysis;
	48	Bool_t fIsOn_AliProtonSpectraCorrection;
	49
	50	AliProtonCorrectionAnalysisTask(const AliProtonCorrectionAnalysisTask&); // not implemented
	51	AliProtonCorrectionAnalysisTask& operator=(const AliProtonCorrectionAnalysisTask&); // not implemented
	52
	53	ClassDef(AliProtonCorrectionAnalysisTask, 1); // example of analysis
	54	};
	55
	56	#endif
	57
	58
	59