TestStore.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
 
#include "SGTools/TestStore.h"
#include "AthenaKernel/DataBucketBase.h"
#include "AthenaKernel/proxyDictFromEventContext.h"
#include <print>
#include <algorithm>
#include <iostream>
 
 
namespace SGTest {
 
 
TestStore store;
 
 
TestStore::TestStore (StoreID::type id /*= StoreID::EVENT_CONTEXT*/)
  : m_id (id)
{
}
 
 
StoreID::type TestStore::storeID() const
{
  return m_id;
}
 
 
std::vector<const SG::DataProxy*> TestStore::proxies() const
{
  std::println ("proxies"); std::abort();
}
 
 
const std::string* TestStore::keyToString (sgkey_t key) const
{
  return m_stringPool.keyToString (key);
}
 
 
void TestStore::registerKey (sgkey_t /*key*/,
                             const std::string& /*str*/,
                             CLID /*clid*/)
{
  std::println ("registerKey"); std::abort();
}
 
 
SG::DataProxy* TestStore::recordObject (SG::DataObjectSharedPtr<DataObject> obj,
                                        const std::string& key,
                                        bool allowMods,
                                        bool returnExisting)
{
  lock_t lock (m_mutex);
  const void* raw_ptr = obj.get();
  if (DataBucketBase* bucket = dynamic_cast<DataBucketBase*> (obj.get())) {
    raw_ptr = bucket->object();
  }
 
  CLID clid = obj->clID();
  SG::DataProxy* proxy = this->proxy (lock, clid, key);
  if (proxy) {
    if (returnExisting)
      return proxy;
    else
      return nullptr;
  }
 
  proxy = this->proxy (lock, raw_ptr);
  if (proxy) {
    sgkey_t sgkey = m_stringPool.stringToKey (key, obj->clID());
    if (!returnExisting)
      return nullptr;
    if (obj->clID() == proxy->clID()) {
      // Alias?
      m_kmap[sgkey] = proxy;
      proxy->addRef();
      proxy->setAlias (key);
      return proxy;
    }
    if (key == proxy->name()) {
      // Symlink?
      m_kmap[sgkey] = proxy;
      proxy->addRef();
      proxy->setTransientID (obj->clID());
      return proxy;
    }
 
    // Error.
    return nullptr;
  }
  
  proxy = record1 (lock, raw_ptr, obj.get(), clid, key);
  if (!allowMods)
    proxy->setConst();
  return proxy;
}
 
 
const std::string& TestStore::name() const
{
  static const std::string nm = "TestStore";
  return nm;
}
 
 
SG::DataProxy* TestStore::proxy(const void* const pTransient) const
{
  lock_t lock (m_mutex);
  return proxy (lock, pTransient);
}
 
 
SG::DataProxy*
TestStore::proxy(lock_t&, const void* const pTransient) const
{ 
  tmap_t::const_iterator i = m_tmap.find (pTransient);
  if (i != m_tmap.end())
    return i->second;
  return 0;
}
 
 
SG::DataProxy* TestStore::proxy(const CLID& id, const std::string& key) const
{
  lock_t lock (m_mutex);
  return proxy (lock, id, key);
}
 
 
SG::DataProxy*
TestStore::proxy(lock_t&, const CLID& id, const std::string& key) const
{
  sgkey_t sgkey = m_stringPool.stringToKey (key, id);
  kmap_t::const_iterator i = m_kmap.find (sgkey);
  if (i != m_kmap.end())
    return i->second;
  m_missedProxies.emplace_back (id, key);
  return 0;
}
 
 
SG::DataProxy* TestStore::proxy_exact (SG::sgkey_t sgkey) const 
{
  lock_t lock (m_mutex);
  kmap_t::const_iterator i = m_kmap.find (sgkey);
  if (i != m_kmap.end())
    return i->second;
  return 0;
}
 
 
sgkey_t TestStore::stringToKey (const std::string& str, CLID clid)
{
  lock_t lock (m_mutex);
  return m_stringPool.stringToKey (str, clid);
}
 
 
const std::string* TestStore::keyToString (sgkey_t key, CLID& clid) const
{
  lock_t lock (m_mutex);
  return m_stringPool.keyToString (key, clid);
}
 
 
bool TestStore::tryELRemap (sgkey_t sgkey_in, size_t index_in,
                            sgkey_t& sgkey_out, size_t& index_out)
{
  lock_t lock (m_mutex);
  remap_t::iterator i = m_remap.find (TestStoreRemap (sgkey_in, index_in));
  if (i == m_remap.end()) return false;
  sgkey_out = i->second.key;
  index_out = i->second.index;
  return true;
}
 
 
StatusCode TestStore::addToStore (CLID /*id*/, SG::DataProxy* proxy)
{
  lock_t lock (m_mutex);
  proxy->setStore (this);
  m_kmap[proxy->sgkey()] = proxy;
  proxy->addRef();
  return StatusCode::SUCCESS;
}
 
 
void TestStore::boundHandle (IResetable* handle)
{
  lock_t lock (m_mutex);
  m_boundHandles.push_back (handle);
}
 
 
void TestStore::unboundHandle (IResetable* handle)
{
  lock_t lock (m_mutex);
  std::vector<IResetable*>::iterator it =
    std::find (m_boundHandles.begin(), m_boundHandles.end(), handle);
  if (it != m_boundHandles.end())
    m_boundHandles.erase (it);
}
 
 
SG::DataProxy* TestStore::record1 (lock_t&, const void* p, DataObject* obj,
                                   CLID clid, const std::string& key)
{
  sgkey_t sgkey = m_stringPool.stringToKey (key, clid);
  if (m_kmap.find (sgkey) != m_kmap.end()) {
    SG::DataProxy* dp = m_kmap[sgkey];
    dp->setObject (obj);
    if (dp->clID() == CLID_NULL)
      dp->setID (clid, key);
    m_tmap[p] = dp;
    return dp;
  }
 
  SG::DataProxy* dp = new SG::DataProxy(obj, SG::TransientAddress(clid, key));
  dp->setStore (this);
  m_tmap[p] = dp;
 
  m_kmap[sgkey] = dp;
  dp->addRef();
  dp->setSGKey (sgkey);
  return dp;
}
 
 
void TestStore::remap (sgkey_t sgkey_in, sgkey_t sgkey_out,
                       size_t index_in, size_t index_out)
{
  lock_t lock (m_mutex);
  m_remap[TestStoreRemap(sgkey_in, index_in)] =
    TestStoreRemap(sgkey_out, index_out);
}
 
 
void TestStore::alias (SG::DataProxy* proxy,
                       const std::string& newKey)
{
  lock_t lock (m_mutex);
  sgkey_t sgkey = m_stringPool.stringToKey (newKey, proxy->clID());
  m_kmap[sgkey] = proxy;
  proxy->addRef();
  proxy->setAlias (newKey);
}
 
 
void initTestStore ATLAS_NOT_THREAD_SAFE ()
{
  SG::CurrentEventStore::setStore (&store);
  Atlas::setProxyDictInEventContext (&store);
}
 
 
std::unique_ptr<TestStore> getTestStore()
{
  auto store = std::make_unique<TestStore>();
  SG::CurrentEventStore::setStore (store.get());
  Atlas::setProxyDictInEventContext (store.get());
  return store;
}
 
 
} // namespace SGTest