TriggerMatchingTool.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#include "TriggerMatchingTool.h"
#include "BuildCombinations.h"
#include "GaudiKernel/ServiceHandle.h"
#include "GaudiKernel/IIncidentSvc.h"
#include "xAODCore/AuxContainerBase.h"
#include "xAODCaloEvent/CaloClusterContainer.h"
#include "FourMomUtils/xAODP4Helpers.h"
#include <memory>
#include <algorithm>
#include "AthenaKernel/IProxyDict.h"
#include "AthenaKernel/ExtendedEventContext.h"
#include "StoreGate/ReadHandle.h"
 
namespace {
  template <typename T>
    using constAcc_t = SG::AuxElement::ConstAccessor<T>;
  template <typename T>
    using acc_t = SG::AuxElement::Accessor<T>;
  template <typename T>
    using vecLink_t = std::vector<ElementLink<T>>;
 
  template <typename T>
    ElementLink<T> makeLink(typename T::const_value_type element, IProxyDict *proxyDct)
    {
      SG::DataProxy *proxy = proxyDct->proxy(element->container());
      // This will be null if the container isn't in the store. This shouldn't really happen
      if (!proxy)
        return {};
      else
        return ElementLink<T>(proxy->sgkey(), element->index(), proxyDct);
    }
} //> end anonymous namespace
 
namespace DerivationFramework {
 
  TriggerMatchingTool::TriggerMatchingTool(
      const std::string& type,
      const std::string& name,
      const IInterface* pSvcLocator) :
    base_class(type, name, pSvcLocator)
  {
    declareProperty("ChainNames", m_chainNames, 
        "The list of trigger chains to match.");
    declareProperty("OnlineParticleTool", m_trigParticleTool,
        "The tool to retrieve online particles from the navigation.");
    declareProperty("InputElectrons",
        m_offlineInputs[xAOD::Type::Electron] = "Electrons",
        "Offline electron candidates for matching.");
    declareProperty("InputPhotons",
        m_offlineInputs[xAOD::Type::Photon] = "Photons",
        "Offline photon candidates for matching.");
    declareProperty("InputMuons",
        m_offlineInputs[xAOD::Type::Muon] = "Muons",
        "Offline muon candidates for matching.");
    declareProperty("InputTaus",
        m_offlineInputs[xAOD::Type::Tau] = "TauJets",
        "Offline tau candidates for matching.");
    declareProperty("DRThreshold", m_drThreshold = 0.1,
        "The maximum dR between an offline and an online particle to consider "
        "a match between them.");
    declareProperty("Rerun", m_rerun = true,
        "Whether to match triggers in rerun mode.");
    declareProperty("OutputContainerPrefix", m_outputPrefix="TrigMatch_",
        "The prefix to add to the output containers.");
    declareProperty("CheckEmptyChainGroups", m_checkEmptyChainGroups = true,
        "If set, discard any empty chain groups. Otherwise these will cause "
        "a job failure.");
    declareProperty("InputDependentConfig", m_inputDependentConfig=false,
        "Warn when a trigger is removed (if the configuration is dependent "
        "on the inputs, removal is not expected).");
  }
 
  StatusCode TriggerMatchingTool::initialize()
  {
    ATH_MSG_INFO( "Initializing " << name() );
 
    // Remove any duplicates from the list of chain names
    std::sort(m_chainNames.begin(), m_chainNames.end() );
    m_chainNames.erase(std::unique(m_chainNames.begin(), m_chainNames.end() ), m_chainNames.end() );
 
    ATH_CHECK( m_trigParticleTool.retrieve() );
    ATH_CHECK( m_scoreTool.retrieve() );
    for (auto &p : m_offlineInputs)
    {
      ATH_CHECK(p.second.initialize(SG::AllowEmpty));
    }
    return StatusCode::SUCCESS;
  }
 
