[u/mrichter/AliRoot.git] / PWGCF / Correlations / DPhi / MuonHadron / AliAnalysisTaskMuonHadronCorrelations.h

#ifndef AliAnalysisTaskMuonHadronCorrelations_H
#define AliAnalysisTaskMuonHadronCorrelations_H

#include "AliAnalysisTaskSE.h"
#include "TString.h"
#include "AliAODEvent.h"
#include "AliAODTrack.h"
#include "TAxis.h"
#include "TH1D.h"
#include "TClonesArray.h"
#include "TH2D.h"

class AliEventPoolManager;

//====================================================================================================================================================

class  AliAnalysisTaskMuonHadronCorrelations : public AliAnalysisTaskSE {

 public:
 
  enum {kSingleEvent, kMixedEvent};

  AliAnalysisTaskMuonHadronCorrelations();
  AliAnalysisTaskMuonHadronCorrelations(const Char_t *name);
  virtual ~AliAnalysisTaskMuonHadronCorrelations();

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

  // ------------- Cuts -----------------

  void SetFilterBitCentralBarrel(Int_t filter) { fFilterBitCentralBarrel = filter; }
  void SetMaxEtaCentralBarrel(Double_t eta) { fMaxEtaCentralBarrel = eta; }
  void SetTriggerMatchLevelMuon(Short_t level) { fTriggerMatchLevelMuon = level; }
  //  void SetMaxChi2Muon(Double_t chi2Max) { fMaxChi2Muon = chi2Max; }
  void SetRAbsRangeMuon (Double_t rAbsMin,Double_t rAbsMax) { fMinRAbsMuon = rAbsMin; fMaxRAbsMuon = rAbsMax; }

  // ------------- Analysis -------------

  Float_t GetV0Multiplicity();
  Double_t GetITSMultiplicity();
  Bool_t IsTriggerFired();
  TObjArray* GetAcceptedTracksCentralBarrel(AliAODEvent *aodEvent);
  TObjArray* GetAcceptedTracksMuonArm(AliAODEvent *aodEvent, Int_t centBin);
  void SetPtBinning(Int_t nBins, Double_t *limits);
  void SetCentBinning(Int_t nBins, Double_t *limits);
  void SetCentMethod(const Char_t *method) { fCentMethod = method; }
  void FillHistograms(Int_t centrality, Int_t option);
  Int_t GetCentBin();

 private:

  static const Int_t fNMaxBinsCentrality = 20;
  static const Int_t fNMaxBinsPt = 10;

  AliAODEvent *fAOD; //!
  AliEventPoolManager *fPoolMgr; //! event pool manager
  AliAODTrack *fTrackCB; //!
  AliAODTrack *fTrackMA; //!
  
  Int_t fFilterBitCentralBarrel;
  Double_t fMaxEtaCentralBarrel;

  Double_t fMaxChi2Muon, fMinRAbsMuon, fMaxRAbsMuon;
  Short_t fTriggerMatchLevelMuon;

  Int_t fNbinsCent, fNbinsPt;
  
  TAxis *fCentAxis;
  TAxis *fPtAxis;

  TH1D *fHistDeltaPhi[fNMaxBinsCentrality][fNMaxBinsPt][fNMaxBinsPt]; //!
  TH1D *fHistDeltaPhiMix[fNMaxBinsCentrality][fNMaxBinsPt][fNMaxBinsPt]; //!
  TH2D *fHistNTracksCB_vs_NTracksMA[fNMaxBinsCentrality]; //!
  TH2D *fHistNTracksCB_vs_NTracksMAmixed[fNMaxBinsCentrality]; //!

  TH1D *fHistV0Multiplicity; //!
  TH1D *fHistITSMultiplicity; //!
  TH1D *fHistCentrality; //!

  TH2D *fHistSingleMuonsEtaVsPt[fNMaxBinsCentrality]; //!
  TH2D *fHistSingleMuonsEtaVsRAbs[fNMaxBinsCentrality]; //!

  TString fCentMethod;

  TList *fOutputList; //!

  AliAnalysisTaskMuonHadronCorrelations(const AliAnalysisTaskMuonHadronCorrelations&);//not implimented
  AliAnalysisTaskMuonHadronCorrelations& operator=(const AliAnalysisTaskMuonHadronCorrelations&);//not implimnted
  
  ClassDef(AliAnalysisTaskMuonHadronCorrelations, 1)  // example of analysis

};

//====================================================================================================================================================

#endif
Commit	Line	Data
07c2615a	1	#ifndef AliAnalysisTaskMuonHadronCorrelations_H
	2	#define AliAnalysisTaskMuonHadronCorrelations_H
	3
	4	#include "AliAnalysisTaskSE.h"
	5	#include "TString.h"
	6	#include "AliAODEvent.h"
	7	#include "AliAODTrack.h"
	8	#include "TAxis.h"
	9	#include "TH1D.h"
	10	#include "TClonesArray.h"
	11	#include "TH2D.h"
	12
