#!/bin/bash

#
# See CK LICENSE.txt for licensing details.
# See CK COPYRIGHT.txt for copyright details.
#
# Developer(s):
# - Leo Gordon, 2019
# - Anton Lokhmotov, 2019
#

################ Global: #############################################

ARMNN_SOURCE_DIR=$INSTALL_DIR/$PACKAGE_SUB_DIR
ARMNN_BUILD_DIR=$INSTALL_DIR/obj
ARMNN_TARGET_DIR=$INSTALL_DIR/install

TF_PB_DIR=$INSTALL_DIR/generated_tf_pb_files
ONNX_SRC_DIR=$INSTALL_DIR/onnx

function exit_if_error() {
    if [ "${?}" != "0" ]; then exit 1; fi
}

echo ""
echo "Building ArmNN '${PACKAGE_VERSION}' in '${INSTALL_DIR}'."
echo "Frontends: USE_TF='${USE_TF}'; USE_TFLITE='${USE_TFLITE}'; USE_ONNX='${USE_ONNX}'."
echo "Backends: USE_NEON='${USE_NEON}'; USE_OPENCL='${USE_OPENCL}'."
echo "Extra flags: '${EXTRA_FLAGS}'"
echo ""

################ Frontend: ###########################################

if [ "$USE_TFLITE" == "YES" ] || [ "$USE_TFLITE" == "yes" ] || [ "$USE_TFLITE" == "ON" ] || [ "$USE_TFLITE" == "on" ] || [ "$USE_TFLITE" == "1" ]
then
    CMAKE_FOR_TFLITE=" -DBUILD_TF_LITE_PARSER=1 -DTF_LITE_GENERATED_PATH=${CK_ENV_LIB_TF_SRC_SRC}/tensorflow/lite/schema -DFLATBUFFERS_ROOT=${CK_ENV_LIB_FLATBUFFERS} -DFLATBUFFERS_LIBRARY=${CK_ENV_LIB_FLATBUFFERS_LIB}/libflatbuffers.a "
else
    CMAKE_FOR_TFLITE=""
fi

if [ -n "$CMAKE_FOR_TFLITE" ] || [ "$USE_TF" == "YES" ] || [ "$USE_TF" == "yes" ] || [ "$USE_TF" == "ON" ] || [ "$USE_TF" == "on" ] || [ "$USE_TF" == "1" ]
then
    echo ""
    echo "Generating Protobuf files from Tensorflow ..."
    echo ""

    rm -rf "${TF_PB_DIR}"
    mkdir ${TF_PB_DIR}
    cd ${CK_ENV_LIB_TF_SRC_SRC}
    ${ARMNN_SOURCE_DIR}/scripts/generate_tensorflow_protobuf.sh ${TF_PB_DIR} ${CK_ENV_LIB_PROTOBUF_HOST}
    exit_if_error

    CMAKE_FOR_TF=" -DBUILD_TF_PARSER=1 -DTF_GENERATED_SOURCES=${TF_PB_DIR} "
else
    CMAKE_FOR_TF=""
fi

if [ "$USE_ONNX" == "YES" ] || [ "$USE_ONNX" == "yes" ] || [ "$USE_ONNX" == "ON" ] || [ "$USE_ONNX" == "on" ] || [ "$USE_ONNX" == "1" ]
then
    echo ""
    echo "Generating Protobuf files from ONNX ..."
    echo ""

    rm -rf "${ONNX_SRC_DIR}"
    export ONNX_ML=1            #To clone ONNX with its ML extension
    git clone ${ONNX_BRANCH} --recursive ${ONNX_SRC_URL} ${ONNX_SRC_DIR}
    unset ONNX_ML
    cd ${ONNX_SRC_DIR}
    ${CK_ENV_LIB_PROTOBUF_HOST_BIN}/protoc onnx/onnx.proto --proto_path=. --proto_path=${CK_ENV_LIB_PROTOBUF_HOST_INCLUDE} --cpp_out ${ONNX_SRC_DIR}
    exit_if_error

    CMAKE_FOR_ONNX=" -DBUILD_ONNX_PARSER=1 -DONNX_GENERATED_SOURCES=${ONNX_SRC_DIR} "
else
    CMAKE_FOR_ONNX=""
fi

################ Backend: ############################################

