我想为python写一个C-extension,如下所示,这段代码的目的是从2个float列表中计算出euclidean-dist。
C代码:
#include <python2.7/Python.h>
#include <math.h>
static PyObject* cutil_euclidean_dist(PyObject* self, PyObject* args) {
PyObject *seq_a, *seq_b;
int n;
float * array_a,* array_b;
PyObject *item;
printf("prepare to check input");
PyArg_ParseTuple(args,"iOO", &n , &seq_a, &seq_b);
if (!PySequence_Check(seq_a) || !PySequence_Check(seq_b)) {
PyErr_SetString(PyExc_TypeError, "expected sequence");
return NULL;
}
array_a =(float *)malloc(sizeof(float)*n);
array_b =(float *)malloc(sizeof(float)*n);
if (NULL == array_a || NULL == array_b){
PyErr_SetString(PyExc_TypeError, "malloc failed!");
return NULL;
}
printf("%d",array_a==NULL);
printf("%d",array_b==NULL);
printf("malloc yes ");
printf("n=%d",n);
int i;
for(i=0;i<n;i++){
printf("111");
item = PySequence_GetItem(seq_a,i);
printf("after get item");
if (!PyFloat_Check(item)) {
printf("in float check");
free(array_a); /* free up the memory before leaving */
free(array_b);
free(seq_a);
free(seq_b);
PyErr_SetString(PyExc_TypeError, "expected sequence of float");
return NULL;
}
array_a[i] = PyFloat_AsDouble(item);
printf("a=%f",array_a[i]);
Py_DECREF(item);
item = PySequence_GetItem(seq_b,i);
if(!PyFloat_Check(item)) {
free(array_a);
free(array_b);
free(seq_a);
free(seq_b);
PyErr_SetString(PyExc_TypeError, "expected sequence of float");
}
array_b[i] = PyFloat_AsDouble(item);
Py_DECREF(item);
}
printf("array analyze yes");
double sum = 0;
for(i=0;i<n;i++){
double delta = array_a[i] - array_b[i];
sum += delta * delta;
}
free(array_a);
free(array_b);
free(seq_a);
free(seq_b);
return Py_BuildValue("d",sqrt(sum));
}
static PyMethodDef cutil_methods[] = {
{"c_euclidean_dist", (PyCFunction)cutil_euclidean_dist,METH_VARARGS,NULL},
{NULL,NULL,0,NULL}
};
PyMODINIT_FUNC initcutil(void) {
Py_InitModule3("cutil", cutil_methods, "liurui's c extension for python");
}
python代码:
import cutil
cutil.c_euclidean_dist(2,[1.0,1.0],[2.0,2.0])
结果:
*** Error in `python': munmap_chunk(): invalid pointer: 0x00007fdfb8c28950 *** (core dumped)
所以我无法成功调用此函数。
谁可以帮我解决这个问题?非常感谢我甚至不知道它有什么问题。
顺便说一下: 当我编译代码时,gcc要求我添加-fPIC,但是当我添加-fPIC时,说gcc找不到Python.h 最终我改变了:
#include <Python.h>
进入这个:
#include <python2.7/Python.h>
然后就可以了
当我评论这两个句子时,代码运行良好:
free(seq_a);
free(seq_b);
但我认为如果没有2个句子,就会出现内存泄漏,因为seq_a,seq_b将不会被释放
答案 0 :(得分:2)
您只能使用free()
取消分配使用malloc()
分配的内存。
seq_a
和seq_b
未通过malloc
分配。
另外,您应该查看Python文档中的PyObject
和引用计数章节。