/cvmfs/atlas.cern.ch/repo/sw/ASG/AnalysisBase/2.4.28/ElectronEfficiencyCorrection/ElectronEfficiencyCorrection/ElectronChargeEfficiencyCorrectionTool.h

#ifndef ELECTRONCHARGECORRECTION__ELECTRONCHARGECORRECTIONTOOL__H
#define ELECTRONCHARGECORRECTION__ELECTRONCHARGECORRECTIONTOOL__H

#include "AsgTools/AsgTool.h"
//#include "ElectronChargeEfficiencyCorrectionTool/IElectronChargeEfficiencyCorrectionTool.h"
#include "AsgAnalysisInterfaces/IEfficiencyScaleFactorTool.h"
#include "PATInterfaces/ISystematicsTool.h"
#include "TH2.h"
// #include "xAODTruth/TruthParticle.h"
#include "AthContainers/AuxElement.h"
#include "xAODEgamma/ElectronFwd.h"

#include <string>
#include <vector>
#include <map>

// Forward declaration(s):
namespace xAOD {
  class IParticle;
}
//class TH2D;


namespace CP {

  class ElectronChargeEfficiencyCorrectionTool
    : virtual public CP::IEfficiencyScaleFactorTool,
    public asg::AsgTool
  {
    ASG_TOOL_CLASS2(ElectronChargeEfficiencyCorrectionTool, CP::IEfficiencyScaleFactorTool, CP:: ISystematicsTool)

  public:

    ElectronChargeEfficiencyCorrectionTool(const std::string name);


    virtual ~ElectronChargeEfficiencyCorrectionTool();

    // METADATA  // TO DO
    //    virtual StatusCode beginInputFile();
    //    virtual StatusCode beginEvent();
    // virtual StatusCode endInputFile();



  public:
    virtual StatusCode initialize();

    virtual StatusCode finalize();

    virtual CP::CorrectionCode getEfficiencyScaleFactor(const xAOD::IParticle& part, double& sf) const;

    virtual CP::CorrectionCode applyEfficiencyScaleFactor(const xAOD::IParticle& part) const;

    //  void applySysVariation();

    virtual bool  isAffectedBySystematic( const SystematicVariation& systematic ) const;

    virtual SystematicSet affectingSystematics() const;

    virtual CP::SystematicSet  recommendedSystematics() const;

    virtual CP::SystematicCode applySystematicVariation( const SystematicSet& systConfig );

    CP::SystematicCode registerSystematics();

  private:

    float getChargeFlipRate( double eta, double pt, TH2D *hrates, double& flipRate) const;

    CP::CorrectionCode getEleTruthCharge( const xAOD::Electron& ele, int& truthcharge) const;

    CP::CorrectionCode isGoodEle( const xAOD::Electron& ele, bool& goodele) const;

    int m_dataTypeOverwrite;
    //
    //
    std::string m_eventInfoCollectionName;

    std::map<std::string, std::vector<TH2D*> > m_SF_SS;     // keys (e.g. RunNumber223333_319200_Nvtx0_10_Phi1.5_1.6) mapping to vector of SF histograms --> vector m_SF: 0=nominal, 1=stat, 2,3,4...n=syst
    std::map<std::string, std::vector<TH2D*> > m_SF_OS;     // keys (e.g. RunNumber223333_319200_Nvtx0_10_Phi1.5_1.6) mapping to vector of SF histograms --> vector m_SF: 0=nominal, 1=stat, 2,3,4...n=syst

    //cuts // further variables to bin in
    std::vector<unsigned int> m_RunNumbers;
    bool m_useRandomRunNumber;
    unsigned int m_defaultRandomRunNumber;

    std::string m_filename;

    std::string m_workingPoint;

    double m_eta_lowlimit;

    double m_eta_uplimit;

    double m_pt_lowlimit;

    double m_pt_uplimit;

    float m_gevmev;


    mutable int m_truthCharge;

    //const xAOD::TruthParticle *m_truthparticle;

    //    TRandom3 *m_Rndm;

    // Systematics
    std::vector<std::string> m_systematics;

    std::map<CP::SystematicSet, CP::SystematicSet> m_filtered_sys_sets;
    //boost::unordered_map<SystematicSet,EffiCollection*> m_sf_sets;
    CP::SystematicSet m_mySysConf;
    CP::SystematicSet m_affectingSys;

    std::string m_sf_decoration_name;
    SG::AuxElement::Decorator< float >* m_sfDec;


  };

} // End namespace CP

#endif

//......virtual CP::CorrectionCode applyCorrection( xAOD::Electron& ele );

//...virtual CP::CorrectionCode correctedCopy( const xAOD::Electron& input,
//...                         xAOD::Electron*& output );

---