SelectedParticlesFilterTool.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
// $Id: SelectedParticlesFilterTool.cxx 486128 2012-03-02 08:54:20Z krasznaa $
#include "SelectedParticlesFilterTool.h"
#include "ParticleEvent/SelectedParticles.h"
#include "AthenaKernel/errorcheck.h"
 
 
namespace D3PD {
 
 
SelectedParticlesFilterTool::SelectedParticlesFilterTool
  (const std::string& type,
   const std::string& name,
   const IInterface* parent)
    : CollectionGetterFilterToolImpl (type, name, parent),
      m_selgetter (this),
      m_selection (0),
      m_index (0)
{
  declareProperty ("SelectionGetter", m_selgetter,
                   "Getter for the SelectedParticles object.");
}
 
 
StatusCode SelectedParticlesFilterTool::initialize()
{
  CHECK( CollectionGetterFilterToolImpl::initialize() );
  CHECK( m_selgetter.retrieve() );
  return StatusCode::SUCCESS;
}
 
 
const std::type_info& SelectedParticlesFilterTool::elementTypeinfo() const
{
  return m_getter->elementTypeinfo();
}
 
 
StatusCode SelectedParticlesFilterTool::reset (bool allowMissing /*= false*/)
{
  m_index = 0;
  if (m_selection) {
    m_selgetter->releaseObject (m_selection);
    m_selection = 0;
  }
  CHECK( CollectionGetterFilterToolImpl::reset (allowMissing) );
  m_selection = m_selgetter->get<SelectedParticles> (allowMissing);
  return StatusCode::SUCCESS;
}
 
 
const void* SelectedParticlesFilterTool::nextUntyped()
{
  // Loop until the filter passes.
  while (const void* p = m_getter->nextUntyped()) {
    ++m_index;
    if (!m_selection || m_selection->isGood (m_index-1))
      return p;
    else 
      m_getter->releaseElementUntyped (p);
  }
  if (m_selection) {
    m_selgetter->releaseObject (m_selection);
    m_selection = 0;
  }
  return 0;
}
 
 
void SelectedParticlesFilterTool::releaseElementUntyped (const void* p)
{
  return m_getter->releaseElementUntyped (p);
}
 
 
} // namespace D3PD