[u/mrichter/AliRoot.git] / PWGHF / hfe / AliAnalysisTaskEMCalHFEpA.h

#ifndef AliAnalysisTaskEMCalHFEpA_cxx
#define AliAnalysisTaskEMCalHFEpA_cxx

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

	////////////////////////////////////////////////////////////////////////
	//                                                                    //
	//      Task for Heavy-flavour electron analysis in pPb collisions    //
	//      (+ Electron-Hadron Jetlike Azimuthal Correlation)             //
	//																	  //
	//		version: March 18, 2014.								      //
	//                                                                    //
	//	    Authors 							                          //
	//		Elienos Pereira de Oliveira Filho (epereira@cern.ch)	      //
	//		Cristiane Jahnke		(cristiane.jahnke@cern.ch)		      //
	//                                                                    //
	////////////////////////////////////////////////////////////////////////

class TH1F;
class TH2F;
class AliESDEvent;
class AliESDtrackCuts;
class AliESDtrack;
class AliHFEcontainer;
class AliHFEcuts;
class AliHFEpid;
class AliHFEpidQAmanager;
class AliCFManager;
class AliPIDResponse;
class AliCentrality;
class AliAODEvent;
class AliVEvent;
class AliAODMCHeader;
class AliSelectNonHFE;
class AliEventPoolManager;
class AliEventPool;
class TObjArray;
	//Lucile
class AliCaloTrackAODReader;
class AliCaloTrackReader;


	//______________________________________________________________________
	//Library
#include "AliAnalysisTaskSE.h"
#include "AliHFEpid.h"
#include "AliLog.h"
	//______________________________________________________________________

	//______________________________________________________________________
class AliAnalysisTaskEMCalHFEpA : public AliAnalysisTaskSE 
{
		//______________________________________________________________________
public:
	AliAnalysisTaskEMCalHFEpA();
	AliAnalysisTaskEMCalHFEpA(const char *name);
	virtual ~AliAnalysisTaskEMCalHFEpA();
	
	virtual void   UserCreateOutputObjects();
	virtual void   UserExec(Option_t *option);
	virtual void   Terminate(Option_t *);
	
		//Setters
	void SetAssHadronPtRange(Double_t AssHadronPtMin, Double_t AssHadronPtMax) {fAssHadronPtMin = AssHadronPtMin; fAssHadronPtMax = AssHadronPtMax; };
	void SetHFECuts(AliHFEcuts * const cuts) {fCuts = cuts;};
	void SetRejectKinkMother(Bool_t rejectKinkMother = kFALSE) {fRejectKinkMother = rejectKinkMother;};
	void SetCorrelationAnalysis(Bool_t CorrelationFlag=kTRUE) {fCorrelationFlag = CorrelationFlag;};
	void SetMCanalysis() {fIsMC = kTRUE;};
	void SetCentrality(Double_t CentralityMin, Double_t CentralityMax) { fCentralityMin = CentralityMin; fCentralityMax = CentralityMax; fHasCentralitySelection = kTRUE; };
	void SetAODanalysis(Bool_t IsAOD) {fIsAOD = IsAOD;};
	void SetEventMixing(Bool_t EventMixingFlag) { fEventMixingFlag = EventMixingFlag;};
	void SetNonHFEmassCut(Double_t MassCut) { fMassCut = MassCut; fMassCutFlag = kTRUE;};
	void SetEtaCut(Double_t EtaCutMin,Double_t EtaCutMax ) { fEtaCutMin = EtaCutMin; fEtaCutMax = EtaCutMax; };
	
	void SetdPhidEtaCut(Double_t dPhiCut, Double_t dEtaCut ) { fdPhiCut = dPhiCut;fdEtaCut = dEtaCut ;};
	
	void SetEoverPCut(Double_t EoverPCutMin,Double_t EoverPCutMax ) { fEoverPCutMin = EoverPCutMin; fEoverPCutMax = EoverPCutMax; };
	
	void SetM02Cut(Double_t M02CutMin,Double_t M02CutMax ) { fM02CutMin = M02CutMin; fM02CutMax = M02CutMax; };
	void SetM20Cut(Double_t M20CutMin,Double_t M20CutMax ) { fM20CutMin = M20CutMin; fM20CutMax = M20CutMax; };
	
	
	void SetNonHFEangleCut(Double_t AngleCut) { fAngleCut = AngleCut; fAngleCutFlag = kTRUE;};
	void SetNonHFEchi2Cut(Double_t Chi2Cut) { fChi2Cut = Chi2Cut; fChi2CutFlag = kTRUE;};
	void SetNonHFEdcaCut(Double_t DCAcut) { fDCAcut = DCAcut; fDCAcutFlag = kTRUE;};
	void SetUseEMCal() { fUseEMCal=kTRUE;};
	void SetUseTrigger() { fUseTrigger=kTRUE;};
	void SetUseShowerShapeCut(Bool_t UseShowerShapeCut=kFALSE) { fUseShowerShapeCut=UseShowerShapeCut;};
	void SetBackground(Bool_t FillBackground=kFALSE) { fFillBackground=FillBackground;};
	void SetEMCalTriggerEG1() { fEMCEG1=kTRUE; };
	void SetEMCalTriggerEG2() { fEMCEG2=kTRUE; };
	void SetCentralityEstimator(Int_t Estimator) { fEstimator=Estimator; }; //0 = V0A, 1 = Other
	void SetAdditionalCuts(Double_t PtMinAsso, Int_t TpcNclsAsso) {fPtMinAsso = PtMinAsso; fTpcNclsAsso = TpcNclsAsso;};
	void SetSPDCutForHadrons() {fAssocWithSPD = kTRUE;};
	
