# The file split_test_helper.h was generated at first run,
# it is now included in test/
# SPDX-FileCopyrightText: The Eigen Authors
# SPDX-License-Identifier: MPL-2.0
if(EXISTS ${CMAKE_CURRENT_BINARY_DIR}/split_test_helper.h)
  file(REMOVE ${CMAKE_CURRENT_BINARY_DIR}/split_test_helper.h)
endif()

set_property(GLOBAL PROPERTY EIGEN_CURRENT_SUBPROJECT "Contrib")
add_custom_target(BuildContrib)

include_directories(../../test ../../contrib
                    ${CMAKE_CURRENT_BINARY_DIR}/../../test)

find_package (Threads)

find_package(GoogleHash)
if(GoogleHash_FOUND)
  add_definitions("-DEIGEN_GOOGLEHASH_SUPPORT")
  include_directories(${GOOGLEHASH_INCLUDES})
  ei_add_property(EIGEN_TESTED_BACKENDS  "GoogleHash, ")
else()
  ei_add_property(EIGEN_MISSING_BACKENDS  "GoogleHash, ")
endif()


find_package(Adolc)
if(Adolc_FOUND)
  include_directories(${ADOLC_INCLUDES})
  ei_add_property(EIGEN_TESTED_BACKENDS "Adolc, ")
  ei_add_test(forward_adolc "" ${ADOLC_LIBRARIES})
else()
  ei_add_property(EIGEN_MISSING_BACKENDS "Adolc, ")
endif()

# this test seems to never have been successful on x87, so is considered to contain a FP-related bug.
# see thread: "non-linear optimization test summary"
ei_add_test(NonLinearOptimization)

ei_add_test(NumericalDiff)
ei_add_test(stencil)
ei_add_test(structured_dpr1)
ei_add_test(autodiff_scalar)
ei_add_test(autodiff)

ei_add_test(BVH)

ei_add_test(matrix_exponential)
ei_add_test(matrix_function)
ei_add_test(matrix_power)
ei_add_test(matrix_square_root)
ei_add_test(alignedvector3)

ei_add_test(FFT)
ei_add_test(structured_bccb)

ei_add_test(EulerAngles)

find_package(MPREAL)
if(MPREAL_FOUND)
  ei_add_property(EIGEN_TESTED_BACKENDS "MPFR C++, ")
  include_directories(${MPREAL_INCLUDES})
  ei_add_test(mpreal_support "" "${MPREAL_LIBRARIES}" )
else()
  ei_add_property(EIGEN_MISSING_BACKENDS "MPFR C++, ")
endif()

ei_add_test(NNLS)
ei_add_test(structured_cauchy)

ei_add_test(sparse_extra   "" "")

find_package(FFTW)
if(FFTW_FOUND)
  ei_add_property(EIGEN_TESTED_BACKENDS "fftw, ")
  include_directories( ${FFTW_INCLUDES} )
  # The concurrent-transform subtest needs std::thread.
  if(FFTWL_LIB)
    ei_add_test(FFTW  "-DEIGEN_HAS_FFTWL -pthread" "${FFTW_LIBRARIES};${CMAKE_THREAD_LIBS_INIT}" )
  else()
    ei_add_test(FFTW  "-pthread" "${FFTW_LIBRARIES};${CMAKE_THREAD_LIBS_INIT}" )
  endif()
else()
  ei_add_property(EIGEN_MISSING_BACKENDS "fftw, ")
endif()

find_path(POCKETFFT  pocketfft_hdronly.h)
if(POCKETFFT)
  ei_add_property(EIGEN_TESTED_BACKENDS "pocketfft, ")
  include_directories( ${POCKETFFT} )
  ei_add_test(pocketfft "-pthread" "${CMAKE_THREAD_LIBS_INIT}" )
else()
  ei_add_property(EIGEN_MISSING_BACKENDS "pocketfft, ")
endif()

if( NOT DUCC_ROOT AND ENV{DUCC_ROOT} )
  set( DUCC_ROOT $ENV{DUCC_ROOT} )
endif()
find_path(DUCCFFT
  NAMES "src/ducc0/fft/fft.h"
  PATHS ${DUCC_ROOT})
