#!/usr/bin/env bash

#############################################################################
#
# The following builds the benchmarks under 5.6.2, 5.6.4 and Master. The
# results can then be compared to ensure an speed penalty is not inadvertently
# taken. Crypto++ 5.6.2 is significant because its the last version Wei worked
# on before turning the library over to the community.
#
# Written and placed in public domain by Jeffrey Walton.
#
# Crypto++ Library is copyrighted as a compilation and (as of version 5.6.2)
# licensed under the Boost Software License 1.0, while the individual files
# in the compilation are all public domain.
#
# See https://www.cryptopp.com/wiki/Benchmarks for more details
#
#############################################################################

# Set to suite your taste. Speed is in GiHz

if [[ -z "$CPU_FREQ" ]]; then
	if [[ ! -z "CRYPTOPP_CPU_SPEED" ]]; then
		CPU_FREQ="$CRYPTOPP_CPU_SPEED"
	else
		CPU_FREQ=2.8
	fi
fi

echo "***************************************************"
echo "Using CPU frequency of $CPU_FREQ GiHz."
echo "Please modify this script if its not correct"
echo

#############################################################################

current=$(git rev-parse --abbrev-ref HEAD 2>/dev/null)
git fetch --all &>/dev/null &>/dev/null
if [[ "$?" -ne "0" ]]; then
	echo "$PWD does not appear to be a Git repository"
	exit 1
fi

#############################################################################
# Try to find a fast option

OPT=

if [[ -z "$OPT" ]]; then
	rm -f "$TMP/adhoc.exe" &>/dev/null
	"$CXX" -DCRYPTOPP_ADHOC_MAIN -O3 adhoc.cpp -o "$TMP/adhoc.exe" &>/dev/null
	if [[ ("$?" -eq "0") ]]; then
		OPT=-O3
	fi
fi

if [[ -z "$OPT" ]]; then
	rm -f "$TMP/adhoc.exe" &>/dev/null
	"$CXX" -DCRYPTOPP_ADHOC_MAIN -xO3 adhoc.cpp -o "$TMP/adhoc.exe" &>/dev/null
	if [[ ("$?" -eq "0") ]]; then
		OPT=-xO3
	fi
fi

if [[ -z "$OPT" ]]; then
	rm -f "$TMP/adhoc.exe" &>/dev/null
	"$CXX" -DCRYPTOPP_ADHOC_MAIN -O2 adhoc.cpp -o "$TMP/adhoc.exe" &>/dev/null
	if [[ ("$?" -eq "0") ]]; then
		OPT=-O2
	fi
fi

if [[ -z "$OPT" ]]; then
	rm -f "$TMP/adhoc.exe" &>/dev/null
	"$CXX" -DCRYPTOPP_ADHOC_MAIN -xO2 adhoc.cpp -o "$TMP/adhoc.exe" &>/dev/null
	if [[ ("$?" -eq "0") ]]; then
		OPT=-xO2
	fi
fi

if [[ -z "$OPT" ]]; then
	rm -f "$TMP/adhoc.exe" &>/dev/null
	"$CXX" -DCRYPTOPP_ADHOC_MAIN -O adhoc.cpp -o "$TMP/adhoc.exe" &>/dev/null
	if [[ ("$?" -eq "0") ]]; then
		OPT=-O
	fi
fi

#############################################################################

echo "***************************************************"
echo "**************** Crypto++ 5.6.2 *******************"
echo "***************************************************"
echo

git checkout -f CRYPTOPP_5_6_2 &>/dev/null
if [[ "$?" -ne "0" ]]; then
	echo "git checkout CRYPTOPP_5_6_2 failed"
else
	rm -f *.o benchmarks.html benchmarks-562.html &>/dev/null

	CXXFLAGS="-DNDEBUG $OPT" make
	if [[ "$?" -eq "0" ]]; then
		echo "Running benchmarks for Crypto++ 5.6.2"
		./cryptest.exe b 3 "$CPU_FREQ" > benchmarks-562.html
		if [[ "$?" -ne "0" ]]; then
			rm -rf benchmarks-562.html &>/dev/null
			echo "Failed to create benchmarks for Crypto++ 5.6.2"
		fi
	else
		echo "Failed to make benchmarks for Crypto++ 5.6.2"
	fi
fi

#############################################################################

echo "***************************************************"
echo "**************** Crypto++ 5.6.4 *******************"
echo "***************************************************"
echo

git checkout -f CRYPTOPP_5_6_4 &>/dev/null
if [[ "$?" -ne "0" ]]; then
	echo "git checkout CRYPTOPP_5_6_4 failed"
else
	rm -f *.o benchmarks.html benchmarks-564.html &>/dev/null

	CXXFLAGS="-DNDEBUG $OPT" make
	if [[ "$?" -eq "0" ]]; then
		echo "Running benchmarks for Crypto++ 5.6.4"
		./cryptest.exe b 3 "$CPU_FREQ" > benchmarks-564.html
		if [[ "$?" -ne "0" ]]; then
			rm -rf benchmarks-564.html &>/dev/null
			echo "Failed to create benchmarks for Crypto++ 5.6.4"
		fi
	else
		echo "Failed to make benchmarks for Crypto++ 5.6.4"
	fi
fi

#############################################################################

echo "***************************************************"
echo "*************** Crypto++ Master *******************"
echo "***************************************************"
echo

git checkout -f master &>/dev/null
if [[ "$?" -ne "0" ]]; then
	echo "git checkout master failed"
else
	rm -f *.o benchmarks.html benchmarks-master.html &>/dev/null

	CXXFLAGS="-DNDEBUG $OPT" make
	if [[ "$?" -eq "0" ]]; then
		echo "Running benchmarks for Crypto++ Master"
		./cryptest.exe b 3 "$CPU_FREQ" > benchmarks-master.html
		if [[ "$?" -ne "0" ]]; then
			rm -rf benchmarks-master.html &>/dev/null
			echo "Failed to create benchmarks for Crypto++ Master"
		fi
	else
		echo "Failed to make benchmarks for Crypto++ Master"
	fi
fi

#############################################################################

if [[ ! -z "$current" ]]; then
	git checkout -f "$current"
fi

exit 0