ftdmp-calc-interface-cadscore-for-reference

#!/bin/bash

function print_help_and_exit
{
cat >&2 << 'EOF'

'ftdmp-calc-interface-cadscore-for-reference' compares docking results produced by 'ftdmp-dock' to a reference

Options:
    --reference               string  *  reference file path
    --input-prefix            string     input file path prefix
    --input-suffix            string     input file path suffix, default is '.pdb'
    --monomer1                string     static monomer input file path
    --monomer2                string     mobile monomer input file path
    --subselect-contacts      string     query to subselect inter-chain contacts, default is '[]'
    --remap-chains            string     flag to calculate and use optimal chains remapping, default is 'false'
    --parallel                number     total number of parallel parts, default is 1
    --sbatch-parameters       string     sbatch parameters to run in parallel, default is ''
    --colnames-prefix         string     prefix for output column names, default is 'CR_'
    --adjoin                             flag to output results joined with input
    --help | -h                          flag to display help message and exit

Standard input:
    space-separated table of docking results

Standard output:
    space-separated table of scoring results
    
Example:

    cat docking_results_table.txt | ftdmp-calc-interface-cadscore-for-reference -m1 monomer1.pdb -m2 monomer2.pdb --reference --parallel 16

EOF
exit 1
}

if [ -z "$1" ]
then
	print_help_and_exit
fi

if [ -z "$FTDMPDIR" ]
then
	export FTDMPDIR="$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )"
	export PATH="${FTDMPDIR}/core/voronota/expansion_js:${FTDMPDIR}:${PATH}"
fi

command -v voronota-js-fast-iface-cadscore &> /dev/null || { echo >&2 "Error: 'voronota-js-fast-iface-cadscore' executable not in binaries path"; exit 1; }

INPUT_PREFIX=""
INPUT_SUFFIX=".pdb"
MONOMERFILE1=""
MONOMERFILE2=""
PARALLEL_PARTS="1"
SBATCH_PARAMETERS=""
SUBSELECT_CONTACTS="[]"
COLNAMES_PREFIX="CR_"
USE_MONOMERS="false"
REMAP_CHAINS="false"
ADJOIN="false"
HELP_MODE="false"

while [[ $# > 0 ]]
do
	OPTION="$1"
	OPTARG="$2"
	shift
	case $OPTION in
	--input-prefix)
		INPUT_PREFIX="$OPTARG"
		shift
		;;
	--input-suffix)
		INPUT_SUFFIX="$OPTARG"
		shift
		;;
	--monomer1)
		MONOMERFILE1="$OPTARG"
		USE_MONOMERS="true"
		shift
		;;
	--monomer2)
		MONOMERFILE2="$OPTARG"
		USE_MONOMERS="true"
		shift
		;;
	--reference)
		REFERENCEFILE="$OPTARG"
		shift
		;;
	--parallel)
		PARALLEL_PARTS="$OPTARG"
		shift
		;;
	--sbatch-parameters)
		SBATCH_PARAMETERS="$OPTARG"
		shift
		;;
	--subselect-contacts)
		SUBSELECT_CONTACTS="$OPTARG"
		shift
		;;
	--colnames-prefix)
		COLNAMES_PREFIX="$OPTARG"
		shift
		;;
	--remap-chains)
		REMAP_CHAINS="$OPTARG"
		shift
		;;
	--adjoin)
		ADJOIN="true"
		;;
	-h|--help)
		HELP_MODE="true"
		;;
	*)
		echo >&2 "Error: invalid command line option '$OPTION'"
		exit 1
		;;
	esac
done

if [ "$HELP_MODE" == "true" ]
then
	print_help_and_exit
fi

if [ "$ADJOIN" == "true" ]
then
	command -v ftdmp-join-tables &> /dev/null || { echo >&2 "Error: 'ftdmp-join-tables' executable not in binaries path"; exit 1; }
fi

if [ -z "$REFERENCEFILE" ]
then
	echo >&2 "Error: no reference input file specified"
	exit 1
fi

if [ ! -s "$REFERENCEFILE" ]
then
	echo >&2 "Error: no reference input file '$REFERENCEFILE'"
	exit 1
fi

