From 0f5e1fe1f68a88bfc764a6d181ca5c37c14ef4c1 Mon Sep 17 00:00:00 2001
From: Levon Gevorgyan <lgevorgyan03@gmail.com>
Date: Sun, 18 Aug 2024 19:17:04 -0500
Subject: [PATCH] small param changes

---
 scripts/benchmark.sh   | 15 +++++++++++----
 scripts/recc_encode.sh |  2 +-
 2 files changed, 12 insertions(+), 5 deletions(-)

diff --git a/scripts/benchmark.sh b/scripts/benchmark.sh
index 6c35c41..207e868 100755
--- a/scripts/benchmark.sh
+++ b/scripts/benchmark.sh
@@ -24,6 +24,13 @@ URL_DL[3]='http://download.opencontent.netflix.com.s3.amazonaws.com/AV1/DVB-DASH
 INPUT[4]='D_2.mp4'
 URL_DL[4]='http://download.opencontent.netflix.com.s3.amazonaws.com/AV1/DVB-DASH/D_2.mp4'
 
+# check for libvmaf
+if [[ "$(ffmpeg 2>&1 | grep 'enable-libvmaf' ; echo $?)" == 1 ]]; then
+    echo "libvmaf not found"
+    echo "add -v flag to build.sh"
+    exit 1
+fi
+
 # download videos
 mkdir -p "$DL_DIR"
 for index in "${!INPUT[@]}"
@@ -53,9 +60,9 @@ ENCODER=('librav1e' 'libaom-av1' 'libsvtav1' 'libx264' 'libx265' 'libvpx-vp9')
 PRESET=(2 4 6 8)
 
 # uncomment for quick testing
-# CRF=(25)
+CRF=(25)
 # ENCODER=('libsvtav1')
-# PRESET=(8)
+PRESET=(8)
 
 # Log for results
 LOG="$BENCHMARK_DIR/results.txt"
@@ -113,7 +120,7 @@ do
                     PARAMS="-cpu-used $preset -row-mt 1 -threads $THREADS -crf $crf"
                 elif [[ "$encoder" == "libsvtav1" ]]; then
                     PARAMS="-preset $preset -crf $crf -svtav1-params \
-                            scd=1:tune=0:enable-overlays=1:enable-hdr=1:fast-decode=1:enable-variance-boost=1"
+                            scd=1:tune=0:enable-overlays=1:fast-decode=1:enable-variance-boost=1"
                 elif [[ ("$encoder" == "libx264") || ("$encoder" == "libx265") ]]; then
                     test "$preset" -eq 2 && preset=veryslow
                     test "$preset" -eq 4 && preset=slower
@@ -125,7 +132,7 @@ do
                     test "$preset" -eq 4 && cpu_used=3
                     test "$preset" -eq 6 && cpu_used=4
                     test "$preset" -eq 8 && cpu_used=5
-                    PARAMS="-cpu-used $cpu_used -crf $crf -row-mt 1 -deadline 0 -quality 0"
+                    PARAMS="-cpu-used $cpu_used -crf $crf -b:v 0 -row-mt 1 -deadline 0 -quality 0"
                 else
                     PARAMS=""
                 fi
diff --git a/scripts/recc_encode.sh b/scripts/recc_encode.sh
index e2e0485..1220736 100755
--- a/scripts/recc_encode.sh
+++ b/scripts/recc_encode.sh
@@ -63,7 +63,7 @@ encode() {
     echo -e '#!/bin/bash\n' > "$ENCODE_FILE"
     echo "export OUTPUT=\"$OUTPUT\"" >> "$ENCODE_FILE"
 
-    SVT_PARAMS="${GRAIN}sharpness=3:tune=3:enable-overlays=1:scd=1:enable-hdr=1:fast-decode=1:enable-variance-boost=1:enable-qm=1:qm-min=0:qm-max=15"
+    SVT_PARAMS="${GRAIN}sharpness=3:tune=3:enable-overlays=1:scd=1:fast-decode=1:enable-variance-boost=1:enable-qm=1:qm-min=0:qm-max=15"
     echo "export SVT_PARAMS=\"$SVT_PARAMS\"" >> "$ENCODE_FILE"
 
     UNMAP=$(unmap_streams "$INPUT")