ContainedAssociationFillerTool.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
// $Id: ContainedAssociationFillerTool.cxx 618070 2014-09-22 19:05:34Z ssnyder $
#include "ContainedAssociationFillerTool.h"
#include "AthenaKernel/errorcheck.h"
 
 
namespace D3PD {
 
 
ContainedAssociationFillerTool::ContainedAssociationFillerTool
   (const std::string& type,
    const std::string& name,
    const IInterface* parent)
     : base_class (type, name, parent),
       AddVariable (m_prefix, m_blockName, m_suffix),
       m_associator (this),
       m_blockFillers (this),
       m_matched (0)
{
  declareProperty ("Associator",   m_associator,
                   "The ISingleAssociationTool instance.");
  declareProperty ("BlockFillers", m_blockFillers,
                   "List of contained block filler tools.");
  declareProperty ("Prefix",       m_prefix = "",
                   "Variable name prefix for the contained blocks.");
  declareProperty ("Suffix",       m_suffix = "",
                   "Variable name suffix for the contained blocks.");
  declareProperty ("BlockName", m_blockName = this->name(),
                   "The name of this block.");
  declareProperty ("Matched",      m_matchedName = "matched",
                   "Variable name to use for the matched flag.  "
                   "Omitted if empty.");
}
 
 
StatusCode
ContainedAssociationFillerTool::initialize()
{
  CHECK( AthAlgTool::initialize() );
  CHECK( m_associator.retrieve() );
  CHECK( m_blockFillers.retrieve() );
  return StatusCode::SUCCESS;
}
 
 
StatusCode
ContainedAssociationFillerTool::configureD3PD (IAddVariable* tree,
                                               const std::type_info& ti)
{
  CHECK( AddVariable::configureD3PD (tree) );
  CHECK( m_associator->configureD3PD (this, ti) );
  const std::type_info& ati = m_associator->typeinfo();
  for (size_t i = 0; i < m_blockFillers.size(); i++)
    CHECK( m_blockFillers[i]->configureD3PD (this, ati ) );
 
  return StatusCode::SUCCESS;
}
 
 
StatusCode ContainedAssociationFillerTool::book()
{
  CHECK( m_associator->book () );
  for (size_t i = 0; i < m_blockFillers.size(); i++)
    CHECK( m_blockFillers[i]->book ( ) );
 
  if (!m_matchedName.empty())
    CHECK( addVariable (m_matchedName, m_matched) );
 
  return StatusCode::SUCCESS;
}
 
 
StatusCode
ContainedAssociationFillerTool::fillUntyped (const void* p,
                                             bool again /*= false*/)
{
  const void* pout = m_associator->getUntyped (p);
  StatusCode scout (EMPTY);
  if (pout) {
    if (!m_matchedName.empty())
      *m_matched = true;
    for (size_t i = 0; i < m_blockFillers.size(); i++) {
      // Set checked flag on scout.
      // Otherwise, if we get an exception in the child, all we'll see
      // is an unchecked status code error with no mention of the
      // actual exception...
      scout.ignore();
      CHECK( againok (m_blockFillers[i]->fillUntyped (pout, again), scout) );
    }
  }
  m_associator->releaseObjectUntyped (pout);
 
  scout = StatusCode(scout.getCode()); // Make sure checked flag is clear again.
  return againok (scout);
}
 
 
 
} // namespace D3PD