1 files changed, 75 insertions, 0 deletions
diff --git a/libshiboken/sbkenum.cpp b/libshiboken/sbkenum.cpp
index e128a059a..e05f5cdad 100644
--- a/libshiboken/sbkenum.cpp
+++ b/libshiboken/sbkenum.cpp
@@ -21,6 +21,7 @@
  */
 
 #include "sbkenum.h"
+#include "basewrapper.h"
 #include <string.h>
 #include <cstring>
 #include <list>
@@ -175,6 +176,80 @@ PyObject* getEnumItemFromValue(PyTypeObject* enumType, long itemValue)
     return 0;
 }
 
+static PyTypeObject* createEnum(const char* fullName, const char* cppName, PyTypeObject* flagsType)
+{
+    PyTypeObject* enumType = newTypeWithName(fullName, cppName);
+    if (flagsType) {
+        if (PyType_Ready(flagsType) < 0)
+            return 0;
+        enumType->tp_as_number = flagsType->tp_as_number;
+    }
+    if (PyType_Ready(enumType) < 0)
+        return 0;
+    return enumType;
+}
+
+PyTypeObject* createGlobalEnum(PyObject* module, const char* name, const char* fullName, const char* cppName, PyTypeObject* flagsType)
+{
+    PyTypeObject* enumType = createEnum(fullName, cppName, flagsType);
+    if (enumType) {
+        if (PyModule_AddObject(module, name, (PyObject*)enumType) < 0)
+            return 0;
+    }
+    if (flagsType) {
+        if (PyModule_AddObject(module, flagsType->tp_name, (PyObject*)flagsType) < 0)
+            return 0;
+    }
+    return enumType;
+}
+
+PyTypeObject* createScopedEnum(SbkObjectType* scope, const char* name, const char* fullName, const char* cppName, PyTypeObject* flagsType)
+{
+    PyTypeObject* enumType = createEnum(fullName, cppName, flagsType);
+    if (enumType) {
+        if (PyDict_SetItemString(scope->super.ht_type.tp_dict, name, (PyObject*)enumType) < 0)
+            return 0;
+    }
+    if (flagsType) {
+        if (PyDict_SetItemString(scope->super.ht_type.tp_dict, flagsType->tp_name, (PyObject*)flagsType) < 0)
+            return 0;
+    }
+    return enumType;
+}
+
+static PyObject* createEnumItem(PyTypeObject* enumType, const char* itemName, long itemValue)
+{
+    PyObject* enumItem = newItem(enumType, itemValue, itemName);
+    if (PyDict_SetItemString(enumType->tp_dict, itemName, enumItem) < 0)
+        return 0;
+    Py_DECREF(enumItem);
+    return enumItem;
+}
+
+bool createGlobalEnumItem(PyTypeObject* enumType, PyObject* module, const char* itemName, long itemValue)
+{
+    PyObject* enumItem = createEnumItem(enumType, itemName, itemValue);
+    if (enumItem) {
+        if (PyModule_AddObject(module, itemName, enumItem) < 0)
+            return false;
+        Py_DECREF(enumItem);
+        return true;
+    }
+    return false;
+}
+
+bool createScopedEnumItem(PyTypeObject* enumType, SbkObjectType* scope, const char* itemName, long itemValue)
+{
+    PyObject* enumItem = createEnumItem(enumType, itemName, itemValue);
+    if (enumItem) {
+        if (PyDict_SetItemString(scope->super.ht_type.tp_dict, itemName, enumItem) < 0)
+            return false;
+        Py_DECREF(enumItem);
+        return true;
+    }
+    return false;
+}
+
 PyObject* newItem(PyTypeObject* enumType, long itemValue, const char* itemName)
 {
     bool newValue = true;