configure

#!/usr/bin/env bash

# A hach to restart the script with Bash
if [ -z "$BASH" ]; then exec bash "$0" "$@"; fi

set -eu

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" >/dev/null 2>&1 && pwd -L | sed 's://:/:g')"
PRIMME_VERSION=3.1.1
WORKDIR="${SCRIPT_DIR}/third_party/build"
export PREFIX="${SCRIPT_DIR}/third_party/primme"
INTERFACE=32
BLAS_SUFFIX=""

get_num_procs() {
  if [[ "$OSTYPE" == "darwin"* ]]; then
    sysctl -n hw.logicalcpu
  else
    nproc
  fi
}

get_ghc_version() {
  ghc --version | tr ' ' '\0' | tail -z -n 1
}

get_blas_target() {
  echo "blas${BLAS_SUFFIX}"
}

get_lapack_target() {
  echo "lapack${BLAS_SUFFIX}"
}

download_and_patch() {
  mkdir -vp "$WORKDIR"
  pushd "$WORKDIR"
  if [[ ! -d "primme" ]]; then
    git clone "https://github.com/primme/primme.git"
  fi
  pushd primme
  git pull
  git checkout eloy/fix-large-mats
  # if [[ ! -f "v${PRIMME_VERSION}.tar.gz" ]]; then
  #   echo "Downloading primme-${PRIMME_VERSION} ..."
  #   wget --no-verbose "https://github.com/primme/primme/archive/v${PRIMME_VERSION}.tar.gz"
  # fi
  # tar xf "v${PRIMME_VERSION}.tar.gz"
  # pushd "primme-${PRIMME_VERSION}"
  if [[ "$OSTYPE" == "darwin"* ]]; then
    echo "Patching primme.h..."
    git checkout include/primme.h
    patch -u include/primme.h "${SCRIPT_DIR}/cbits/primme.h.patch"
  elif [[ "$OSTYPE" == "linux-musl"* ]]; then
    echo "Patching common.h..."
    git checkout src/include/common.h
    patch -u src/include/common.h "${SCRIPT_DIR}/cbits/common.h.patch"
  fi
  popd
  popd
}

check_dependencies() {
  [[ "$OSTYPE" == "darwin"* ]] && return
  if ! pkg-config --exists $(get_blas_target); then
    echo "Missing pkg-config dependencies: $(get_blas_target)" >&2
    exit 1
  fi
  if ! pkg-config --exists $(get_lapack_target); then
    echo "Missing pkg-config dependencies: $(get_lapack_target)" >&2
    exit 1
  fi
}

get_blas_libraries() {
  pkg-config --static --libs-only-l $(get_lapack_target) $(get_blas_target) | tr ' ' '\n' | sed 's/^-l/    /'
  # echo "    mkl_rt"
  # echo "    gomp"
  # echo "    m"
}

get_blas_cflags() {
  pkg-config --cflags $(get_lapack_target) $(get_blas_target)
  # echo "-I/vol/opt/intelcompilers/intel-2019u5/mkl/include"
}

get_blas_lib_dirs() {
  pkg-config --static --libs-only-L $(get_lapack_target) $(get_blas_target) | tr ' ' '\n' | sed 's/^-L/    /'
  # echo "    /vol/opt/intelcompilers/intel-2019u5/mkl/lib/intel64"
}

build_static_lib() {
  check_dependencies
  export CFLAGS="-O3 -march=nocona -mtune=haswell -fPIC -DNDEBUG -DPRIMME_BLASINT_SIZE=${INTERFACE} -DPRIMME_INT_SIZE=64"
  [[ "$OSTYPE" != "darwin"* ]] && export CFLAGS="$CFLAGS $(get_blas_cflags)"
  export FFLAGS="-fno-second-underscore -O3 -march=nocona -mtune=haswell"
  if [[ "$OSTYPE" == "darwin"* ]]; then
    export LDFLAGS="-framework Accelerate"
  else
    export LDFLAGS=
  fi
  export PRIMME_WITH_HALF=no PRIMME_WITH_FLOAT=yes
  echo "Using CFLAGS: $CFLAGS"
  echo "Using FFLAGS: $FFLAGS"
  echo "Using LDFLAGS: $LDFLAGS"
  pushd "${WORKDIR}/primme"
  make -j$(get_num_procs) lib
  make install
  if [[ "$OSTYPE" == "darwin"* ]]; then
    rm -v "${PREFIX}"/lib/libprimme*.dylib
  else
    rm -v "${PREFIX}"/lib/libprimme.so*
  fi
  cp -v "lib/libprimme.a" "${PREFIX}/lib"
  popd
}

generate_buildinfo_osx() {
  cat <<-EOF
extra-libraries:
    primme
ghc-options:
    -DPRIMME_INT_SIZE=64
    -DPRIMME_BLASINT_SIZE=${INTERFACE}
extra-lib-dirs:
    ${PREFIX}/lib
include-dirs:
    ${PREFIX}/include
frameworks:
    Accelerate
EOF
}

generate_buildinfo_linux() {
  cat <<-EOF
include-dirs:
    ${PREFIX}/include
ghc-options:
    -DPRIMME_INT_SIZE=64
    -DPRIMME_BLASINT_SIZE=${INTERFACE}
extra-libraries:
    primme
$(get_blas_libraries)
extra-lib-dirs:
    ${PREFIX}/lib
$(get_blas_lib_dirs)
EOF
}

generate_buildinfo() {
  if [[ "$OSTYPE" == "darwin"* ]]; then
    generate_buildinfo_osx
  else
    generate_buildinfo_linux
  fi
}

print_help() {
  echo ""
  echo "Usage: ./configure [--help] [--clean]"
  echo ""
  echo "This script builds PRIMME C library and generates primme-hs.buildinfo"
  echo "file for Cabal."
  echo ""
  echo "Options:"
  echo "  --help         Display this message."
  echo "  --clean        Run make clean."
  echo ""
}

main() {
  CLEAN=0
  while [ $# -gt 0 ]; do
    key="$1"
    case $key in
    --clean)
      CLEAN=1
      shift
      ;;
    --help)
      print_help
      exit 0
      ;;
    CC=*)
      eval "export $key"
      shift
      ;;
    --blas-interface=*)
      INTERFACE="${key#*=}"
      if [[ "$INTERFACE" -eq 64 ]]; then
        BLAS_SUFFIX=64
      elif [[ "$INTERFACE" -eq 32 ]]; then
        :
      else
        echo "Error: invalid --blas-interface argument: $INTERFACE; expected either 32 or 64"
        exit 1
      fi
      shift
      ;;
    *)
      echo "Warning: argument '$1' ignored"
      shift
      ;;
    esac
  done
  if [ $CLEAN -eq 0 ]; then
    echo "Running ./configure on $OSTYPE..."
    download_and_patch
    build_static_lib
    generate_buildinfo >"primme-hs.buildinfo"
    echo "$PWD contains"
    ls -l
    echo "Generated build info:"
    cat "primme-hs.buildinfo"
  else
    if [ -d "${WORKDIR}/primme" ]; then
      cd "${WORKDIR}/primme"
      make clean
    fi
  fi
}

main "$@"