  StatusCode TriggerMatchingTool::addBranches(const EventContext& ctx) const
  {
    [[maybe_unused]] static const bool firstEvent = [&](){
      auto itr = m_chainNames.begin();
      while (itr != m_chainNames.end() ) {
        const Trig::ChainGroup* cg = m_tdt->getChainGroup(*itr);
        if (cg->getListOfTriggers().size() == 0){
          if (m_inputDependentConfig)
            ATH_MSG_WARNING("Removing trigger " << (*itr) << " -- suggests a bad tool configuration (asking for triggers not in the menu)");
          // We are now modifying the mutable m_chainNames but since it is done
          // within this static initialization this is thread-safe:
          itr = m_chainNames.erase(itr);
        } else
          ++itr;
      }
      return false;
    }();
 
    const Atlas::ExtendedEventContext &extendedCtx = Atlas::getExtendedEventContext(ctx);
 
    // Now, get all the possible offline candidates
    std::map<xAOD::Type::ObjectType, const xAOD::IParticleContainer *> offlineContainers;
    std::map<xAOD::Type::ObjectType, particleVec_t> offlineCandidates;
    for (const auto& p : m_offlineInputs) {
      // Skip any that may have been disabled by providing an empty string to
      // the corresponding property
      if (p.second.empty() )
        continue;
      auto handle = SG::makeHandle(p.second, ctx);
      if (!handle.isValid())
      {
        ATH_MSG_ERROR("Failed to retrieve " << p.second);
        return StatusCode::FAILURE;
      }
      offlineContainers[p.first] = handle.ptr();
      offlineCandidates.emplace(std::make_pair(
          p.first, 
          particleVec_t(handle->begin(), handle->end() )
      ) );
    }
 
    // Store possible matches from online to offline particles here
    // We do this as multiple chains may use the same online particles so we can
    // reuse this information.
    std::map<const xAOD::IParticle*, particleVec_t> matches;
 
    // Iterate through the chains to get the matchings
    for (const std::string& chain : m_chainNames) {
      // Create the output
      xAOD::TrigCompositeContainer* container(nullptr);
      ATH_CHECK( createOutputContainer(container, chain) );
 
      // Get the list of online combinations
      std::vector<particleVec_t> onlineCombinations;
      ATH_CHECK( m_trigParticleTool->retrieveParticles(onlineCombinations, chain, m_rerun) );
 
      ATH_MSG_DEBUG(
          onlineCombinations.size() << " combinations found for chain" << chain);
      
      // If no combinations were found (because the trigger was not passed) then
      // we can skip this trigger
      if (onlineCombinations.size() == 0)
        continue;
 
      using particleRange_t = TriggerMatchingUtils::RangedItr<typename particleVec_t::const_iterator>;
      // Now build up the list of offline combinations;
      std::vector<particleVec_t> offlineCombinations;
 
      // Projection operator to use for comparing vectors
      // of IParticle*.  Compare by using the sum of the pts
      // of all particles in the vector.
      struct outerproj {
        double operator() (const particleVec_t& v) const
        {
          // Unfortunately std::ranges::accumulate hasn't made it into C++
          // as of C++23 or we could use that directly.
          auto r = std::views::transform (v, &xAOD::IParticle::pt);
          return std::accumulate (std::begin(r), std::end(r), 0);
        }
      };
 
      for (const particleVec_t& combination : onlineCombinations) {
        // Here we store the possible candidates for the matching. We use the
        // range type as a lightweight method to carry around a view of a vector
        // without copying it (the RangedItr is essentially a combination of 3
        // iterators).
        std::vector<particleRange_t> matchCandidates;
        for (const xAOD::IParticle* part : combination) {
          const particleVec_t& possibleMatches = getCandidateMatchesFor(
              part, offlineCandidates, matches);
          matchCandidates.emplace_back(possibleMatches.begin(), possibleMatches.end() );
        } //> end loop over particles
        // Get all possible combinations of offline objects that could match to
        // this particular online combination.
        auto theseOfflineCombinations = 
          TriggerMatchingUtils::getAllDistinctCombinations<const xAOD::IParticle*>(
              matchCandidates,
              &xAOD::IParticle::pt,
              outerproj());
        if (msgLvl(MSG::VERBOSE) ) {
          // Spit out some verbose information
          ATH_MSG_VERBOSE(
              "Checking matching for chain " << chain
              << " with " << matchCandidates.size() << " legs");
          std::size_t idx = 0;
          for (const particleRange_t& range : matchCandidates)
            ATH_MSG_VERBOSE( "Leg #" << idx++ << " has " << range.size()
                << " offline candidates." );
          ATH_MSG_VERBOSE(
              "Matching generated " << theseOfflineCombinations.size()
              << " offline combinations");
        }
        // Now push them back into the output. Use a specialised function for
        // inserting into a sorted vector that ensures that we only output
        // unique combinations
        for (const particleVec_t& vec : theseOfflineCombinations)
          TriggerMatchingUtils::insertIntoSortedVector(offlineCombinations, vec, outerproj());
      } //> end loop over combinations
 
 
      // Decorate the found combinations onto trigger composites.
      for (const particleVec_t& foundCombination : offlineCombinations) {
        xAOD::TrigComposite* composite = new xAOD::TrigComposite();
        container->push_back(composite);
        static const acc_t<vecLink_t<xAOD::IParticleContainer>> dec_links(
            "TrigMatchedObjects");
        vecLink_t<xAOD::IParticleContainer>& links = dec_links(*composite);
        for (const xAOD::IParticle* part : foundCombination) {
          const xAOD::IParticleContainer *container = offlineContainers.at(part->type());
          // If we have an owning container then things are relatively simple:
          // we can just construct the element link from its SG key and the
          // index. Otherwise we have to locate the correct proxy from the element
          if (container->trackIndices())
            links.emplace_back(
                m_offlineInputs.at(part->type()).key(),
                part->index(),
                extendedCtx.proxy());
          else
            links.push_back(makeLink<xAOD::IParticleContainer>(part, extendedCtx.proxy()));
        }
      } //> end loop over the found combinations
    } //> end loop over chains
    return StatusCode::SUCCESS;
  }
 
