#!/bin/sh

# Configurable build tool parameters
BAZEL_VERSION=0.19.2
NDK_VERSION=18
SDK_VERSION=28
TF_VERSION=v1.13.1
TF_PATCH=yes
TF_PATCH_DIR=patches

# Setup build tools - use global config or custom tools plugin
ANDROID_NDK_HOME=$BUILD_TOOLS/android/android-ndk-r$NDK_VERSION
ANDROID_SDK_HOME=$BUILD_TOOLS/android/android-sdk
BAZEL_HOME=$BUILD_TOOLS/bazel/bazel-$BAZEL_VERSION
VENV_HOME=$HOME/.local
# Add them to path
PATH=$BAZEL_HOME/bin:$VENV_HOME/bin:$PATH

# Create and activate Python venv
virtualenv .
virtualenv --relocatable .
. ./bin/activate

# Install pip requirements
# From REQUIRED_PACKAGES in tensorflow/tools/pip_package/setup.py
pip install -U pip six numpy wheel setuptools mock future>=0.17.1
pip install -U keras_applications==1.0.6 --no-deps
pip install -U keras_preprocessing==1.0.5 --no-deps

# Download patches
GIST_URL=https://gist.github.com/muendelezaji/47b32812a15b622f05019b0e433de9b1/raw
mkdir -p $TF_PATCH_DIR
for patch in '0001-tflite-allow-fp16-for-fp32-models.patch'; do
    curl -fsL $GIST_URL/$patch > $TF_PATCH_DIR/$patch
done

# Get build script and run it
BUILD_SCRIPT='build-tflite.sh'
curl -fsL $GIST_URL/$BUILD_SCRIPT > $BUILD_SCRIPT
chmod +x $BUILD_SCRIPT
bash -x ./$BUILD_SCRIPT

deactivate