[u/mrichter/AliRoot.git] / PWGGA / GammaConv / AliAnalysisTaskResolution.h

#ifndef AliAnalysisTaskResolution_cxx
#define AliAnalysisTaskResolution_cxx

#include "AliAnalysisTaskSE.h"
#include "AliConversionPhotonBase.h"
#include "TH1.h"
#include "TH2.h"
#include "TTreeStream.h"
#include "AliLog.h"
#include <vector>
#include "AliV0ReaderV1.h"
#include "AliConvEventCuts.h"
#include "AliConversionPhotonCuts.h"
#include "TList.h"
#include "AliStack.h"
#include "TClonesArray.h"


using namespace std;


class AliAnalysisTaskResolution : public AliAnalysisTaskSE{

	public:
		AliAnalysisTaskResolution();
		AliAnalysisTaskResolution(const char *name);
		virtual ~AliAnalysisTaskResolution();

		virtual void   UserCreateOutputObjects();
		virtual void   UserExec(Option_t *option);
		virtual void   Terminate(Option_t *);

		void SetV0Reader(AliV0ReaderV1 *v0Reader){fV0Reader=v0Reader;}
		void SetConversionCuts(AliConversionPhotonCuts* conversionCuts,Int_t IsHeavyIon ){
			fConversionCuts=conversionCuts;
			fIsHeavyIon = IsHeavyIon;
		}
		void SetEventCuts(AliConvEventCuts* conversionCuts,Int_t IsHeavyIon ){
			fEventCuts=conversionCuts;
			fIsHeavyIon = IsHeavyIon;
		}
		void SetIsMC(Bool_t isMC){fIsMC=isMC;}
	
	private:

		void ProcessPhotons();
		Int_t CountTracks0914();
		Int_t CountTracks09();

		AliV0ReaderV1 *fV0Reader;
		TClonesArray *fConversionGammas; //Reconstructed Photons;
		AliConvEventCuts *fEventCuts; // Cuts used by the V0Reader
		AliConversionPhotonCuts *fConversionCuts; // Cuts used by the V0Reader
		TTree *fTreeEvent;
		TTree *fTreeResolution;
		Float_t fPrimVtxZ;
		Int_t fNContrVtx;
		Int_t fNESDtracksEta09;
		Int_t fNESDtracksEta0914;
		Int_t fNESDtracksEta14;
		TVectorF fGammaRecCoords;
		TVectorF fGammaMCCoords;
		Float_t fChi2ndf;
		Int_t fIsHeavyIon;
		Bool_t fIsMC;
		TList *fOutputList;
		TList *fEventList;
		TList *fResolutionList;
		AliESDEvent *fESDEvent;
		AliMCEvent *fMCEvent;

		AliAnalysisTaskResolution(const AliAnalysisTaskResolution&); // not implemented
		AliAnalysisTaskResolution& operator=(const AliAnalysisTaskResolution&); // not implemented


	ClassDef(AliAnalysisTaskResolution, 2);
};

#endif
Commit	Line	Data
72395bd9	1	#ifndef AliAnalysisTaskResolution_cxx
	2	#define AliAnalysisTaskResolution_cxx
	3
	4	#include "AliAnalysisTaskSE.h"
	5	#include "AliConversionPhotonBase.h"
	6	#include "TH1.h"
	7	#include "TH2.h"
	8	#include "TTreeStream.h"
	9	#include "AliLog.h"
	10	#include <vector>
	11	#include "AliV0ReaderV1.h"
344100c4	12	#include "AliConvEventCuts.h"
344100c4	13	#include "AliConversionPhotonCuts.h"
72395bd9	14	#include "TList.h"
	15	#include "AliStack.h"
	16	#include "TClonesArray.h"
	17
	18
	19	using namespace std;
	20
	21
	22	class AliAnalysisTaskResolution : public AliAnalysisTaskSE{
	23
63674d67	24	public:
	25	AliAnalysisTaskResolution();
	26	AliAnalysisTaskResolution(const char *name);
	27	virtual ~AliAnalysisTaskResolution();
72395bd9	28
63674d67	29	virtual void UserCreateOutputObjects();
	30	virtual void UserExec(Option_t *option);
	31	virtual void Terminate(Option_t *);
72395bd9	32
63674d67	33	void SetV0Reader(AliV0ReaderV1 *v0Reader){fV0Reader=v0Reader;}
344100c4	34	void SetConversionCuts(AliConversionPhotonCuts* conversionCuts,Int_t IsHeavyIon ){
63674d67	35	fConversionCuts=conversionCuts;
	36	fIsHeavyIon = IsHeavyIon;
	37	}
344100c4	38	void SetEventCuts(AliConvEventCuts* conversionCuts,Int_t IsHeavyIon ){
	39	fEventCuts=conversionCuts;
	40	fIsHeavyIon = IsHeavyIon;
	41	}
63674d67	42	void SetIsMC(Bool_t isMC){fIsMC=isMC;}
	43
	44	private:
72395bd9	45
63674d67	46	void ProcessPhotons();
	47	Int_t CountTracks0914();
	48	Int_t CountTracks09();
72395bd9	49
63674d67	50	AliV0ReaderV1 *fV0Reader;
63674d67	51	TClonesArray *fConversionGammas; //Reconstructed Photons;
344100c4	52	AliConvEventCuts *fEventCuts; // Cuts used by the V0Reader
344100c4	53	AliConversionPhotonCuts *fConversionCuts; // Cuts used by the V0Reader
63674d67	54	TTree *fTreeEvent;
	55	TTree *fTreeResolution;
	56	Float_t fPrimVtxZ;
	57	Int_t fNContrVtx;
	58	Int_t fNESDtracksEta09;
	59	Int_t fNESDtracksEta0914;
	60	Int_t fNESDtracksEta14;
	61	TVectorF fGammaRecCoords;
	62	TVectorF fGammaMCCoords;
	63	Float_t fChi2ndf;
	64	Int_t fIsHeavyIon;
	65	Bool_t fIsMC;
	66	TList *fOutputList;
	67	TList *fEventList;
	68	TList *fResolutionList;
	69	AliESDEvent *fESDEvent;
	70	AliMCEvent *fMCEvent;
72395bd9	71
63674d67	72	AliAnalysisTaskResolution(const AliAnalysisTaskResolution&); // not implemented
63674d67	73	AliAnalysisTaskResolution& operator=(const AliAnalysisTaskResolution&); // not implemented
72395bd9	74
72395bd9	75
344100c4	76	ClassDef(AliAnalysisTaskResolution, 2);
72395bd9	77	};
	78
	79	#endif