#ifndef ALIANALYSISTASKSEITSSASPECTRA_H
#define ALIANALYSISTASKSEITSSASPECTRA_H

///////////////////////////////////////////////////////////////////////////
// AliAnalysisTaskSE for the extraction of the various histograms to
// study the pt spectra of identified hadrons:
// - log(dEdx)-log(dEdxBB) distributions for pions, kaons and protons in pt bins
// - Pt distributions of pions, kaons and protons with nSigma PID
// Authors: 
// E. Biolcati, biolcati@to.infn.it
// L. Milano, milano@to.infn.it
// F. Prino, prino@to.infn.it
///////////////////////////////////////////////////////////////////////////

/* $Id$ */

class TString;
class TTree;
class TH1F;
class TF1;
class TH2F;
class TRandom3;
class AliESDEvent;
class TNtuple;
class AliITSPIDResponse;

#include "AliAnalysisTaskSE.h"

class AliAnalysisTaskSEITSsaSpectra : public AliAnalysisTaskSE {
 public:
  AliAnalysisTaskSEITSsaSpectra();
  virtual ~AliAnalysisTaskSEITSsaSpectra();

  virtual void   UserCreateOutputObjects();
  virtual void Init();
  virtual void LocalInit() {Init();}
  virtual void   UserExec(Option_t *);
  virtual void   Terminate(Option_t *);

  void SetMaxChi2Clu(Double_t chi=2.5){
    fMaxChi2Clu=chi;
  }
  void SetRapidityRange(Double_t dy=0.5){
    fMaxY=dy;
  }
  void SetMinNSigma(Double_t ns=1.5){
    fMinNSigma=ns;
  }
  void SetMindEdx(Double_t mind=0.){
    fMindEdx=mind;
  }
  void SetMinSPDPoints(Int_t np=1){
    fMinSPDPts=np;
  }
  void SetMinNdEdxSamples(Int_t np=3){
    fMinNdEdxSamples=np;
  }
  void SetDCACuts(Double_t nsxy=7., Double_t nsz=7.){
    fNSigmaDCAxy=nsxy;
    fNSigmaDCAz=nsz; 
  }
  void SetMultBin(Int_t LowBin=-1,Int_t UpBin=-1){
    fLowMult=LowBin;
    fUpMult=UpBin;
  }
  void SetCentralityCut(Float_t low, Float_t up){
    if((up>low)&&(!(low<0.0))&&(!(up>100.0))){
      fLowCentrality=low; fUpCentrality=up;
    }
  }
 void SetHImode(){fHImode=kTRUE;}
 void SetUseTrackMultiplicityEstimator(){fMultEstimator=0;}
 void SetUseTrackletsMultiplicityEstimator(){fMultEstimator=1;}
 void SetUseClustersSPD1MultiplicityEstimator(){fMultEstimator=2;}
 
  
  void SetEtaMax(Double_t maxeta){
    fEtaRange=maxeta;
  }
  
  void SetReadMC(Bool_t flag = kTRUE) {fMC = flag;}
  void SetFillNtuple(Bool_t fill=kTRUE) {fFillNtuple=fill;}
  void SetLowEnergypp(Bool_t opt=kTRUE) {fLowEnergypp=opt;}

  void SetSmearMC(Double_t smearp, Double_t smeardedx){
    fSmearMC=kTRUE;
    fSmearP=smearp;
    fSmeardEdx=smeardedx;
  }
  Double_t CookdEdx(Double_t *s) const; 
  Double_t Eta2y(Double_t pt, Double_t m, Double_t eta) const;
  Bool_t DCAcut(Double_t impactXY, Double_t impactZ, Double_t pt) const;

 private:
  AliAnalysisTaskSEITSsaSpectra(const AliAnalysisTaskSEITSsaSpectra &source); 
  AliAnalysisTaskSEITSsaSpectra& operator=(const AliAnalysisTaskSEITSsaSpectra &source);

  enum {kNbins=22};
  
  AliESDEvent *fESD; //ESD object
  
  TList *fOutput; //! tlist with output
  TList *fListCuts; // list of functions storing DCA cut 
  TH1F *fHistNEvents; //! histo with number of events
  TH1F *fHistMult; //! histo with multiplicity of the events
  TH1F *fHistCen; //! histo with multiplicity of the events
  TH1F *fHistNTracks; //! histo with number of tracks
  TH1F *fHistNTracksPos; //! histo with number of tracks
  TH1F *fHistNTracksNeg; //! histo with number of tracks
  
