# 
# Copyright (C) 2018 Xilinx, Inc
# Xilinx XRT setup functions
#
# Author: ryan.radjabi@xilinx.com
# 
# Licensed under the Apache License, Version 2.0 (the "License"). You may
# not use this file except in compliance with the License. A copy of the
# License is located at
# 
#     http://www.apache.org/licenses/LICENSE-2.0
# 
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
#

XRT_PATH="${SDACCEL_DIR}/Runtime/XRT_${RELEASE_VER}"

function get_install_cmd {
    xrt_inst_cmd="install"
    aws_inst_cmd="install"
    if [[ $(lsb_release -si) == "Ubuntu" ]]; then
        dpkg -s xrt
        ret=$?
        if [[ $ret == "0" ]]; then
            xrt_inst_cmd="install --reinstall"
        fi
        dpkg -s xrt-aws
        ret=$?
        if [[ $ret == "0" ]]; then
            aws_inst_cmd="install --reinstall"
        fi
    elif [[ $(lsb_release -si) == "CentOS" ]]; then
        rpm -q xrt
        ret=$?
        if [[ $ret == "0" ]]; then
            xrt_inst_cmd="reinstall"
        fi
        rpm -q xrt-aws
        ret=$?
        if [[ $ret == "0" ]]; then
            aws_inst_cmd="reinstall"
        fi
    fi
}

function build_xrt {
    info_msg "xrt-path: $XRT_PATH"
    if [ -z "$(ls -A $XRT_PATH)" ]; then
        # XRT_PATH is empty, this is the first run, so call submodule update
        git submodule update --init -- $XRT_PATH
    else
        # XRT_PATH is not empty, only call init, this allows local changes to exist in XRT 
        git submodule init -- $XRT_PATH
    fi
    info_msg "XRT Developer Flow: Building SDAccel runtime XRT..."
    sudo sh -c "cd $XRT_PATH;./src/runtime_src/tools/scripts/xrtdeps.sh;"
    ret=$?
    if [[ $ret != 0 ]]; then
        err_msg "XRT Developer Flow: Failed to install dependencies: xrtdeps.sh: {$?}"
        return $?
    fi
    if [[ $(lsb_release -si) == "CentOS" ]]; then
        scl enable devtoolset-6 "cd ${XRT_PATH}/build/; ./build.sh;"
    elif [[ $(lsb_release -si) == "Ubuntu" ]]; then
        sudo sh -c "cd ${XRT_PATH}/build/; ./build.sh;"
    fi
    ret=$?
    if [[ $ret != 0 ]]; then
        err_msg "XRT Developer Flow: Failed to build XRT: {$?}"
    fi

    return $?
}

# takes the path to RPM/DEB package as argument
function install_xrt_package {
    get_install_cmd
    if [[ $(lsb_release -si) == "CentOS" ]]; then
        sudo sh -c "cd $1; yum ${xrt_inst_cmd} -y xrt_*-xrt.rpm; yum ${aws_inst_cmd} -y xrt_*-aws.rpm;"
    elif [[ $(lsb_release -si) == "Ubuntu" ]]; then
        sudo sh -c "cd $1; apt ${xrt_inst_cmd} ./xrt_*-xrt.deb; apt ${aws_inst_cmd} ./xrt_*-aws.deb;"
    fi
    ret=$?
    if [[ $ret != 0 ]]; then
        err_msg "XRT Developer Flow: Failed to install XRT: {$?}"
    else
        info_msg "SDAccel runtime installed"
    fi
    return $?
}

function setup_runtime {
    if [ -e /opt/xilinx/xrt ]; then # Check if XRT is installed
        info_msg "XRT Install, non-dev"
        export XILINX_XRT=/opt/xilinx/xrt
        export LD_LIBRARY_PATH=$XILINX_XRT/lib:$LD_LIBRARY_PATH
        # copy libstdc++ from $XILINX_SDX/lib
        if [[ $(lsb_release -si) == "Ubuntu" ]]; then
            sudo cp $XILINX_SDX/lib/lnx64.o/Ubuntu/libstdc++.so* /opt/xilinx/xrt/lib/
        elif [[ $(lsb_release -si) == "CentOS" ]]; then
            sudo cp $XILINX_SDX/lib/lnx64.o/Default/libstdc++.so* /opt/xilinx/xrt/lib/
        else
            info_msg "Unsupported OS."
            return 1
        fi
    elif [[ $RELEASE_VER == "2017.4" ]]; then # Build 2017.4 Runtime
        info_msg "SDACCEL Install"
        cd $SDACCEL_DIR
        info_msg "Building SDAccel runtime v$RELEASE_VER"
        if ! make ec2=1 debug=1 RELEASE_VER=$RELEASE_VER; then
            err_msg "Build of SDAccel runtime v$RELEASE_VER FAILED"
            return 1
        fi
        info_msg "Installing SDAccel runtime v$RELEASE_VER"
        export INSTALL_ROOT=/opt/Xilinx/SDx/${RELEASE_VER}.rte.dyn
        if ! sudo make ec2=1 debug=1 INSTALL_ROOT=$INSTALL_ROOT SDK_DIR=$SDK_DIR XILINX_SDX=$XILINX_SDX SDACCEL_DIR=$SDACCEL_DIR RELEASE_VER=$RELEASE_VER DSA=xilinx_aws-vu9p-f1-04261818_dynamic_5_0 install ; then
            err_msg "Install of SDAccel runtime v$RELEASE_VER FAILED"
            return 1
        fi
        info_msg "SDAccel runtime v$RELEASE_VER installed"
        cd $current_dir
    else # No XRT available and not 2017.4
        info_msg "SDAccel runtime not installed - This is required if you are running on an F1 instance."
        # Placeholder for code to download pre-compiled RPM/DEB package and remove above message
        # install_xrt_package <Path to RPM/DEB>
    fi
}