#!/bin/bash

# Copyright (C) 2019 Intel Corporation.  All rights reserved.
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

source /opt/emsdk/emsdk_env.sh

PLATFORM=$(uname -s | tr A-Z a-z)

OUT_DIR=$PWD/out
WAMRC_CMD=$PWD/../../../wamr-compiler/build/wamrc

mkdir -p jetstream
mkdir -p tsf-src
mkdir -p ${OUT_DIR}

if [[ $1 != "--no-simd" ]];then
    NATIVE_SIMD_FLAGS="-msse2 -msse3 -msse4"
    WASM_SIMD_FLAGS="-msimd128 -msse2 -msse3 -msse4"
else
    NATIVE_SIMD_FLAGS=""
    WASM_SIMD_FLAGS=""
fi

cd jetstream

echo "Download source files .."
wget -N https://browserbench.org/JetStream/wasm/gcc-loops.cpp
wget -N https://browserbench.org/JetStream/wasm/quicksort.c
wget -N https://browserbench.org/JetStream/wasm/HashSet.cpp
wget -N https://browserbench.org/JetStream/simple/float-mm.c

if [[ $? != 0 ]]; then
    exit
fi

echo "Patch source files .."
patch -p1 -N < ../jetstream.patch

echo "Build gcc-loops with g++ .."
g++ -O3 ${NATIVE_SIMD_FLAGS} -o ${OUT_DIR}/gcc-loops_native gcc-loops.cpp

echo "Build gcc-loops with em++ .."
em++ -O3 -s STANDALONE_WASM=1 ${WASM_SIMD_FLAGS} \
         -s INITIAL_MEMORY=1048576 \
         -s TOTAL_STACK=32768 \
         -s "EXPORTED_FUNCTIONS=['_main']" \
         -s ERROR_ON_UNDEFINED_SYMBOLS=0 \
         -o ${OUT_DIR}/gcc-loops.wasm gcc-loops.cpp

echo "Compile gcc-loops.wasm to gcc-loops.aot"
${WAMRC_CMD} -o ${OUT_DIR}/gcc-loops.aot ${OUT_DIR}/gcc-loops.wasm

if [[ ${PLATFORM} == "linux" ]]; then
    echo "Compile gcc-loops.wasm to gcc-loops_segue.aot"
    ${WAMRC_CMD} --enable-segue -o ${OUT_DIR}/gcc-loops_segue.aot ${OUT_DIR}/gcc-loops.wasm
fi

echo "Build quicksort with gcc .."
gcc -O3 ${NATIVE_SIMD_FLAGS} -o ${OUT_DIR}/quicksort_native quicksort.c

echo "Build quicksort with emcc .."
emcc -O3 -s STANDALONE_WASM=1 ${WASM_SIMD_FLAGS} \
         -s INITIAL_MEMORY=1048576 \
         -s TOTAL_STACK=32768 \
         -s "EXPORTED_FUNCTIONS=['_main']" \
         -o ${OUT_DIR}/quicksort.wasm quicksort.c

echo "Compile quicksort.wasm to quicksort.aot"
${WAMRC_CMD} -o ${OUT_DIR}/quicksort.aot ${OUT_DIR}/quicksort.wasm

if [[ ${PLATFORM} == "linux" ]]; then
    echo "Compile quicksort.wasm to quicksort_segue.aot"
    ${WAMRC_CMD} --enable-segue -o ${OUT_DIR}/quicksort_segue.aot ${OUT_DIR}/quicksort.wasm
fi

echo "Build HashSet with g++ .."
g++ -O3 ${NATIVE_SIMD_FLAGS} -o ${OUT_DIR}/HashSet_native HashSet.cpp \
        -lstdc++

echo "Build HashSet with em++ .."
em++ -O3 -s STANDALONE_WASM=1 ${WASM_SIMD_FLAGS} \
         -s INITIAL_MEMORY=1048576 \
         -s TOTAL_STACK=32768 \
         -s "EXPORTED_FUNCTIONS=['_main']" \
         -o ${OUT_DIR}/HashSet.wasm HashSet.cpp

echo "Compile HashSet.wasm to HashSet.aot"
${WAMRC_CMD} -o ${OUT_DIR}/HashSet.aot ${OUT_DIR}/HashSet.wasm

if [[ ${PLATFORM} == "linux" ]]; then
    echo "Compile HashSet.wasm to HashSet_segue.aot"
    ${WAMRC_CMD} --enable-segue -o ${OUT_DIR}/HashSet_segue.aot ${OUT_DIR}/HashSet.wasm
fi

echo "Build float-mm with gcc .."
gcc -O3 ${NATIVE_SIMD_FLAGS} -o ${OUT_DIR}/float-mm_native float-mm.c

