From e32714c456236327269d6e46dfb10c67fa4280f3 Mon Sep 17 00:00:00 2001
From: Jason Turner <jason@emptycrate.com>
Date: Sat, 2 Jan 2016 19:45:10 -0700
Subject: [PATCH] Add some operators for Enums made with helper class

---
 include/chaiscript/utility/utility.hpp | 9 ++++++++-
 unittests/compiled_tests.cpp           | 5 +++++
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/include/chaiscript/utility/utility.hpp b/include/chaiscript/utility/utility.hpp
index 4dfd079..05c49c5 100644
--- a/include/chaiscript/utility/utility.hpp
+++ b/include/chaiscript/utility/utility.hpp
@@ -15,6 +15,7 @@
 #include "../chaiscript.hpp"
 #include "../dispatchkit/proxy_functions.hpp"
 #include "../dispatchkit/type_info.hpp"
+#include "../dispatchkit/operators.hpp"
 
 
 namespace chaiscript 
@@ -63,7 +64,7 @@ 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,
@@ -76,6 +77,12 @@ namespace chaiscript
         t_module.add(chaiscript::constructor<Enum ()>(), t_class_name);
         t_module.add(chaiscript::constructor<Enum (const Enum &)>(), t_class_name);
 
+        t_module.add([](){
+              // add some comparison and assignment operators
+              using namespace chaiscript::bootstrap::operators;
+              return assign<Enum>(not_equal<Enum>(equal<Enum>()));
+            }());
+
         for (const auto &constant : t_constants)
         {
           t_module.add_global_const(constant.first, constant.second);
diff --git a/unittests/compiled_tests.cpp b/unittests/compiled_tests.cpp
index ca5e9ec..a028526 100644
--- a/unittests/compiled_tests.cpp
+++ b/unittests/compiled_tests.cpp
@@ -562,6 +562,11 @@ TEST_CASE("Utility_Test utility class wrapper for enum")
   CHECK_NOTHROW(chai.eval("do_something_with_enum_vector([ONE])"));
   CHECK_NOTHROW(chai.eval("[ONE]"));
 
+  CHECK(chai.eval<bool>("ONE == ONE"));
+  CHECK(chai.eval<bool>("ONE != TWO"));
+  CHECK_NOTHROW(chai.eval("var o = ONE; o = TWO"));
+
+
 }