		//Getters
	AliHFEpid *GetPID() const {return fPID;};
		//______________________________________________________________________
	
		//______________________________________________________________________
private:
	
		//Function to process track cuts
	Bool_t ProcessCutStep(Int_t cutStep, AliVParticle *track);
		//Function to process eh analysis
	void ElectronHadronCorrelation(AliVTrack *track, Int_t trackIndex, AliVParticle *vtrack);
		//Function to find non-HFE and fill histos
	void Background(AliVTrack *track, Int_t trackIndex, AliVParticle *vtrack, Bool_t IsTPConly);
		//Selected Hadrons, for mixed event analysis
	TObjArray* SelectedHadrons();
		//DiHadron Correlation Background
	void DiHadronCorrelation(AliVTrack *track, Int_t trackIndex);
		//Find Mothers (Finde HFE and NonHFE from MC information)
	Bool_t FindMother(Int_t mcIndex);
	
		//Flags for specifics analysis
	Bool_t 				fCorrelationFlag;
	Bool_t				fIsMC;
	Bool_t				fUseEMCal;
	Bool_t				fUseTrigger;
	Bool_t				fUseShowerShapeCut;
	Bool_t				fFillBackground;
	Bool_t				fAssocWithSPD;
	
	
	Bool_t				fEMCEG1;
	Bool_t				fEMCEG2;
	
		//Used in the function FindMother
	Bool_t				fIsHFE1;
	Bool_t				fIsHFE2;
	Bool_t				fIsNonHFE;
	Bool_t				fIsFromD;
	Bool_t				fIsFromB;
	Bool_t				fIsFromPi0;
	Bool_t				fIsFromEta;
	Bool_t				fIsFromGamma;
	
		//General variables
	AliESDEvent 			*fESD;
	AliAODEvent 		   	*fAOD;				/// new
	AliVEvent 		      	*fVevent;			/// new
	AliESDtrackCuts         *fPartnerCuts;
	TList       			*fOutputList;
	AliPIDResponse 			*fPidResponse;
	AliSelectNonHFE 		*fNonHFE;
	
		//For the case of AOD analysis
	Bool_t					fIsAOD;					//flag for AOD analysis
	
		//For Centrality Selection
	AliCentrality			*fCentrality;
	Double_t				fCentralityMin;
	Double_t				fCentralityMax;
	Bool_t					fHasCentralitySelection;
	TH1F					*fCentralityHist;
	TH1F					*fCentralityHistPass;
	Float_t					fZvtx;	
	Int_t					fEstimator;
	
		//EMCal
	
	AliVCluster				*fClus;
		//AliESDCaloCluster 		*fClusESD;
	
		//Histograms
	TH1F				*fNevent;
	TH1F				*fPtElec_Inc;
	
	TH1F				*fPtPrim;
	TH1F				*fPtSec;
	TH1F				*fPtPrim2;
	TH1F				*fPtSec2;
	
	
	TH1F				*fCharge_n;
	TH1F				*fCharge_p;
	
	TH2D				*fTime;
	TH2D				*fTime2;
	TH2D				*ftimingEle;
	TH2D				*ftimingEle2;	
	
	TH1F				*fPtElec_ULS;
	TH1F				*fPtElec_LS;
	
	TH1F				*fPtElec_ULS_NoPid;
	TH1F				*fPtElec_LS_NoPid;
	
	TH1F				*fPtElec_ULS_MC;
	TH1F				*fPtElec_ULS_MC_weight;
	
	TH1F				*fPtElec_ULS2;
	TH1F				*fPtElec_LS2;
	
	TH1F				*fPtElec_ULS_weight;
	TH1F				*fPtElec_LS_weight;
	TH1F				*fPtElec_ULS2_weight;
	TH1F				*fPtElec_LS2_weight;
	
		//PID Histograms
	
	TH2F				*fTOF01;
	TH2F				*fTOF02;
	TH2F				*fTOF03;

	TH1F				*fpid;		
	
	TH2F				**fEoverP_pt;
	TH2F				**fEoverP_tpc;
	
	TH1F				**fTPC_pt;
	TH2F				**fTPC_p;
	
	TH1F				**fTPCnsigma_pt;
	TH2F				**fTPCnsigma_p;
	TH2F				*fTPCnsigma_pt_2D;
	TH2F				*fShowerShapeCut;
	TH2F				*fShowerShapeM02_EoverP;
	TH2F				*fShowerShapeM20_EoverP;
	TH2F				*fShowerShape_ha;
	TH2F				*fShowerShape_ele;
	
	TH2F				*fTPCnsigma_eta;
	TH2F				*fTPCnsigma_phi;
	
	
	TH1F				**fECluster;
	TH1F				*fECluster_pure;
	TH2F				**fEtaPhi;
	TH2F				*fEtaPhi_num;
	TH2F				*fEtaPhi_den;
	
	TH2F				*fpt_reco_pt_MC_num;
	TH2F				*fpt_reco_pt_MC_den;
	
	TH1F				**fVtxZ;
	
	TH1F				*fVtxZ_new1;
	TH1F				*fVtxZ_new2;
	TH1F				*fVtxZ_new3;
	TH1F				*fVtxZ_new4;
	
