CMAKE_MINIMUM_REQUIRED(VERSION 3.12)

PROJECT(HITSZ-RM-CV)
SET(CMAKE_CXX_STANDARD 14)
SET(CMAKE_BUILD_TYPE RELEASE)
SET(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -O2")
SET(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DPATH=\"\\\"${PROJECT_SOURCE_DIR}\\\"\"")
SET(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -D${CMAKE_SYSTEM_NAME}")
SET(BIN_NAME "run")


FIND_FILE(CONFIG_FOUND "config.h" "others/include/config")
if (CONFIG_FOUND)
    MESSAGE("Found config.h")
    SET(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DWITH_CONFIG")
endif()


FIND_PROGRAM(CCACHE_FOUND ccache)
IF(CCACHE_FOUND)
    SET_PROPERTY(GLOBAL PROPERTY RULE_LAUNCH_COMPILE ccache)
    SET_PROPERTY(GLOBAL PROPERTY RULE_LAUNCH_LINK ccache)
    MESSAGE("< Use ccache for compiler >")
ENDIF()

FIND_PACKAGE(Eigen3 REQUIRED)
FIND_PACKAGE(OpenCV 3 REQUIRED)
FIND_PACKAGE(Threads)

LINK_DIRECTORIES(${PROJECT_SOURCE_DIR}/others)
######################## added for GX start ########################
link_directories(${PROJECT_SOURCE_DIR}/others/genicam/bin/Linux64_x64)
link_directories(${PROJECT_SOURCE_DIR}/others/genicam/bin/Linux64_x64/GenApi/Generic)
link_directories(${PROJECT_SOURCE_DIR}/others/lib)
######################## added for GX end   ########################



INCLUDE_DIRECTORIES(${EIGEN3_INCLUDE_DIR})
INCLUDE_DIRECTORIES(${PROJECT_SOURCE_DIR}/energy/include)
INCLUDE_DIRECTORIES(${PROJECT_SOURCE_DIR}/armor/include)
INCLUDE_DIRECTORIES(${PROJECT_SOURCE_DIR}/others/include)
######################## added for GX start ########################
include_directories(${PROJECT_SOURCE_DIR}/others/include/camera)
include_directories(${PROJECT_SOURCE_DIR}/others/genicam/library/CPP/include)
include_directories(${OpenCV_INCLUDE_DIR})
######################## added for GX end   ########################



FILE(GLOB_RECURSE sourcefiles "others/src/*.cpp" "energy/src/*cpp" "armor/src/*.cpp")
ADD_EXECUTABLE(${BIN_NAME} main.cpp ${sourcefiles} armor/include/armor_finder/classifier/classifier.cpp)



######################## added for GX start ########################
target_link_libraries(${BIN_NAME} libgxiapi.so)
target_link_libraries(${BIN_NAME} libdhgentl.so)
target_link_libraries(${BIN_NAME} libdx_image_proc.so)
target_link_libraries(${BIN_NAME} libGCBase_gcc40_v2_3.so)
target_link_libraries(${BIN_NAME} libGenApi_gcc40_v2_3.so)
target_link_libraries(${BIN_NAME} liblog4cpp_gcc40_v2_3.so)
target_link_libraries(${BIN_NAME} libLog_gcc40_v2_3.so)
target_link_libraries(${BIN_NAME} libMathParser_gcc40_v2_3.so)
######################## added for GX end   ########################
TARGET_LINK_LIBRARIES(${BIN_NAME} ${CMAKE_THREAD_LIBS_INIT})
TARGET_LINK_LIBRARIES(${BIN_NAME} ${OpenCV_LIBS})
IF (CMAKE_SYSTEM_NAME MATCHES "Linux")
    MESSAGE(STATUS "current platform: Linux ")
    TARGET_LINK_LIBRARIES(${BIN_NAME} "${PROJECT_SOURCE_DIR}/others/libMVSDK.so")
ELSEIF (CMAKE_SYSTEM_NAME MATCHES "Windows")
    MESSAGE(STATUS "current platform: Windows")
    TARGET_LINK_LIBRARIES(${BIN_NAME} "${PROJECT_SOURCE_DIR}/others/MVCAMSDK_X64.lib")
ELSEIF (CMAKE_SYSTEM_NAME MATCHES "Darwin")
    MESSAGE(STATUS "current platform: Mac")
    TARGET_LINK_LIBRARIES(${BIN_NAME} "${PROJECT_SOURCE_DIR}/others/libmvsdk.dylib")
ELSE ()
    MESSAGE(STATUS "Unsupport platform: ${CMAKE_SYSTEM_NAME}")
ENDIF()

ADD_CUSTOM_TARGET(create-startup COMMAND "${PROJECT_SOURCE_DIR}/tools/create-startup.sh" "${PROJECT_SOURCE_DIR}" "${CMAKE_BINARY_DIR}")
ADD_CUSTOM_TARGET(train-cnn COMMAND "gnome-terminal" "--" "bash" "-c" "${PROJECT_SOURCE_DIR}/tools/TrainCNN/backward.py")