anytovcd.sh

来自「Motion JPEG编解码器源代码」· Shell 代码 · 共 1,015 行 · 第 1/2 页

#!/bin/sh
#
#  anytovcd.sh
#
#  Copyright (C) 2004,2005 Nicolas Boos <[e-mail]>
# 
#  This program is free software; you can redistribute it and/or modify
#  it under the terms of the GNU General Public License as published by
#  the Free Software Foundation; either version 2 of the License, or
#  (at your option) any later version.
#
#  This program is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU General Public License for more details.
#
#  You should have received a copy of the GNU General Public License
#  along with this program; if not, write to the Free Software
#  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
#

AUD_TRACK="1"
VOLUMES="1"
FILTERING="none"
FILTER_TYPE="median"
PREFIX_OUT="out"
VCD_TYPE="dvd"
DEC_TOOL="ffmpeg"
ENC_TOOL="mpeg2enc"
MATRICES="tmpgenc"
QUALITY="best"
CQMFILE="matrices.txt"
STATFILE="statfile.log"

BFR="bfr"
FFMPEG="ffmpeg"
MPEG2DEC="mpeg2dec"
MPEG2ENC="mpeg2enc"
MPLEX="mplex"
PGMTOY4M="pgmtoy4m"
SOX="sox"
TCCAT="tccat"
TCDEMUX="tcdemux"
TCPROBE="tcprobe"
TRANSCODE="transcode"
Y4MDENOISE="y4mdenoise"
Y4MSCALER="y4mscaler"
Y4MSPATIALFILTER="y4mspatialfilter"
Y4MUNSHARP="y4munsharp"
YUVDENOISE="yuvdenoise"
YUVFPS="yuvfps"
YUVMEDIANFILTER="yuvmedianfilter"

SCRIPT_NAME="anytovcd.sh"
SCRIPT_VERSION="9"


# custom quant. matrices
INTRA_KVCD="8,9,12,22,26,27,29,34,9,10,14,26,27,29,34,37,12,14,18,27,29,34,37,38,22,26,27,31,36,37,38,40,26,27,29,36,39,38,40,48,27,29,34,37,38,40,48,58,29,34,37,38,40,48,58,69,34,37,38,40,48,58,69,79"
INTER_KVCD="16,18,20,22,24,26,28,30,18,20,22,24,26,28,30,32,20,22,24,26,28,30,32,34,22,24,26,30,32,32,34,36,24,26,28,32,34,34,36,38,26,28,30,32,34,36,38,40,28,30,32,34,36,38,42,42,30,32,34,36,38,40,42,44"

INTRA_MVCD="8,10,16,20,25,28,33,44,10,10,14,26,27,29,34,47,16,14,18,27,29,34,37,48,20,26,27,31,36,37,38,50,25,27,29,32,37,40,48,58,28,33,36,42,45,50,58,68,33,37,39,44,48,56,66,79,37,39,45,48,56,66,79,83"
INTER_MVCD="16,18,20,22,23,27,32,40,18,20,22,24,26,32,36,42,20,22,24,26,33,38,42,44,22,24,26,34,38,42,44,46,23,26,33,38,42,44,46,48,27,32,38,42,44,46,48,50,32,36,42,44,46,48,50,52,40,42,44,46,48,50,52,58"

INTRA_TMPGENC="8,16,19,22,26,27,29,34,16,16,22,24,27,29,34,37,19,22,26,27,29,34,34,38,22,22,26,27,29,34,37,40,22,26,27,29,32,35,40,48,26,27,29,32,35,40,40,58,26,27,29,34,38,46,56,69,27,29,35,38,46,56,69,83"
INTER_TMPGENC="16,17,18,19,20,21,22,23,17,18,19,20,21,22,23,24,18,19,20,21,22,23,24,25,19,20,21,22,23,24,26,27,20,21,22,23,25,26,27,28,21,22,23,24,26,27,28,30,22,23,24,26,27,28,30,31,23,24,25,27,28,30,31,33"

matrix_copy ()
{
    # TODO: find less ugly func.
    echo "`echo "$1" | \
    awk -F, '{for (i=1; i<=NF; i++) {n=n$i","; z+=1; if (i==64) {z=0; sub(",$","",n)}; if (z==8) {z=0; n=n"\n"}}; print n}'`"
}

