python – Swig从Base *向下转换为Derived *

我有以下c类(简化),我使用SWIG向Python公开：

struct Component
{
    virtual void update();
}

struct DerivedComponent : public Component
{
    void update() { cout << "DerivedComponent::update()" << endl; }
    void speak() { cout << "DerivedComponent::speak()" << endl; }
}

class Entity
{
public:
    Component* component(const std::string& class_name)
    {
        return m_components[class_name];
    }

private:
    std::unordered_map<std::string, Component*> m_components;
}

现在,在Python中,我可以在Entity实例上成功调用component(“DerivedComponent”).update().但是,我不能调用component(“DerivedComponent”).speak(),因为组件返回的类型(“DerivedComponent”)被报告为< class'module.Component'>.

我显然需要向下转换component()函数的结果,以便调用DerivedComponent中定义的方法.我曾希望Swig会像我相信Boost.Python那样执行自动向下转换.

如果没有在c中定义一大堆类型转换函数并将它们暴露给Python,那么使用Swig或Python进行向下转换是否有更好的解决方案？我有什么选择？

解决方法:

您可以通过一些工作在Python中完成您想要的任务.它可以正常工作,因为在Python中,向下转换是没有意义的,因为函数的返回类型(或一般类型)不是强类型的,所以我们可以修改你的Entity :: component函数,以便始终返回最派生的类型,无论如何这是什么.

要使用C / Python绑定工作,您需要为Entity :: component编写一个’out’类型映射.我写了一个如何工作的例子.在这种情况下,我们必须略微提出它,因为知道向下传播它的唯一方法来自函数的参数. (例如,如果您的基类有一个方法,将其作为字符串/枚举返回,您可以进一步简化此操作而不依赖于输入参数).

%module test

%{
#include "test.hh"
%}

%include <std_string.i>

%typemap(out) Component * Entity::component {
    const std::string lookup_typename = *arg2 + " *";
    swig_type_info * const outtype = SWIG_TypeQuery(lookup_typename.c_str());
    $result = SWIG_NewPointerObj(SWIG_as_voidptr($1), outtype, $owner);
}

%include "test.hh"

这使用SWIG_TypeQuery函数要求Python运行时基于arg2查找类型(对于您的示例,这是字符串).

我必须对你的示例标题(在我的示例中命名为test.hh)进行一些更改以修复一些问题,然后才能将其转换为完全正常运行的演示,它最终看起来像：

#include <iostream>
#include <map>
#include <string>

struct Component
{
    virtual void update() = 0;
    virtual ~Component() {}
};

struct DerivedComponent : public Component
{
    void update() { std::cout << "DerivedComponent::update()" << std::endl; }
    void speak() { std::cout << "DerivedComponent::speak()" << std::endl; }
};

class Entity
{
public:
    Entity() {
       m_components["DerivedComponent"] = new DerivedComponent;
    }

    Component* component(const std::string& class_name)
    {
        return m_components[class_name];
    }

private:
    std::map<std::string, Component*> m_components;
};

然后我用它构建它：

swig -py3 -c++ -python -Wall test.i
g++ -Wall -Wextra test_wrap.cxx -I/usr/include/python3.4/ -lpython3.4m -shared -o _test.so

有了这个,我就可以运行以下Python：

from test import *

e=Entity()
print(e)

c=e.component("DerivedComponent")
print(c)
print(type(c))

c.update()
c.speak()

这有效,你希望：

<test.Entity; proxy of <Swig Object of type 'Entity *' at 0xb7230458> >
Name is: DerivedComponent *, type is: 0xb77661d8
<test.DerivedComponent; proxy of <Swig Object of type 'DerivedComponent *' at 0xb72575d8> >
<class 'test.DerivedComponent'>
DerivedComponent::update()
DerivedComponent::speak()

标签：downcast,python,c,swig
来源： https://codeday.me/bug/20191008/1872543.html