egammaTransformerSvc.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef EGAMMATRANSFORMERCALIB_EGAMMATRANSFORMERSVC_H
#define EGAMMATRANSFORMERCALIB_EGAMMATRANSFORMERSVC_H
 
#include "xAODEgamma/EgammaEnums.h"
#include "EgammaAnalysisInterfaces/IegammaMVASvc.h"
#include "EgammaAnalysisInterfaces/IegammaMVACalibTool.h"
// Framework includes
#include "AsgServices/AsgService.h"
#include "AsgTools/PropertyWrapper.h"
 
#include "xAODEgamma/PhotonFwd.h"
 
 
#include <string>
 
 
class egammaTransformerSvc : public extends<asg::AsgService, IegammaMVASvc>
{
public:
  using extends::extends;  // base class constructor
 
  virtual ~egammaTransformerSvc() override {};
  virtual StatusCode initialize() override;

  // Note for both egammaTransformerSvc::getEnergy overloads below:
  // zero-response handling is delegated to the underlying calibration tools
  // (m_mvaElectron/m_mvaFwdElectron/m_mvaConvertedPhoton/m_mvaUnconvertedPhoton).
  // By default their property useClusterIf0 (m_clusterEif0) is true, so a zero
  // model response returns the uncalibrated cluster energy instead of 0.
  // Therefore mvaE is not expected to be 0 by default unless the user
  // explicitly changes that tool property in the calibration configuration.
  StatusCode getEnergy(const xAOD::CaloCluster& cluster,
                       const xAOD::Egamma& eg,
                       double& mvaE,
               const egammaMVACalib::GlobalEventInfo& gei = egammaMVACalib::GlobalEventInfo()) const override final;

  StatusCode getEnergy(const xAOD::CaloCluster& cluster,
                       const xAOD::EgammaParameters::EgammaType egType,
                       double& mvaE,
               const egammaMVACalib::GlobalEventInfo& gei = egammaMVACalib::GlobalEventInfo()) const override final;

  StatusCode execute(xAOD::CaloCluster& cluster,
                     const xAOD::Egamma& eg,
             const egammaMVACalib::GlobalEventInfo& gei = egammaMVACalib::GlobalEventInfo()) const override final;

  StatusCode execute(xAOD::CaloCluster& cluster,
                     const xAOD::EgammaParameters::EgammaType egType,
             const egammaMVACalib::GlobalEventInfo& gei = egammaMVACalib::GlobalEventInfo()) const override final;
 
 
private:

  ToolHandle<IegammaMVACalibTool> m_mvaElectron {this,
      "ElectronTool", "", "Tool to handle MVA trees for electrons"};

  ToolHandle<IegammaMVACalibTool> m_mvaFwdElectron {this,
      "FwdElectronTool", "", "Tool to handle MVA trees for forward electrons"};

  ToolHandle<IegammaMVACalibTool> m_mvaUnconvertedPhoton {this,
      "UnconvertedPhotonTool", "", "Tool to handle MVA trees for unconverted photons"};

  ToolHandle<IegammaMVACalibTool> m_mvaConvertedPhoton {this,
      "ConvertedPhotonTool", "", "Tool to handle MVA trees for converted photons"};
 
  Gaudi::Property<float> m_maxConvR {this,
      "MaxConvRadius", 800.0,
      "The maximum conversion radius for a photon to be considered converted"};
 
  Gaudi::Property<int> m_removeTRTConvBarrel {this,
      "RemoveTRTConvBarrel", -1,
      "Remove TRT converted photons in the barrel: no=0, yes=1, automatic=-1"};
 
  Gaudi::Property<std::string> m_folder {this,
      "folder", "", "folder for weight files"};
 

  bool isConvCalib(const xAOD::Photon& ph) const;
 
  StatusCode resolve_flags();
 
 
};
 
#endif