在 Python 中从 C++ 继承 Base,使用 SWIG 调用抽象方法

Inheriting Base from C++ in Python, call abstract method using SWIG

我在将 C++ (98) 与 python 耦合时遇到了一些问题。我在 C++ 中有一些基础 classes,我想在 Python 中进行扩展。一些有问题的方法在 C++ 端是纯虚拟的,因此将在 Python 端实现。

目前,我可以从 C++ 调用抽象方法,并且通过 swig,可以在 Python 中调用特化。凉爽的。我无法将参数移交给 Python..

简化我的问题的最小完整示例:

// iBase.h
#pragma once
#include <memory>

typedef enum EMyEnumeration{
    EMyEnumeration_Zero,
    EMyEnumeration_One,
    EMyEnumeration_Two

}TEMyEnumeration;


class FooBase{
protected:  
    int a;
public:
    virtual int getA() = 0 ; 
};

class Foo : public FooBase{
public:
    Foo() {a = 2;}
    int getA(){return a;}
};

class iBase{
    public:

    virtual void start() =0;
    virtual void run(std::shared_ptr<FooBase> p, TEMyEnumeration enumCode) = 0;
};

痛饮方面:

// myif.i
%module(directors="1") DllWrapper

%{
#include <iostream>
#include "iBase.h"
%}

%include <std_shared_ptr.i>
%shared_ptr(FooBase)
%shared_ptr(Foo)
%feature("director") FooBase;
%feature("director") iBase;
%include "iBase.h"

运行 痛饮为:

swig -c++ -python myif.i
swig -Wall -c++ -python -external-runtime runtime.h

编译 myif_wrap.cxx -> _DllWrapper.pyd

使用以下代码创建一个 *.exe,它将加载 _DllWrapper.pyd 库(确保它在同一目录中!)。另外将swig生成的DllWrapper.py复制到exe目录下

//Main_SmartPtr.cpp
#include "stdafx.h"
#include <Python.h>
#include <windows.h> 
#include <string>
#include <memory>
#include "iBase.h"
#include "runtime.h"
using namespace std;

int _tmain(int argc, _TCHAR* argv[])
{
    string moduleName = "ExampleSmartPtr";

    // load *.pyd  (actually a dll file which implements PyInit__<swigWrapperName>)
    auto handle =LoadLibrary("_DllWrapper.pyd");

    // getting an instance handle..
    Py_Initialize();
    PyObject *main = PyImport_AddModule("__main__");
    PyObject *dict = PyModule_GetDict(main);

    PyObject *module = PyImport_Import(PyString_FromString(moduleName.c_str()));
    PyModule_AddObject(main, moduleName.c_str(), module);
    PyObject *instance = PyRun_String(string(moduleName+string(".")+moduleName+string("()")).c_str(), Py_eval_input, dict, dict);


    //calling start() in the Python derived class..
    //PyObject *result = PyObject_CallMethod(instance, "start", (char *)"()");

    // trying to call run in the Python derived class..

    shared_ptr<Foo> foo = make_shared<Foo>();
    EMyEnumeration enumCode = EMyEnumeration_Two;


    string  typeName1 = "std::shared_ptr <FooBase> *"; 
    swig_type_info* info1 = SWIG_TypeQuery(typeName1.c_str());
    auto swigData1 = SWIG_NewPointerObj((void*)(&foo), info1, SWIG_POINTER_OWN);

    string  typeName2 = "TEMyEnumeration *"; 
    swig_type_info* info2 = SWIG_TypeQuery(typeName2.c_str());
    auto swigData2 = SWIG_NewPointerObj((void*)(&enumCode), info2, SWIG_POINTER_OWN);

    auto result = PyObject_CallMethod(instance, "run", (char *)"(O)(O)", swigData1, swigData2);

    return 0;
}

创建一个新的Python文件并将其放在exe目录中:

#ExampleSmartPtr.py
import DllWrapper

class ExampleSmartPtr(DllWrapper.iBase):
    def __init__(self): # constructor
        print("__init__!!")
        DllWrapper.iBase.__init__(self)

    def start(self):
        print("start")
        return 0

    def run(self, data, enumCode):
        print("run")
        print("-> data: "+str(data))
        print("-> enumCode: "+str(enumCode))

        print (data.getA())
        return 1

运行 执行 exe 的输出是:

__init__!!
run
-> data: (<DllWrapper.FooBase; proxy of <Swig Object of type 'std::shared_ptr< FooBase > *' at 0x00000000014F8B70> >,)
-> enumCode: (<Swig Object of type 'TEMyEnumeration *' at 0x00000000014F89F0>,)

如何将一个 'dereference' 的 enumCode 转换为一个简单的 int?如何在 python class 运行() 中打印 (data.getA())?目前的形式它不打印任何东西..

这不是一个真正的答案,但我阅读了讨论 Discussion from 2005 并且它是有道理的,它不应该是可能的。如果您站在 Python 一边,请执行以下操作,您会将枚举 'dereferenced' 变成一个简单的整数。

import ExampleSmartPtr

instance = ExampleSmartPtr.ExampleSmartPtr()
swigData1 = ExampleSmartPtr.DllWrapper.Foo()
swigData2 = ExampleSmartPtr.DllWrapper.EMyEnumeration_Two
instance.run(swigData1,swigData2)

这将打印

__init__!!
run
-> data: <DllWrapper.Foo; proxy of <Swig Object of type 'std::shared_ptr< Foo > *' at 0x7f8825c0b7e0> >
-> enumCode: 2

我认为问题在于两个不同的虚表在起作用。原始的 C++ vtable 和 Swig Object 的 vtable。只是好奇,在什么情况下使用 C++ class 的 Python 后代对 C++ 感兴趣?

似乎有人尝试过 the exact same thing !

我所做的是用 -DSWIG_TYPE_TABLE=iBase 编译 *.pyd。

然后我将这个添加到cpp端的主应用程序中:

iBase *python2interface(PyObject *obj) {
  void *argp1 = 0;
  swig_type_info * pTypeInfo = SWIG_TypeQuery("iBase *");

  const int res = SWIG_ConvertPtr(obj, &argp1,pTypeInfo, 0);
  if (!SWIG_IsOK(res)) {
    abort();
  }
  return reinterpret_cast<iBase*>(argp1);
}

并像这样调用实施形式 python:

auto foo = make_shared<Foo>();
TEMyEnumeration enumCode = EMyEnumeration_Two;
python2interface(instance)->run(foo, enumCode);

最后,我用 -DSWIG_TYPE_TABLE=iBase 再次编译了 C++ 实现。

很有魅力!