// Efficiency scale factors macro
// Date: 11/01/2010
// Author: Olivier Arnaez <olivier.arnaez@cern.ch>
//         Jan Kretzschmar <jan.kretzschmar@cern.ch>
//
// Usage: 
// std::pair<float,float> sf_error = objsf->scaleFactor(eta(cluster), ET(MeV), set, range, rel, etcorrection)
//
// Please note: alternative accessors scaleFactorLoose, scaleFactorMedium, ... 
//              have been disabled, as the amount of sets is expanding and this was not maintainable
//              Read the "set=X" documentation below and use the correct arguments for scaleFactor() function
//
// The first number (sf_error.first) returns the efficiency scale factor,
// the second number is its uncertainty (sf_errror.second)
// 
// The combined (W/Z/jpsi) scale factor and uncertainty vs eta and ET (in MeV) are given
// 
// different "sets" of numbers available (not all with all sets):
//    * Loose SF (set=0)
//    * Medium SF (set=1)
//    * Tight SF (set=2)
//    * e20_medium trigger SF (set=3) (use set 8 or 10 for release 17 2011 data/MC11a)
//    * reco+trkqual SF (set=4)
//    * Loose++ SF (set=5)
//    * Medium++ SF (set=6)
//    * Tight++ SF (set=7)
//    * e20_medium trigger SF w.r.t Medium++ offline (set=8)
//    * e20_medium MC efficiency w.r.t Medium++ offline (set=9)
//    * e20_medium trigger SF w.r.t Tight++ offline (set=10)
//    * e20_medium MC efficiency w.r.t Tight++ offline (set=11)
//    * e22_medium trigger SF w.r.t Medium++ offline (set=12)
//    * e22_medium MC efficiency w.r.t Medium++ offline (set=13)
//    * e22_medium trigger SF w.r.t Tight++ offline (set=14)
//    * e22_medium MC efficiency w.r.t Tight++ offline (set=15)
//    * e22vh_medium1 trigger SF (using e22_medium1 on MC11a) w.r.t Medium++ offline (set=16)
//    * e22_medium1 MC efficiency w.r.t Medium++ offline (set=17)
//    * e22vh_medium1 trigger SF (using e22_medium1 on MC11a) w.r.t Tight++ offline (set=18)
//    * e22_medium1 MC efficiency w.r.t Tight++ offline (set=19)
//    * e20_medium MC efficiency w.r.t Loose++ offline (set=20)
//    * e22_medium1 MC efficiency w.r.t Loose++ offline (set=21)
//    * e22vh_medium1 MC efficiency w.r.t Loose++ offline (set=22)
// data and MC release selection:
//    * release 15 2010 data/MC09 (rel=0)
//    * release 16 2010 data/MC10 (rel=1)
//    * release 16.6 2011 data/MC10ab (estimated from 2010 data) (rel=2)
//    * release 16.6 estimated from 2011 data "EPS recommendations" and MC10b (rel=3)
//    * release 16.6 estimated from 2011 data "EPS recommendations" including Jpsi measurements (rel=4)
//    * release 17 estimated from 2011 data/MC11a "CERN council recommendations" (rel=5)
// measured with probes in the ranges:
//    * 20-50 GeV range (range=0)
//    * 30-50 GeV (range=1)
// and correcting (etcorrection=1) or not (etcorrection=0) for the ET-dependence
//
// Eta binning is changing from release to release
//
// Note that for rel>=4 range should be left at 0 and etcorrection=1 always
// 
// For now separete function for Forward Electrons (|eta|>=2.5)
// and ONLY for release 16.6, 2011 data in analogy to function for central electrons
// std::pair<float,float> sf_error = objsf->scaleFactorForward(eta, set)
// where eta = electron eta = cluster eta (as no track)
// cuts are ForwardLoose (set=0) or ForwardTight (set=2)
// 

#ifndef egammaSFclass_h
#define egammaSFclass_h

#include <iostream>
#include <string>
#include <map>
#include <vector>
//#include <pair>

using namespace std;

class egammaSFclass {

public:

  egammaSFclass();
  ~egammaSFclass() {};
  
