/cvmfs/atlas.cern.ch/repo/sw/ASG/AnalysisBase/2.4.31/QuickAna/QuickAna/ElectronTool.h

#ifndef QUICK_ANA__ELECTRON_TOOL_H
#define QUICK_ANA__ELECTRON_TOOL_H

// This module still needs to be documented.  The interface provided
// in this module is intended for experts only.  The module is
// considered to be in the pre-alpha stage.

#include <QuickAna/Global.h>

#include <QuickAna/AnaToolCorrect.h>
#include <AsgTools/AnaToolHandle.h>
#include <QuickAna/AnaToolSelect.h>
#include <QuickAna/AnaToolWeight.h>
#include <QuickAna/SelectionCut.h>
#include <QuickAna/xAODInclude.h>

class IAsgElectronIsEMSelector;
class IAsgElectronLikelihoodTool;
namespace CP{
  class IEgammaCalibrationAndSmearingTool;
  class IIsolationSelectionTool;
  class IIsolationCorrectionTool;
}
class IAsgElectronEfficiencyCorrectionTool;

namespace ana
{
  class ElectronToolPreCorrect : virtual public AnaToolCorrect<xAOD::ElectronContainer>
  {

    ASG_TOOL_CLASS (ElectronToolPreCorrect, ana::IAnaTool)

  public:
    // Public interface methods

    ElectronToolPreCorrect (const std::string& name);

  public:
    virtual AnalysisStep step () const override;

    StatusCode initialize() override;

    virtual StatusCode
    correctObject (xAOD::Electron& electron) override;

  private:
    // CP tools

    asg::AnaToolHandle<IAsgElectronLikelihoodTool> m_likelihoodTool_loose;
    asg::AnaToolHandle<IAsgElectronLikelihoodTool> m_likelihoodTool_medium;
    asg::AnaToolHandle<IAsgElectronLikelihoodTool> m_likelihoodTool_tight;
  };



  class ElectronToolCorrect : virtual public AnaToolCorrect<xAOD::ElectronContainer>
  {

    ASG_TOOL_CLASS (ElectronToolCorrect, ana::IAnaTool)

  public:
    // Public interface methods

    ElectronToolCorrect (const std::string& name);

    StatusCode initialize() override;

    virtual StatusCode
    correctObject (xAOD::Electron& electron) override;

  private:
    // Configuration

    bool m_isAF2;
    bool m_isData;

    // CP tools

    asg::AnaToolHandle<CP::IEgammaCalibrationAndSmearingTool> m_calibration;
    asg::AnaToolHandle<CP::IIsolationCorrectionTool> m_isoCorrTool;

    SelectionCut cut_correction;
    SelectionCut cut_isoCorr_tool;
  };



  class ElectronToolSelect : virtual public AnaToolSelect<xAOD::ElectronContainer>
  {

    ASG_TOOL_CLASS (ElectronToolSelect, ana::IAnaTool)

  public:
    // Public interface methods

    ElectronToolSelect (const std::string& name);

    StatusCode initialize() override;

    virtual StatusCode
    selectObject (xAOD::Electron& electron) override;

  private:

    // Configuration

    std::string m_idStr;
    float m_ptMin;
    float m_etaMax;

    bool m_isolationOn;
    std::string m_isolationWP;

    std::string m_idAna;
    bool m_looseImpactCut;


    // CP tools

    asg::AnaToolHandle<CP::IIsolationSelectionTool> m_isolationTool;

    // the different cuts
    SelectionCut cut_id;
    SelectionCut cut_OQ;
    SelectionCut cut_ptMin;
    SelectionCut cut_clusEta;
    SelectionCut cut_d0;
    SelectionCut cut_z0;
    SelectionCut cut_isolationTool;
    SelectionCut cut_idAna;
  };



  class ElectronToolWeight : virtual public AnaToolWeight<xAOD::ElectronContainer>
  {

    ASG_TOOL_CLASS (ElectronToolWeight, ana::IAnaTool)

  public:
    // Public interface methods

    ElectronToolWeight (const std::string& name);

    StatusCode initialize() override;

    virtual StatusCode
    objectWeight (const xAOD::Electron& elecron, float& weight) override;

  public:
    // CP tools

    asg::AnaToolHandle<IAsgElectronEfficiencyCorrectionTool> m_efficiencyTool_reco;
    asg::AnaToolHandle<IAsgElectronEfficiencyCorrectionTool> m_efficiencyTool_id;
    asg::AnaToolHandle<IAsgElectronEfficiencyCorrectionTool> m_efficiencyTool_iso;

  private:
    // Configuration

    std::string m_idStr;
    std::string m_isoStr;
    std::string m_idAna;
    bool m_isAF2;
  };


  StatusCode makeElectronTool(DefinitionArgs& args,
                              const std::string& id,
                              const bool isolationOn = false,
                              const std::string& isolationWP = "Gradient",
                              const std::string& idAna = "",
                              const bool looseImpactCut=false);

}

#endif

---