ContainerFlagFillerTool.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
// $Id$
#include "ContainerFlagFillerTool.h"
#include "AthenaKernel/errorcheck.h"
 
 
namespace D3PD {
 
 
ContainerFlagFillerTool::ContainerFlagFillerTool (const std::string& type,
                                                  const std::string& name,
                                                  const IInterface* parent)
  : BlockFillerToolImpl (type, name, parent),
    m_getters (this),
    m_map (this, m_getters, m_allowMissing)
{
  declareProperty ("FlagNames", m_flagNames,
                   "Names of the flags added to the D3PD."
"May be followed by a string of the form `@TYPE/KEY', in which case the "
"corresponding entry in Getters is automatically initialized (only available "
"when the Python configuration wrapper is used).  May be followed by a "
"docstring following a colon.");
  declareProperty ("Getters", m_getters,
                   "Getter tools for the INav4MomLinkContainer objects.");
  declareProperty ("AllowMissing", m_allowMissing,
                   "If true, missing containers will not be considered errors.");
}
 
 
StatusCode ContainerFlagFillerTool::initialize()
{
  // Check that the configuration makes sense:
  if( m_getters.size() != m_flagNames.size() ) {
    REPORT_MESSAGE( MSG::FATAL )
      << "Different number of getter tools and flags specified";
    return StatusCode::FAILURE;
  }
 
  CHECK( BlockFillerToolImpl::initialize() );
 
  return StatusCode::SUCCESS;
}
 
 
StatusCode ContainerFlagFillerTool::configureD3PD (IAddVariable* tree,
                                                   const std::type_info& ti)
{
  CHECK( m_map.configureD3PD (ti) );
  CHECK( AddVariable::configureD3PD (tree) );
  return StatusCode::SUCCESS;
}
 
 
StatusCode ContainerFlagFillerTool::book()
{
  // Resize the flag vector to the right size:
  m_flags.resize( m_flagNames.size(), 0 );
 
  // Book the flag variable(s):
  for( size_t i = 0; i < m_flagNames.size(); ++i ) {
    std::string flagname = m_flagNames[i];
    std::string desc = "Was this object selected according to X criteria?";
    std::string::size_type ipos = flagname.find (':');
    if (ipos != std::string::npos) {
      desc = flagname.substr (ipos+1, std::string::npos);
      flagname.resize (ipos);
    }
    CHECK( addVariable (flagname, m_flags[ i ], desc) );
  }
 
  return StatusCode::SUCCESS;
}
 
 
StatusCode ContainerFlagFillerTool::fillUntyped (const void* p,
                                                 bool /*again = false*/)
{
  if (m_flags.empty())
    return StatusCode(EMPTY);
 
  CHECK( m_map.reset() );
  for (size_t i = 0; i < m_flags.size(); i++)
    *m_flags[i] = (m_map.findIndex (p, i) != -1);
  return StatusCode::SUCCESS;
}
 
 
} // namespace D3PD