egammaMVACalibTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 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>
 
 
class egammaMVACalibTool : public asg::AsgTool, virtual public IegammaMVACalibTool{
  ASG_TOOL_CLASS(egammaMVACalibTool, IegammaMVACalibTool)
public:
  egammaMVACalibTool(const std::string& type);
  virtual ~egammaMVACalibTool() override = default;
 
  virtual StatusCode initialize() override;
 
  enum CalibrationType {correctEaccordion, fullCalibration};
 
  enum ShiftType {NOSHIFT=0, MEAN10TOTRUE};
 
  float getEnergy(const xAOD::CaloCluster& clus,
                  const xAOD::Egamma* eg) 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;
 
  std::vector<std::vector<std::function<float(const xAOD::Egamma*, const xAOD::CaloCluster*)> > > 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