run x11docker as a systemd service

2026-10-18 07:35:19 +00:00 · 2024-04-12 18:48:00 +05:30
parent 0e71970bfc
commit 44e3fa1d06
6 changed files with 158 additions and 14 deletions
@@ -0,0 +1,79 @@
+#!/usr/bin/env bash
+
+
+set -ex
+
+# This script is used to run studio-headless
+
+#parse arguments from command line
+# usage: runner.sh -i input_file -v output.mp4 -o output_dir.zip -f 1000 -b path/to/studio-headless
+
+while getopts "i:o:f:c:s:b:" opt; do
+  case $opt in
+    i) INPUT_FILE=$OPTARG ;;
+    o) OUTPUT_DIR=$OPTARG ;;
+    f) FRAME_COUNT=$OPTARG ;;
+    c) STUDIO_CAMERA_SETTING=$OPTARG ;;
+    s) STUDIO_CAMERA_SPEED=$OPTARG ;;
+    b) SKYBOX=$OPTARG ;;
+    \?) echo "Invalid option -$OPTARG" >&2
+    ;;
+  esac
+done
+
+# error out if input file is not provided
+if [ -z "$INPUT_FILE" ]; then
+  echo "Input file is required"
+  exit 1
+fi
+
+# error out if output directory is not provided
+if [ -z "$OUTPUT_DIR" ]; then
+  echo "Output directory is required"
+  exit 1
+fi
+
+#source secrets.env
+source /home/ubuntu/secrets.env
+
+OUTPUT_DIR=${OUTPUT_DIR:-/tmp/studio-headless}
+FPS=${FPS:-24}
+RESOLUTION=${RESOLUTION:-1920x1080}
+FRAME_COUNT=${FRAME_COUNT:-100}
+STUDIO_HEADLESS_PATH=${STUDIO_HEADLESS_PATH:-/home/ubuntu/storyteller-bevy/target/release/studio-headless}
+MAX_FRAME_COUNT=240 # 10 seconds @ 24 fps
+STUDIO_CAMERA_SPEED=${STUDIO_CAMERA_SPEED:-1}
+SKYBOX=${SKYBOX:-ff5555}
+STUDIO_TIMEOUT_SECONDS=${STUDIO_TIMEOUT_SECONDS:-120}
+
+mkdir -p target/release/assets/
+
+cp $INPUT_FILE target/release/assets/
+
+# extract input file name without prefix
+INPUT_FILE_NAME=$(basename $INPUT_FILE)
+
+mkdir -p $OUTPUT_DIR/frames
+
+# run studio-headless
+#$STUDIO_HEADLESS_PATH bvh -o $OUTPUT_DIR/frames --frames $FRAME_COUNT $INPUT_FILE_NAME # NB(bt): commented out
+# NB: Output directory might have to end in slash
+#RUST_LOG="debug" $STUDIO_HEADLESS_PATH scene \
+#RUST_LOG="warn,bevy_gltf::loader=error" $STUDIO_HEADLESS_PATH scene \
+# NB(bt,2024-03-25): Removing the camera params since we no longer use the manual non-timeline camera 
+	#--camera-motion $STUDIO_CAMERA_SETTING \
+	#--camera-motion-speed $STUDIO_CAMERA_SPEED \
+#RUST_BACKTRACE=full RUST_LOG="warn,bevy_gltf::loader=error" $STUDIO_HEADLESS_PATH scene \
+RUST_BACKTRACE=full RUST_LOG="info" timeout --kill-after 30 $STUDIO_TIMEOUT_SECONDS $STUDIO_HEADLESS_PATH scene \
+	-o $OUTPUT_DIR/frames/ \
+	--frames $MAX_FRAME_COUNT \
+	--skybox $SKYBOX \
+	$INPUT_FILE_NAME
+
+# use ffmpeg to create a video from the images
+
+#ffmpeg -framerate $FPS -pattern_type glob -i "$OUTPUT_DIR/frames/*-img.png"  -c:v libx264 -pix_fmt yuv420p $OUTPUT_DIR/../output.mp4
+ffmpeg -framerate $FPS -pattern_type glob -i "$OUTPUT_DIR/frames/*.png"  -c:v libx264 -pix_fmt yuv420p $OUTPUT_DIR/../output.mp4
+
+# zip the images and the video
+zip -r $OUTPUT_DIR/../output.zip $OUTPUT_DIR/frames