blob: 68ed90992ca2fb92877903396868c718d9055146 [file] [log] [blame]
#!/bin/bash
# Encode each .yuv file in the specified input directory to H.264 format,
# and compute the PSNR.
# Input Parameters:
# $1=Input directory
tempyuvfile=$(mktemp ./tempXXXXX.yuv)
if [ ! -d "$1" ]; then
echo "No such directory: $1"
exit 1
fi
if [ ! -d encoded_clips ]; then
mkdir encoded_clips
mkdir encoded_clips/h264
mkdir encoded_clips/vp8
fi
if [ ! -d logs ]; then
mkdir logs
mkdir logs/h264
mkdir logs/vp8
fi
if [ ! -d stats ]; then
mkdir stats
mkdir stats/h264
mkdir stats/vp8
fi
for filename in $1/*.yuv
do
echo "Processing ${filename}"
# filename format: <path>/<clip_name>_<width>_<height>_<frame_rate>.yuv
pathless=$(basename ${filename})
clip_stem=${pathless%.*}
part=($(echo $clip_stem | tr "_" "\n"))
width=${part[1]}
height=${part[2]}
frame_rate=${part[3]}
# Reset previous run data
rm -f ./stats/h264/${clip_stem}.txt
# Data-rate range depends on input format
if [ ${width} -gt 640 ]; then
rate_start=800
rate_end=1500
rate_step=100
else
rate_start=100
rate_end=800
rate_step=100
fi
for (( rate=rate_start; rate<=rate_end; rate+=rate_step ))
do
echo "Encoding for $rate"
# Encode into ./<clip_name>_<width>_<height>_<frame_rate>_<rate>kbps.yuv
./bin/x264 --nal-hrd cbr --vbv-maxrate ${rate} --vbv-bufsize ${rate} \
--vbv-init 0.8 --bitrate ${rate} --fps ${frame_rate} \
--profile baseline --no-scenecut --keyint infinite --preset veryslow \
--input-res ${width}x${height} \
--tune psnr \
-o ./encoded_clips/h264/${clip_stem}_${rate}kbps.mkv ${filename} \
2> ./logs/h264/${clip_stem}_${rate}kbps.txt
# Decode the clip to a temporary file in order to compute PSNR and extract
# bitrate.
rm -f $tempyuvfile
encoded_rate=( `bin/ffmpeg -i ./encoded_clips/h264/${clip_stem}_${rate}kbps.mkv \
$tempyuvfile 2>&1 | awk '/bitrate:/ { print $6 }'` )
if expr $encoded_rate + 0 > /dev/null; then
# Compute the global PSNR.
psnr=$(./bin/psnr ${filename} $tempyuvfile ${width} ${height} 9999)
if [ $rate -ne $encoded_rate ]; then
# Rename the file to reflect the encoded datarate.
mv ./encoded_clips/h264/${clip_stem}_${rate}kbps.mkv \
./encoded_clips/h264/${clip_stem}_${encoded_rate}kbps.mkv
fi
echo "${encoded_rate} ${psnr}" >> ./stats/h264/${clip_stem}.txt
else
echo "Non-numeric bitrate $encoded_rate"
exit 1
fi
rm -f $tempyuvfile
done
done