cmake_minimum_required(VERSION 3.20)
project(cnc_sim_wasm_core LANGUAGES CXX)

option(CNC_SIM_ENABLE_LINUXCNC_BRIDGE "Build the experimental LinuxCNC canon bridge" OFF)
option(CNC_SIM_ENABLE_LINUXCNC_RS274_BACKEND "Build the native LinuxCNC librs274 backend" OFF)
option(CNC_SIM_ENABLE_LINUXCNC_WASM_SAFE_PROBE "Compile the browser-safe subset of LinuxCNC rs274 sources" OFF)
set(CNC_SIM_LINUXCNC_ROOT "" CACHE PATH "LinuxCNC source root for the experimental bridge")
set(CNC_SIM_LINUXCNC_WASM_SOURCE_MANIFEST
    "${CMAKE_CURRENT_SOURCE_DIR}/../linuxcnc-rs274-wasm-source-files.txt"
    CACHE FILEPATH
    "Manifest that partitions LinuxCNC rs274 sources into wasm-safe core and blocked groups"
)

set(cnc_sim_core_sources
    src/canon_event_sink.cpp
    src/cnc_sim_api.cpp
    src/gcode_backend.cpp
    src/rtcp_kinematics.cpp
    src/simulator_gcode_controls.cpp
    src/smoke_gcode_parser.cpp
)

if(CNC_SIM_ENABLE_LINUXCNC_RS274_BACKEND)
    if(EMSCRIPTEN)
        message(FATAL_ERROR "CNC_SIM_ENABLE_LINUXCNC_RS274_BACKEND is native-only; it links LinuxCNC librs274")
    endif()
    if(NOT CNC_SIM_LINUXCNC_ROOT)
        message(FATAL_ERROR "Set CNC_SIM_LINUXCNC_ROOT to the LinuxCNC source root")
    endif()
    list(APPEND cnc_sim_core_sources
        src/linuxcnc_canon_bridge.cpp
        src/linuxcnc_rs274_backend.cpp
        src/linuxcnc_tooldata_fixture.cpp
    )
endif()

