[u/mrichter/AliRoot.git] / PWGGA / EMCALTasks / AliAnalysisTaskEMCALIsoPhoton.h

#ifndef AliAnalysisTaskEMCALIsoPhoton_h
#define AliAnalysisTaskEMCALIsoPhoton_h

// $Id$

class TH1F;
class TH2F;
class TH3F;
class THnSparse;
class TList;
class TObjArray;
class AliEMCALGeometry;
class AliOADBContainer;
class AliESDCaloCells;
class AliESDEvent;
class AliESDtrack;
class AliESDtrackCuts;
class AliAODEvent;
class AliAODCaloCells;
class AliVCluster;
class AliMCEvent;
class AliStack;
class TParticle;
class TGeoHMatrix;

#include "AliAnalysisTaskSE.h"

class AliAnalysisTaskEMCALIsoPhoton : public AliAnalysisTaskSE {
 public:
  AliAnalysisTaskEMCALIsoPhoton();
  AliAnalysisTaskEMCALIsoPhoton(const char *name);
  virtual ~AliAnalysisTaskEMCALIsoPhoton() {}

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

  void                   GetCeIso(TVector3 vec, Int_t maxid, Float_t &iso, Float_t &phiband, Float_t &core);
  Double_t               GetCrossEnergy(const AliVCluster *cluster, Short_t &idmax);
  Double_t               GetMaxCellEnergy(const AliVCluster *cluster, Short_t &id) const; 
  void                   GetTrIso(TVector3 vec, Float_t &iso, Float_t &phiband, Float_t &core);
  Double_t               GetTrackMatchedPt(Int_t matchIndex);
  void                   FillClusHists();
  void                   FillMcHists();
  void                   FillQA();
  Float_t                GetClusSource(const AliVCluster *cluster);
  void                   FollowGamma();
  void                   GetDaughtersInfo(int firstd, int lastd, int selfid, const char *indputindent);
  Float_t                GetMcPtSumInCone(Float_t etaclus, Float_t phiclus, Float_t R);
  void                   LoopOnCells();
  void                   SetExotCut(Double_t c)                 { fExoticCut          = c;       }
  void                   SetGeoName(const char *n)              { fGeoName            = n;       }
  void                   SetIsoConeR(Double_t r)                { fIsoConeR           = r;       }
  void                   SetPeriod(const char *n)               { fPeriod             = n;       }
  void                   SetTriggerBit(const char *tb)          { fTrigBit            = tb;      }
  void                   SetPrimTrackCuts(AliESDtrackCuts *c)   { fPrTrCuts           = c;       }
  void                   SetTrainMode(Bool_t t)                 { fIsTrain            = t;       }
  void                   SetMcMode(Bool_t mc)                   { fIsMc               = mc;      }
  void                   SetDebugOn(Bool_t d)                   { fDebug              = d;       }
  void                   SetPathStringSelect(char *p)           { fPathStrOpt         = p;       }
  void                   SetEtCut(Double_t ec)                  { fECut               = ec;      }
  void                   SetImportGeometryFromFile(Bool_t  im, 
                                           TString pa = "")     { fImportGeometryFromFile = im ; 
                                                                  fImportGeometryFilePath = pa ; }    
  void                  SetTrackFilterBit(ULong_t bit)          { fFilterBit = bit;  }
  

 protected:
  TObjArray             *fESDClusters;           //!pointer to EMCal clusters
  TObjArray             *fAODClusters;           //!pointer to EMCal clusters
  TObjArray             *fSelPrimTracks;         //!pointer to ESD primary tracks
  TClonesArray          *fTracks;                //!track input array
  AliESDCaloCells       *fESDCells;              //!pointer to EMCal cells, esd
  AliAODCaloCells       *fAODCells;              //!pointer to EMCal cells, aod  
  AliESDtrackCuts       *fPrTrCuts;              //pointer to hold the prim track cuts
  AliEMCALGeometry      *fGeom;                  // geometry utils
  TString                fGeoName;               // geometry name (def = EMCAL_FIRSTYEARV1)
  AliOADBContainer      *fOADBContainer;         //!OADB container used to load misalignment matrices
  TString                fPeriod;                // string to the LHC period
  TString                fTrigBit;               // string to the trigger bit name
  Bool_t                 fIsTrain;               // variable to set train mode
  Bool_t                 fIsMc;                  // variable to set mc mode
  Bool_t                 fDebug;                 // variable to set on/off debugging printouts
  TString                fPathStrOpt;            // variable to set the name of files to be analyzed (MC only)
  Double_t               fExoticCut;             // variable to set the cut on exotic clusters
  Double_t               fIsoConeR;              // variable to set the isolation cone radius
  Int_t                  fNDimensions;           // variable to set the number of dimensions of n-sparse
  Double_t               fECut;                  // variable to set the minimum E of a cluster
  Int_t                  fTrackMult;             // global variable with the event multiplicity        
  TString                fMcIdFamily;            // string that holds the ids of all particles originated from the prompt photon
  Int_t                  fNClusForDirPho;        // number of clusters from prompt photon per event
  Float_t                fDirPhoPt;              // prompt photon pt (assumes only one per event)
  Float_t                fHigherPtCone;          // higher pt inside the cone around the candidate
  Bool_t                 fImportGeometryFromFile;  // Import geometry settings in geometry.root file
  TString                fImportGeometryFilePath;  // path fo geometry.root file
  Double_t               fMaxPtTrack;            //track with highest pt in event
  Double_t               fMaxEClus;              //cluster with highest energy in event
  Int_t                  fNCells50;              // variable to keep the number of cells with E>50 MeV
  ULong_t                fFilterBit;             // Track selection bit, for AODs 


