From c73469a0d96930194080699dbcc8d98df9f8393f Mon Sep 17 00:00:00 2001
From: Levon <lgevorgyan03@gmail.com>
Date: Tue, 16 Apr 2024 08:18:37 -0500
Subject: [PATCH] wip recc_encode

---
 .gitignore                     |  2 +-
 scripts/recc_encode.sh         | 36 ++++++++++++++++++++++++++++++++++
 scripts/recc_encode_install.sh |  3 +++
 3 files changed, 40 insertions(+), 1 deletion(-)
 create mode 100755 scripts/recc_encode.sh
 create mode 100755 scripts/recc_encode_install.sh

diff --git a/.gitignore b/.gitignore
index e6f34ce..b2af0ae 100644
--- a/.gitignore
+++ b/.gitignore
@@ -9,4 +9,4 @@ benchmark
 graphs
 rkmpp
 rkrga
-
+x265*
diff --git a/scripts/recc_encode.sh b/scripts/recc_encode.sh
new file mode 100755
index 0000000..e3c8677
--- /dev/null
+++ b/scripts/recc_encode.sh
@@ -0,0 +1,36 @@
+#!/bin/bash
+
+# this is simply my recommended encoding method.
+# do not take this as a holy grail.
+
+encode() {
+    FILENAME="$1"
+    OUTPUT_NAME=""
+    # allow optional output filename
+    if [[ -n "$2" ]];
+    then
+        OUTPUT_NAME="$2"
+    else
+        OUTPUT_NAME="${HOME}/av1_${FILENAME}"
+    fi
+    
+    echo ffmpeg -i \""$FILENAME"\" -map 0 \
+        -af '"aformat=channel_layouts=7.1|5.1|stereo|mono"' -c:a libopus $(get_bitrate_audio "$FILENAME") \
+        -c:s copy -c:v libsvtav1 -pix_fmt yuv420p10le -crf 20 -preset 3 -g 240 \
+        -svtav1-params \"tune=0:enable-overlays=1:scd=1:enable-hdr=1:fast-decode=1:enable-variance-boost=1\" \
+        \""$OUTPUT_NAME"\"
+}
+
+get_bitrate_audio() {
+    FILENAME="$1"
+    bitrate_cmd=""
+    num_streams=$(ffprobe -v error -select_streams a -show_entries stream=index -of csv=p=0 "$FILENAME" | wc -l)
+    for ((i = 0; i < num_streams; i++)); do
+        num_channels=$(ffprobe -v error -select_streams "a:$i" -show_entries stream=channels -of default=noprint_wrappers=1:nokey=1 "$FILENAME")
+        bitrate=$((num_channels * 64))
+        bitrate_cmd+="-b:a:$i ${bitrate}k "
+    done
+    echo "$bitrate_cmd"
+}
+
+encode "$@"
diff --git a/scripts/recc_encode_install.sh b/scripts/recc_encode_install.sh
new file mode 100755
index 0000000..febc10c
--- /dev/null
+++ b/scripts/recc_encode_install.sh
@@ -0,0 +1,3 @@
+#!/bin/bash
+
+sudo ln -sf "$(pwd)/scripts/recc_encode.sh" /usr/local/bin/encode
\ No newline at end of file