cmake_minimum_required(VERSION 3.10)
project(SweepLB LANGUAGES CXX)

if(NOT CMAKE_BUILD_TYPE)
  set(CMAKE_BUILD_TYPE Release)
endif()

set(SIMD_MODE "AVX256" CACHE STRING "Generate either AVX256 or AVX512 instructions")
option(USE_SIMD_CELL_LIST "Generate gather / scatter instructions for cell list processing" OFF)

if(NOT SIMD_MODE)
	set(SIMD_MODE "AVX256")
endif()

set(CXX_FLAGS "-O3 -march=native -mtune=native")

if(${SIMD_MODE} MATCHES "AVX256")
	message("Selecting AVX256")
	set(SIMD_MODE_FLAGS "-mavx2")
elseif(${SIMD_MODE} MATCHES "AVX512")
	message("Selecting AVX512")
	add_compile_definitions(ENABLE_AVX512)
	set(SIMD_MODE_FLAGS "-mavx512f -mavx512dq")
else()
  message(FATAL_ERROR "Invalid SIMD mode, choose either AVX256 or AVX512")
endif()

if(USE_SIMD_CELL_LIST)
  message("Enabling SIMD cell list processing.")
  add_compile_definitions(SIMD_CELL_LIST)
endif()

set(CMAKE_CXX_FLAGS "${CXX_FLAGS} ${SIMD_MODE_FLAGS}")

message("Using flags \"${CMAKE_CXX_FLAGS}\"")

include_directories(
  src/
)

find_package(OpenMP)

set(EXAMPLES ldc box channel)
set(PRECISIONS float double)
set(PATTERNS sss ps)

foreach(example IN LISTS EXAMPLES)
	foreach(precision IN LISTS PRECISIONS)
		foreach(pattern IN LISTS PATTERNS)
			set(generated_name ${example}_${precision}_${pattern})
			string(TOUPPER ${pattern} pattern)
			add_executable(${generated_name} ${example}.cpp)
			target_compile_definitions(${generated_name}
				PRIVATE
				SWEEPLB_PRECISION=${precision}
				SWEEPLB_PATTERN=${pattern})
			target_compile_features(${generated_name}
				PRIVATE cxx_std_20)
			target_link_libraries(
				${generated_name}
				PUBLIC
				  rt)
			if(OpenMP_CXX_FOUND)
				target_link_libraries(
					${generated_name}
					PUBLIC
					  OpenMP::OpenMP_CXX
				)
			endif()
	  endforeach()
  endforeach()
endforeach()