AsgFlagSelectionTool.cxx

Go to the documentation of this file.

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

 
 
 
//
// includes
//
 
#include <AsgAnalysisAlgorithms/AsgFlagSelectionTool.h>
 
//
// method implementations
//
 
namespace CP
{
 
    StatusCode AsgFlagSelectionTool ::
    initialize ()
    {
      if(m_invertFlags.size()!=m_selFlags.size() && !m_invertFlags.empty()) {
          ATH_MSG_ERROR("Property invertFlags has different size to selectionFlags. Please check your configuration");
          return StatusCode::FAILURE;
      }
      // Could also warn if there are fewer values, but we don't have to force users to set where irrelevant.
      // Maybe warn unless the size is 0, in which case assume all default?
      if (m_invertFlags.empty()) {
          m_invertFlags = std::vector<bool>(m_selFlags.size(), false);
      }
 
      for(size_t index=0; index<m_selFlags.size(); ++index) {
          const std::string& thisflag = m_selFlags[index];
          if (thisflag.empty()) {
              ATH_MSG_ERROR("Empty string passed as selection flag!");
              return StatusCode::FAILURE;
          } else {
              std::string doInvertStr = m_invertFlags[index] ? "!" : "";
              m_accept.addCut (doInvertStr + thisflag, doInvertStr + thisflag);
              std::unique_ptr<ISelectionReadAccessor> accessor;
              ATH_CHECK (makeSelectionReadAccessor (thisflag, accessor, true));
              m_acc_selFlags.push_back (std::move (accessor));
          }
      }
 
      return StatusCode::SUCCESS;
    }
 
 
 
    const asg::AcceptInfo& AsgFlagSelectionTool ::
    getAcceptInfo () const
    {
      return m_accept;
    }
 
 
 
    asg::AcceptData AsgFlagSelectionTool ::
    accept (const xAOD::IParticle *particle) const
    {
      asg::AcceptData accept (&m_accept);
      for(std::size_t cutIndex=0; cutIndex<m_accept.getNCuts(); ++cutIndex) {
          // Test against the opposite of the invert value
          bool testval = !m_invertFlags[cutIndex];
          ATH_MSG_VERBOSE("Now testing flag \"" << m_selFlags[cutIndex] << "\" requiring value " << testval);
          accept.setCutResult (cutIndex, m_acc_selFlags[cutIndex]->getBool (*particle)==testval);
      }
 
      return accept;
    }
}