react-native-fast-tflite/android/CMakeLists.txt

High-performance TensorFlow Lite library for React Native

project(VisionCameraTflite)
cmake_minimum_required(VERSION 3.9.0)

set (CMAKE_VERBOSE_MAKEFILE ON)
set (PACKAGE_NAME "VisionCameraTflite")
set (CMAKE_CXX_STANDARD 17)

find_package(ReactAndroid REQUIRED CONFIG)
find_package(fbjni REQUIRED CONFIG)

find_library(
  TFLITE
  tensorflowlite_jni
  PATHS "./src/main/cpp/lib/litert/jni/${ANDROID_ABI}"
  NO_DEFAULT_PATH
  NO_CMAKE_FIND_ROOT_PATH
)

find_library(
  TFLITE_GPU
  tensorflowlite_gpu_jni
  PATHS "./src/main/cpp/lib/litert/jni/${ANDROID_ABI}"
  NO_DEFAULT_PATH
  NO_CMAKE_FIND_ROOT_PATH
)

string(APPEND CMAKE_CXX_FLAGS " -DANDROID")

add_library(
  ${PACKAGE_NAME}
  SHARED
  ../cpp/jsi/Promise.cpp
  ../cpp/jsi/TypedArray.cpp
  ../cpp/TensorflowPlugin.cpp
  ../cpp/TensorHelpers.cpp
  src/main/cpp/Tflite.cpp
)

target_link_options(${PACKAGE_NAME}
    PRIVATE
    "-Wl,-z,max-page-size=16384"
    "-Wl,-z,common-page-size=16384"
)

# Specifies a path to native header files.
target_include_directories(
  ${PACKAGE_NAME}
  PRIVATE
  "../cpp"
  "src/main/cpp"
  "src/main/cpp/lib/litert/headers"
  "src/main/cpp/lib/litert/headers/tflite"
  "${NODE_MODULES_DIR}/react-native/ReactCommon"
  "${NODE_MODULES_DIR}/react-native/ReactCommon/callinvoker"
  "${NODE_MODULES_DIR}/react-native/ReactAndroid/src/main/jni/react/turbomodule" # <-- CallInvokerHolder JNI wrapper
)

set_target_properties(${PACKAGE_NAME} PROPERTIES LINKER_LANGUAGE CXX)

target_link_libraries(
  ${PACKAGE_NAME}
  android                         # <-- log
  ReactAndroid::jsi               # <-- jsi.h
  fbjni::fbjni                    # <-- fbjni.h
  ${TFLITE}
  ${TFLITE_GPU}
)

if(ReactAndroid_VERSION_MINOR GREATER_EQUAL 76)
  target_link_libraries(
    ${PACKAGE_NAME}
    ReactAndroid::reactnative     # <-- RN merged so
  )
else()
  target_link_libraries(
    ${PACKAGE_NAME}
    ReactAndroid::reactnativejni  # <-- CallInvokerImpl
  )
endif()