#!/bin/sh

set -e

# TODO: Externalize to site-specific configfile
[ $# -gt 0 ] || set -- morla 5002 -- dvcam ../../content/icon_small.png

exit1() {
	echo >&2 "ERROR: $1"
	exit 1
}

while [ $# -gt 0 ]; do
	case $1 in
	  --)
		shift; break;;
	  *)
		if [ -z "$HOST" ]; then
			HOST=$1
		elif [ -z "$FIRSTPORT" ]; then
			FIRSTPORT=$1
		else
			exit1 "Too many arguments: Max. 2 about target"
		fi
		;;
	esac
	shift
done

# TODO: Externalize to site-specific configfile
[ $# -gt 0 ] || set -- dvcam ../../content/icon_small.png

while [ $# -gt 0 ]; do
	case $1 in
		alsa=*) ALSA=${1#*=}; AINPUT=$((AINPUT+1));;
		alsa) ALSA=default; AINPUT=$((AINPUT+1));;
		dvcam=*) DVCAM=${1#*=}; XINPUT=$((XINPUT+1));;
		dvcam) DVCAM=auto; XINPUT=$((XINPUT+1));;
		dc=*) IIDC=${1#*=}; VINPUT=$((VINPUT+1));;
		dc) IIDC=/dev/fw1; VINPUT=$((VINPUT+1));;
		videofile=*) VFILE=${1#*=}; VINPUT=$((VINPUT+1));;
		*.ffv1|*.yuv|*.vp8|*.vp9) VFILE=$1; VINPUT=$((VINPUT+1));;
		container=*) XFILE=${1#*=}; XINPUT=$((XINPUT+1));;
		*.avi|*.mkv|*.mov|*.mp4|*.ogg|*.ogv|*.webm) XFILE=$1; XINPUT=$((XINPUT+1));;
		*.png) LOGO=$1; WINPUT=$((WINPUT+1));;
		--) shift; break;;
		*) exit1 "Unsupported input: $1";;
	esac
	shift
done

HOST=${HOST:-127.0.0.1}
if [ "$HOST" = "$(hostname --short)" ]; then
	IP=127.0.0.1
else
	IP=$(host "$HOST" | grep -Po 'address \K\S+')
fi

[ -n "$AINPUT$VINPUT$XINPUT" ] || exit1 "Too few arguments: Min. 1 A/V source"
[ -z "$AINPUT" ] || [ -z "$VINPUT" ] || [ -z "$XINPUT" ] || exit1 "Too many arguments: Max. 2 A/V sources"
[ -z "$AINPUT" ] || [ $AINPUT -eq 1 ] || exit1 "Too many arguments: Max. 1 audio source"
[ -z "$VINPUT" ] || [ $VINPUT -eq 1 ] || exit1 "Too many arguments: Max. 1 video source"
[ -z "$WINPUT" ] || [ $WINPUT -eq 1 ] || exit1 "Too many arguments: Max. 1 watermark source"
[ -z "$XINPUT" ] || [ $XINPUT -eq 1 ] || exit1 "Too many arguments: Max. 1 multimedia source"
[ -n "$NOAUDIO" ] || [ -z "$AINPUT$XINPUT" ] || HASAUDIO=1
[ -n "$NOVIDEO" ] || [ -z "$VINPUT$XINPUT" ] || HASVIDEO=1
[ "$AINPUT$VINPUT$XINPUT" = "1" ] || TWOSOURCES=1
VSTREAMINDEX=1
[ -n "$AINPUT" ] || VSTREAMINDEX=0

FIRSTPORT=${FIRSTPORT:-5002} # even number - next 7 ports used too

ACHANNELS=2
AFRAMERATE=48000
ABITRATE=32000

HEIGHT=360
VBITRATE=256000

# FIXME: support multiple heights
HEIGHTCOUNT=1
height=$HEIGHT
MKWEBM="[v${height}webm]"
MKMPEG="[v${height}mpeg]"

[ -z "$MKWEBM" ] || VCODECCOUNT=$((VCODECCOUNT+1))
[ -z "$MKMPEG" ] || VCODECCOUNT=$((VCODECCOUNT+1))

