Refactor project structure

2026-03-28 19:56:39 +11:00
parent 1d267378b2
commit 8a2e721058
511 changed files with 59 additions and 48 deletions
--- a/modules/ANSLPR/ANSLPR_RT.h
+++ b/modules/ANSLPR/ANSLPR_RT.h
@@ -0,0 +1,97 @@
+#ifndef ANSLPRRT_H
+#define ANSLPRRT_H
+#pragma once
+#include "ANSLPR.h"
+#include <onnxruntime_c_api.h>
+#include <onnxruntime_cxx_api.h>
+#include "yolov5_alpr_onnx_detector.h"
+#include "ONNX_detector.h"
+#include <thread>         // std::thread
+#include <mutex>          // std::mutex, std::unique_lock, std::defer_lock
+
+namespace ANSCENTER
+{
+	class ANSLPR_API ANSALPR_RT :public ANSALPR {
+	private:
+		std::mutex mtxMutex;      // mutex for critical section
+		std::string _globalModelFileName{};
+		std::string _focusedOnLPRModelFileName{};
+		std::string _platesTypesClassifierFileName{};
+		std::string _platesTypesLabelsFileName{};
+
+		size_t _globalViewId;
+		size_t _focusedOnLPRId;
+		size_t _platesTypesClassifierId;
+		[[nodiscard]] bool   CloseReferences();
+
+	public:
+		ANSALPR_RT();
+		~ANSALPR_RT();
+		[[nodiscard]] bool Initialize(const std::string& licenseKey, const std::string& modelZipFilePath, const std::string& modelZipPassword) override;
+		[[nodiscard]] bool Inference(const cv::Mat& input, std::string& lprResult);
+		[[nodiscard]] bool Inference(const cv::Mat& input, const std::vector<cv::Rect>& Bbox, std::string& lprResult);
+		[[nodiscard]] bool Destroy() override;
+
+	private:
+		[[nodiscard]] std::string VectorDetectionToJsonString(const std::vector<Object>& dets);
+		void CheckStatus(OrtStatus* status);
+		cv::Rect GetGlobalROI(std::list<cv::Rect> ROIs);
+		std::list<Yolov5_alpr_onxx_detector*>::const_iterator	GetDetector(unsigned int id, const std::list<Yolov5_alpr_onxx_detector*>& detectors,
+			const std::list<unsigned int>& detectors_ids);
+		std::list<Plates_types_classifier*>::const_iterator		GetPlatesTypesClassifier(unsigned int id, const std::list<Plates_types_classifier*>& plates_types_classifiers,
+			const std::list<unsigned int>& plates_types_classifiers_ids);
+		unsigned int GetNewId(const std::list<unsigned int>& detectors_ids);
+		bool CloseDetector(unsigned int id, std::list<Ort::Env*>& _envs, std::list<Ort::SessionOptions*>& _lsessionOptions, std::list<Yolov5_alpr_onxx_detector*>& _detectors,
+			std::list<unsigned int>& _detectors_ids);
+		bool CloseDetector(unsigned int id, std::list<Ort::Env*>& _envs, std::list<Ort::SessionOptions*>& _lsessionOptions, std::list<Plates_types_classifier*>& _detectors,
+			std::list<unsigned int>& _detectors_ids);
+
+		unsigned int InitYoloDetector(unsigned int len, const char* model_file);
+		unsigned int InitPlatesClassifer(unsigned int len_models_filename, const char* model_file, unsigned int len_labels_filename, const char* labels_file);
+		bool TwoStageALPR(const int width,//width of image
+			const int height,//height of image i.e. the specified dimensions of the image
+			const int pixOpt,// pixel type : 1 (8 bpp greyscale image) 3 (RGB 24 bpp image) or 4 (RGBA 32 bpp image)
+			void* pbData, unsigned int step,// source image bytes buffer
+			unsigned int id_global_view,
+			unsigned int id_focused_on_lp,
+			unsigned int lpn_len, char* lpn, cv::Rect& bbox);
+		bool TwoStageLPRPlatesTypeDetection(const int width,//width of image
+			const int height,//height of image i.e. the specified dimensions of the image
+			const int pixOpt,// pixel type : 1 (8 bpp greyscale image) 3 (RGB 24 bpp image) or 4 (RGBA 32 bpp image)
+			void* pbData,
+			unsigned int step,// source image bytes buffer
+			unsigned int id_global_view,
+			unsigned int id_focused_on_lp,
+			unsigned int id_plates_types_classifier,
+			unsigned int lpn_len,
+			char* lpn,
+			cv::Rect& bbox);
+		bool CloseDetector(unsigned int id);
+		bool ClosePlatesTypesClassifier(unsigned int id);
+
+	private:
+		// Version-negotiated OrtApi pointer: caps at the DLL's max supported API level so
+		// that a newer SDK header (ORT_API_VERSION=22) paired with an older runtime DLL
+		// (e.g. ORT 1.17.1, API ≤17) does not emit "[ORT ERROR] API version not available".
+		// Mirrors the negotiation pattern used in EPLoader.cpp.
+		const OrtApi* g_ort = []() -> const OrtApi* {
+			const OrtApiBase* base = OrtGetApiBase();
+			int dllMaxApi = ORT_API_VERSION;
+			const char* verStr = base->GetVersionString();
+			int major = 0, minor = 0;
+			if (verStr && sscanf(verStr, "%d.%d", &major, &minor) == 2)
+				dllMaxApi = minor;
+			int targetApi = (ORT_API_VERSION < dllMaxApi) ? ORT_API_VERSION : dllMaxApi;
+			return base->GetApi(targetApi);
+		}();
+		std::list<Ort::Env*> detectors_envs;
+		std::list<Ort::SessionOptions*> l_detectors_sessionOptions;
+		std::list<Yolov5_alpr_onxx_detector*> detectors;
+		std::list<unsigned int> detectors_ids;
+		std::list<Ort::Env*> plates_types_envs;
+		std::list<Ort::SessionOptions*> l_plates_types_classifier_sessionOptions;
+		std::list<Plates_types_classifier*> plates_types_classifiers;
+		std::list<unsigned int> plates_types_classifier_ids;
+	};
+}
+#endif