probe_ffmpeg_version ()
{
    echo "`${FFMPEG} 2>&1 | \
    awk '$4 == "build" {print $5}' | sed s/,// | head -1`"
}

probe_aud_bitrate ()
{
    # $1 = audio input
    # $2 = audio track
    echo "`${FFMPEG} -i "$1" 2>&1 | \
    awk '/Audio:/ {print $8}' | head -${2} | tail -1`"
}

probe_aud_fmt ()
{
    # $1 = audio input
    # $2 = audio track
    echo "`${FFMPEG} -i "$1" 2>&1 | \
    awk '/Audio:/ {print $4}' | sed s/,// | head -${2} | tail -1`"
}

probe_aud_freq ()
{
    # $1 = audio input
    # $2 = audio track
    echo "`${FFMPEG} -i "$1" 2>&1 | \
    awk '/Audio:/ {print $5}' | sed s/,// | head -${2} | tail -1`"
}

probe_vid_fps ()
{
    ${FFMPEG} -i "$1" -f yuv4mpegpipe -t 1 -y /tmp/tmp.y4m >/dev/null 2>&1
    echo "`head -1 /tmp/tmp.y4m | awk '{print $4}' | cut -c 2-`"
    rm -f /tmp/tmp.y4m
}

probe_vid_ilace ()
{
    ${FFMPEG} -i "$1" -f yuv4mpegpipe -t 1 -y /tmp/tmp.y4m >/dev/null 2>&1
    echo "`head -1 /tmp/tmp.y4m | awk '{print $5}' | cut -c 2-`"
    rm -f /tmp/tmp.y4m
}

probe_vid_sar ()
{
    ${TRANSCODE} -i "$1" -c 0-1 -y yuv4mpeg,null -o /tmp/tmp.y4m >/dev/null 2>&1
    echo "`head -1 /tmp/tmp.y4m | awk '{print $6}' | cut -c 2-`"
    rm -f /tmp/tmp.y4m
}

probe_vid_size ()
{
    echo "`${FFMPEG} -i "$1" 2>&1 | \
    awk '/Video:/ {print $5}' | sed s/,// | head -1`"
}

probe_vid_fmt ()
{
    echo "`${FFMPEG} -i "$1" 2>&1 | \
    awk '/Video:/ {print $4}' | sed s/,// | head -1`"
}

probe_mpeg_length ()
{
    echo "`${TCCAT} -i "$1" | ${TCDEMUX} -W 2>&1 | \
    awk '/unit/ {print sum+=$2}' | tail -1`"
}

probe_vid_length ()
{
    echo "`${TCPROBE} -i "$1" 2>/dev/null | \
    awk '/length:/ {print $2}'`"
}

range_check ()
{
    # if input value < min. value
    if test "$1" -lt "$2"; then
        
	# = min. value
	echo "$2"

    # if input value > max. value
    elif test "$1" -gt "$3"; then
        
	# = max. value
	echo "$3"
    
    else
        
	echo "$1"
    
    fi
}

mpeg2enc_statfile_analyse ()
{
    # calculate average quant value from pseudo statfile (mpeg2enc output)
    echo "`awk -F= '/quant=/ {print $2}' "$1" | \
    awk '{iquant=$1; \
    oquant=iquant; \
    if (iquant>=  10) {oquant=9}; \
    if (iquant>=  12) {oquant=10}; \
    if (iquant>=  14) {oquant=11}; \
    if (iquant>=  16) {oquant=12}; \
    if (iquant>=  18) {oquant=13}; \
    if (iquant>=  20) {oquant=14}; \
    if (iquant>=  22) {oquant=15}; \
    if (iquant>=  24) {oquant=16}; \
    if (iquant>=  28) {oquant=17}; \
    if (iquant>=  32) {oquant=18}; \
    if (iquant>=  36) {oquant=19}; \
    if (iquant>=  40) {oquant=20}; \
    if (iquant>=  44) {oquant=21}; \
    if (iquant>=  48) {oquant=22}; \
    if (iquant>=  52) {oquant=23}; \
    if (iquant>=  56) {oquant=24}; \
    if (iquant>=  64) {oquant=25}; \
    if (iquant>=  72) {oquant=26}; \
    if (iquant>=  80) {oquant=27}; \
    if (iquant>=  88) {oquant=28}; \
    if (iquant>=  96) {oquant=29}; \
    if (iquant>= 104) {oquant=30}; \
    if (iquant>= 112) {oquant=31}; \
    nquant+=1; \
    sum_quant+=oquant; \
    print sum_quant/nquant}' | \
    tail -1 | awk -F. '{print $1}'`"
}

