fix(cuda): use correct hwaccel arg, add configuration via env in entrypoint (#89)

Author: fivethreeo

Dockerfile.cuda

@@ -140,6 +140,11 @@ ENV HTTP_PORT 80
 ENV HTTPS_PORT 443
 ENV RTMP_PORT 1935
 
+# Set default options.
+ENV SINGLE_STREAM ""
+ENV MAX_MUXING_QUEUE_SIZE ""
+ENV ANALYZEDURATION ""
+
 RUN apt update && apt install -y --no-install-recommends \
   ca-certificates \
   curl \
@@ -158,13 +163,13 @@ COPY --from=build-ffmpeg /usr/lib/x86_64-linux-gnu/libfdk-aac.so.1 /usr/lib/x86_
 
 # Add NGINX path, config and static files.
 ENV PATH "${PATH}:/usr/local/nginx/sbin"
-ADD nginx-cuda.conf /etc/nginx/nginx.conf.template
 RUN mkdir -p /opt/data && mkdir /www
+ADD nginx-cuda.conf /etc/nginx/nginx.conf.template
+ADD entrypoint.cuda.sh /opt/entrypoint.sh
+RUN chmod gu+x /opt/entrypoint.sh
 ADD static /www/static
 
 EXPOSE 1935
 EXPOSE 80
 
-CMD envsubst "$(env | sed -e 's/=.*//' -e 's/^/\$/g')" < \
-  /etc/nginx/nginx.conf.template > /etc/nginx/nginx.conf && \
-  nginx
+CMD /opt/entrypoint.sh

entrypoint.cuda.sh

@@ -0,0 +1,37 @@
+#!/usr/bin/env bash
+set -e
+
+if [ ! -v "${MAX_MUXING_QUEUE_SIZE}" ]; then
+  MAX_MUXING_QUEUE_SIZE_ARG="-max_muxing_queue_size ${MAX_MUXING_QUEUE_SIZE} "
+fi
+
+if [ ! -v "${ANALYZEDURATION}" ]; then
+  ANALYZEDURATION_ARG="-analyzeduration ${ANALYZEDURATION} "
+fi
+
+quality1=('480' '256k' '64k' 'low' '448000')
+quality2=('720' '768k' '128k' 'mid' '448000')
+quality3=('960' '1240k' '128k' 'high' '1152000')
+quality4=('1280' '1920k' '128k' 'hd720' '2048000')
+
+if [ -v ${SINGLE_STREAM} ]; then
+  qualities=(quality1 quality2 quality3 quality4)
+else
+  qualities=(quality4)
+fi
+
+output_execpush="/usr/local/bin/ffmpeg $ANALYZEDURATION_ARG-async 1 -vsync -1 -hwaccel cuda -hwaccel_output_format cuda -c:v h264_cuvid -i rtmp://localhost:1935/stream/\$name "
+output_hlsvariants=""
+for quality in "${qualities[@]}"; do
+  declare -n qualitylist=$quality
+  output_execpush="$output_execpush"$'\n\t\t'"-c:v h264_nvenc -c:a aac -b:v ${qualitylist[1]} -b:a ${qualitylist[2]} -vf \"scale_npp=${qualitylist[0]}:trunc(ow/a/2)*2\" -zerolatency 1 $MAX_MUXING_QUEUE_SIZE_ARG-f flv rtmp://localhost:1935/hls/\$name_${qualitylist[3]}"
+  output_hlsvariants=$'\n\t\t'"hls_variant _${qualitylist[3]} BANDWIDTH=${qualitylist[4]};"$'\n'"${output_hlsvariants}"
+done
+
+export EXECPUSH="$output_execpush"
+export HLSVARIANTS="$output_hlsvariants"
+
+envsubst "$(env | sed -e 's/=.*//' -e 's/^/\$/g')" < \
+  /etc/nginx/nginx.conf.template > /etc/nginx/nginx.conf
+
+nginx

nginx-cuda.conf

@@ -14,12 +14,7 @@ rtmp {
         application stream {
             live on;
 
-            exec ffmpeg -hwaccel cuvid -c:v h264_cuvid -i rtmp://localhost:1935/stream/$name
-              -c:a libfdk_aac -b:a 128k -c:v h264_nvenc -b:v 2500k -f flv -g 30 -r 30 -s 1280x720 -preset superfast -profile:v baseline rtmp://localhost:1935/hls/$name_720p2628kbs
-              -c:a libfdk_aac -b:a 128k -c:v h264_nvenc -b:v 1000k -f flv -g 30 -r 30 -s 854x480 -preset superfast -profile:v baseline rtmp://localhost:1935/hls/$name_480p1128kbs
-              -c:a libfdk_aac -b:a 128k -c:v h264_nvenc -b:v 750k -f flv -g 30 -r 30 -s 640x360 -preset superfast -profile:v baseline rtmp://localhost:1935/hls/$name_360p878kbs
-              -c:a libfdk_aac -b:a 128k -c:v h264_nvenc -b:v 400k -f flv -g 30 -r 30 -s 426x240 -preset superfast -profile:v baseline rtmp://localhost:1935/hls/$name_240p528kbs
-              -c:a libfdk_aac -b:a 64k -c:v h264_nvenc -b:v 200k -f flv -g 15 -r 15 -s 426x240 -preset superfast -profile:v baseline rtmp://localhost:1935/hls/$name_240p264kbs;
+            exec ${EXECPUSH};
         }
 
         application hls {
@@ -31,11 +26,7 @@ rtmp {
             hls_path /opt/data/hls;
             hls_nested on;
 
-            hls_variant _720p2628kbs BANDWIDTH=2628000,RESOLUTION=1280x720;
-            hls_variant _480p1128kbs BANDWIDTH=1128000,RESOLUTION=854x480;
-            hls_variant _360p878kbs BANDWIDTH=878000,RESOLUTION=640x360;
-            hls_variant _240p528kbs BANDWIDTH=528000,RESOLUTION=426x240;
-            hls_variant _240p264kbs BANDWIDTH=264000,RESOLUTION=426x240;
+            ${HLSVARIANTS}
         }
     }
 }
@@ -57,7 +48,7 @@ http {
 
         # Uncomment these lines to enable SSL.
         # Update the ssl paths with your own certificate and private key.
-            
+
         # listen ${HTTPS_PORT} ssl;
         # ssl_certificate     /opt/certs/example.com.crt;
         # ssl_certificate_key /opt/certs/example.com.key;