if(CNC_SIM_ENABLE_LINUXCNC_WASM_SAFE_PROBE)
    if(NOT CNC_SIM_LINUXCNC_ROOT)
        message(FATAL_ERROR "Set CNC_SIM_LINUXCNC_ROOT to the LinuxCNC source root")
    endif()
    get_filename_component(CNC_SIM_LINUXCNC_ROOT_ABS
        "${CNC_SIM_LINUXCNC_ROOT}"
        ABSOLUTE
        BASE_DIR "${CMAKE_CURRENT_SOURCE_DIR}/.."
    )
    if(NOT IS_DIRECTORY "${CNC_SIM_LINUXCNC_ROOT_ABS}")
        message(FATAL_ERROR "LinuxCNC source root does not exist: ${CNC_SIM_LINUXCNC_ROOT_ABS}")
    endif()
    if(NOT EXISTS "${CNC_SIM_LINUXCNC_WASM_SOURCE_MANIFEST}")
        message(FATAL_ERROR "Missing LinuxCNC wasm source manifest: ${CNC_SIM_LINUXCNC_WASM_SOURCE_MANIFEST}")
    endif()
    if(IS_DIRECTORY "${CNC_SIM_LINUXCNC_WASM_SOURCE_MANIFEST}")
        message(FATAL_ERROR "LinuxCNC wasm source manifest is not a file: ${CNC_SIM_LINUXCNC_WASM_SOURCE_MANIFEST}")
    endif()

    set(linuxcnc_wasm_safe_probe_sources)
    set(linuxcnc_wasm_blocked_sources)
    file(STRINGS "${CNC_SIM_LINUXCNC_WASM_SOURCE_MANIFEST}" linuxcnc_wasm_manifest_lines)
    foreach(manifest_line IN LISTS linuxcnc_wasm_manifest_lines)
        string(STRIP "${manifest_line}" manifest_line)
        if(manifest_line STREQUAL "" OR manifest_line MATCHES "^#")
            continue()
        endif()
        if(NOT manifest_line MATCHES "^([^:]+):([^:]+):(.*)$")
            message(FATAL_ERROR "Bad LinuxCNC wasm source manifest line: ${manifest_line}")
        endif()

        set(manifest_group "${CMAKE_MATCH_1}")
        set(manifest_path "${CMAKE_MATCH_2}")
        if(NOT manifest_group STREQUAL "core" AND NOT manifest_group STREQUAL "blocked")
            message(FATAL_ERROR "Unknown LinuxCNC wasm source manifest group: ${manifest_group}")
        endif()

        set(manifest_source "${CNC_SIM_LINUXCNC_ROOT_ABS}/${manifest_path}")
        if(NOT EXISTS "${manifest_source}")
            message(FATAL_ERROR "LinuxCNC wasm manifest source does not exist: ${manifest_source}")
        endif()
        if(IS_DIRECTORY "${manifest_source}")
            message(FATAL_ERROR "LinuxCNC wasm manifest source is not a file: ${manifest_source}")
        endif()

        if(manifest_group STREQUAL "core")
            list(APPEND linuxcnc_wasm_safe_probe_sources "${manifest_source}")
        else()
            list(APPEND linuxcnc_wasm_blocked_sources "${manifest_source}")
        endif()
    endforeach()

    list(LENGTH linuxcnc_wasm_safe_probe_sources linuxcnc_wasm_safe_probe_source_count)
    list(LENGTH linuxcnc_wasm_blocked_sources linuxcnc_wasm_blocked_source_count)
    if(linuxcnc_wasm_safe_probe_source_count EQUAL 0)
        message(FATAL_ERROR "LinuxCNC wasm source manifest has no core entries")
    endif()
    if(linuxcnc_wasm_blocked_source_count EQUAL 0)
        message(FATAL_ERROR "LinuxCNC wasm source manifest has no blocked entries")
    endif()
    message(STATUS "LinuxCNC wasm-safe core sources: ${linuxcnc_wasm_safe_probe_source_count}")
    message(STATUS "LinuxCNC wasm blocked sources: ${linuxcnc_wasm_blocked_source_count}")

    set(linuxcnc_wasm_safe_probe_shims
        ${CMAKE_CURRENT_SOURCE_DIR}/wasm_shims/dlfcn.cc
        ${CMAKE_CURRENT_SOURCE_DIR}/wasm_shims/emc_status_shim.cc
        ${CMAKE_CURRENT_SOURCE_DIR}/wasm_shims/gettext_shim.cc
        ${CMAKE_CURRENT_SOURCE_DIR}/wasm_shims/linuxcnc_runtime_shim.cc
        ${CMAKE_CURRENT_SOURCE_DIR}/wasm_shims/python_c_api_shim.cc
        ${CMAKE_CURRENT_SOURCE_DIR}/wasm_shims/rtapi_compat.cc
        ${CMAKE_CURRENT_SOURCE_DIR}/wasm_shims/tooldata/tooldata_mmap_backend.cc
        ${CMAKE_CURRENT_SOURCE_DIR}/wasm_shims/tooldata/tooldata_runtime_stubs.cc
        ${CMAKE_CURRENT_SOURCE_DIR}/wasm_shims/pythonplugin/python_plugin.cc
    )

    add_library(linuxcnc_rs274_wasm_safe_probe_objects OBJECT
        ${linuxcnc_wasm_safe_probe_shims}
        ${linuxcnc_wasm_safe_probe_sources}
    )
    target_compile_features(linuxcnc_rs274_wasm_safe_probe_objects PUBLIC cxx_std_20)
    target_compile_definitions(linuxcnc_rs274_wasm_safe_probe_objects PRIVATE ULAPI)
    target_compile_options(linuxcnc_rs274_wasm_safe_probe_objects PRIVATE -include wctype.h)
    target_include_directories(linuxcnc_rs274_wasm_safe_probe_objects
        PRIVATE
            ${CMAKE_CURRENT_SOURCE_DIR}/include
            ${CMAKE_CURRENT_SOURCE_DIR}/src
            ${CMAKE_CURRENT_SOURCE_DIR}/wasm_shims
            ${CNC_SIM_LINUXCNC_ROOT_ABS}/src
            ${CNC_SIM_LINUXCNC_ROOT_ABS}/src/emc
            ${CNC_SIM_LINUXCNC_ROOT_ABS}/src/emc/nml_intf
            ${CNC_SIM_LINUXCNC_ROOT_ABS}/src/emc/rs274ngc
            ${CNC_SIM_LINUXCNC_ROOT_ABS}/src/emc/motion
            ${CNC_SIM_LINUXCNC_ROOT_ABS}/include
    )

    set(linuxcnc_rs274_wasm_safe_probe_project_sources
        ${CMAKE_CURRENT_SOURCE_DIR}/src/canon_event_sink.cpp
        ${CMAKE_CURRENT_SOURCE_DIR}/src/linuxcnc_canon_bridge.cpp
        ${CMAKE_CURRENT_SOURCE_DIR}/src/linuxcnc_tooldata_fixture.cpp
        ${CMAKE_CURRENT_SOURCE_DIR}/src/rtcp_kinematics.cpp
    )

    add_executable(linuxcnc_rs274_wasm_safe_probe EXCLUDE_FROM_ALL
        $<TARGET_OBJECTS:linuxcnc_rs274_wasm_safe_probe_objects>
        ${linuxcnc_rs274_wasm_safe_probe_project_sources}
        ${CMAKE_CURRENT_SOURCE_DIR}/wasm_shims/rs274ngc_pre_probe_main.cc
    )
    target_compile_features(linuxcnc_rs274_wasm_safe_probe PRIVATE cxx_std_20)
    target_compile_definitions(linuxcnc_rs274_wasm_safe_probe PRIVATE ULAPI)
    target_compile_options(linuxcnc_rs274_wasm_safe_probe PRIVATE -include wctype.h)
    target_include_directories(linuxcnc_rs274_wasm_safe_probe
        PRIVATE
            ${CMAKE_CURRENT_SOURCE_DIR}/include
            ${CMAKE_CURRENT_SOURCE_DIR}/src
            ${CMAKE_CURRENT_SOURCE_DIR}/wasm_shims
            ${CNC_SIM_LINUXCNC_ROOT_ABS}/src
            ${CNC_SIM_LINUXCNC_ROOT_ABS}/src/emc
            ${CNC_SIM_LINUXCNC_ROOT_ABS}/src/emc/nml_intf
            ${CNC_SIM_LINUXCNC_ROOT_ABS}/src/emc/rs274ngc
            ${CNC_SIM_LINUXCNC_ROOT_ABS}/src/emc/motion
            ${CNC_SIM_LINUXCNC_ROOT_ABS}/include
    )
    target_link_libraries(linuxcnc_rs274_wasm_safe_probe PRIVATE fmt)

    if(linuxcnc_wasm_blocked_sources)
        add_custom_target(linuxcnc_rs274_wasm_blocked_sources
            SOURCES ${linuxcnc_wasm_blocked_sources}
        )
    endif()
