batch-av1

#!/bin/bash

# https://github.com/Jaded-Encoding-Thaumaturgy/JET-guide/blob/master/docs/encoding/svtav1.md
# TODO: check if anime mode is necessary

# Default values
CRF=25
PRESET=4
ANIME_MODE=false
DRY_RUN=false
USE_OPUS=false
FILES=()

# Display help message
show_help() {
    echo "Usage: $(basename "$0") [OPTIONS] [FILES...]"
    echo ""
    echo "Batch convert video files to AV1 using SVT-AV1."
    echo ""
    echo "Options:"
    echo "  -a, --anime         Enable anime mode (adds more film-grain), TODO: check if this is necessary"
    echo "  -c, --crf VALUE     Set CRF value (default: $CRF)"
    echo "  -p, --preset VALUE  Set SVT-AV1 preset (default: $PRESET)"
    echo "  -o, --opus          Transcode audio to Opus (default: copy)"
    echo "  -d, --dry-run       Show what would be done without encoding"
    echo "  -h, --help          Show this help message"
    echo ""
    echo "If no files are provided, common video files (.mp4, .mkv, .webm, .mov, .avi) in the current directory will be processed."
}

# Parse arguments
while [[ $# -gt 0 ]]; do
    case "$1" in
        -a|--anime)
            ANIME_MODE=true
            shift
            ;;
        -c|--crf)
            CRF="$2"
            shift 2
            ;;
        -p|--preset)
            PRESET="$2"
            shift 2
            ;;
        -o|--opus)
            USE_OPUS=true
            shift
            ;;
        -d|--dry-run)
            DRY_RUN=true
            shift
            ;;
        -h|--help)
            show_help
            exit 0
            ;;
        -*)
            echo "Unknown option: $1"
            show_help
            exit 1
            ;;
        *)
            FILES+=("$1")
            shift
            ;;
    esac
done

# Check for ffmpeg and libsvtav1
if ! command -v ffmpeg &> /dev/null; then
    echo "Error: ffmpeg is not installed."
    exit 1
fi

if ! ffmpeg -encoders 2>/dev/null | grep -q libsvtav1; then
    echo "Error: ffmpeg does not support libsvtav1. Please install a compatible version."
    exit 1
fi

# If no files provided, default to common video formats
if [ ${#FILES[@]} -eq 0 ]; then
    shopt -s nullglob
    FILES=(*.mp4 *.mkv *.webm *.mov *.avi)
fi

if [ ${#FILES[@]} -eq 0 ]; then
    echo "Error: No files to process."
    exit 1
fi

# Construct SVT-AV1 parameters
SVTAV1_PARAMS="tune=1:enable-variance-boost=1:tf-strength=1:sharpness=1:tile-columns=1:film-grain=4:luminance-qp-bias=25"

if [ "$ANIME_MODE" = true ]; then
    SVTAV1_PARAMS="${SVTAV1_PARAMS/film-grain=4/film-grain=8}"
fi

AUDIO_OPTS=(-c:a copy)
if [ "$USE_OPUS" = true ]; then
    AUDIO_OPTS=(-c:a libopus -b:a 128k)
fi

FFMPEG_OPTIONS=(
    -map 0
    -c:v libsvtav1
    -crf "$CRF"
    -preset "$PRESET"
    -pix_fmt yuv420p10le
    -svtav1-params "$SVTAV1_PARAMS"
    "${AUDIO_OPTS[@]}"
    -c:s copy
)

echo "--- Starting AV1 Batch Conversion ---"
echo "--- CRF: $CRF, Preset: $PRESET, Anime: $ANIME_MODE, Opus: $USE_OPUS"
[ "$DRY_RUN" = true ] && echo "--- DRY RUN MODE ENABLED ---"

for INPUT_FILE in "${FILES[@]}"; do
    if [[ ! -f "$INPUT_FILE" ]]; then
        echo "--- Skipping: $INPUT_FILE (Not a file)"
        continue
    fi

    # Determine output filename in the same directory as input
    DIR=$(dirname "$INPUT_FILE")
    BASE=$(basename "$INPUT_FILE")
    NAME="${BASE%.*}"
    OUTPUT_FILE="${DIR}/${NAME}_AV1.mkv"

    # Avoid self-overwrite (e.g. if input is already _AV1.mkv)
    if [[ "$INPUT_FILE" == "$OUTPUT_FILE" ]]; then
        OUTPUT_FILE="${DIR}/${NAME}_AV1_new.mkv"
    fi

    echo "--- Processing: $INPUT_FILE"
    echo "--- Output: $OUTPUT_FILE"

    if [ "$DRY_RUN" = true ]; then
        echo "ffmpeg -i \"$INPUT_FILE\" ${FFMPEG_OPTIONS[*]} \"$OUTPUT_FILE\""
    else
        ffmpeg -i "$INPUT_FILE" "${FFMPEG_OPTIONS[@]}" "$OUTPUT_FILE" -y

        if [ $? -eq 0 ]; then
            echo "--- Finished: $INPUT_FILE (Success)"
        else
            echo "--- Error: $INPUT_FILE (FFmpeg failed)"
            echo "$(date): Error processing $INPUT_FILE" >> ./ffmpeg-error-log.txt
        fi
    fi
    echo "-------------------------------------"
done

echo "--- Batch processing complete! ---"