6de71f1d	13	class AliEventPoolManager;
6de71f1d	14
07c2615a	15	//====================================================================================================================================================
	16
	17	class AliAnalysisTaskMuonHadronCorrelations : public AliAnalysisTaskSE {
	18
	19	public:
	20
	21	enum {kSingleEvent, kMixedEvent};
	22
	23	AliAnalysisTaskMuonHadronCorrelations();
	24	AliAnalysisTaskMuonHadronCorrelations(const Char_t *name);
	25	virtual ~AliAnalysisTaskMuonHadronCorrelations();
	26
	27	virtual void UserCreateOutputObjects();
	28	virtual void UserExec(Option_t *option);
	29	virtual void Terminate(Option_t *);
	30
	31	// ------------- Cuts -----------------
	32
	33	void SetFilterBitCentralBarrel(Int_t filter) { fFilterBitCentralBarrel = filter; }
	34	void SetMaxEtaCentralBarrel(Double_t eta) { fMaxEtaCentralBarrel = eta; }
	35	void SetTriggerMatchLevelMuon(Short_t level) { fTriggerMatchLevelMuon = level; }
	36	// void SetMaxChi2Muon(Double_t chi2Max) { fMaxChi2Muon = chi2Max; }
	37	void SetRAbsRangeMuon (Double_t rAbsMin,Double_t rAbsMax) { fMinRAbsMuon = rAbsMin; fMaxRAbsMuon = rAbsMax; }
	38
07c2615a	39	// ------------- Analysis -------------
	40
	41	Float_t GetV0Multiplicity();
	42	Double_t GetITSMultiplicity();
	43	Bool_t IsTriggerFired();
6de71f1d	44	TObjArray* GetAcceptedTracksCentralBarrel(AliAODEvent *aodEvent);
57a9cc15	45	TObjArray* GetAcceptedTracksMuonArm(AliAODEvent *aodEvent, Int_t centBin);
07c2615a	46	void SetPtBinning(Int_t nBins, Double_t *limits);
07c2615a	47	void SetCentBinning(Int_t nBins, Double_t *limits);
07c2615a	48	void SetCentMethod(const Char_t *method) { fCentMethod = method; }
	49	void FillHistograms(Int_t centrality, Int_t option);
	50	Int_t GetCentBin();
07c2615a	51
	52	private:
	53
	54	static const Int_t fNMaxBinsCentrality = 20;
	55	static const Int_t fNMaxBinsPt = 10;
	56
6de71f1d	57	AliAODEvent *fAOD; //!
	58	AliEventPoolManager *fPoolMgr; //! event pool manager
	59	AliAODTrack *fTrackCB; //!
	60	AliAODTrack *fTrackMA; //!
07c2615a	61
	62	Int_t fFilterBitCentralBarrel;
	63	Double_t fMaxEtaCentralBarrel;
	64
	65	Double_t fMaxChi2Muon, fMinRAbsMuon, fMaxRAbsMuon;
	66	Short_t fTriggerMatchLevelMuon;
	67
07c2615a	68	Int_t fNbinsCent, fNbinsPt;
07c2615a	69
6de71f1d	70	TAxis *fCentAxis;
6de71f1d	71	TAxis *fPtAxis;
07c2615a	72
6de71f1d	73	TH1D *fHistDeltaPhi[fNMaxBinsCentrality][fNMaxBinsPt][fNMaxBinsPt]; //!
	74	TH1D *fHistDeltaPhiMix[fNMaxBinsCentrality][fNMaxBinsPt][fNMaxBinsPt]; //!
	75	TH2D *fHistNTracksCB_vs_NTracksMA[fNMaxBinsCentrality]; //!
57a9cc15	76	TH2D *fHistNTracksCB_vs_NTracksMAmixed[fNMaxBinsCentrality]; //!
07c2615a	77
6de71f1d	78	TH1D *fHistV0Multiplicity; //!
	79	TH1D *fHistITSMultiplicity; //!
	80	TH1D *fHistCentrality; //!
07c2615a	81
57a9cc15	82	TH2D *fHistSingleMuonsEtaVsPt[fNMaxBinsCentrality]; //!
	83	TH2D *fHistSingleMuonsEtaVsRAbs[fNMaxBinsCentrality]; //!
	84
07c2615a	85	TString fCentMethod;
07c2615a	86
6de71f1d	87	TList *fOutputList; //!
07c2615a	88
	89	AliAnalysisTaskMuonHadronCorrelations(const AliAnalysisTaskMuonHadronCorrelations&);//not implimented
	90	AliAnalysisTaskMuonHadronCorrelations& operator=(const AliAnalysisTaskMuonHadronCorrelations&);//not implimnted
	91
	92	ClassDef(AliAnalysisTaskMuonHadronCorrelations, 1) // example of analysis
	93
	94	};
	95
	96	//====================================================================================================================================================
	97
	98	#endif