 private:
  AliESDEvent *fESD;      //! ESD object
  AliAODEvent *fAOD;      //! AOD object
  AliMCEvent  *fMCEvent;  //! MC event object
  AliStack    *fStack;    //!MC particles stack object
  TGeoHMatrix *fGeomMatrix[12];//! Geometry misalignment matrices for EMCal
  TList       *fOutputList; //! Output list
  //histograms for events with 1+ track pt>1
  TH1F        *fEvtSel;                    //!evt selection counter: 0=all trg, 1=pv cut 
  TH1F        *fNClusEt10;                 //!number of clusters w/ Et>10 in the event
  TH1F        *fRecoPV;                    //!histogram to record if an event has a prim. vert.
  TH1F        *fPVtxZ;                     //!primary vertex Z before cut
  TH1F        *fTrMultDist;                //!track multiplicity distribution
  TH3F        *fMCDirPhotonPtEtaPhi;       //!direct produced photon pt, eta, phi
  TH3F        *fMCIsoDirPhotonPtEtaPhi;    //!direct produced photon pt, eta, phi, isolated @ mc level
  TH1F        *fDecayPhotonPtMC;           //!decay photon pt
  TH2F        *fCellAbsIdVsAmpl;           //!cell abs id vs cell amplitude (energy)
  TH2F        *fNClusHighClusE;            //!total number of clusters vs. highest clus energy in the event
  TH2F        *fHigherPtConeM02;           //!M02 vs. the higher pt of a track inside the cone
  TH2F        *fClusEtMcPt;                //!cluster et x mc-pt
  TH2F        *fClusMcDetaDphi;            //!delta-eta x delta-phi(reco-mc)
  TH2F        *fNClusPerPho;               //!delta-eta x delta-phi(reco-mc)
  TH2F        *fMcPtInConeBG;              //!sum of mc-pt of "primary" particles inside de cone, as a function of NET-ISO in BG template
  TH2F        *fMcPtInConeSBG;             //!sum of mc-pt of "primary" particles inside de cone, as a function of NET-ISO in SBG range
  TH2F        *fMcPtInConeBGnoUE;          //!sum of mc-pt of "primary" particles inside de cone, as a function of NET-ISO in BG template no UE sub
  TH2F        *fMcPtInConeSBGnoUE;         //!sum of mc-pt of "primary" particles inside de cone, as a function of NET-ISO in SBG range no UE sub
  TH2F        *fAllIsoEtMcGamma;           //!all iso distribution vs. Et clus for clusters comming from a MC prompt photon
  TH2F        *fAllIsoNoUeEtMcGamma;       //!all iso distribution (without UE subtraction) vs. Et clus for clusters comming from a MC prompt photon
  TH3F        *fMCDirPhotonPtEtaPhiNoClus; //!pt x eta x phi for prompt photons that didn't produce clusters
  THnSparse   *fHnOutput;                  //!Output matrix with 7 dimensions