  std::pair<float,float> etCorrection(float ET, int set, int rel=5);
  std::pair<float,float> scaleFactor(float eta, float ET, int set, int range=0, int rel=5, bool etcorrection=true);

/*   std::pair<float,float> scaleFactorLoose(float eta, float ET=20000., int range=0, int rel=5, bool etcorrection=false) { return scaleFactor(eta, ET, 0, range, rel, etcorrection); }; */
/*   std::pair<float,float> scaleFactorMedium(float eta, float ET=20000., int range=0, int rel=5, bool etcorrection=false) { return scaleFactor(eta, ET, 1, range, rel, etcorrection); }; */
/*   std::pair<float,float> scaleFactorTight(float eta, float ET=20000., int range=0, int rel=5, bool etcorrection=false) { return scaleFactor(eta, ET, 2, range, rel, etcorrection); }; */

/*   std::pair<float,float> scaleFactorTrigger(float eta, float ET=20000., int range=0, int rel=5, bool etcorrection=false) { return scaleFactor(eta, ET, 3, range, rel, etcorrection); }; */
/*   std::pair<float,float> scaleFactorRecoTrkQual(float eta, float ET, int range=0, int rel=5, bool etcorrection=false) { return scaleFactor(eta, ET, 4, range, rel, etcorrection); }; */

/*   std::pair<float,float> scaleFactorMediumETcorrected(float eta, float ET, int rel=5) { return scaleFactorMedium(eta, ET, 0, rel, true); }; */
/*   std::pair<float,float> scaleFactorTightETcorrected(float eta, float ET, int rel=5) { return scaleFactorTight(eta, ET, 0, rel, true); }; */

  // for now separate "forward electron classes"
  std::pair<float,float> scaleFactorForward(float eta, int set);
  std::pair<float,float> scaleFactorForwardLoose(float eta)
    { return scaleFactorForward(eta, 0); };
  std::pair<float,float> scaleFactorForwardTight(float eta)
    { return scaleFactorForward(eta, 2); };

  //For the binning
  std::vector<float> m_Etabins;
  std::vector<float> m_FineEtabins;
  std::vector<float> m_11Etabins;

  std::vector<float> m_ETbins;
  std::vector<float> m_ETbinsFullRange;
  std::vector<float> m_ETbinsTrigger;

  //For the scale factors of the standard egamma cuts 
  //Release 15
  //Probes between 30 and 50 GeV (plateau region)
  std::vector<float> efficienciesRel15Loose3050;
  std::vector<float> uncertaintiesRel15Loose3050;
  std::vector<float> efficienciesRel15Medium3050;
  std::vector<float> uncertaintiesRel15Medium3050;
  std::vector<float> efficienciesRel15Tight3050;
  std::vector<float> uncertaintiesRel15Tight3050;
  //Probes between 20 and 50 GeV
  std::vector<float> efficienciesRel15Loose2050;
  std::vector<float> uncertaintiesRel15Loose2050;
  std::vector<float> efficienciesRel15Medium2050;
  std::vector<float> uncertaintiesRel15Medium2050;
  std::vector<float> efficienciesRel15Tight2050;
  std::vector<float> uncertaintiesRel15Tight2050;

  //Release 16
  //Probes between 30 and 50 GeV (plateau region)
  std::vector<float> efficienciesRel16Medium3050;
  std::vector<float> uncertaintiesRel16Medium3050;
  std::vector<float> efficienciesRel16Tight3050;
  std::vector<float> uncertaintiesRel16Tight3050;
  //Probes between 20 and 50 GeV
  std::vector<float> efficienciesRel16Medium2050;
  std::vector<float> uncertaintiesRel16Medium2050;
  std::vector<float> efficienciesRel16Tight2050;
  std::vector<float> uncertaintiesRel16Tight2050;

  //Release 16.6 with 2010 data
  //Probes between 30 and 50 GeV (plateau region)
  std::vector<float> efficienciesRel166Data2010Medium3050;
  std::vector<float> uncertaintiesRel166Data2010Medium3050;
  std::vector<float> efficienciesRel166Data2010Tight3050;
  std::vector<float> uncertaintiesRel166Data2010Tight3050;
  //Probes between 20 and 50 GeV
  std::vector<float> efficienciesRel166Data2010Medium2050;
  std::vector<float> uncertaintiesRel166Data2010Medium2050;
  std::vector<float> efficienciesRel166Data2010Tight2050;
  std::vector<float> uncertaintiesRel166Data2010Tight2050;

