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

#ifndef ALIANALYSISTASKSDMGAMMAMC_H
#define ALIANALYSISTASKSDMGAMMAMC_H

class TF1;
class TH1F;
class TH2F;
class TH3F;
class TH1D;
class TH2D;
class TH3D;
class TNtuple;
class TList;
class AliESDEvent;
class AliAODEvent;
class AliESDtrackCuts;
class AliESDCaloCluster;
class AliAODCaloCluster;
class AliMCEvent;
class AliMCParticle;
class AliEMCALGeometry;

#ifndef ALIANALYSISTASKSE_H
#include "AliAnalysisTaskSE.h"
#endif

class AliAnalysisTaskSDMGammaMC : public AliAnalysisTaskSE {
 public:
  AliAnalysisTaskSDMGammaMC();
  AliAnalysisTaskSDMGammaMC(const char *name);
  virtual ~AliAnalysisTaskSDMGammaMC();
    
  virtual void     UserCreateOutputObjects();
  virtual void     UserExec(Option_t *option);
  virtual void     Terminate(Option_t *);
    
  void         SetMcMode(Bool_t b) { fMcMode = b; }
  void         SetRecalScheme(Int_t kRecalibrator) { fRecalibrator = kRecalibrator; }
  void         SetFidPhiMinMax(Double_t kPhimin, Double_t kPhimax){ fPhimin = kPhimin; fPhimax = kPhimax; }
  void         SetFidEtaMinMax(Double_t kEtamin, Double_t kEtamax){ fEtamin = kEtamin; fEtamax = kEtamax; }
  
 private:
  static const int zvtx_bins = 16;
  static const int mult_bins = 25;
  static const unsigned int poolDepth = 25;
  
  Int_t GetMultBin(Int_t mult);
  Int_t GetZvtxBin(Double_t vertZ);
  Int_t isGoodEsdCluster(AliESDCaloCluster* esdclust);
  Int_t isGoodAodCluster(AliAODCaloCluster* aodclust);
  Double_t getDeltaPhi(TLorentzVector p1, TLorentzVector p2);
  Double_t getDeltaEta(TLorentzVector p1, TLorentzVector p2);
  Double_t PrivateEnergyRecal(Double_t energy, Int_t iCalib);

  TList           *fOutput;        //! Output list
  Bool_t           fMcMode;                 // monte carlo mode
  Int_t            fRecalibrator;           // custom recalibrator? 
  Double_t         fPhimin;           // Cuts. 
  Double_t         fPhimax;           // Cuts. 
  Double_t         fEtamin;           // Cuts. 
  Double_t         fEtamax;           // Cuts. 
  AliESDtrackCuts *fTrackCuts;     // Track cuts
  AliESDEvent     *fEsdEv;                  //!pointer to input esd event
  AliAODEvent     *fAodEv;                  //!pointer to input aod event
  TH1F            *h1_nClusters;  //!  # of clusters/evt
  TH1F            *h1_zvtx;  //!  # of clusters/evt
  TH1F            *h1_trigger;  //!  # of clusters/evt
  TH1F            *h1_E;        //!  energy spectrum
  TH1F            *h1_Phi;        //!  Pt spectrum

  TH2F            *h2_PiMotherID; //! first bin is primaries, second is non-primaries
  TH2F            *h2_GaMotherID; //! first bin is primaries, second is non-primaries
  TH3F            *h3_gE_RecTruth; //! E_in/E_rec vs E_rec (for 4 categories)
  TH3F            *h3_gE_RecTruth_ncellscut; //! E_in/E_rec vs E_rec (for 4 categories)
  
  TH1F            *h1_Pi0TruthPt;        //!  Pt spectrum from MC! 
  TH1F            *h1_PriPi0TruthPt;        //!  Pt spectrum from MC! 
  TH1F            *h1_Pi0TruthPtEmcal;        //!  Pt spectrum from MC! 
  TH1F            *h1_PriPi0TruthPtEmcal;        //!  Pt spectrum from MC! 

  TH2F            *h2_Pi0TruthPhiEta;    //!  etaphi spectrum from MC! 
  TH2F            *h2_PriPi0TruthPhiEta;    //!  etaphi spectrum from MC! 
  TH2F            *h2_Pi0TruthPhiEtaEmcal;    //!  etaphi spectrum from MC! 
  TH2F            *h2_PriPi0TruthPhiEtaEmcal;    //!  etaphi spectrum from MC! 

