Merge pull request #163 from xyloon/feature/ffmpeg_stream_transcoding

Adding stream transcoding in ffmpeg kernel

Author: hephaex

ffmpeg/Dockerfile

@@ -161,6 +161,10 @@ RUN set -xe && \
     apt-get clean && \
     rm -rf /usr/local/ffmpeg_sources
 
+# Stream transcoding
+COPY ./scripts/bootstrap.sh /opt/container/bootstrap.sh
+RUN  chmod +x /opt/container/bootstrap.sh
+
 # Backend.AI specifics
 COPY ./service-defs /etc/backend.ai/service-defs
 COPY ./policy.yml /etc/backend.ai/jail/policy.yml

ffmpeg/scripts/bootstrap.sh

@@ -0,0 +1,16 @@
+#!/bin/sh
+
+USER_ID=${LOCAL_USER_ID:-9001}
+USER_NAME=work
+
+export HOME="/home/$USER_NAME"
+
+su - $USER_NAME -c "touch /home/work/ffmpeglog"
+
+if (! [ -z ${INPUT_STREAM}] ) && ( ! [ -z ${OUTPUT_STREAM} ] ); then
+    echo "nohup /usr/local/bin/ffmpeg -i $INPUT_STREAM -vcodec hevc_nvenc -f rtsp $OUTPUT_STREAM >> /home/work/ffmpeglog &" >> /home/work/ffmpeglog
+    su - $USER_NAME -c "nohup /usr/local/bin/ffmpeg -i $INPUT_STREAM -vcodec hevc_nvenc -f rtsp $OUTPUT_STREAM >> /home/work/ffmpeglog &"
+else
+    echo "INPUT_STREAM: $INPUT_STREAM" >> /home/work/ffmpeglog
+    echo "OUTPUT_STREAM: $OUTPUT_STREAM" >> /home/work/ffmpeglog
+fi