show_help ()
{
    echo "----------------------------------"
    echo "-a    audio track number (1)      "
    echo "-A    force input pixel aspect ratio (X:Y)"
    echo "-b    blind mode (no video)       "
    echo "-c    enable audio pass-through   "
    echo "-d    video decoder tool          "
    echo "      avail. : ffmpeg (default) or mpeg2dec"
    echo "-e    video encoder tool          "
    echo "      avail. : mpeg2enc (default) "
    echo "      or ffmpeg (experimental)    "
    echo "-i    input file                  "
    echo "-I    force input ilace flag      "
    echo "      avail. : none, top_first and"
    echo "               bottom_first       "
    echo "-J    force output ilace flag (if possible)"
    echo "      avail. : none, top_first and bottom_first"
    echo "-f    filtering method (*none*)   "
    echo "      avail.: light, medium, heavy"
    echo "-m    mute mode (no audio)        "
    echo "-n    output norm in case of input"
    echo "      with non-standard framerate "
    echo "      pal (default), ntsc or ntsc_film"
    echo "-o    output files prefix (default = \"out\")"
    echo "-p    output type (default = dvd) "
    echo "      avail. : cvd, cvd_wide, dvd, dvd_wide, mvcd,"
    echo "               svcd and vcd       "
    echo "-q    quality (default = best)    "
    echo "      avail. : best, good, fair   "
    echo "-r    enable two passes encoding mode"
    echo "-R    force input framerate (X:Y) "
    echo "-t    filter type to use with     "
    echo "      the chosen filter method    "
    echo "      (default = median)          "
    echo "      avail. : mean (yuvmedianfilter -f) "
    echo "               median (yuvmedianfilter)  "
    echo "               spatial (y4mspatialfilter)"
    echo "               temporal (yuvdenoise)     "
    echo "               unsharp (y4munsharp)      "
    echo "               hqdenoise (y4mdenoise)    "
    echo "-T    force input length (minutes)"
    echo "-v    script version              "
    echo "-V    number of volumes (1)       "
}

show_error ()
{
    echo "**ERROR: [${SCRIPT_NAME}] $1"
}

show_info ()
{
    echo "   INFO: [${SCRIPT_NAME}] $1"
}

show_warn ()
{
    echo "++ WARN: [${SCRIPT_NAME}] $1"
}

test_bin ()
{
    if ! which ${1} >/dev/null; then

        show_error "${1} binary not present."
        exit 2

    fi
}

for BIN in ${FFMPEG} ${MPLEX} ${PGMTOY4M}; do

    test_bin ${BIN}

done

while getopts a:A:bcd:e:f:i:I:J:mn:o:p:q:rR:t:T:vV: OPT
do
    case ${OPT} in
    a)    AUD_TRACK="${OPTARG}";;
    A)    VID_SAR_SRC="${OPTARG}";;
    b)    BLIND_MODE="1";;
    c)    ACOPY_MODE="1";;
    d)    DEC_TOOL="${OPTARG}";;
    e)    ENC_TOOL="${OPTARG}";;
    i)    VIDEO_SRC="${OPTARG}"; AUDIO_SRC="${VIDEO_SRC}";;
    I)    INTERLACING="${OPTARG}";;
    J)    INTERLACING_OUT="${OPTARG}";;
    f)    FILTERING="${OPTARG}";;
    m)    MUTE_MODE="1";;
    n)    OUT_NORM="${OPTARG}";;
    o)    PREFIX_OUT="${OPTARG}";;
    p)    VCD_TYPE="${OPTARG}";;
    q)    QUALITY="${OPTARG}";;
    r)    ENC_TYPE="abr";;
    R)    VID_FPS_SRC="${OPTARG}";;
    t)    FILTER_TYPE="${OPTARG}";;
    T)    DURATION="${OPTARG}";;
    v)    echo "${SCRIPT_NAME} version ${SCRIPT_VERSION}"; exit 0;;
    V)    VOLUMES="${OPTARG}";;
    \?)   show_help; exit 0;;
    esac
