CMakeLists.txt

cmake_minimum_required(VERSION 3.10 FATAL_ERROR)
project(PokerProject)
set(CMAKE_CXX_STANDARD 17)
set(CMAKE_CXX_STANDARD_REQUIRED ON)
set(CMAKE_CXX_EXTENSIONS OFF)

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

# Option 1: Using add_compile_options
if(CMAKE_CXX_COMPILER_ID MATCHES "GNU|Clang")
    add_compile_options(-w)  # Disables all warnings
elseif(CMAKE_CXX_COMPILER_ID MATCHES "MSVC")
    add_compile_options(/w)  # Disables all warnings for MSVC
endif()

# near top, after project declaration:
if(APPLE)
   find_package(BLAS REQUIRED)
endif()

include_directories(${CMAKE_CURRENT_SOURCE_DIR}/OMPEval)
link_directories(${CMAKE_CURRENT_SOURCE_DIR}/OMPEval/lib)
find_library(OMPEVAL_LIB ompeval PATHS ${CMAKE_CURRENT_SOURCE_DIR}/OMPEval/lib NO_DEFAULT_PATH)

if(NOT OMPEVAL_LIB)
  message(FATAL_ERROR "OMPEval library not found in ${CMAKE_CURRENT_SOURCE_DIR}/OMPEval/lib")
endif()

if(UNIX AND NOT APPLE)
  find_package(OpenMP REQUIRED)
  set(OPENMP_LIB OpenMP::OpenMP_CXX)
elseif(APPLE)
  set(OPENMP_INCLUDE_DIR /opt/homebrew/opt/libomp/include)
  set(OPENMP_LIB_DIR /opt/homebrew/opt/libomp/lib)
  include_directories(${OPENMP_INCLUDE_DIR})
  link_directories(${OPENMP_LIB_DIR})
  if(NOT EXISTS "${OPENMP_LIB_DIR}/libomp.dylib")
    message(FATAL_ERROR "libomp.dylib not found in ${OPENMP_LIB_DIR}")
  endif()
  set(OPENMP_LIB "${OPENMP_LIB_DIR}/libomp.dylib")
else()
  message(FATAL_ERROR "Unsupported operating system")
endif()

file(GLOB_RECURSE NON_MODEL_SRCS 
    "${CMAKE_CURRENT_SOURCE_DIR}/src/*.cpp" 
    "${CMAKE_CURRENT_SOURCE_DIR}/src/*.cxx"
)
list(FILTER NON_MODEL_SRCS EXCLUDE REGEX "${CMAKE_CURRENT_SOURCE_DIR}/src/tests/.*")
list(APPEND MODEL_SRCS "${CMAKE_CURRENT_SOURCE_DIR}/src/model/model.h")
set(ALL_SRCS ${NON_MODEL_SRCS} ${MODEL_SRCS})
add_executable(main ${ALL_SRCS})

# mlx config
if(APPLE)
    set(MLX_DIR ${CMAKE_CURRENT_SOURCE_DIR}/mlx)
    include_directories(${MLX_DIR})
    target_link_libraries(main PRIVATE 
        ${MLX_DIR}/build/libmlx.a
        ${OMPEVAL_LIB} 
        "${TORCH_LIBRARIES}" 
        ${OPENMP_LIB}
        "-framework Foundation"
        "-framework Metal"
        "-framework MetalKit"
        "-framework Accelerate" 
        "-framework CoreFoundation"
        ${BLAS_LIBRARIES}
    )
else()
    target_link_libraries(main PRIVATE
        ${OMPEVAL_LIB} 
        "${TORCH_LIBRARIES}" 
        ${OPENMP_LIB}
    )
endif()

target_compile_options(main PRIVATE "${TORCH_CXX_FLAGS}")

if(APPLE)
  set_target_properties(main PROPERTIES
      BUILD_RPATH "${OPENMP_LIB_DIR};${Torch_DIR}/../lib"
      INSTALL_RPATH "${OPENMP_LIB_DIR};${Torch_DIR}/../lib"
      CXX_VISIBILITY_PRESET hidden
      CXX_STANDARD_REQUIRED ON
      CXX_EXTENSIONS OFF
  )
endif()

if(CMAKE_CXX_COMPILER_ID MATCHES "GNU|Clang")
    set(CMAKE_CXX_FLAGS_RELEASE "-O3 -march=native -DNDEBUG")
    set(CMAKE_CXX_FLAGS_RELWITHDEBINFO "-O3 -march=native -g -DNDEBUG") 
elseif(CMAKE_CXX_COMPILER_ID MATCHES "MSVC")
    set(CMAKE_CXX_FLAGS_RELEASE "/O2 /Oi /Ot /Gy /DNDEBUG")
    set(CMAKE_CXX_FLAGS_RELWITHDEBINFO "/O2 /Oi /Ot /Gy /DEBUG /DNDEBUG")
endif()

set(CMAKE_RUNTIME_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/bin)