cmake_minimum_required(VERSION 3.15)

project(cnn)

set(CMAKE_EXPORT_COMPILE_COMMANDS ON)

set(CMAKE_CXX_STANDARD 11)
set(CMAKE_BUILD_TYPE "Debug")
set(CMAKE_C_FLAGS_DEBUG "${CMAKE_C_FLAGS_DEBUG} -fno-omit-frame-pointer -fsanitize=address")                                             
set(CMAKE_CXX_FLAGS_DEBUG "${CMAKE_CXX_FLAGS_DEBUG} -fno-omit-frame-pointer -fsanitize=address")
set(CMAKE_LINKER_FLAGS_DEBUG "${CMAKE_LINKER_FLAGS_DEBUG} -fno-omit-frame-pointer -fsanitize=address")

include("cmake/overlook.cmake")

set(ecnn_api_files
  ecnn/api/ecnn/refcount.hpp
  ecnn/api/ecnn/tensor.hpp
  ecnn/api/ecnn/layer.hpp
  ecnn/api/ecnn/padding.h
)

set(ecnn_impl_files
  ecnn/impl/tensor.cpp
  ecnn/impl/layer.cpp
  ecnn/impl/padding.cpp
  ecnn/impl/tensor_debug.cpp

  ecnn/impl/convolution.h
  ecnn/impl/convolution.cpp

  ecnn/impl/max_pooling.h
  ecnn/impl/max_pooling.cpp
  ecnn/impl/avg_pooling.h
  ecnn/impl/avg_pooling.cpp

  ecnn/impl/innerproduct.h
  ecnn/impl/innerproduct.cpp

  ecnn/impl/activations.h
  ecnn/impl/activations.cpp
)

add_library(ecnn STATIC
  ${ecnn_api_files}
  ${ecnn_impl_files}
)

target_include_directories(ecnn PUBLIC ${CMAKE_SOURCE_DIR}/ecnn/api)

add_executable(testbed
  #ecnn/test/main.cpp
  
  #ecnn/test/lenet_to_ncnn.cpp
  #ecnn/test/ncnn_infer.cpp

  ecnn/test/ecnn_new.cpp
)
target_include_directories(testbed PRIVATE
  ${CMAKE_SOURCE_DIR}/ecnn/impl/
)

find_package(OpenCV REQUIRED)

set(ncnn_DIR "/home/zz/work/ncnn/build/linux/install/lib/cmake/ncnn" CACHE PATH "")
find_package(ncnn REQUIRED)


target_link_libraries(testbed
  ecnn
  ${OpenCV_LIBS}
  ncnn
)

add_custom_command(TARGET testbed
  POST_BUILD

  COMMAND ${CMAKE_COMMAND} -E copy_if_different
  ${CMAKE_SOURCE_DIR}/assets/model.dat
  ${CMAKE_BINARY_DIR}/

  COMMAND ${CMAKE_COMMAND} -E copy_if_different
  ${CMAKE_SOURCE_DIR}/assets/mobilenet.param
  ${CMAKE_BINARY_DIR}/

  COMMAND ${CMAKE_COMMAND} -E copy_if_different
  ${CMAKE_SOURCE_DIR}/assets/2_1002.bmp
  ${CMAKE_BINARY_DIR}/

  COMMAND ${CMAKE_COMMAND} -E copy_if_different
  ${CMAKE_SOURCE_DIR}/assets/7.bmp
  ${CMAKE_BINARY_DIR}/
)