我有一个BaseClass
和一些派生类
#ifndef TEST_H__
#define TEST_H__
#include <iostream>
#include <memory>
class BaseClass
{
public:
virtual double eval(double x) const = 0;
};
class Square: public BaseClass
{
public:
double eval(double x) const {return x*x;}
};
class Add1: public BaseClass
{
public:
Add1(BaseClass & obj): obj_(obj) {}
double eval(double x) const {return obj_.eval(x) + 1.0;}
private:
BaseClass & obj_;
};
#endif /* TEST_H__ */
用SWIGàla
处理%module test
%{
#define SWIG_FILE_WITH_INIT
%}
%{
#include "test.h"
%}
%include "test.h"
这可以在Python中使用,如
import test
s = test.Square()
a = test.Add1(s)
print(a.eval(2.0))
什么是 segfaulting :
import test
a = test.Add1(test.Square())
print(a.eval(2.0))
为什么呢? test.Square()
未分配给变量,因此在分配到a
后不再存在,obj_
指向无效存储。
为避免此类行为,请使用std::shared_ptr<BaseClass>
代替BaseClass&
,即
class Add1: public BaseClass
{
public:
Add1(std::shared_ptr<BaseClass> & obj): obj_(obj) {}
double eval(double x) const {return obj_->eval(x) + 1.0;}
private:
std::shared_ptr<BaseClass> obj_;
};
这个确切的代码不适用于
TypeError: in method 'new_Add1', argument 1 of type 'std::shared_ptr< BaseClass > &'
也有道理:test.Square()
不会返回std::shared_ptr<BaseClass>
,只会返回Square
又称BaseClass
个实例。
是否可以让test.Square()
返回共享指针std::shared_ptr<Square>
?
答案 0 :(得分:5)
SWIG非常好support for std::smart_ptr
。这一切都非常透明,因此您需要对.i文件进行的更改只是:
%module test
%{
#define SWIG_FILE_WITH_INIT
#include "test.h"
%}
%include <std_shared_ptr.i>
%shared_ptr(Square);
%shared_ptr(BaseClass);
%shared_ptr(Add1); // Not actually needed to make your demo work, but a good idea still
%include "test.h"
这足以使您的演示Python代码工作,我还添加了onlySquare()
作为Square
的成员函数,并调整了演示来说明它:
import test
sq=test.Square()
test.Add1(sq) # implicitly converted to shared_ptr<BaseClass> here
sq.onlySquare()
print sq
# <test.Square; proxy of <Swig Object of type 'std::shared_ptr< Square > *' at 0xf7424950> >
它应该只是工作&#39;对于非智能指针参数也是如此,但请注意,现在所有 Python在该层次结构中创建的实例将是“聪明的”。
(如果您有兴趣,我之前也已经介绍过std::unique_ptr
和std::weak_ptr
)。