mirror of
https://github.com/kerberos-io/openalpr-base.git
synced 2025-10-07 07:10:53 +08:00
92 lines
2.9 KiB
Python
92 lines
2.9 KiB
Python
import ctypes
|
|
import json
|
|
import platform
|
|
|
|
class Alpr():
|
|
def __init__(self, country, config_file, runtime_dir):
|
|
|
|
# Load the .dll for Windows and the .so for Unix-based
|
|
if platform.system().lower().find("windows") != -1:
|
|
self._openalprpy_lib = ctypes.cdll.LoadLibrary("openalprpy.dll")
|
|
else:
|
|
self._openalprpy_lib = ctypes.cdll.LoadLibrary("libopenalprpy.so")
|
|
|
|
|
|
self._initialize_func = self._openalprpy_lib.initialize
|
|
self._initialize_func.argtypes = [ctypes.c_char_p, ctypes.c_char_p, ctypes.c_char_p]
|
|
|
|
self._dispose_func = self._openalprpy_lib.dispose
|
|
|
|
self._is_loaded_func = self._openalprpy_lib.isLoaded
|
|
self._is_loaded_func.restype = ctypes.c_bool
|
|
|
|
self._recognize_file_func = self._openalprpy_lib.recognizeFile
|
|
self._recognize_file_func.restype = ctypes.c_void_p
|
|
self._recognize_file_func.argtypes = [ctypes.c_char_p]
|
|
|
|
self._recognize_array_func = self._openalprpy_lib.recognizeArray
|
|
self._recognize_array_func.restype = ctypes.c_void_p
|
|
self._recognize_array_func.argtypes = [ctypes.POINTER(ctypes.c_ubyte), ctypes.c_uint]
|
|
|
|
self._free_json_mem_func = self._openalprpy_lib.freeJsonMem
|
|
|
|
|
|
self._set_default_region_func = self._openalprpy_lib.setDefaultRegion
|
|
self._set_default_region_func.argtypes = [ctypes.c_char_p]
|
|
|
|
self._set_detect_region_func = self._openalprpy_lib.setDetectRegion
|
|
self._set_detect_region_func.argtypes = [ctypes.c_bool]
|
|
|
|
|
|
self._set_top_n_func = self._openalprpy_lib.setTopN
|
|
self._set_top_n_func.argtypes = [ctypes.c_int]
|
|
|
|
self._get_version_func = self._openalprpy_lib.getVersion
|
|
self._get_version_func.restype = ctypes.c_void_p
|
|
|
|
self._initialize_func(country, config_file, runtime_dir)
|
|
|
|
|
|
def unload(self):
|
|
self._openalprpy_lib.dispose()
|
|
|
|
def is_loaded(self):
|
|
return self._is_loaded_func()
|
|
|
|
def recognize_file(self, file_path):
|
|
ptr = self._recognize_file_func(file_path)
|
|
json_data = ctypes.cast(ptr, ctypes.c_char_p).value
|
|
response_obj = json.loads(json_data)
|
|
self._free_json_mem_func(ptr)
|
|
|
|
return response_obj
|
|
|
|
def recognize_array(self, byte_array):
|
|
|
|
pb = ctypes.cast(byte_array, ctypes.POINTER(ctypes.c_ubyte))
|
|
ptr = self._recognize_array_func(pb, len(byte_array))
|
|
json_data = ctypes.cast(ptr, ctypes.c_char_p).value
|
|
response_obj = json.loads(json_data)
|
|
self._free_json_mem_func(ptr)
|
|
|
|
return response_obj
|
|
|
|
def get_version(self):
|
|
|
|
ptr = self._get_version_func()
|
|
version_number = ctypes.cast(ptr, ctypes.c_char_p).value
|
|
self._free_json_mem_func(ptr)
|
|
|
|
return version_number
|
|
|
|
def set_top_n(self, topn):
|
|
self._set_top_n_func(topn)
|
|
|
|
def set_default_region(self, region):
|
|
self._set_default_region_func(region)
|
|
|
|
def set_detect_region(self, enabled):
|
|
self._set_detect_region_func(enabled)
|
|
|
|
|