set(IMC_BENCH_NAME "vAes" CACHE STRING "")
set(IMC_BENCH_INFINITE_LOOP 0 CACHE BOOL "")
set(IMC_PRINT_RECOVERY_MESSAGE 0 CACHE BOOL "")

set(AVAILABLE_BENCHES "vBasicMath" "vStringSearch" "vFFT" "vSha" "vCrc" "vMatMul" "vConv2d" "adc_demo" "vAes")
if(NOT IMC_BENCH_NAME IN_LIST AVAILABLE_BENCHES)
    message( FATAL_ERROR "incorrect benchmark name: ${IMC_BENCH_NAME}")
endif()

add_compile_definitions(imcBENCH_NAME=${IMC_BENCH_NAME})
if(IMC_BENCH_INFINITE_LOOP)
    add_compile_definitions(imcBENCH_INFINITE_LOOP=1)
endif()
if(IMC_PRINT_RECOVERY_MESSAGE)
    add_compile_definitions(imcPRINT_RECOVERY_MESSAGE=1)
endif()

if(IMC_BENCH_NAME MATCHES "vConv2d")
    set(BENCHMARK_SRC_FILES
    )
    set(IMC_APP_FILES
        Core/Src/benchmarks/conv2d/conv2d.c
    )
    set(IMC_LINK_OBJS
    )
endif()

if(IMC_BENCH_NAME MATCHES "adc_demo")
    set(BENCHMARK_SRC_FILES
        Core/Src/benchmarks/adc_demo/adc_demo.c
    )
    set(IMC_APP_FILES
    )
    set(IMC_LINK_OBJS
    )
endif()

if(IMC_BENCH_NAME MATCHES "vBasicMath")
    set(BENCHMARK_SRC_FILES
        Core/Src/benchmarks/basicmath/cubic.c
        Core/Src/benchmarks/basicmath/isqrt.c
        Core/Src/benchmarks/basicmath/rad2deg.c
    )
    set(IMC_APP_FILES
        Core/Src/benchmarks/basicmath/basicmath.c
    )
    set(IMC_LINK_OBJS
    )
    include_directories(
        Core/Inc/benchmarks/basicmath
    )
endif()

if(IMC_BENCH_NAME MATCHES "vCrc")
    set(BENCHMARK_SRC_FILES
    )
    set(IMC_APP_FILES
        Core/Src/benchmarks/crc/crc_32.c
    )
    set(IMC_LINK_OBJS
        ${CMAKE_SOURCE_DIR}/Core/Src/benchmarks/crc/crc_input.o
    )
    include_directories(
        Core/Inc/benchmarks/crc
    )
endif()

if(IMC_BENCH_NAME MATCHES "vFFT")
    set(BENCHMARK_SRC_FILES
        Core/Src/benchmarks/fft/fftmisc.c
        Core/Src/benchmarks/fft/fourierf.c
    )
    set(IMC_APP_FILES
        Core/Src/benchmarks/fft/fft.c
    )
    set(IMC_LINK_OBJS
    )
    include_directories(
        Core/Inc/benchmarks/fft
    )
endif()

if(IMC_BENCH_NAME MATCHES "vSha")
    set(BENCHMARK_SRC_FILES
    )
    set(IMC_APP_FILES
        Core/Src/benchmarks/sha/sha.c
    )
    set(IMC_LINK_OBJS
        ${CMAKE_SOURCE_DIR}/Core/Src/benchmarks/sha/sha_input.o
    )
    include_directories(
        Core/Inc/benchmarks/sha
    )
endif()

if(IMC_BENCH_NAME MATCHES "vMatMul")
    set(BENCHMARK_SRC_FILES
    )
    set(IMC_APP_FILES
        Core/Src/benchmarks/matmul/matmul.c
    )
    set(IMC_LINK_OBJS
    )
endif()