	TH1F				**fEtad;
	TH1F				**fNTracks;
	
	TH2F				**fNTracks_pt;
	TH2F				**fNTracks_eta;
	TH2F				**fNTracks_phi;
	
	TH1F				**fNClusters;
	TH2F				**fTPCNcls_EoverP;
	TH2F				**fTPCNcls_pid;
	
	TH1F				**fEta;
	TH1F				**fPhi;
	TH1F				**fR;
	TH2F				**fR_EoverP;
	TH1F				**fNcells;
	TH2F				**fNcells_EoverP;
	TH1F				**fNcells_electrons;
	TH1F				**fNcells_hadrons;
	TH1F				**fECluster_ptbins;
	TH1F				**fEoverP_ptbins;
	TH1F				**fEoverP_wSSCut;
	TH2F				**fM02_EoverP;
	TH2F				**fM20_EoverP;
	TH2F				**fTPCnsigma_eta_electrons;
	TH2F				**fTPCnsigma_eta_hadrons;
	
	TH2F				*fEoverP_pt_pions;
	
	TH2F				*ftpc_p_EoverPcut;
	TH2F				*fnsigma_p_EoverPcut;
	
	TH2F				*fEoverP_pt_pions2;
	TH2F				*fNcells_pt;
	TH2F				*fEoverP_pt_hadrons;
	
		//Electron-Hadron Correlation Histograms
	TH2F				**fCEtaPhi_Inc;
	
	TH2F				**fCEtaPhi_ULS;
	TH2F				**fCEtaPhi_LS;
	TH2F				**fCEtaPhi_ULS_NoP;
	TH2F				**fCEtaPhi_LS_NoP;
	
	TH2F				**fCEtaPhi_ULS_Weight;
	TH2F				**fCEtaPhi_LS_Weight;
	TH2F				**fCEtaPhi_ULS_NoP_Weight;
	TH2F				**fCEtaPhi_LS_NoP_Weight;
	
	TH1F				*fInvMass;
	TH1F				*fInvMassBack;
	TH1F				*fDCA;
	TH1F				*fDCABack;
	TH1F				*fOpAngle;
	TH1F				*fOpAngleBack;
	
	TH1F				*fInvMass2;
	TH1F				*fInvMassBack2;
	TH1F				*fDCA2;
	TH1F				*fDCABack2;
	TH1F				*fOpAngle2;
	TH1F				*fOpAngleBack2;
	
	Double_t			fMassCut;
	Double_t			fEtaCutMin;
	Double_t			fEtaCutMax;
	
	Double_t			fdPhiCut;
	Double_t			fdEtaCut;
	
	Double_t			fEoverPCutMin;
	Double_t			fEoverPCutMax;
	Double_t			fM20CutMin;
	Double_t			fM20CutMax;
	Double_t			fM02CutMin;
	Double_t			fM02CutMax;
	
	Double_t			fAngleCut;
	Double_t			fChi2Cut;
	Double_t			fDCAcut;
	Bool_t				fMassCutFlag;
	Bool_t				fAngleCutFlag;
	Bool_t				fChi2CutFlag;
	Bool_t				fDCAcutFlag;
	
	//Correlation Function
	Double_t			fAssHadronPtMin;
	Double_t			fAssHadronPtMax;
	
		//Non-HFE reconstruction efficiency
	TH1F				*fPtBackgroundBeforeReco;
	TH1F				*fPtBackgroundBeforeReco2;
	TH1F				*fPtBackgroundBeforeReco_weight;
	TH1F				*fPtBackgroundBeforeReco2_weight;
	
	TH2F				*fpT_m_electron;
	TH2F				*fpT_gm_electron;
	
	TH1F				*fPtBackgroundAfterReco;
	
	Double_t			fPtMinAsso;
	Int_t			fTpcNclsAsso;
	
		//Tracking Efficiency
	TH1F				*fPtMCparticleAll;
	TH1F				*fPtMCparticleAll_nonPrimary;
	TH1F				*fPtMCparticleAlle_nonPrimary;
	TH1F				*fPtMCparticleAlle_Primary;

	TH1F				*fPtMCparticleReco;
	TH1F				*fPtMCparticleReco_nonPrimary;
	TH1F				*fPtMCparticleAllHfe1;
	TH1F				*fPtMCparticleRecoHfe1;
	TH1F				*fPtMCparticleAllHfe2;
	TH1F				*fPtMCparticleRecoHfe2;
	TH1F				*fPtMCelectronAfterAll;
	TH1F				*fPtMCelectronAfterAll_unfolding;
	TH1F				*fPtMCelectronAfterAll_nonPrimary;
	TH1F				*fPtMCelectronAfterAll_Primary;
	
	TH1F				*fPtMCpi0;
	TH1F				*fPtMCeta;
	TH1F				*fPtMCpi02;
	TH1F				*fPtMCeta2;
	
	TH1F				*fPtMC_EMCal_All;
	TH1F				*fPtMC_EMCal_Selected;
	TH1F				*fPtMC_TPC_All;
	TH1F				*fPtMC_TPC_Selected;
	TH1F				*fPt_track_match_den;
	TH1F				*fPt_track_match_num;
	
		
	TH1F				*fPtMCWithLabel;
	TH1F				*fPtMCWithoutLabel;
	TH1F				*fPtIsPhysicaPrimary;
	
