# Copyright (c) Huawei Technologies Co., Ltd. 2019. All rights reserved.

# CMake lowest version requirement
cmake_minimum_required(VERSION 3.5.1)

# project information
project(ACL_RESNET50)


set(MODULE_NAME fcos3d)

# # Compile options
# add_compile_options(-std=c++11)
# Compile options
add_compile_options(-std=c++17)

option(ENABLE_wrappers "Enable the wrappers" OFF)

# set(CMAKE_CXX_FLAGS_DEBUG "-fPIC -O0 -g -Wall")
# set(CMAKE_CXX_FLAGS_RELEASE "-fPIC -O2 -Wall")
set(CMAKE_BUILD_TYPE "Release")
set(CMAKE_CXX_FLAGS_RELEASE "${CMAKE_CXX_FLAGS_RELEASE} -O3")
#set(yaml-cpp_DIR "/usr/lib/x86_64-linux-gnu/cmake/yaml-cpp/")
# Find OpenMP package
find_package(OpenMP REQUIRED)

#find_package(yaml-cpp REQUIRED)
#find_package(yaml-cpp)
#find_package(  yaml-cpp REQUIRED )
#find_package(PkgConfig REQUIRED)
#pkg_check_modules(YAML_CPP REQUIRED yaml-cpp)
#find_package(yaml-cpp REQUIRED)
# Header path
include_directories(
        "/usr/include/tbb/"
        "/usr/local/mdc_sdk_llvm/dp_gea/mdc_cross_compiler/sysroot/usr/include"
        "/usr/local/mdc_sdk_llvm/dp_gea/mdc_cross_compiler/sysroot/usr/local/Ascend/runtime/include/"
        inc/
        includeMOT/
        includeSlam_fcos/
        
)

if(ENABLE_wrappers)
  include_directories(wrappers/*.h)
endif()

# add host lib path
link_directories(
        "/usr/local/mdc_sdk_llvm/dp_gea/mdc_cross_compiler/sysroot/usr/lib64"
        "/usr/local/mdc_sdk_llvm/dp_gea/mdc_cross_compiler/sysroot/usr/local/Ascend/runtime/lib64/"
        "/app/share/yaml-cpp/build"
)

set(Eigen "./eigen-3.4.0")

# include_directories(${Eigen})
include_directories(/home/kasra/mds_workspace/fcos3d/modules/fcos3d_swc/eigen-3.4.0)
include_directories(${Eigen}/unsupported)
include_directories(./include)
#include_directories(${YAML_CPP_INCLUDEDIR})
file(GLOB SRC_FILES ${CMAKE_CURRENT_SOURCE_DIR}/source/*.cc)

if(ENABLE_wrappers)
   file(GLOB SRC_FILES ${CMAKE_CURRENT_SOURCE_DIR}/source/*.cc
                       ${CMAKE_CURRENT_SOURCE_DIR}/wrappers/*.cpp)
endif()

add_executable(${MODULE_NAME}
        ${SRC_FILES}
        src/utils.cpp
        src/model_process.cpp
        src/main.cpp
        inc/fcos3d_pipeline.h
        src/fcos3d_pipeline.cpp
        inc/fcos_parameters.h
        src/postprocess.cpp
        inc/postprocess.h
        src/preprocess.cpp
        inc/preprocess.h
        inc/utils_output.h)

target_include_directories(${MODULE_NAME}
        PRIVATE
        ${CMAKE_SOURCE_DIR}/includeMOT
        ${CMAKE_SOURCE_DIR}/generated/includes
        /usr/local/mdc_sdk_llvm/dp_gea/mdc_cross_compiler/sysroot/usr/include/opencv4/opencv2
        /usr/local/mdc_sdk_llvm/dp_gea/mdc_cross_compiler/sysroot/usr/include/opencv4
        ${AP_WRAPPERS_INCLUDE}
)
include(FetchContent)

FetchContent_Declare(
        yaml-cpp
        GIT_REPOSITORY https://github.com/jbeder/yaml-cpp.git
        GIT_TAG master
)
FetchContent_GetProperties(yaml-cpp)

if(NOT yaml-cpp_POPULATED)
    message(STATUS "Fetching yaml-cpp...")
    # FetchContent_Populate(yaml-cpp)
    # add_subdirectory(${yaml-cpp_SOURCE_DIR} ${yaml-cpp_BINARY_DIR})
endif()

target_link_libraries(${MODULE_NAME} PUBLIC yaml-cpp)
#target_link_libraries(main PUBLIC yaml-cpp::yaml-cpp)
#target_link_libraries(main  TBB::tbb)

target_link_libraries(${MODULE_NAME}
        PUBLIC
        ascendcl stdc++
        OpenMP::OpenMP_CXX
        -lopencv_core
        -lopencv_highgui
        -lopencv_imgcodecs
        -lopencv_imgproc
        -lopencv_videoio)