From ed25726d2992f8a395a8d4ed9bd5e85d231c471e Mon Sep 17 00:00:00 2001
From: root <root@localhost.localdomain>
Date: 星期五, 28 四月 2023 17:00:56 +0800
Subject: [PATCH] add offline websocket support

---
 funasr/runtime/onnxruntime/src/CMakeLists.txt |   57 +++++++++++++++++++++++++++------------------------------
 1 files changed, 27 insertions(+), 30 deletions(-)

diff --git a/funasr/runtime/onnxruntime/src/CMakeLists.txt b/funasr/runtime/onnxruntime/src/CMakeLists.txt
index 4842072..9c8d51c 100644
--- a/funasr/runtime/onnxruntime/src/CMakeLists.txt
+++ b/funasr/runtime/onnxruntime/src/CMakeLists.txt
@@ -1,43 +1,40 @@
 
 file(GLOB files1 "*.cpp")
-file(GLOB files4 "paraformer/*.cpp")
+file(GLOB files2 "*.cc")
 
-set(files ${files1} ${files2} ${files3} ${files4})
+set(files ${files1} ${files2})
+set(CMAKE_RUNTIME_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/bin)
 
-# message("${files}")
-
-add_library(rapidasr ${files})
+add_library(funasr ${files})
 
 if(WIN32)
-
-        set(EXTRA_LIBS libfftw3f-3 webrtcvad)
-        if(CMAKE_CL_64)
-            target_link_directories(rapidasr PUBLIC ${CMAKE_SOURCE_DIR}/win/lib/x64)
-        else()
-            target_link_directories(rapidasr PUBLIC ${CMAKE_SOURCE_DIR}/win/lib/x86)
-        endif()
-        target_include_directories(rapidasr PUBLIC ${CMAKE_SOURCE_DIR}/win/include )
-        
-
+    set(EXTRA_LIBS pthread yaml-cpp csrc glog)
+    if(CMAKE_CL_64)
+        target_link_directories(funasr PUBLIC ${CMAKE_SOURCE_DIR}/win/lib/x64)
+    else()
+        target_link_directories(funasr PUBLIC ${CMAKE_SOURCE_DIR}/win/lib/x86)
+    endif()
+    target_include_directories(funasr PUBLIC ${CMAKE_SOURCE_DIR}/win/include )
+    
+    target_compile_definitions(funasr PUBLIC -D_FUNASR_API_EXPORT)
 else()
-
-    set(EXTRA_LIBS fftw3f webrtcvad pthread)
-    target_include_directories(rapidasr PUBLIC "/usr/local/opt/fftw/include")
-    target_link_directories(rapidasr PUBLIC "/usr/local/opt/fftw/lib")
-
-    target_include_directories(rapidasr PUBLIC "/usr/local/opt/openblas/include")
-    target_link_directories(rapidasr PUBLIC "/usr/local/opt/openblas/lib")
-
-    target_include_directories(rapidasr PUBLIC "/usr/include")
-    target_link_directories(rapidasr PUBLIC "/usr/lib64")
-
-    target_include_directories(rapidasr PUBLIC  ${FFTW3F_INCLUDE_DIR})
-    target_link_directories(rapidasr PUBLIC ${FFTW3F_LIBRARY_DIR})
-    include_directories(${ONNXRUNTIME_DIR}/include)    
+    set(EXTRA_LIBS pthread yaml-cpp csrc glog )
+    include_directories(${ONNXRUNTIME_DIR}/include)
 endif()
 
 include_directories(${CMAKE_SOURCE_DIR}/include)
-target_link_libraries(rapidasr PUBLIC onnxruntime ${EXTRA_LIBS})
+target_link_libraries(funasr PUBLIC onnxruntime ${EXTRA_LIBS})
 
+if(ENABLE_WEBSOCKET)
+  add_executable(websocketmain "websocketmain.cpp")
+  add_executable(websocketclient "websocketclient.cpp")
 
+  target_link_libraries(websocketclient PUBLIC funasr)
+  target_link_libraries(websocketmain PUBLIC funasr)
+endif()
+
+add_executable(funasr-onnx-offline "funasr-onnx-offline.cpp")
+add_executable(funasr-onnx-offline-rtf "funasr-onnx-offline-rtf.cpp")
+target_link_libraries(funasr-onnx-offline PUBLIC funasr)
+target_link_libraries(funasr-onnx-offline-rtf PUBLIC funasr)
 

--
Gitblit v1.9.1