rename

api4cpp/rapidocr_api.cpp

@@ -0,0 +1,134 @@
+#include "../include/precomp.h"
+#ifdef __cplusplus
+extern "C"
+{
+#endif 
+
+	typedef struct
+	{
+		OcrLite OcrObj;
+		std::string strRes;
+	}OCR_OBJ;
+
+	_QM_OCR_API BPHANDLE   BPOcrInit(const char* szDetModel, const char* szClsModel, const char* szRecModel,const char *szKeyPath, int nThreads)
+	{
+
+		omp_set_num_threads(nThreads);
+		OCR_OBJ* pOcrObj=new OCR_OBJ;
+		if(pOcrObj)
+		{
+			pOcrObj->OcrObj.setNumThread(nThreads);
+
+			 pOcrObj->OcrObj.initModels(szDetModel, szClsModel, szRecModel, szKeyPath);
+			 
+			return pOcrObj;
+		}
+		else
+		{
+			return nullptr;
+		}
+		
+	}
+
+	
+	_QM_OCR_API BOOL  BPOcrDoOcr(BPHANDLE handle, const char* szImgPath, BOOL bAutoParam,BOOL bLongPic,BPOCR_PARAM *pParam)
+	{
+
+		OCR_OBJ* pOcrObj=(OCR_OBJ*)handle;
+		if(!pOcrObj)
+			return FALSE;
+
+		BPOCR_PARAM Param = *pParam;
+		if(bAutoParam)
+		{
+
+		}
+		else
+		{
+
+		// 	-numThread %NUMBER_OF_PROCESSORS% ^
+		// --padding 0 ^
+		// --maxSideLen %MAX_SIDE_LEN% ^
+		// --boxScoreThresh 0.5 ^
+		// --boxThresh 0.3 ^
+		// --unClipRatio %CLIP_RATE% ^
+		// --doAngle 1 ^
+		// --mostAngle 0
+			if(Param.boxScoreThresh ==0)
+				Param.boxScoreThresh = 0.5;
+			if(Param.boxThresh ==0)
+				Param.boxThresh =0.3;
+			if(Param.flagDoAngle ==0)
+				Param.flagDoAngle =1;
+			if(Param.flagMostAngle ==0)
+				Param.flagMostAngle =0;
+			
+			if( bLongPic)
+			{
+				if(Param.maxSideLen ==0)
+					Param.maxSideLen =0;
+				if(Param.unClipRatio ==0)
+					Param.unClipRatio = 3.0;
+			}
+			else
+			{
+				if(Param.maxSideLen ==0)
+					Param.maxSideLen =1024;
+				if(Param.unClipRatio ==0)
+					Param.unClipRatio = 1.5;
+			}
+		}
+		std::string imgPath,imgDir,imgName;
+		imgPath=szImgPath;
+		imgDir.assign(imgPath.substr(0, imgPath.find_last_of('\\') + 1));
+		imgName.assign(imgPath.substr(imgPath.find_last_of('\\') + 1));
+		OcrResult result=pOcrObj->OcrObj.detect(imgDir.c_str(),imgName.c_str(),Param.padding,Param.maxSideLen,Param.boxScoreThresh,Param.boxThresh,Param.unClipRatio,Param.flagDoAngle?true:false,Param.flagMostAngle?true:false);
+		if (result.strRes.length() > 0)
+		{
+	
+			pOcrObj->strRes = result.strRes; 
+			return TRUE;
+		}
+		else
+			return FALSE;
+	}
+
+
+	_QM_OCR_API  int BPOcrGetLen(BPHANDLE handle)
+	{
+		OCR_OBJ* pOcrObj = (OCR_OBJ*)handle;
+		if (!pOcrObj)
+			return 0;
+		return pOcrObj->strRes.size()+1;
+	}
+
+	_QM_OCR_API  BOOL BPOcrGetResult(BPHANDLE handle, char* szBuf, int nLen)
+	{
+
+		OCR_OBJ* pOcrObj = (OCR_OBJ*)handle;
+		if (!pOcrObj)
+			return FALSE;
+
+		if (nLen > pOcrObj->strRes.size())
+		{
+			strncpy(szBuf, pOcrObj->strRes.c_str(), pOcrObj->strRes.size());
+			szBuf[pOcrObj->strRes.size() - 1] = 0;
+		}
+		
+		return pOcrObj->strRes.size();
+	}
+
+	_QM_OCR_API void  BPOcrDeinit(BPHANDLE handle)
+	{
+
+		OCR_OBJ* pOcrObj=(OCR_OBJ*)handle;
+		if(pOcrObj)
+			delete pOcrObj;
+
+
+	}
+
+
+#ifdef __cplusplus
+}
+#endif