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

/**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *               
 *                                                                        *               
 * Author: The ALICE Off-line Project.                                    *               
 * Contributors are mentioned in the code where appropriate.              *               
 *                                                                        *               
 * Permission to use, copy, modify and distribute this software and its   *               
 * documentation strictly for non-commercial purposes is hereby granted   *               
 * without fee, provided that the above copyright notice appears in all   *               
 * copies and that both the copyright notice and this permission notice   *               
 * appear in the supporting documentation. The authors make no claims     *               
 * about the suitability of this software for any purpose. It is          *               
 * provided "as is" without express or implied warranty.                  *               
 **************************************************************************/

#ifndef ALIANALYSISTASKFlowTPCEMCalEP_H
#define ALIANALYSISTASKFlowTPCEMCalEP_H

class THnSparse;
class TH2F;
class TLorentzVector;

class AliEMCALTrack;
class AliMagF;
class AliESDEvent;
class AliAODEvent;
class AliEMCALGeometry;
class AliEMCALRecoUtils;
class AliAnalysisFilter;
class AliESDtrackCuts;
class AliESDtrack;
class AliHFEcontainer;
class AliHFEcuts;
class AliHFEpid;
class AliHFEpidQAmanager;
class AliCFManager;

#include "AliAnalysisTaskSE.h"

class AliAnalysisTaskFlowTPCEMCalEP : public AliAnalysisTaskSE {
 public:
  AliAnalysisTaskFlowTPCEMCalEP();
  AliAnalysisTaskFlowTPCEMCalEP(const char *name);
  virtual ~AliAnalysisTaskFlowTPCEMCalEP();
  
  virtual void   UserCreateOutputObjects();
  virtual void   UserExec(Option_t *option);
  virtual void   Terminate(Option_t *);
  
  void SetHFECuts(AliHFEcuts * const cuts) { fCuts = cuts; };
  void SetOpeningAngleCut (Double_t openingAngle) {fOpeningAngleCut = openingAngle;};
  void SetInvariantMassCut (Double_t invmass) {fInvmassCut = invmass;};
  AliHFEpid *GetPID() const { return fPID; }
  void SetRejectKinkMother(Bool_t rejectKinkMother = kFALSE) { fRejectKinkMother = rejectKinkMother; };
  void SelectPhotonicElectron(Int_t iTracks,AliESDtrack *track,Bool_t &fFlagPhotonicElec, Bool_t &fFlagPhotonicElecBCG);
  
  Double_t GetCos2DeltaPhi(Double_t phiA,Double_t phiB)		const;
  Double_t GetDeltaPhi(Double_t phiA,Double_t phiB)	const;
  Double_t GetPi0weight(Double_t mcPi0pT) const;
  Double_t GetEtaweight(Double_t mcEtapT) const;
  Double_t GetDweight(Int_t whichD, Double_t mcDpT) const;
  
 private:
  
  Bool_t ProcessCutStep(Int_t cutStep, AliVParticle *track);
  
  AliESDEvent 		*fESD;			//!ESD object
  AliMCEvent            *fMC;                   //!MC object
    
  TList       		*fOutputList;		//! output list
  
  AliESDtrackCuts	*fTrackCuts;		//! ESD track cuts
  AliHFEcuts 		*fCuts;                 //! Cut Collection
  Bool_t 		fIdentifiedAsOutInz;    //Out Of Range in z
  Bool_t 		fPassTheEventCut;       //Pass The Event Cut
  Bool_t 		fRejectKinkMother;      //Reject Kink Mother
  Bool_t 		fIsMC;
  Double_t 		fVz;                    //z position of the primary vertex
  AliCFManager 		*fCFM;                  //! Correction Framework Manager
  AliHFEpid 		*fPID;                  //! PID
  AliHFEpidQAmanager 	*fPIDqa;		//! PID QA manager
  Double_t 		fOpeningAngleCut;	//openingAngle cut value
  Double_t		fInvmassCut;		//invariant mass cut value
  
  TH1F			*fNoEvents;		//! no of events
  TH1F			*fTrkpt;		//! track pt
  TH2F			*fTrkEovPBef;		//! track E/p before HFE pid
  TH2F			*fTrkEovPAft;		//! track E/p after HFE pid
  TH2F			*fdEdxBef;		//! track dEdx vs p before HFE pid
  TH2F			*fdEdxAft;		//! track dEdx vs p after HFE pid
  TH1F			*fInvmassLS;		//! Inv mass of LS (e,e)
  TH1F			*fInvmassULS;		//! Inv mass of ULS (e,e)
  TH1F			*fOpeningAngleLS;	//! opening angle for LS pairs
  TH1F			*fOpeningAngleULS;	//! opening angle for ULS pairs
  TH1F			*fPhotoElecPt;		//! photonic elec pt 
  TH1F			*fSemiInclElecPt;	//! Semi inclusive ele pt
  THnSparse		*fMCphotoElecPt;	//! pt distribution (MC)
  
