#!/bin/bash
V="v3"

# Commit changes before running this build script

echo "Rendering GNAL ${V}"

bash ./scripts/deps.sh
bash ./scripts/license.sh

VERSION=`bash ./scripts/version.sh`
CPU=`bash ./scripts/cpu.sh`
DIST=./stl
CSG=./csg
IMG=./img
NOTES=./notes/${V}.csv
DB="./notes/renders.sqlite"
STEP=false
LOGGING=true

#"quarter_a" "quarter_b" "quarter_c" "quarter_d"
#quarter pieces not rendering properly

FILES=( 
	"spindle_bottom" 
	"spindle_top" 
	"spindle_single"
	"spindle_stacking"  
	"insert_s8" 
	"insert_16" 
	"spacer"
	"spacer_16" 
	"insert_single"
	"top" 
	"spiral" 
)
SIZES=( "50ft" "100ft" ) 


mkdir -p "${DIST}"

if [ $STEP = true ]; then
	mkdir -p "${CSG}"
fi

if [[ ! -f "${DB}" ]]; then
	cat "./notes/setup.sql" | sqlite3 "${DB}"
fi

db () {
	sqlite3 "${DB}" "${1}"
}

render_part () {
	scad="${1}"
	SIZE="${2}"
	FILE="${3}"
	stl="${DIST}/${SIZE}_${V}/gnal_${SIZE}_${FILE}.stl"
	csg="${CSG}/${SIZE}_${V}/gnal_${SIZE}_${FILE}.csg"
	png="${IMG}/gnal_${SIZE}_${V}_${FILE}.png"

	echo "${scad} - ${FILE}"

	start=`date +%s`
	if [[ "${SIZE}" == "100ft" ]]; then
		openscad --csglimit=20000000 -o "$stl" -D "PART=\"${FILE}\"" -D "FN=800" -D "DEBUG=false" "${scad}"
	else
		openscad --csglimit=20000000 -o "$stl" -D "PART=\"${FILE}\"" -D "FN=600" -D "DEBUG=false" "${scad}"
	fi
	
	end=`date +%s`
	runtime=$((end-start))

	fileSize=`wc -c < "$stl"`
	fileSize=`echo $fileSize | xargs`

	if ! [ -x "$(command -v admesh)" ]; then
		facets="-1"
		volume="-1"
		X="-1"
		Y="-1"
		Z="-1"
	else
		firstline=`head -n 1 "$stl"`
		if [[ $firstline == solid* ]]; then
			#order stl file if ascii
			python3 scripts/c14n_stl.py "$stl"
			#convert from ascii to binary
			tmpBinary=`mktemp`
			admesh -c -b "$tmpBinary" "$stl"
			newSize=`wc -c < "$tmpBinary"`
			newSize=`echo $newSize | xargs`

			if [ $newSize -lt $fileSize ]; then
				cp "$tmpBinary" "$stl"
				percent=`echo "scale=1;($newSize/$fileSize)*100" | bc`
				fileSize="${newSize}"
				echo "Binary conversion created STL file ${percent}% of original"
			else
				echo "Binary STL is larger than ASCII original, skipping conversion..."
			fi
			rm "$tmpBinary"
		fi
		ao=`admesh -c "$stl"`
		facets=`echo "$ao" | grep "Number of facets" | awk '{print $5}'`
		volume=`echo "$ao" | grep "Number of parts"  | awk '{print $8}'`

		minX=`echo "$ao" | grep "Min X" | awk '{print $4}'`
		minX=`echo "${minX//,/}"`
		maxX=`echo "$ao" | grep "Min X" | awk '{print $8}'`
		minY=`echo "$ao" | grep "Min Y" | awk '{print $4}'`
		minY=`echo "${minY//,/}"`
		maxY=`echo "$ao" | grep "Min Y" | awk '{print $8}'`
		minZ=`echo "$ao" | grep "Min Z" | awk '{print $4}'`
		minZ=`echo "${minZ//,/}"`
		maxZ=`echo "$ao" | grep "Min Z" | awk '{print $8}'`
		X=`echo "scale=5;($maxX)-($minX)" | bc`
		Y=`echo "scale=5;($maxY)-($minY)" | bc`
		Z=`echo "scale=5;($maxZ)-($minZ)" | bc`
	fi

	hash=`sha256sum "$stl" | awk '{ print $1 }'`
	commit=`git rev-parse --short HEAD`

	if [ ${LOGGING} = true ]; then
		line="${VERSION},${CPU},$stl,$hash,$fileSize,$srchash,$srcsize,$facets,$volume,$runtime,$commit"
		echo "$line" >> $NOTES
		echo "$line"
	fi

	TIME=`date '+%s'`
	QUERY="INSERT OR IGNORE INTO renders ( \
		time, \
		commit_id, \
		source, \
		model, \
		stl, \
		stl_size, \
		facets, \
		volume, \
		x, \
		y, \
		z, \
		render_time, \
		source_hash, \
		stl_hash, \
		openscad, \
		cpu \
	 ) \
	VALUES ( \
		$TIME, \
		'$commit', \
		'$scad', \
		'$FILE', \
		'$stl', \
		$fileSize, \
		$facets, \
		$volume, \
		$X, \
		$Y, \
		$Z, \
		$runtime, \
		'$srchash', \
		'$hash', \
		'$VERSION', \
		'$CPU' \
	)"
	#echo -n "${QUERY}"
	db "${QUERY}"

	if [ ${STEP} = true ] && [[ "${FILE}" == "spiral" ]]; then
		mkdir -p "${CSG}/${SIZE}_${V}/"
		start=`date +%s`
		if [[ "${SIZE}" == "100ft" ]]; then
			openscad --csglimit=20000000 -o "$csg" -D "PART=\"${FILE}\"" -D "FN=800" -D "DEBUG=false" "${scad}"
		else
			openscad --csglimit=20000000 -o "$csg" -D "PART=\"${FILE}\"" -D "FN=600" -D "DEBUG=false" "${scad}"
		fi
		end=`date +%s`
		runtime=$((end-start))
		echo "Compiling CSG took ${runtime}sec"
	fi

	echo "Rendering image of ${stl}..."

	if [[ "${FILE}" == "spiral" ]]; then
		tmp=`mktemp`
		fullPath=`realpath "${stl}"`
		data="import(\"${fullPath}\");"
		echo data > "${tmp}.scad"
		openscad -o "$png" --csglimit=20000000 --imgsize=2048,2048 --colorscheme=DeepOcean  "${tmp}.scad"
	else
		openscad -o "$png" --csglimit=20000000 --imgsize=2048,2048 --colorscheme=DeepOcean  -D "DEBUG=false" -D "PART=\"${FILE}\"" "${scad}"
	fi
}

