From f0796b51c8541319378bfa53e180b9ab73be14be Mon Sep 17 00:00:00 2001
From: Viktor Rennert <vrennert@squarepx.com>
Date: Sat, 2 Jan 2016 14:17:36 +0100
Subject: [PATCH] Added template specialization in
 chaiscript::utility::add_class<Enum> to register bulk constants.

---
 include/chaiscript/utility/utility.hpp | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/include/chaiscript/utility/utility.hpp b/include/chaiscript/utility/utility.hpp
index ecd5a3f..2704929 100644
--- a/include/chaiscript/utility/utility.hpp
+++ b/include/chaiscript/utility/utility.hpp
@@ -8,6 +8,7 @@
 #define CHAISCRIPT_UTILITY_UTILITY_HPP_
 
 #include <string>
+#include <type_traits>
 #include <utility>
 #include <vector>
 
@@ -62,6 +63,21 @@ namespace chaiscript
           t_module.add(fun.first, fun.second);
         }
       }
+      
+    template<typename Enum, typename ModuleType>
+      typename std::enable_if<std::is_enum<Enum>::value, void>::type
+      add_class(ModuleType &t_module,
+        const std::string &t_class_name,
+        const std::vector<chaiscript::Proxy_Function> &t_constructors,
+        const std::vector<std::pair<chaiscript::Boxed_Value, std::string>> &t_constants)
+      {
+        t_module.add(chaiscript::user_type<Enum>(), t_class_name);
+  
+        for (const auto &constant : t_constants)
+        {
+          t_module.add_global_const(constant.first, constant.second);
+        }
+      }
   }
 }