Python調用C++程式的方法詳解,python詳解
前言
大家都知道Python的優點是開發效率高,使用方便,C++則是運行效率高,這兩者可以相輔相成,不管是在Python項目中嵌入C++代碼,或是在C++項目中用Python實現外圍功能,都可能遇到Python調用C++模組的需求,下面列舉出集中c++代碼匯出成Python介面的幾種基本方法,一起來學習學習吧。
原生態匯出
Python解譯器就是用C實現,因此只要我們的C++的資料結構能讓Python認識,理論上就是可以被直接調用的。我們實現test1.cpp如下
#include <Python.h>int Add(int x, int y){ return x + y;}int Del(int x, int y){ return x - y;}PyObject* WrappAdd(PyObject* self, PyObject* args){ int x, y; if (!PyArg_ParseTuple(args, "ii", &x, &y)) { return NULL; } return Py_BuildValue("i", Add(x, y));}PyObject* WrappDel(PyObject* self, PyObject* args){ int x, y; if (!PyArg_ParseTuple(args, "ii", &x, &y)) { return NULL; } return Py_BuildValue("i", Del(x, y));}static PyMethodDef test_methods[] = { {"Add", WrappAdd, METH_VARARGS, "something"}, {"Del", WrappDel, METH_VARARGS, "something"}, {NULL, NULL}};extern "C"void inittest1(){ Py_InitModule("test1", test_methods);}
編譯命令如下
g++ -fPIC -shared test1.cpp -I/usr/include/python2.6 -o test1.so
運行Python解譯器,測試如下
>>> import test1>>> test1.Add(1,2)3
這裡要注意一下幾點
- 如果產生的動態庫名字為test1,則源檔案裡必須有inittest1這個函數,且Py_InitModule的第一個參數必須是“test1”,否則Python匯入模組會失敗
- 如果是cpp源檔案,inittest1函數必須用extern "C"修飾,如果是c源檔案,則不需要。原因是Python解譯器在匯入庫時會尋找initxxx這樣的函數,而C和C++對函數符號的編碼方式不同,C++在對函數符號進行編碼時會考慮函數長度和參數類型,具體可以通過
nm test1.so查看函數符號,c++filt工具可通過符號反解出函數原型
通過boost實現
我們使用和上面同樣的例子,實現test2.cpp如下
#include <boost/python/module.hpp>#include <boost/python/def.hpp>using namespace boost::python;int Add(const int x, const int y){ return x + y;}int Del(const int x, const int y){ return x - y;}BOOST_PYTHON_MODULE(test2){ def("Add", Add); def("Del", Del);}
其中BOOST_PYTHON_MODULE的參數為要匯出的模組名字
編譯命令如下
g++ test2.cpp -fPIC -shared -o test2.so -I/usr/include/python2.6 -I/usr/local/include -L/usr/local/lib -lboost_python
注意: 編譯時間需要指定boost標頭檔和庫的路徑,我這裡分別是/usr/local/include和/usr/local/lib
或者通過setup.py匯出模組
#!/usr/bin/env pythonfrom distutils.core import setupfrom distutils.extension import Extensionsetup(name="PackageName", ext_modules=[ Extension("test2", ["test2.cpp"], libraries = ["boost_python"]) ])
Extension的第一個參數為模組名,第二個參數為檔案名稱
執行如下命令
python setup.py build
這時會產生build目錄,找到裡面的test2.so,並進入同一級目錄,驗證如下
>>> import test2>>> test2.Add(1,2)3>>> test2.Del(1,2)-1
匯出類
test3.cpp實現如下
#include <boost/python.hpp>using namespace boost::python;class Test{public: int Add(const int x, const int y) { return x + y; } int Del(const int x, const int y) { return x - y; }};BOOST_PYTHON_MODULE(test3){ class_<Test>("Test") .def("Add", &Test::Add) .def("Del", &Test::Del);}
注意:BOOST_PYTHON_MODULE裡的.def使用方法有點類似Python的文法,等同於
class_<Test>("Test").def("Add", &Test::Add);class_<Test>("Test").def("Del", &Test::Del);
編譯命令如下
g++ test3.cpp -fPIC -shared -o test3.so -I/usr/include/python2.6 -I/usr/local/include/boost -L/usr/local/lib -lboost_python
測試如下
>>> import test3>>> test = test3.Test()>>> test.Add(1,2)3>>> test.Del(1,2)-1
匯出變參函數
test4.cpp實現如下
#include <boost/python.hpp>using namespace boost::python;class Test{public: int Add(const int x, const int y, const int z = 100) { return x + y + z; }};int Del(const int x, const int y, const int z = 100){ return x - y - z;}BOOST_PYTHON_MEMBER_FUNCTION_OVERLOADS(Add_member_overloads, Add, 2, 3)BOOST_PYTHON_FUNCTION_OVERLOADS(Del_overloads, Del, 2, 3)BOOST_PYTHON_MODULE(test4){ class_<Test>("Test") .def("Add", &Test::Add, Add_member_overloads(args("x", "y", "z"), "something")); def("Del", Del, Del_overloads(args("x", "y", "z"), "something"));}
這裡Add和Del函數均採用了預設參數,Del為普通函數,Add為類成員函數,這裡分別調用了不同的宏,宏的最後兩個參數分別代表函數的最少參數個數和最多參數個數
編譯命令如下
g++ test4.cpp -fPIC -shared -o test4.so -I/usr/include/python2.6 -I/usr/local/include/boost -L/usr/local/lib -lboost_python
測試如下
>>> import test4>>> test = test4.Test()>>> print test.Add(1,2)103>>> print test.Add(1,2,z=3)6>>> print test4.Del(1,2)-1>>> print test4.Del(1,2,z=3)-1
匯出帶Python對象的介面
既然是匯出為Python介面,調用者難免會使用Python特有的資料結構,比如tuple,list,dict,由於原生態方法太麻煩,這裡只記錄boost的使用方法,假設要實現如下的Python函數功能
def Square(list_a){ return [x * x for x in list_a]}
即對傳入的list每個元素計算平方,返回list類型的結果
代碼如下
#include <boost/python.hpp>boost::python::list Square(boost::python::list& data){ boost::python::list ret; for (int i = 0; i < len(data); ++i) { ret.append(data[i] * data[i]); } return ret;}BOOST_PYTHON_MODULE(test5){ def("Square", Square);}
編譯命令如下
g++ test5.cpp -fPIC -shared -o test5.so -I/usr/include/python2.6 -I/usr/local/include/boost -L/usr/local/lib -lboost_python
測試如下
>>> import test5>>> test5.Square([1,2,3])[1, 4, 9]
boost實現了boost::python::tuple, boost::python::list, boost::python::dict這幾個資料類型,使用方法基本和Python保持一致,具體方法可以查看boost標頭檔裡的boost/python/tuple.hpp及其它對應檔案
另外比較常用的一個函數是boost::python::make_tuple() ,使用方法如下
boost::python::tuple(int a, int b, int c){ return boost::python::make_tuple(a, b, c);}
總結
以上就是這篇文章的全部內容了,希望本文的內容對大家的學習或者工作能帶來一定的協助,如果有疑問大家可以留言交流。