#===============================================================================
# Copyright 2020-2022 Intel Corporation.
#
# This software and the related documents are Intel copyrighted  materials,  and
# your use of  them is  governed by the  express license  under which  they were
# provided to you (License).  Unless the License provides otherwise, you may not
# use, modify, copy, publish, distribute,  disclose or transmit this software or
# the related documents without Intel's prior written permission.
#
# This software and the related documents  are provided as  is,  with no express
# or implied  warranties,  other  than those  that are  expressly stated  in the
# License.
#===============================================================================

cmake_minimum_required(VERSION 3.13)
enable_testing()

# Add cmake scripts and modules to CMake search path
list(APPEND CMAKE_MODULE_PATH "${CMAKE_CURRENT_SOURCE_DIR}/../cmake")
list(APPEND CMAKE_MODULE_PATH "${CMAKE_CURRENT_SOURCE_DIR}/cmake")

# Define language and compiler
set(TEST_LANG CXX)
set(TEST_EXT cpp)
include(setup_examples)

# Device Library can be linked only statically

if(NOT MKL_LINK STREQUAL "static")
message(NOTICE "MKL_LINK is set to static unconditionally due to toolchain requirements")
set(MKL_LINK static)
endif()

project(MKL_Examples LANGUAGES ${TEST_LANG})
find_package(MKL CONFIG REQUIRED)

# Generate domainList and ${domain}_funcList
include(generate_examples_list)

if(WIN32)
  set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wno-unused-function -w")
  set(CMAKE_CXX_CREATE_STATIC_LIBRARY "lib <OBJECTS> /out:<TARGET>")
  set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} /MANIFEST:NO")
  if(CMAKE_VERSION VERSION_LESS "3.25.2")
    set(CMAKE_CXX_COMPILE_OBJECT "<CMAKE_CXX_COMPILER> <DEFINES> <INCLUDES> <FLAGS> /Fo<OBJECT> -c <SOURCE>")
    set(CMAKE_CXX_LINK_EXECUTABLE "<CMAKE_CXX_COMPILER> <OBJECTS> -o <TARGET> <LINK_LIBRARIES>")
  endif()
endif()

if(FAIL_ON_MISSING_DEVICES)
  list(APPEND DEVICE_OPT -DFAIL_ON_MISSING_DEVICES)
  message(STATUS "FAIL_ON_MISSING_DEVICES: `${FAIL_ON_MISSING_DEVICES}`")
endif()

foreach(device IN LISTS TARGET_DEVICES)
  list(APPEND DEVICE_OPT -DSYCL_DEVICES_${device})
endforeach()

# Define target for each function from each domain
if(domainList)
foreach(domain IN LISTS domainList)
  set(TEST_INCLUDE "")
  set(TEST_COPT "")
  set(TEST_LOPT "")

  list(APPEND TEST_INCLUDE "${PROJECT_SOURCE_DIR}/common" "${PROJECT_SOURCE_DIR}/${domain}")
  list(APPEND TEST_COPT -fsycl-unnamed-lambda ${DEVICE_OPT})
  list(APPEND TEST_LOPT -fsycl-device-code-split=per_kernel)
  if(CXX_COMPILER_NAME STREQUAL "icpx" OR CXX_COMPILER_NAME STREQUAL "icx.exe")
    list(APPEND TEST_COPT "-fsycl")
    list(APPEND TEST_LOPT "-fsycl")
  endif()

  # Build target for each example
  message(STATUS "Functions list ${domain}: ${${domain}_funcList}")
  foreach(func IN LISTS ${domain}_funcList)
    string(STRIP ${func} func_name)
    set(func "${domain}-${func_name}")

    file(GLOB_RECURSE ${domain}_${func}_SRC ${PROJECT_SOURCE_DIR}/${domain}/*/${func_name}.${TEST_EXT})
    if(NOT ${domain}_${func}_SRC)
      message(FATAL_ERROR "${domain} source file ${func_name}.${TEST_EXT} was not found")
    endif()

    add_executable(${func} ${${domain}_${func}_SRC})
    target_include_directories(${func} PUBLIC ${TEST_INCLUDE} $<TARGET_PROPERTY:MKL::MKL_SYCL,INTERFACE_INCLUDE_DIRECTORIES>)

    target_compile_options(${func} PUBLIC ${TEST_COPT} $<TARGET_PROPERTY:MKL::MKL_SYCL,INTERFACE_COMPILE_OPTIONS>)

    if(func_name STREQUAL "gaussian_vml_based" OR func_name MATCHES "^vm")
      if(MKL_LINK STREQUAL "static")
        target_link_libraries(${func} PUBLIC ${TEST_LOPT} $<LINK_ONLY:MKL::MKL_SYCL>)
      else()
        message(FATAL_ERROR "${func_name} must be statically linked. Set MKL_LINK to static")
      endif()
    else()
      target_link_libraries(${func} PUBLIC ${TEST_LOPT})
    endif()
    # Register example as ctest
    add_test(NAME ${func} COMMAND ${func})

  endforeach() #${domain}_funcList
endforeach() #domainList
endif() #not empty domainList
