Question

我是SWIG的新手。我创建了一个python模块来使用c ++类。

我的cpp标题代码是

GradedComplex.h：

class GradedComplex
{
public:
  typedef std::complex<double> dcomplex;
  typedef Item<dcomplex> item_type;
  typedef ItemComparator<dcomplex> comparator;
  typedef std::set<item_type, comparator> grade_type;

private:
  int n_;
  std::vector<grade_type *> grade_;
  std::vector<double> thre_;

public:
  GradedComplex(int n, double *thre);
  ~GradedComplex();

  void push(item_type item);
  void avg(double *buf);
};

CPP代码是

#include <iostream>
#include "GradedComplex.h"
using namespace std;

GradedComplex::GradedComplex(int n, double *thre)
{
  n_ = n;
  for (int i = 0; i < n_; ++i)
  {
    thre_.push_back(thre[i]);
    grade_.push_back(new grade_type());
  }
}

GradedComplex::~GradedComplex()
{
  while (0 < grade_.size())
  {
    delete grade_.back();
    grade_.pop_back();
  }
}

void GradedComplex::push(item_type item)
{
  for (int i = 0; i < n_; ++i)
  {
    if (item.norm() < thre_[i])
    {
      grade_[i]->insert(item);
      break;
    }
  }
}

void GradedComplex::avg(double *buf)
{
  for (int i = 0; i < n_; ++i)
  {
    int n = 0;
    double acc = .0l;
    for (grade_type::iterator it = grade_[i]->begin(); it != grade_[i]->end(); ++it)
    {
      acc += (*it).norm();
      ++n;
    }
    buf[i] = acc / n;
  }
}

我的SWIG界面文件是：

example.i

/* File: example.i */
%module example
%{
#include "Item.h"
#include "GradedComplex.h"
#include "GradedDouble.h"
%}

%include <std_string.i>
%include <std_complex.i>
%include "Item.h"
%include "GradedComplex.h"
%include "GradedDouble.h"
%template(Int) Item<int>;
%template(Complex) Item<std::complex<double> >;

我通过运行 * python setup.py build_ext --inplace * 这个命令生成了python模块。

现在我想从python

访问 GradedComplex（int n，double * thre）

当我尝试访问 GradedComplex 时，会显示 ** TypeError：在方法'new_GradedComplex'中，参数2的类型为'double '错误*

如何从python模块传递双指针？请帮我解决这个问题。

Answer 1

在构造函数中直接使用向量并利用SWIG的向量支持更简单：

在.i文件中：

%include <std_vector.i>
%template(DoubleVector) std::vector<double>;
%include "GradedComplex.h"

在.h：

中

GradedComplex(const std::vector<double>& dbls);

在.cpp：

中

GradedComplex::GradedComplex(const vector<double>& dbls) : thre_(dbls)
{
}

n_可以消失，因为thre_.size()是同一回事。

用以下方式调用：

c=Item.GradedComplex([1.2,3.4,5.6])

SWIG也可以处理返回的向量，因此avg可以是：

std::vector<double> GradedComplex::avg() { ... }

SWIG：无法使用双指针访问构造函数

1 个答案: