#!/bin/bash

set -vex

sudo apt-get update
if [[ "$OPENCV_VERSION" == "3.2.0" ]]; then
	sudo apt-get -y install libopencv-dev=3.2.0*
else
	BUILD_FLAGS="
		-D BUILD_CUDA_STUBS=OFF
		-D BUILD_DOCS=OFF
		-D BUILD_EXAMPLES=OFF
		-D BUILD_IPP_IW=ON
		-D BUILD_ITT=ON
		-D BUILD_JASPER=OFF
		-D BUILD_JAVA=OFF
		-D BUILD_JPEG=OFF
		-D BUILD_OPENEXR=OFF
		-D BUILD_PERF_TESTS=OFF
		-D BUILD_PNG=OFF
		-D BUILD_PROTOBUF=ON
		-D BUILD_SHARED_LIBS=ON
		-D BUILD_TBB=OFF
		-D BUILD_TESTS=OFF
		-D BUILD_TIFF=OFF
		-D BUILD_WEBP=OFF
		-D BUILD_WITH_DEBUG_INFO=OFF
		-D BUILD_WITH_DYNAMIC_IPP=OFF
		-D BUILD_ZLIB=OFF
		-D BUILD_opencv_apps=ALL
		-D BUILD_opencv_python2=OFF
		-D BUILD_opencv_python3=OFF
		-D CMAKE_BUILD_TYPE=Release
		-D CMAKE_INSTALL_PREFIX=/usr
		-D CV_DISABLE_OPTIMIZATION=OFF
		-D ENABLE_CONFIG_VERIFICATION=OFF
		-D ENABLE_FAST_MATH=OFF
		-D CV_ENABLE_INTRINSICS=ON
		-D ENABLE_LTO=OFF
		-D ENABLE_PIC=ON
		-D ENABLE_PRECOMPILED_HEADERS=OFF
		-D INSTALL_CREATE_DISTRIB=OFF
		-D INSTALL_PYTHON_EXAMPLES=OFF
		-D INSTALL_C_EXAMPLES=OFF
		-D INSTALL_TESTS=OFF
		-D OPENCV_ENABLE_NONFREE=ON
		-D OPENCV_FORCE_3RDPARTY_BUILD=OFF
		-D OPENCV_GENERATE_PKGCONFIG=OFF
		-D PROTOBUF_UPDATE_FILES=OFF
		-D WITH_1394=ON
		-D WITH_ADE=ON
		-D WITH_ARAVIS=OFF
		-D WITH_CLP=OFF
		-D WITH_CUDA=OFF
		-D WITH_EIGEN=ON
		-D WITH_FFMPEG=ON
		-D WITH_GDAL=ON
		-D WITH_GDCM=OFF
		-D WITH_GIGEAPI=OFF
		-D WITH_GPHOTO2=ON
		-D WITH_GSTREAMER=ON
		-D WITH_GSTREAMER_0_10=OFF
		-D WITH_GTK=OFF
		-D WITH_GTK_2_X=OFF
		-D WITH_HALIDE=OFF
		-D WITH_IMGCODEC_HDcR=ON
		-D WITH_IMGCODEC_PXM=ON
		-D WITH_IMGCODEC_SUNRASTER=ON
		-D WITH_INF_ENGINE=OFF
		-D WITH_IPP=ON
		-D WITH_ITT=ON
		-D WITH_JASPER=OFF
		-D WITH_JPEG=ON
		-D WITH_LAPACK=ON
		-D WITH_LIBV4L=OFF
		-D WITH_MATLAB=OFF
		-D WITH_MFX=OFF
		-D WITH_OPENCL=OFF
		-D WITH_OPENCLAMDBLAS=OFF
		-D WITH_OPENCLAMDFFT=OFF
		-D WITH_OPENCL_SVM=OFF
		-D WITH_OPENEXR=ON
		-D WITH_OPENGL=ON
		-D WITH_OPENMP=OFF
		-D WITH_OPENNI=OFF
		-D WITH_OPENNI2=OFF
		-D WITH_OPENVX=OFF
		-D WITH_PNG=ON
		-D WITH_PROTOBUF=ON
		-D WITH_PTHREADS_PF=ON
		-D WITH_PVAPI=OFF
		-D WITH_QT=ON
		-D WITH_QUIRC=ON
		-D WITH_TBB=ON
		-D WITH_TIFF=ON
		-D WITH_UNICAP=OFF
		-D WITH_V4L=ON
		-D WITH_VA=ON
		-D WITH_VA_INTEL=ON
		-D WITH_VTK=ON
		-D WITH_WEBP=ON
		-D WITH_XIMEA=OFF
		-D WITH_XINE=OFF
		-D OPENCV_ENABLE_MEMALIGN=OFF
	"

	# runtime deps
	sudo apt-get -y install \
		libatlas3-base \
		libavcodec57 \
		libavformat57 \
		libavresample3 \
		libceres1 \
		libdc1394-22 \
		libfreetype6 \
		libgdal20 \
		libgflags2.2 \
		libgoogle-glog0v5 \
		libgphoto2-6 \
		libgstreamer-plugins-base1.0-0 \
		libharfbuzz0b \
		libhdf5-100 \
		libjpeg8 \
		liblapacke \
		liblept5 \
		libopenexr22 \
		libpng16-16 \
		libswscale4 \
		libtbb2 \
		libtesseract4 \
		libvtk7.1 \
		libwebp6 \
		libqt5core5a \
		libqt5gui5 \
		libqt5opengl5 \
		libqt5test5 \
		libqt5widgets5

	# build deps
	sudo apt-get -y install \
		build-essential \
		cmake \
		libatlas-base-dev \
		libavcodec-dev \
		libavformat-dev \
		libavresample-dev \
		libceres-dev \
		libdc1394-22-dev \
		libeigen3-dev \
		libfreetype6-dev \
		libgdal-dev \
		libgflags-dev \
		libgoogle-glog-dev \
		libgphoto2-dev \
		libgstreamer-plugins-base1.0-dev \
		libharfbuzz-dev \
		libhdf5-dev \
		libjpeg-dev \
		liblapacke-dev \
		libleptonica-dev \
		libopenexr-dev \
		libpng-dev \
		libswscale-dev \
		libtbb-dev \
		libtesseract-dev \
		libtiff-dev \
		libvtk7-dev \
		libwebp-dev \
		qtbase5-dev

	base_dir="$HOME/build/opencv/"
	mkdir -p "$base_dir"

	curl -L "https://github.com/opencv/opencv/archive/$OPENCV_VERSION.tar.gz" | tar -xzC "$base_dir"
	curl -L "https://github.com/opencv/opencv_contrib/archive/$OPENCV_VERSION.tar.gz" | tar -xzC "$base_dir"

	build_dir="$base_dir/opencv-$OPENCV_VERSION-build/"
	mkdir -p "$build_dir"

	pushd "$build_dir" > /dev/null
	cmake $BUILD_FLAGS \
		-D CMAKE_INSTALL_PREFIX=/usr \
		-D OPENCV_EXTRA_MODULES_PATH="$base_dir/opencv_contrib-$OPENCV_VERSION/modules" \
		"$base_dir/opencv-$OPENCV_VERSION"
	make -j"$(nproc)"
	sudo make -j"$(nproc)" install
	popd > /dev/null
fi

sudo apt-get install -y clang
# workaround to make clang_sys crate detect installed libclang
sudo ln -s libclang.so.1 /usr/lib/llvm-6.0/lib/libclang.so