  StatusCode TriggerMatchingTool::createOutputContainer(
      xAOD::TrigCompositeContainer*& container,
      const std::string& chain) const
  {
    auto uniqueContainer = std::make_unique<xAOD::TrigCompositeContainer>();
    auto aux = std::make_unique<xAOD::AuxContainerBase>();
    uniqueContainer->setStore(aux.get() );
    container = uniqueContainer.get();
    std::string name = m_outputPrefix+chain;
    // We have to replace '.' characters with '_' characters so that these are
    // valid container names...
    std::replace(name.begin(), name.end(), '.', '_');
    ATH_CHECK( evtStore()->record(std::move(uniqueContainer), name) );
    ATH_CHECK( evtStore()->record(std::move(aux), name+"Aux.") );
    return StatusCode::SUCCESS;
  }
 
  const TriggerMatchingTool::particleVec_t& TriggerMatchingTool::getCandidateMatchesFor(
      const xAOD::IParticle* part,
      std::map<xAOD::Type::ObjectType, particleVec_t>& offlineParticles,
      std::map<const xAOD::IParticle*, particleVec_t>& cache) const
  {
    // Build up all the possible matches between online and offline particles
    auto cacheItr = cache.find(part);
    if (cacheItr == cache.end() ) {
      xAOD::Type::ObjectType type = part->type();
      particleVec_t candidates;
      if (type == xAOD::Type::CaloCluster) {
        // If it's a calo cluster then we need to get the cluster from the
        // egamma types.
        static const constAcc_t<vecLink_t<xAOD::CaloClusterContainer>> acc_calo("caloClusterLinks");
        for (xAOD::Type::ObjectType egType : {
            xAOD::Type::Electron, xAOD::Type::Photon})
        {
          for (const xAOD::IParticle* cand : offlineParticles[egType]) {
            const vecLink_t<xAOD::CaloClusterContainer>& links = acc_calo(*cand);
            if (links.size() == 0 || !links.at(0).isValid() )
              continue;
            const xAOD::CaloCluster* clus = *links.at(0);
            if (matchParticles(part, clus) )
              candidates.push_back(cand);
          }
        }
      }
      else {
        for (const xAOD::IParticle* cand : offlineParticles[type])
          if (matchParticles(part, cand) )
            candidates.push_back(cand);
      }
      cacheItr = cache.emplace(
          std::make_pair(part, std::move(candidates) ) ).first;
    }
    return cacheItr->second;
  }
 
  bool TriggerMatchingTool::matchParticles(
      const xAOD::IParticle* lhs,
      const xAOD::IParticle* rhs) const
  {
    return m_scoreTool->score(*lhs, *rhs) < m_drThreshold;
  }
 
} //> end namespace DerivationFramework