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

#ifndef ALIANALYSISTASKETATOPIPLPIMIGAMMA_H
#define ALIANALYSISTASKETATOPIPLPIMIGAMMA_H

/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */


#include "AliAnalysisTaskSE.h"
#include "AliV0ReaderV1.h"
#include "AliKFConversionPhoton.h"
#include "AliPrimaryPionSelector.h"
#include "AliConversionMesonCuts.h"
#include "AliConvEventCuts.h"
#include "AliGammaConversionAODBGHandler.h"
#include "TProfile2D.h"

class AliESDInputHandler;
class AliMCEventHandler;
class AliESDEvent;
class AliESDtrack;
class AliESDtrackCuts;
class AliESDpidCuts;
class AliV0Reader;
class AliTriggerAnalysis;

class AliAnalysisTaskEtaToPiPlPiMiGamma: public AliAnalysisTaskSE
{
	public:

		AliAnalysisTaskEtaToPiPlPiMiGamma();
		AliAnalysisTaskEtaToPiPlPiMiGamma( const char* name );
		virtual ~AliAnalysisTaskEtaToPiPlPiMiGamma();

		virtual void UserExec(Option_t *);
		virtual void UserCreateOutputObjects();
		virtual Bool_t Notify();
		virtual void Terminate(const Option_t *);

			
		void SetMoveParticleAccordingToVertex(Bool_t flag){fMoveParticleAccordingToVertex = flag;}
			
		void SetIsHeavyIon(Int_t flag){
			if (flag == 1 || flag ==2 ){
				fIsHeavyIon = 1;    
			} else {
				fIsHeavyIon = 0;    
			}
		}
		
		void SetIsMC(Bool_t isMC){fIsMC=isMC;}
		void SetConversionCutList(Int_t nCuts, TList *CutArray){
			fnCuts= nCuts;
			fGammaCutArray = CutArray;
		}
		void SetEventCutList(Int_t nCuts, TList *CutArray){
			fnCuts= nCuts;
			fEventCutArray = CutArray;
		}

		void SetPionCutList(TList *CutArray){
			fPionCutArray = CutArray;
		}
		void SetMesonCutList(TList *CutArray){
			fMesonCutArray = CutArray;
		}
		void SetDoMesonQA(Bool_t flag){ fDoMesonQA = flag; }
	

	private:

		void InitBack();
		void ProcessPhotonCandidates();
		void ProcessTruePhotonCandidates(AliAODConversionPhoton*);
		void ProcessTrueMesonCandidates(AliAODConversionMother *Pi0Candidate, AliAODConversionPhoton *TrueGammaCandidate, AliAODConversionPhoton *TrueVirtualGammaCandidate);
		void MoveParticleAccordingToVertex(AliAODConversionPhoton* particle,const AliGammaConversionAODBGHandler::GammaConversionVertex *vertex);
    	void ProcessPionCandidates();
		void ProcessMCParticles();
		void CalculateMesonCandidates();
        void CalculateBackground();
		void UpdateEventByEventData();
        
		Bool_t IsPiPlPiMiGammaDecay(TParticle *fMCMother) const;
		Bool_t IsEtaPiPlPiMiGammaDaughter( Int_t label ) const;
		
		
		AliV0ReaderV1 					*fV0Reader;									//
		AliPrimaryPionSelector			*fPionSelector;								//
		AliGammaConversionAODBGHandler 	**fBGHandler;								//
		AliESDEvent 					*fESDEvent;									//
		AliMCEvent 						*fMCEvent;									//
		AliStack 						*fMCStack;									//
		TList 							**fCutFolder;								//
		TList 							**fESDList;									//
		TList 							**fBackList;								//
		TList 							**fMotherList;								//
		TList 							**fTrueList;								//
		TList 							**fMCList;									//
		TList 							*fOutputContainer;							//
		TClonesArray 					*fReaderGammas;								//
		vector<Int_t> 					fSelectorNegPionIndex;						//
		vector<Int_t> 					fSelectorPosPionIndex;						//
		TList 							*fGoodGammas;								//
		TList 							*fGoodVirtualParticles;						//
		TList 							*fEventCutArray;							//
		TList 							*fGammaCutArray;							//
		TList 							*fPionCutArray;								//
		TList 							*fMesonCutArray;							//
		AliConvEventCuts 				*fEventCuts;								//
		AliConversionPhotonCuts 		*fConversionCuts;							//
		
