TrigPrescalesAlg.cxx

Go to the documentation of this file.

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

 
 
//
// includes
//
 
#include <TriggerAnalysisAlgorithms/TrigPrescalesAlg.h>
 
#include <algorithm>
#include <xAODEventInfo/EventInfo.h>
#include <format>
//
// method implementations
//
 
namespace CP
{
 
  StatusCode TrigPrescalesAlg ::
  initialize ()
  {
    if (m_prescaleDecoration.empty())
    {
      ANA_MSG_ERROR ("The prescale decoration should not be empty");
      return StatusCode::FAILURE;
    }
 
    if (m_trigList.empty() && m_trigFormula.empty())
    {
      ANA_MSG_ERROR ("Either a list of triggers or trigger formula need to be provided");
      return StatusCode::FAILURE;
    }
 
    if (!m_trigList.empty() && !m_trigFormula.empty())
    {
      ANA_MSG_ERROR ("Provide either only a list of triggers or only a trigger formula");
      return StatusCode::FAILURE;
    }
 
    ANA_CHECK (m_pileupReweightingTool.retrieve());
 
    if (!m_selectionDecoration.empty()) {
      for (const std::string &chain : m_trigList) {
        std::string chainfix = chain;
        std::replace(chainfix.begin(), chainfix.end(), '.', 'p');
        std::replace(chainfix.begin(), chainfix.end(), '-', '_');
        m_selectionAccessors.emplace(chain, std::format("{}_{}", m_selectionDecoration.value(), chainfix));
      }
    }
 
    if (!m_trigFormula.empty())
    {
      m_prescaleAccessors.emplace_back(m_prescaleDecoration);
      m_prescaleFunctions.emplace_back([this](const xAOD::EventInfo *evtInfo, const std::string &trigger)
      {
        if(m_prescaleMC) return m_pileupReweightingTool->getPrescaleWeight(*evtInfo, trigger, true);
        return m_pileupReweightingTool->getDataWeight (*evtInfo, trigger, true);
      });
      // By putting the formula into` m_trigListAll`
      // the logic in `execute` does not have to change
      // depending on if `m_trigFormula` or `m_trigList` is used
      std::vector<std::string> formulaVector = {m_trigFormula.value()};
      m_trigListAll = std::move(formulaVector);
      return StatusCode::SUCCESS;
    }
 
    if (m_trigListAll.empty())
    {
      m_trigListAll = m_trigList;
    }
    const std::string prefix = m_prescaleDecoration + "_";
    for (const std::string &chain : m_trigListAll)
    {
      std::string chainfix = chain;
      std::replace(chainfix.begin(), chainfix.end(), '.', 'p');
      std::replace(chainfix.begin(), chainfix.end(), '-', '_');
      m_prescaleAccessors.emplace_back(prefix + chainfix);
 
      // Generate helper functions
      if (std::find(m_trigList.begin(), m_trigList.end(), chain) != m_trigList.end())
      {
        m_prescaleFunctions.emplace_back([this](const xAOD::EventInfo *evtInfo, const std::string &trigger)
        {
          auto it = m_selectionAccessors.find(trigger);
          if (it != m_selectionAccessors.end())
          {
            if (!it->second(*evtInfo))
            {
              return invalidTriggerPrescale();
            }
          }
 
          if(m_prescaleMC) return m_pileupReweightingTool->getPrescaleWeight(*evtInfo, trigger, true);
          return m_pileupReweightingTool->getDataWeight (*evtInfo, trigger, true);
        });
      }
      else
      {
        m_prescaleFunctions.emplace_back([](const xAOD::EventInfo *, const std::string &)
        {
          return invalidTriggerPrescale();
        });
      }
    }
 
 
    return StatusCode::SUCCESS;
  }
 
 
 
  StatusCode TrigPrescalesAlg ::
  execute (const EventContext& /*ctx*/)
  {
    const xAOD::EventInfo *evtInfo{};
    ANA_CHECK (evtStore()->retrieve(evtInfo, "EventInfo"));
 
    for (size_t i = 0; i < m_trigListAll.size(); i++)
    {
      (m_prescaleAccessors[i]) (*evtInfo) = (m_prescaleFunctions[i]) (evtInfo, m_trigListAll[i]);
    }
 
    return StatusCode::SUCCESS;
  }
}