HIJetAugmentationTool.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#include "HIJetAugmentationTool.h"
#include "AsgDataHandles/WriteDecorHandle.h"
#include <format>
 
namespace DerivationFramework
{
  HIJetAugmentationTool::HIJetAugmentationTool(   const std::string& t,
                                                                const std::string& n,
                                                                const IInterface* p ) :   base_class(t,n,p)
  {
  }
 
  // Destructor
  HIJetAugmentationTool::~HIJetAugmentationTool()
  {
  }
 
  // Athena initialize and finalize
  StatusCode HIJetAugmentationTool::initialize()
  {
    ATH_CHECK(m_eventInfoKey.initialize());
    ATH_CHECK(m_hiJet_key.initialize());
    ATH_CHECK(m_caloJet_key.initialize());
    ATH_CHECK(m_jvtUpdateTool.retrieve());
    
    m_jvtMatchedKey = m_hiJet_key.key() + "." + m_jvtMatchedKey.key();
    ATH_CHECK(m_jvtMatchedKey.initialize());
    m_jvtMediumPassedKey = m_hiJet_key.key() + "." + m_jvtMediumPassedKey.key();
    ATH_CHECK(m_jvtMediumPassedKey.initialize());
    m_jvtTightPassedKey = m_hiJet_key.key() + "." + m_jvtTightPassedKey.key();
    ATH_CHECK(m_jvtTightPassedKey.initialize());
    
    ATH_MSG_INFO("DeltaRJetMatching = "<< m_deltaR.value());
    
    
    return StatusCode::SUCCESS;
  }
  
  StatusCode HIJetAugmentationTool::finalize()
  {
 
    ATH_CHECK(m_jvtUpdateTool->finalize());
 
    return StatusCode::SUCCESS;
  }
 
  double deltaR(double eta1, double eta2, double phi1, double phi2) {
    double deltaPhi = TVector2::Phi_mpi_pi(phi1 - phi2);
    double deltaEta = eta1 - eta2;
    return std::sqrt(deltaEta * deltaEta + deltaPhi * deltaPhi);
  }
 
  StatusCode HIJetAugmentationTool::addBranches(const EventContext &ctx) const {
    SG::ReadHandle<xAOD::EventInfo> eventInfo(m_eventInfoKey, ctx);
 
    // Load jet containers
    SG::ReadHandle<xAOD::JetContainer> hiJets(m_hiJet_key, ctx);
    if (!hiJets.isValid()) {
      ATH_MSG_ERROR("Couldn't retrieve JetContainer with key " << m_hiJet_key);
      return StatusCode::FAILURE;
    }
    SG::ReadHandle<xAOD::JetContainer> caloJets(m_caloJet_key, ctx);
    if (!caloJets.isValid()) {
      ATH_MSG_ERROR("Couldn't retrieve JetContainer with key "
                    << m_caloJet_key);
      return StatusCode::FAILURE;
    }
 
    // calibrate topo jets
    //
 
 
    SG::WriteDecorHandle<xAOD::JetContainer, float> jvtMatchedHandle(m_jvtMatchedKey,ctx);
    SG::WriteDecorHandle<xAOD::JetContainer, bool>  jvtMediumPassedHandle(m_jvtMediumPassedKey,ctx);
    SG::WriteDecorHandle<xAOD::JetContainer, bool>  jvtTightPassedHandle(m_jvtTightPassedKey,ctx);
 
    // first loop over calibrated HI jets
    for (const auto *hjet : *hiJets) {
      float mindR = 999.;
      float matchedEta = 999.;
      float matchedJvt = -1;
      // temporary HI jvt selection
      bool passJvtMedium = false;
      bool passJvtTight  = false;
      // second loop over topo jets
      for (const auto *tjet : *caloJets) {
        float newjvt = m_jvtUpdateTool->updateJvt(*tjet);
 
        // perform the matching
        float dR =
            deltaR(tjet->eta(), hjet->eta(), hjet->phi(), tjet->phi());
        
        if (dR < m_deltaR.value() && dR < mindR) {
          mindR = dR;
          matchedJvt = newjvt;
      matchedEta = tjet->eta();
        }
      }
    
      if (mindR < m_deltaR.value()) {
        (jvtMatchedHandle)(*hjet) = matchedJvt;
    // HI jet sub-group pre-recommendation for JVT: https://atlas-heavy-ions.docs.cern.ch/Jets/jetselection/
    // applying JVT selection on EMTopo jet matched to HI jet
    // using selections from: https://twiki.cern.ch/twiki/bin/viewauth/AtlasProtected/PileupJetRecommendations#JVT
        if (std::abs(matchedEta) < 2.4) {
            passJvtMedium = matchedJvt > 0.59;
            passJvtTight  = matchedJvt > 0.91;
        } else if (std::abs(matchedEta) < 2.5) {
            passJvtMedium = matchedJvt > 0.11;
            passJvtTight  = true; // no tight JVT in this region
        } else {
            passJvtMedium = true;
            passJvtTight  = true;
        }
        (jvtMediumPassedHandle)(*hjet) = passJvtMedium;
        (jvtTightPassedHandle)(*hjet)  = passJvtTight;
 
      } else {
        (jvtMatchedHandle)(*hjet) = -1;
        (jvtMediumPassedHandle)(*hjet) = false;
        (jvtTightPassedHandle)(*hjet)  = false;
      }
    }
 
    return StatusCode::SUCCESS;
  }
}