install-script.sh

#!/usr/bin/env bash
if [ ! -d darknet ]; then
		git clone https://github.com/single9/darknet.git;

		if [ $? -ne 0 ]; then
				echo "Could not clone darknet repo";
				exit 1;
		fi
fi

# dive in the darknet folder and make
cd darknet

# look for exported variables for GPU and CUDNN
GPU="${DARKNET_BUILD_WITH_GPU:-0}";
CUDNN="${DARKNET_BUILD_WITH_CUDNN:-0}";
OPENCV="${DARKNET_BUILD_WITH_OPENCV:-0}";
OPENMP="${DARKNET_BUILD_WITH_OPENMP:-0}";

case "$GPU" in
		1|0);;
		*) echo "Interpreting DARKNET_BUILD_WITH_GPU=$GPU as 0"; GPU=0;;
esac

case "$CUDNN" in
		1|0);;
		*) echo "Interpreting DARKNET_BUILD_WITH_CUDNN=$CUDNN as 0"; CUDNN=0;;
esac

case "$OPENCV" in
		1|0);;
		*) echo "Interpreting DARKNET_BUILD_WITH_OPENCV=$OPENCV as 0"; OPENCV=0;;
esac

case "$OPENMP" in
		1|0);;
		*) echo "Interpreting DARKNET_BUILD_WITH_OPENMP=$OPENMP as 0"; OPENMP=0;;
esac

if [[ "$OSTYPE" == "darwin"* ]]; then
		./build.sh
else
		sed -i -e "s/LIBSO=[01]/LIBSO=1/g" ./Makefile
		sed -i -e "s/GPU=[01]/GPU=${GPU}/g" ./Makefile
		sed -i -e "s/CUDNN=[01]/CUDNN=${CUDNN}/g" ./Makefile
		sed -i -e "s/OPENCV=[01]/OPENCV=${OPENCV}/g" ./Makefile
		sed -i -e "s/OPENMP=[01]/OPENMP=${OPENMP}/g" ./Makefile

		make clean && make -j
fi

if [ $? -ne 0 ]; then
		echo "Could not compile darknet";
		exit 2;
fi

# dive out
cd ../

# if macos make .dylib symlink
if [[ "$OSTYPE" == "darwin"* ]]; then
		if [ -e libdarknet.dylib ]; then
			rm libdarknet.dylib
		fi
		ln -s darknet/libdark.dylib libdarknet.dylib
else
		ln -s darknet/libdarknet.so libdarknet.so
fi