[u/mrichter/AliRoot.git] / PWGHF / hfe / AliAnalysisTaskHFECal.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 ALIANALYSISTASKHFECAL_H
#define ALIANALYSISTASKHFECAL_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"
#include "AliStack.h"

class AliAnalysisTaskHFECal : public AliAnalysisTaskSE {
 public:
  AliAnalysisTaskHFECal();
  AliAnalysisTaskHFECal(const char *name);
  virtual ~AliAnalysisTaskHFECal();
  
  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;};
  void SetQAHist (int qahist) {fqahist = qahist;};
  AliHFEpid *GetPID() const { return fPID; }
  void SetRejectKinkMother(Bool_t rejectKinkMother = kFALSE) { fRejectKinkMother = rejectKinkMother; };
  void SelectPhotonicElectron(Int_t itrack, Double_t cent, AliESDtrack *track, Bool_t &fFlagPhotonicElec, Bool_t &fFlagConvinatElec, Double_t nSig, Double_t shower, Double_t ep, Double_t mce, Double_t w, Int_t ibgevent, Bool_t tagpi0, Bool_t tageta);
  void FindMother(TParticle* part, int &label, int &pid);
  double GetMCweight(double mcPi0pT);
  double GetMCweightEta(double mcEtapT);
  void FindTriggerClusters();
 private:
  
  Bool_t ProcessCutStep(Int_t cutStep, AliVParticle *track);
  
  AliESDEvent 		*fESD;			//!ESD object
  AliMCEvent 		*fMC;			//!MC object
  AliStack 		*stack;			//!MC object
  AliEMCALGeometry  	*fGeom; 		// emcal geometry 
    
  TList       		*fOutputList;		//! output list
  Int_t                 fqahist;  

  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                fmcData;
  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
 
  int ftriggers[48][60];//!
  int ftriggersCut[48][60];//!
  int ftriggersTime[48][60];//!
 

  TH1F			*fNoEvents;		//! no of events
  THnSparseD		*fEMCAccE;		//! EMC acc
  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
  TH2F			*fIncpT;		//! HFE pid electron vs centrality
  TH2F			*fIncpTM20;		//! HFE pid electron vs centrality
  THnSparseD		*fInvmassLS;		//! Inv mass of LS (e,e)
  THnSparseD		*fInvmassULS;		//! Inv mass of ULS (e,e)
  THnSparseD		*fInvmassLSmc;		//! Inv mass of LS (e,e)
  THnSparseD		*fInvmassULSmc;		//! Inv mass of ULS (e,e)
  TH2D		*fInvmassLSmc0;		//! Inv mass of ULS (e,e)
  TH2D		*fInvmassLSmc1;		//! Inv mass of ULS (e,e)
  TH2D		*fInvmassLSmc2;		//! Inv mass of ULS (e,e)
  TH2D		*fInvmassLSmc3;		//! Inv mass of ULS (e,e)
  TH2D		*fInvmassULSmc0;		//! Inv mass of ULS (e,e)
  TH2D		*fInvmassULSmc1;		//! Inv mass of ULS (e,e)
  TH2D		*fInvmassULSmc2;		//! Inv mass of ULS (e,e)
  TH2D		*fInvmassULSmc3;		//! 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 
  TH2F			*fPhoElecPt;	        //! Pho inclusive ele pt
  TH2F			*fPhoElecPtM20;	        //! Pho inclusive ele pt
  TH2F			*fSameElecPt;	        //! Same inclusive ele pt
  TH2F			*fSameElecPtM20;	        //! Same inclusive ele pt

  TH1F			*fTrackPtBefTrkCuts;	//! Track pt before track cuts	
  TH1F			*fTrackPtAftTrkCuts;	//! Track pt after track cuts
  TH2F			*fTPCnsigma;		//! TPC n sigma vs p	
  
  TH1F			*fCent;			//! centrality
  THnSparseD		*fEleInfo;		//! EMC acc
  /*
  //<---- trigger info
  TH1F	      *fClsEBftTrigCut;	//Cluster E before trigger selection
  TH1F        *fClsEAftTrigCut;	//Cluster E after trigger selection
  TH1F	      *fClsEAftTrigCut1;	//Cluster E after trigger selection
  TH1F	      *fClsEAftTrigCut2;	//Cluster E after trigger selection
  TH1F	      *fClsEAftTrigCut3;	//Cluster E after trigger selection
  TH1F	      *fClsEAftTrigCut4;	//Cluster E after trigger selection
  TH2F        *fClsETime; //ClsE vs time distribution
  TH2F        *fClsETime1; //ClsE vs time distribution
  TH1F        *fTrigTimes;// trigger time
  TH2F        *fCellCheck;// trigger time
  */
  //<------ MC
  TH2F                  *fInputHFEMC;
  TH2F                  *fInputAlle;
  TH2F			*fIncpTMChfe;		//! MC HFE pid electron vs centrality
  TH2F			*fIncpTMChfeAll;		//! MC HFE pid electron vs centrality
  TH2F			*fIncpTMCM20hfe;	//! MC HFE pid electron vs centrality
  TH2F			*fIncpTMCM20hfeAll;	//! MC HFE pid electron vs centrality
  TH2F			*fIncpTMCM20hfeCheck;	//! MC HFE pid electron vs centrality
 THnSparseD		*fInputHFEMC_weight;		//! MC HFE pid electron vs centrality
 THnSparseD		*fIncpTMCM20hfeCheck_weight;		//! MC HFE pid electron vs centrality
 THnSparseD		*fIncpTMCpho;		//! MC HFE pid electron vs centrality
 THnSparseD 		*fIncpTMCM20pho;	//! MC HFE pid electron vs centrality
 THnSparseD 		*fPhoElecPtMC;	        //! Pho inclusive ele pt
 THnSparseD 		*fPhoElecPtMCM20;	        //! Pho inclusive ele pt
 THnSparseD 		*fSameElecPtMC;	        //! Same inclusive ele pt
 THnSparseD 		*fSameElecPtMCM20;	        //! Same inclusive ele pt
 THnSparseD 		*fIncpTMCM20pho_pi0e;	//! MC HFE pid electron vs centrality
 THnSparseD 		*fPhoElecPtMCM20_pi0e;	        //! Pho inclusive ele pt
 THnSparseD 		*fSameElecPtMCM20_pi0e;	        //! Same inclusive ele pt
 THnSparseD 		*fIncpTMCM20pho_eta;	//! MC HFE pid electron vs centrality
 THnSparseD 		*fPhoElecPtMCM20_eta;	        //! Pho inclusive ele pt
 THnSparseD 		*fSameElecPtMCM20_eta;	        //! Same inclusive ele pt
 THnSparseD 		*fIncpTMCpho_pi0e_TPC;	//! MC HFE pid electron vs centrality
 THnSparseD 		*fPhoElecPtMC_pi0e_TPC;	        //! Pho inclusive ele pt
 THnSparseD 		*fSameElecPtMC_pi0e_TPC;	        //! Same inclusive ele pt
 TH1D                   *CheckNclust;  
 TH1D                   *CheckNits;  
 THnSparseD             *Hpi0pTcheck; 
 THnSparseD             *HETApTcheck; 
 TH2D                   *HphopTcheck; 
 TH2D                   *HDpTcheck; 
 TH2D                   *HBpTcheck; 
 TH1D                   *fpTCheck; 
 TH2D                   *fMomDtoE; 
 TH2D                   *fLabelCheck;
 TH2D                   *fgeoFake;
 
  AliAnalysisTaskHFECal(const AliAnalysisTaskHFECal&); // not implemented
  AliAnalysisTaskHFECal& operator=(const AliAnalysisTaskHFECal&); // not implemented
  
  ClassDef(AliAnalysisTaskHFECal, 1); //!example of analysis
};

