#! /bin/bash

#
# CK installation script
#
# See CK LICENSE for licensing details.
# See CK COPYRIGHT for copyright details.
#

# PACKAGE_DIR
# INSTALL_DIR
# PYTHON_PACKAGE_NAME
# PIP_INSTALL_OPTIONS


    # This is where pip will install the modules.
    # It has its own funny structure we don't control :
    #
EXTRA_PYTHON_SITE=${INSTALL_DIR}/python_deps_site

SHORT_PYTHON_VERSION=`${CK_ENV_COMPILER_PYTHON_FILE} -c 'import sys;print(sys.version[:3])'`
export PACKAGE_LIB_DIR="${EXTRA_PYTHON_SITE}/lib/python${SHORT_PYTHON_VERSION}/site-packages"
export PACKAGE_LIB64_DIR="${EXTRA_PYTHON_SITE}/lib64/python${SHORT_PYTHON_VERSION}/site-packages"
export PYTHONPATH=$PACKAGE_LIB64_DIR:$PACKAGE_LIB_DIR:$PYTHONPATH

echo "**************************************************************"
echo ""
echo "Cleanup: removing ${EXTRA_PYTHON_SITE}"
rm -rf "${EXTRA_PYTHON_SITE}"

######################################################################################
echo "Building '${PYTHON_PACKAGE_NAME}' from source ..."
cd ${CK_ENV_MLPERF_INFERENCE_LOADGEN}

CC=$CK_CXX_FULL_PATH CFLAGS="-std=c++14" ${CK_ENV_COMPILER_PYTHON_FILE} setup.py bdist_wheel

######################################################################################
echo "Installing '${PYTHON_PACKAGE_NAME}' and its dependencies to '$PACKAGE_LIB64_DIR' or '${PACKAGE_LIB_DIR}' ..."

${CK_ENV_COMPILER_PYTHON_FILE} -m pip install `ls dist/mlperf_loadgen*.whl` --prefix=${EXTRA_PYTHON_SITE} ${PIP_INSTALL_OPTIONS} --force-reinstall

FOUND_PACKAGE_LIB_DIR=`find $EXTRA_PYTHON_SITE -name site-packages -type d | sort -r | head -1`
touch ${FOUND_PACKAGE_LIB_DIR}/__init__.py

if [ "${?}" != "0" ] ; then
  echo "Error: installation failed!"
  exit 1
fi

    # Because we have to provide a fixed name via meta.json ,
    # and $PACKAGE_LIB_DIR or $PACKAGE_LIB64_DIR depends on the Python version,
    # we solve it by creating a symbolic link with a fixed name.
    #
ln -s $FOUND_PACKAGE_LIB_DIR ${INSTALL_DIR}/build