done

if test "${VIDEO_SRC}" == "" || ! test -r "${VIDEO_SRC}"; then

    show_error "input file not specified or not present."
    show_help
    exit 2

fi

FFMPEG_VERSION="`probe_ffmpeg_version`"

AUD_TRACK="`range_check ${AUD_TRACK} 1 99`"
FFMPEG_AUD_TRACK="`${FFMPEG} -i \"${AUDIO_SRC}\" 2>&1 | awk '/Audio:/ {sub("^#","",$2); print $2}' | head -${AUD_TRACK} | tail -1`"

AUD_BITRATE_SRC="`probe_aud_bitrate "${AUDIO_SRC}" ${AUD_TRACK}`"
AUD_FMT_SRC="`probe_aud_fmt "${AUDIO_SRC}" ${AUD_TRACK}`"
AUD_FREQ_SRC="`probe_aud_freq "${AUDIO_SRC}" ${AUD_TRACK}`"
VID_FMT_SRC="`probe_vid_fmt "${VIDEO_SRC}"`"
VID_SIZE_SRC="`probe_vid_size "${VIDEO_SRC}"`"

# input interlacing
if test "${INTERLACING}" == "none"; then

    VID_ILACE_SRC="p"

elif test "${INTERLACING}" == "top_first"; then

    VID_ILACE_SRC="t"

elif test "${INTERLACING}" == "bottom_first"; then

    VID_ILACE_SRC="b"

else

    VID_ILACE_SRC="`probe_vid_ilace "${VIDEO_SRC}"`"

fi

# output interlacing
if test "${INTERLACING_OUT}" == "none"; then

    VID_ILACE_OUT="p"

elif test "${INTERLACING_OUT}" == "top_first"; then

    VID_ILACE_OUT="t"

elif test "${INTERLACING_OUT}" == "bottom_first"; then

    VID_ILACE_OUT="b"

else

    VID_ILACE_OUT="${VID_ILACE_SRC}"

fi

# input framerate
if test "${VID_FPS_SRC}" == ""; then

    VID_FPS_SRC="`probe_vid_fps "${VIDEO_SRC}"`"

fi

# input pixel/sample aspect ratio
if test "${VID_SAR_SRC}" == ""; then

    test_bin ${TRANSCODE}
    VID_SAR_SRC="`probe_vid_sar "${VIDEO_SRC}"`"

fi

VID_SAR_N_SRC="`echo ${VID_SAR_SRC} | cut -d: -f1`"
VID_SAR_D_SRC="`echo ${VID_SAR_SRC} | cut -d: -f2`"

if test "${VID_SAR_N_SRC}" == "0" || test "${VID_SAR_D_SRC}" == "0"; then

    show_warn "Unknown pixel aspect ratio, defaulting to 1:1"
    VID_SAR_SRC="1:1"

fi

FF_DEC="${FFMPEG} -i \"${VIDEO_SRC}\""
FF_DEC_AFLAGS="-map ${FFMPEG_AUD_TRACK} -i \"${AUDIO_SRC}\""
FF_ENC="${FFMPEG} -v 0 -f yuv4mpegpipe -i /dev/stdin -bf 2"
MPEG2ENC="${MPEG2ENC} -v 0 -M 0 -R 2 -P -s -2 1 -E -5"
MPLEX="${MPLEX} -v 1"
PGMTOY4M="${PGMTOY4M} -r ${VID_FPS_SRC} -i ${VID_ILACE_SRC} -a ${VID_SAR_SRC}"
Y4MDENOISE_FLAGS="-v 0"
Y4MSCALER="${Y4MSCALER} -v 0 -S option=cubicCR"