#endif
Commit	Line	Data
bfc7c23b	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
	16	#ifndef ALIANALYSISTASKHFECAL_H
	17	#define ALIANALYSISTASKHFECAL_H
	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"
8806ce6c	39	#include "AliStack.h"
bfc7c23b	40
	41	class AliAnalysisTaskHFECal : public AliAnalysisTaskSE {
	42	public:
	43	AliAnalysisTaskHFECal();
	44	AliAnalysisTaskHFECal(const char *name);
	45	virtual ~AliAnalysisTaskHFECal();
	46
	47	virtual void UserCreateOutputObjects();
	48	virtual void UserExec(Option_t *option);
	49	virtual void Terminate(Option_t *);
	50
	51	void SetHFECuts(AliHFEcuts * const cuts) { fCuts = cuts; };
	52	void SetOpeningAngleCut (Double_t openingAngle) {fOpeningAngleCut = openingAngle;};
	53	void SetInvariantMassCut (Double_t invmass) {fInvmassCut = invmass;};
85985bb0	54	void SetQAHist (int qahist) {fqahist = qahist;};
bfc7c23b	55	AliHFEpid *GetPID() const { return fPID; }
bfc7c23b	56	void SetRejectKinkMother(Bool_t rejectKinkMother = kFALSE) { fRejectKinkMother = rejectKinkMother; };
2b4460a6	57	void SelectPhotonicElectron(Int_t itrack, Double_t cent, AliESDtrack *track, Bool_t &fFlagPhotonicElec, Bool_t &fFlagConvinatElec, Double_t nSig, Double_t shower, Double_t ep, Double_t mce, Double_t w, Int_t ibgevent, Bool_t tagpi0, Bool_t tageta);
46305ed6	58	void FindMother(TParticle* part, int &label, int &pid);
8806ce6c	59	double GetMCweight(double mcPi0pT);
93c189c5	60	double GetMCweightEta(double mcEtapT);
fb87d707	61	void FindTriggerClusters();
bfc7c23b	62	private:
	63
	64	Bool_t ProcessCutStep(Int_t cutStep, AliVParticle *track);
	65
	66	AliESDEvent *fESD; //!ESD object
e7d87aef	67	AliMCEvent *fMC; //!MC object
46305ed6	68	AliStack *stack; //!MC object
fb87d707	69	AliEMCALGeometry *fGeom; // emcal geometry
bfc7c23b	70
bfc7c23b	71	TList *fOutputList; //! output list
85985bb0	72	Int_t fqahist;
85985bb0	73
bfc7c23b	74	AliESDtrackCuts *fTrackCuts; //! ESD track cuts
	75	AliHFEcuts *fCuts; //! Cut Collection
	76	Bool_t fIdentifiedAsOutInz; //Out Of Range in z
	77	Bool_t fPassTheEventCut; //Pass The Event Cut
	78	Bool_t fRejectKinkMother; //Reject Kink Mother
e7d87aef	79	Bool_t fmcData;
bfc7c23b	80	Double_t fVz; //z position of the primary vertex
	81	AliCFManager *fCFM; //! Correction Framework Manager
	82	AliHFEpid *fPID; //! PID
	83	AliHFEpidQAmanager *fPIDqa; //! PID QA manager
	84	Double_t fOpeningAngleCut; //openingAngle cut value
	85	Double_t fInvmassCut; //invariant mass cut value
fb87d707	86
	87	int ftriggers[48][60];//!
	88	int ftriggersCut[48][60];//!
	89	int ftriggersTime[48][60];//!
	90
e7d87aef	91
bfc7c23b	92	TH1F *fNoEvents; //! no of events
	93	THnSparseD *fEMCAccE; //! EMC acc
	94	TH1F *fTrkpt; //! track pt
	95	TH2F *fTrkEovPBef; //! track E/p before HFE pid
	96	TH2F *fTrkEovPAft; //! track E/p after HFE pid
	97	TH2F *fdEdxBef; //! track dEdx vs p before HFE pid
	98	TH2F *fdEdxAft; //! track dEdx vs p after HFE pid
	99	TH2F *fIncpT; //! HFE pid electron vs centrality
fb87d707	100	TH2F *fIncpTM20; //! HFE pid electron vs centrality
f09766dd	101	THnSparseD *fInvmassLS; //! Inv mass of LS (e,e)
f09766dd	102	THnSparseD *fInvmassULS; //! Inv mass of ULS (e,e)
8806ce6c	103	THnSparseD *fInvmassLSmc; //! Inv mass of LS (e,e)
8806ce6c	104	THnSparseD *fInvmassULSmc; //! Inv mass of ULS (e,e)
93c189c5	105	TH2D *fInvmassLSmc0; //! Inv mass of ULS (e,e)
	106	TH2D *fInvmassLSmc1; //! Inv mass of ULS (e,e)
	107	TH2D *fInvmassLSmc2; //! Inv mass of ULS (e,e)
	108	TH2D *fInvmassLSmc3; //! Inv mass of ULS (e,e)
	109	TH2D *fInvmassULSmc0; //! Inv mass of ULS (e,e)
	110	TH2D *fInvmassULSmc1; //! Inv mass of ULS (e,e)
	111	TH2D *fInvmassULSmc2; //! Inv mass of ULS (e,e)
	112	TH2D *fInvmassULSmc3; //! Inv mass of ULS (e,e)
bfc7c23b	113	TH1F *fOpeningAngleLS; //! opening angle for LS pairs
	114	TH1F *fOpeningAngleULS; //! opening angle for ULS pairs
	115	TH1F *fPhotoElecPt; //! photonic elec pt
f09766dd	116	TH2F *fPhoElecPt; //! Pho inclusive ele pt
fb87d707	117	TH2F *fPhoElecPtM20; //! Pho inclusive ele pt
f09766dd	118	TH2F *fSameElecPt; //! Same inclusive ele pt
fb87d707	119	TH2F *fSameElecPtM20; //! Same inclusive ele pt
a6df418c	120
bfc7c23b	121	TH1F *fTrackPtBefTrkCuts; //! Track pt before track cuts
	122	TH1F *fTrackPtAftTrkCuts; //! Track pt after track cuts
	123	TH2F *fTPCnsigma; //! TPC n sigma vs p
	124
	125	TH1F *fCent; //! centrality
	126	THnSparseD *fEleInfo; //! EMC acc
feffe705	127	/*
fb87d707	128	//<---- trigger info
	129	TH1F *fClsEBftTrigCut; //Cluster E before trigger selection
	130	TH1F *fClsEAftTrigCut; //Cluster E after trigger selection
	131	TH1F *fClsEAftTrigCut1; //Cluster E after trigger selection
	132	TH1F *fClsEAftTrigCut2; //Cluster E after trigger selection
	133	TH1F *fClsEAftTrigCut3; //Cluster E after trigger selection
	134	TH1F *fClsEAftTrigCut4; //Cluster E after trigger selection
	135	TH2F *fClsETime; //ClsE vs time distribution
	136	TH2F *fClsETime1; //ClsE vs time distribution
	137	TH1F *fTrigTimes;// trigger time
42c75692	138	TH2F *fCellCheck;// trigger time
feffe705	139	*/
e7d87aef	140	//<------ MC
e7d87aef	141	TH2F *fInputHFEMC;
feffe705	142	TH2F *fInputAlle;
e7d87aef	143	TH2F *fIncpTMChfe; //! MC HFE pid electron vs centrality
3db00c72	144	TH2F *fIncpTMChfeAll; //! MC HFE pid electron vs centrality
e7d87aef	145	TH2F *fIncpTMCM20hfe; //! MC HFE pid electron vs centrality
3db00c72	146	TH2F *fIncpTMCM20hfeAll; //! MC HFE pid electron vs centrality
60544aea	147	TH2F *fIncpTMCM20hfeCheck; //! MC HFE pid electron vs centrality
bd6ee6dd	148	THnSparseD *fInputHFEMC_weight; //! MC HFE pid electron vs centrality
bd6ee6dd	149	THnSparseD *fIncpTMCM20hfeCheck_weight; //! MC HFE pid electron vs centrality
44be9e1d	150	THnSparseD *fIncpTMCpho; //! MC HFE pid electron vs centrality
	151	THnSparseD *fIncpTMCM20pho; //! MC HFE pid electron vs centrality
	152	THnSparseD *fPhoElecPtMC; //! Pho inclusive ele pt
	153	THnSparseD *fPhoElecPtMCM20; //! Pho inclusive ele pt
	154	THnSparseD *fSameElecPtMC; //! Same inclusive ele pt
	155	THnSparseD *fSameElecPtMCM20; //! Same inclusive ele pt
e4b0faf2	156	THnSparseD *fIncpTMCM20pho_pi0e; //! MC HFE pid electron vs centrality
	157	THnSparseD *fPhoElecPtMCM20_pi0e; //! Pho inclusive ele pt
	158	THnSparseD *fSameElecPtMCM20_pi0e; //! Same inclusive ele pt
93c189c5	159	THnSparseD *fIncpTMCM20pho_eta; //! MC HFE pid electron vs centrality
	160	THnSparseD *fPhoElecPtMCM20_eta; //! Pho inclusive ele pt
	161	THnSparseD *fSameElecPtMCM20_eta; //! Same inclusive ele pt
697ecf6b	162	THnSparseD *fIncpTMCpho_pi0e_TPC; //! MC HFE pid electron vs centrality
	163	THnSparseD *fPhoElecPtMC_pi0e_TPC; //! Pho inclusive ele pt
	164	THnSparseD *fSameElecPtMC_pi0e_TPC; //! Same inclusive ele pt
d113d7cd	165	TH1D *CheckNclust;
d113d7cd	166	TH1D *CheckNits;
68d718e7	167	THnSparseD *Hpi0pTcheck;
68d718e7	168	THnSparseD *HETApTcheck;
e4b0faf2	169	TH2D *HphopTcheck;
5e0e45b3	170	TH2D *HDpTcheck;
5e0e45b3	171	TH2D *HBpTcheck;
93c189c5	172	TH1D *fpTCheck;
50919258	173	TH2D *fMomDtoE;
70448e3b	174	TH2D *fLabelCheck;
70448e3b	175	TH2D *fgeoFake;
d113d7cd	176
bfc7c23b	177	AliAnalysisTaskHFECal(const AliAnalysisTaskHFECal&); // not implemented
	178	AliAnalysisTaskHFECal& operator=(const AliAnalysisTaskHFECal&); // not implemented
	179
	180	ClassDef(AliAnalysisTaskHFECal, 1); //!example of analysis
	181	};
	182
	183	#endif
	184
	185