From 096d99e0689b69777aeedf98689bb344efb70b39 Mon Sep 17 00:00:00 2001
From: Sven <xiang.zhang@verisilicon.com>
Date: Thu, 13 May 2021 22:28:36 +0800
Subject: [PATCH] Support local sdk for cmake (#41)

* Support local sdk for cmake

cmake -DEXTERNAL_VIV_SDK=<Local_Driver_build_sdk_dir>

Signed-off-by: xiang.zhang <xiang.zhang@verisilicon.com>
---
 CMakeLists.txt        | 19 ++++++++++++-------
 cmake/local_sdk.cmake | 19 +++++++++++++++++++
 2 files changed, 31 insertions(+), 7 deletions(-)
 create mode 100644 cmake/local_sdk.cmake

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 17e1ecf..a106f30 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -11,14 +11,19 @@ set(OVXLIB_API_ATTR "__attribute__\(\(visibility\(\"default\"\)\)\)")
 message(${OVXLIB_API_ATTR})
 add_definitions(-DOVXLIB_API=${OVXLIB_API_ATTR})
 
-if("${CONFIG}" STREQUAL "A311D")
-    include(cmake/A311D.cmake)
-elseif("${CONFIG}" STREQUAL "S905D3")
-    include(cmake/S905D3.cmake)
-elseif("${CONFIG}" STREQUAL "vim3_android")
-    include(cmake/vim3_android.cmake)
+if(EXTERNAL_VIV_SDK AND EXISTS ${EXTERNAL_VIV_SDK})
+    # this is for internal development purpose
+    include(cmake/local_sdk.cmake)
 else()
-    include(cmake/X86_64_linux.cmake)
+    if("${CONFIG}" STREQUAL "A311D")
+        include(cmake/A311D.cmake)
+    elseif("${CONFIG}" STREQUAL "S905D3")
+        include(cmake/S905D3.cmake)
+    elseif("${CONFIG}" STREQUAL "vim3_android")
+        include(cmake/vim3_android.cmake)
+    else()
+        include(cmake/X86_64_linux.cmake)
+    endif()
 endif()
 
 include_directories(${PROJECT_SOURCE_DIR}/include/tim/vx)
diff --git a/cmake/local_sdk.cmake b/cmake/local_sdk.cmake
new file mode 100644
index 0000000..2875f69
--- /dev/null
+++ b/cmake/local_sdk.cmake
@@ -0,0 +1,19 @@
+set(PKG_NAME "OVXDRV")
+
+message("include driver sdk from ${EXTERNAL_VIV_SDK}")
+set(OVXDRV_INCLUDE_DIRS)
+list(APPEND OVXDRV_INCLUDE_DIRS
+        ${EXTERNAL_VIV_SDK}include
+        ${EXTERNAL_VIV_SDK}include/CL)
+
+set(OVXDRV_LIBRARIES)
+list(APPEND OVXDRV_LIBRARIES
+    ${EXTERNAL_VIV_SDK}/drivers/libCLC.so
+    ${EXTERNAL_VIV_SDK}/drivers/libGAL.so
+    ${EXTERNAL_VIV_SDK}/drivers/libOpenVX.so
+    ${EXTERNAL_VIV_SDK}/drivers/libOpenVXU.so
+    ${EXTERNAL_VIV_SDK}/drivers/libVSC.so
+    ${EXTERNAL_VIV_SDK}/drivers/libArchModelSw.so
+    ${EXTERNAL_VIV_SDK}/drivers/libNNArchPerf.so)
+
+mark_as_advanced(${OVXDRV_INCLUDE_DIRS} ${OVXDRV_LIBRARIES})