//_________________________________________________________________________
//  Utility Class for transverse energy studies
//  Selection class for EMCAL
//
//*-- Authors: Oystein Djuvsland (Bergen)
//_________________________________________________________________________


#include "AliAnalysisEtRecEffCorrection.h"

ClassImp(AliAnalysisEtRecEffCorrection);

AliAnalysisEtRecEffCorrection::AliAnalysisEtRecEffCorrection() : TNamed("RecEff","RecEff")
							       ,fEnergyCorrection(new TF1())//"ReCorr","1",0.01)
									 ,fRecoEff(0)
    ,fMaxEnergy(0)
{
  fEnergyCorrection->SetName("ReCorr");
}

AliAnalysisEtRecEffCorrection::AliAnalysisEtRecEffCorrection(TString name, const TF1 &correction,const TH2F &recoEff, Double_t maxEnergy) : TNamed(name, name)
    ,fEnergyCorrection(new TF1(correction))										    	,fRecoEff(new TH2F(recoEff))
    ,fMaxEnergy(maxEnergy)
{}

//! Copy constructor
AliAnalysisEtRecEffCorrection::AliAnalysisEtRecEffCorrection(const AliAnalysisEtRecEffCorrection &obj) : TNamed(obj)
    ,fEnergyCorrection(new TF1(*(obj.fEnergyCorrection)))
    ,fRecoEff(new TH2F(*(obj.fRecoEff)))
    ,fMaxEnergy(obj.fMaxEnergy)
{}

//! Destructor
AliAnalysisEtRecEffCorrection::~AliAnalysisEtRecEffCorrection()
{
}

//! Assignment operator
AliAnalysisEtRecEffCorrection& AliAnalysisEtRecEffCorrection::operator=(const AliAnalysisEtRecEffCorrection &other)
{
    if (this != &other)
    {
        fEnergyCorrection = other.fEnergyCorrection;
	fRecoEff = other.fRecoEff;
        fMaxEnergy = other.fMaxEnergy;
    }
    return *this;
}

//! Equality operator
bool AliAnalysisEtRecEffCorrection::operator==(const AliAnalysisEtRecEffCorrection &other) const
{
    if (this == &other) return true;
    return false;
    //return (fMaxEnergy == other.fMaxEnergy && fEnergyCorrection == other.fEnergyCorrection);
}

Double_t AliAnalysisEtRecEffCorrection::CorrectedEnergy(Double_t energy)
{
  return fEnergyCorrection->Eval(energy);
  
}
Double_t AliAnalysisEtRecEffCorrection::CorrectedEnergy(Double_t energy, int cent)
{
  if(fRecoEff){
    Double_t eff = ReconstructionEfficiency(energy, cent);
    if(eff>1e-5) return 1.0/eff*energy;
  }
  //cerr<<"Warning:  I should not be here!"<<endl;
  return 1.0;
  
}

Double_t AliAnalysisEtRecEffCorrection::ReconstructionEfficiency(float energy, int cent) const {
  Double_t eff = 1.0;
  if(fRecoEff) eff =  fRecoEff->GetBinContent(fRecoEff->GetXaxis()->FindBin(energy),fRecoEff->GetYaxis()->FindBin(cent));
  //cout <<"eff "<<eff<<" bin energy "<<fRecoEff->GetXaxis()->FindBin(energy)<<" bin centrality "<<fRecoEff->GetYaxis()->FindBin(cent)<<endl;
  return eff;
}