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

set(CMAKE_CXX_STANDARD 17)
set(CMAKE_CXX_STANDARD_REQUIRED ON)

#-- - Configuration Options -- -
option(DEMO_SIZE_OPT "Enable size optimization flags" OFF)
option(DEMO_STRIP_ALL "Strip all unnecessary code for final build" OFF)
option(DEMO_BUILD_TESTS "Build tests" OFF)
option(DEMO_BUILD_TOOLS "Build tools" OFF)
option(DEMO_ALL_OPTIONS "Activate all options at once" OFF)

if (DEMO_ALL_OPTIONS)
    set(DEMO_SIZE_OPT ON)
    set(DEMO_STRIP_ALL ON)
    set(DEMO_BUILD_TESTS ON)
    set(DEMO_BUILD_TOOLS ON)
endif()

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

#-- - Platform Detection and Core Setup -- -
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()

#-- - Dependencies & Includes -- -
set(CORE_INCLUDES src third_party ${CMAKE_CURRENT_BINARY_DIR}/src)

if (DEMO_CROSS_COMPILE_WIN32)
    add_definitions(-DDEMO_CROSS_COMPILE_WIN32)
    set(WINDOWS_DEPS_DIR "${CMAKE_CURRENT_SOURCE_DIR}/third_party/windows")
    set(WGPU_INCLUDE_DIR "${WINDOWS_DEPS_DIR}/include/webgpu")
    set(WGPU_LIBRARY "${WINDOWS_DEPS_DIR}/lib/libwgpu_native.dll.a")
    set(GLFW3_INCLUDE_DIR "${WINDOWS_DEPS_DIR}/include")
    set(GLFW3_LIBRARY "${WINDOWS_DEPS_DIR}/lib/libglfw3.a")
    
    list(APPEND CORE_INCLUDES ${WGPU_INCLUDE_DIR} ${GLFW3_INCLUDE_DIR})
    set(DEMO_LIBS ${GLFW3_LIBRARY} ${WGPU_LIBRARY} -lgdi32 -lws2_32 -luser32 -lkernel32 -lshell32 -ladvapi32 -ldwmapi)
    set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} -static-libgcc -static-libstdc++")
else()
    find_library(WGPU_LIBRARY NAMES wgpu_native libwgpu_native REQUIRED)
    find_path(WGPU_ROOT_INCLUDE_DIR NAMES wgpu.h REQUIRED)
    list(APPEND CORE_INCLUDES ${WGPU_ROOT_INCLUDE_DIR} ${WGPU_ROOT_INCLUDE_DIR}/webgpu-headers)
    find_package(glfw3 REQUIRED)
    set(DEMO_LIBS glfw ${WGPU_LIBRARY})
endif()

list(APPEND CORE_INCLUDES third_party/glfw3webgpu)
include_directories(${CORE_INCLUDES})

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")
elseif (NOT DEMO_CROSS_COMPILE_WIN32)
    list(APPEND DEMO_LIBS pthread m dl)
endif()

#-- - Source Groups -- -
set(AUDIO_SOURCES src/audio/audio.cc src/audio/gen.cc src/audio/fdct.cc src/audio/idct.cc src/audio/window.cc src/audio/synth.cc)
set(PROCEDURAL_SOURCES src/procedural/generator.cc)
set(GPU_SOURCES
    src/gpu/gpu.cc
    src/gpu/effect.cc
    src/gpu/effects/heptagon_effect.cc
    src/gpu/effects/particles_effect.cc
    src/gpu/effects/passthrough_effect.cc
    src/gpu/effects/moving_ellipse_effect.cc
    src/gpu/effects/particle_spray_effect.cc
    src/gpu/effects/gaussian_blur_effect.cc
    src/gpu/effects/solarize_effect.cc
    src/gpu/effects/distort_effect.cc
    src/gpu/effects/chroma_aberration_effect.cc
    src/gpu/effects/post_process_helper.cc
    src/gpu/effects/shaders.cc
    src/gpu/effects/hybrid_3d_effect.cc
    src/gpu/texture_manager.cc
    src/3d/renderer.cc
)
set(UTIL_SOURCES src/util/asset_manager.cc ${GEN_DEMO_CC})
set(PLATFORM_SOURCES src/platform.cc third_party/glfw3webgpu/glfw3webgpu.c)

