ElectronChargeEfficiencyCorrectionTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ELECTRONCHARGECORRECTION__ELECTRONCHARGECORRECTIONTOOL__H
#define ELECTRONCHARGECORRECTION__ELECTRONCHARGECORRECTIONTOOL__H
 
#include "AsgTools/AsgTool.h"
#include "EgammaAnalysisInterfaces/IAsgElectronEfficiencyCorrectionTool.h"
#include "PATInterfaces/ISystematicsTool.h"
#include "TH2.h"
#include "AthContainers/AuxElement.h"
#include "xAODEgamma/ElectronFwd.h"
#include "ColumnarMode/ColumnarMode.h"
 
#include <map>
#include <string>
#include <vector>
 
// Forward declaration(s):
namespace xAOD {
class IParticle;
}
 
namespace ADD_CMODE(CP) {
 
class ElectronChargeEfficiencyCorrectionTool final
  : virtual public IAsgElectronEfficiencyCorrectionTool
  , public asg::AsgTool
{
  ASG_TOOL_CLASS(ElectronChargeEfficiencyCorrectionTool,
                 IAsgElectronEfficiencyCorrectionTool)
 
public:
  ElectronChargeEfficiencyCorrectionTool(const std::string& name);

  virtual ~ElectronChargeEfficiencyCorrectionTool();
 
public:
  virtual StatusCode initialize() override final;
 

  virtual CP::CorrectionCode getEfficiencyScaleFactor(
    const xAOD::Electron& inputObject,
    double& sf) const override final;

  virtual CP::CorrectionCode applyEfficiencyScaleFactor(
    const xAOD::Electron& inputObject) const override final;

  virtual bool isAffectedBySystematic(
    const CP::SystematicVariation& systematic) const override final;

  virtual CP::SystematicSet affectingSystematics() const override final;

  virtual CP::SystematicSet recommendedSystematics() const override final;
 
  virtual StatusCode applySystematicVariation(const CP::SystematicSet& systConfig) override final;
 

  virtual const CP::SystematicSet& appliedSystematics() const override  final
  {
    return *m_appliedSystematics;
  }
 
  virtual int systUncorrVariationIndex(const xAOD::Electron&) const override final
  {
    ATH_MSG_ERROR(
        "systUncorrVariationIndex is not implemented in "
        "ElectronChargeEfficiencyCorrectionTool");
    return -999;
  }
 
  virtual CP::CorrectionCode getEfficiencyScaleFactor(
      const double, const double, const unsigned int,
      double&) const override final{
    ATH_MSG_ERROR(
        "No toysimplemented in "
        "ElectronChargeEfficiencyCorrectionTool");
    return CP::CorrectionCode::Error;
  }
  //
  virtual int getNumberOfToys() const override final
  {
    ATH_MSG_ERROR(
        "No toysimplemented in "
        "ElectronChargeEfficiencyCorrectionTool");
    return -1;
  };
  virtual void printCorrelationModels() const override final
  {
    ATH_MSG_INFO(
      "ONLY A DEFAULT Correlation model available for now");
  };
 
 
private:
  StatusCode registerSystematics();
  float getChargeFlipRate(double eta,
                          double pt,
                          TH2* hrates,
                          double& flipRate) const;

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

  std::map<std::string, std::vector<TH2*>>
    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<TH2*>>
    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;
 
  // Systematics
  std::vector<std::string> m_systematics;
 
  std::map<CP::SystematicSet, CP::SystematicSet> m_filtered_sys_sets;
  // boost::unordered_map<CP::SystematicSet,EffiCollection*> m_sf_sets;
  CP::SystematicSet m_mySysConf;
  CP::SystematicSet m_affectingSys;

  CP::SystematicSet* m_appliedSystematics;

  std::string m_sf_decoration_name;
  SG::Decorator<float>* m_sfDec;
};
 
} // End namespace ADD_CMODE(CP)
 
#endif