JvtSelectionTool.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#include "JetJvtEfficiency/JvtSelectionTool.h"
#include "AsgDataHandles/ReadDecorHandle.h"
 
namespace {
    const static std::map<std::string, float> pflowWPMap{{"Medium", 0.2}, {"Tight", 0.5}};
    const static std::map<std::string, float> topoWPMap{
            {"Loose", 0.11}, {"Medium", 0.59}, {"Tight", 0.91}};
} // namespace
 
namespace CP {
 
    StatusCode JvtSelectionTool::initialize() {
        if(m_isPFlow.value()){
      ATH_MSG_WARNING("Jvt is deprecated, please move to using NNJvt");
    }
 
        ATH_CHECK(JvtSelectionToolBase::initialize());
 
        if (m_wp != "Custom") {
            if (m_wp == "Default")
                m_wp = m_isPFlow.value() ? "Tight" : "Medium";
            m_jvtCutBorder = (!m_isPFlow.value() && m_wp == "Medium") ? 0.11 : -2.;
            const auto &map = m_isPFlow.value() ? pflowWPMap : topoWPMap;
            auto itr = map.find(m_wp);
            if (itr == map.end()) {
                ATH_MSG_ERROR("Invalid Jvt working point name");
                return StatusCode::FAILURE;
            }
            m_jvtCut = itr->second;
        }
 
        return StatusCode::SUCCESS;
    }
 
    bool JvtSelectionTool::select(const xAOD::IParticle *jet) const {
        if(!isInRange(jet)) return true;
        SG::ReadDecorHandle<xAOD::JetContainer, float> jvtHandle(m_jvtMomentKey);
        float eta = m_etaAcc(*jet);
        if (std::abs(eta) > 2.4 && std::abs(eta) < 2.5)
            return jvtHandle(*jet) > m_jvtCutBorder;
        return jvtHandle(*jet) > m_jvtCut;
    }
 
} // namespace CP