#!/bin/bash

# This script will bundle all required TensorFlow headers into a single location.
# libtensorflow.so must be built first because many of the headers are generated
# automatically.

set -x
set -e

if [ -z "$1" ]; then
    echo "No argument (header installation path) supplied."
    exit 1
fi

mkdir -p $1/tensorflow/core/public
mkdir -p $1/tensorflow/core/lib/core
mkdir -p $1/tensorflow/core/lib/strings
mkdir -p $1/tensorflow/core/lib/gtl
mkdir -p $1/tensorflow/core/framework
mkdir -p $1/tensorflow/core/platform/default
mkdir -p $1/tensorflow/core/protobuf
mkdir -p $1/google/protobuf/stubs
mkdir -p $1/google/protobuf/io
mkdir -p $1/google/protobuf/compiler
mkdir -p $1/google/protobuf/util
mkdir -p $1/third_party/eigen3/unsupported/Eigen/CXX11/src/FixedPoint
mkdir -p $1/third_party/eigen3/unsupported/Eigen/CXX11/src/util
mkdir -p $1/third_party/eigen3/unsupported/Eigen/CXX11/src/Tensor
mkdir -p $1/third_party/eigen3/Eigen/src/Core/util
mkdir -p $1/third_party/eigen3/Eigen/src/Core/arch/SSE
mkdir -p $1/third_party/eigen3/Eigen/src/Core/arch/AVX
mkdir -p $1/third_party/eigen3/Eigen/src/Core/arch/CUDA
mkdir -p $1/third_party/eigen3/Eigen/src/Core/arch/Default
mkdir -p $1/third_party/eigen3/Eigen/src/Core/functors
mkdir -p $1/third_party/eigen3/Eigen/src/Core/products
mkdir -p $1/third_party/eigen3/Eigen/src/plugins

cp /tensorflow/bazel-tensorflow/tensorflow/core/public/*.h $1/tensorflow/core/public/
cp /tensorflow/bazel-tensorflow/tensorflow/core/lib/core/*.h $1/tensorflow/core/lib/core/
cp /tensorflow/bazel-genfiles/tensorflow/core/lib/core/*.h $1/tensorflow/core/lib/core/
cp /tensorflow/bazel-tensorflow/tensorflow/core/lib/strings/*.h $1/tensorflow/core/lib/strings/
cp /tensorflow/bazel-tensorflow/tensorflow/core/lib/gtl/*.h $1/tensorflow/core/lib/gtl/
cp /tensorflow/bazel-genfiles/tensorflow/core/framework/*.h $1/tensorflow/core/framework/
cp /tensorflow/bazel-tensorflow/tensorflow/core/framework/*.h $1/tensorflow/core/framework/
cp /tensorflow/bazel-tensorflow/tensorflow/core/platform/*.h $1/tensorflow/core/platform/
cp /tensorflow/bazel-tensorflow/tensorflow/core/platform/default/*.h $1/tensorflow/core/platform/default/
cp /tensorflow/bazel-genfiles/tensorflow/core/protobuf/*.h $1/tensorflow/core/protobuf/
cp /tensorflow/bazel-tensorflow/external/protobuf/src/google/protobuf/stubs/*.h $1/google/protobuf/stubs/
cp /tensorflow/bazel-tensorflow/external/protobuf/src/google/protobuf/*.h $1/google/protobuf/
cp /tensorflow/bazel-tensorflow/external/protobuf/src/google/protobuf/io/*.h $1/google/protobuf/io/
cp /tensorflow/bazel-tensorflow/external/protobuf/src/google/protobuf/compiler/*.h $1/google/protobuf/compiler/
cp /tensorflow/bazel-tensorflow/external/protobuf/src/google/protobuf/util/*.h $1/google/protobuf/util/
cp /tensorflow/tensorflow/core/framework/tensor.h $1/tensorflow/core/framework/
cp /tensorflow/bazel-tensorflow/external/eigen_archive/eigen-eigen-d02e6a705c30/unsupported/Eigen/CXX11/Tensor $1/third_party/eigen3/unsupported/Eigen/CXX11/
cp /tensorflow/third_party/eigen3/unsupported/Eigen/CXX11/FixedPoint $1/third_party/eigen3/unsupported/Eigen/CXX11/
cp /tensorflow/third_party/eigen3/unsupported/Eigen/CXX11/src/FixedPoint/*.h $1/third_party/eigen3/unsupported/Eigen/CXX11/src/FixedPoint/
cp /tensorflow/bazel-tensorflow/external/eigen_archive/eigen-eigen-d02e6a705c30/unsupported/Eigen/CXX11/src/util/*.h $1/third_party/eigen3/unsupported/Eigen/CXX11/src/util/
cp /tensorflow/bazel-tensorflow/external/eigen_archive/eigen-eigen-d02e6a705c30/unsupported/Eigen/CXX11/src/Tensor/*.h $1/third_party/eigen3/unsupported/Eigen/CXX11/src/Tensor/
cp /tensorflow/bazel-tensorflow/external/eigen_archive/eigen-eigen-d02e6a705c30/Eigen/Core $1/third_party/eigen3/Eigen/
cp /tensorflow/bazel-tensorflow/external/eigen_archive/eigen-eigen-d02e6a705c30/Eigen/src/Core/util/*.h $1/third_party/eigen3/Eigen/src/Core/util/
cp /tensorflow/bazel-tensorflow/external/eigen_archive/eigen-eigen-d02e6a705c30/Eigen/src/Core/*.h $1/third_party/eigen3/Eigen/src/Core/
cp /tensorflow/bazel-tensorflow/external/eigen_archive/eigen-eigen-d02e6a705c30/Eigen/src/Core/arch/SSE/*.h $1/third_party/eigen3/Eigen/src/Core/arch/SSE/
cp /tensorflow/bazel-tensorflow/external/eigen_archive/eigen-eigen-d02e6a705c30/Eigen/src/Core/arch/AVX/*.h $1/third_party/eigen3/Eigen/src/Core/arch/AVX/
cp /tensorflow/bazel-tensorflow/external/eigen_archive/eigen-eigen-d02e6a705c30/Eigen/src/Core/arch/CUDA/*.h $1/third_party/eigen3/Eigen/src/Core/arch/CUDA/
cp /tensorflow/bazel-tensorflow/external/eigen_archive/eigen-eigen-d02e6a705c30/Eigen/src/Core/arch/Default/*.h $1/third_party/eigen3/Eigen/src/Core/arch/Default/
cp /tensorflow/bazel-tensorflow/external/eigen_archive/eigen-eigen-d02e6a705c30/Eigen/src/Core/functors/*.h $1/third_party/eigen3/Eigen/src/Core/functors/
cp /tensorflow/bazel-tensorflow/external/eigen_archive/eigen-eigen-d02e6a705c30/Eigen/src/plugins/*.h $1/third_party/eigen3/Eigen/src/plugins/
cp /tensorflow/bazel-tensorflow/external/eigen_archive/eigen-eigen-d02e6a705c30/Eigen/src/Core/products/*.h $1/third_party/eigen3/Eigen/src/Core/products/