#!/bin/bash 

# 14-08-19 20:02:41 
# this script launches a batch of gencase runs that draw floating objects
# and reads out the moments of inertia out of the inertia tensor
# (diagonal elements); it does not inspect inertia products off the diagonal
# Usage note: before starting select the paths, ids and data sets in the section user-defined parts

# 23-08-19 19:54:02 
# Extended so that you can also run the flow solver and some postprocessing (surface and floating information)

# 23-08-19 19:56:28 
# This script forked from inertiaconvergence.sh; simplified so that it does not do large for loops

# 1 USER-DEFINED PARTS

## functions
  
fail () { 
	echo Execution aborted. 
	#    read -n1 -r -p "Press any key to continue..." key 
	exit 1 
}

function version {

	# return name and version of $1
	# makes some assumption on how binary files are named
	echo
	lookfor=$(basename "$1" | sed -e "s/[\._].*//" )
	echo version $(basename $1): $( "$1" -h | grep ${lookfor} | grep "[[:digit:]]\.[[:digit:]]" )

}


## functions to extract a single diagonal element of inertia tensor from xml file

Ixx() {
	cat ${diroutgencase}/${name}.xml | grep "<floating mkbound=\"${mkbound}\"" -A7 | grep v11 | cut -d\" -f2 
}

Iyy() {
	cat ${diroutgencase}/${name}.xml | grep "<floating mkbound=\"${mkbound}\"" -A7 | grep v22 | cut -d\" -f4  
}

Izz() {
	cat ${diroutgencase}/${name}.xml | grep "<floating mkbound=\"${mkbound}\"" -A7 | grep v33 | cut -d\" -f6 
}

## data sets

### select the data set you want to use ()

given=0.1200000000  # just one tricky value 
given=0.1250000000  # just one nice value 
given=0.0139238217  # a particularly evil high L/Dp ~ 72
given=0.0424280136  # a particularly evil low L/Dp ~ 24

## select simulation paths, object ids, and parameters

export name=InertiaBug04rotation
export dirbin=~/Solvers/sph/dsph/gitlab44/bin/linux/  ## v4.0.003 
export dirbin=~/Solvers/sph/dsph/github/bin/linux/  ## v4.0.010 21 june 2019
export mkbound1=10
export mkbound2=30
export mkbound3=31

### select the gencase version if ready for use

export gencase="${dirbin}/GenCase4_linux64-40027" # 11 may 2016
export gencase="${dirbin}/GenCase4_MkWord_linux64" # 4.0.092 23 march 2019
export gencase="${dirbin}/GenCase4_linux64-40109" # 4.0.109 21 june 2019
export gencase="${dirbin}/GenCase4_linux64" 

### processing

export runcpu="${dirbin}/DualSPHysics4.4CPU_linux64"
export rungpu="${dirbin}/DualSPHysics4.4_linux64"

### postprocessing

export boundaryvtk="${dirbin}/BoundaryVTK4_linux64"
export floatinginfo="${dirbin}/FloatingInfo4_linux64-40030" # 4.0.030 21 june 2019
export floatinginfo="${dirbin}/FloatingInfo4_linux64" 

# 2 DONE FOR YOU BY THE COMPUTER IN SOLITUDE

export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:${dirbin}

## remove and remake directory

export dirout=${name}
export diroutgencase=${dirout}/gencase
export diroutdata=${dirout}/data
export diroutsurfaces=${dirout}/surfaces
export diroutfloat=${dirout}/floatinginfo

if [ -e ${dirout} ]; then rm -r ${dirout}; fi

## clear old output files

rm $0*.out

## loops through the list of givens (one item nowadays)

for dx in $given; do

	echo
	echo particle spacing $dx
	sleep 0.5

	${gencase} -dp:$dx ${name}_Def ${diroutgencase}/${name} -save:all #-save:vtkbound #-createdirs:1
	if [ $? -ne 0 ] ; then fail; fi
	echo

	### read out information on floating object (extended)

	cat ${diroutgencase}/${name}.xml | grep "<inertia" -A3 -B4 | sed s/' '*\</\</g 
	echo
	
	### read out diagonal moments of inertia only

	#### first object
	mkbound=${mkbound1}
	a1=$(Ixx)
	b1=$(Iyy)
	c1=$(Izz)
#	Ixx && Iyy && Izz
	echo $dx $a1 $b1 $c1 2>&1 | tee -a $0-${mkbound}.out
	echo

	#### second object
	mkbound=${mkbound2}
	a2=$(Ixx)
	b2=$(Iyy)
	c2=$(Izz)
#	Ixx && Iyy && Izz
	echo $dx $a2 $b2 $c2 2>&1 | tee -a $0-${mkbound}.out
	echo

	#### third object
	mkbound=${mkbound3}
	a3=$(Ixx)
	b3=$(Iyy)
	c3=$(Izz)
#	Ixx && Iyy && Izz
	echo $dx $a3 $b3 $c3 2>&1 | tee -a $0-${mkbound}.out
	echo

	sleep 0.5

	### if so requested...

	if true; then

		#### starts dualsphysics

		${runcpu} -name ${diroutgencase}/${name} -dirout ${dirout} -dirdataout data -stable -svres -sv:vtk,binx
#		${runcpu} -name ${diroutgencase}/${name} -dirout ${dirout} -dirdataout data -stable -svres -sv:vtk,binx -gpu:0
		if [ $? -ne 0 ] ; then fail; fi

		#### surfaces (mkbound=10)

		${boundaryvtk} -loadvtk ${diroutgencase}/${name}_10_Dp.vtk -filexml ${diroutgencase}/${name}.xml -motiondata ${diroutdata} -savevtkdata ${diroutsurfaces}/Surfaces -onlymk:31,51,52
		if [ $? -ne 0 ] ; then fail; fi

		# motion information (mkbound=10)

		${floatinginfo} -dirin ${diroutdata} -onlymk:31,51,52 -savedata ${diroutfloat}/PrismInfo
		if [ $? -ne 0 ] ; then fail; fi

	fi

done

## shows software versions and close

version ${runcpu}
version ${rungpu}
version ${floatinginfo}
version ${gencase}
version ${boundaryvtk}

echo
echo "End of script $0"