Merge pull request #273 from alibaba-damo-academy/dev_onnx

zhifu gao

2023-03-21 4986ec2dd3012607e7bf771041cff45edc6c5045

 funasr/runtime/onnxruntime/src/librapidasrapi.cpp

@@ -4,24 +4,16 @@
extern "C" {
#endif


   // APIs for qmasr
   _RAPIDASRAPI RPASR_HANDLE  RapidAsrInit(const char* szModelDir, int nThreadNum)
   _RAPIDASRAPI RPASR_HANDLE  RapidAsrInit(const char* szModelDir, int nThreadNum, bool quantize)
   {


      Model* mm = create_model(szModelDir, nThreadNum); 

      Model* mm = create_model(szModelDir, nThreadNum, quantize);
      return mm;
   }


   _RAPIDASRAPI RPASR_RESULT RapidAsrRecogBuffer(RPASR_HANDLE handle, const char* szBuf, int nLen, RPASR_MODE Mode, QM_CALLBACK fnCallback)
   {


      Model* pRecogObj = (Model*)handle;

      if (!pRecogObj)
         return nullptr;

@@ -46,15 +38,12 @@
            fnCallback(nStep, nTotal);
      }


      return pResult;
   }

   _RAPIDASRAPI RPASR_RESULT RapidAsrRecogPCMBuffer(RPASR_HANDLE handle, const char* szBuf, int nLen, RPASR_MODE Mode, QM_CALLBACK fnCallback)
   {

      Model* pRecogObj = (Model*)handle;

      if (!pRecogObj)
         return nullptr;

@@ -79,16 +68,12 @@
            fnCallback(nStep, nTotal);
      }


      return pResult;

   }

   _RAPIDASRAPI RPASR_RESULT RapidAsrRecogPCMFile(RPASR_HANDLE handle, const char* szFileName, RPASR_MODE Mode, QM_CALLBACK fnCallback)
   {

      Model* pRecogObj = (Model*)handle;

      if (!pRecogObj)
         return nullptr;

@@ -113,15 +98,12 @@
            fnCallback(nStep, nTotal);
      }


      return pResult;

   }

   _RAPIDASRAPI RPASR_RESULT RapidAsrRecogFile(RPASR_HANDLE handle, const char* szWavfile, RPASR_MODE Mode, QM_CALLBACK fnCallback)
   {
      Model* pRecogObj = (Model*)handle;

      if (!pRecogObj)
         return nullptr;

@@ -146,9 +128,6 @@
            fnCallback(nStep, nTotal);
      }
   
	


      return pResult;
   }

@@ -158,7 +137,6 @@
         return 0;

      return 1;
		
   }


@@ -168,7 +146,6 @@
         return 0.0f;

      return ((RPASR_RECOG_RESULT*)Result)->snippet_time;

   }

   _RAPIDASRAPI const char* RapidAsrGetResult(RPASR_RESULT Result,int nIndex)
@@ -178,33 +155,25 @@
         return nullptr;

      return pResult->msg.c_str();
	
   }

   _RAPIDASRAPI void RapidAsrFreeResult(RPASR_RESULT Result)
   {

      if (Result)
      {
         delete (RPASR_RECOG_RESULT*)Result;

      }
   }

   _RAPIDASRAPI void RapidAsrUninit(RPASR_HANDLE handle)
   {

      Model* pRecogObj = (Model*)handle;


      if (!pRecogObj)
         return;

      delete pRecogObj;

   }



#ifdef __cplusplus