type_map.h

Go to the documentation of this file.

#pragma once
// get type info for each storage object
#include <typeindex>
// store objects
#include <unordered_map>

template <class GenericObjectT> class TypeMap {
  std::unordered_map<std::type_index, GenericObjectT *> object_storage_map_;

public:
  template <class ObjectT> void setObject(ObjectT &object) {
    object_storage_map_[typeid(ObjectT)] = &object;
  }
  template <class ObjectT> ObjectT *getObject() {
    return static_cast<ObjectT *>(object_storage_map_[typeid(ObjectT)]);
  }

  template <class ObjectT> const ObjectT *getObject() const {
    return (
        static_cast<const ObjectT *>(object_storage_map_.at(typeid(ObjectT))));
  }
};