echo "Build float-mm with emcc .."
emcc -O3 -s STANDALONE_WASM=1 ${WASM_SIMD_FLAGS} \
         -s INITIAL_MEMORY=1048576 \
         -s TOTAL_STACK=32768 \
         -s "EXPORTED_FUNCTIONS=['_main']" \
         -o ${OUT_DIR}/float-mm.wasm float-mm.c

echo "Compile float-mm.wasm to float-mm.aot"
${WAMRC_CMD} -o ${OUT_DIR}/float-mm.aot ${OUT_DIR}/float-mm.wasm

if [[ ${PLATFORM} == "linux" ]]; then
    echo "Compile float-mm.wasm to float-mm_segue.aot"
    ${WAMRC_CMD} --enable-segue -o ${OUT_DIR}/float-mm_segue.aot ${OUT_DIR}/float-mm.wasm
fi

cd ../tsf-src

tsf_srcs="tsf_asprintf.c tsf_buffer.c tsf_error.c tsf_reflect.c tsf_st.c \
          tsf_type.c tsf_io.c tsf_native.c tsf_generator.c tsf_st_typetable.c \
          tsf_parser.c tsf_buf_writer.c tsf_buf_reader.c tsf_primitive.c \
          tsf_type_table.c tsf_copier.c tsf_destructor.c tsf_gpc_code_gen.c \
          gpc_code_gen_util.c gpc_threaded.c gpc_intable.c gpc_instruction.c \
          gpc_program.c gpc_proto.c gpc_stack_height.c tsf_serial_in_man.c \
          tsf_serial_out_man.c tsf_type_in_map.c tsf_type_out_map.c \
          tsf_stream_file_input.c tsf_stream_file_output.c tsf_sort.c \
          tsf_version.c tsf_named_type.c tsf_io_utils.c tsf_zip_attr.c \
          tsf_zip_reader.c tsf_zip_writer.c tsf_zip_abstract.c tsf_limits.c \
          tsf_ra_type_man.c tsf_adaptive_reader.c tsf_sha1.c tsf_sha1_writer.c \
          tsf_fsdb.c tsf_fsdb_protocol.c tsf_define_helpers.c tsf_ir.c \
          tsf_ir_different.c tsf_ir_speed.c"

tsf_files="${tsf_srcs} config.h gpc_worklist.h \
           tsf_config_stub.h tsf.h tsf_internal.h tsf_region.h tsf_types.h \
           gpc.h tsf_atomics.h tsf_define_helpers.h tsf_indent.h tsf_inttypes.h \
           tsf_serial_protocol.h tsf_util.h gpc_int_common.h tsf_build_defines.h \
           tsf_format.h tsf_internal_config.h tsf_ir_different.h tsf_sha1.h \
           tsf_zip_abstract.h gpc_internal.h tsf_config.h tsf_fsdb_protocol.h \
           tsf_internal_config_stub.h tsf_ir.h tsf_st.h \
           gpc_instruction_dispatch.gen gpc_instruction_stack_effects.gen \
           gpc_instruction_to_string.gen gpc_instruction_size.gen \
           gpc_instruction_static_size.gen gpc_interpreter.gen"

echo "Download tsf source files .."
for t in ${tsf_files}
do
    wget -N "https://browserbench.org/JetStream/wasm/TSF/${t}"
    if [[ $? != 0 ]]; then
        exit
    fi
done

patch -p1 -N < ../tsf.patch

echo "Build tsf with gcc .."
gcc \
    -o ${OUT_DIR}/tsf_native -O3 ${NATIVE_SIMD_FLAGS} \
    -I. -DTSF_BUILD_SYSTEM=1 \
    ${tsf_srcs} -lm

echo "Build tsf standalone with wasi-sdk .."
/opt/wasi-sdk/bin/clang -O3 ${WASM_SIMD_FLAGS} -z stack-size=1048576 \
    -Wl,--initial-memory=52428800 \
    -Wl,--export=main \
    -Wl,--export=__heap_base,--export=__data_end \
    -I. -DTSF_BUILD_SYSTEM=1 \
    -Wl,--allow-undefined \
    -o ${OUT_DIR}/tsf.wasm \
    ${tsf_srcs}

echo "Compile tsf.wasm to tsf.aot"
${WAMRC_CMD} -o ${OUT_DIR}/tsf.aot ${OUT_DIR}/tsf.wasm

if [[ ${PLATFORM} == "linux" ]]; then
    echo "Compile tsf.wasm to tsf_segue.aot"
    ${WAMRC_CMD} --enable-segue -o ${OUT_DIR}/tsf_segue.aot ${OUT_DIR}/tsf.wasm
fi