  TH1F            *h1_TruthPhotonsEmcal;        //!  Pt spectrum from MC! 
  TH2F            *h2_TruthPhotonsPhiEta;        //!  Pt spectrum from MC! 
  TH1F            *h1_PhotonsEmcal;        //!  Pt spectrum from MC! 
  TH1F            *h1_PhotonsNCellsCut;        //!  Pt spectrum from MC! 
  TH1F            *h1_PhotonsTrackMatchCut;        //!  Pt spectrum from MC! 
  TH1F            *h1_PhotonsAllCut;        //!  Pt spectrum from MC! 
  TH2F            *h2_PhotonsPhiEtaIsEmcal;        //!  Pt spectrum from MC! 

  TH1F            *h1_dR_RealMC;        //!  Pt spectrum from MC! 

  TH1F            *h1_Eta;       //!  pseudorapidity spectrum
  TH1F            *h1_Chi2;       //!  pseudorapidity spectrum
  TH1F            *h1_nTrkMatch;       //!  pseudorapidity spectrum
  TH1F            *h1_nCells;       //!  pseudorapidity spectrum
  TH1F            *h1_ClusterDisp;       //!  cluster dispersion
  TH2F            *h2_Ellipse;       //!  ellipse axis?
  TH2F            *h2_EtaPt;       //!  2d histogram Y - pseudorap spectrum
  TH2F            *h2_dphi_deta;       //!  2dimensional E vs mom
  TH2F            *h2_dphi_deta_mix;       //!  2dimensional E vs mom
  TH2F            *h2_DispRes;       //!  2dimensional E vs mom
  TH2F            *h2_cells_M02;       //!  

  std::vector<TLorentzVector> Photons[poolDepth][zvtx_bins][mult_bins];
  std::vector<Int_t> TriggerList;
  
  AliAnalysisTaskSDMGammaMC(const AliAnalysisTaskSDMGammaMC&); // not implemented
  AliAnalysisTaskSDMGammaMC& operator=(const AliAnalysisTaskSDMGammaMC&); // not implemented
    