  //QA histos
  TList       *fQAList;           //!output list holding QA histos
  TH1F        *fNTracks;          //!number of tracks from Array->GetEntries()
  TH1F        *fEmcNCells;        //!number of emcal cells in the event
  TH1F        *fEmcNClus;         //!# of emcal clusters
  TH1F        *fEmcNClusCut;      //!# of clusters in an event with at least 1 clus with E > fECut ("triggered event")
  TH1F        *fNTracksECut;      //!number of tracks from Array->GetEntries() in "triggered event"
  TH1F        *fEmcNCellsCut;     //!number of emcal cells in a in "triggered event"
  TH2F        *fEmcClusEPhi;      //!cluster E spectrum vs. phi
  TH2F        *fEmcClusEPhiCut;   //!cluster E spectrum vs. phi in "triggered event"
  TH2F        *fEmcClusEEta;      //!cluster E spectrum vs. eta
  TH2F        *fEmcClusEEtaCut;   //!cluster E spectrum vs. eta in "triggered event"
  TH2F        *fTrackPtPhi;       //!selected tracks pt vs. phi
  TH2F        *fTrackPtPhiCut;    //!selected tracks pt vs. phi in "triggered event"
  TH2F        *fTrackPtEta;       //!selected tracks pt vs. eta
  TH2F        *fTrackPtEtaCut;    //!selected tracks pt vs. eta in "triggered event"
  TH2F        *fMaxCellEPhi;      //!max cell energy vs. cell phi


  AliAnalysisTaskEMCALIsoPhoton(const AliAnalysisTaskEMCALIsoPhoton&); // not implemented
  AliAnalysisTaskEMCALIsoPhoton& operator=(const AliAnalysisTaskEMCALIsoPhoton&); // not implemented
  
  ClassDef(AliAnalysisTaskEMCALIsoPhoton, 1); // Class to analyse isolated photons
};
#endif
Commit	Line	Data
bd092f0f	1	#ifndef AliAnalysisTaskEMCALIsoPhoton_h
	2	#define AliAnalysisTaskEMCALIsoPhoton_h
	3
	4	// $Id$
	5
	6	class TH1F;
	7	class TH2F;
caaf99d3	8	class TH3F;
965c985f	9	class THnSparse;
a62631a9	10	class TList;
bd092f0f	11	class TObjArray;
bd092f0f	12	class AliEMCALGeometry;
4d4ce2d2	13	class AliOADBContainer;
bd092f0f	14	class AliESDCaloCells;
	15	class AliESDEvent;
	16	class AliESDtrack;
	17	class AliESDtrackCuts;
2e20efe5	18	class AliAODEvent;
2e20efe5	19	class AliAODCaloCells;
bd092f0f	20	class AliVCluster;
c7bb0b43	21	class AliMCEvent;
	22	class AliStack;
	23	class TParticle;
