JetCalibrationAlg.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/

 
 
//
// includes
//
 
#include <JetAnalysisAlgorithms/JetCalibrationAlg.h>
 
//
// method implementations
//
 
namespace CP
{
 
  JetCalibrationAlg ::
  JetCalibrationAlg (const std::string& name, ISvcLocator* pSvcLocator)
    : EL::AnaAlgorithm (name, pSvcLocator)
  {
    declareProperty ("calibrationTool", m_calibrationTool, "The calibration tool we apply");
  }
 
  StatusCode JetCalibrationAlg ::
  initialize ()
  {
    ANA_CHECK (m_calibrationTool.retrieve());
    ANA_CHECK (m_jetHandle.initialize (m_systematicsList));
    ANA_CHECK (m_systematicsList.initialize());
    return StatusCode::SUCCESS;
  }
 
 
 
  StatusCode JetCalibrationAlg ::
  execute (const EventContext& ctx)
  {
    for (const auto& sys : m_systematicsList.systematicsVector())
    {
      xAOD::JetContainer *jets = nullptr;
      ANA_CHECK (m_jetHandle.getCopy (jets, sys, ctx));
 
      if (m_HIsetup.value())
      {
        for(auto jet : *jets )
    {
          xAOD::JetFourMom_t jet_4mom_subtracted = (*jet).jetP4("JetSubtractedScaleMomentum");
      (*jet).setJetP4("JetConstitScaleMomentum",jet_4mom_subtracted);
      (*jet).setJetP4("JetEMScaleMomentum",jet_4mom_subtracted);
    }
      }
 
      ANA_CHECK (m_calibrationTool->applyCalibration(*jets));
    }
 
    return StatusCode::SUCCESS;
  }
}