使用C ++(和GSL)扩展和嵌入Python(和NumPy):将gsl_matrix传递给python并返回

时间:2013-09-11 18:40:01

标签: c++ python numpy gsl

我的问题“应该”很简单,但我仍然无法解决。

我目前正在开发一个项目,需要一些繁重的计算(在C ++中完成)和一些模拟后的数据分析(在Python中完成)。

但是,现在我正在改变主算法,我需要从C ++和Python中“循环”一些计算来回。 也就是说,我需要在C ++和Python中来回移动双打矩阵。

在C ++中,数据矩阵是“gsl_matrix”对象,而在python中,相同的矩阵实现为“numpy数组”。

目前,我正在运行我的C ++代码,将矩阵保存到文件,从python中读取它,将其写回文件,然后在C ++中再次打开它以进行进一步的计算。

由于这是非常低效的,我想问一下是否有人可以给我一个如何以“干净”的方式做到这一点的例子。

我一直在阅读(并尝试了10天)SWIG,Cython,Boost.Python和Boost.Numpy但我仍然无法破解它。

有没有人有一个有用的例子可以分享?

谢谢!

1 个答案:

答案 0 :(得分:1)

我认为您不需要自己实现包装器,因为您可以使用pygsl。如果你真的想要实现自己的版本,这里是pygsl的例程,可能值得你

%{
#include <gsl/gsl_matrix_double.h>
#include <gsl/gsl_matrix_complex_double.h>
%}

%include typemaps.i

// gsl_matrix typemaps
%typemap(in) gsl_matrix* %{
  PyArrayObject *_PyMatrix$argnum;
  gsl_matrix_view matrix$argnum;
  {
    _PyMatrix$argnum = (PyArrayObject*)
      PyArray_ContiguousFromObject($input, PyArray_DOUBLE, 2, 2);
    if (_PyMatrix$argnum == NULL)
      return NULL;
    matrix$argnum
      = gsl_matrix_view_array((double*)_PyMatrix$argnum->data,
                  _PyMatrix$argnum->dimensions[0],
                  _PyMatrix$argnum->dimensions[1]);    
    $1 = &matrix$argnum.matrix;
  }
%}