message(INFO " ${DUCC_ROOT} ${DUCCFFT}")
if(DUCCFFT)
  ei_add_property(EIGEN_TESTED_BACKENDS "duccfft, ")
  include_directories( "${DUCCFFT}/src" )
  add_library(ducc_lib "${DUCCFFT}/src/ducc0/infra/string_utils.cc" "${DUCCFFT}/src/ducc0/infra/threading.cc")
  target_compile_definitions(ducc_lib PUBLIC "DUCC0_NO_THREADING=1")
  ei_add_test(duccfft "-DEIGEN_DUCCFFT_DEFAULT -DDUCC0_NO_THREADING=1" "ducc_lib" )
  set_target_properties(ducc_lib duccfft PROPERTIES CXX_STANDARD 17)
else()
  ei_add_property(EIGEN_MISSING_BACKENDS "duccfft, ")
endif()


option(EIGEN_TEST_OPENGL "Enable OpenGL support in unit tests" OFF)
if(EIGEN_TEST_OPENGL)
  find_package(OpenGL)
  find_package(GLUT)
  find_package(GLEW)
  if(OPENGL_FOUND AND GLUT_FOUND AND GLEW_FOUND)
    include_directories(${OPENGL_INCLUDE_DIR} ${GLUT_INCLUDE_DIR} ${GLEW_INCLUDE_DIRS})
    ei_add_property(EIGEN_TESTED_BACKENDS "OpenGL, ")
    set(EIGEN_GL_LIB ${GLUT_LIBRARIES} ${GLEW_LIBRARIES} ${OPENGL_LIBRARIES})
    ei_add_test(openglsupport  "" "${EIGEN_GL_LIB}" )
  else()
    ei_add_property(EIGEN_MISSING_BACKENDS "OpenGL, ")
  endif()
else()
    ei_add_property(EIGEN_MISSING_BACKENDS "OpenGL, ")
endif()

ei_add_test(structured_vandermonde)
ei_add_test(structured_vandermonde_int_index)
ei_add_test(polynomialsolver)
ei_add_test(polynomialutils)
ei_add_test(splines)
ei_add_test(structured_dplr)
ei_add_test(levenberg_marquardt)
ei_add_test(kronecker_product)
ei_add_test(structured_matrices)
ei_add_test(structured_kronecker)
ei_add_test(structured_kronecker_sum)
ei_add_test(structured_kronecker_sparse_view)
ei_add_test(bessel_functions)
ei_add_test(special_functions)
ei_add_test(special_packetmath "-DEIGEN_FAST_MATH=1")

if(EIGEN_TEST_SYCL)
  set(EIGEN_SYCL ON)
  include(SyclConfigureTesting)

  ei_add_test(tensor_sycl)
  ei_add_test(tensor_image_op_sycl)
  ei_add_test(tensor_math_sycl)
  ei_add_test(tensor_forced_eval_sycl)
  ei_add_test(tensor_broadcast_sycl)
  ei_add_test(tensor_device_sycl)
  ei_add_test(tensor_reduction_sycl)
  ei_add_test(tensor_morphing_sycl)
  ei_add_test(tensor_shuffling_sycl)
  ei_add_test(tensor_padding_sycl)
  ei_add_test(tensor_builtins_sycl)
  ei_add_test(tensor_contract_sycl)
  ei_add_test(tensor_concatenation_sycl)
  ei_add_test(tensor_reverse_sycl)
  ei_add_test(tensor_convolution_sycl)
  ei_add_test(tensor_striding_sycl)
  ei_add_test(tensor_chipping_sycl)
  ei_add_test(tensor_layout_swap_sycl)
  ei_add_test(tensor_inflation_sycl)
  ei_add_test(tensor_random_sycl)
  ei_add_test(tensor_generator_sycl)
  ei_add_test(tensor_patch_sycl)
  ei_add_test(tensor_image_patch_sycl)
  ei_add_test(tensor_volume_patch_sycl)
  ei_add_test(tensor_argmax_sycl)
  ei_add_test(tensor_custom_op_sycl)
  ei_add_test(tensor_scan_sycl)
  ei_add_test(tensor_of_float16_sycl)
  set(EIGEN_SYCL OFF)
endif()

