mirror of
https://github.com/kerberos-io/openalpr-base.git
synced 2025-10-06 16:27:07 +08:00
Fixed thread safety issue in Python binding
This commit is contained in:
@@ -9,52 +9,56 @@ class Alpr():
|
||||
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._openalprpy_lib = ctypes.cdll.LoadLibrary("/storage/projects/alpr/src/build/Debug/libopenalprpy.so")
|
||||
|
||||
|
||||
self._initialize_func = self._openalprpy_lib.initialize
|
||||
self._initialize_func.restype = ctypes.c_void_p
|
||||
self._initialize_func.argtypes = [ctypes.c_char_p, ctypes.c_char_p, ctypes.c_char_p]
|
||||
|
||||
self._dispose_func = self._openalprpy_lib.dispose
|
||||
self._dispose_func.argtypes = [ctypes.c_void_p]
|
||||
|
||||
self._is_loaded_func = self._openalprpy_lib.isLoaded
|
||||
self._is_loaded_func.argtypes = [ctypes.c_void_p]
|
||||
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_file_func.argtypes = [ctypes.c_void_p, 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._recognize_array_func.argtypes = [ctypes.c_void_p, 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_default_region_func.argtypes = [ctypes.c_void_p, ctypes.c_char_p]
|
||||
|
||||
self._set_detect_region_func = self._openalprpy_lib.setDetectRegion
|
||||
self._set_detect_region_func.argtypes = [ctypes.c_bool]
|
||||
self._set_detect_region_func.argtypes = [ctypes.c_void_p, ctypes.c_bool]
|
||||
|
||||
|
||||
self._set_top_n_func = self._openalprpy_lib.setTopN
|
||||
self._set_top_n_func.argtypes = [ctypes.c_int]
|
||||
self._set_top_n_func.argtypes = [ctypes.c_void_p, ctypes.c_int]
|
||||
|
||||
self._get_version_func = self._openalprpy_lib.getVersion
|
||||
self._get_version_func.argtypes = [ctypes.c_void_p]
|
||||
self._get_version_func.restype = ctypes.c_void_p
|
||||
|
||||
self._initialize_func(country, config_file, runtime_dir)
|
||||
self.alpr_pointer = self._initialize_func(country, config_file, runtime_dir)
|
||||
|
||||
|
||||
def unload(self):
|
||||
self._openalprpy_lib.dispose()
|
||||
self._openalprpy_lib.dispose(self.alpr_pointer)
|
||||
|
||||
def is_loaded(self):
|
||||
return self._is_loaded_func()
|
||||
return self._is_loaded_func(self.alpr_pointer)
|
||||
|
||||
def recognize_file(self, file_path):
|
||||
ptr = self._recognize_file_func(file_path)
|
||||
ptr = self._recognize_file_func(self.alpr_pointer, file_path)
|
||||
json_data = ctypes.cast(ptr, ctypes.c_char_p).value
|
||||
response_obj = json.loads(json_data)
|
||||
self._free_json_mem_func(ctypes.c_void_p(ptr))
|
||||
@@ -64,7 +68,7 @@ class Alpr():
|
||||
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))
|
||||
ptr = self._recognize_array_func(self.alpr_pointer, 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(ctypes.c_void_p(ptr))
|
||||
@@ -73,19 +77,19 @@ class Alpr():
|
||||
|
||||
def get_version(self):
|
||||
|
||||
ptr = self._get_version_func()
|
||||
ptr = self._get_version_func(self.alpr_pointer)
|
||||
version_number = ctypes.cast(ptr, ctypes.c_char_p).value
|
||||
self._free_json_mem_func(ctypes.c_void_p(ptr))
|
||||
|
||||
return version_number
|
||||
|
||||
def set_top_n(self, topn):
|
||||
self._set_top_n_func(topn)
|
||||
self._set_top_n_func(self.alpr_pointer, topn)
|
||||
|
||||
def set_default_region(self, region):
|
||||
self._set_default_region_func(region)
|
||||
self._set_default_region_func(self.alpr_pointer, region)
|
||||
|
||||
def set_detect_region(self, enabled):
|
||||
self._set_detect_region_func(enabled)
|
||||
self._set_detect_region_func(self.alpr_pointer, enabled)
|
||||
|
||||
|
||||
|
@@ -16,10 +16,8 @@ extern "C" {
|
||||
|
||||
using namespace alpr;
|
||||
|
||||
bool initialized = false;
|
||||
static Alpr* nativeAlpr;
|
||||
|
||||
OPENALPR_EXPORT void initialize(char* ccountry, char* cconfigFile, char* cruntimeDir)
|
||||
OPENALPR_EXPORT Alpr* initialize(char* ccountry, char* cconfigFile, char* cruntimeDir)
|
||||
{
|
||||
//printf("Initialize");
|
||||
|
||||
@@ -29,35 +27,29 @@ extern "C" {
|
||||
std::string runtimeDir(cruntimeDir);
|
||||
|
||||
//std::cout << country << std::endl << configFile << std::endl << runtimeDir << std::endl;
|
||||
nativeAlpr = new alpr::Alpr(country, configFile, runtimeDir);
|
||||
Alpr* nativeAlpr = new alpr::Alpr(country, configFile, runtimeDir);
|
||||
|
||||
initialized = true;
|
||||
return;
|
||||
return nativeAlpr;
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
OPENALPR_EXPORT void dispose()
|
||||
OPENALPR_EXPORT void dispose(Alpr* nativeAlpr)
|
||||
{
|
||||
//printf("Dispose");
|
||||
initialized = false;
|
||||
delete nativeAlpr;
|
||||
}
|
||||
|
||||
|
||||
OPENALPR_EXPORT bool isLoaded()
|
||||
OPENALPR_EXPORT bool isLoaded(Alpr* nativeAlpr)
|
||||
{
|
||||
//printf("IS LOADED");
|
||||
|
||||
if (!initialized)
|
||||
return false;
|
||||
|
||||
return nativeAlpr->isLoaded();
|
||||
|
||||
}
|
||||
|
||||
OPENALPR_EXPORT char* recognizeFile(char* cimageFile)
|
||||
OPENALPR_EXPORT char* recognizeFile(Alpr* nativeAlpr, char* cimageFile)
|
||||
{
|
||||
//printf("Recognize file");
|
||||
|
||||
@@ -83,12 +75,13 @@ extern "C" {
|
||||
}
|
||||
|
||||
|
||||
OPENALPR_EXPORT char* recognizeArray(unsigned char* buf, int len)
|
||||
OPENALPR_EXPORT char* recognizeArray(Alpr* nativeAlpr, unsigned char* buf, int len)
|
||||
{
|
||||
//printf("Recognize byte array");
|
||||
//printf("buffer pointer: %p\n", buf);
|
||||
//printf("buffer length: %d\n", len);
|
||||
|
||||
//std::cout << "Using instance: " << nativeAlpr << std::endl;
|
||||
|
||||
std::vector<char> cvec(buf, buf+len);
|
||||
|
||||
@@ -103,7 +96,7 @@ extern "C" {
|
||||
return membuffer;
|
||||
}
|
||||
|
||||
OPENALPR_EXPORT void setDefaultRegion(char* cdefault_region)
|
||||
OPENALPR_EXPORT void setDefaultRegion(Alpr* nativeAlpr, char* cdefault_region)
|
||||
{
|
||||
// Convert strings from java to C++ and release resources
|
||||
std::string default_region(cdefault_region);
|
||||
@@ -111,17 +104,17 @@ extern "C" {
|
||||
nativeAlpr->setDefaultRegion(default_region);
|
||||
}
|
||||
|
||||
OPENALPR_EXPORT void setDetectRegion(bool detect_region)
|
||||
OPENALPR_EXPORT void setDetectRegion(Alpr* nativeAlpr, bool detect_region)
|
||||
{
|
||||
nativeAlpr->setDetectRegion(detect_region);
|
||||
}
|
||||
|
||||
OPENALPR_EXPORT void setTopN(int top_n)
|
||||
OPENALPR_EXPORT void setTopN(Alpr* nativeAlpr, int top_n)
|
||||
{
|
||||
nativeAlpr->setTopN(top_n);
|
||||
}
|
||||
|
||||
OPENALPR_EXPORT char* getVersion()
|
||||
OPENALPR_EXPORT char* getVersion(Alpr* nativeAlpr)
|
||||
{
|
||||
std::string version = nativeAlpr->getVersion();
|
||||
|
||||
|
Reference in New Issue
Block a user