  ClassDef(AliAnalysisTaskSDMGammaMC, 1); // example of analysis
};
#endif
Commit	Line	Data
3a66195e	1	#ifndef ALIANALYSISTASKSDMGAMMAMC_H
	2	#define ALIANALYSISTASKSDMGAMMAMC_H
	3
	4	class TF1;
	5	class TH1F;
	6	class TH2F;
	7	class TH3F;
	8	class TH1D;
	9	class TH2D;
	10	class TH3D;
	11	class TNtuple;
	12	class TList;
	13	class AliESDEvent;
	14	class AliAODEvent;
	15	class AliESDtrackCuts;
	16	class AliESDCaloCluster;
	17	class AliAODCaloCluster;
	18	class AliMCEvent;
	19	class AliMCParticle;
	20	class AliEMCALGeometry;
	21
	22	#ifndef ALIANALYSISTASKSE_H
	23	#include "AliAnalysisTaskSE.h"
	24	#endif
	25
	26	class AliAnalysisTaskSDMGammaMC : public AliAnalysisTaskSE {
	27	public:
	28	AliAnalysisTaskSDMGammaMC();
	29	AliAnalysisTaskSDMGammaMC(const char *name);
	30	virtual ~AliAnalysisTaskSDMGammaMC();
	31
	32	virtual void UserCreateOutputObjects();
	33	virtual void UserExec(Option_t *option);
	34	virtual void Terminate(Option_t *);
	35
	36	void SetMcMode(Bool_t b) { fMcMode = b; }
	37	void SetRecalScheme(Int_t kRecalibrator) { fRecalibrator = kRecalibrator; }
	38	void SetFidPhiMinMax(Double_t kPhimin, Double_t kPhimax){ fPhimin = kPhimin; fPhimax = kPhimax; }
	39	void SetFidEtaMinMax(Double_t kEtamin, Double_t kEtamax){ fEtamin = kEtamin; fEtamax = kEtamax; }
	40
	41	private:
	42	static const int zvtx_bins = 16;
	43	static const int mult_bins = 25;
	44	static const unsigned int poolDepth = 25;
	45
	46	Int_t GetMultBin(Int_t mult);
	47	Int_t GetZvtxBin(Double_t vertZ);
	48	Int_t isGoodEsdCluster(AliESDCaloCluster* esdclust);
	49	Int_t isGoodAodCluster(AliAODCaloCluster* aodclust);
	50	Double_t getDeltaPhi(TLorentzVector p1, TLorentzVector p2);
	51	Double_t getDeltaEta(TLorentzVector p1, TLorentzVector p2);
	52	Double_t PrivateEnergyRecal(Double_t energy, Int_t iCalib);
	53
	54	TList *fOutput; //! Output list
	55	Bool_t fMcMode; // monte carlo mode
	56	Int_t fRecalibrator; // custom recalibrator?
	57	Double_t fPhimin; // Cuts.
	58	Double_t fPhimax; // Cuts.
	59	Double_t fEtamin; // Cuts.
	60	Double_t fEtamax; // Cuts.
	61	AliESDtrackCuts *fTrackCuts; // Track cuts
	62	AliESDEvent *fEsdEv; //!pointer to input esd event
	63	AliAODEvent *fAodEv; //!pointer to input aod event
	64	TH1F *h1_nClusters; //! # of clusters/evt
65	TH1F *h1_zvtx; //! # of clusters/evt
66	TH1F *h1_trigger; //! # of clusters/evt
67	TH1F *h1_E; //! energy spectrum
68	TH1F *h1_Phi; //! Pt spectrum
69
70	TH2F *h2_PiMotherID; //! first bin is primaries, second is non-primaries
71	TH2F *h2_GaMotherID; //! first bin is primaries, second is non-primaries
72	TH3F *h3_gE_RecTruth; //! E_in/E_rec vs E_rec (for 4 categories)
73	TH3F *h3_gE_RecTruth_ncellscut; //! E_in/E_rec vs E_rec (for 4 categories)
74
75	TH1F *h1_Pi0TruthPt; //! Pt spectrum from MC!
76	TH1F *h1_PriPi0TruthPt; //! Pt spectrum from MC!
77	TH1F *h1_Pi0TruthPtEmcal; //! Pt spectrum from MC!
78	TH1F *h1_PriPi0TruthPtEmcal; //! Pt spectrum from MC!
79
80	TH2F *h2_Pi0TruthPhiEta; //! etaphi spectrum from MC!
81	TH2F *h2_PriPi0TruthPhiEta; //! etaphi spectrum from MC!
82	TH2F *h2_Pi0TruthPhiEtaEmcal; //! etaphi spectrum from MC!
83	TH2F *h2_PriPi0TruthPhiEtaEmcal; //! etaphi spectrum from MC!
84
85	TH1F *h1_TruthPhotonsEmcal; //! Pt spectrum from MC!
86	TH2F *h2_TruthPhotonsPhiEta; //! Pt spectrum from MC!
87	TH1F *h1_PhotonsEmcal; //! Pt spectrum from MC!
88	TH1F *h1_PhotonsNCellsCut; //! Pt spectrum from MC!
89	TH1F *h1_PhotonsTrackMatchCut; //! Pt spectrum from MC!
90	TH1F *h1_PhotonsAllCut; //! Pt spectrum from MC!
91	TH2F *h2_PhotonsPhiEtaIsEmcal; //! Pt spectrum from MC!
92
93	TH1F *h1_dR_RealMC; //! Pt spectrum from MC!
94
95	TH1F *h1_Eta; //! pseudorapidity spectrum
96	TH1F *h1_Chi2; //! pseudorapidity spectrum
97	TH1F *h1_nTrkMatch; //! pseudorapidity spectrum
98	TH1F *h1_nCells; //! pseudorapidity spectrum
99	TH1F *h1_ClusterDisp; //! cluster dispersion
100	TH2F *h2_Ellipse; //! ellipse axis?
101	TH2F *h2_EtaPt; //! 2d histogram Y - pseudorap spectrum
102	TH2F *h2_dphi_deta; //! 2dimensional E vs mom
103	TH2F *h2_dphi_deta_mix; //! 2dimensional E vs mom
104	TH2F *h2_DispRes; //! 2dimensional E vs mom
105	TH2F *h2_cells_M02; //!
106
107	std::vector<TLorentzVector> Photons[poolDepth][zvtx_bins][mult_bins];
108	std::vector<Int_t> TriggerList;
109
110	AliAnalysisTaskSDMGammaMC(const AliAnalysisTaskSDMGammaMC&); // not implemented
111	AliAnalysisTaskSDMGammaMC& operator=(const AliAnalysisTaskSDMGammaMC&); // not implemented
112
113	ClassDef(AliAnalysisTaskSDMGammaMC, 1); // example of analysis
114	};
115	#endif