summaryrefslogtreecommitdiff
path: root/bin/stream
blob: 6e8e0696f50e285263d1e1ddc576021b2c8cc433 (plain)
  1. #!/bin/sh
  2. set -eu
  3. # TODO: Externalize to site-specific configfile
  4. [ $# -gt 0 ] || set -- morla 5002 -- dvcam ../../content/icon_small.png
  5. exit1() {
  6. echo >&2 "ERROR: $1"
  7. exit 1
  8. }
  9. while [ $# -gt 0 ]; do
  10. case $1 in
  11. --)
  12. shift; break;;
  13. *)
  14. if [ -z "${HOST:-}" ]; then
  15. HOST=$1
  16. elif [ -z "${FIRSTPORT:-}" ]; then
  17. FIRSTPORT=$1
  18. else
  19. exit1 "Too many arguments: Max. 2 about target"
  20. fi
  21. ;;
  22. esac
  23. shift
  24. done
  25. # TODO: Externalize to site-specific configfile
  26. [ $# -gt 0 ] || set -- dvcam ../../content/icon_small.png
  27. AINPUT=
  28. VINPUT=
  29. WINPUT=
  30. XINPUT=
  31. while [ $# -gt 0 ]; do
  32. case $1 in
  33. alsa=*) ALSA=${1#*=}; AINPUT=$((AINPUT+1));;
  34. alsa) ALSA=default; AINPUT=$((AINPUT+1));;
  35. dvcam=*) DVCAM=${1#*=}; XINPUT=$((XINPUT+1));;
  36. dvcam) DVCAM=auto; XINPUT=$((XINPUT+1));;
  37. dc=*) IIDC=${1#*=}; VINPUT=$((VINPUT+1));;
  38. dc) IIDC=/dev/fw1; VINPUT=$((VINPUT+1));;
  39. videofile=*) VFILE=${1#*=}; VINPUT=$((VINPUT+1));;
  40. *.ffv1|*.yuv|*.vp8|*.vp9) VFILE=$1; VINPUT=$((VINPUT+1));;
  41. container=*) XFILE=${1#*=}; XINPUT=$((XINPUT+1));;
  42. *.avi|*.mkv|*.mov|*.mp4|*.ogg|*.ogv|*.webm) XFILE=$1; XINPUT=$((XINPUT+1));;
  43. *.png) LOGO=$1; WINPUT=$((WINPUT+1));;
  44. --) shift; break;;
  45. *) exit1 "Unsupported input: $1";;
  46. esac
  47. shift
  48. done
  49. HOST=${HOST:-127.0.0.1}
  50. if [ "$HOST" = "$(hostname --short)" ]; then
  51. IP=127.0.0.1
  52. else
  53. IP=$(host "$HOST" | grep -Po 'address \K\S+')
  54. fi
  55. [ -n "$AINPUT$VINPUT$XINPUT" ] || exit1 "Too few arguments: Min. 1 A/V source"
  56. [ -z "$AINPUT" ] || [ -z "$VINPUT" ] || [ -z "$XINPUT" ] || exit1 "Too many arguments: Max. 2 A/V sources"
  57. [ -z "$AINPUT" ] || [ $AINPUT -eq 1 ] || exit1 "Too many arguments: Max. 1 audio source"
  58. [ -z "$VINPUT" ] || [ $VINPUT -eq 1 ] || exit1 "Too many arguments: Max. 1 video source"
  59. [ -z "$WINPUT" ] || [ $WINPUT -eq 1 ] || exit1 "Too many arguments: Max. 1 watermark source"
  60. [ -z "$XINPUT" ] || [ $XINPUT -eq 1 ] || exit1 "Too many arguments: Max. 1 multimedia source"
  61. [ -n "${NOAUDIO:-}" ] || [ -z "$AINPUT$XINPUT" ] || HASAUDIO=1
  62. [ -n "${NOVIDEO:-}" ] || [ -z "$VINPUT$XINPUT" ] || HASVIDEO=1
  63. [ "$AINPUT$VINPUT$XINPUT" = "1" ] || TWOSOURCES=1
  64. VSTREAMINDEX=1
  65. [ -n "$AINPUT" ] || VSTREAMINDEX=0
  66. FIRSTPORT=${FIRSTPORT:-5002} # even number - next 7 ports used too
  67. ACHANNELS=2
  68. AFRAMERATE=48000
  69. ABITRATE=32000
  70. HEIGHTS_WEBM=360
  71. HEIGHTS_MPEG=360
  72. VBITRATE=256000
  73. # shellcheck disable=SC2048,SC2059
  74. echo_n() {
  75. printf "$*"
  76. }
  77. # shellcheck disable=SC2048,SC2059
  78. printf_each() {
  79. skel=$1; shift
  80. for string in $*; do
  81. printf "$skel" "$string"
  82. done
  83. }
  84. uniqwords() {
  85. echo "$@" | tr ' ' '\012' | sort -u
  86. }
  87. HEIGHTS=$(uniqwords "$HEIGHTS_WEBM $HEIGHTS_MPEG")
  88. HEIGHTCOUNT=$(echo "$HEIGHTS" | wc --words)
  89. # FIXME: support multiple heights
  90. height=$HEIGHTS
  91. VCODECCOUNT=
  92. [ -z "$HEIGHTS_WEBM" ] || VCODECCOUNT=$((VCODECCOUNT+1))
  93. [ -z "$HEIGHTS_MPEG" ] || VCODECCOUNT=$((VCODECCOUNT+1))
  94. [ -z "${SAVEDIR:-}" ] || SAVESTEM="${SAVEDIR:-}/$(date +%Y%m%d-%H%M%S)"
  95. [ -z "${SAVEDIR:-}" ] || export FFREPORT=file="$SAVESTEM.log"
  96. tee_rtp() { stream=$1; pt=$2; port=$3;
  97. echo_n "[select=\'$stream\':f=rtp:payload_type=$pt]rtp://$IP:$port?pkt_size=1200"
  98. }
  99. # * scale+watermark trick based on http://stackoverflow.com/a/10937357
  100. # * routing based on http://trac.ffmpeg.org/wiki/Creating%20multiple%20outputs#Teepseudo-muxer
  101. # * VP8 encoding based on http://www.webmproject.org/docs/encoder-parameters/#real-time-cbr-encoding-and-streaming
  102. # + Add 1s latency (deadline)
  103. # * Use same RTP payload types as GStreamer
  104. ffmpeg -hide_banner -threads auto -re \
  105. ${ALSA:+-f alsa -sample_rate "$AFRAMERATE" -channels "$ACHANNELS" -thread_queue_size 1024 -i "$ALSA"} \
  106. ${DVCAM:+-f iec61883 -thread_queue_size 64 -i $DVCAM} \
  107. ${XFILE:+-i "$XFILE"} \
  108. ${IIDC:+-f libdc1394 -video_size 640x480 -framerate 15 -thread_queue_size 256 -i "$IIDC"} \
  109. ${VFILE:+-i "$VFILE"} \
  110. ${LOGO:+-i "$LOGO"} \
  111. ${HASVIDEO:+-filter_complex \
  112. "[$VSTREAMINDEX:v]split=$HEIGHTCOUNT$(printf_each '[s%s]' "$HEIGHTS");
  113. [s$height]scale=-2:$height${WINPUT:+[bg$height]${LOGO:+;
  114. [bg$height][$((VSTREAMINDEX+1)):v]overlay=main_w-overlay_w-20:main_h-overlay_h-20}}[v$height];
  115. [v$height]split=$VCODECCOUNT$(printf_each '[v%swebm]' "$HEIGHTS_WEBM")$(printf_each '[v%smpeg]' "$HEIGHTS_MPEG")" } \
  116. ${SAVEDIR:+-map '0:' ${TWOSOURCES:+-map '1:'} \
  117. -codec copy \
  118. -f segment -segment_format matroska -segment_format_options live=1:reserve_index_space=512kB \
  119. -segment_list "$SAVESTEM.ffconcat" -segment_list_flags live \
  120. -segment_time 600 -segment_atclocktime 1 -strftime 1 "$SAVESTEM-%H%M.mkv"} \
  121. ${HEIGHTS_WEBM:+\
  122. ${HASAUDIO:+-map '0:a' }${HASVIDEO:+$(printf_each ' -map [v%swebm]' "$HEIGHTS_WEBM") } \
  123. ${HASAUDIO:+\
  124. -codec:a libopus -ac "$ACHANNELS" -ar "$AFRAMERATE" -b:a "$ABITRATE" } \
  125. ${HASVIDEO:+\
  126. -pix_fmt yuv420p \
  127. -codec:v vp8 -quality realtime -deadline 1000000 -cpu-used 15 \
  128. -b:v "$VBITRATE" -minrate "$VBITRATE" -maxrate "$VBITRATE" \
  129. -undershoot-pct 95 -bufsize $((6000*VBITRATE/1000)) -rc_init_occupancy $((4000*VBITRATE/1000)) \
  130. -max-intra-rate 0 \
  131. -qmin 4 -qmax 56 } \
  132. -f tee \
  133. "${HASAUDIO:+\
  134. $(tee_rtp a 111 "$FIRSTPORT")|\
  135. }${HASVIDEO:+\
  136. $(tee_rtp v 100 $((FIRSTPORT+2)))}" } \
  137. ${HEIGHTS_MPEG:+\
  138. ${HASAUDIO:+-map '0:a' }${HASVIDEO:+$(printf_each ' -map [v%smpeg]' "$HEIGHTS_MPEG") } \
  139. ${HASAUDIO:+\
  140. -codec:a aac -strict experimental -ac "$ACHANNELS" -ar 44100 -b:a $((ACHANNELS*64))k } \
  141. ${HASVIDEO:+\
  142. -pix_fmt yuv420p \
  143. -codec:v libx264 -tune zerolatency -preset ultrafast \
  144. -maxrate "$VBITRATE" -bufsize "$((VBITRATE*2))" -crf 23 } \
  145. -f tee \
  146. "${HASAUDIO:+\
  147. $(tee_rtp a 96 $((FIRSTPORT+4)))|\
  148. }${HASVIDEO:+\
  149. $(tee_rtp v 97 $((FIRSTPORT+6)))}" }