CollectionGetterRegistryTool.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#include "CollectionGetterRegistryTool.h"
#include "D3PDMakerInterfaces/ICollectionGetterTool.h"
#include "AthenaKernel/errorcheck.h"
#include "GaudiKernel/IToolSvc.h"
#include "CxxUtils/starts_with.h"
#include <sstream>
 
 
namespace D3PD {
 
 
CollectionGetterRegistryTool::CollectionGetterRegistryTool
  (const std::string& type,
   const std::string& name,
   const IInterface* parent)
    : base_class (type, name, parent),
      m_jos       ("JobOptionsSvc", name),
      m_toolsvc   ("ToolSvc",       name)
{
  declareProperty ("JobOptionsSvc", m_jos,
                   "The JobOptionsSvc instance.");
  declareProperty ("ToolSvc",       m_toolsvc,
                   "The ToolSvc instance.");
}
 
 
StatusCode
CollectionGetterRegistryTool::initialize()
{
  CHECK( AthAlgTool::initialize() );
  CHECK( m_jos.retrieve() );
  CHECK( m_toolsvc.retrieve() );
  return StatusCode::SUCCESS;
}
 
 
StatusCode CollectionGetterRegistryTool::add (const std::string& label,
                                              ICollectionGetterTool* tool)
{
  std::pair<map_t::iterator, bool> res =
    m_collection_map.insert (std::make_pair (label, tool));
  if (!res.second) {
    REPORT_MESSAGE_WITH_CONTEXT (MSG::ERROR,
                                 "D3PD::CollectionGetterRegistryTool")
      << "Duplicate collection label " << label
      << " for tools " << res.first->second->name()
      << " and " << tool->name();
    return StatusCode::FAILURE;
  }
  return StatusCode::SUCCESS;
}
 
 
StatusCode
CollectionGetterRegistryTool::get (const std::string& label,
                                   const INamedInterface* parent,
                                   ICollectionGetterTool*& tool)
{
  // Loop up the tool in the map.
  tool = 0;
  map_t::iterator i = m_collection_map.find (label);
  if (i == m_collection_map.end()) {
    REPORT_MESSAGE_WITH_CONTEXT (MSG::WARNING,
                                 "D3PD::CollectionGetterRegistryTool")
      << "Can't find collection label " << label;
    std::ostringstream os;
    os << "  Known collections:";
    for (const auto& p : m_collection_map) {
      os << " " << p.first;
    }
    REPORT_MESSAGE_WITH_CONTEXT (MSG::WARNING,
                                 "D3PD::CollectionGetterRegistryTool")
      << os.str();
    return StatusCode::SUCCESS;
  }
 
  // Get the properties for the source tool.
  const auto& props = m_jos->items([&i](const auto& p) {
    return CxxUtils::starts_with(std::get<0>(p), i->second->name()+".");
  });
 
  // Copy them to the destination tool (except for Label).
  std::string fullname = parent->name() + "." + label;
  for (const auto& p : props) {
    const std::string& oldname = std::get<0>(p);
    std::string::size_type ipos = oldname.rfind ('.');
    if (ipos != std::string::npos) {
      std::string pname = oldname.substr (ipos, std::string::npos);
      if (pname != ".Label") {
        m_jos->set(fullname + pname, std::get<1>(p));
      }
    }
  }
 
  // Create the new tool.
  return m_toolsvc->retrieveTool (i->second->type(), label, tool, parent);
}
 
 
} // namespace D3PD