Add utility functions for getting an item from an std::map

[libs/core.git] / source / core / maputils.h

diff --git a/source/core/maputils.h b/source/core/maputils.h
new file mode 100644 (file)
index 0000000..1a7c0ae
--- /dev/null
+++ b/source/core/maputils.h
@@ -0,0 +1,38 @@
+#ifndef MSP_CORE_MAPUTILS_H_
+#define MSP_CORE_MAPUTILS_H_
+
+#include <stdexcept>
+#include <typeinfo>
+
+namespace Msp {
+
+class key_error: public std::runtime_error
+{
+public:
+       key_error(const std::type_info &);
+};
+
+
+template<typename T>
+typename T::mapped_type &get_item(T &map, const typename T::key_type &key)
+{
+       typename T::iterator i = map.find(key);
+       if(i==map.end())
+               throw key_error(typeid(T));
+
+       return i->second;
+}
+
+template<typename T>
+const typename T::mapped_type &get_item(const T &map, const typename T::key_type &key)
+{
+       typename T::const_iterator i = map.find(key);
+       if(i==map.end())
+               throw key_error(typeid(T));
+
+       return i->second;
+}
+
+} // namespace Msp
+
+#endif