如何编写Swig Out Typemap实现从C++包装返回Python枚举对象

你已经搞定了Python Enum转C枚举类的入参typemap，要实现反向的out typemap，核心就是把C枚举的整数值映射到你自定义的Python Enum实例上，下面是具体的实现代码和注意事项：

实现思路 #

1. 拿到你自己定义的Python Color Enum类对象
2. 把C++里的枚举值转成Python整数
3. 用这个整数获取对应的Color枚举实例
4. 处理好Python对象的引用计数，确保返回正确的对象

完整Out Typemap代码 #

假设你的Python Enum定义在my_enums.py模块里，对应的out typemap代码如下：

%typemap(out) Color {
    // 导入包含Color Enum的Python模块
    PyObject* enum_module = PyImport_ImportModule("my_enums");
    if (!enum_module) {
        SWIG_fail;
    }

    // 从模块字典里取出Color Enum类
    PyObject* color_enum = PyDict_GetItemString(PyModule_GetDict(enum_module), "Color");
    Py_DECREF(enum_module); // 模块用完就释放引用
    if (!color_enum || !PyType_Check(color_enum)) {
        SWIG_fail;
    }

    // 把C++枚举值转成Python整数
    PyObject* enum_value = PyLong_FromLong((long)$1);
    if (!enum_value) {
        SWIG_fail;
    }

    // 调用Color(enum_value)得到对应的枚举实例
    PyObject* result = PyObject_CallFunctionObjArgs(color_enum, enum_value, NULL);
    Py_DECREF(enum_value); // 临时整数用完释放
    if (!result) {
        SWIG_fail;
    }

    // 设置返回值，Swig会处理后续的引用计数
    $result = result;
}

关键细节提醒 #

• 模块路径：如果你的Color Enum直接写在Swig生成的Python模块里，不用单独导入，直接用PyModule_GetDict(SWIG_GetModule())获取当前模块的字典来拿Color类就行。
• 错误检查：每一步的NULL检查不能少，不然容易留异常隐患，SWIG_fail会触发Swig自带的错误处理流程。
• 引用计数：手动释放没用的Python对象（比如导入的模块、临时整数），避免内存泄漏。
• 值匹配：一定要保证C++枚举的整数值和你Python Enum定义的完全一致（你例子里red=0、green=1、blue=2是对的），不然会返回错误的枚举实例。

简化版（同一模块场景） #

要是你的Python Enum和Swig生成的代码在同一个模块里，代码可以简化成这样：

%typemap(out) Color {
    PyObject* module_dict = PyModule_GetDict(SWIG_GetModule());
    PyObject* color_enum = PyDict_GetItemString(module_dict, "Color");
    if (!color_enum || !PyType_Check(color_enum)) {
        SWIG_fail;
    }

    PyObject* result = PyObject_CallFunctionObjArgs(color_enum, PyLong_FromLong((long)$1), NULL);
    if (!result) {
        SWIG_fail;
    }

    $result = result;
}

内容的提问来源于stack exchange，提问作者gogoer