  //Release 16.6, EPS recommendations
  //Identification for probes between 20 and 50 GeV
  std::vector<float> efficienciesRel166EPSMedium2050;
  std::vector<float> uncertaintiesRel166EPSMedium2050;
  std::vector<float> efficienciesRel166EPSTight2050;
  std::vector<float> uncertaintiesRel166EPSTight2050;
  //Identification for low ET probes
  std::vector<float> efficienciesRel166EPSMediumLowET;
  std::vector<float> uncertaintiesRel166EPSMediumLowET;
  std::vector<float> efficienciesRel166EPSTightLowET;
  std::vector<float> uncertaintiesRel166EPSTightLowET;
  //For trigger efficiencies on the plateau
  std::vector<float> efficienciesRel166EPSTrigger;
  std::vector<float> uncertaintiesRel166EPSTrigger;
  //For reco+trkquality efficiencies
  std::vector<float> efficienciesRel166EPSRecoTrkQual;
  std::vector<float> uncertaintiesRel166EPSRecoTrkQual;

  //For the ET-corrections of the scale factors
  //Release 16
  //Medium
  std::vector<float> ETCorrectionsMediumRel16;
  std::vector<float> uncertaintiesETCorrectionsMediumRel16;
  //Tight
  std::vector<float> ETCorrectionsTightRel16;
  std::vector<float> uncertaintiesETCorrectionsTightRel16;
  //Release 16.6 with 2010 data
  //Medium
  std::vector<float> ETCorrectionsMediumRel166Data2010;
  std::vector<float> uncertaintiesETCorrectionsMediumRel166Data2010;
  //Tight
  std::vector<float> ETCorrectionsTightRel166Data2010;
  std::vector<float> uncertaintiesETCorrectionsTightRel166Data2010;
  //Release 16.6, EPS recommendations
  //Medium
  std::vector<float> ETCorrectionsMediumRel166EPS;
  std::vector<float> uncertaintiesETCorrectionsMediumRel166EPS;
  //Tight
  std::vector<float> ETCorrectionsTightRel166EPS;
  std::vector<float> uncertaintiesETCorrectionsTightRel166EPS;
  //Release 16.6, EPS recommendations including low ET electrons
  //Medium
  std::vector<float> ETCorrectionsMediumRel166EPSFullRange;
  std::vector<float> uncertaintiesETCorrectionsMediumRel166EPSFullRange;
  //Tight
  std::vector<float> ETCorrectionsTightRel166EPSFullRange;
  std::vector<float> uncertaintiesETCorrectionsTightRel166EPSFullRange;


  // Release 17, "CERN Council" recommendations
  // converter
  void copyToVector(const float *myarray, int n, std::vector<float> &dest, double renorm = 100.);

  // For reco+trkquality efficiencies
  std::vector<float> efficienciesRel17CCRecoTrkQual;
  std::vector<float> uncertaintiesRel17CCRecoTrkQual;

  // Identification eta for probes between 15 and 50 GeV
  std::vector<float> efficienciesRel17CCLoosePP1550;
  std::vector<float> uncertaintiesRel17CCLoosePP1550;
  std::vector<float> efficienciesRel17CCMediumPP1550;
  std::vector<float> uncertaintiesRel17CCMediumPP1550;
  std::vector<float> efficienciesRel17CCTightPP1550;
  std::vector<float> uncertaintiesRel17CCTightPP1550;
  //Identification eta for low ET probes
  std::vector<float> efficienciesRel17CCLoosePP415;
  std::vector<float> uncertaintiesRel17CCLoosePP415;
  std::vector<float> efficienciesRel17CCMediumPP415;
  std::vector<float> uncertaintiesRel17CCMediumPP415;
  std::vector<float> efficienciesRel17CCTightPP415;
  std::vector<float> uncertaintiesRel17CCTightPP415;
  // ET correction
  std::vector<float> ETCorrectionsRel17CCLoosePP;
  std::vector<float> uncertaintiesETCorrectionsRel17CCLoosePP;
  std::vector<float> ETCorrectionsRel17CCMediumPP;
  std::vector<float> uncertaintiesETCorrectionsRel17CCMediumPP;
  std::vector<float> ETCorrectionsRel17CCTightPP;
  std::vector<float> uncertaintiesETCorrectionsRel17CCTightPP;

