summaryrefslogtreecommitdiff
path: root/bin/stream
blob: 0734f83fc2cfefdce5520f3558f46756d1065355 (plain)
  1. #!/bin/sh
  2. #
  3. # TODO: Set predictive GOP size for WebM (when tested that it works)
  4. set -eu
  5. # TODO: Externalize to site-specific configfile
  6. [ $# -gt 0 ] || set -- morla 5002 -- dvcam ../../content/icon_small.png
  7. exit1() {
  8. echo >&2 "ERROR: $1"
  9. exit 1
  10. }
  11. while [ $# -gt 0 ]; do
  12. case $1 in
  13. --)
  14. shift; break;;
  15. *)
  16. if [ -z "${HOST:-}" ]; then
  17. HOST=$1
  18. elif [ -z "${FIRSTPORT:-}" ]; then
  19. FIRSTPORT=$1
  20. else
  21. exit1 "Too many arguments: Max. 2 about target"
  22. fi
  23. ;;
  24. esac
  25. shift
  26. done
  27. # TODO: Externalize to site-specific configfile
  28. [ $# -gt 0 ] || set -- dvcam ../../content/icon_small.png
  29. AINPUT=
  30. VINPUT=
  31. WINPUT=
  32. XINPUT=
  33. while [ $# -gt 0 ]; do
  34. case $1 in
  35. alsa=*) ALSA=${1#*=}; AINPUT=$((AINPUT+1));;
  36. alsa) ALSA=default; AINPUT=$((AINPUT+1));;
  37. dvcam=*) DVCAM=${1#*=}; XINPUT=$((XINPUT+1));;
  38. dvcam) DVCAM=auto; XINPUT=$((XINPUT+1));;
  39. dc=*) IIDC=${1#*=}; VINPUT=$((VINPUT+1));;
  40. dc) IIDC=/dev/fw1; VINPUT=$((VINPUT+1));;
  41. videofile=*) VFILE=${1#*=}; VINPUT=$((VINPUT+1));;
  42. *.ffv1|*.yuv|*.vp8|*.vp9) VFILE=$1; VINPUT=$((VINPUT+1));;
  43. container=*) XFILE=${1#*=}; XINPUT=$((XINPUT+1));;
  44. *.avi|*.dv|*.mkv|*.mov|*.mp4|*.nut|*.ogg|*.ogv|*.webm) XFILE=$1; XINPUT=$((XINPUT+1));;
  45. *.png) LOGO=$1; WINPUT=$((WINPUT+1));;
  46. --) shift; break;;
  47. *) exit1 "Unsupported input: $1";;
  48. esac
  49. shift
  50. done
  51. HOST=${HOST:-127.0.0.1}
  52. if [ "$HOST" = "$(hostname --short)" ]; then
  53. IP=127.0.0.1
  54. else
  55. IP=$(host "$HOST" | grep -Po 'address \K\S+')
  56. fi
  57. [ -n "$AINPUT$VINPUT$XINPUT" ] || exit1 "Too few arguments: Min. 1 A/V source"
  58. [ -z "$AINPUT" ] || [ -z "$VINPUT" ] || [ -z "$XINPUT" ] || exit1 "Too many arguments: Max. 2 A/V sources"
  59. [ -z "$AINPUT" ] || [ $AINPUT -eq 1 ] || exit1 "Too many arguments: Max. 1 audio source"
  60. [ -z "$VINPUT" ] || [ $VINPUT -eq 1 ] || exit1 "Too many arguments: Max. 1 video source"
  61. [ -z "$WINPUT" ] || [ $WINPUT -eq 1 ] || exit1 "Too many arguments: Max. 1 watermark source"
  62. [ -z "$XINPUT" ] || [ $XINPUT -eq 1 ] || exit1 "Too many arguments: Max. 1 multimedia source"
  63. [ -n "${NOAUDIO:-}" ] || [ -z "$AINPUT$XINPUT" ] || HASAUDIO=1
  64. [ -n "${NOVIDEO:-}" ] || [ -z "$VINPUT$XINPUT" ] || HASVIDEO=1
  65. [ "$AINPUT$VINPUT$XINPUT" = "1" ] || TWOSOURCES=1
  66. VSTREAMINDEX=1
  67. [ -n "$AINPUT" ] || VSTREAMINDEX=0
  68. FIRSTPORT=${FIRSTPORT:-5002} # even number - next 7 ports used too
  69. ACHANNELS=1
  70. AFRAMERATE_SRC=24000
  71. AFRAMERATE_OPUS=24000
  72. AFRAMERATE_AAC=44100
  73. ABITRATE=16000
  74. # FIXME: support multiple heights
  75. HEIGHT=234
  76. HEIGHTS_WEBM="$HEIGHT"
  77. HEIGHTS_MPEG="$HEIGHT"
  78. # inspired by Apple HLS recommendations
  79. if [ $HEIGHT -le 234 ]; then VBITRATE=145000;
  80. elif [ $HEIGHT -le 270 ]; then VBITRATE=365000;
  81. elif [ $HEIGHT -le 360 ]; then VBITRATE=730000;
  82. elif [ $HEIGHT -le 432 ]; then VBITRATE=1100000;
  83. elif [ $HEIGHT -le 540 ]; then VBITRATE=2000000;
  84. elif [ $HEIGHT -le 720 ]; then VBITRATE=3000000;
  85. fi
  86. # shellcheck disable=SC2048,SC2059
  87. echo_n() {
  88. printf "$*"
  89. }
  90. # shellcheck disable=SC2048,SC2059
  91. printf_each() {
  92. skel=$1; shift
  93. for string in $*; do
  94. printf "$skel" "$string"
  95. done
  96. }
  97. uniqwords() {
  98. echo "$@" | tr ' ' '\012' | sort -u
  99. }
  100. valuedargcount() {
  101. nonempty=
  102. while [ $# -gt 0 ]; do
  103. [ -z "$1" ] || nonempty=$((nonempty+1))
  104. shift
  105. done
  106. echo_n "$nonempty"
  107. }
  108. HEIGHTS=$(uniqwords "$HEIGHTS_WEBM $HEIGHTS_MPEG")
  109. [ -z "${SAVEDIR:-}" ] || SAVESTEM="${SAVEDIR:-}/$(date +%Y%m%d-%H%M%S)"
  110. [ -z "${SAVEDIR:-}" ] || export FFREPORT=file="$SAVESTEM.log"
  111. filter_split_height() {
  112. heightcount=$(echo "$HEIGHTS" | wc --words)
  113. echo_n "[$VSTREAMINDEX:v]split=$heightcount"
  114. printf_each '[s%s]' "$HEIGHTS"
  115. }
  116. filter_scale() { outstem=${1:-v};
  117. for height in $HEIGHTS; do
  118. echo_n "[s$height]scale=-2:$height[$outstem$height]"
  119. done
  120. }
  121. filter_watermark() {
  122. for height in $HEIGHTS; do
  123. echo_n "[bg$height][$((VSTREAMINDEX+1)):v]overlay=main_w-overlay_w-20:main_h-overlay_h-20[v$height]"
  124. done
  125. }
  126. filter_split_codec() { heights_webm=$1; heights_mpeg=$2;
  127. codeccount=$(valuedargcount "$@")
  128. printf_each "[v%s]split=$codeccount$(printf_each '[v%swebm]' "$heights_webm")$(printf_each '[v%smpeg]' "$heights_mpeg")" "$HEIGHTS"
  129. }
  130. tee_rtp() { stream=$1; pt=$2; port=$3;
  131. echo_n "[select=\'$stream\':f=rtp:payload_type=$pt]rtp://$IP:$port?pkt_size=1200"
  132. }
  133. # * scale+watermark trick based on http://stackoverflow.com/a/10937357
  134. # * routing based on http://trac.ffmpeg.org/wiki/Creating%20multiple%20outputs#Teepseudo-muxer
  135. # * VP8 encoding based on http://www.webmproject.org/docs/encoder-parameters/#real-time-cbr-encoding-and-streaming
  136. # + Add 1s latency (deadline)
  137. # * Predictive GOP size preserving scene-change GOP based on https://superuser.com/a/1098329
  138. # * Use same RTP payload types as GStreamer
  139. ffmpeg -hide_banner -threads auto \
  140. ${ALSA:+-f alsa -sample_rate "$AFRAMERATE_SRC" -channels "$ACHANNELS" -thread_queue_size 2048 -i "$ALSA"} \
  141. ${DVCAM:+-f iec61883 -thread_queue_size 64 -i $DVCAM} \
  142. ${XFILE:+-re${SEEK:+ -ss $SEEK} -i "$XFILE"} \
  143. ${IIDC:+-f libdc1394 -video_size 640x480 -framerate 15 -thread_queue_size 256 -i "$IIDC"} \
  144. ${VFILE:+-re${SEEK:+ -ss $SEEK} -i "$VFILE"} \
  145. ${LOGO:+-i "$LOGO"} \
  146. ${HASVIDEO:+-filter_complex \
  147. "$(filter_split_height);
  148. $(filter_scale "${WINPUT:+bg}")${WINPUT:+${LOGO:+;
  149. $(filter_watermark)}};
  150. $(filter_split_codec "$HEIGHTS_WEBM" "$HEIGHTS_MPEG")" } \
  151. ${SAVEDIR:+\
  152. ${DVCAM:+-map $((0${ALSA:++1})) \
  153. -codec copy -f dv "$SAVESTEM.dv" }} \
  154. ${HEIGHTS_WEBM:+\
  155. ${HASAUDIO:+-map '0:a' }${HASVIDEO:+$(printf_each ' -map [v%swebm]' "$HEIGHTS_WEBM") } -shortest \
  156. ${HASAUDIO:+\
  157. -codec:a libopus -ac "$ACHANNELS" -ar "$AFRAMERATE_OPUS" -b:a "$ABITRATE" } \
  158. ${HASVIDEO:+\
  159. -pix_fmt yuv420p \
  160. -codec:v vp8 -quality realtime -deadline 1000000 -cpu-used 15 \
  161. -b:v "$VBITRATE" -minrate "$VBITRATE" -maxrate "$VBITRATE" \
  162. -undershoot-pct 95 -bufsize $((6000*VBITRATE/1000)) -rc_init_occupancy $((4000*VBITRATE/1000)) \
  163. -max-intra-rate 0 \
  164. -qmin 4 -qmax 56 } \
  165. -f tee \
  166. "${HASAUDIO:+\
  167. $(tee_rtp a 111 "$FIRSTPORT")|\
  168. }${HASVIDEO:+\
  169. $(tee_rtp v 100 $((FIRSTPORT+2)))}" } \
  170. ${HEIGHTS_MPEG:+\
  171. ${HASAUDIO:+-map '0:a' }${HASVIDEO:+$(printf_each ' -map [v%smpeg]' "$HEIGHTS_MPEG") } -shortest \
  172. ${HASAUDIO:+\
  173. -codec:a aac -strict experimental -ac "$ACHANNELS" -ar "$AFRAMERATE_AAC" -b:a $((ACHANNELS*64))k } \
  174. ${HASVIDEO:+\
  175. -pix_fmt yuv420p \
  176. -codec:v libx264 -tune zerolatency -preset ultrafast \
  177. -maxrate "$VBITRATE" -bufsize "$((VBITRATE*2))" -crf 23 } \
  178. -force_key_frames 'expr:gte(t,n_forced*2)' \
  179. -f rtp_mpegts rtp://$IP:10000 }