  TH1F			*fTrackPtBefTrkCuts;	//! Track pt before track cuts	
  TH1F			*fTrackPtAftTrkCuts;	//! Track pt after track cuts
  TH2F			*fTPCnsigma;		//! TPC n sigma vs p	
  
  TH1F			*fCent;			//! centrality
  TH2F			*fevPlaneV0A;		//! V0A event plane distribution
  TH2F			*fevPlaneV0C;		//! V0C event plane distribution
  TH2F			*fevPlaneTPC;		//! TPC event plane distribution
  TH2F			*fTPCsubEPres;		//! TPC event plane resolution
  THnSparse		*fEPres;		//! event plane resolution
  THnSparse		*fCorr;			//! correlations
  THnSparse		*feTPCV2;		//! inclusive eletron v2 (only TPC PID)
  THnSparse		*feV2;			//! inclusive eletron v2 (TPC + EMCAL PID)
  THnSparse		*fphoteV2;		//! photonic electron v2 (TPC + EMCAL PID)
  THnSparse		*fChargPartV2;		//! charged particle v2
    
  TH2F			*fGammaWeight;		//! gamma weight
  TH2F			*fPi0Weight;		//! pi0 weight
  TH2F			*fEtaWeight;		//! eta weight
  TH2F			*fD0Weight;		//! D0 weight
  TH2F			*fDplusWeight;		//! D+ weight
  TH2F			*fDminusWeight;		//! D- weight
  
  TH1F			*fDe[6];
  TH1F			*fD0e[6];
  TH1F			*fDpluse[6];
  TH1F			*fDminuse[6];
  
  TH2F			*fD0_e;
  
  TH1F			*fTot_pi0e;		//! inclusive electron
  TH1F			*fPhot_pi0e;		//! ULS pair 
  TH1F			*fPhotBCG_pi0e;		//! LS pair
  TH1F			*fTot_etae;		//! inclusive electron
  TH1F			*fPhot_etae;		//! ULS pair 
  TH1F			*fPhotBCG_etae;		//! LS pair
    
  AliAnalysisTaskFlowTPCEMCalEP(const AliAnalysisTaskFlowTPCEMCalEP&); // not implemented
  AliAnalysisTaskFlowTPCEMCalEP& operator=(const AliAnalysisTaskFlowTPCEMCalEP&); // not implemented
  
  ClassDef(AliAnalysisTaskFlowTPCEMCalEP, 1); //!example of analysis
};

