summaryrefslogtreecommitdiff
path: root/bin/stream
blob: 8ecbf78a078a1dbbb9ff05ee24548286ed8ccd41 (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. # FIXME: support multiple heights
  71. HEIGHTS_WEBM=360
  72. HEIGHTS_MPEG=360
  73. VBITRATE=256000
  74. # shellcheck disable=SC2048,SC2059
  75. echo_n() {
  76. printf "$*"
  77. }
  78. # shellcheck disable=SC2048,SC2059
  79. printf_each() {
  80. skel=$1; shift
  81. for string in $*; do
  82. printf "$skel" "$string"
  83. done
  84. }
  85. uniqwords() {
  86. echo "$@" | tr ' ' '\012' | sort -u
  87. }
  88. valuedargcount() {
  89. nonempty=
  90. while [ $# -gt 0 ]; do
  91. [ -z "$1" ] || nonempty=$((nonempty+1))
  92. shift
  93. done
  94. echo_n "$nonempty"
  95. }
  96. HEIGHTS=$(uniqwords "$HEIGHTS_WEBM $HEIGHTS_MPEG")
  97. [ -z "${SAVEDIR:-}" ] || SAVESTEM="${SAVEDIR:-}/$(date +%Y%m%d-%H%M%S)"
  98. [ -z "${SAVEDIR:-}" ] || export FFREPORT=file="$SAVESTEM.log"
  99. filter_split_height() {
  100. heightcount=$(echo "$HEIGHTS" | wc --words)
  101. echo_n "[$VSTREAMINDEX:v]split=$heightcount"
  102. printf_each '[s%s]' "$HEIGHTS"
  103. }
  104. filter_scale() { outstem=${1:-v};
  105. for height in $HEIGHTS; do
  106. echo_n "[s$height]scale=-2:$height[$outstem$height]"
  107. done
  108. }
  109. filter_watermark() {
  110. for height in $HEIGHTS; do
  111. echo_n "[bg$height][$((VSTREAMINDEX+1)):v]overlay=main_w-overlay_w-20:main_h-overlay_h-20[v$height]"
  112. done
  113. }
  114. filter_split_codec() { heights_webm=$1; heights_mpeg=$2;
  115. codeccount=$(valuedargcount "$@")
  116. printf_each "[v%s]split=$codeccount$(printf_each '[v%swebm]' "$heights_webm")$(printf_each '[v%smpeg]' "$heights_mpeg")" "$HEIGHTS"
  117. }
  118. tee_rtp() { stream=$1; pt=$2; port=$3;
  119. echo_n "[select=\'$stream\':f=rtp:payload_type=$pt]rtp://$IP:$port?pkt_size=1200"
  120. }
  121. # * scale+watermark trick based on http://stackoverflow.com/a/10937357
  122. # * routing based on http://trac.ffmpeg.org/wiki/Creating%20multiple%20outputs#Teepseudo-muxer
  123. # * VP8 encoding based on http://www.webmproject.org/docs/encoder-parameters/#real-time-cbr-encoding-and-streaming
  124. # + Add 1s latency (deadline)
  125. # * Use same RTP payload types as GStreamer
  126. ffmpeg -hide_banner -threads auto -re \
  127. ${ALSA:+-f alsa -sample_rate "$AFRAMERATE" -channels "$ACHANNELS" -thread_queue_size 2048 -i "$ALSA"} \
  128. ${DVCAM:+-f iec61883 -thread_queue_size 64 -i $DVCAM} \
  129. ${XFILE:+-i "$XFILE"} \
  130. ${IIDC:+-f libdc1394 -video_size 640x480 -framerate 15 -thread_queue_size 256 -i "$IIDC"} \
  131. ${VFILE:+-i "$VFILE"} \
  132. ${LOGO:+-i "$LOGO"} \
  133. ${HASVIDEO:+-filter_complex \
  134. "$(filter_split_height);
  135. $(filter_scale "${WINPUT:+bg}")${WINPUT:+${LOGO:+;
  136. $(filter_watermark)}};
  137. $(filter_split_codec "$HEIGHTS_WEBM" "$HEIGHTS_MPEG")" } \
  138. ${SAVEDIR:+\
  139. ${DVCAM:+-map $((0${ALSA:++1})) \
  140. -codec copy -f dv "$SAVESTEM.dv" }} \
  141. ${HEIGHTS_WEBM:+\
  142. ${HASAUDIO:+-map '0:a' }${HASVIDEO:+$(printf_each ' -map [v%swebm]' "$HEIGHTS_WEBM") } \
  143. ${HASAUDIO:+\
  144. -codec:a libopus -ac "$ACHANNELS" -ar "$AFRAMERATE" -b:a "$ABITRATE" } \
  145. ${HASVIDEO:+\
  146. -pix_fmt yuv420p \
  147. -codec:v vp8 -quality realtime -deadline 1000000 -cpu-used 15 \
  148. -b:v "$VBITRATE" -minrate "$VBITRATE" -maxrate "$VBITRATE" \
  149. -undershoot-pct 95 -bufsize $((6000*VBITRATE/1000)) -rc_init_occupancy $((4000*VBITRATE/1000)) \
  150. -max-intra-rate 0 \
  151. -qmin 4 -qmax 56 } \
  152. -f tee \
  153. "${HASAUDIO:+\
  154. $(tee_rtp a 111 "$FIRSTPORT")|\
  155. }${HASVIDEO:+\
  156. $(tee_rtp v 100 $((FIRSTPORT+2)))}" } \
  157. ${HEIGHTS_MPEG:+\
  158. ${HASAUDIO:+-map '0:a' }${HASVIDEO:+$(printf_each ' -map [v%smpeg]' "$HEIGHTS_MPEG") } \
  159. ${HASAUDIO:+\
  160. -codec:a aac -strict experimental -ac "$ACHANNELS" -ar 44100 -b:a $((ACHANNELS*64))k } \
  161. ${HASVIDEO:+\
  162. -pix_fmt yuv420p \
  163. -codec:v libx264 -tune zerolatency -preset ultrafast \
  164. -maxrate "$VBITRATE" -bufsize "$((VBITRATE*2))" -crf 23 } \
  165. -f tee \
  166. "${HASAUDIO:+\
  167. $(tee_rtp a 96 $((FIRSTPORT+4)))|\
  168. }${HASVIDEO:+\
  169. $(tee_rtp v 97 $((FIRSTPORT+6)))}" }