| // feature_export_dll.cpp
|
| // Full C++ DLL implementation (stdcall, Python embedding, raw protobuf bytes)
|
| // Compile with: cl /LD feature_export_dll.cpp /I<python_include> /link /LIBPATH:<python_lib> python3x.lib
|
| #include <windows.h>
|
| #include <Python.h>
|
| #include <string>
|
| #include <stdexcept>
|
| #include <cstring>
|
|
|
| // Helper: Convert UTF-8 std::string to wide string
|
| std::wstring utf8_to_wide(const std::string& str) {
|
| int size_needed = MultiByteToWideChar(CP_UTF8, 0, str.c_str(), (int)str.size(), NULL, 0);
|
| std::wstring wstrTo(size_needed, 0);
|
| MultiByteToWideChar(CP_UTF8, 0, str.c_str(), (int)str.size(), &wstrTo[0], size_needed);
|
| return wstrTo;
|
| }
|
|
|
| extern "C" __declspec(dllexport) int __stdcall ExportFeatureBatch(const void* pb_bytes, int pb_len, char* out_path, int out_path_len) {
|
| int result = 0;
|
| Py_Initialize();
|
| try {
|
| PyObject* sys_path = PySys_GetObject("path");
|
| PyList_Append(sys_path, PyUnicode_FromString("./python")); // or absolute path
|
| PyObject* module = PyImport_ImportModule("feature_export");
|
| if (!module) throw std::runtime_error("Could not import feature_export.py");
|
| PyObject* func = PyObject_GetAttrString(module, "write_batches_to_parquet");
|
| if (!func || !PyCallable_Check(func)) throw std::runtime_error("No write_batches_to_parquet");
|
| PyObject* py_bytes = PyBytes_FromStringAndSize((const char*)pb_bytes, pb_len);
|
| PyObject* args = PyTuple_Pack(1, py_bytes);
|
| PyObject* ret = PyObject_CallObject(func, args);
|
| if (!ret) throw std::runtime_error("Python call failed");
|
| const char* path = PyUnicode_AsUTF8(ret);
|
| if (!path) throw std::runtime_error("Return value not string");
|
| strncpy(out_path, path, out_path_len - 1);
|
| out_path[out_path_len - 1] = '\0';
|
| Py_XDECREF(ret);
|
| Py_XDECREF(args);
|
| Py_XDECREF(py_bytes);
|
| Py_XDECREF(func);
|
| Py_XDECREF(module);
|
| } catch (...) {
|
| result = -1;
|
| }
|
| Py_Finalize();
|
| return result;
|
| }
|
|
|
| BOOL APIENTRY DllMain(HMODULE hModule, DWORD ul_reason_for_call, LPVOID lpReserved) {
|
| return TRUE;
|
| }
|
