引言
很多时候会遇到需要自己写python库的问题,但是又要兼容考虑numpy这种常用数据类型。本文目的在于提供一个简单的tutorial,方面大家入门swig给python写wrapper。实际上,wrapper有很多方法,除了swig,还可以用boost库,boost的部分将放在以后的文章中讨论。
实现
一个最简单的wrapper目录如下所示,假设你有一个写好的类myClass。另外numpy.i是官方提供的swig wrapper,直接下载就行。我们需要做的是,自己写一个.i描述文件,告诉swig哪些指针类型是具体的多少维度的数据。
这里以简单的c++ class为例。头文件myClass.hpp声明两个public函数,用于产生2D和1D数组。具体内容如下:
class myClass
{
public:
myClass();
~myClass();
void generateData2D(float *ptr_data1, int dimx1, int dimy1, int dim1, int dim2);
void generateData1D(float *ptr_data2, int dimx2, int dim1);
};
其实现如下所示,myClass.cpp。
#include "myClass.hpp"
myClass::myClass()