82 lines
2.3 KiB
C++
82 lines
2.3 KiB
C++
#include <pybind11/pybind11.h>
|
|
#include <pybind11/embed.h>
|
|
#include <string>
|
|
#include <iostream>
|
|
|
|
namespace py = pybind11;
|
|
|
|
// 全局解释器持有对象
|
|
py::scoped_interpreter* g_python_interpreter = nullptr;
|
|
|
|
// 初始化 Python 解释器
|
|
extern "C" __declspec(dllexport) bool InitializePython()
|
|
{
|
|
try {
|
|
if (!g_python_interpreter) {
|
|
g_python_interpreter = new py::scoped_interpreter();
|
|
}
|
|
return true;
|
|
} catch (const std::exception& e) {
|
|
std::cerr << "Failed to initialize Python: " << e.what() << std::endl;
|
|
return false;
|
|
}
|
|
}
|
|
|
|
// 清理 Python 解释器
|
|
extern "C" __declspec(dllexport) void FinalizePython()
|
|
{
|
|
if (g_python_interpreter) {
|
|
delete g_python_interpreter;
|
|
g_python_interpreter = nullptr;
|
|
}
|
|
}
|
|
|
|
// 执行 doWork 函数
|
|
extern "C" __declspec(dllexport) bool RunConversion(const char* script_path)
|
|
{
|
|
try {
|
|
if (!g_python_interpreter) {
|
|
std::cerr << "Python interpreter not initialized!" << std::endl;
|
|
return false;
|
|
}
|
|
|
|
// 添加脚本路径到 sys.path
|
|
py::module sys = py::module::import("sys");
|
|
py::object path = sys.attr("path");
|
|
|
|
// 获取脚本目录
|
|
std::string script_dir = script_path;
|
|
size_t last_sep = script_dir.find_last_of("/\\");
|
|
if (last_sep != std::string::npos) {
|
|
script_dir = script_dir.substr(0, last_sep);
|
|
}
|
|
|
|
// 添加到 sys.path
|
|
path.attr("insert")(0, script_dir);
|
|
|
|
// 导入脚本模块
|
|
py::module script_module = py::module::import("main");
|
|
|
|
// 调用 doWork 函数
|
|
script_module.attr("doWork")();
|
|
|
|
return true;
|
|
} catch (const py::error_already_set& e) {
|
|
std::cerr << "Python error: " << e.what() << std::endl;
|
|
return false;
|
|
} catch (const std::exception& e) {
|
|
std::cerr << "Error: " << e.what() << std::endl;
|
|
return false;
|
|
}
|
|
}
|
|
|
|
// 简单的 C API 封装
|
|
PYBIND11_MODULE(uf2touft3, m) {
|
|
m.doc() = "UFT2 to UFT3 code conversion module";
|
|
|
|
m.def("initialize", &InitializePython, "Initialize Python interpreter");
|
|
m.def("finalize", &FinalizePython, "Finalize Python interpreter");
|
|
m.def("run_conversion", &RunConversion, "Run the conversion process",
|
|
py::arg("script_path"));
|
|
}
|