JetFFSmearingAlg.cxx

Go to the documentation of this file.

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

 
//
// includes
//
 
#include <JetAnalysisAlgorithms/JetFFSmearingAlg.h>
#include <algorithm>
 
//
// method implementations
//
 
namespace CP {
 
StatusCode JetFFSmearingAlg ::initialize() {
  ANA_CHECK(m_FFSmearingTool.retrieve());
  ANA_CHECK(m_jetHandle.initialize(m_systematicsList));
  ANA_CHECK(m_preselection.initialize(m_systematicsList, m_jetHandle,
                                      SG::AllowEmpty));
 
  ANA_CHECK(m_systematicsList.addSystematics(*m_FFSmearingTool));
  ANA_CHECK(m_systematicsList.initialize());
  ANA_CHECK(m_outOfValidity.initialize());
 
  // FIXME: while the FFJetSmearingTool is unable to accept external systematics,
  // we make our own vector of systematics
  const CP::SystematicSet& recommendedSysts = m_FFSmearingTool->recommendedSystematics();
  for (const auto &sys : recommendedSysts) m_systematicsVector.push_back(CP::SystematicSet({sys}));
 
  return StatusCode::SUCCESS;
}
 
StatusCode JetFFSmearingAlg ::execute(const EventContext& ctx) {
  for (const auto &sys : m_systematicsList.systematicsVector()) {
    xAOD::JetContainer *jets = nullptr;
    ANA_CHECK(m_jetHandle.getCopy(jets, sys, ctx));
 
    // FIXME: while the FFJetSmearingTool is unable to accept external systematics,
    // only run over the internal ones
    if ( std::find(m_systematicsVector.begin(), m_systematicsVector.end(), sys) == m_systematicsVector.end() ) continue;
 
    ANA_CHECK(m_FFSmearingTool->applySystematicVariation(sys));
 
    for (xAOD::Jet *jet : *jets) {
      if (m_preselection.getBool(*jet, sys)) {
        ANA_CHECK_CORRECTION(m_outOfValidity, *jet,
                             m_FFSmearingTool->applyCorrection(*jet));
      }
    }
  }
  return StatusCode::SUCCESS;
}
}  // namespace CP