rx.sh

#!/bin/bash
#
# Some automation around rx.py to help with testing

OPUS=build/src
PATH=${PATH}:${OPUS}

features_out=features_rx_out.f32

if [ $# -lt 3 ]; then
    echo "usage (write output to file):"
    echo "  ./rx.sh model rx.iqf32 out.wav [optional rx.py args]"
    echo "usage (play output with aplay):"
    echo "  ./rx.sh model rx.iqf32 - [optional rx.py args]"
    exit 1
fi
if [ ! -f $1 ]; then
    echo "can't find $1"
    exit 1
fi
if [ ! -f $2 ]; then
    echo "can't find $2"
    exit 1
fi

model=$1
input_iqf32=$2
output_speech=$3

# eat first 3 args before passing rest to inference.py in $@
shift; shift; shift

python3 ./rx.py ${model} ${input_iqf32} ${features_out} "$@"
if [ $output_speech == "-" ]; then
    tmp=$(mktemp)
    lpcnet_demo -fargan-synthesis ${features_out} ${tmp}
    aplay $tmp -r 16000 -f S16_LE
elif [ $output_speech != "/dev/null" ]; then
    tmp=$(mktemp)
    lpcnet_demo -fargan-synthesis ${features_out} ${tmp}
    sox -t .s16 -r 16000 -c 1 ${tmp} ${output_speech}
fi