python - 编译多个模块时 import_array() 出现 Numpy/CAPI 错误

Question

我正在尝试编译一个 C++ 模块以在 scipy.weave 中使用，它由几个头文件和源 C++ 文件组成。这些文件包含广泛使用 Numpy/C-API 接口(interface)的类和方法。但我没能弄清楚如何成功地包含 import_array() 。过去一周我一直在为此苦苦挣扎，我快疯了。我希望你能帮助我，因为 weave help不是很解释。

在实践中，我首先有一个名为 pycapi_utils 的模块，其中包含一些将 C 对象与 Python 对象连接起来的例程。它由一个头文件pycapi_utils.h和一个源文件pycapi_utils.cpp组成，例如:

//pycapi_utils.h
#if ! defined _PYCAPI_UTILS_H
#define _PYCAPI_UTILS_H 1

#include <stdlib.h>
#include <Python.h>
#include <numpy/arrayobject.h>
#include <tuple>
#include <list>

typedef std::tuple<const char*,PyObject*> pykeyval; //Tuple type (string,Pyobj*) as dictionary entry (key,val)
typedef std::list<pykeyval> kvlist;                    

//Declaration of methods
PyObject* array_double_to_pyobj(double* v_c, long int NUMEL); //Convert from array to Python list (double)
...
...
#endif

和

//pycapi_utils.cpp

#include "pycapi_utils.h"

PyObject* array_double_to_pyobj(double* v_c, long int NUMEL){
    //Convert a double array to a Numpy array
    PyObject* out_array = PyArray_SimpleNew(1, &NUMEL, NPY_DOUBLE);
    double* v_b = (double*) ((PyArrayObject*) out_array)->data;
    for (int i=0;i<NUMEL;i++) v_b[i] = v_c[i];
    free(v_c);
    return out_array;
}

然后我还有一个模块 model，其中包含处理某些数学模型的类和例程。同样，它由一个头文件和源文件组成，例如:

//model.h
#if ! defined _MODEL_H
#define _MODEL_H 1

//model class
class my_model{
    int i,j;
    public:
        my_model();
        ~my_model();
        double* update(double*); 
}

//Simulator   
PyObject* simulate(double* input);
#endif  

和

//model.cpp

#include "pycapi_utils.h"
#include "model.h"

//Define class and methods
model::model{
...
...
}

...
...

double* model::update(double* input){
    double* x = (double*)calloc(N,sizeof(double));
    ...
    ...

    // Do something
    ...
    ...

    return x;
}

PyObject* simulate(double* input){
    //Initialize Python interface 
    Py_Initialize;
    import_array();

    model random_network;
    double* output;

    output = random_network.update(input);
    return array_double_to_pyobj(output);  // from pycapi_utils.h     
}

以上代码包含在 Python 中的 scipy.weave 模块中

def model_py(input):
    support_code="""
                 #include "model.h"
                 """
    code = """
           return_val = simulate(input.data());
           """
    libs=['gsl','gslcblas','m']
    vars = ['input']
    out = weave.inline(code,
                       vars,
                       support_code=support_code,
                       sources = source_files,
                       libraries=libs
                       type_converters=converters.blitz,
                       compiler='gcc',
                       extra_compile_args=['-std=c++11'],
                       force=1) 

编译失败:

error: int _import_array() was not declared in this scope

值得注意的是，如果我将 pycapi_utils.h 和源代码 pycapi_utils.cpp 混为一谈，一切正常。但我不想使用这个解决方案，因为实际上我的模块需要包含在其他几个也使用 PyObjects 的模块中，并且需要调用 import_array()。

我正在寻找 this post在堆栈交换上，但我不知道是否以及如何在我的案例中正确定义 #define 指令。此外，该帖子中的示例并不完全是我的情况，import_array() 是在 main() 的全局范围内调用的，而在我的情况下是 import_array( ) 在我的 simulate 例程中调用，该例程由 scipy.weave 构建的 main() 调用。

Answer 1

我有一个类似的问题，正如您发布的链接所指出的，万恶之源是 PyArray_API 被定义为静态的，这意味着每个翻译单元都有自己的 PyArray_API 默认情况下使用 PyArray_API = NULL 初始化。因此，必须为每个 *.cpp 文件调用一次 import_array()。在您的情况下，在 pycapi_utils.cpp 中调用它就足够了，并且在 model.cpp 中调用一次就足够了。您还可以在实际调用之前测试是否需要 array_import:

if(PyArray_API == NULL)
{
    import_array(); 
}