cmake_minimum_required(VERSION 3.16)
project(demo64k LANGUAGES C CXX)

if(APPLE)
    add_definitions(-DGLFW_EXPOSE_NATIVE_COCOA)
elseif(WIN32)
    add_definitions(-DGLFW_EXPOSE_NATIVE_WIN32)
elseif(UNIX)
    if(DEFINED CMAKE_USE_WAYLAND)
        add_definitions(-DGLFW_EXPOSE_NATIVE_WAYLAND)
    else()
        add_definitions(-DGLFW_EXPOSE_NATIVE_X11)
    endif()
endif()

option(DEMO_SIZE_OPT "Enable size optimization flags" OFF)
option(DEMO_STRIP_ALL "Strip all unnecessary code for final build" OFF)

if (DEMO_STRIP_ALL)
    add_definitions(-DSTRIP_ALL)
    set(DEMO_SIZE_OPT ON)
endif()

set(CMAKE_CXX_STANDARD 17)
set(CMAKE_CXX_STANDARD_REQUIRED ON)

# Find wgpu-native (system install)
find_library(WGPU_LIBRARY NAMES wgpu_native libwgpu_native REQUIRED)
find_path(WGPU_INCLUDE_DIR NAMES webgpu.h PATH_SUFFIXES webgpu-headers REQUIRED)

include_directories(
    src
    third_party
    ${WGPU_INCLUDE_DIR}
    third_party/glfw3webgpu
)

find_package(glfw3 REQUIRED)

set(DEMO_LIBS glfw ${WGPU_LIBRARY})

# Platform-specific dependencies
if (APPLE)
    set_source_files_properties(
        src/platform.cc 
        third_party/glfw3webgpu/glfw3webgpu.c
        PROPERTIES COMPILE_FLAGS "-x objective-c++"
    )
    list(APPEND DEMO_LIBS
        "-framework Metal"
        "-framework Foundation"
        "-framework Cocoa"
        "-framework QuartzCore"
    )
else()
    # Assume Linux/other POSIX-like systems might need these
    list(APPEND DEMO_LIBS pthread m dl)
endif()

option(DEMO_BUILD_TOOLS "Build tools" OFF)

# Asset Packing Tool (needed during build if any assets exist)
if(DEMO_BUILD_TOOLS OR NOT EXISTS ${CMAKE_CURRENT_BINARY_DIR}/src/assets_data.cc)
    add_executable(asset_packer
        tools/asset_packer.cc
    )
endif()

# Configure asset generation
set(ASSETS_TXT_PATH ${CMAKE_CURRENT_SOURCE_DIR}/assets/final/assets.txt)
set(GENERATED_ASSETS_H ${CMAKE_CURRENT_BINARY_DIR}/src/assets.h)
set(GENERATED_ASSETS_DATA_CC ${CMAKE_CURRENT_BINARY_DIR}/src/assets_data.cc)

add_custom_command(
    OUTPUT ${GENERATED_ASSETS_H} ${GENERATED_ASSETS_DATA_CC}
    COMMAND ${CMAKE_COMMAND} -E make_directory ${CMAKE_CURRENT_BINARY_DIR}/src
    COMMAND asset_packer ${ASSETS_TXT_PATH} ${GENERATED_ASSETS_H} ${GENERATED_ASSETS_DATA_CC}
    DEPENDS asset_packer ${ASSETS_TXT_PATH}
    COMMENT "Generating assets.h and assets_data.cc"
)

add_custom_target(generate_assets ALL
    DEPENDS ${GENERATED_ASSETS_H} ${GENERATED_ASSETS_DATA_CC}
)

# Main Demo Target
add_executable(demo64k
    src/main.cc
    src/platform.cc
    src/gpu/gpu.cc
    src/audio/audio.cc
    src/audio/fdct.cc
    src/audio/idct.cc
    src/audio/window.cc
    src/audio/synth.cc
    third_party/glfw3webgpu/glfw3webgpu.c
    ${GENERATED_ASSETS_DATA_CC}
)

add_dependencies(demo64k generate_assets)
target_link_libraries(demo64k PRIVATE ${DEMO_LIBS})
target_include_directories(demo64k PRIVATE ${CMAKE_CURRENT_BINARY_DIR}/src)

if (DEMO_SIZE_OPT)
    if (MSVC)
        target_compile_options(demo64k PRIVATE /Os /GS-)
        target_link_options(demo64k PRIVATE /OPT:REF /OPT:ICF /INCREMENTAL:NO)
    elseif (APPLE)
        target_compile_options(demo64k PRIVATE -Os)
        target_link_options(demo64k PRIVATE -Wl,-dead_strip)
    else()
        target_compile_options(demo64k PRIVATE -Os -ffunction-sections -fdata-sections)
        target_link_options(demo64k PRIVATE -Wl,--gc-sections -s)
    endif()
endif()

# Tests
option(DEMO_BUILD_TESTS "Build tests" OFF)
enable_testing()

if(DEMO_BUILD_TESTS)
    add_executable(test_window
        src/tests/test_window.cc
        src/audio/window.cc
    )
    target_include_directories(test_window PRIVATE src)
    add_test(NAME HammingWindowTest COMMAND test_window)

    add_executable(test_synth
        src/tests/test_synth.cc
        src/audio/synth.cc
        src/audio/idct.cc
        src/audio/window.cc
    )
    target_include_directories(test_synth PRIVATE src)
    add_test(NAME SynthEngineTest COMMAND test_synth)

    add_executable(test_spectool
        src/tests/test_spectool.cc
        src/audio/audio.cc
        src/audio/window.cc
        src/audio/fdct.cc
        src/audio/synth.cc
        src/audio/idct.cc
        third_party/glfw3webgpu/glfw3webgpu.c
    )
    target_include_directories(test_spectool PRIVATE
        src
        third_party
        ${WGPU_INCLUDE_DIR}
        third_party/glfw3webgpu
    )
    target_link_libraries(test_spectool PRIVATE ${DEMO_LIBS})
    add_test(NAME SpectoolEndToEndTest COMMAND test_spectool)

    add_executable(test_assets
        src/tests/test_assets.cc
        ${GENERATED_ASSETS_DATA_CC}
    )
    target_include_directories(test_assets PRIVATE ${CMAKE_CURRENT_BINARY_DIR}/src src)
    add_dependencies(test_assets generate_assets)
    add_test(NAME AssetManagerTest COMMAND test_assets)
endif()

# Tools
if(DEMO_BUILD_TOOLS)
    add_executable(spectool
        tools/spectool.cc
        src/platform.cc
        src/audio/audio.cc
        src/audio/fdct.cc
        src/audio/idct.cc
        src/audio/window.cc
        src/audio/synth.cc
        third_party/glfw3webgpu/glfw3webgpu.c
    )
    target_include_directories(spectool PRIVATE
        src
        third_party
        ${WGPU_INCLUDE_DIR}
        third_party/glfw3webgpu
    )
    target_link_libraries(spectool PRIVATE ${DEMO_LIBS})

    add_executable(specview
        tools/specview.cc
    )
    target_include_directories(specview PRIVATE src)
endif()