AsgUnionSelectionAlg.cxx

Go to the documentation of this file.

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

 
//
// includes
//
 
#include <AsgAnalysisAlgorithms/AsgUnionSelectionAlg.h>
 
#include <xAODBase/IParticle.h>
 
//
// method implementations
//
 
namespace CP
{
 
  StatusCode AsgUnionSelectionAlg ::
  initialize ()
  {
    if (m_selectionDecoration.empty())
    {
      ANA_MSG_ERROR("Selection decoration can not be empty.");
      return StatusCode::FAILURE;
    }
 
    ANA_CHECK (makeSelectionWriteAccessor (m_selectionDecoration, m_selectionAccessor));
 
    ANA_CHECK (m_particlesHandle.initialize(m_systematicsList));
    ANA_CHECK (m_preselection.initialize (m_systematicsList, m_particlesHandle));
    ANA_CHECK (m_systematicsList.initialize());
 
    return StatusCode::SUCCESS;
  }
 
 
 
  StatusCode AsgUnionSelectionAlg ::
  execute (const EventContext& ctx)
  {
    std::vector<bool> selections;
 
    // first loop through systematics and define if particle passes each of them
    for (const auto& sys : m_systematicsList.systematicsVector())
    {
      const xAOD::IParticleContainer *particles{};
      ANA_CHECK (m_particlesHandle.retrieve (particles, sys, ctx));
 
      if (selections.empty())
      {
        selections.resize(particles->size(), false);
      }
      else if (selections.size() != particles->size())
      {
        ANA_MSG_ERROR("All input containers should have the same size.");
        return StatusCode::FAILURE;
      }
 
      for (size_t i{}; i < particles->size(); i++)
      {
        selections[i] = selections[i] || m_preselection.getBool (*particles->at(i), sys);
      }
    }
 
    // Now decorate the selection decoration
    // Looping over systematics is needed to ensure all containers are processed
    for (const auto& sys : m_systematicsList.systematicsVector())
    {
      const xAOD::IParticleContainer *particles{};
      ANA_CHECK (m_particlesHandle.retrieve (particles, sys, ctx));
 
      for (size_t i{}; i < particles->size(); i++)
      {
        m_selectionAccessor->setBool (*particles->at(i), selections[i]);
      }
    }
 
    return StatusCode::SUCCESS;
  }
 
} // namespace CP