From 0523a28e4930fc599ff2de504aa081e8416ed8f1 Mon Sep 17 00:00:00 2001
From: Levon <lgevorgyan03@gmail.com>
Date: Fri, 12 Jan 2024 19:44:09 -0600
Subject: [PATCH] log is done

---
 .gitignore              |  1 +
 scripts/benchmark.sh    | 16 +++++++++-------
 scripts/install_deps.sh |  2 +-
 3 files changed, 11 insertions(+), 8 deletions(-)

diff --git a/.gitignore b/.gitignore
index 7cd7391..2b12632 100644
--- a/.gitignore
+++ b/.gitignore
@@ -2,4 +2,5 @@ ffmpeg/
 rav1e/
 svt/
 aom/
+vmaf/
 benchmark/
\ No newline at end of file
diff --git a/scripts/benchmark.sh b/scripts/benchmark.sh
index d84a0a2..44194ce 100755
--- a/scripts/benchmark.sh
+++ b/scripts/benchmark.sh
@@ -18,17 +18,16 @@ test -f "$INPUT_DIR/${INPUT[2]}" || wget -O "$INPUT_DIR/${INPUT[2]}" 'https://ww
 rm -rf "$OUTPUT_DIR" && mkdir -p "$OUTPUT_DIR"
 
 # Different variables to test
-CRF=(40)
+CRF=(20 25 30)
 ENCODER=('libsvtav1' 'librav1e' 'libaom-av1')
-ENCODER=('libsvtav1')
-PRESET=(13)
+PRESET=(4 8 12)
 
 # Log for results
 LOG="$OUTPUT_DIR/results.txt"
+VMAF_RESULTS="$OUTPUT_DIR/vmaf.json"
 
 for input in "${INPUT[@]}"
 do
-    # echo "$input"
     for encoder in "${ENCODER[@]}"
     do
         for preset in "${PRESET[@]}"
@@ -38,11 +37,14 @@ do
                 OUTPUT="$OUTPUT_DIR/${encoder}_preset${preset}_crf${crf}_$input"
                 echo "output: $OUTPUT" >> "$LOG"
                 TIME_BEFORE=$(date +%s)
-                ffmpeg -i "$INPUT_DIR/$input" -c:a copy -c:v "$encoder" -preset "$preset" -crf "$crf" "$OUTPUT" 2> /dev/null || exit 1
+                ffmpeg -i "$INPUT_DIR/$input" -c:a copy -c:v "$encoder" \
+                    -preset "$preset" -crf "$crf" "$OUTPUT" 2> /dev/null || exit 1
                 TIME_AFTER=$(date +%s)
                 TIME_DIFF=$((TIME_AFTER - TIME_BEFORE))
-                echo -e "\t time taken: $TIME_DIFF" >> "$LOG"
-                ffmpeg -i "$OUTPUT" -i "$INPUT_DIR/$input" -lavfi libvmaf=n_threads="$(nproc)" -f 'null' - | grep "VMAF score:" >> "$LOG" || exit 1
+                echo -e "\t time taken: $TIME_DIFF seconds" >> "$LOG"
+                ffmpeg -an -sn -i "$OUTPUT" -i "$INPUT_DIR/$input" -lavfi \
+                    libvmaf=n_threads="$(nproc)":log_path="$VMAF_RESULTS":log_fmt='json' -f 'null' -
+                echo -e "\t mean vmaf: $(cat "$VMAF_RESULTS" | jq '.pooled_metrics.vmaf.mean')" >> "$LOG" || exit 1
             done
         done    
     done
diff --git a/scripts/install_deps.sh b/scripts/install_deps.sh
index fc5945b..e7ae08b 100755
--- a/scripts/install_deps.sh
+++ b/scripts/install_deps.sh
@@ -6,7 +6,7 @@ sudo apt-get install autoconf automake build-essential cmake git-core \
   libass-dev libfreetype6-dev libsdl2-dev libtool libva-dev libvdpau-dev \
   libvorbis-dev libxcb1-dev libxcb-shm0-dev libxcb-xfixes0-dev pkg-config \
   texinfo wget zlib1g-dev nasm yasm libssl-dev time python3 meson ninja-build\
-  doxygen xxd -y || exit 1
+  doxygen xxd jq -y || exit 1
 
 curl https://sh.rustup.rs -sSf | sh -s -- -y
 source "$HOME/.cargo/env"