#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 SetMinEtaCentralBarrel(Double_t eta) { fMinEtaCentralBarrel = 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 SetEtaBinning(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;
  static const Int_t fNMaxBinsEta = 10;

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

  Double_t fMaxChi2Muon, fMinRAbsMuon, fMaxRAbsMuon;
  Short_t fTriggerMatchLevelMuon;

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

  TH1D *fHistDeltaPhi[fNMaxBinsCentrality][fNMaxBinsPt][fNMaxBinsPt]; //!
  TH1D *fHistDeltaPhiMix[fNMaxBinsCentrality][fNMaxBinsPt][fNMaxBinsPt]; //!
  TH2D *fHistEtaDeltaPhi[fNMaxBinsCentrality][fNMaxBinsPt][fNMaxBinsPt]; //!
  TH2D *fHistEtaDeltaPhiMix[fNMaxBinsCentrality][fNMaxBinsPt][fNMaxBinsPt]; //!
  TH2D *fHistNTracksCB_vs_NTracksMA[fNMaxBinsCentrality]; //!
  TH2D *fHistNTracksCB_vs_NTracksMAmixed[fNMaxBinsCentrality]; //!
  TH2D *fHistTracksEtaMAvsEtaCB[fNMaxBinsCentrality]; //!
  TH2D *fHistTracksEtaMAvsEtaCBmixed[fNMaxBinsCentrality]; //!
  TH1D *fHistSingleMuonsPt[fNMaxBinsCentrality]; //!
  TH1D *fHistSingleMuonsPtmixed[fNMaxBinsCentrality]; //!
  TH2D *fHistSingleMuonsEtaPt[fNMaxBinsCentrality]; //!
  TH2D *fHistSingleMuonsEtaPtmixed[fNMaxBinsCentrality]; //!

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

  TH1D *fHistSingleMuonsTrigMatch[fNMaxBinsCentrality]; //!
  TH1D *fHistSingleMuonsChi2[fNMaxBinsCentrality]; //!
  TH2D *fHistSingleMuonsEtaVsPt[fNMaxBinsCentrality]; //!
  TH2D *fHistSingleMuonsEtaVsRAbs[fNMaxBinsCentrality]; //!

  TString fCentMethod;

  TList *fOutputList; //!

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

};

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

#endif