ContainedMultiAssociationFillerTool.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
// $Id: ContainedMultiAssociationFillerTool.cxx 618070 2014-09-22 19:05:34Z ssnyder $
#include "ContainedMultiAssociationFillerTool.h"
#include "AthenaKernel/errorcheck.h"
 
 
namespace D3PD {
 
 
ContainedMultiAssociationFillerTool::ContainedMultiAssociationFillerTool
   (const std::string& type,
    const std::string& name,
    const IInterface* parent)
     : base_class (type, name, parent),
       AddVariable (m_prefix, m_blockName),
       m_associator (this),
       m_blockFillers (this),
       m_next (0)
{
  declareProperty ("Associator",   m_associator,
                   "The IMultiAssociationTool instance.");
  declareProperty ("BlockFillers", m_blockFillers,
                   "List of contained block filler tools.");
  declareProperty ("BlockName", m_blockName = this->name(),
                   "The name of this block.");
  declareProperty ("Prefix",       m_prefix = "",
                   "Variable name prefix for the contained blocks.");
}
 
 
StatusCode
ContainedMultiAssociationFillerTool::initialize()
{
  CHECK( AthAlgTool::initialize() );
  CHECK( m_associator.retrieve() );
  CHECK( m_blockFillers.retrieve() );
  return StatusCode::SUCCESS;
}
 
 
StatusCode
ContainedMultiAssociationFillerTool::finalize()
{
  CHECK( AlgTool::finalize() );
  if (m_next) {
    m_associator->releaseElementUntyped (m_next);
    m_next = 0;
  }
  return StatusCode::SUCCESS;
}
 
 
StatusCode
ContainedMultiAssociationFillerTool::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->elementTypeinfo();
  for (size_t i = 0; i < m_blockFillers.size(); i++)
    CHECK( m_blockFillers[i]->configureD3PD (this, ati ) );
 
  return StatusCode::SUCCESS;
}
 
 
StatusCode ContainedMultiAssociationFillerTool::book()
{
  CHECK( m_associator->book () );
  for (size_t i = 0; i < m_blockFillers.size(); i++)
    CHECK( m_blockFillers[i]->book ( ) );
 
  return StatusCode::SUCCESS;
}
 
 
StatusCode
ContainedMultiAssociationFillerTool::fillUntyped (const void* p,
                                                  bool again /*= false*/)
{
  // ??? Doesn't handle nested AGAIN.
 
  if (!again) {
    if (m_next) {
      m_associator->releaseElementUntyped (m_next);
      m_next = 0;
    }
 
    // Start the association.
    CHECK( m_associator->resetUntyped (p) );
 
    // Get the first result.
    m_next = m_associator->nextUntyped();
  }
 
  if (!m_next)
    return StatusCode(EMPTY);
 
  for (size_t i = 0; i < m_blockFillers.size(); i++) {
    StatusCode sc = m_blockFillers[i]->fillUntyped (m_next);
    CHECK( IBlockFillerTool::againok (sc) );
    if (!sc.isSuccess()) {
      REPORT_MESSAGE (MSG::WARNING)
        << "Got AGAIN/EMPTY: "
        << "Nested ContainedMultiAssociationFillerTool not supported.";
    }
  }
 
  m_associator->releaseElementUntyped (m_next);
  m_next = m_associator->nextUntyped();
 
  // Always return AGAIN for the first object, even if it's the last one.
  // We'll return EMPTY on the next call.
  // If we don't do this, then we can have problems if multiple
  // ContainedMultiAssociationFillerTools are used; any that return
  // a list of size 1 will have that object duplicated in all rows.
  if (m_next || !again)
    return StatusCode(AGAIN);
 
  return StatusCode::SUCCESS;
}
 
 
 
} // namespace D3PD