endif()

add_library(cnc_sim_objects OBJECT
    ${cnc_sim_core_sources}
)

target_include_directories(cnc_sim_objects
    PUBLIC
        ${CMAKE_CURRENT_SOURCE_DIR}/include
    PRIVATE
        ${CMAKE_CURRENT_SOURCE_DIR}/src
)

target_compile_features(cnc_sim_objects PUBLIC cxx_std_17)

if(CNC_SIM_ENABLE_LINUXCNC_RS274_BACKEND)
    find_package(Python3 REQUIRED COMPONENTS Development)
    target_compile_definitions(cnc_sim_objects
        PRIVATE
            CNC_SIM_ENABLE_LINUXCNC_RS274_BACKEND
    )
    target_include_directories(cnc_sim_objects
        PRIVATE
            ${Python3_INCLUDE_DIRS}
            ${CNC_SIM_LINUXCNC_ROOT}/src
            ${CNC_SIM_LINUXCNC_ROOT}/src/emc
            ${CNC_SIM_LINUXCNC_ROOT}/src/emc/nml_intf
            ${CNC_SIM_LINUXCNC_ROOT}/src/emc/rs274ngc
            ${CNC_SIM_LINUXCNC_ROOT}/src/emc/motion
            ${CNC_SIM_LINUXCNC_ROOT}/include
    )
endif()

add_library(cnc_sim_core STATIC
    $<TARGET_OBJECTS:cnc_sim_objects>
)

target_include_directories(cnc_sim_core
    PUBLIC
        ${CMAKE_CURRENT_SOURCE_DIR}/include
    PRIVATE
        ${CMAKE_CURRENT_SOURCE_DIR}/src
)

if(CNC_SIM_ENABLE_LINUXCNC_RS274_BACKEND)
    target_link_directories(cnc_sim_core
        PUBLIC
            ${CNC_SIM_LINUXCNC_ROOT}/lib
    )
    target_link_libraries(cnc_sim_core
        PUBLIC
            rs274
            tooldata
            Python3::Python
    )
    target_link_options(cnc_sim_core
        PUBLIC
            "-Wl,-rpath,${CNC_SIM_LINUXCNC_ROOT}/lib"
    )
endif()