		//For the HFE package
	AliHFEcuts 			*fCuts;                 		// Cut Collection for HFE
														//Lucile
														//AliCaloTrackAODReader 			*reader; 
	AliCFManager 		*fCFM;                  		// Correction Framework Manager
	AliHFEpid 			*fPID;                  		// PID
	AliHFEpidQAmanager 	*fPIDqa;						// PID QA manager
	
		//Others
	AliStack 			*fMCstack;						//
	Bool_t              fRejectKinkMother;				//
	TParticle 			*fMCtrack;
	TParticle 			*fMCtrackMother;
	TParticle 			*fMCtrackGMother;
	TParticle 			*fMCtrackGGMother;
	TParticle 			*fMCtrackGGGMother;
	TClonesArray 		*fMCarray;
	AliAODMCHeader 		*fMCheader;
	AliAODMCParticle 	*fMCparticle;
	AliAODMCParticle 	*fMCparticle2;
	AliAODMCParticle 	*fMCparticleMother;
	AliAODMCParticle 	*fMCparticleGMother;
	AliAODMCParticle 	*fMCparticleGGMother;
	AliAODMCParticle 	*fMCparticleGGGMother;
	AliMCEventHandler	*fEventHandler;
	AliMCEvent			*fMCevent;
	
		//______________________________________________________________________
		//Mixed event analysis
	AliEventPoolManager *fPoolMgr;
	AliEventPool		*fPool;
	TObjArray			*fTracksClone;
	TObjArray			*fTracks;
	
	TH2F				**fCEtaPhi_Inc_EM;
	
	TH2F				**fCEtaPhi_ULS_EM;
	TH2F				**fCEtaPhi_LS_EM;
	
	TH2F				**fCEtaPhi_ULS_Weight_EM;
	TH2F				**fCEtaPhi_LS_Weight_EM;
	
	TH1F				*fPoolNevents;
	
	Bool_t				fEventMixingFlag;
		//______________________________________________________________________
	
		//______________________________________________________________________
		//Di-hadron correlation
	TH2F				**fCEtaPhi_Inc_DiHadron;
	TH1F				*fPtTrigger_Inc;
		//______________________________________________________________________
	
	AliAnalysisTaskEMCalHFEpA(const AliAnalysisTaskEMCalHFEpA&); 			// not implemented
	AliAnalysisTaskEMCalHFEpA& operator=(const AliAnalysisTaskEMCalHFEpA&); 		// not implemented
	
	ClassDef(AliAnalysisTaskEMCalHFEpA, 1); 								// example of analysis
																			//______________________________________________________________________
};

	///_________________________________________________________________________________________________
	///Class copied from : $ALICE_ROOT/PWGCF/Correlations/DPhi/AliAnalysisTaskLongRangeCorrelations.h
	///Author: Christoph Mayer
class AliEHCParticle : public TObject {
public:
	AliEHCParticle(Double_t eta=0, Double_t phi=0, Double_t pt=0)
    : fEta(eta), fPhi(phi), fPt(pt) {}
	virtual ~AliEHCParticle() {}
	
	Double_t Eta() const { return fEta; }
	Double_t Phi() const { return fPhi; }
	Double_t Pt() const { return fPt; }
	
protected:
private:
	AliEHCParticle(const AliEHCParticle&);
	AliEHCParticle& operator=(const AliEHCParticle&);
	
	Double_t fEta;
	Double_t fPhi;
	Double_t fPt;
	
	ClassDef(AliEHCParticle, 1);
} ;
	///_________________________________________________________________________________________________

#endif
Commit	Line	Data
c852fdae	1	#ifndef AliAnalysisTaskEMCalHFEpA_cxx
	2	#define AliAnalysisTaskEMCalHFEpA_cxx
	3
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
9a1f999b	7	////////////////////////////////////////////////////////////////////////
	8	// //
	9	// Task for Heavy-flavour electron analysis in pPb collisions //
	10	// (+ Electron-Hadron Jetlike Azimuthal Correlation) //
	11	// //
8e2d649d	12	// version: March 18, 2014. //
9a1f999b	13	// //
	14	// Authors //
	15	// Elienos Pereira de Oliveira Filho (epereira@cern.ch) //
	16	// Cristiane Jahnke (cristiane.jahnke@cern.ch) //
	17	// //
	18	////////////////////////////////////////////////////////////////////////
c852fdae	19
	20	class TH1F;
	21	class TH2F;
	22	class AliESDEvent;
	23	class AliESDtrackCuts;
	24	class AliESDtrack;
	25	class AliHFEcontainer;
	26	class AliHFEcuts;
	27	class AliHFEpid;
	28	class AliHFEpidQAmanager;
	29	class AliCFManager;
	30	class AliPIDResponse;
	31	class AliCentrality;
	32	class AliAODEvent;
	33	class AliVEvent;
	34	class AliAODMCHeader;
	35	class AliSelectNonHFE;
	36	class AliEventPoolManager;
	37	class AliEventPool;
	38	class TObjArray;
8e2d649d	39	//Lucile
	40	class AliCaloTrackAODReader;
	41	class AliCaloTrackReader;
	42
c852fdae	43
9a1f999b	44	//______________________________________________________________________
9a1f999b	45	//Library
c852fdae	46	#include "AliAnalysisTaskSE.h"
	47	#include "AliHFEpid.h"
	48	#include "AliLog.h"
