cmake_minimum_required(VERSION 3.15)
include(cmake/Utilities.cmake)

set (CMAKE_CXX_STANDARD 11)

OPTION(SECONDARY_LANGUAGES "Secondary language support in the firmware" ON)

get_dependency_directory(prusa3dboards PRUSA_BOARDS_DIR)
project(Prusa-Firmware)

FILE(STRINGS ${CMAKE_CURRENT_SOURCE_DIR}/Firmware/config.h MAX_SIZE_LINE REGEX "^#define \+LANG_SIZE_RESERVED \+")
STRING(REGEX MATCH "0x[0-9]+" MAX_SIZE_HEX "${MAX_SIZE_LINE}")
math(EXPR LANG_MAX_SIZE "${MAX_SIZE_HEX}" OUTPUT_FORMAT DECIMAL)
message("Language maximum size (from config.h): ${LANG_MAX_SIZE} bytes")

set (LANG_BIN_MAX 249856) # Ditto, this in xflash_layout.h

get_recommended_gcc_version(RECOMMENDED_TOOLCHAIN_VERSION)
if(CMAKE_CROSSCOMPILING AND NOT CMAKE_CXX_COMPILER_VERSION VERSION_EQUAL
                            ${RECOMMENDED_TOOLCHAIN_VERSION}
   )
  message(WARNING "Recommended AVR toolchain is ${RECOMMENDED_TOOLCHAIN_VERSION}"
                  ", but you have ${CMAKE_CXX_COMPILER_VERSION}"
          )

elseif(NOT CMAKE_CROSSCOMPILING AND NOT CMAKE_CXX_COMPILER_ID STREQUAL "GNU")
  message(
    WARNING
      "Recommended compiler for host tools and unittests is GCC, you have ${CMAKE_CXX_COMPILER_ID}."
    )
endif()

# append custom C/C++ flags
if(CUSTOM_COMPILE_OPTIONS)
  string(REPLACE " " ";" CUSTOM_COMPILE_OPTIONS "${CUSTOM_COMPILE_OPTIONS}")
  add_compile_options(${CUSTOM_COMPILE_OPTIONS})
endif()

#
# Global Compiler & Linker Configuration
#

# include symbols
add_compile_options(-g)

# optimizations
if(CMAKE_CROSSCOMPILING)
  if(CMAKE_BUILD_TYPE STREQUAL "Debug")
    add_compile_options(-Og)
  else()
    add_compile_options(-Os)
  endif()

  # mcu related settings
  set(MCU_FLAGS -mmcu=atmega2560 -DF_CPU=16000000L)
  add_compile_options(${MCU_FLAGS})
  add_link_options(${MCU_FLAGS})

  # split and gc sections
  add_compile_options(-ffunction-sections -fdata-sections)
  add_link_options(-Wl,--gc-sections)

  # disable exceptions and related metadata
  add_compile_options(-fno-exceptions -fno-unwind-tables)
  add_compile_options($<$<COMPILE_LANGUAGE:CXX>:-fno-rtti>)
  add_link_options(-Wl,--defsym,__exidx_start=0,--defsym,__exidx_end=0)
else()
  if(CMAKE_BUILD_TYPE STREQUAL "Debug")
    add_compile_options(-O0)
  else()
    add_compile_options(-O2)
  endif()
endif()

# enable all warnings (well, not all, but some)
add_compile_options(-Wall -Wsign-compare)
add_compile_options($<$<COMPILE_LANGUAGE:CXX>:-std=c++14>)

# support _DEBUG macro (some code uses to recognize debug builds)
if(CMAKE_BUILD_TYPE STREQUAL "Debug")
  add_compile_definitions(_DEBUG)
endif()