  TH2F *fHistDEDX; //! histo with dedx versus momentum
  TH2F *fHistDEDXdouble; //! histo with dedx versus signed momentum
  TH2F *fHistPosNSigmaSep[3]; //! histo nsigma separation vs momentum
  TH2F *fHistNegNSigmaSep[3]; //! histo nsigma separation vs momentum
  
  TH1F *fHistBeforeEvSel; //! histo with pt distribution before my event selection
  TH1F *fHistAfterEvSel; //! histo with pt distribution after my event selection
  
  TH1F *fHistPrimMCpos[3]; //! histo with spectra of primaries from the MC truth (positive)
  TH1F *fHistPrimMCneg[3]; //! histo with spectra of primaries from the MC truth (negative)
  TH1F *fHistSecStrMCpos[3]; //! histo with spectra of strange decays from the MC truth (positive)
  TH1F *fHistSecStrMCneg[3]; //! histo with spectra of strange decays from the MC truth (negative)
  TH1F *fHistSecMatMCpos[3]; //! histo with spectra of sec. from material from the MC truth (positive)
  TH1F *fHistSecMatMCneg[3]; //! histo with spectra of sec. from material from the MC truth (negative)
  TH1F *fHistPrimMCposBefEvSel[3]; //! histo with spectra of primaries from the MC truth (positive)
  TH1F *fHistPrimMCposBefEvSelEtaY[3]; //! histo with spectra of primaries from the MC truth with both cut on y and eta(positive)
  TH1F *fHistPrimMCposBefEvSelEta[3]; //! histo with spectra of primaries from the MC truth with cut on eta only(positive)
  TH1F *fHistPrimMCnegBefEvSel[3]; //! histo with spectra of primaries from the MC truth (negative)
  TH1F *fHistPrimMCnegBefEvSelEtaY[3]; //! histo with spectra of primaries from the MC truth with both cut on y and eta(negative)
  TH1F *fHistPrimMCnegBefEvSelEta[3]; //! histo with spectra of primaries from the MC truth with cut on eta only(negative)
  TH1F *fHistSecStrMCposBefEvSel[3]; //! histo with spectra of strange decays from the MC truth (positive)
  TH1F *fHistSecStrMCnegBefEvSel[3]; //! histo with spectra of strange decays from the MC truth (negative)
  TH1F *fHistSecMatMCposBefEvSel[3]; //! histo with spectra of sec. from material from the MC truth (positive)
  TH1F *fHistSecMatMCnegBefEvSel[3]; //! histo with spectra of sec. from material from the MC truth (negative)
  TH1F *fHistPrimMCposReco[3]; //! histo with spectra of primaries from the MC truth (positive)
  TH1F *fHistPrimMCposRecoEtaY[3]; //! histo with spectra of primaries from the MC truth with both cut on y and eta(positive)
  TH1F *fHistPrimMCnegReco[3]; //! histo with spectra of primaries from the MC truth (negative)
  TH1F *fHistPrimMCnegRecoEtaY[3]; //! histo with spectra of primaries from the MC truth with both cut on y and eta(negative)
  TH1F *fHistSecStrMCposReco[3]; //! histo with spectra of strange decays from the MC truth (positive)
  TH1F *fHistSecStrMCnegReco[3]; //! histo with spectra of strange decays from the MC truth (negative)
  TH1F *fHistSecMatMCposReco[3]; //! histo with spectra of sec. from material from the MC truth (positive)
  TH1F *fHistSecMatMCnegReco[3]; //! histo with spectra of sec. from material from the MC truth (negative)
  
  TH1F *fHistCharge[4]; //! histo with charge distribution to check the calibration 
  
  TH1F *fHistPosPi[kNbins]; //! histo with dedx distibution in the pions hypotesis (positive)
  TH1F *fHistPosK[kNbins]; //! histo with dedx distibution in the kaons hypotesis (positive)
  TH1F *fHistPosP[kNbins]; //! histo with dedx distibution in the protons hypotesis (positive)
  TH1F *fHistNegPi[kNbins]; //! histo with dedx distibution in the pions hypotesis (negative)
  TH1F *fHistNegK[kNbins]; //! histo with dedx distibution in the kaons hypotesis (negative)
  TH1F *fHistNegP[kNbins]; //! histo with dedx distibution in the protons hypotesis (negative)
  
