我正在构建一个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'
错误。
那么,我在做什么错了?
答案 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)