C++ 调用 python 注意事项

C++ 调用 python 注意事项 (2010-03-11 15:15:09)

标签：  杂谈

分类： 图像处理+Python

C++ 调用 python 注意事项

注意事项有：
1. python文件里面不能有主函数，只能全部写一般的函数，不然在C++调用的时候会找不到*.py文件在哪个地方，加载不进去。
2. 三种基本参数类型的传递 整型 字符串 浮点 i s f



1. python文件 pytest.py

#test function

def add(a,b):

print "in python function add"

print "a = " + str(a)

print "b = " + str(b)

print "ret = " + str(a+b)

return


def foo(a):

print "in python function foo"

print "a = " + str(a)

print "ret = " + str(a * a)

return


2. lwk.cpp

//C++调用python代码例子
#include "Python.h"

int main(int argc, char** argv)
{
//初始化Python
//在使用Python系统前，必须使用Py_Initialize对其
//进行初始化。它会载入Python的内建模块并添加系统路
//径到模块搜索路径中。这个函数没有返回值，检查系统
//是否初始化成功需要使用Py_IsInitialized。

Py_Initialize();

// 检查初始化是否成功
if ( !Py_IsInitialized() )
{
return -1;
}

// 添加当前路径
//把输入的字符串作为Python代码直接运行，返回0
//表示成功，-1表示有错。大多时候错误都是因为字符串
//中有语法错误。
PyRun_SimpleString("import sys");
PyRun_SimpleString("sys.path.append('./')");
PyObject *pName,*pModule,*pDict,*pFunc,*pArgs;

// 载入名为pytest的脚本
pName = PyString_FromString("pytest");
pModule = PyImport_Import(pName);
if ( !pModule )
{
printf("can't find pytest.py");
getchar();
return -1;
}
pDict = PyModule_GetDict(pModule);
if ( !pDict )
{
return -1;
}

// 找出函数名为add的函数
pFunc = PyDict_GetItemString(pDict, "add");
if ( !pFunc || !PyCallable_Check(pFunc) )
{
printf("can't find function [add]");
getchar();
return -1;
}

// 参数进栈
*pArgs;
pArgs = PyTuple_New(2);

// PyObject* Py_BuildValue(char *format, ...)
// 把C++的变量转换成一个Python对象。当需要从
// C++传递变量到Python时，就会使用这个函数。此函数
// 有点类似C的printf，但格式不同。常用的格式有
// s 表示字符串，
// i 表示整型变量，
// f 表示浮点数，
// O 表示一个Python对象。

PyTuple_SetItem(pArgs, 0, Py_BuildValue("i",3));
PyTuple_SetItem(pArgs, 1, Py_BuildValue("i",4));

// 调用Python函数
PyObject_CallObject(pFunc, pArgs);

//下面这段是查找函数foo 并执行foo
pFunc = PyDict_GetItemString(pDict, "foo");
if ( !pFunc || !PyCallable_Check(pFunc) )
{
printf("can't find function [foo]");
getchar();
return -1;
}

pArgs = PyTuple_New(1);
PyTuple_SetItem(pArgs, 0, Py_BuildValue("i",2)); //

PyObject_CallObject(pFunc, pArgs);


Py_DECREF(pName);
Py_DECREF(pArgs);
Py_DECREF(pModule);

// 关闭Python
Py_Finalize();

return 0;
}


3. Makefile文件

# kevinew
# Makefile Nov 2009
# feel free to hack this to pieces

#### local macros
# remove without fussing about it
RM = /bin/rm -f

# compiler name and flags
CCC = g++
CCFLAGS = -I/usr/include/python2.4 -L/usr/lib/python2.4 -lpython2.4

# loader flags
LDFLAGS =

### local program information
EXEC = lwk
SOURCES = lwk.cpp

### intermediate objects
OBJECTS = $(SOURCES: .cpp=.o)

### includes
INCLUDES =