#endif
Commit	Line	Data
1d08b6e8	1	/**************************************************************************
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
	15
914042c2	16	#ifndef ALIANALYSISTASKFlowTPCEMCalEP_H
914042c2	17	#define ALIANALYSISTASKFlowTPCEMCalEP_H
1d08b6e8	18
	19	class THnSparse;
	20	class TH2F;
	21	class TLorentzVector;
	22
	23	class AliEMCALTrack;
	24	class AliMagF;
	25	class AliESDEvent;
	26	class AliAODEvent;
	27	class AliEMCALGeometry;
	28	class AliEMCALRecoUtils;
	29	class AliAnalysisFilter;
	30	class AliESDtrackCuts;
	31	class AliESDtrack;
	32	class AliHFEcontainer;
	33	class AliHFEcuts;
	34	class AliHFEpid;
	35	class AliHFEpidQAmanager;
	36	class AliCFManager;
	37
	38	#include "AliAnalysisTaskSE.h"
	39
914042c2	40	class AliAnalysisTaskFlowTPCEMCalEP : public AliAnalysisTaskSE {
1d08b6e8	41	public:
914042c2	42	AliAnalysisTaskFlowTPCEMCalEP();
	43	AliAnalysisTaskFlowTPCEMCalEP(const char *name);
	44	virtual ~AliAnalysisTaskFlowTPCEMCalEP();
1d08b6e8	45
	46	virtual void UserCreateOutputObjects();
	47	virtual void UserExec(Option_t *option);
	48	virtual void Terminate(Option_t *);
	49
	50	void SetHFECuts(AliHFEcuts * const cuts) { fCuts = cuts; };
	51	void SetOpeningAngleCut (Double_t openingAngle) {fOpeningAngleCut = openingAngle;};
	52	void SetInvariantMassCut (Double_t invmass) {fInvmassCut = invmass;};
	53	AliHFEpid *GetPID() const { return fPID; }
	54	void SetRejectKinkMother(Bool_t rejectKinkMother = kFALSE) { fRejectKinkMother = rejectKinkMother; };
06d74a7b	55	void SelectPhotonicElectron(Int_t iTracks,AliESDtrack *track,Bool_t &fFlagPhotonicElec, Bool_t &fFlagPhotonicElecBCG);
06d74a7b	56
1d08b6e8	57	Double_t GetCos2DeltaPhi(Double_t phiA,Double_t phiB) const;
1d08b6e8	58	Double_t GetDeltaPhi(Double_t phiA,Double_t phiB) const;
1c424962	59	Double_t GetPi0weight(Double_t mcPi0pT) const;
1c424962	60	Double_t GetEtaweight(Double_t mcEtapT) const;
937b2ff2	61	Double_t GetDweight(Int_t whichD, Double_t mcDpT) const;
1c424962	62
1d08b6e8	63	private:
	64
	65	Bool_t ProcessCutStep(Int_t cutStep, AliVParticle *track);
	66
	67	AliESDEvent *fESD; //!ESD object
3a94d6fe	68	AliMCEvent *fMC; //!MC object
1d08b6e8	69
	70	TList *fOutputList; //! output list
	71
	72	AliESDtrackCuts *fTrackCuts; //! ESD track cuts
	73	AliHFEcuts *fCuts; //! Cut Collection
	74	Bool_t fIdentifiedAsOutInz; //Out Of Range in z
	75	Bool_t fPassTheEventCut; //Pass The Event Cut
	76	Bool_t fRejectKinkMother; //Reject Kink Mother
3a94d6fe	77	Bool_t fIsMC;
1d08b6e8	78	Double_t fVz; //z position of the primary vertex
	79	AliCFManager *fCFM; //! Correction Framework Manager
	80	AliHFEpid *fPID; //! PID
	81	AliHFEpidQAmanager *fPIDqa; //! PID QA manager
	82	Double_t fOpeningAngleCut; //openingAngle cut value
	83	Double_t fInvmassCut; //invariant mass cut value
	84
	85	TH1F *fNoEvents; //! no of events
	86	TH1F *fTrkpt; //! track pt
	87	TH2F *fTrkEovPBef; //! track E/p before HFE pid
	88	TH2F *fTrkEovPAft; //! track E/p after HFE pid
	89	TH2F *fdEdxBef; //! track dEdx vs p before HFE pid
	90	TH2F *fdEdxAft; //! track dEdx vs p after HFE pid
	91	TH1F *fInvmassLS; //! Inv mass of LS (e,e)
	92	TH1F *fInvmassULS; //! Inv mass of ULS (e,e)
	93	TH1F *fOpeningAngleLS; //! opening angle for LS pairs
	94	TH1F *fOpeningAngleULS; //! opening angle for ULS pairs
	95	TH1F *fPhotoElecPt; //! photonic elec pt
	96	TH1F *fSemiInclElecPt; //! Semi inclusive ele pt
3a94d6fe	97	THnSparse *fMCphotoElecPt; //! pt distribution (MC)
1d08b6e8	98
	99	TH1F *fTrackPtBefTrkCuts; //! Track pt before track cuts
	100	TH1F *fTrackPtAftTrkCuts; //! Track pt after track cuts
	101	TH2F *fTPCnsigma; //! TPC n sigma vs p
	102
	103	TH1F *fCent; //! centrality
5945f829	104	TH2F *fevPlaneV0A; //! V0A event plane distribution
	105	TH2F *fevPlaneV0C; //! V0C event plane distribution
	106	TH2F *fevPlaneTPC; //! TPC event plane distribution
1d08b6e8	107	TH2F *fTPCsubEPres; //! TPC event plane resolution
	108	THnSparse *fEPres; //! event plane resolution
	109	THnSparse *fCorr; //! correlations
	110	THnSparse *feTPCV2; //! inclusive eletron v2 (only TPC PID)
	111	THnSparse *feV2; //! inclusive eletron v2 (TPC + EMCAL PID)
	112	THnSparse *fphoteV2; //! photonic electron v2 (TPC + EMCAL PID)
	113	THnSparse *fChargPartV2; //! charged particle v2
5945f829	114
6522b3b5	115	TH2F *fGammaWeight; //! gamma weight
	116	TH2F *fPi0Weight; //! pi0 weight
	117	TH2F *fEtaWeight; //! eta weight
def484dc	118	TH2F *fD0Weight; //! D0 weight
	119	TH2F *fDplusWeight; //! D+ weight
	120	TH2F *fDminusWeight; //! D- weight
	121
937b2ff2	122	TH1F *fDe[6];
	123	TH1F *fD0e[6];
	124	TH1F *fDpluse[6];
	125	TH1F *fDminuse[6];
	126
	127	TH2F *fD0_e;
1c424962	128
	129	TH1F *fTot_pi0e; //! inclusive electron
	130	TH1F *fPhot_pi0e; //! ULS pair
	131	TH1F *fPhotBCG_pi0e; //! LS pair
	132	TH1F *fTot_etae; //! inclusive electron
	133	TH1F *fPhot_etae; //! ULS pair
	134	TH1F *fPhotBCG_etae; //! LS pair
	135
914042c2	136	AliAnalysisTaskFlowTPCEMCalEP(const AliAnalysisTaskFlowTPCEMCalEP&); // not implemented
914042c2	137	AliAnalysisTaskFlowTPCEMCalEP& operator=(const AliAnalysisTaskFlowTPCEMCalEP&); // not implemented
1d08b6e8	138
914042c2	139	ClassDef(AliAnalysisTaskFlowTPCEMCalEP, 1); //!example of analysis
1d08b6e8	140	};
	141
	142	#endif
	143
	144