TrigPrescalesAlg.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
 
 
//
// includes
//
 
#include <TriggerAnalysisAlgorithms/TrigPrescalesAlg.h>
 
#include <RootCoreUtils/StringUtil.h>
#include <xAODEventInfo/EventInfo.h>
 
//
// method implementations
//
 
namespace CP
{
 
  StatusCode TrigPrescalesAlg ::
  initialize ()
  {
    if (m_prescaleDecoration.empty())
    {
      ANA_MSG_ERROR ("The 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_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 = formulaVector;
      return StatusCode::SUCCESS;
    }
 
    if (m_trigListAll.empty())
    {
      m_trigListAll = m_trigList;
    }
 
    for (const std::string &chain : m_trigListAll)
    {
      m_prescaleAccessors.emplace_back(m_prescaleDecoration + "_" + RCU::substitute (chain, "-", "_"));
 
      // 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)
        {
          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 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;
  }
}