  TH1F *fHistDCAPosPi[kNbins]; //! histo with DCA distibution in the kaons hypotesis (positive)
  TH1F *fHistDCAPosK[kNbins]; //! histo with DCA distibution in the kaons hypotesis (positive)
  TH1F *fHistDCAPosP[kNbins]; //! histo with DCA distibution in the protons hypotesis (positive)
  TH1F *fHistDCANegPi[kNbins]; //! histo with DCA distibution in the pions hypotesis (negative)
  TH1F *fHistDCANegK[kNbins]; //! histo with DCA distibution in the kaons hypotesis (negative)
  TH1F *fHistDCANegP[kNbins]; //! histo with DCA distibution in the protons hypotesis (negative)
  
  TH1F *fHistMCPrimDCAPosPi[kNbins]; //! histo with DCA distibution, MC truth 
  TH1F *fHistMCPrimDCAPosK[kNbins]; //! histo with DCA distibution, MC truth 
  TH1F *fHistMCPrimDCAPosP[kNbins]; //! histo with DCA distibution, MC truth 
  TH1F *fHistMCPrimDCANegPi[kNbins]; //! histo with DCA distibution, MC truth 
  TH1F *fHistMCPrimDCANegK[kNbins]; //! histo with DCA distibution, MC truth 
  TH1F *fHistMCPrimDCANegP[kNbins]; //! histo with DCA distibution, MC truth
  
  TH1F *fHistMCSecStDCAPosPi[kNbins]; //! histo with DCA distibution, MC truth 
  TH1F *fHistMCSecStDCAPosK[kNbins]; //! histo with DCA distibution, MC truth 
  TH1F *fHistMCSecStDCAPosP[kNbins]; //! histo with DCA distibution, MC truth 
  TH1F *fHistMCSecStDCANegPi[kNbins]; //! histo with DCA distibution, MC truth 
  TH1F *fHistMCSecStDCANegK[kNbins]; //! histo with DCA distibution, MC truth 
  TH1F *fHistMCSecStDCANegP[kNbins]; //! histo with DCA distibution, MC truth
  
  TH1F *fHistMCSecMatDCAPosPi[kNbins]; //! histo with DCA distibution, MC truth 
  TH1F *fHistMCSecMatDCAPosK[kNbins]; //! histo with DCA distibution, MC truth 
  TH1F *fHistMCSecMatDCAPosP[kNbins]; //! histo with DCA distibution, MC truth 
  TH1F *fHistMCSecMatDCANegPi[kNbins]; //! histo with DCA distibution, MC truth 
  TH1F *fHistMCSecMatDCANegK[kNbins]; //! histo with DCA distibution, MC truth 
  TH1F *fHistMCSecMatDCANegP[kNbins]; //! histo with DCA distibution, MC truth
  
  TH1F *fHistMCPosOtherHypPion[kNbins]; //! histo with dedx using the MC truth 
  TH1F *fHistMCPosOtherHypKaon[kNbins]; //! histo with dedx using the MC truth
  TH1F *fHistMCPosOtherHypProton[kNbins]; //! histo with dedx using the MC truth
  TH1F *fHistMCPosElHypPion[kNbins]; //! histo with dedx using the MC truth 
  TH1F *fHistMCPosElHypKaon[kNbins]; //! histo with dedx using the MC truth
  TH1F *fHistMCPosElHypProton[kNbins]; //! histo with dedx using the MC truth
  TH1F *fHistMCPosPiHypPion[kNbins]; //! histo with dedx using the MC truth 
  TH1F *fHistMCPosPiHypKaon[kNbins]; //! histo with dedx using the MC truth
  TH1F *fHistMCPosPiHypProton[kNbins]; //! histo with dedx using the MC truth
  TH1F *fHistMCPosKHypPion[kNbins]; //! histo with dedx using the MC truth
  TH1F *fHistMCPosKHypKaon[kNbins]; //! histo with dedx using the MC truth
  TH1F *fHistMCPosKHypProton[kNbins]; //! histo with dedx using the MC truth
  TH1F *fHistMCPosPHypPion[kNbins]; //! histo with dedx using the MC truth
  TH1F *fHistMCPosPHypKaon[kNbins]; //! histo with dedx using the MC truth
  TH1F *fHistMCPosPHypProton[kNbins]; //! histo with dedx using the MC truth
  