if [ "$USE_NEON" == "YES" ] || [ "$USE_NEON" == "yes" ] || [ "$USE_NEON" == "ON" ] || [ "$USE_NEON" == "on" ] || [ "$USE_NEON" == "1" ]
then
    CMAKE_FOR_NEON=" -DARMCOMPUTENEON=1 "
else
    CMAKE_FOR_NEON=""
fi

if [ "$USE_OPENCL" == "YES" ] || [ "$USE_OPENCL" == "yes" ] || [ "$USE_OPENCL" == "ON" ] || [ "$USE_OPENCL" == "on" ] || [ "$USE_OPENCL" == "1" ]
then
    CMAKE_FOR_OPENCL=" -DARMCOMPUTECL=1 "
else
    CMAKE_FOR_OPENCL=""
fi

################ OSX's "native" LLVM support: ########################

if [ "$CK_DLL_EXT" == ".dylib" ] && [ -n "$CK_ENV_COMPILER_LLVM_SET" ] && [ "$CK_ARMNN_BUILD_TESTS" == "" ]
then
    CK_ARMNN_BUILD_TESTS="NO"
fi

################ Default for Linuxes is "YES": #######################

if [ "$CK_ARMNN_BUILD_TESTS" == "" ]
then
    CK_ARMNN_BUILD_TESTS="YES"
fi

############################################################
echo ""
echo "=====[ Running the following cmake command line to build ArmNN: ]====="
echo ""

rm -rf "${ARMNN_BUILD_DIR}"
mkdir ${ARMNN_BUILD_DIR}
cd ${ARMNN_BUILD_DIR}

#####################################################################################
#
# When using cmake to link a project, make sure to avoid "library path stripping"
# by emptying $LIBRARY_PATH prior to running cmake:
#
#   https://cmake.org/pipermail/cmake/2011-June/044782.html
#   https://github.com/ARM-software/armnn/issues/226
#
#####################################################################################
export LIBRARY_PATH=
echo "LIBRARY_PATH === ${LIBRARY_PATH}"
echo "LD_LIBRARY_PATH === ${LD_LIBRARY_PATH}"

read -d '' CMK_CMD <<EO_CMK_CMD
cmake ${ARMNN_SOURCE_DIR} \
    -DCMAKE_C_COMPILER="${CK_CC_PATH_FOR_CMAKE}" \
    -DCMAKE_C_FLAGS="${CK_CC_FLAGS_FOR_CMAKE} ${EXTRA_FLAGS}" \
    -DCMAKE_CXX_COMPILER="${CK_CXX_PATH_FOR_CMAKE}" \
    -DCMAKE_CXX_FLAGS="${CK_CXX_FLAGS_FOR_CMAKE} ${CK_COMPILER_FLAG_CPP14} ${EXTRA_FLAGS} ${CK_CXX_COMPILER_STDLIB}" \
    -DCMAKE_AR="${CK_AR_PATH_FOR_CMAKE}" \
    -DCMAKE_RANLIB="${CK_RANLIB_PATH_FOR_CMAKE}" \
    -DCMAKE_LINKER="${CK_LD_PATH_FOR_CMAKE}" \
    -DARMCOMPUTE_ROOT="${CK_ENV_LIB_ARMCL_SRC}" \
    -DARMCOMPUTE_BUILD_DIR="${CK_ENV_LIB_ARMCL_SRC}/build" \
    -DBoost_NO_BOOST_CMAKE=TRUE \
    -DBoost_NO_SYSTEM_PATHS=ON \
    -DBOOST_ROOT="${CK_ENV_LIB_BOOST}" \
    -DPROTOBUF_ROOT="${CK_ENV_LIB_PROTOBUF_HOST}" \
    ${CMAKE_FOR_TF} ${CMAKE_FOR_TFLITE} ${CMAKE_FOR_ONNX} \
    ${CMAKE_FOR_NEON} ${CMAKE_FOR_OPENCL} \
    -DCMAKE_INSTALL_PREFIX="${ARMNN_TARGET_DIR}" \
    -DCMAKE_VERBOSE_MAKEFILE:BOOL="${CK_ARMNN_CMAKE_VERBOSE}" \
    -DBUILD_UNIT_TESTS="${CK_ARMNN_BUILD_TESTS}"
EO_CMK_CMD

# See the EXACT command we are about to run (with all interpolations that have not been specifically blocked) :
echo $CMK_CMD
echo ""

# Now run it!
eval $CMK_CMD

exit_if_error