TestStore.h

Go to the documentation of this file.

// This file's extension implies that it's C, but it's really -*- C++ -*-.
/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
#ifndef SGTOOLS_TESTSTORE_H
#define SGTOOLS_TESTSTORE_H
 
 
#include "AthenaKernel/IProxyDict.h"
#include "SGTools/StringPool.h"
#include "SGTools/DataProxy.h"
#include "SGTools/CurrentEventStore.h"
#include "CxxUtils/checker_macros.h"
#include <unordered_map>
#include <mutex>
 
 
namespace SGTest {
 
 
typedef SG::sgkey_t sgkey_t;
 
 
struct TestStoreRemap
{
  TestStoreRemap (SG::sgkey_t the_key=0, size_t the_index = 0)
    : key (the_key), index (the_index) {}
  bool operator== (const TestStoreRemap& other) const
  { return other.key == key && other.index == index; }
  SG::sgkey_t key;
  size_t index;
};
 
 
struct TestStoreRemapHash
{
  size_t operator() (const TestStoreRemap& m) const
  { return m.key + m.index; }
};
 
 
class TestStore
  : public implements<IProxyDict>
{
public:
  virtual std::vector<const SG::DataProxy*> proxies() const override;
  virtual const std::string* keyToString (sgkey_t /*key*/) const override;
  virtual void registerKey (sgkey_t /*key*/,
                            const std::string& /*str*/,
                            CLID /*clid*/) override;
 
  
  // These have dummy implementations.
  virtual const std::string& name() const override;
  virtual SG::DataProxy* proxy(const void* const pTransient) const override;
  virtual SG::DataProxy* proxy(const CLID& id, const std::string& key) const override;
  virtual SG::DataProxy* proxy_exact (SG::sgkey_t sgkey) const override;
  virtual sgkey_t stringToKey (const std::string& str, CLID clid) override;
  virtual const std::string* keyToString (sgkey_t key, CLID& clid) const override;
  virtual bool tryELRemap (sgkey_t sgkey_in, size_t index_in,
                           sgkey_t& sgkey_out, size_t& index_out) override;
  virtual StatusCode addToStore (CLID /*id*/, SG::DataProxy* proxy) override;
  virtual void boundHandle (IResetable* handle) override;
  virtual void unboundHandle (IResetable* handle) override;
 
  virtual SG::DataProxy* recordObject (SG::DataObjectSharedPtr<DataObject> obj,
                                       const std::string& key,
                                       bool allowMods,
                                       bool returnExisting) override;
 
 
  template <class T>
  void record (const T* p, const std::string& key)
  {
    lock_t lock (m_mutex);
    T* p_nc ATLAS_THREAD_SAFE = const_cast<T*>(p);
    DataObject* obj = SG::asStorable<T>(p_nc);
    CLID clid = ClassID_traits<T>::ID();
    record1 (lock, p, obj, clid, key);
  }
 
 
  template <class T>
  void record (std::unique_ptr<T> up, const std::string& key)
  {
    lock_t lock (m_mutex);
    T* p = up.get();
    DataObject* obj = SG::asStorable<T>(std::move (up));
    CLID clid = ClassID_traits<T>::ID();
    record1 (lock, p, obj, clid, key);
  }
 
 
  void remap (sgkey_t sgkey_in, sgkey_t sgkey_out,
              size_t index_in = 0, size_t index_out = 0);
 
 
  template <class T>
  void remap (const std::string& key_in, const std::string& key_out,
              size_t index_in = 0, size_t index_out = 0)
  {
    CLID clid = ClassID_traits<T>::ID();
    sgkey_t sgkey_in = stringToKey (key_in, clid);
    sgkey_t sgkey_out = stringToKey (key_out, clid);
    remap (sgkey_in, sgkey_out, index_in, index_out);
  }
 
  void alias (SG::DataProxy* proxy, const std::string& newKey);
 
  typedef std::unordered_map<const void*, SG::DataProxy*> tmap_t;
  tmap_t m_tmap;
 
  typedef SG::SGKeyMap<SG::DataProxy*> kmap_t;
  kmap_t m_kmap;
 
  typedef std::unordered_map<TestStoreRemap, TestStoreRemap, TestStoreRemapHash> remap_t;
  remap_t m_remap;
 
  mutable SG::StringPool m_stringPool ATLAS_THREAD_SAFE;
 
  std::vector<IResetable*> m_boundHandles;
 
  // Log failed calls to proxy(CLID, std::string).
  mutable std::vector<std::pair<CLID, std::string> > m_missedProxies ATLAS_THREAD_SAFE;
 
  typedef std::mutex mutex_t;
  typedef std::lock_guard<mutex_t> lock_t;
  mutable mutex_t m_mutex;
 
  SG::DataProxy* proxy(lock_t&, const void* const pTransient) const;
  SG::DataProxy* proxy(lock_t&, const CLID& id, const std::string& key) const;
  SG::DataProxy* record1 (lock_t&, const void* p, DataObject* obj,
                          CLID clid, const std::string& key);
};
 
 
// Initialize the global, static test store.
extern TestStore store;
void initTestStore ATLAS_NOT_THREAD_SAFE ();
 
 
// Create a new test store and install it.
std::unique_ptr<TestStore> getTestStore();
 
 
} // namespace SGTest
 
 
#endif // not SGTOOLS_TESTSTORE_H