C++中調用Python腳本

C++中調用Python腳本的意義就不講了，至少你可以把它當成文本形式的動態鏈接庫，

需要的時候還可以改一改，只要不改變接口， C++的程序一旦編譯好了，再改就沒那么方便了

先看Python的代碼

#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


把上面的Python代碼存為pytest.py

接下來是c++ 的代碼
代碼:
#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 = 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("l",3));
  PyTuple_SetItem(pArgs, 1, Py_BuildValue("l",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("l",2)); //

  PyObject_CallObject(pFunc, pArgs);


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

  // 關閉Python
  Py_Finalize();
  return 0;
}