#-- - Tools Setup -- -
if (DEFINED ASSET_PACKER_PATH)
    set(ASSET_PACKER_CMD ${ASSET_PACKER_PATH})
    set(ASSET_PACKER_DEPENDS ${ASSET_PACKER_PATH})
else()
    add_executable(asset_packer tools/asset_packer.cc ${PROCEDURAL_SOURCES})
    set(ASSET_PACKER_CMD $<TARGET_FILE:asset_packer>)
    set(ASSET_PACKER_DEPENDS asset_packer)
endif()

if (DEFINED SEQ_COMPILER_PATH)
    set(SEQ_COMPILER_CMD ${SEQ_COMPILER_PATH})
    set(SEQ_COMPILER_DEPENDS ${SEQ_COMPILER_PATH})
else()
    add_executable(seq_compiler tools/seq_compiler.cc)
    set(SEQ_COMPILER_CMD $<TARGET_FILE:seq_compiler>)
    set(SEQ_COMPILER_DEPENDS seq_compiler)
endif()

#-- - Code Generation Helpers -- -
function(pack_assets NAME INPUT_TXT HEADER_VAR DATA_CC_VAR TARGET_NAME)
    set(OUT_H ${CMAKE_CURRENT_BINARY_DIR}/src/generated/${NAME}.h)
    set(OUT_CC ${CMAKE_CURRENT_BINARY_DIR}/src/generated/${NAME}_data.cc)
    add_custom_command(
        OUTPUT ${OUT_H} ${OUT_CC}
        COMMAND ${CMAKE_COMMAND} -E make_directory ${CMAKE_CURRENT_BINARY_DIR}/src/generated
        COMMAND ${ASSET_PACKER_CMD} ${INPUT_TXT} ${OUT_H} ${OUT_CC}
        DEPENDS ${ASSET_PACKER_DEPENDS} ${INPUT_TXT}
        COMMENT "Generating assets for ${NAME}..."
    )
    set(${HEADER_VAR} ${OUT_H} PARENT_SCOPE)
    set(${DATA_CC_VAR} ${OUT_CC} PARENT_SCOPE)
    add_custom_target(${TARGET_NAME} DEPENDS ${OUT_H} ${OUT_CC})
endfunction()

#-- - Generation Targets -- -
set(DEMO_SEQ_PATH ${CMAKE_CURRENT_SOURCE_DIR}/assets/demo.seq)
set(GENERATED_TIMELINE_CC ${CMAKE_CURRENT_BINARY_DIR}/src/generated/timeline.cc)
add_custom_command(
    OUTPUT ${GENERATED_TIMELINE_CC}
    COMMAND ${CMAKE_COMMAND} -E make_directory ${CMAKE_CURRENT_BINARY_DIR}/src/generated
    COMMAND ${SEQ_COMPILER_CMD} ${DEMO_SEQ_PATH} ${GENERATED_TIMELINE_CC}
    DEPENDS ${SEQ_COMPILER_DEPENDS} ${DEMO_SEQ_PATH}
            src/gpu/demo_effects.h
    COMMENT \"Compiling demo sequence...\"
)
add_custom_target(generate_timeline ALL DEPENDS ${GENERATED_TIMELINE_CC})

pack_assets(assets ${CMAKE_CURRENT_SOURCE_DIR}/assets/final/demo_assets.txt GEN_DEMO_H GEN_DEMO_CC generate_demo_assets)
pack_assets(test_assets ${CMAKE_CURRENT_SOURCE_DIR}/assets/final/test_assets_list.txt GEN_TEST_H GEN_TEST_CC generate_test_assets)

#-- - Main Demo -- -
add_executable(demo64k src/main.cc ${PLATFORM_SOURCES} ${GPU_SOURCES} ${AUDIO_SOURCES} ${PROCEDURAL_SOURCES} ${UTIL_SOURCES} ${GEN_DEMO_CC} ${GENERATED_TIMELINE_CC})
add_dependencies(demo64k generate_demo_assets generate_timeline)
target_link_libraries(demo64k PRIVATE ${DEMO_LIBS})

