Swig:接口简单的C ++类,返回向量

时间:2017-09-23 07:41:09

标签: python c++ swig

我使用swig是全新的,如果看起来很简单,请原谅我。

我想为C++中的一个类创建一个接口,该接口获取并返回double的向量:

/*  example.h file */
#include <iostream>
#include <cstdio>
#include <vector>

class my_class 
{
    private:    
        int N;
    public:
        my_class(){ }
        std::vector<double> half(const std::vector<double>& ); 
};

/* example.cpp  file */
#include "example.h"

std::vector<double> my_class::half(const std::vector<double>& v) {
    std::vector<double> w(v);
    for (unsigned int i=0; i<w.size(); i++)
        w[i] /= 2.0;
    return w;
}

example.i 接口文件

%module example

%{
#define SWIG_FILE_WITH_INIT
#include "example.h"
%}

%include "std_vector.i"

namespace std {
    %template(DoubleVector)  vector<double>;
}


#include "example.h"

然后我尝试在python中使用该模块:

import example
A = example.my_class()

AttributeError                            Traceback (most recent call last)
<ipython-input-5-d2af384d4adf> in <module>()
----> 1 example.my_class()

AttributeError: 'module' object has no attribute 'my_class'

感谢您提供任何指导或评论。

1 个答案:

答案 0 :(得分:0)

example.i中的最后一行应为%include "example.h"而不是#include "example.h"