Python 3.8:运行外部C ++代码:无法导入模块

时间:2020-06-11 11:29:54

标签: python c++ python-3.x

我正在构建一个C ++ DLL文件,试图将其加载到Python模块中。
我正在使用the official Python Documentation。 C ++代码(包括所有样板)如下(一个酒窝“ hello”函数):

#include <Python.h>

#include <iostream>

PyObject* fft_hello(PyObject *self, PyObject *args){
  std::cout << "Insid fft hello" << std::endl;
  return nullptr;
}

static PyMethodDef fft_methods[2] = {
    {"hello", fft_hello, METH_VARARGS, "Runs hello"},
    {nullptr, nullptr, 0, nullptr}
};

static struct PyModuleDef fft_module = {
    PyModuleDef_HEAD_INIT,
    "fft",
    "fft module",
    0,
    fft_methods
};

PyMODINIT_FUNC
PyInit_fft(void){
  std::cout << "INITING" << std::endl;
  return PyModule_Create(&fft_module);
}

编译模块的CMAKE是:

cmake_minimum_required(VERSION 3.16)
project(fft)

set(CMAKE_CXX_STANDARD 14)
link_directories("C:/Users/guyy/AppData/Local/Programs/Python/Python38/libs")

add_library(fft SHARED fft.cpp )

include_directories("C:/Users/guyy/AppData/Local/Programs/Python/Python38/include")

这将产生一个fft.dll模块。

接下来,我将这个dll放在python文件目录中,并尝试导入它:

import fft

if __name__ == '__main__':
    pass

我遇到了ModuleNotFoundError: No module named 'fft'错误。

那么,我在做什么错了?

  • C ++文件写错了吗?
  • DLL模块是否构建错误?
  • DLL模块放置在错误的位置吗?

2 个答案:

答案 0 :(得分:0)

检查:https://docs.python.org/2/library/ctypes.html或CPython。

[编辑] 您也可以使用use distutils。在根文件夹中,创建一个名为setup.py的文件,该文件具有以下实现:

从distutils.core导入设置(扩展程序)

def main():
    setup(name="fft",
          version="1.0.0",
          description="Python interface for the fft C library function",
          author="<your name>",
          author_email="your_email@gmail.com",
          ext_modules=[Extension("fft", ["fft.cpp"])])

if __name__ == "__main__":
    main()

运行以下命令:

python setup.py install

基本上,这将创建一个名为fft%python_version_and_distribution%.pyc的模块,您可以将其添加到文件夹的根目录中。现在,您的python文件可以读取此模块了。 您也可以在此模块上运行cmake,但它稍微复杂一些。 我建议检查distutils库以获取更多详细信息。

我希望这会有所帮助

答案 1 :(得分:0)

您是否绝对需要将 DLL 作为“模块”加载?

使用 CTYPES 并没有那么复杂。我将它与 Python 3.8 和使用 VC++2010 生成的 DLL 一起使用(此 DLL 实现了多线程代码并且运行良好)。

在 C/C++ 代码方面: - 为所有可调用函数加上前缀“extern "C"”,最好使用标准类型的变量和指针,例如:

extern "C" int myTestFunc(int par1, int par2, float *array1)
{
  ...
  return 0;
}

在 Python 代码方面: -导入以下模块: 导入 ctypes 导入_ctypes

-在您的代码/类中,将 DLL 加载为:

self.lib = ctypes.CDLL('<full_path_to_your_DLL>')

-声明函数的返回类型(函数名必须与C代码中的相同):

self.lib.myTestFunc.restype = ctypes.c_int

-定义函数参数类型(以下是标准值参数以及指向 numpy 浮点数组的指针):

self.lib.myTestFunc.argtypes = [ctypes.c_int, ctypes.c_int, np.ctypeslib.ndpointer(dtype=np.float32)]

-在您的代码中,只需调用该函数:

val1 = 1
val2 = 2
float_array = np.zeros((20, 50),dtype=np.float32)
retval = self.lib.myTestFunc(val1,val2,float_array)

-如果需要释放DLL访问权限,可以使用如下:

_ctypes.dlclose(self.lib._handle)