EGammaEnergyCalibrationWrapper.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
// Author: Chengxi Yang (cxyang@berkeley.edu)
 
#ifndef DERIVATIONFRAMEWORK_EGAMMAENERGYCALIBRATIONWRAPPER_H
#define DERIVATIONFRAMEWORK_EGAMMAENERGYCALIBRATIONWRAPPER_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "DerivationFrameworkInterfaces/IAugmentationTool.h"
//
#include "StoreGate/ReadHandleKey.h"
#include "GaudiKernel/ServiceHandle.h"
#include "xAODEgamma/EgammaContainer.h"
// name space of global event info is defined here
#include "EgammaAnalysisInterfaces/IegammaMVASvc.h"
#include "xAODEventInfo/EventInfo.h"
//
#include <string>
 
 
namespace DerivationFramework {
 
  class EGammaEnergyCalibrationWrapper :
    public extends<AthAlgTool, IAugmentationTool>
  {
  public:
    using base_class::base_class;
    StatusCode initialize() override final;
    StatusCode addBranches(const EventContext& ctx) const override final;
 
  private:
    
    SG::ReadHandleKey<xAOD::EgammaContainer> m_electronContainerKey{
      this, "ElectronContainerName", "Electrons", "Electron container"
    };
 
    SG::ReadHandleKey<xAOD::EgammaContainer> m_photonContainerKey{
      this, "PhotonContainerName", "Photons", "Photon container"
    };
 
    SG::WriteDecorHandleKey<xAOD::EgammaContainer> m_electronEnergyDecoKey{
      this, "decoratorTransformerEnergy", m_electronContainerKey, "", "Calibrated energy decoration for electrons"
    };
 
    SG::WriteDecorHandleKey<xAOD::EgammaContainer> m_photonEnergyDecoKey{
      this, "decoratorTransformerEnergyPhoton", m_photonContainerKey, "", "Calibrated energy decoration for photons"
    };
 
    SG::ReadHandleKey<xAOD::EventInfo> m_eventInfo_key{this, "EventInfo", "EventInfo", "Input event information"};
 
    ServiceHandle<IegammaMVASvc> m_MVACalibSvc{
      this, "TransformerCalibSvc", "", "Calibration service using Transformer model"
    };
  };
 
} // namespace DerivationFramework
 
#endif