egammaMVASvc.h

Go to the documentation of this file.

// Dear Emacs, this is -*- C++ -*-
 
/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef EGAMMAMVACALIB_EGAMMAMVASVC_H
#define EGAMMAMVACALIB_EGAMMAMVASVC_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 egammaMVASvc : public extends<asg::AsgService, IegammaMVASvc>
{
public:
  using extends::extends;  // base class constructor
 
  virtual ~egammaMVASvc() override {};
  virtual StatusCode initialize() override;
 
 
  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