		// reconstructed particles
		TH1F 							**fHistoConvGammaPt;						//
		TH1F 							**fHistoConvGammaEta;						//
		TH1F 							**fHistoNegPionPt;							//
		TH1F 							**fHistoPosPionPt;							//
		TH1F 							**fHistoNegPionPhi;							//
		TH1F 							**fHistoPosPionPhi;							//
		TH1F 							**fHistoNegPionEta;							//
		TH1F 							**fHistoPosPionEta;							//
		TH2F 							**fHistoNegPionClsTPC;						//
		TH2F 							**fHistoPosPionClsTPC;						//
		TH2F 							**fHistoPionDCAxy;							//
		TH2F 							**fHistoPionDCAz;							//
		TH2F 							**fHistoPionTPCdEdxNSigma;					//
		TH2F 							**fHistoPionTPCdEdx;						//
		TH2F 							**fHistoPionPionInvMassPt;					//
		TH2F	 						**fHistoMotherInvMassPt;					//
		THnSparseF 						**fTHnSparseMotherInvMassPtZM;				//
		TH2F 							**fHistoMotherBackInvMassPt;				//
		THnSparseF				 		**fTHnSparseMotherBackInvMassPtZM;			//
		
		// pure MC properties
		TH1F 							**fHistoMCAllGammaPt;						//
		TH1F 							**fHistoMCConvGammaPt;						//
		TH1F 							**fHistoMCAllPosPionsPt;					//
		TH1F 							**fHistoMCAllNegPionsPt;					//
		TH1F 							**fHistoMCGammaFromEtaPt;					//
		TH1F 							**fHistoMCPosPionsFromEtaPt;				//
		TH1F 							**fHistoMCNegPionsFromEtaPt;				//
		TH1F 							**fHistoMCEtaPiPlPiMiGammaPt;				//
		TH1F 							**fHistoMCEtaGGPt;							//
		TH1F 							**fHistoMCEtaDalitzPt;						//
		TH1F 							**fHistoMCEtaPiPlPiMiGammaInAccPt;			//

		// reconstructed particles MC validated
		TH2F 							**fHistoTrueMotherPiPlPiMiGammaInvMassPt;	//
		TH2F 							**fHistoTrueMotherGammaGammaInvMassPt;		//
		TH2F 							**fHistoTrueMotherDalitzInvMassPt;			//
		TH1F 							**fHistoTrueConvGammaPt;					//
		TH1F 							**fHistoTrueConvGammaFromEtaPt;				//
		TH1F				 			**fHistoTruePosPionPt;						//
		TH1F 							**fHistoTruePosPionFromEtaPt;				//
		TH1F 							**fHistoTrueNegPionPt;						//
		TH1F 							**fHistoTrueNegPionFromEtaPt;				//
		TH2F 							**fHistoTruePionPionInvMassPt;				//
		TH2F 							**fHistoTruePionPionFromEtaInvMassPt;		//
		// Event properties
		TH1I 							**fHistoNEvents;							//
		TH1I 							**fHistoNGoodESDTracks;						//
		TProfile 						**fProfileEtaShift;							//
			
		TRandom3 						fRandom;
		Int_t 							fnCuts;
		Int_t 							fiCut;
		Int_t 							fNumberOfESDTracks;
		Bool_t 							fMoveParticleAccordingToVertex;
		Bool_t 							fIsHeavyIon;
		Bool_t 							fDoMesonAnalysis;
		Bool_t 							fDoMesonQA;
		Bool_t 							fIsFromMBHeader;
		Bool_t 							fIsMC;
		Bool_t 							fIsGammaEtaCand;
	private:
		AliAnalysisTaskEtaToPiPlPiMiGamma( const AliAnalysisTaskEtaToPiPlPiMiGamma& ); // Not implemented
		AliAnalysisTaskEtaToPiPlPiMiGamma& operator=( const AliAnalysisTaskEtaToPiPlPiMiGamma& ); // Not implemented

		ClassDef( AliAnalysisTaskEtaToPiPlPiMiGamma, 2 );
};

#endif // ALIANALYSISTASKETATOPIPLPIMIGAMMA_H
Commit	Line	Data
344100c4	1	#ifndef ALIANALYSISTASKETATOPIPLPIMIGAMMA_H
344100c4	2	#define ALIANALYSISTASKETATOPIPLPIMIGAMMA_H
209b710e	3
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
209b710e	7
	8	#include "AliAnalysisTaskSE.h"
	9	#include "AliV0ReaderV1.h"
	10	#include "AliKFConversionPhoton.h"
	11	#include "AliPrimaryPionSelector.h"
	12	#include "AliConversionMesonCuts.h"