822c9944	24	class TGeoHMatrix;
bd092f0f	25
	26	#include "AliAnalysisTaskSE.h"
	27
	28	class AliAnalysisTaskEMCALIsoPhoton : public AliAnalysisTaskSE {
	29	public:
	30	AliAnalysisTaskEMCALIsoPhoton();
	31	AliAnalysisTaskEMCALIsoPhoton(const char *name);
	32	virtual ~AliAnalysisTaskEMCALIsoPhoton() {}
	33
	34	void UserCreateOutputObjects();
	35	void UserExec(Option_t *option);
	36	void Terminate(Option_t *);
	37
f224025f	38	void GetCeIso(TVector3 vec, Int_t maxid, Float_t &iso, Float_t &phiband, Float_t &core);
bd092f0f	39	Double_t GetCrossEnergy(const AliVCluster *cluster, Short_t &idmax);
	40	Double_t GetMaxCellEnergy(const AliVCluster *cluster, Short_t &id) const;
	41	void GetTrIso(TVector3 vec, Float_t &iso, Float_t &phiband, Float_t &core);
1c86c72c	42	Double_t GetTrackMatchedPt(Int_t matchIndex);
bd092f0f	43	void FillClusHists();
c7bb0b43	44	void FillMcHists();
2b7205ad	45	void FillQA();
b8b781cb	46	Float_t GetClusSource(const AliVCluster *cluster);
f507c09b	47	void FollowGamma();
22ad7981	48	void GetDaughtersInfo(int firstd, int lastd, int selfid, const char *indputindent);
f912f9a9	49	Float_t GetMcPtSumInCone(Float_t etaclus, Float_t phiclus, Float_t R);
112eb594	50	void LoopOnCells();
bd092f0f	51	void SetExotCut(Double_t c) { fExoticCut = c; }
	52	void SetGeoName(const char *n) { fGeoName = n; }
	53	void SetIsoConeR(Double_t r) { fIsoConeR = r; }
	54	void SetPeriod(const char *n) { fPeriod = n; }
751194e8	55	void SetTriggerBit(const char *tb) { fTrigBit = tb; }
bd092f0f	56	void SetPrimTrackCuts(AliESDtrackCuts *c) { fPrTrCuts = c; }
bd092f0f	57	void SetTrainMode(Bool_t t) { fIsTrain = t; }
c7bb0b43	58	void SetMcMode(Bool_t mc) { fIsMc = mc; }
ecd47673	59	void SetDebugOn(Bool_t d) { fDebug = d; }
f3843637	60	void SetPathStringSelect(char *p) { fPathStrOpt = p; }
f507c09b	61	void SetEtCut(Double_t ec) { fECut = ec; }
26a8db0f	62	void SetImportGeometryFromFile(Bool_t im,
	63	TString pa = "") { fImportGeometryFromFile = im ;
	64	fImportGeometryFilePath = pa ; }
48aab590	65	void SetTrackFilterBit(ULong_t bit) { fFilterBit = bit; }
bd092f0f	66
112eb594	67
bd092f0f	68	protected:
2e20efe5	69	TObjArray *fESDClusters; //!pointer to EMCal clusters
2e20efe5	70	TObjArray *fAODClusters; //!pointer to EMCal clusters
bd092f0f	71	TObjArray *fSelPrimTracks; //!pointer to ESD primary tracks
3f4073ba	72	TClonesArray *fTracks; //!track input array
2e20efe5	73	AliESDCaloCells *fESDCells; //!pointer to EMCal cells, esd
2e20efe5	74	AliAODCaloCells *fAODCells; //!pointer to EMCal cells, aod
ae41967e	75	AliESDtrackCuts *fPrTrCuts; //pointer to hold the prim track cuts
bd092f0f	76	AliEMCALGeometry *fGeom; // geometry utils
bd092f0f	77	TString fGeoName; // geometry name (def = EMCAL_FIRSTYEARV1)
4d4ce2d2	78	AliOADBContainer *fOADBContainer; //!OADB container used to load misalignment matrices
bd092f0f	79	TString fPeriod; // string to the LHC period
751194e8	80	TString fTrigBit; // string to the trigger bit name
bd092f0f	81	Bool_t fIsTrain; // variable to set train mode
c7bb0b43	82	Bool_t fIsMc; // variable to set mc mode
ecd47673	83	Bool_t fDebug; // variable to set on/off debugging printouts
f3843637	84	TString fPathStrOpt; // variable to set the name of files to be analyzed (MC only)
bd092f0f	85	Double_t fExoticCut; // variable to set the cut on exotic clusters
bd092f0f	86	Double_t fIsoConeR; // variable to set the isolation cone radius
965c985f	87	Int_t fNDimensions; // variable to set the number of dimensions of n-sparse
965c985f	88	Double_t fECut; // variable to set the minimum E of a cluster
16a4050e	89	Int_t fTrackMult; // global variable with the event multiplicity
f507c09b	90	TString fMcIdFamily; // string that holds the ids of all particles originated from the prompt photon
	91	Int_t fNClusForDirPho; // number of clusters from prompt photon per event
	92	Float_t fDirPhoPt; // prompt photon pt (assumes only one per event)
f9e2362a	93	Float_t fHigherPtCone; // higher pt inside the cone around the candidate
26a8db0f	94	Bool_t fImportGeometryFromFile; // Import geometry settings in geometry.root file
26a8db0f	95	TString fImportGeometryFilePath; // path fo geometry.root file
112eb594	96	Double_t fMaxPtTrack; //track with highest pt in event
	97	Double_t fMaxEClus; //cluster with highest energy in event
	98	Int_t fNCells50; // variable to keep the number of cells with E>50 MeV
48aab590	99	ULong_t fFilterBit; // Track selection bit, for AODs
48aab590	100
16a4050e	101
bd092f0f	102
	103	private:
	104	AliESDEvent *fESD; //! ESD object
2e20efe5	105	AliAODEvent *fAOD; //! AOD object
c7bb0b43	106	AliMCEvent *fMCEvent; //! MC event object
c7bb0b43	107	AliStack *fStack; //!MC particles stack object
822c9944	108	TGeoHMatrix *fGeomMatrix[12];//! Geometry misalignment matrices for EMCal
bd092f0f	109	TList *fOutputList; //! Output list
bd092f0f	110	//histograms for events with 1+ track pt>1
092ceec8	111	TH1F *fEvtSel; //!evt selection counter: 0=all trg, 1=pv cut
092ceec8	112	TH1F *fNClusEt10; //!number of clusters w/ Et>10 in the event
cc57d293	113	TH1F *fRecoPV; //!histogram to record if an event has a prim. vert.
092ceec8	114	TH1F *fPVtxZ; //!primary vertex Z before cut
092ceec8	115	TH1F *fTrMultDist; //!track multiplicity distribution
e53ab710	116	TH3F *fMCDirPhotonPtEtaPhi; //!direct produced photon pt, eta, phi
e53ab710	117	TH3F *fMCIsoDirPhotonPtEtaPhi; //!direct produced photon pt, eta, phi, isolated @ mc level
092ceec8	118	TH1F *fDecayPhotonPtMC; //!decay photon pt
	119	TH2F *fCellAbsIdVsAmpl; //!cell abs id vs cell amplitude (energy)
	120	TH2F *fNClusHighClusE; //!total number of clusters vs. highest clus energy in the event
f9e2362a	121	TH2F *fHigherPtConeM02; //!M02 vs. the higher pt of a track inside the cone
092ceec8	122	TH2F *fClusEtMcPt; //!cluster et x mc-pt
	123	TH2F *fClusMcDetaDphi; //!delta-eta x delta-phi(reco-mc)
	124	TH2F *fNClusPerPho; //!delta-eta x delta-phi(reco-mc)
f912f9a9	125	TH2F *fMcPtInConeBG; //!sum of mc-pt of "primary" particles inside de cone, as a function of NET-ISO in BG template
	126	TH2F *fMcPtInConeSBG; //!sum of mc-pt of "primary" particles inside de cone, as a function of NET-ISO in SBG range
	127	TH2F *fMcPtInConeBGnoUE; //!sum of mc-pt of "primary" particles inside de cone, as a function of NET-ISO in BG template no UE sub
	128	TH2F *fMcPtInConeSBGnoUE; //!sum of mc-pt of "primary" particles inside de cone, as a function of NET-ISO in SBG range no UE sub
81f2660f	129	TH2F *fAllIsoEtMcGamma; //!all iso distribution vs. Et clus for clusters comming from a MC prompt photon
81f2660f	130	TH2F *fAllIsoNoUeEtMcGamma; //!all iso distribution (without UE subtraction) vs. Et clus for clusters comming from a MC prompt photon
092ceec8	131	TH3F *fMCDirPhotonPtEtaPhiNoClus; //!pt x eta x phi for prompt photons that didn't produce clusters
092ceec8	132	THnSparse *fHnOutput; //!Output matrix with 7 dimensions
965c985f	133
2b7205ad	134	//QA histos
	135	TList *fQAList; //!output list holding QA histos
	136	TH1F *fNTracks; //!number of tracks from Array->GetEntries()
	137	TH1F *fEmcNCells; //!number of emcal cells in the event
	138	TH1F *fEmcNClus; //!# of emcal clusters
	139	TH1F *fEmcNClusCut; //!# of clusters in an event with at least 1 clus with E > fECut ("triggered event")
	140	TH1F *fNTracksECut; //!number of tracks from Array->GetEntries() in "triggered event"
	141	TH1F *fEmcNCellsCut; //!number of emcal cells in a in "triggered event"
e4ba80ad	142	TH2F *fEmcClusEPhi; //!cluster E spectrum vs. phi
	143	TH2F *fEmcClusEPhiCut; //!cluster E spectrum vs. phi in "triggered event"
	144	TH2F *fEmcClusEEta; //!cluster E spectrum vs. eta
	145	TH2F *fEmcClusEEtaCut; //!cluster E spectrum vs. eta in "triggered event"
	146	TH2F *fTrackPtPhi; //!selected tracks pt vs. phi
	147	TH2F *fTrackPtPhiCut; //!selected tracks pt vs. phi in "triggered event"
	148	TH2F *fTrackPtEta; //!selected tracks pt vs. eta
	149	TH2F *fTrackPtEtaCut; //!selected tracks pt vs. eta in "triggered event"
3ae97198	150	TH2F *fMaxCellEPhi; //!max cell energy vs. cell phi
2b7205ad	151
2b7205ad	152
bd092f0f	153	AliAnalysisTaskEMCALIsoPhoton(const AliAnalysisTaskEMCALIsoPhoton&); // not implemented
	154	AliAnalysisTaskEMCALIsoPhoton& operator=(const AliAnalysisTaskEMCALIsoPhoton&); // not implemented
	155
	156	ClassDef(AliAnalysisTaskEMCALIsoPhoton, 1); // Class to analyse isolated photons
	157	};
	158	#endif