使用pythoncapi twi调用C函数时出现分段错误

Question

最近我想用C包装器从Fortran调用Python（参见here）。下面我发布了一个简单的例子。函数的第二次调用导致pModule = PyImport_Import(pName)处的分段错误。我发现问题出在from scipy.optimize import newton行上——如果我注释它，一切都会正常工作。有什么办法解决吗？在

rootC.c

#include "rootC.h"
#include <Python.h>

void root_(double* A, double* B, double* t, double* x)
{
    PyObject *pName, *pModule, *pFunc;
    PyObject *pArgs, *pValue, *sys, *path;

    Py_Initialize();

    sys  = PyImport_ImportModule("sys");
    path = PyObject_GetAttrString(sys, "path");
    PyList_append(path, PyString_FromString("."));

    pName = PyString_FromString("rootPY");
    pModule = PyImport_Import(pName);

    if (!pModule)
    {
        PyErr_Print();
        printf("ERROR in pModule\n");
        exit(1);
    }

    pFunc = PyObject_GetAttrString(pModule, "root");
    pArgs = PyTuple_New(3);
    PyTuple_SetItem(pArgs, 0, PyFloat_FromDouble((*A)));
    PyTuple_SetItem(pArgs, 1, PyFloat_FromDouble((*B)));
    PyTuple_SetItem(pArgs, 2, PyFloat_FromDouble((*t)));

    pValue = PyObject_CallObject(pFunc, pArgs);
    *x     = PyFloat_AsDouble(pValue);

    Py_Finalize();
}

rootC.h

^{pr2}$

根py.py

from mpmath import polylog, exp
from scipy.optimize import newton

def root(A,B,t):   
    return 1

main.c

#include "rootC.h"
#include <stdio.h>

int main() 
{
    double A = 0.4, B = 0.3, t = 0.1, x = 0.0;

    root_(&A,&B,&t,&x);
    printf("x = %.15f\n", x);

    root_(&A,&B,&t,&x);
    printf("x = %.15f\n", x);

    return 0;
}

生成文件

CC = gcc
FC = gfortran
CFLAGS = -I/usr/include/python2.7
LFLAGS = -L/usr/local/lib -lpython2.7 -lm

.PHONY: all clean

all: main

main: main.o rootC.o
    $(CC) $^ -o $@ $(LFLAGS)

main.o: main.c
    $(CC) $(CFLAGS) -c $< -o $@

rootC.o: rootC.c
    $(CC) $(CFLAGS) -c $< -o $@

clean:
    rm -f *.o

Answer 1

问题在于每次调用函数时都要初始化和完成pythonshell。（它多次尝试初始化.dll）

编辑：Link to resource

这是我的解决办法。。。（不是工作的最佳代码） 在你的主楼

int main() 
{
    double A = 0.4, B = 0.3, t = 0.1, x = 0.0;
    bool Stop = false;
    root_(&A,&B,&t,&x,&Stop);
    printf("x = %.15f\n", x);

    Stop=true;
    root_(&A,&B,&t,&x,&Stop);
    printf("x = %.15f\n", x);



    return 0;
}

然后在你的根C.h

^{pr2}$

然后在rootC.c

#include "rootC.h"
#include <Python.h>

void root_(double* A, double* B, double* t, double* x,bool* Stop)
{
    PyObject *pName, *pModule, *pFunc;
    PyObject *pArgs, *pValue, *sys, *path;
    if (*Stop==false)
    {
        Py_Initialize();
    }
    sys  = PyImport_ImportModule("sys");
    path = PyObject_GetAttrString(sys, "path");
    PyList_Append(path, PyString_FromString("."));

    pName = PyString_FromString("rootPY");
    pModule = PyImport_Import(pName);

    if (!pModule)
    {
        PyErr_Print();
        printf("ERROR in pModule\n");
        exit(1);
    }

    pFunc = PyObject_GetAttrString(pModule, "root");
    pArgs = PyTuple_New(3);
    PyTuple_SetItem(pArgs, 0, PyFloat_FromDouble((*A)));
    PyTuple_SetItem(pArgs, 1, PyFloat_FromDouble((*B)));
    PyTuple_SetItem(pArgs, 2, PyFloat_FromDouble((*t)));

    pValue = PyObject_CallObject(pFunc, pArgs);
    *x     = PyFloat_AsDouble(pValue);
    if (*Stop==true)
    {
        Py_Finalize();
    }
}

这应该行得通：）

Answer 2

你也可以这样做：

#include "rootC.h"
#include <Python.h>
#include <stdlib.h>

void  Initialize ()
{ 
    Py_Initialize();
}

void Finalize ()
{
    Py_Finalize();
}

void root_(double* A, double* B, double* t, double* x)
{
    PyObject *pName, *pModule, *pFunc;
    PyObject *pArgs, *pValue, *sys, *path;
    static int i;

    if (i == 0)
    {   
        ++i;
        Initialize();
        atexit(Finalize);
    }

    ....
}