egammaMVACalibTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef EGAMMAMVACALIB_EGAMMAMVACALIBTOOL_H
#define EGAMMAMVACALIB_EGAMMAMVACALIBTOOL_H
 
// Package includes
#include "EgammaAnalysisInterfaces/IegammaMVACalibTool.h"
#include "xAODEgamma/EgammaEnums.h"
#include "MVAUtils/BDT.h"
#include "egammaMVACalib/egammaMVAFunctions.h"
 
// Framework includes
#include "AsgTools/AsgTool.h"
#include "AsgTools/PropertyWrapper.h"
 
//Root includes
#include "TH2Poly.h"
#include "TObject.h"
#include "TString.h"
#include "TFormula.h"
 
// STL includes
#include <string>
#include <memory>
#include <functional>
 
namespace egammaMVACalibTool_detail {
struct Funcs;
}
 

 
class egammaMVACalibTool : public asg::AsgTool, virtual public IegammaMVACalibTool{
  ASG_TOOL_CLASS(egammaMVACalibTool, IegammaMVACalibTool)
public:
  egammaMVACalibTool(const std::string& type);
  virtual ~egammaMVACalibTool() override;
 
  virtual StatusCode initialize() override;

  enum CalibrationType {correctEaccordion, fullCalibration};

  enum ShiftType {NOSHIFT=0, MEAN10TOTRUE};

  float getEnergy(const xAOD::CaloCluster& clus,
                  const xAOD::Egamma* eg,
          const egammaMVACalib::GlobalEventInfo& gei = egammaMVACalib::GlobalEventInfo())
    const override final;
 
private:
  Gaudi::Property<int> m_particleType {this,
      "ParticleType", xAOD::EgammaParameters::NumberOfEgammaTypes,
      "What type of particle do we use"};
 
  Gaudi::Property<int> m_shiftType {this,
      "ShiftType", MEAN10TOTRUE,
      "Shift corrections to apply to value"};
 
  Gaudi::Property<int> m_calibrationType {this,
      "CalibrationType", correctEaccordion,
      "What type of calibration to apply"};
 
  Gaudi::Property<bool> m_clusterEif0 {this,
      "useClusterIf0", true,
      "Use cluster energy if MVA response is 0"};

  Gaudi::Property<std::string> m_folder {this,
      "folder", "",
      "string with folder for weight files"};
 
  Gaudi::Property<bool> m_useLayerCorrected {this,
      "use_layer_corrected", false,
      "whether to use layer corrections"};
  
  std::unique_ptr<TH2Poly> m_hPoly;

  std::vector<MVAUtils::BDT> m_BDTs;

  // (Need to hide this type to avoid a cling bug:
  //  https://github.com/root-project/root/issues/18833)
  std::unique_ptr<egammaMVACalibTool_detail::Funcs> m_funcs;

  std::vector<TFormula> m_shifts;

  StatusCode setupBDT(const egammaMVAFunctions::funcMap_t& funcLibrary,
                      const std::string& fileName);

  static const TString& getString(TObject* obj) ;
 
};
 
#endif