if(NOT EMSCRIPTEN)
    add_executable(cnc_sim_api_smoke
        tests/cnc_sim_api_smoke.cpp
    )
    target_link_libraries(cnc_sim_api_smoke PRIVATE cnc_sim_core)
    target_include_directories(cnc_sim_api_smoke PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/src)

    add_executable(canon_event_sink_smoke
        tests/canon_event_sink_smoke.cpp
    )
    target_link_libraries(canon_event_sink_smoke PRIVATE cnc_sim_core)
    target_include_directories(canon_event_sink_smoke PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/src)

    add_executable(rtcp_kinematics_smoke
        tests/rtcp_kinematics_smoke.cpp
    )
    target_link_libraries(rtcp_kinematics_smoke PRIVATE cnc_sim_core)
    target_include_directories(rtcp_kinematics_smoke PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/src)

    add_executable(linuxcnc_gees_table_smoke
        tests/linuxcnc_gees_table_smoke.cpp
    )
    target_include_directories(linuxcnc_gees_table_smoke PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/src)

    add_executable(simulator_gcode_controls_smoke
        tests/simulator_gcode_controls_smoke.cpp
    )
    target_link_libraries(simulator_gcode_controls_smoke PRIVATE cnc_sim_core)
    target_include_directories(simulator_gcode_controls_smoke PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/src)

    add_executable(cnc_sim_dump
        tools/cnc_sim_dump.cpp
    )
    target_link_libraries(cnc_sim_dump PRIVATE cnc_sim_core)
    target_include_directories(cnc_sim_dump PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/src)

    if(CNC_SIM_ENABLE_LINUXCNC_RS274_BACKEND)
        add_executable(cnc_sim_api_linuxcnc_rs274_smoke
            tests/cnc_sim_api_linuxcnc_rs274_smoke.cpp
        )
        target_link_libraries(cnc_sim_api_linuxcnc_rs274_smoke PRIVATE cnc_sim_core)
        target_include_directories(cnc_sim_api_linuxcnc_rs274_smoke PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/src)
    endif()
endif()

if(CNC_SIM_ENABLE_LINUXCNC_BRIDGE)
    if(NOT CNC_SIM_LINUXCNC_ROOT)
        message(FATAL_ERROR "Set CNC_SIM_LINUXCNC_ROOT to the LinuxCNC source root")
    endif()
    add_library(cnc_sim_linuxcnc_canon_bridge STATIC
        src/linuxcnc_canon_bridge.cpp
    )
    target_include_directories(cnc_sim_linuxcnc_canon_bridge
        PRIVATE
            ${CMAKE_CURRENT_SOURCE_DIR}/include
            ${CMAKE_CURRENT_SOURCE_DIR}/src
            ${CNC_SIM_LINUXCNC_ROOT}/src/emc
            ${CNC_SIM_LINUXCNC_ROOT}/src/emc/nml_intf
            ${CNC_SIM_LINUXCNC_ROOT}/src/emc/rs274ngc
            ${CNC_SIM_LINUXCNC_ROOT}/src
            ${CNC_SIM_LINUXCNC_ROOT}/include
    )
    target_link_directories(cnc_sim_linuxcnc_canon_bridge
        PRIVATE
            ${CNC_SIM_LINUXCNC_ROOT}/lib
    )
    target_link_libraries(cnc_sim_linuxcnc_canon_bridge PRIVATE cnc_sim_core tooldata)
    target_link_options(cnc_sim_linuxcnc_canon_bridge
        PRIVATE
            "-Wl,-rpath,${CNC_SIM_LINUXCNC_ROOT}/lib"
    )
endif()

if(EMSCRIPTEN)
    add_executable(cnc_sim_wasm
        $<TARGET_OBJECTS:cnc_sim_objects>
    )

    target_include_directories(cnc_sim_wasm
        PRIVATE
            ${CMAKE_CURRENT_SOURCE_DIR}/include
            ${CMAKE_CURRENT_SOURCE_DIR}/src
    )

    set_target_properties(cnc_sim_wasm PROPERTIES
        OUTPUT_NAME "cnc_sim"
        SUFFIX ".js"
    )

    target_link_options(cnc_sim_wasm PRIVATE
        "--no-entry"
        "-sMODULARIZE=1"
        "-sEXPORT_NAME=createCncSimModule"
        "-sALLOW_MEMORY_GROWTH=1"
        "-sALLOW_TABLE_GROWTH=1"
        "-sEXPORTED_FUNCTIONS=['_malloc','_free','_cnc_sim_create','_cnc_sim_destroy','_cnc_sim_reset','_cnc_sim_set_dialect','_cnc_sim_set_event_callback','_cnc_sim_load_config_json','_cnc_sim_parse_program','_cnc_sim_last_error']"
        "-sEXPORTED_RUNTIME_METHODS=['ccall','cwrap','UTF8ToString','stringToUTF8','lengthBytesUTF8','addFunction','removeFunction']"
    )
endif()
