type_map.h

Go to the documentation of this file.

 
#ifndef TYPE_MAP_H
#define TYPE_MAP_H
 
#include <unordered_map>
#include <memory>
#include <typeindex>
 
template <class element_type>
class type_map {
  private:
    std::unordered_map<std::type_index, std::unique_ptr<element_type>> table;
  public:
    template <typename T>
    void add (const T &input) {
      static_assert(std::is_base_of<element_type, T>::value,
                    "T must be derived from element_type.");
      table[std::type_index(typeid(T))] = std::make_unique<T>(input);
    }
 
    template <typename T>
    T& get () {
      static_assert(std::is_base_of<element_type, T>::value,
                    "T must be derived from element_type.");
      auto it = table.find(std::type_index(typeid(T)));
      if (it != table.end()) {
          return *static_cast<T*>(it->second.get());
      } else {
          throw std::runtime_error("Element type not found.");
      }
    }
};
 
#endif // TYPE_MAP_H