SAVESTEM="$SAVEDIR/$(date +%Y%m%d-%H%M%S)"
${SAVEDIR:+export FFREPORT=file="$SAVESTEM.log"}

# * scale+watermark trick based on http://stackoverflow.com/a/10937357
# * routing based on http://trac.ffmpeg.org/wiki/Creating%20multiple%20outputs#Teepseudo-muxer
# * VP8 encoding based on http://www.webmproject.org/docs/encoder-parameters/#real-time-cbr-encoding-and-streaming
#   + Add 1s latency (deadline)
# * Use same RTP payload types as GStreamer
ffmpeg -hide_banner -threads auto -re \
 ${ALSA:+-f alsa -sample_rate "$AFRAMERATE" -channels "$ACHANNELS" -thread_queue_size 1024 -i "$ALSA"} \
 ${DVCAM:+-f iec61883 -thread_queue_size 64 -i $DVCAM} \
 ${XFILE:+-i "$XFILE"} \
 ${IIDC:+-f libdc1394 -video_size 640x480 -framerate 15 -thread_queue_size 256 -i "$IIDC"} \
 ${VFILE:+-i "$VFILE"} \
 ${LOGO:+-i "$LOGO"} \
 ${HASVIDEO:+-filter_complex \
 "[$VSTREAMINDEX:v]split=$HEIGHTCOUNT[s$height];
  [s$height]scale=-2:$height${WATERMARK:-[v$height]}${WATERMARK:+[bg$height]${LOGO:+;
  [bg$height][$((VSTREAMINDEX+1)):v]overlay=main_w-overlay_w-20:main_h-overlay_h-20[v$height]}};
  [v$height]split=$VCODECCOUNT$MKWEBM$MKMPEG" } \
 ${SAVEDIR:+-map '0:' ${TWOSOURCES:+-map '1:'} \
 -codec copy \
 -f segment -segment_format matroska -segment_format_options live=1:reserve_index_space=512kB \
 -segment_list "$SAVESTEM.ffconcat" -segment_list_flags live \
 -segment_time 600 -segment_atclocktime 1 -strftime 1 "$SAVESTEM-%H%M.mkv"} \
 ${MKWEBM:+\
 ${HASAUDIO:+-map '0:a' } ${HASVIDEO:+-map "[v${height}webm]" } \
 ${HASAUDIO:+\
 -codec:a libopus -ac "$ACHANNELS" -ar "$AFRAMERATE" -b:a "$ABITRATE" } \
 ${HASVIDEO:+\
 -pix_fmt yuv420p \
 -codec:v vp8 -quality realtime -deadline 1000000 -cpu-used 15 \
 -b:v "$VBITRATE" -minrate "$VBITRATE" -maxrate "$VBITRATE" \
 -undershoot-pct 95 -bufsize $((6000*VBITRATE/1000)) -rc_init_occupancy $((4000*VBITRATE/1000)) \
 -max-intra-rate 0 \
 -qmin 4 -qmax 56 } \
 -f tee \
 "${HASAUDIO:+[select=\'a\':f=rtp:payload_type=111]rtp://$IP:$FIRSTPORT?pkt_size=1200| \
  }${HASVIDEO:+[select=\'v\':f=rtp:payload_type=100]rtp://$IP:$((FIRSTPORT+2))?pkt_size=1200}" } \
 ${MKMPEG:+\
 ${HASAUDIO:+-map '0:a' } ${HASVIDEO:+-map "[v${height}mpeg]" } \
 ${HASAUDIO:+\
 -codec:a aac -strict experimental -ac "$ACHANNELS" -ar 44100 -b:a $((ACHANNELS*64))k } \
 ${HASVIDEO:+\
 -pix_fmt yuv420p \
 -codec:v libx264 -tune zerolatency -preset ultrafast \
 -maxrate "$VBITRATE" -bufsize "$((VBITRATE*2))" -crf 23 } \
 -f tee \
 "${HASAUDIO:+[select=\'a\':f=rtp:payload_type=96]rtp://$IP:$((FIRSTPORT+4))?pkt_size=1200| \
  }${HASVIDEO:+[select=\'v\':f=rtp:payload_type=97]rtp://$IP:$((FIRSTPORT+6))?pkt_size=1200}" }