ffmpeg-smart beta - multi-encoder support

ffmpeg-smart-beta.sh

#!/bin/bash
set -euo pipefail

LOG_PREFIX="[ffmpeg-smart]"

# Parse args
AGENT=""
URL=""
# Auto-detect default accel based on available hardware
detect_accel() {
    # macOS - VideoToolbox
    if [[ "$(uname -s)" == "Darwin" ]]; then
        echo "videotoolbox"; return
    fi
    # NVIDIA
    if [[ -e /dev/nvidia0 ]]; then
        echo "nvenc"; return
    fi
    # Check DRI vendor for Intel/AMD
    local vendor
    for v in /sys/class/drm/renderD*/device/vendor; do
        [[ -r "$v" ]] || continue
        vendor=$(cat "$v" 2>/dev/null)
        case "$vendor" in
            0x8086) echo "qsv"; return ;;   # Intel
            0x1002) echo "vaapi"; return ;; # AMD (vaapi, not deprecated amf)
        esac
    done
    # V4L2 M2M (ARM: Raspberry Pi, Rockchip, etc.)
    for n in /sys/class/video4linux/video*/name; do
        [[ -r "$n" ]] && grep -qiE 'm2m|codec' "$n" && echo "v4l2m2m" && return
    done
    # Fallback to vaapi if DRI exists, else software
    if [[ -e /dev/dri/renderD128 ]]; then
        echo "vaapi"
    else
        echo "software"
    fi
}
ACCEL=$(detect_accel)

while [[ $# -gt 0 ]]; do
    case "$1" in
        -user_agent) AGENT="$2"; shift 2 ;;
        -i) URL="$2"; shift 2 ;;
        -accel) ACCEL="$2"; shift 2 ;;
        *) shift ;;
    esac
done

# Validate
if [[ -z "$URL" ]]; then
    echo "$LOG_PREFIX ERROR: No stream URL provided" >&2
    exit 1
fi

# Cache encoder list for codec selection
ENCODERS="$(ffmpeg -hide_banner -encoders 2>/dev/null || true)"

# Validate accel type
case "$ACCEL" in
    qsv|vaapi|nvenc|v4l2m2m|videotoolbox|software) ;;
    *)
        echo "$LOG_PREFIX ERROR: Unknown accel type: $ACCEL (use: qsv, vaapi, nvenc, v4l2m2m, videotoolbox, software)" >&2
        exit 1
        ;;
esac

# Set hwaccel args and encoder-specific options
set_hwaccel_args() {
    ACCEL_OPTS=""
    case "$ACCEL" in
        qsv)
            HWACCEL_ARGS=( -hwaccel qsv -hwaccel_output_format qsv )
            ACCEL_OPTS="-look_ahead 0"
            ;;
        vaapi)
            HWACCEL_ARGS=( -hwaccel vaapi -hwaccel_output_format vaapi -vaapi_device "${VAAPI_DEVICE:-/dev/dri/renderD128}" )
            ;;
        nvenc)
            HWACCEL_ARGS=( -hwaccel cuda -hwaccel_output_format cuda )
            ;;
        videotoolbox)
            HWACCEL_ARGS=( -hwaccel videotoolbox )
            ;;
        *)
            HWACCEL_ARGS=()
            ;;
    esac
}

# Probe stream
PROBE=$(ffprobe -user_agent "$AGENT" -v quiet -print_format json -show_streams "$URL" 2>&1) || {
    echo "$LOG_PREFIX ERROR: ffprobe failed - cannot access stream" >&2
    exit 1
}

# Parse streams by codec_type
VCODEC=$(echo "$PROBE" | jq -r '.streams[] | select(.codec_type=="video") | .codec_name' | head -n1)
FPS_FRAC=$(echo "$PROBE" | jq -r '.streams[] | select(.codec_type=="video") | .r_frame_rate' | head -n1)
PIX_FMT=$(echo "$PROBE" | jq -r '.streams[] | select(.codec_type=="video") | .pix_fmt // empty' | head -n1)
COLOR_TRANSFER=$(echo "$PROBE" | jq -r '.streams[] | select(.codec_type=="video") | .color_transfer // empty' | head -n1)
WIDTH=$(echo "$PROBE" | jq -r '.streams[] | select(.codec_type=="video") | .width // 0' | head -n1)
HEIGHT=$(echo "$PROBE" | jq -r '.streams[] | select(.codec_type=="video") | .height // 0' | head -n1)
ABITRATE_RAW=$(echo "$PROBE" | jq -r '.streams[] | select(.codec_type=="audio") | .bit_rate // empty' | head -n1)
ACHANNELS=$(echo "$PROBE" | jq -r '.streams[] | select(.codec_type=="audio") | .channels // empty' | head -n1)