#Size optimizations
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 -- -
enable_testing()
if(DEMO_BUILD_TESTS)
    add_executable(test_window src/tests/test_window.cc src/audio/window.cc)
    add_test(NAME HammingWindowTest COMMAND test_window)

    add_executable(test_maths src/tests/test_maths.cc)
    add_test(NAME MathUtilsTest COMMAND test_maths)

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

    add_executable(test_spectool src/tests/test_spectool.cc ${AUDIO_SOURCES} ${PLATFORM_SOURCES})
    target_compile_definitions(test_spectool PRIVATE DEMO_BUILD_TOOLS)
    target_link_libraries(test_spectool PRIVATE ${DEMO_LIBS})
    add_test(NAME SpectoolEndToEndTest COMMAND test_spectool)

    add_executable(test_assets src/tests/test_assets.cc ${UTIL_SOURCES} ${PROCEDURAL_SOURCES} ${GEN_TEST_CC})
    target_compile_definitions(test_assets PRIVATE USE_TEST_ASSETS)
    add_dependencies(test_assets generate_test_assets)
    set_source_files_properties(src/tests/test_assets.cc PROPERTIES COMPILE_DEFINITIONS "USE_TEST_ASSETS")
    add_test(NAME AssetManagerTest COMMAND test_assets)

    add_executable(test_sequence src/tests/test_sequence.cc ${GPU_SOURCES} ${GENERATED_TIMELINE_CC} ${PLATFORM_SOURCES} ${UTIL_SOURCES})
    target_link_libraries(test_sequence PRIVATE ${DEMO_LIBS})
    add_dependencies(test_sequence generate_timeline)
    add_test(NAME SequenceSystemTest COMMAND test_sequence)

    add_executable(test_procedural src/tests/test_procedural.cc ${PROCEDURAL_SOURCES})
    add_test(NAME ProceduralGenTest COMMAND test_procedural)

    add_executable(test_3d src/tests/test_3d.cc)
    add_test(NAME ThreeDSystemTest COMMAND test_3d)

    add_executable(test_texture_manager src/tests/test_texture_manager.cc ${GPU_SOURCES} ${PLATFORM_SOURCES} ${PROCEDURAL_SOURCES} ${AUDIO_SOURCES} ${GENERATED_TIMELINE_CC} ${UTIL_SOURCES})
    target_link_libraries(test_texture_manager PRIVATE ${DEMO_LIBS})
    add_test(NAME TextureManagerTest COMMAND test_texture_manager)

    add_executable(test_3d_render src/tests/test_3d_render.cc src/3d/renderer.cc ${GPU_SOURCES} ${PLATFORM_SOURCES} ${PROCEDURAL_SOURCES} ${AUDIO_SOURCES} ${GENERATED_TIMELINE_CC} ${UTIL_SOURCES})
    target_link_libraries(test_3d_render PRIVATE ${DEMO_LIBS})
endif()

#-- - Extra Tools -- -
if(DEMO_BUILD_TOOLS OR DEMO_BUILD_TESTS)
    add_executable(spectool tools/spectool.cc ${PLATFORM_SOURCES} ${AUDIO_SOURCES})
    target_compile_definitions(spectool PRIVATE DEMO_BUILD_TOOLS)
    target_link_libraries(spectool PRIVATE ${DEMO_LIBS})

    add_executable(specview tools/specview.cc)
endif()

#-- - Global Target Configuration -- -
add_custom_target(final
    COMMAND ${CMAKE_CURRENT_SOURCE_DIR}/scripts/gen_assets.sh
    COMMAND ${CMAKE_CURRENT_SOURCE_DIR}/scripts/crunch_demo.sh
    DEPENDS demo64k
    WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}
)

add_custom_target(pack_source
    COMMAND tar -czf demo_all.tgz --exclude=.git --exclude=build* --exclude=.gemini* --exclude=*.tgz --exclude=*.zip --exclude=.DS_Store .
    WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}
)