if [ -z "$MONOMERFILE1" ] && [ -z "$MONOMERFILE2" ]
then
	USE_MONOMERS="false"
fi

if [ "$USE_MONOMERS" == "true" ]
then
	if [ -z "$MONOMERFILE1" ]
	then
		echo >&2 "Error: no static input file specified"
		exit 1
	fi
	
	if [ ! -s "$MONOMERFILE1" ]
	then
		echo >&2 "Error: no static input file '$MONOMERFILE1'"
		exit 1
	fi
	
	if [ -z "$MONOMERFILE2" ]
	then
		echo >&2 "Error: no mobile input file specified"
		exit 1
	fi
	
	if [ ! -s "$MONOMERFILE2" ]
	then
		echo >&2 "Error: no mobile input file '$MONOMERFILE2'"
		exit 1
	fi
fi

if [ -z "$PARALLEL_PARTS" ] || [ "$PARALLEL_PARTS" -ne "$PARALLEL_PARTS" ] || [ "$PARALLEL_PARTS" -lt "1" ]
then
	echo >&2 "Error: invalid number of parallel parts '$PARALLEL_PARTS', must be a positive number"
	exit 1
fi

MORE_FAST_IFACE_CADSCORE_PARAMETERS=()

if [ "$REMAP_CHAINS" == "true" ]
then
	MORE_FAST_IFACE_CADSCORE_PARAMETERS+=("--remap-chains")
fi

readonly TMPLDIR=$(mktemp -d)
trap "rm -r $TMPLDIR" EXIT

cat > "${TMPLDIR}/input"

if [ ! -s "${TMPLDIR}/input" ]
then
	echo >&2 "Error: no input data in stdin"
	exit 1
fi

{
if [ "$USE_MONOMERS" == "true" ]
then
	mkdir -p "${TMPLDIR}/docklines"
	
	cd "${TMPLDIR}/docklines"
	
	cat ../input \
	| awk -v m1="$MONOMERFILE1" -v m2="$MONOMERFILE2" '{if($1!="ID"){print "import-docking-result -static-file \"" m1 "\" -mobile-file \"" m2 "\" -generic-transform " $3 " " $4 " " $5 " " $6 " " $7 " " $8 " " $9 " " $10 " " $11 " " $12 " " $13 " " $14 " " $15 > $1 ; close($1)}}'
	
	cd - &> /dev/null
	
	find "${TMPLDIR}/docklines" -type f -not -empty \
	| voronota-js-fast-iface-cadscore --lt --target "$REFERENCEFILE" --model _list --model-input-is-script --processors "$PARALLEL_PARTS" --sbatch-parameters "$SBATCH_PARAMETERS" --subselect-contacts "$SUBSELECT_CONTACTS" "${MORE_FAST_IFACE_CADSCORE_PARAMETERS[@]}"
else
	cat "${TMPLDIR}/input" \
	| awk -v prefix="$INPUT_PREFIX" -v suffix="$INPUT_SUFFIX" '{if($1!="ID"){print prefix $1 suffix}}' \
	| voronota-js-fast-iface-cadscore --lt --target "$REFERENCEFILE" --model _list --processors "$PARALLEL_PARTS" --sbatch-parameters "$SBATCH_PARAMETERS" --subselect-contacts "$SUBSELECT_CONTACTS" "${MORE_FAST_IFACE_CADSCORE_PARAMETERS[@]}" \
	| sed "s/^\(\S\+\)${INPUT_SUFFIX}\s\+\(\S\+\)${INPUT_SUFFIX} /\1 \2 /"
fi
} \
| sed 's/^\(\S\+\)\s\+\(\S\+\)/\2 \1/' \
| sed '1 s/^\(\S\+\) /ID /' \
| sed "1 s/\s\+/ ${COLNAMES_PREFIX}/g" \
> "${TMPLDIR}/output"

if [ ! -s "${TMPLDIR}/output" ]
then
	echo >&2 "Error: no output produced"
	exit 1
fi

if [ "$ADJOIN" == "true" ]
then
	ftdmp-join-tables "${TMPLDIR}/input" "${TMPLDIR}/output"
else
	cat "${TMPLDIR}/output"
fi