# Passthrough conditions:
# - 4K+ resolution (no point re-encoding)
# - 10-bit content on QSV (can't encode properly)
# - HDR content (preserve HDR, don't tone map)
PASSTHROUGH=""
if [[ "$WIDTH" -ge 3840 || "$HEIGHT" -ge 2160 ]]; then
    PASSTHROUGH="4K+"
elif [[ "$PIX_FMT" == *"10"* && "$ACCEL" == "qsv" ]]; then
    PASSTHROUGH="10-bit"
elif [[ "$COLOR_TRANSFER" == "smpte2084" || "$COLOR_TRANSFER" == "arib-std-b67" ]]; then
    PASSTHROUGH="HDR"
fi

# Validate video stream
if [[ -z "$VCODEC" || "$VCODEC" == "null" ]]; then
    echo "$LOG_PREFIX ERROR: No video stream found" >&2
    exit 1
fi

# Select output video codec
if [[ "$ACCEL" == "software" ]]; then
    if [[ "$VCODEC" == "hevc" ]] && grep -qw 'libx265' <<<"$ENCODERS"; then
        VCODEC_OUT="libx265"
        TAG_ARGS="-tag:v hvc1"
    else
        VCODEC_OUT="libx264"
        TAG_ARGS=""
    fi
else
    if [[ "$VCODEC" == "hevc" ]] && grep -qw "hevc_${ACCEL}" <<<"$ENCODERS"; then
        VCODEC_OUT="hevc_${ACCEL}"
        TAG_ARGS="-tag:v hvc1"
    elif [[ "$VCODEC" == "mpeg2video" ]] && grep -qw "mpeg2_${ACCEL}" <<<"$ENCODERS"; then
        VCODEC_OUT="mpeg2_${ACCEL}"
        TAG_ARGS=""
    elif grep -qw "h264_${ACCEL}" <<<"$ENCODERS"; then
        VCODEC_OUT="h264_${ACCEL}"
        TAG_ARGS=""
    else
        echo "$LOG_PREFIX ERROR: No usable video encoder for accel=${ACCEL}" >&2
        exit 1
    fi
fi

# Set hwaccel args (after probe so we can check for 10-bit)
set_hwaccel_args

# Validate audio bitrate (reject sample rates like 44100/48000)
if [[ -n "$ABITRATE_RAW" ]] && [[ "$ABITRATE_RAW" =~ ^[0-9]+$ ]] && [[ "$ABITRATE_RAW" -ge 60000 ]] && [[ "$ABITRATE_RAW" -le 500000 ]]; then
    ABITRATE="$ABITRATE_RAW"
else
    ABITRATE="128000"
fi

# Set channel layout based on channel count
case "$ACHANNELS" in
    1) CHANNEL_LAYOUT="-ch_layout mono" ;;
    2) CHANNEL_LAYOUT="-ch_layout stereo" ;;
    6) CHANNEL_LAYOUT="-ch_layout 5.1" ;;
    8) CHANNEL_LAYOUT="-ch_layout 7.1" ;;
    *)
        CHANNEL_LAYOUT="-ac 2 -ch_layout stereo"
        ACHANNELS="${ACHANNELS:-0} -> 2 (forced)"
        ;;
esac

# Bitrates
VBITRATE="8000000"
MAXRATE="10000000"
BUFSIZE="20000000"

