compiler.sh

#!/bin/bash

#TODO add options for clean/noclean, make/nomake, cython/nocython
#TODO include instuctions on blowing away entire package for fresh install e.g. rm -r ~/venvs/mesh/lib/python3.6/site-packages/troute/*

#set root folder of github repo (should be named t-route)
REPOROOT=`pwd`

#For each build step, you can set these to true to make it build
#or set it to anything else (or unset) to skip that step
build_mc_kernel=true
build_diffusive_tulane_kernel=true
build_reservoir_kernel=true
build_framework=true
build_routing=true
build_config=true
build_nwm=true
build_bmi=true

if [ -z "$F90" ]
then
    export F90="gfortran"
    echo "using F90=${F90}"
fi

if [ -z "$CC" ]
then
    export CC="gcc"
    echo "using CC=${CC}"
fi

# Parse command line arguments
WITH_EDITABLE=true
USE_UV=false

while [[ $# -gt 0 ]]; do
    case $1 in
        no-e)
            WITH_EDITABLE=false
            shift
            ;;
        --uv)
            USE_UV=true
            shift
            ;;
        *)
            echo "Unknown option: $1"
            echo "Usage: $0 [no-e] [--uv]"
            echo "  no-e: Install packages without editable mode"
            echo "  --uv: Use uv pip instead of pip"
            exit 1
            ;;
    esac
done

# set pip command based on UV flag
if [[ "$USE_UV" == true ]]; then
    PIP_CMD="uv pip"
    echo "Using uv pip for package installation"
else
    PIP_CMD="pip"
    echo "Using standard pip for package installation"
fi

#if you have custom static library paths, uncomment below and export them
#export LIBRARY_PATH=<paths>:$LIBRARY_PATH
#if you have custom dynamic library paths, uncomment below and export them
#export LD_LIBRARY_PATHS=<paths>:$LD_LIBRARY_PATHS

if [ -z "$NETCDF" ]
then
    export NETCDFINC="/usr/include/openmpi-$(uname -m)/"
    # set alternative NETCDF variable include path, for example for WSL
    # (Windows Subsystems for Linux).
    #
    # EXAMPLE USAGE: export NETCDFALTERNATIVE=$HOME/.conda/envs/py39/include/
    # (before ./compiler.sh)
    if [ -n "$NETCDFALTERNATIVE" ]
    then
        echo "using alternative NETCDF inc ${NETCDFALTERNATIVE}"
        export NETCDFINC=$NETCDFALTERNATIVE
    fi
else
    export NETCDFINC="${NETCDF}"
fi
echo "using NETCDFINC=${NETCDFINC}"



if [[ "$build_mc_kernel" == true ]]; then
    #building reach and resevoir kernel files .o
    cd $REPOROOT/src/kernel/muskingum/
    make clean
    # ``TROUTE_NATIVE=1 ./compiler.sh`` opts into -mcpu=native / -march=native
    # for the kernel build. Default is the portable -O3 build (no native
    # arch flags), which is required for any artifact that may run on a
    # different CPU than the build host. See src/kernel/muskingum/makefile.
    make TROUTE_NATIVE=${TROUTE_NATIVE:-0} || exit
    make install || exit

fi

if  [[ "$build_diffusive_tulane_kernel" == true ]]; then
  #building reach and resevoir kernel files .o
  cd $REPOROOT/src/kernel/diffusive/
  make clean
  make diffusive.o
  make pydiffusive.o
  make chxsec_lookuptable.o
  make pychxsec_lookuptable.o
  make install || exit
fi

if [[ "$build_reservoir_kernel" == true ]]; then
    cd $REPOROOT/src/kernel/reservoir/
    make clean
    # The makefile's `all` target lists bind_hybrid.a whose build rule
    # is commented out -- bare `make` fails. Build only the two
    # archives we actually need + install them, matching what the
    # pixi-build path does.
    make binding_lp.a bind_rfc.a || exit
    make install_lp || exit
    make install_rfc || exit
fi

if [[ "$build_framework" == true ]]; then
  cd $REPOROOT/src/troute-network
  if [[ ${WITH_EDITABLE} == true ]]; then
    USE_CYTHON=1 pip install --no-build-isolation --config-setting='--build-option=--use-cython' --editable . --config-setting='editable_mode=compat' || exit
  else
    USE_CYTHON=1 pip install --no-build-isolation --config-setting='--build-option=--use-cython' . || exit
  fi
fi

if [[ "$build_routing" == true ]]; then
    #updates troute package with the execution script
    cd $REPOROOT/src/troute-routing
    rm -rf build
    
    if [[ ${WITH_EDITABLE} == true ]]; then
        USE_CYTHON=1 $PIP_CMD install --no-build-isolation --config-setting='--build-option=--use-cython' --editable . --config-setting='editable_mode=compat' || exit
    else
        USE_CYTHON=1 $PIP_CMD install --no-build-isolation --config-setting='--build-option=--use-cython' . || exit
    fi
fi

if [[ "$build_config" == true ]]; then
    #updates troute package with the execution script
    cd $REPOROOT/src/troute-config
    if [[ ${WITH_EDITABLE} == true ]]; then
        $PIP_CMD install --editable . || exit
    else
        $PIP_CMD install . || exit
    fi
fi

if [[ "$build_nwm" == true ]]; then
    #updates troute package with the execution script
    cd $REPOROOT/src/troute-nwm
    if [[ ${WITH_EDITABLE} == true ]]; then
        $PIP_CMD install --editable . || exit
    else
        $PIP_CMD install . || exit
    fi
fi

if [[ "$build_bmi" == true ]]; then
  cd $REPOROOT/src/troute-bmi
  if [[ ${WITH_EDITABLE} == true ]]; then
    pip install --editable . || exit
  else
    pip install . || exit
  fi
fi