29 lines
991 B
Python
29 lines
991 B
Python
 | import ctypes
| ||
import os
| |||
| |||
# Define the argument and result types for the DLL
| |||
# Assume DLL is built as ExportFeatureBatch(void* pb_bytes, int pb_len, char* out_path, int out_path_len)
| |||
| |||
DLL_PATH = r"C:\DualEA_FeatureBatches\feature_export.dll" # Adjust as needed
| |||
| |||
dll = ctypes.CDLL(DLL_PATH)
| |||
| |||
ExportFeatureBatch = dll.ExportFeatureBatch
| |||
ExportFeatureBatch.argtypes = [ctypes.c_void_p, ctypes.c_int, ctypes.c_char_p, ctypes.c_int]
| |||
ExportFeatureBatch.restype = ctypes.c_int
| |||
| |||
def export_feature_batch(pb_bytes: bytes) -> str:
| |||
out_path_buf = ctypes.create_string_buffer(512)
| |||
result = ExportFeatureBatch(
| |||
ctypes.c_char_p(pb_bytes),
| |||
len(pb_bytes),
| |||
out_path_buf,
| |||
ctypes.sizeof(out_path_buf)
| |||
)
| |||
if result != 0:
| |||
raise RuntimeError(f"DLL ExportFeatureBatch failed with code {result}")
| |||
return out_path_buf.value.decode('utf-8')
| |||
| |||
# Usage example (assuming you have pb_bytes):
| |||
# path = export_feature_batch(pb_bytes)
| |||
# print("Exported to:", path)
|