# ffmpeg decoder version
if test "${FFMPEG_VERSION}" -ge "4731"; then

    FF_DEC="${FF_DEC} -f image2pipe -vcodec pgmyuv"

else

    FF_DEC="${FF_DEC} -f imagepipe -img pgmyuv"

fi

# filtering method(s)
if test "${FILTERING}" == "none"; then

    YUVMEDIANFILTER=""
    YUVDENOISE=""
    Y4MSPATIALFILTER=""
    MEAN_FILTER=""
    Y4MUNSHARP=""

elif test "${FILTERING}" == "light"; then

    YUVMEDIANFILTER="${YUVMEDIANFILTER} -t 0"
    YUVDENOISE="${YUVDENOISE} -l 1 -t 4 -S 0"
    Y4MSPATIALFILTER="${Y4MSPATIALFILTER} -x 0 -y 0"
    MEAN_FILTER="${YUVMEDIANFILTER} -f -r 1 -R 1"
    Y4MDENOISE_FLAGS="${Y4MDENOISE_FLAGS} -t 1"
    Y4MUNSHARP="${Y4MUNSHARP} -L 1.5,0.2,0"

elif test "${FILTERING}" == "medium"; then

    YUVMEDIANFILTER="${YUVMEDIANFILTER} -t 1 -T 1"
    YUVDENOISE="${YUVDENOISE} -l 2 -t 6 -S 0"
    Y4MSPATIALFILTER="${Y4MSPATIALFILTER} -X 4,0.64 -Y 4,0.8"
    MEAN_FILTER="${YUVMEDIANFILTER} -f -r 1 -R 1 -w 2.667"
    Y4MDENOISE_FLAGS="${Y4MDENOISE_FLAGS} -t 2"
    Y4MUNSHARP="${Y4MUNSHARP} -L 2.0,0.3,0"

elif test "${FILTERING}" == "heavy"; then

    YUVMEDIANFILTER="${YUVMEDIANFILTER}"
    YUVDENOISE="${YUVDENOISE} -l 3 -t 8 -S 0"
    Y4MSPATIALFILTER="${Y4MSPATIALFILTER} -X 4,0.5 -Y 4,0.5 -x 2,0.5 -y 2,0.5"
    MEAN_FILTER="${YUVMEDIANFILTER} -f"
    Y4MDENOISE_FLAGS="${Y4MDENOISE_FLAGS} -t 3"
    Y4MUNSHARP="${Y4MUNSHARP} -L 5.0,0.5,0"

else

    show_error "the specified filtering method is inexistant."
    show_help
    exit 2

fi

# quality presets
if test "${QUALITY}" == "best"; then

    QUANT="4"

elif test "${QUALITY}" == "good"; then

    QUANT="5"

elif test "${QUALITY}" == "fair"; then

    QUANT="6"

else

    show_error "the specified quality preset is inexistant."
    show_help
    exit 2

fi

# profile(s)
if test "${VCD_TYPE}" == "cvd"; then

    FF_ENC="${FF_ENC} -vcodec mpeg2video -f rawvideo -bufsize 224"
    MPEG2ENC="${MPEG2ENC} -f 8"
    Y4MSCALER="${Y4MSCALER} -O preset=cvd"
    MPLEX="${MPLEX} -f 8"
    
    VID_FMT_OUT="m2v"
    VID_MINRATE_OUT="0"
    VID_MAXRATE_OUT="7500"

    VID_SAR_525_OUT="20:11"
    VID_SAR_625_OUT="59:27"
    VID_SIZE_525_OUT="352x480"
    VID_SIZE_625_OUT="352x576"
    
    AUD_FMT_OUT="ac3"
    AUD_BITRATE_OUT="192"
    AUD_FREQ_OUT="48000"
    AUD_CHANNELS_OUT="2"
    
    # default volume size in mbytes (m = 1000)
    VOL_SIZE="4700"

elif test "${VCD_TYPE}" == "cvd_wide"; then

    FF_ENC="${FF_ENC} -vcodec mpeg2video -f rawvideo -bufsize 224"
    MPEG2ENC="${MPEG2ENC} -f 8"
    MPLEX="${MPLEX} -f 8"