# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      https://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

cmake_minimum_required(VERSION 3.18)
project(tfl_interp CXX)

# check requirements
find_package(Patch)
if(NOT Patch_FOUND)
  message(FATAL_ERROR "Patch not found patch command")
endif()

# add Tensorflow sources and modify
message("** Download Tensorflow lite etc.")
include(FetchContent)
FetchContent_Declare(tensorflow
  GIT_REPOSITORY https://github.com/tensorflow/tensorflow.git
  GIT_TAG        v2.7.0
  SOURCE_SUBDIR tensorflow/lite
  SOURCE_DIR "${CMAKE_BINARY_DIR}/tensorflow_src"
)
FetchContent_MakeAvailable(tensorflow)

IF((${CMAKE_HOST_SYSTEM_NAME} MATCHES "Windows") AND (NOT EXISTS ${CMAKE_BINARY_DIR}/patched))
  execute_process(
    COMMAND patch --verbos -p1 -i ${CMAKE_SOURCE_DIR}/msys2.patch
    WORKING_DIRECTORY ${CMAKE_BINARY_DIR}
  )
  file(TOUCH ${CMAKE_BINARY_DIR}/patched)
endif()

# add Nlohmann JSON
FetchContent_Declare(nlohmann_json
  GIT_REPOSITORY https://github.com/nlohmann/json.git
  SOURCE_DIR "${CMAKE_BINARY_DIR}/nlohmann_json"
)
FetchContent_MakeAvailable(nlohmann_json)


# main
add_executable(tfl_interp
  src/tfl_interp.cc
  src/io_port.cc
  src/tfl_nonmaxsuppression.cc
)
target_link_libraries(tfl_interp
  tensorflow-lite
  nlohmann_json
)

install(TARGETS tfl_interp
  RUNTIME
  DESTINATION ${CMAKE_SOURCE_DIR}/priv
)