set -e

TF_VERSION_UPPER_BOUND=1.14
VALID_TF_VERSION=$(python -c "import tensorflow as tf; from pkg_resources import parse_version; print(parse_version(tf.VERSION) < parse_version('$TF_VERSION_UPPER_BOUND'))")

if [ $VALID_TF_VERSION != "True" ]; then
  TF_VERSION=$(python -c "import tensorflow as tf; print(tf.VERSION)")
  echo "tensorflow version $TF_VERSION too new, must be < $TF_VERSION_UPPER_BOUND. See README.md for explanation."
  exit
fi

GIT_BRANCH=$(git branch | sed -n '/\* /s///p')
echo "*** git branch for silot is <"$GIT_BRANCH">."

# Install dps
echo "*** Installing dps on branch "$GIT_BRANCH
git clone https://github.com/e2crawfo/dps.git
cd dps
git checkout "$GIT_BRANCH"
echo "*** git branch for dps is <"$GIT_BRANCH">."
pip install -r requirements.txt
pip install -e .

# Install auto_yolo, and custom tf op "render_sprites"
echo "*** Installing auto_yolo on branch "$GIT_BRANCH
cd ../
git clone https://github.com/e2crawfo/auto_yolo.git
cd auto_yolo
git checkout "$GIT_BRANCH"
echo "*** git branch for auto_yolo is <"$GIT_BRANCH">."
pip install -r requirements.txt
pip install -e .
cd auto_yolo/tf_ops/render_sprites/
make
cd ../resampler_edge
make
cd ../../../

# Optional: install fork of SQAIR.
echo "*** Installing sqair on branch "$GIT_BRANCH
cd ../
git clone https://github.com/e2crawfo/sqair.git
cd sqair
git checkout "$GIT_BRANCH"
echo "*** git branch for sqair is <"$GIT_BRANCH">."
pip install -r requirements.txt
pip install -e .

# Install silot
cd  ../
pip install -r requirements.txt
pip install -e .