  // Trigger efficiencies
  // e20_medium B-J
  std::vector<float> MCefficienciesRel17CCe20_mediumLoosePP;
  std::vector<float> MCefficienciesRel17CCe20_mediumLoosePPET;

  std::vector<float> efficienciesRel17CCe20_mediumMediumPP;
  std::vector<float> uncertaintiesRel17CCe20_mediumMediumPP;
  std::vector<float> efficienciesRel17CCe20_mediumMediumPPET;
  std::vector<float> uncertaintiesRel17CCe20_mediumMediumPPET;

  std::vector<float> MCefficienciesRel17CCe20_mediumMediumPP;
  std::vector<float> MCefficienciesRel17CCe20_mediumMediumPPET;

  std::vector<float> efficienciesRel17CCe20_mediumTightPP;
  std::vector<float> uncertaintiesRel17CCe20_mediumTightPP;
  std::vector<float> efficienciesRel17CCe20_mediumTightPPET;
  std::vector<float> uncertaintiesRel17CCe20_mediumTightPPET;

  std::vector<float> MCefficienciesRel17CCe20_mediumTightPP;
  std::vector<float> MCefficienciesRel17CCe20_mediumTightPPET;


  // e22_medium K

  std::vector<float> MCefficienciesRel17CCe22_mediumLoosePP;
  std::vector<float> MCefficienciesRel17CCe22_mediumLoosePPET;

  std::vector<float> efficienciesRel17CCe22_mediumMediumPP;
  std::vector<float> uncertaintiesRel17CCe22_mediumMediumPP;
  std::vector<float> efficienciesRel17CCe22_mediumMediumPPET;
  std::vector<float> uncertaintiesRel17CCe22_mediumMediumPPET;

  std::vector<float> MCefficienciesRel17CCe22_mediumMediumPP;
  std::vector<float> MCefficienciesRel17CCe22_mediumMediumPPET;

  std::vector<float> efficienciesRel17CCe22_mediumTightPP;
  std::vector<float> uncertaintiesRel17CCe22_mediumTightPP;
  std::vector<float> efficienciesRel17CCe22_mediumTightPPET;
  std::vector<float> uncertaintiesRel17CCe22_mediumTightPPET;

  std::vector<float> MCefficienciesRel17CCe22_mediumTightPP;
  std::vector<float> MCefficienciesRel17CCe22_mediumTightPPET;


  // e22vh_medium1 L-M
  std::vector<float> MCefficienciesRel17CCe22vh_medium1LoosePP;
  std::vector<float> MCefficienciesRel17CCe22vh_medium1LoosePPET;

  std::vector<float> efficienciesRel17CCe22vh_medium1MediumPP;
  std::vector<float> uncertaintiesRel17CCe22vh_medium1MediumPP;
  std::vector<float> efficienciesRel17CCe22vh_medium1MediumPPET;
  std::vector<float> uncertaintiesRel17CCe22vh_medium1MediumPPET;

  std::vector<float> MCefficienciesRel17CCe22vh_medium1MediumPP;
  std::vector<float> MCefficienciesRel17CCe22vh_medium1MediumPPET;

  std::vector<float> efficienciesRel17CCe22vh_medium1TightPP;
  std::vector<float> uncertaintiesRel17CCe22vh_medium1TightPP;
  std::vector<float> efficienciesRel17CCe22vh_medium1TightPPET;
  std::vector<float> uncertaintiesRel17CCe22vh_medium1TightPPET;

  std::vector<float> MCefficienciesRel17CCe22vh_medium1TightPP;
  std::vector<float> MCefficienciesRel17CCe22vh_medium1TightPPET;


};

#endif