-
Notifications
You must be signed in to change notification settings - Fork 399
/
entrypoint.cuda.sh
37 lines (29 loc) · 1.39 KB
/
entrypoint.cuda.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
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