feat: add Windows and Linux wrapper scripts for VMAF optimizer

2025-12-31 18:09:05 +04:00
commit 2edf9f51c7
9 changed files with 2312 additions and 0 deletions
--- a/run_optimisation.sh
+++ b/run_optimisation.sh
@@ -0,0 +1,151 @@
+#!/bin/bash
+
+# VMAF Library Optimiser (Linux/Server runner)
+# This script wraps optimize_library.py with the same interface as the Windows PowerShell version
+
+set -e
+
+COLOR_RED='\033[0;31m'
+COLOR_GREEN='\033[0;32m'
+COLOR_CYAN='\033[0;36m'
+COLOR_YELLOW='\033[1;33m'
+COLOR_WHITE='\033[0;37m'
+COLOR_RESET='\033[0m'
+
+log_info() {
+    echo -e "${COLOR_CYAN}$*${COLOR_RESET}"
+}
+
+log_error() {
+    echo -e "${COLOR_RED}ERROR: $*${COLOR_RESET}" >&2
+}
+
+log_success() {
+    echo -e "${COLOR_GREEN}$*${COLOR_RESET}"
+}
+
+# Default values matching optimize_library.py defaults
+DIRECTORY="."
+VMAF="95.0"
+PRESET="6"
+WORKERS="1"
+SAMPLES="4"
+THOROUGH=""
+ENCODER="svt-av1"
+HWACCEL=""
+
+# Parse command line arguments
+while [[ $# -gt 0 ]]; do
+    case "$1" in
+        --directory)
+            DIRECTORY="$2"
+            shift 2
+            ;;
+        --vmaf)
+            VMAF="$2"
+            shift 2
+            ;;
+        --preset)
+            PRESET="$2"
+            shift 2
+            ;;
+        --workers)
+            WORKERS="$2"
+            shift 2
+            ;;
+        --samples)
+            SAMPLES="$2"
+            shift 2
+            ;;
+        --thorough)
+            THOROUGH="--thorough"
+            shift
+            ;;
+        --encoder)
+            ENCODER="$2"
+            shift 2
+            ;;
+        --hwaccel)
+            HWACCEL="$2"
+            shift 2
+            ;;
+        *)
+            DIRECTORY="$1"
+            shift
+            ;;
+    esac
+done
+
+# Check if python3 is available
+if ! command -v python3 &> /dev/null; then
+    if ! command -v python &> /dev/null; then
+        log_error "Python 3 not found. Please install Python 3."
+        exit 1
+    else
+        PYTHON_CMD="python"
+    fi
+else
+    PYTHON_CMD="python3"
+fi
+
+# Check if optimize_library.py exists
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+SCRIPT_PATH="$SCRIPT_DIR/optimize_library.py"
+
+if [[ ! -f "$SCRIPT_PATH" ]]; then
+    log_error "optimize_library.py not found in: $SCRIPT_DIR"
+    exit 1
+fi
+
+# Build command arguments
+ARGS=(
+    "$PYTHON_CMD" "$SCRIPT_PATH"
+    "$DIRECTORY"
+    --vmaf "$VMAF"
+    --preset "$PRESET"
+    --workers "$WORKERS"
+    --samples "$SAMPLES"
+    --encoder "$ENCODER"
+)
+
+if [[ -n "$THOROUGH" ]]; then
+    ARGS+=(--thorough)
+fi
+
+if [[ -n "$HWACCEL" ]]; then
+    ARGS+=(--hwaccel "$HWACCEL")
+fi
+
+# Print configuration
+log_info "========================================"
+log_info "VMAF Library Optimiser (Linux/Server)"
+log_info "========================================"
+echo ""
+log_info "Directory: $DIRECTORY"
+log_info "Target VMAF: $VMAF"
+log_info "Preset: $PRESET"
+log_info "Workers: $WORKERS"
+log_info "Samples: $SAMPLES"
+log_info "Encoder: $ENCODER"
+if [[ -n "$THOROUGH" ]]; then
+    log_info "Thorough: Yes"
+fi
+if [[ -n "$HWACCEL" ]]; then
+    log_info "HW Accel: $HWACCEL"
+fi
+echo ""
+log_info "Running optimize_library.py..."
+echo ""
+
+# Run the optimisation
+"${ARGS[@]}"
+EXIT_CODE=$?
+
+# Handle exit code
+if [ $EXIT_CODE -eq 0 ]; then
+    log_success "SUCCESS: Library optimisation completed"
+else
+    log_error "optimize_library.py exited with code $EXIT_CODE"
+fi
+
+exit $EXIT_CODE