JetCalibrationDecoratorAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef FLAVOR_TAG_DISCRIMINANTS_JET_CALIBRATION_DECORATOR_ALG_H
#define FLAVOR_TAG_DISCRIMINANTS_JET_CALIBRATION_DECORATOR_ALG_H
 
#include <string>
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteDecorHandleKey.h"
#include "xAODJet/JetContainer.h"
#include "JetInterface/IJetModifier.h"
 
 
namespace FlavorTagDiscriminants {
 
  class JetCalibrationDecoratorAlg : public AthReentrantAlgorithm {
  public:
    JetCalibrationDecoratorAlg(const std::string& name,
                               ISvcLocator* pSvcLocator);
 
    virtual StatusCode initialize() override;
    virtual StatusCode execute(const EventContext& ctx) const override;
 
  private:
    ToolHandle<IJetModifier> m_calibTool;
 
    SG::ReadHandleKey<xAOD::JetContainer> m_jetKey {
      this, "JetContainer", "AntiKt4EMPFlowJets",
        "Input jet container"};
 
    SG::WriteDecorHandleKey<xAOD::JetContainer> m_ptCalibKey {
      this, "ptCalibratedKey", "AntiKt4EMPFlowJets.pt_calibrated",
        "Decorated calibrated pT"};
 
    SG::WriteDecorHandleKey<xAOD::JetContainer> m_etaCalibKey {
      this, "etaCalibratedKey", "AntiKt4EMPFlowJets.eta_calibrated",
        "Decorated calibrated eta"};
 
    SG::WriteDecorHandleKey<xAOD::JetContainer> m_massCalibKey {
      this, "massCalibratedKey", "AntiKt4EMPFlowJets.mass_calibrated",
        "Decorated calibrated mass"};
 
    SG::WriteDecorHandleKey<xAOD::JetContainer> m_phiCalibKey {
      this, "phiCalibratedKey", "AntiKt4EMPFlowJets.phi_calibrated",
        "Decorated calibrated phi"};
  };
 
} // namespace FlavorTagDiscriminants
 
#endif