#!/bin/bash
# ping service script - sends bursts to the frame repeater and logs reply

# TODO
# [X] receive it's own packets
# [X] log one one logfile
# [X] clean start up and shut down
# [X] send a sequence of packets
# [ ] test with Pi running frame_repeater OTC

TERM_ADDR=0x1
RS=10000
DATA_BITS_PER_FRAME=256
CODE=H_${DATA_BITS_PER_FRAME}_512_4
PIRIP_PATH=${PWD}/..
NAME=ping
PIDFILE=/var/run/${NAME}.pid
LOGFILE=/var/log/${NAME}.log
PATH=${PIRIP_PATH}/tx:${PIRIP_PATH}/librtlsdr/build_rtlsdr/src:${PATH}
PATH=${PIRIP_PATH}/codec2/build_linux/misc:${PIRIP_PATH}/codec2/build_linux/src:${PATH}
PAUSE=10

function tx_burst_hackrf {
    num_bursts=$1
    pause_between_bursts=$2
    tmp=$(mktemp)
    if [ -z ${verbose+x} ]; then
        # quiet version
        freedv_data_raw_tx --source 0x1 -c --testframes 3 --burst 1 --Fs 100000 --Rs ${RS} --tone1 ${RS} --shift ${RS} -a 30000 FSK_LDPC /dev/zero - 2>/dev/null | tlininterp - - 40 -d -f > ${tmp}
        for (( i=1; i<=$num_bursts; i++ ))
        do
            printf "%d Tx burst %d from HackRF...\n" `date +%s` ${i} 1>&2
            hackrf_transfer -t ${tmp} -s 4E6 -f 143.5E6 2>/dev/null 1>/dev/null
        sleep $pause_between_bursts
        done
    else
        # verbose version
        freedv_data_raw_tx --source 0x1 -c --testframes 3 --burst 1 --Fs 100000 --Rs ${RS} --tone1 ${RS} --shift ${RS} -a 30000 FSK_LDPC /dev/zero - | tlininterp - - 40 -d -f > ${tmp}
        for (( i=1; i<=$num_bursts; i++ ))
        do
            printf "%d Tx burst %d from HackRF...\n" `date +%s` ${i} 1>&2
            hackrf_transfer -t ${tmp} -s 4E6 -f 143.5E6
        sleep $pause_between_bursts
        done
    fi
}

function start_rx {
    rtl_fsk -g 49 -f 144490000 - -a 200000 -r 10000 --code  H_256_512_4 --mask 10000 -L $1 > /dev/null &
    echo $!>${PIDFILE}
}

function stop_service {
    echo "service stopping - bye!" 1>&2
    parent=$(cat ${PIDFILE})
    kill ${parent}
    rm ${PIDFILE}
}

case "$1" in 
    start)
        ( start_rx "--filter ${TERM_ADDR}" && sleep 1 && tx_burst_hackrf $2 ${PAUSE} && stop_service) 2>>${LOGFILE} &
        ;;
    start_verbose)
        # Show all tool outputs and log output to stderr rather than logfile
        verbose=1
        start_rx "--filter ${TERM_ADDR}" && sleep 1 && tx_burst_hackrf 1 1 && stop_service
        ;;
    start_loopback)
        # Send packets from HackRF to RTLSDR on this machine (no filtering of packets)
        verbose=1
        start_rx && sleep 1 && tx_burst_hackrf 1 1 && stop_service
        ;;
    stop)
        stop_service
        ;;
    restart)
        $0 stop
        $0 start
        ;;
    status)
        if [ -e ${PIDFILE} ]; then
            echo ${NAME} is running, pid=`cat ${PIDFILE}`
        else
            echo$ {NAME} is NOT running
            exit 1
        fi
        ;;
    *)
    echo "Usage: $0 {start|stop|status|restart}"
        echo ""
        echo "start numPings - send numPing test frames to frame repeater, one every ${PAUSE} seconds, logfile is ${LOGFILE}"
        echo "start_verbose  - send single test frame in foreground, no logfile"
        echo "start_loopback - local loopback test, Tx/Rx a single packet"
esac

exit 0