# Calculate GOP with rounding
if [[ "$FPS_FRAC" =~ ^([0-9]+)/([0-9]+)$ ]]; then
    NUM=${BASH_REMATCH[1]}
    DEN=${BASH_REMATCH[2]}
    if [[ $DEN -gt 0 ]]; then
        GOP=$(( (NUM + DEN/2) / DEN ))
        FPS_OUT="$FPS_FRAC"
        GOP_WARN=""
    else
        GOP=50
        FPS_OUT="25/1"
        GOP_WARN=" (invalid fps denominator)"
    fi
else
    GOP=50
    FPS_OUT="25/1"
    GOP_WARN=" (fps parse failed)"
fi

if [[ -n "$PASSTHROUGH" ]]; then
    echo "$LOG_PREFIX Detected ${WIDTH}x${HEIGHT} $VCODEC/$PIX_FMT @ $FPS_FRAC -> passthrough (${PASSTHROUGH}) audio=${ABITRATE}bps ${ACHANNELS}ch" >&2
    exec ffmpeg \
        -user_agent "$AGENT" \
        -reconnect 1 \
        -reconnect_at_eof 1 \
        -reconnect_streamed 1 \
        -reconnect_delay_max 30 \
        -rw_timeout 15000000 \
        -fflags +genpts+igndts+discardcorrupt \
        -err_detect ignore_err \
        -i "$URL" \
        -map 0:v:0 \
        -map 0:a:0? \
        -c:v copy \
        -c:a aac \
        -b:a "$ABITRATE" \
        $CHANNEL_LAYOUT \
        -af "aresample=async=1" \
        -avoid_negative_ts make_zero \
        -start_at_zero \
        -mpegts_copyts 0 \
        -mpegts_flags +pat_pmt_at_frames+resend_headers \
        -flush_packets 1 \
        -max_muxing_queue_size 4096 \
        -f mpegts \
        pipe:1
fi

echo "$LOG_PREFIX Detected ${WIDTH}x${HEIGHT} $VCODEC/$PIX_FMT @ $FPS_FRAC -> $VCODEC_OUT GOP=$GOP${GOP_WARN} audio=${ABITRATE}bps ${ACHANNELS}ch accel=${ACCEL}" >&2

exec ffmpeg \
    -user_agent "$AGENT" \
    "${HWACCEL_ARGS[@]}" \
    -reconnect 1 \
    -reconnect_at_eof 1 \
    -reconnect_streamed 1 \
    -reconnect_delay_max 30 \
    -rw_timeout 15000000 \
    -fflags +genpts+igndts+discardcorrupt \
    -err_detect ignore_err \
    -i "$URL" \
    -map 0:v:0 \
    -map 0:a:0? \
    -c:v "$VCODEC_OUT" \
    -preset fast \
    -b:v "$VBITRATE" \
    -maxrate "$MAXRATE" \
    -bufsize "$BUFSIZE" \
    -g "$GOP" \
    -bf 0 \
    ${ACCEL_OPTS} \
    -fps_mode cfr \
    -r "$FPS_OUT" \
    $TAG_ARGS \
    -c:a aac \
    -b:a "$ABITRATE" \
    $CHANNEL_LAYOUT \
    -af "aresample=async=1" \
    -avoid_negative_ts make_zero \
    -start_at_zero \
    -mpegts_copyts 0 \
    -mpegts_flags +pat_pmt_at_frames+resend_headers \
    -flush_packets 1 \
    -max_muxing_queue_size 4096 \
    -f mpegts \
    pipe:1

testing

Hardware auto-detection:

• nvenc (NVIDIA) → qsv (Intel) → vaapi (AMD/fallback) → v4l2m2m (ARM) → software
• Override with -accel <type>
• CUDA/QSV/VAAPI hwaccel for decode+encode
• Removed deprecated AMF

Passthrough (no re-encode):

• 4K+ resolution
• 10-bit on QSV
• HDR (PQ/HLG)

Logging:

• Shows resolution and pixel format
• Shows passthrough reason

Revision 60a130f - macOS support + deprecation fixes

• Added VideoToolbox support for macOS (auto-detected via uname -s)
• -channel_layout → -ch_layout (deprecated in FFmpeg 5.1+)
• Removed -async 1 (redundant with aresample=async=1 filter)
• -look_ahead 0 now QSV-specific only (was causing warnings on other encoders)