344100c4	13	#include "AliConvEventCuts.h"
209b710e	14	#include "AliGammaConversionAODBGHandler.h"
	15	#include "TProfile2D.h"
	16
	17	class AliESDInputHandler;
	18	class AliMCEventHandler;
	19	class AliESDEvent;
	20	class AliESDtrack;
	21	class AliESDtrackCuts;
	22	class AliESDpidCuts;
	23	class AliV0Reader;
	24	class AliTriggerAnalysis;
	25
	26	class AliAnalysisTaskEtaToPiPlPiMiGamma: public AliAnalysisTaskSE
	27	{
	28	public:
	29
	30	AliAnalysisTaskEtaToPiPlPiMiGamma();
	31	AliAnalysisTaskEtaToPiPlPiMiGamma( const char* name );
	32	virtual ~AliAnalysisTaskEtaToPiPlPiMiGamma();
	33
	34	virtual void UserExec(Option_t *);
	35	virtual void UserCreateOutputObjects();
	36	virtual Bool_t Notify();
	37	virtual void Terminate(const Option_t *);
	38
	39
	40	void SetMoveParticleAccordingToVertex(Bool_t flag){fMoveParticleAccordingToVertex = flag;}
	41
	42	void SetIsHeavyIon(Int_t flag){
	43	if (flag == 1 \|\| flag ==2 ){
	44	fIsHeavyIon = 1;
	45	} else {
	46	fIsHeavyIon = 0;
	47	}
	48	}
	49
	50	void SetIsMC(Bool_t isMC){fIsMC=isMC;}
	51	void SetConversionCutList(Int_t nCuts, TList *CutArray){
	52	fnCuts= nCuts;
	53	fGammaCutArray = CutArray;
	54	}
344100c4	55	void SetEventCutList(Int_t nCuts, TList *CutArray){
	56	fnCuts= nCuts;
	57	fEventCutArray = CutArray;
	58	}
	59
209b710e	60	void SetPionCutList(TList *CutArray){
	61	fPionCutArray = CutArray;
	62	}
	63	void SetMesonCutList(TList *CutArray){
	64	fMesonCutArray = CutArray;
	65	}
	66	void SetDoMesonQA(Bool_t flag){ fDoMesonQA = flag; }
	67
	68
	69	private:
	70
	71	void InitBack();
	72	void ProcessPhotonCandidates();
	73	void ProcessTruePhotonCandidates(AliAODConversionPhoton*);
	74	void ProcessTrueMesonCandidates(AliAODConversionMother Pi0Candidate, AliAODConversionPhoton TrueGammaCandidate, AliAODConversionPhoton *TrueVirtualGammaCandidate);
	75	void MoveParticleAccordingToVertex(AliAODConversionPhoton* particle,const AliGammaConversionAODBGHandler::GammaConversionVertex *vertex);
	76	void ProcessPionCandidates();
	77	void ProcessMCParticles();
	78	void CalculateMesonCandidates();
	79	void CalculateBackground();
	80	void UpdateEventByEventData();
	81
	82	Bool_t IsPiPlPiMiGammaDecay(TParticle *fMCMother) const;
	83	Bool_t IsEtaPiPlPiMiGammaDaughter( Int_t label ) const;
	84
	85
	86
344100c4	87	AliV0ReaderV1 *fV0Reader; //
	88	AliPrimaryPionSelector *fPionSelector; //
	89	AliGammaConversionAODBGHandler **fBGHandler; //
	90	AliESDEvent *fESDEvent; //
	91	AliMCEvent *fMCEvent; //
	92	AliStack *fMCStack; //
	93	TList **fCutFolder; //
	94	TList **fESDList; //
	95	TList **fBackList; //
	96	TList **fMotherList; //
	97	TList **fTrueList; //
	98	TList **fMCList; //
	99	TList *fOutputContainer; //
	100	TClonesArray *fReaderGammas; //
	101	vector<Int_t> fSelectorNegPionIndex; //
	102	vector<Int_t> fSelectorPosPionIndex; //
	103	TList *fGoodGammas; //
	104	TList *fGoodVirtualParticles; //
	105	TList *fEventCutArray; //
	106	TList *fGammaCutArray; //
	107	TList *fPionCutArray; //
	108	TList *fMesonCutArray; //
	109	AliConvEventCuts *fEventCuts; //
	110	AliConversionPhotonCuts *fConversionCuts; //
209b710e	111
209b710e	112	// reconstructed particles
344100c4	113	TH1F **fHistoConvGammaPt; //
	114	TH1F **fHistoConvGammaEta; //
	115	TH1F **fHistoNegPionPt; //
	116	TH1F **fHistoPosPionPt; //
	117	TH1F **fHistoNegPionPhi; //
	118	TH1F **fHistoPosPionPhi; //
	119	TH1F **fHistoNegPionEta; //
	120	TH1F **fHistoPosPionEta; //
	121	TH2F **fHistoNegPionClsTPC; //
	122	TH2F **fHistoPosPionClsTPC; //
	123	TH2F **fHistoPionDCAxy; //
	124	TH2F **fHistoPionDCAz; //
	125	TH2F **fHistoPionTPCdEdxNSigma; //
	126	TH2F **fHistoPionTPCdEdx; //
	127	TH2F **fHistoPionPionInvMassPt; //
	128	TH2F **fHistoMotherInvMassPt; //
	129	THnSparseF **fTHnSparseMotherInvMassPtZM; //
	130	TH2F **fHistoMotherBackInvMassPt; //
	131	THnSparseF **fTHnSparseMotherBackInvMassPtZM; //
209b710e	132
209b710e	133	// pure MC properties
344100c4	134	TH1F **fHistoMCAllGammaPt; //
	135	TH1F **fHistoMCConvGammaPt; //
	136	TH1F **fHistoMCAllPosPionsPt; //
	137	TH1F **fHistoMCAllNegPionsPt; //
	138	TH1F **fHistoMCGammaFromEtaPt; //
	139	TH1F **fHistoMCPosPionsFromEtaPt; //
	140	TH1F **fHistoMCNegPionsFromEtaPt; //
	141	TH1F **fHistoMCEtaPiPlPiMiGammaPt; //
	142	TH1F **fHistoMCEtaGGPt; //
	143	TH1F **fHistoMCEtaDalitzPt; //
	144	TH1F **fHistoMCEtaPiPlPiMiGammaInAccPt; //
	145
209b710e	146	// reconstructed particles MC validated
344100c4	147	TH2F **fHistoTrueMotherPiPlPiMiGammaInvMassPt; //
	148	TH2F **fHistoTrueMotherGammaGammaInvMassPt; //
	149	TH2F **fHistoTrueMotherDalitzInvMassPt; //
	150	TH1F **fHistoTrueConvGammaPt; //
	151	TH1F **fHistoTrueConvGammaFromEtaPt; //
	152	TH1F **fHistoTruePosPionPt; //
	153	TH1F **fHistoTruePosPionFromEtaPt; //
	154	TH1F **fHistoTrueNegPionPt; //
	155	TH1F **fHistoTrueNegPionFromEtaPt; //
	156	TH2F **fHistoTruePionPionInvMassPt; //
	157	TH2F **fHistoTruePionPionFromEtaInvMassPt; //
209b710e	158	// Event properties
344100c4	159	TH1I **fHistoNEvents; //
	160	TH1I **fHistoNGoodESDTracks; //
	161	TProfile **fProfileEtaShift; //
209b710e	162
344100c4	163	TRandom3 fRandom;
	164	Int_t fnCuts;
	165	Int_t fiCut;
	166	Int_t fNumberOfESDTracks;
	167	Bool_t fMoveParticleAccordingToVertex;
	168	Bool_t fIsHeavyIon;
	169	Bool_t fDoMesonAnalysis;
	170	Bool_t fDoMesonQA;
	171	Bool_t fIsFromMBHeader;
	172	Bool_t fIsMC;
	173	Bool_t fIsGammaEtaCand;
209b710e	174	private:
	175	AliAnalysisTaskEtaToPiPlPiMiGamma( const AliAnalysisTaskEtaToPiPlPiMiGamma& ); // Not implemented
	176	AliAnalysisTaskEtaToPiPlPiMiGamma& operator=( const AliAnalysisTaskEtaToPiPlPiMiGamma& ); // Not implemented
	177
344100c4	178	ClassDef( AliAnalysisTaskEtaToPiPlPiMiGamma, 2 );
209b710e	179	};
209b710e	180
344100c4	181	#endif // ALIANALYSISTASKETATOPIPLPIMIGAMMA_H
209b710e	182