if(IMC_BENCH_NAME MATCHES "vStringSearch")
    set(BENCHMARK_SRC_FILES
    )
    set(IMC_APP_FILES
        Core/Src/benchmarks/stringsearch/stringsearch_local.c
    )
    set(IMC_LINK_OBJS
    )
endif()

if(IMC_BENCH_NAME MATCHES "vAes")
    set(BENCHMARK_SRC_FILES
    )
    set(IMC_APP_FILES
        Core/Src/benchmarks/aes/aes.c
    )
    set(IMC_LINK_OBJS
    )
    include_directories(
        Core/Inc/benchmarks/aes
    )
endif()

foreach(BENCH_SRC ${BENCHMARK_SRC_FILES})
    list(APPEND SRC_FILES ${BENCH_SRC})
endforeach()

function(split_loop imc_files)
    list(APPEND IMC_APP_FILES
        Core/Src/ImC/imc_extension.c
    )
    get_target_property(SRC_INCLUDES imc_freeRTOS INCLUDE_DIRECTORIES)
    foreach(dir ${SRC_INCLUDES})
        list(APPEND INCLUDE_DIRS "-I${dir}")
    endforeach()
    string(REPLACE " " ";" C_FLAGS ${CMAKE_C_FLAGS})

    set(BASIC_OPT_PASSES "mem2reg,indvars,instnamer,unify-loop-exits,lcssa,simplifycfg,loop-simplify")
    set(IMC_OPT_PASSES "imc-loop-opt")
    set(IMC_FLAGS ${IMC_BASE})
    set(LOOP_SPLIT_PLUGIN /home/ybkim/workspace/imc/loop_duplication/build/lib/libImcLoopOpt.so)

    # set(TARGET_TRIPPLE "--target=armv7em-none-eabi --sysroot=${CMAKE_SYSROOT}")

    foreach(APP_SRC ${IMC_APP_FILES})
        add_custom_command(
            OUTPUT ${PROJECT_BINARY_DIR}/${APP_SRC}.s
            DEPENDS ${APP_SRC}
            # COMMAND ${CLANG_BIN} --target=armv8m-none-eabi --sysroot=${CMAKE_SYSROOT} -g ${C_FLAGS} -O0 -Xclang -disable-O0-optnone ${INCLUDE_DIRS} -D USE_HAL_DRIVER -D STM32L552xx -c -emit-llvm ${PROJECT_SOURCE_DIR}/${APP_SRC} -o ${PROJECT_BINARY_DIR}/${APP_SRC}.bc
            COMMAND ${CMAKE_C_COMPILER} -g ${C_FLAGS} -O0 -Xclang -disable-O0-optnone ${INCLUDE_DIRS} -D USE_HAL_DRIVER -D STM32L552xx -c -emit-llvm ${PROJECT_SOURCE_DIR}/${APP_SRC} -o ${PROJECT_BINARY_DIR}/${APP_SRC}.bc
            COMMAND ${OPT_BIN} -o ${PROJECT_BINARY_DIR}/${APP_SRC}_imc.bc -load-pass-plugin=${LOOP_SPLIT_PLUGIN} -passes=${BASIC_OPT_PASSES},${IMC_OPT_PASSES} ${PROJECT_BINARY_DIR}/${APP_SRC}.bc
            COMMAND ${OPT_BIN} ${OPT} -o ${PROJECT_BINARY_DIR}/${APP_SRC}.opt.bc ${PROJECT_BINARY_DIR}/${APP_SRC}_imc.bc
            COMMAND ${LLC_BIN} ${IMC_FLAGS} ${CPU} ${OPT} -o ${PROJECT_BINARY_DIR}/${APP_SRC}.s ${PROJECT_BINARY_DIR}/${APP_SRC}.opt.bc
        )
        list(APPEND APP_OPTS ${PROJECT_BINARY_DIR}/${APP_SRC}.s)
    endforeach()
    add_library(imc_apps "${APP_OPTS}")
endfunction()