if [[ "${1}" != "" ]]; then
	LOGGING=false
	SIZE="${1}"
	scad="./scad/${SIZE}_${V}/gnal_${SIZE}.scad"
	srchash=`sha256sum "${scad}" | awk '{ print $1 }'`
	srcsize=`wc -c < "${scad}"`
	srcsize=`echo $srcsize | xargs`

	mkdir -p "${DIST}/${SIZE}_${V}"
	if [[ "${2}" != "" ]]; then
		FILE="${2}"
		render_part "${scad}" "${SIZE}" "${FILE}"
	else 
		for FILE in "${FILES[@]}"; do
			render_part "${scad}" "${SIZE}" "${FILE}"
		done
	fi
	exit 0
fi

echo "openscad,cpu,stl,stl_hash,stl_size,source_hash,source_size,facets,volume,render_time,commit" > $NOTES

for SIZE in "${SIZES[@]}"
do
	:
	scad="./scad/${SIZE}_${V}/gnal_${SIZE}.scad"
	srchash=`sha256sum "${scad}" | awk '{ print $1 }'`
	srcsize=`wc -c < "${scad}"`
	srcsize=`echo $srcsize | xargs`
	
	mkdir -p "${DIST}/${SIZE}_${V}"

	for FILE in "${FILES[@]}"; do
		render_part "${scad}" "${SIZE}" "${FILE}"
	done

	# add license to directories for zip
	cp ./LICENSE.txt "./stl/${SIZE}_v3/"
	# zip all
	zip -x ".*" -r "./releases/gnal_${SIZE}_v3.zip" "./stl/${SIZE}_v3/"
	# tar all
	tar --exclude=".*" -czvf "./releases/gnal_${SIZE}_v3.tar.gz" "./stl/${SIZE}_v3/"
done