Python 调用 C/C++ 程序

贡献者： addis

• 参考这里。
• 大概就是先编译一个动态链接库。但注意 python 只能导入其中的 C 语言函数，不能直接导入对象等。
• 所以如果写 cpp 文件，必须在 extern "C" {} 定义要导出的函数（可以只是简单把 cpp 函数和对象封装一下，但不能使用相同的函数名否则编译器会报错）。
  代码 1：test.cpp

  int myfun_imp() {
  	return 123456;
  }
  
  extern "C" {
  	int myfun() { return myfun_imp(); }
  }
  

• 编译 出动态链接库（test.so）：g++ -shared -fPIC -o test.so test.cpp
• 在同目录打开 python，确定当前路径用 os.getcwd()
• from ctypes import cdll
• lib = cdll.LoadLibrary('./test.so')
• 现在调用 lib.myfun() 就会返回 123456 了！

传递数组

• 参考这里。例如定义以下函数

  void cfun(const double *indatav, size_t size, double *outdatav) 
  {
      size_t i;
      for (i = 0; i < size; ++i)
          outdatav[i] = indatav[i] * 2.0;
  }
  

• 修改函数的 protorype 使用 numpy 的指针

  import ctypes
  from numpy.ctypeslib import ndpointer
  lib = ctypes.cdll.LoadLibrary("./ctest.so")
  fun = lib.cfun
  fun.restype = None
  fun.argtypes = [ndpointer(ctypes.c_double, flags="C_CONTIGUOUS"),
                  ctypes.c_size_t,
                  ndpointer(ctypes.c_double, flags="C_CONTIGUOUS")]
  

• 然后就可以调用了

  indata = numpy.ones((5,6))
  outdata = numpy.empty((5,6))
  fun(indata, indata.size, outdata)