9a1f999b	49	//______________________________________________________________________
c852fdae	50
9a1f999b	51	//______________________________________________________________________
c852fdae	52	class AliAnalysisTaskEMCalHFEpA : public AliAnalysisTaskSE
c852fdae	53	{
9a1f999b	54	//______________________________________________________________________
9a1f999b	55	public:
c852fdae	56	AliAnalysisTaskEMCalHFEpA();
	57	AliAnalysisTaskEMCalHFEpA(const char *name);
	58	virtual ~AliAnalysisTaskEMCalHFEpA();
9a1f999b	59
c852fdae	60	virtual void UserCreateOutputObjects();
	61	virtual void UserExec(Option_t *option);
	62	virtual void Terminate(Option_t *);
9a1f999b	63
9a1f999b	64	//Setters
0131ce6c	65	void SetAssHadronPtRange(Double_t AssHadronPtMin, Double_t AssHadronPtMax) {fAssHadronPtMin = AssHadronPtMin; fAssHadronPtMax = AssHadronPtMax; };
c852fdae	66	void SetHFECuts(AliHFEcuts * const cuts) {fCuts = cuts;};
c852fdae	67	void SetRejectKinkMother(Bool_t rejectKinkMother = kFALSE) {fRejectKinkMother = rejectKinkMother;};
a89620c8	68	void SetCorrelationAnalysis(Bool_t CorrelationFlag=kTRUE) {fCorrelationFlag = CorrelationFlag;};
c852fdae	69	void SetMCanalysis() {fIsMC = kTRUE;};
	70	void SetCentrality(Double_t CentralityMin, Double_t CentralityMax) { fCentralityMin = CentralityMin; fCentralityMax = CentralityMax; fHasCentralitySelection = kTRUE; };
	71	void SetAODanalysis(Bool_t IsAOD) {fIsAOD = IsAOD;};
	72	void SetEventMixing(Bool_t EventMixingFlag) { fEventMixingFlag = EventMixingFlag;};
	73	void SetNonHFEmassCut(Double_t MassCut) { fMassCut = MassCut; fMassCutFlag = kTRUE;};
	74	void SetEtaCut(Double_t EtaCutMin,Double_t EtaCutMax ) { fEtaCutMin = EtaCutMin; fEtaCutMax = EtaCutMax; };
a6f21076	75
	76	void SetdPhidEtaCut(Double_t dPhiCut, Double_t dEtaCut ) { fdPhiCut = dPhiCut;fdEtaCut = dEtaCut ;};
	77
c852fdae	78	void SetEoverPCut(Double_t EoverPCutMin,Double_t EoverPCutMax ) { fEoverPCutMin = EoverPCutMin; fEoverPCutMax = EoverPCutMax; };
a6f21076	79
	80	void SetM02Cut(Double_t M02CutMin,Double_t M02CutMax ) { fM02CutMin = M02CutMin; fM02CutMax = M02CutMax; };
	81	void SetM20Cut(Double_t M20CutMin,Double_t M20CutMax ) { fM20CutMin = M20CutMin; fM20CutMax = M20CutMax; };
	82
	83
c852fdae	84	void SetNonHFEangleCut(Double_t AngleCut) { fAngleCut = AngleCut; fAngleCutFlag = kTRUE;};
	85	void SetNonHFEchi2Cut(Double_t Chi2Cut) { fChi2Cut = Chi2Cut; fChi2CutFlag = kTRUE;};
	86	void SetNonHFEdcaCut(Double_t DCAcut) { fDCAcut = DCAcut; fDCAcutFlag = kTRUE;};
	87	void SetUseEMCal() { fUseEMCal=kTRUE;};
8e2d649d	88	void SetUseTrigger() { fUseTrigger=kTRUE;};
a89620c8	89	void SetUseShowerShapeCut(Bool_t UseShowerShapeCut=kFALSE) { fUseShowerShapeCut=UseShowerShapeCut;};
a89620c8	90	void SetBackground(Bool_t FillBackground=kFALSE) { fFillBackground=FillBackground;};
c852fdae	91	void SetEMCalTriggerEG1() { fEMCEG1=kTRUE; };
	92	void SetEMCalTriggerEG2() { fEMCEG2=kTRUE; };
	93	void SetCentralityEstimator(Int_t Estimator) { fEstimator=Estimator; }; //0 = V0A, 1 = Other
9a1f999b	94	void SetAdditionalCuts(Double_t PtMinAsso, Int_t TpcNclsAsso) {fPtMinAsso = PtMinAsso; fTpcNclsAsso = TpcNclsAsso;};
2c8e1f6c	95	void SetSPDCutForHadrons() {fAssocWithSPD = kTRUE;};
c852fdae	96
9a1f999b	97	//Getters
c852fdae	98	AliHFEpid *GetPID() const {return fPID;};
9a1f999b	99	//______________________________________________________________________
c852fdae	100
9a1f999b	101	//______________________________________________________________________
	102	private:
	103
	104	//Function to process track cuts
c852fdae	105	Bool_t ProcessCutStep(Int_t cutStep, AliVParticle *track);
9a1f999b	106	//Function to process eh analysis
c852fdae	107	void ElectronHadronCorrelation(AliVTrack track, Int_t trackIndex, AliVParticle vtrack);
9a1f999b	108	//Function to find non-HFE and fill histos
9a4682da	109	void Background(AliVTrack track, Int_t trackIndex, AliVParticle vtrack, Bool_t IsTPConly);
9a1f999b	110	//Selected Hadrons, for mixed event analysis
c852fdae	111	TObjArray* SelectedHadrons();
9a1f999b	112	//DiHadron Correlation Background
c852fdae	113	void DiHadronCorrelation(AliVTrack *track, Int_t trackIndex);
9a1f999b	114	//Find Mothers (Finde HFE and NonHFE from MC information)
c852fdae	115	Bool_t FindMother(Int_t mcIndex);
9a1f999b	116
9a1f999b	117	//Flags for specifics analysis
c852fdae	118	Bool_t fCorrelationFlag;
	119	Bool_t fIsMC;
	120	Bool_t fUseEMCal;
8e2d649d	121	Bool_t fUseTrigger;
a89620c8	122	Bool_t fUseShowerShapeCut;
a89620c8	123	Bool_t fFillBackground;
2c8e1f6c	124	Bool_t fAssocWithSPD;
9a4682da	125
9a1f999b	126
c852fdae	127	Bool_t fEMCEG1;
c852fdae	128	Bool_t fEMCEG2;
9a1f999b	129
9a1f999b	130	//Used in the function FindMother
c852fdae	131	Bool_t fIsHFE1;
	132	Bool_t fIsHFE2;
	133	Bool_t fIsNonHFE;
	134	Bool_t fIsFromD;
	135	Bool_t fIsFromB;
	136	Bool_t fIsFromPi0;
	137	Bool_t fIsFromEta;
	138	Bool_t fIsFromGamma;
	139
9a1f999b	140	//General variables
c852fdae	141	AliESDEvent *fESD;
	142	AliAODEvent *fAOD; /// new
	143	AliVEvent *fVevent; /// new
	144	AliESDtrackCuts *fPartnerCuts;
	145	TList *fOutputList;
	146	AliPIDResponse *fPidResponse;
	147	AliSelectNonHFE *fNonHFE;
	148
9a1f999b	149	//For the case of AOD analysis
c852fdae	150	Bool_t fIsAOD; //flag for AOD analysis
c852fdae	151
9a1f999b	152	//For Centrality Selection
c852fdae	153	AliCentrality *fCentrality;
	154	Double_t fCentralityMin;
	155	Double_t fCentralityMax;
	156	Bool_t fHasCentralitySelection;
	157	TH1F *fCentralityHist;
	158	TH1F *fCentralityHistPass;
	159	Float_t fZvtx;
	160	Int_t fEstimator;
	161
9a1f999b	162	//EMCal
a89620c8	163
c852fdae	164	AliVCluster *fClus;
9a1f999b	165	//AliESDCaloCluster *fClusESD;
c852fdae	166
9a1f999b	167	//Histograms
c852fdae	168	TH1F *fNevent;
c852fdae	169	TH1F *fPtElec_Inc;
9a1f999b	170
8e2d649d	171	TH1F *fPtPrim;
	172	TH1F *fPtSec;
	173	TH1F *fPtPrim2;
	174	TH1F *fPtSec2;
	175
a6f21076	176
	177	TH1F *fCharge_n;
	178	TH1F *fCharge_p;
	179
a89620c8	180	TH2D *fTime;
	181	TH2D *fTime2;
	182	TH2D *ftimingEle;
	183	TH2D *ftimingEle2;
	184
c852fdae	185	TH1F *fPtElec_ULS;
c852fdae	186	TH1F *fPtElec_LS;
8e2d649d	187
	188	TH1F *fPtElec_ULS_NoPid;
	189	TH1F *fPtElec_LS_NoPid;
	190
	191	TH1F *fPtElec_ULS_MC;
	192	TH1F *fPtElec_ULS_MC_weight;
	193
9a4682da	194	TH1F *fPtElec_ULS2;
9a4682da	195	TH1F *fPtElec_LS2;
c852fdae	196
2c8e1f6c	197	TH1F *fPtElec_ULS_weight;
	198	TH1F *fPtElec_LS_weight;
	199	TH1F *fPtElec_ULS2_weight;
	200	TH1F *fPtElec_LS2_weight;
	201
9a1f999b	202	//PID Histograms
2c8e1f6c	203
	204	TH2F *fTOF01;
	205	TH2F *fTOF02;
	206	TH2F *fTOF03;
	207
c852fdae	208	TH1F *fpid;
	209
	210	TH2F **fEoverP_pt;
	211	TH2F **fEoverP_tpc;
	212
	213	TH1F **fTPC_pt;
	214	TH2F **fTPC_p;
	215
	216	TH1F **fTPCnsigma_pt;
	217	TH2F **fTPCnsigma_p;
	218	TH2F *fTPCnsigma_pt_2D;
a89620c8	219	TH2F *fShowerShapeCut;
9a4682da	220	TH2F *fShowerShapeM02_EoverP;
	221	TH2F *fShowerShapeM20_EoverP;
	222	TH2F *fShowerShape_ha;
	223	TH2F *fShowerShape_ele;
c852fdae	224
	225	TH2F *fTPCnsigma_eta;
	226	TH2F *fTPCnsigma_phi;
	227
9a1f999b	228
c852fdae	229	TH1F **fECluster;
8e2d649d	230	TH1F *fECluster_pure;
c852fdae	231	TH2F **fEtaPhi;
0131ce6c	232	TH2F *fEtaPhi_num;
0131ce6c	233	TH2F *fEtaPhi_den;
8e2d649d	234
	235	TH2F *fpt_reco_pt_MC_num;
	236	TH2F *fpt_reco_pt_MC_den;
	237
c852fdae	238	TH1F **fVtxZ;
8e2d649d	239
	240	TH1F *fVtxZ_new1;
	241	TH1F *fVtxZ_new2;
	242	TH1F *fVtxZ_new3;
	243	TH1F *fVtxZ_new4;
	244
0131ce6c	245	TH1F **fEtad;
c852fdae	246	TH1F **fNTracks;
8e2d649d	247
	248	TH2F **fNTracks_pt;
	249	TH2F **fNTracks_eta;
	250	TH2F **fNTracks_phi;
	251
c852fdae	252	TH1F **fNClusters;
c852fdae	253	TH2F **fTPCNcls_EoverP;
aaec953d	254	TH2F **fTPCNcls_pid;
c852fdae	255
	256	TH1F **fEta;
	257	TH1F **fPhi;
	258	TH1F **fR;
	259	TH2F **fR_EoverP;
	260	TH1F **fNcells;
	261	TH2F **fNcells_EoverP;
	262	TH1F **fNcells_electrons;
	263	TH1F **fNcells_hadrons;
	264	TH1F **fECluster_ptbins;
	265	TH1F **fEoverP_ptbins;
	266	TH1F **fEoverP_wSSCut;
	267	TH2F **fM02_EoverP;
	268	TH2F **fM20_EoverP;
	269	TH2F **fTPCnsigma_eta_electrons;
	270	TH2F **fTPCnsigma_eta_hadrons;
	271
	272	TH2F *fEoverP_pt_pions;
	273
	274	TH2F *ftpc_p_EoverPcut;
	275	TH2F *fnsigma_p_EoverPcut;
	276
	277	TH2F *fEoverP_pt_pions2;
	278	TH2F *fNcells_pt;
	279	TH2F *fEoverP_pt_hadrons;
	280
9a1f999b	281	//Electron-Hadron Correlation Histograms
c852fdae	282	TH2F **fCEtaPhi_Inc;
	283
	284	TH2F **fCEtaPhi_ULS;
	285	TH2F **fCEtaPhi_LS;
	286	TH2F **fCEtaPhi_ULS_NoP;
	287	TH2F **fCEtaPhi_LS_NoP;
	288
	289	TH2F **fCEtaPhi_ULS_Weight;
	290	TH2F **fCEtaPhi_LS_Weight;
	291	TH2F **fCEtaPhi_ULS_NoP_Weight;
	292	TH2F **fCEtaPhi_LS_NoP_Weight;
	293
	294	TH1F *fInvMass;
	295	TH1F *fInvMassBack;
	296	TH1F *fDCA;
	297	TH1F *fDCABack;
	298	TH1F *fOpAngle;
	299	TH1F *fOpAngleBack;
	300
9a4682da	301	TH1F *fInvMass2;
	302	TH1F *fInvMassBack2;
	303	TH1F *fDCA2;
	304	TH1F *fDCABack2;
	305	TH1F *fOpAngle2;
	306	TH1F *fOpAngleBack2;
	307
c852fdae	308	Double_t fMassCut;
	309	Double_t fEtaCutMin;
	310	Double_t fEtaCutMax;
a6f21076	311
	312	Double_t fdPhiCut;
	313	Double_t fdEtaCut;
	314
c852fdae	315	Double_t fEoverPCutMin;
c852fdae	316	Double_t fEoverPCutMax;
a6f21076	317	Double_t fM20CutMin;
	318	Double_t fM20CutMax;
	319	Double_t fM02CutMin;
	320	Double_t fM02CutMax;
	321
c852fdae	322	Double_t fAngleCut;
	323	Double_t fChi2Cut;
	324	Double_t fDCAcut;
	325	Bool_t fMassCutFlag;
	326	Bool_t fAngleCutFlag;
	327	Bool_t fChi2CutFlag;
	328	Bool_t fDCAcutFlag;
	329
0131ce6c	330	//Correlation Function
	331	Double_t fAssHadronPtMin;
	332	Double_t fAssHadronPtMax;
	333
9a1f999b	334	//Non-HFE reconstruction efficiency
c852fdae	335	TH1F *fPtBackgroundBeforeReco;
9a4682da	336	TH1F *fPtBackgroundBeforeReco2;
2c8e1f6c	337	TH1F *fPtBackgroundBeforeReco_weight;
	338	TH1F *fPtBackgroundBeforeReco2_weight;
	339
2235208e	340	TH2F *fpT_m_electron;
	341	TH2F *fpT_gm_electron;
	342
9a4682da	343	TH1F *fPtBackgroundAfterReco;
9a4682da	344
9a1f999b	345	Double_t fPtMinAsso;
9a1f999b	346	Int_t fTpcNclsAsso;
9a4682da	347
9a1f999b	348	//Tracking Efficiency
c852fdae	349	TH1F *fPtMCparticleAll;
9a1f999b	350	TH1F *fPtMCparticleAll_nonPrimary;
9a1f999b	351	TH1F *fPtMCparticleAlle_nonPrimary;
2c8e1f6c	352	TH1F *fPtMCparticleAlle_Primary;
9a1f999b	353
c852fdae	354	TH1F *fPtMCparticleReco;
9a1f999b	355	TH1F *fPtMCparticleReco_nonPrimary;
c852fdae	356	TH1F *fPtMCparticleAllHfe1;
	357	TH1F *fPtMCparticleRecoHfe1;
	358	TH1F *fPtMCparticleAllHfe2;
	359	TH1F *fPtMCparticleRecoHfe2;
	360	TH1F *fPtMCelectronAfterAll;
8e2d649d	361	TH1F *fPtMCelectronAfterAll_unfolding;
9a1f999b	362	TH1F *fPtMCelectronAfterAll_nonPrimary;
2c8e1f6c	363	TH1F *fPtMCelectronAfterAll_Primary;
c852fdae	364
c852fdae	365	TH1F *fPtMCpi0;
9a1f999b	366	TH1F *fPtMCeta;
8e2d649d	367	TH1F *fPtMCpi02;
8e2d649d	368	TH1F *fPtMCeta2;
c852fdae	369
	370	TH1F *fPtMC_EMCal_All;
	371	TH1F *fPtMC_EMCal_Selected;
	372	TH1F *fPtMC_TPC_All;
	373	TH1F *fPtMC_TPC_Selected;
8e2d649d	374	TH1F *fPt_track_match_den;
8e2d649d	375	TH1F *fPt_track_match_num;
c852fdae	376
8e2d649d	377
c852fdae	378	TH1F *fPtMCWithLabel;
	379	TH1F *fPtMCWithoutLabel;
	380	TH1F *fPtIsPhysicaPrimary;
	381
9a1f999b	382	//For the HFE package
c852fdae	383	AliHFEcuts *fCuts; // Cut Collection for HFE
8e2d649d	384	//Lucile
8e2d649d	385	//AliCaloTrackAODReader *reader;
c852fdae	386	AliCFManager *fCFM; // Correction Framework Manager
	387	AliHFEpid *fPID; // PID
	388	AliHFEpidQAmanager *fPIDqa; // PID QA manager
	389
9a1f999b	390	//Others
c852fdae	391	AliStack *fMCstack; //
	392	Bool_t fRejectKinkMother; //
	393	TParticle *fMCtrack;
	394	TParticle *fMCtrackMother;
	395	TParticle *fMCtrackGMother;
	396	TParticle *fMCtrackGGMother;
	397	TParticle *fMCtrackGGGMother;
	398	TClonesArray *fMCarray;
	399	AliAODMCHeader *fMCheader;
	400	AliAODMCParticle *fMCparticle;
8e2d649d	401	AliAODMCParticle *fMCparticle2;
c852fdae	402	AliAODMCParticle *fMCparticleMother;
	403	AliAODMCParticle *fMCparticleGMother;
	404	AliAODMCParticle *fMCparticleGGMother;
	405	AliAODMCParticle *fMCparticleGGGMother;
	406	AliMCEventHandler *fEventHandler;
	407	AliMCEvent *fMCevent;
9a1f999b	408
	409	//______________________________________________________________________
	410	//Mixed event analysis
c852fdae	411	AliEventPoolManager *fPoolMgr;
	412	AliEventPool *fPool;
	413	TObjArray *fTracksClone;
	414	TObjArray *fTracks;
	415
	416	TH2F **fCEtaPhi_Inc_EM;
	417
	418	TH2F **fCEtaPhi_ULS_EM;
	419	TH2F **fCEtaPhi_LS_EM;
	420
	421	TH2F **fCEtaPhi_ULS_Weight_EM;
	422	TH2F **fCEtaPhi_LS_Weight_EM;
	423
	424	TH1F *fPoolNevents;
	425
	426	Bool_t fEventMixingFlag;
9a1f999b	427	//______________________________________________________________________
	428
	429	//______________________________________________________________________
	430	//Di-hadron correlation
c852fdae	431	TH2F **fCEtaPhi_Inc_DiHadron;
c852fdae	432	TH1F *fPtTrigger_Inc;
9a1f999b	433	//______________________________________________________________________
9a1f999b	434
c852fdae	435	AliAnalysisTaskEMCalHFEpA(const AliAnalysisTaskEMCalHFEpA&); // not implemented
c852fdae	436	AliAnalysisTaskEMCalHFEpA& operator=(const AliAnalysisTaskEMCalHFEpA&); // not implemented
9a1f999b	437
c852fdae	438	ClassDef(AliAnalysisTaskEMCalHFEpA, 1); // example of analysis
9a1f999b	439	//______________________________________________________________________
c852fdae	440	};
c852fdae	441
9a1f999b	442	///_________________________________________________________________________________________________
	443	///Class copied from : $ALICE_ROOT/PWGCF/Correlations/DPhi/AliAnalysisTaskLongRangeCorrelations.h
	444	///Author: Christoph Mayer
c852fdae	445	class AliEHCParticle : public TObject {
c852fdae	446	public:
9a1f999b	447	AliEHCParticle(Double_t eta=0, Double_t phi=0, Double_t pt=0)
c852fdae	448	: fEta(eta), fPhi(phi), fPt(pt) {}
9a1f999b	449	virtual ~AliEHCParticle() {}
	450
	451	Double_t Eta() const { return fEta; }
	452	Double_t Phi() const { return fPhi; }
	453	Double_t Pt() const { return fPt; }
	454
c852fdae	455	protected:
c852fdae	456	private:
9a1f999b	457	AliEHCParticle(const AliEHCParticle&);
	458	AliEHCParticle& operator=(const AliEHCParticle&);
	459
	460	Double_t fEta;
	461	Double_t fPhi;
	462	Double_t fPt;
	463
	464	ClassDef(AliEHCParticle, 1);
c852fdae	465	} ;
9a1f999b	466	///_________________________________________________________________________________________________
c852fdae	467
c852fdae	468	#endif