fix: resolve anonymous enum

compiler-research · Sep 24, 2024 · b1c76d0 · b1c76d0
1 parent 098f455
commit b1c76d0
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/src/CPPDataMember.cxx b/src/CPPDataMember.cxx
@@ -81,6 +81,26 @@ static PyObject* dm_get(CPPDataMember* dm, CPPInstance* pyobj, PyObject* /* kls
             Py_INCREF(dm->fDescription);
             return dm->fDescription;
         }
+
+        if (Cppyy::IsEnumConstant(dm->fScope)) {
+            // anonymous enum
+            long long llval = Cppyy::GetEnumDataValue(dm->fScope);
+            const std::string& enum_type = Cppyy::ResolveEnum(dm->fScope);
+
+            PyObject* bval;
+            if (enum_type == "bool") {
+                bval = (bool)llval ? Py_True : Py_False;
+                Py_INCREF(bval);
+            } else if (enum_type == "char") {
+                char val = (char)llval;
+                bval = CPyCppyy_PyText_FromStringAndSize(&val, 1);
+            } else if (enum_type == "int" || enum_type == "unsigned int")
+                bval = PyInt_FromLong((long)llval);
+            else
+                bval = PyLong_FromLongLong(llval);
+
+            return bval;
+        }
     }
 
 // non-initialized or public data accesses through class (e.g. by help())