  TH1F *fHistMCNegOtherHypPion[kNbins]; //! histo with dedx using the MC truth
  TH1F *fHistMCNegOtherHypKaon[kNbins]; //! histo with dedx using the MC truth
  TH1F *fHistMCNegOtherHypProton[kNbins]; //! histo with dedx using the MC truth
  TH1F *fHistMCNegElHypPion[kNbins]; //! histo with dedx using the MC truth
  TH1F *fHistMCNegElHypKaon[kNbins]; //! histo with dedx using the MC truth
  TH1F *fHistMCNegElHypProton[kNbins]; //! histo with dedx using the MC truth
  TH1F *fHistMCNegPiHypPion[kNbins]; //! histo with dedx using the MC truth
  TH1F *fHistMCNegPiHypKaon[kNbins]; //! histo with dedx using the MC truth
  TH1F *fHistMCNegPiHypProton[kNbins]; //! histo with dedx using the MC truth
  TH1F *fHistMCNegKHypPion[kNbins]; //! histo with dedx using the MC truth
  TH1F *fHistMCNegKHypKaon[kNbins]; //! histo with dedx using the MC truth
  TH1F *fHistMCNegKHypProton[kNbins]; //! histo with dedx using the MC truth
  TH1F *fHistMCNegPHypPion[kNbins]; //! histo with dedx using the MC truth
  TH1F *fHistMCNegPHypKaon[kNbins]; //! histo with dedx using the MC truth
  TH1F *fHistMCNegPHypProton[kNbins]; //! histo with dedx using the MC truth
  
  TH1F *fHistPosNSigmaMean[3];       //! NSigma histos for 6 species
  TH1F *fHistPosNSigmaMCMean[3];       //! NSigma histos for 6 species
  TH1F *fHistPosNSigmaPrimMean[3];   //! NSigma histos for 6 species
  TH1F *fHistPosNSigmaPrimMCMean[3]; //! NSigma histos for 6 species
  TH1F *fHistNegNSigmaMean[3];       //! NSigma histos for 6 species
  TH1F *fHistNegNSigmaMCMean[3];       //! NSigma histos for 6 species
  TH1F *fHistNegNSigmaPrimMean[3];   //! NSigma histos for 6 species
  TH1F *fHistNegNSigmaPrimMCMean[3]; //! NSigma histos for 6 species

  TH1F *fHistPosNSigma[3];       //! NSigma histos for 6 species
  TH1F *fHistPosNSigmaMC[3];       //! NSigma histos for 6 species
  TH1F *fHistPosNSigmaPrim[3];   //! NSigma histos for 6 species
  TH1F *fHistPosNSigmaPrimMC[3]; //! NSigma histos for 6 species
  TH1F *fHistNegNSigma[3];       //! NSigma histos for 6 species
  TH1F *fHistNegNSigmaMC[3];       //! NSigma histos for 6 species
  TH1F *fHistNegNSigmaPrim[3];   //! NSigma histos for 6 species
  TH1F *fHistNegNSigmaPrimMC[3]; //! NSigma histos for 6 species

  TF1* fDCAxyCutFunc;  // function with DCAz cut vs. pt
  TF1* fDCAzCutFunc;   // function with DCAxy cut vs. pt

  Double_t fPtBinLimits[kNbins+1]; // limits of Pt Bins
  AliITSPIDResponse* fITSPIDResponse; //! class with BB parameterizations
  Int_t    fMinSPDPts;       // minimum number of SPD Points
  Int_t    fMinNdEdxSamples; // minimum number of SDD+SSD points
  Double_t fMindEdx;         // minimum dE/dx value in a layer (to cut noise)
  Double_t fMinNSigma;       // minimum number of sigmas
  Double_t fMaxY;            // maximum rapidity
  Double_t fMaxChi2Clu;      // maximum cluster
  Double_t fNSigmaDCAxy;     // DCA cut in bend. plane
  Double_t fNSigmaDCAz;      // DCA cut along z
  Double_t fEtaRange;        // limits in pseudorap
  Int_t fLowMult;      // Multiplicity bin
  Int_t fUpMult;      // Multiplicity bin
  Float_t fLowCentrality;//low Centrality cut
  Float_t fUpCentrality;//up  Centrality cut
  Int_t fMultEstimator; // multiplicty estimator 
  Bool_t fHImode;//use spd2 as mulestimator 
  Int_t fYear;        // Year (2009, 2010)
  Bool_t   fMC;        //flag to switch on the MC analysis for the efficiency estimation
  Bool_t   fSmearMC;   // flag to apply extra smearing on MC 
  Double_t fSmearP;    // extra relative smearing on simulated momentum
  Double_t fSmeardEdx; // extra relative smearing on simulated dE/dx
  TRandom3* fRandGener;    // generator for smearing
  Bool_t       fFillNtuple;      // fill ntuple  
  Bool_t   fLowEnergypp; // flag for counrint ev. in p-p 2.76
  TNtuple     *fNtupleNSigma;//! output ntuple
  TNtuple     *fNtupleMC;//! output MC ntuple
  
  ClassDef(AliAnalysisTaskSEITSsaSpectra, 8);
};

#endif