BJetCalibrationAlg.cxx

Go to the documentation of this file.

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

 
 
//
// includes
//
 
#include <JetAnalysisAlgorithms/BJetCalibrationAlg.h>
 
//
// method implementations
//
 
namespace CP
{
 
  StatusCode BJetCalibrationAlg ::
  initialize ()
  {
    ANA_CHECK (m_jetHandle.initialize (m_systematicsList));
    ANA_CHECK (m_jetPreselection.initialize
           (m_systematicsList, m_jetHandle, SG::AllowEmpty));
    ANA_CHECK (m_muonHandle.initialize (m_systematicsList));
    ANA_CHECK (m_muonPreselection.initialize
           (m_systematicsList, m_muonHandle, SG::AllowEmpty));
 
    ANA_CHECK (m_nmuons.initialize(m_systematicsList, m_jetHandle));
 
    if(!m_muonSelectionTool.empty()){
      ANA_CHECK (m_muonSelectionTool.retrieve());
      if(!m_muonPreselection.empty()){
        ATH_MSG_ERROR ("muonSelectionTool and muonPreselection should not be configured simultaneously");
        return StatusCode::FAILURE;
      }
    }
 
    ANA_CHECK (m_muonInJetTool.retrieve());
    if(!m_bJetTool.empty()) ANA_CHECK (m_bJetTool.retrieve());
 
    ANA_CHECK (m_systematicsList.initialize());
    return StatusCode::SUCCESS;
  }
 
  StatusCode BJetCalibrationAlg ::
  execute (const EventContext& ctx)
  {
    for (const auto& sys : m_systematicsList.systematicsVector())
    {
      xAOD::JetContainer *jets = nullptr;
      ANA_CHECK (m_jetHandle.getCopy (jets, sys, ctx));
 
      const xAOD::MuonContainer *muons = nullptr;
      ANA_CHECK (m_muonHandle.retrieve (muons, sys, ctx));
 
      std::vector<const xAOD::Muon*> muons_for_correction;
      muons_for_correction.reserve(muons->size());
      for(const xAOD::Muon* muon : *muons)
      {
          if (// For analysis frameworks
            (m_muonSelectionTool.empty() && m_muonPreselection.getBool(*muon, sys)) ||
             // For other frameworks
            (!m_muonSelectionTool.empty() && m_muonSelectionTool->accept(*muon)))
        {
            muons_for_correction.emplace_back(muon);
        }
      }
 
      for (xAOD::Jet* jet : *jets)
      {
        jet->setJetP4("NoBJetCalibMomentum", jet->jetP4());
        jet->setJetP4("MuonCorrMomentum", jet->jetP4());
        if (!m_bJetTool.empty())
        {
          jet->setJetP4("BJetCalibMomentum", jet->jetP4());
        }
 
        int nmuons = 0;
        if (m_jetPreselection.getBool(*jet, sys))
        {
          ANA_CHECK (m_muonInJetTool->applyMuonInJetCorrection(*jet, muons_for_correction, nmuons));
          jet->setJetP4("MuonCorrMomentum", jet->jetP4());
          if (!m_bJetTool.empty())
          {
            ANA_CHECK (m_bJetTool->applyBJetCorrection(*jet, nmuons>0));
            jet->setJetP4("BJetCalibMomentum", jet->jetP4());
          }
 
          if (m_onlyDecorate)
          {
            // Reset the jet 4-vector to the original one
            jet->setJetP4(jet->jetP4("NoBJetCalibMomentum"));
          }
        }
 
        m_nmuons.set(*jet, nmuons, sys);
      }
    }
 
    return StatusCode::SUCCESS;
  }
}