ei_add_test(tensor_argmax)
ei_add_test(tensor_assign)
ei_add_test(tensor_block_access)
ei_add_test(tensor_block_eval)
ei_add_test(tensor_block_view "-pthread" "${CMAKE_THREAD_LIBS_INIT}")
ei_add_test(tensor_block_io)
ei_add_test(tensor_broadcasting)
ei_add_test(tensor_casts)
ei_add_test(tensor_chipping)
ei_add_test(tensor_comparisons)
ei_add_test(tensor_concatenation)
ei_add_test(tensor_const)
ei_add_test(tensor_contraction)
ei_add_test(tensor_cost_model)
ei_add_test(tensor_convolution)
ei_add_test(tensor_custom_index)
ei_add_test(tensor_custom_op)
ei_add_test(tensor_dimension)
ei_add_test(tensor_empty)
ei_add_test(tensor_executor "-pthread" "${CMAKE_THREAD_LIBS_INIT}")
ei_add_test(tensor_expr)
ei_add_test(tensor_fft)
ei_add_test(tensor_fixed_size)
ei_add_test(tensor_forced_eval)
ei_add_test(tensor_generator)
ei_add_test(tensor_ifft)
ei_add_test(tensor_image_patch)
ei_add_test(tensor_index_list)
ei_add_test(tensor_inflation "-pthread" "${CMAKE_THREAD_LIBS_INIT}")
ei_add_test(tensor_intdiv)
ei_add_test(tensor_io)
ei_add_test(tensor_layout_swap)
ei_add_test(tensor_lvalue)
ei_add_test(tensor_map)
ei_add_test(tensor_math)
ei_add_test(tensor_mixed_indices)
ei_add_test(tensor_morphing)
ei_add_test(tensor_move)
ei_add_test(tensor_notification "-pthread" "${CMAKE_THREAD_LIBS_INIT}")
ei_add_test(tensor_of_complex)
ei_add_test(tensor_of_const_values)
ei_add_test(tensor_of_strings)
ei_add_test(tensor_padding)
ei_add_test(tensor_patch)
ei_add_test(tensor_random)
ei_add_test(tensor_reinclude "-pthread" "${CMAKE_THREAD_LIBS_INIT}")
ei_add_test(tensor_reverse)
ei_add_test(tensor_reduction)
ei_add_test(tensor_ref)
ei_add_test(tensor_roll)
ei_add_test(tensor_roundings)
ei_add_test(tensor_scan)
ei_add_test(tensor_shuffling)
ei_add_test(tensor_simple)
ei_add_test(tensor_special_members)
ei_add_test(tensor_striding)
ei_add_test(tensor_sugar)
ei_add_test(tensor_thread_local "-pthread" "${CMAKE_THREAD_LIBS_INIT}")
ei_add_test(tensor_thread_pool "-pthread" "${CMAKE_THREAD_LIBS_INIT}")
ei_add_test(tensor_trace)
ei_add_test(tensor_volume_patch)
ei_add_test(tensor_symmetry)
if("${CMAKE_SIZEOF_VOID_P}" EQUAL "8" AND NOT CMAKE_CXX_COMPILER_ID STREQUAL "MSVC")
  # This test requires __uint128_t which is only available on 64bit systems
  ei_add_test(tensor_uint128)
endif()

# CUDA and HIP tests. The options, the toolkit and the language are set up once in the top-level CMakeLists
# through cmake/EigenGpuTesting.cmake.
if(EIGEN_GPU_TEST_MODE MATCHES "^(cuda|hip)")
  set(EIGEN_ADD_TEST_FILENAME_EXTENSION "cu")

  if(EIGEN_GPU_TEST_MODE MATCHES "^cuda")
    # complex is not yet supported by HIP.
    ei_add_test(tensor_complex_gpu)
    ei_add_test(tensor_complex_cwise_ops_gpu)
  endif()
  ei_add_test(tensor_reduction_gpu)
  ei_add_test(tensor_argmax_gpu)
  ei_add_test(tensor_cast_float16_gpu)
  ei_add_test(tensor_scan_gpu)
  ei_add_test(tensor_device)
  ei_add_test(tensor_gpu_device_attributes)
  ei_add_test(tensor_gpu)
  ei_add_test(tensor_gpu_runtime_check)
  ei_add_test(tensor_gpu_runtime_check_ndebug)
  ei_add_test(tensor_contract_gpu)
  ei_add_test(tensor_of_float16_gpu)
  ei_add_test(tensor_of_bfloat16_gpu)
  ei_add_test(tensor_random_gpu)

  unset(EIGEN_ADD_TEST_FILENAME_EXTENSION)

  # The GPU module's own tests (cuSOLVER, cuDSS) are CUDA-only.
  if(EIGEN_GPU_TEST_MODE MATCHES "^cuda")
    add_subdirectory(GPU)
  endif()
endif()