#
# Firmware - get file lists.
#
file(GLOB FW_SOURCES RELATIVE ${PROJECT_SOURCE_DIR} ${PROJECT_SOURCE_DIR}/Firmware/*.c*)
file(GLOB FW_HEADERS RELATIVE ${PROJECT_SOURCE_DIR} ${PROJECT_SOURCE_DIR}/Firmware/*.h*)
file(GLOB AVR_SOURCES RELATIVE ${PROJECT_SOURCE_DIR} ${PRUSA_BOARDS_DIR}/cores/prusa_einsy_rambo/*.c*)

  # Setup language resources:
file(GLOB LANG_VARIANTS RELATIVE ${PROJECT_SOURCE_DIR}/lang/po ${PROJECT_SOURCE_DIR}/lang/po/Firmware_??.po)
string(REPLACE "Firmware_" "" LANG_VARIANTS "${LANG_VARIANTS}")
string(REPLACE ".po" "" LANG_VARIANTS "${LANG_VARIANTS}")
list(SORT LANG_VARIANTS)
message("Languages found: ${LANG_VARIANTS}")

add_library(avr_core STATIC ${AVR_SOURCES})
target_include_directories(avr_core PRIVATE
  ${PRUSA_BOARDS_DIR}/cores/prusa_einsy_rambo/
  ${PRUSA_BOARDS_DIR}/variants/prusa_einsy_rambo/
)
target_compile_options(avr_core PUBLIC -mmcu=atmega2560)

# Meta target to build absolutely everything
add_custom_target(ALL_FIRMWARE)

function(fw_add_variant variant_name)

  # Create the Configuration_Prusa.h for this variant so it can be #included.
  set(VARIANT_CFG_DIR "${CMAKE_CURRENT_BINARY_DIR}/${variant_name}_include")
  set(VARIANT_CFG_FILE "${VARIANT_CFG_DIR}/Configuration_prusa.h")
  add_custom_command(OUTPUT ${VARIANT_CFG_FILE}
      COMMAND ${CMAKE_COMMAND} -E copy ${PROJECT_SOURCE_DIR}/Firmware/variants/${variant_name}.h ${VARIANT_CFG_FILE}
      COMMENT "Generating Configuration_prusa.h for ${variant_name}"
      BYPRODUCTS ${VARIANT_CFG_DIR}
  )
  STRING(REPLACE "1_75mm_" "" variant_name "${variant_name}")
  STRING(REPLACE "-E3Dv6full" "" variant_name "${variant_name}")

  add_executable(${variant_name} ${FW_SOURCES} ${FW_HEADERS} ${VARIANT_CFG_FILE})

  set_target_properties(${variant_name} PROPERTIES CXX_STANDARD 14)

  target_include_directories(${variant_name} PRIVATE Firmware
    ${PRUSA_BOARDS_DIR}/cores/prusa_einsy_rambo/
    ${PRUSA_BOARDS_DIR}/variants/prusa_einsy_rambo/
    ${VARIANT_CFG_DIR} # Include the header for this variant.
    )

#   # configure linker script
   set(LINKER_SCRIPT ${PRUSA_BOARDS_DIR}/ldscripts/avr6.xn)
   target_link_options(${variant_name} PUBLIC -Wl,-T,${LINKER_SCRIPT})

  # limit the text section to 248K (256K - 8k reserved for the bootloader)
  target_link_options(${variant_name} PUBLIC -Wl,--defsym=__TEXT_REGION_LENGTH__=248K)

  # generate firmware.bin file
  objcopy(${variant_name} "ihex" ".hex")

  # produce ASM listing. Note we also specify the .map as a byproduct so it gets cleaned
  # because link_options doesn't have a "generated outputs" feature.
  add_custom_command(
    TARGET ${variant_name} POST_BUILD COMMAND ${CMAKE_OBJDUMP} -CSd ${variant_name} > ${variant_name}.asm
    BYPRODUCTS ${variant_name}.asm ${variant_name}.map
    )

  # inform about the firmware's size in terminal
  add_custom_command(
    TARGET ${variant_name} POST_BUILD COMMAND ${CMAKE_SIZE_UTIL} -C --mcu=atmega2560 ${variant_name}
    )
  report_size(${variant_name})

  # generate linker map file
  target_link_options(${variant_name} PUBLIC -Wl,-Map=${variant_name}.map)

  target_compile_options(${variant_name} PRIVATE) # turn this on for lolz -Wdouble-promotion)
  target_compile_definitions(${variant_name} PRIVATE
    ARDUINO=10600
    __AVR_ATmega2560__
    CMAKE_LANG_CONTROL
    )
  target_link_libraries(${variant_name} avr_core)
  if (SECONDARY_LANGUAGES)
    target_compile_definitions(${variant_name} PUBLIC LANG_MODE=1)
  else()
    target_compile_definitions(${variant_name} PUBLIC LANG_MODE=0)
    add_dependencies(ALL_FIRMWARE "${variant_name}")
    return() #Done, if no languages there's nothing else to do.
  endif()

  #Construct language map
  set(LANG_TMP_DIR ${CMAKE_CURRENT_BINARY_DIR}/lang)
  set(LANG_MAP ${LANG_TMP_DIR}/${variant_name}_lang.map)
  set(LANG_FWBIN ${CMAKE_BINARY_DIR}/${variant_name}.bin)
  set(LANG_FINAL_BIN ${LANG_TMP_DIR}/${variant_name}_lang.bin)
  set(LANG_FINAL_HEX ${LANG_TMP_DIR}/${variant_name}_lang.hex)

  add_custom_command(OUTPUT ${LANG_FWBIN}
    COMMAND "${CMAKE_OBJCOPY}" -I ihex -O binary ${CMAKE_BINARY_DIR}/${variant_name}.hex ${LANG_FWBIN}
    DEPENDS ${variant_name}
  )
  add_custom_command(OUTPUT ${LANG_MAP}
    COMMAND ${CMAKE_CURRENT_SOURCE_DIR}/lang/lang-map.py "${CMAKE_BINARY_DIR}/${variant_name}" "${LANG_FWBIN}" > "${LANG_MAP}"
    DEPENDS ${LANG_FWBIN}
  )

  set(LANG_BINS "")

  foreach (LANG IN LISTS LANG_VARIANTS)
    set(LANG_BIN ${LANG_TMP_DIR}/${variant_name}_${LANG}.bin)

    set(PO_FILE "${CMAKE_CURRENT_SOURCE_DIR}/lang/po/Firmware_${LANG}.po")
    add_custom_command(OUTPUT ${LANG_BIN}
  #      COMMAND ${CMAKE_CURRENT_SOURCE_DIR}/lang/lang-check.py --no-warning --map "${LANG_MAP}" "${PO_FILE}"
  #      COMMAND ${CMAKE_COMMAND} -E echo "Building lang_${LANG}.bin"
        COMMAND ${CMAKE_CURRENT_SOURCE_DIR}/lang/lang-build.py ${LANG_MAP} ${PO_FILE} ${LANG_BIN}
        # Check bin size:
        COMMAND ${CMAKE_COMMAND} -DLANG_MAX_SIZE=${LANG_MAX_SIZE}
            -DLANG_FILE=${LANG_BIN}
            -P ${PROJECT_CMAKE_DIR}/Check_lang_size.cmake
        DEPENDS ${LANG_MAP}
        COMMENT "Generating ${variant_name}_${LANG}.bin from .po"
        )
        LIST(APPEND LANG_BINS ${LANG_BIN})

  endforeach()
  string(FIND ${variant_name} "MK3" HAS_XFLASH)
  if (${HAS_XFLASH} GREATER_EQUAL 0)
    if (WIN32)
        STRING(REPLACE ";" " + " LANG_CMD_TMP "${LANG_BINS}")
        add_custom_command( OUTPUT ${LANG_FINAL_BIN}
            COMMAND copy /b ${LANG_CMD_TMP} ${LANG_FINAL_BIN}
            DEPENDS ${LANG_BINS}
            COMMENT "Merging language binaries (W32)"
        )
    else()
        add_custom_command( OUTPUT ${LANG_FINAL_BIN}
            COMMAND cat ${LANG_BINS} > ${LANG_FINAL_BIN}
            DEPENDS ${LANG_BINS}
            COMMENT "Merging language binaries (Non-W32)"
        )
    endif()
    add_custom_command( OUTPUT ${LANG_FINAL_BIN}
        COMMAND ${CMAKE_COMMAND} -DLANG_MAX_SIZE=${LANG_BIN_MAX}
            -DLANG_FILE=${LANG_FINAL_BIN}
            -P ${PROJECT_CMAKE_DIR}/Check_final_lang_bin_size.cmake
        APPEND
    )
    add_custom_command( OUTPUT ${LANG_FINAL_HEX}
        COMMAND ${CMAKE_OBJCOPY} -I binary -O ihex ${LANG_FINAL_BIN} ${LANG_FINAL_HEX}
        DEPENDS ${LANG_FINAL_BIN}
        COMMENT "Generating Hex for language data"
    )
    set(LANG_HEX ${CMAKE_BINARY_DIR}/${variant_name}-lang.hex)
    if (WIN32)
        SET(TEXT_MERGE_CMD "type")
    else()
        SET(TEXT_MERGE_CMD "cat")
    endif()
    add_custom_target(${variant_name}-languages
        COMMAND ${CMAKE_COMMAND} -E copy ${CMAKE_BINARY_DIR}/${variant_name}.hex ${LANG_HEX}
        COMMAND ${TEXT_MERGE_CMD} ${LANG_FINAL_HEX} >> ${LANG_HEX}
        COMMENT "Generating final ${variant_name}-lang.hex"
        BYPRODUCTS ${LANG_HEX}
        DEPENDS ${LANG_FINAL_HEX}
    )
    add_dependencies(ALL_FIRMWARE ${variant_name}-languages)
  else()
    set (ALL_VARIANT_HEXES "")
    # Non-xflash, e.g. MK2.5
    foreach(LANG IN LISTS LANG_VARIANTS)
        SET(LANG_HEX_FN ${variant_name}-en_${LANG})
        SET(LANG_HEX ${CMAKE_BINARY_DIR}/${LANG_HEX_FN}.hex)
        SET(LANG_BIN ${LANG_TMP_DIR}/${variant_name}_${LANG}.bin)
        SET(LANG_FWBIN_TMP ${LANG_TMP_DIR}/${variant_name}-en_${LANG}.bin)

        #Intermediate 2-lang bin
        add_custom_command(OUTPUT ${LANG_FWBIN_TMP}
            COMMAND ${CMAKE_COMMAND} -E copy ${LANG_FWBIN} ${LANG_FWBIN_TMP}
            COMMAND ${CMAKE_CURRENT_SOURCE_DIR}/lang/lang-patchsec.py ${CMAKE_BINARY_DIR}/${variant_name} ${LANG_BIN} ${LANG_FWBIN_TMP}
            DEPENDS ${LANG_FWBIN} ${LANG_BIN}
            COMMENT "Generating ${variant_name}-en_${LANG}.bin"
        )

        #Final hex:
        add_custom_target(${LANG_HEX_FN}
            COMMAND ${CMAKE_OBJCOPY} -I binary -O ihex ${LANG_FWBIN_TMP} ${LANG_HEX}
            BYPRODUCTS ${LANG_HEX}
            DEPENDS ${LANG_FWBIN_TMP}
            COMMENT "Creating ${LANG_HEX_FN}.hex"
        )
        LIST(APPEND ALL_VARIANT_HEXES ${LANG_HEX})
    endforeach()
    add_custom_target("${variant_name}-All-Languages"
        DEPENDS ${ALL_VARIANT_HEXES}
    )
    add_dependencies(ALL_FIRMWARE "${variant_name}-All-Languages")
  endif()
endfunction()


if(CMAKE_CROSSCOMPILING)



  file(GLOB FW_VARIANTS RELATIVE ${PROJECT_SOURCE_DIR}/Firmware/variants ${PROJECT_SOURCE_DIR}/Firmware/variants/*.h)
  foreach(THIS_VAR IN LISTS FW_VARIANTS)
    string(REPLACE ".h" "" TRIMMED_NAME "${THIS_VAR}")
    message("Variant added: ${TRIMMED_NAME}")
    fw_add_variant(${TRIMMED_NAME})

  endforeach(THIS_VAR IN LISTS FW_VARIANTS)

endif()

if(NOT CMAKE_CROSSCOMPILING)
  # do not build the firmware by default (tests are the focus if not crosscompiling)
  project(cmake_test)

  # Prepare "Catch" library for other executables
  set(CATCH_INCLUDE_DIR Catch2)
  add_library(Catch INTERFACE)
  target_include_directories(Catch INTERFACE ${CATCH_INCLUDE_DIR})

  # Make test executable
  set(TEST_SOURCES
    Tests/tests.cpp
    Tests/Example_test.cpp
    Tests/Timer_test.cpp
    Tests/AutoDeplete_test.cpp
    Tests/PrusaStatistics_test.cpp
    Firmware/Timer.cpp
    Firmware/AutoDeplete.cpp
  )
  add_executable(tests ${TEST_SOURCES})
  target_include_directories(tests PRIVATE Tests)
  target_link_libraries(tests Catch)

endif()