| #!/bin/sh |
| # |
| # Generate one or more settings files for codec comparision tests. |
| # |
| # Usage: |
| # |
| # generate-settings [--list] |
| # |
| # With --list, will list the config directories that would be generated. |
| # Without --list, will create the config directories and put an encoder |
| # and a measurer within each. |
| set -e |
| |
| list=0 |
| finished=0 |
| while [ $finished == 0 ]; do |
| case $1 in |
| --list) |
| list=1 |
| shift |
| ;; |
| --*) |
| echo "Unknown option $1" |
| exit 1 |
| ;; |
| *) |
| finished=1 |
| ;; |
| esac |
| done |
| |
| TEMPFILE=$(mktemp ./tempXXXXencoder) |
| |
| extension() { |
| if [ $CODEC == "vp8" ]; then |
| echo "webm" |
| elif [ $CODEC == "h264" ]; then |
| echo "mkv" |
| else |
| echo "Unknown codec $CODEC" |
| exit 1 |
| fi |
| } |
| |
| vp8encoder() { |
| # Generate the encoder file. Note that the fingerprint on this is the ID of the |
| # encoder instance, so don't make changes here lightly! |
| cat << EoF > $TEMPFILE |
| #!/bin/sh |
| # Generated encoder for encoding VP8 at $rate kbps. |
| # Command line format: ./encoder <filename> <framerate> <width> <height> |
| # The script must be executed within the directory it's placed in. |
| filename=\$1 |
| frame_rate=\$2 |
| width=\$3 |
| height=\$4 |
| rate=$rate |
| |
| basename=\$(basename \$filename .yuv) |
| |
| ../../bin/vpxenc --lag-in-frames=0 --target-bitrate=\${rate} --kf-min-dist=3000 \ |
| --kf-max-dist=3000 --cpu-used=0 --fps=\${frame_rate}/1 --static-thresh=0 \ |
| --token-parts=1 --drop-frame=0 --end-usage=cbr --min-q=2 --max-q=56 \ |
| --undershoot-pct=100 --overshoot-pct=15 --buf-sz=1000 \ |
| --buf-initial-sz=800 --buf-optimal-sz=1000 --max-intra-rate=1200 \ |
| --resize-allowed=0 --drop-frame=0 --passes=1 --good --noise-sensitivity=0 \ |
| -w \${width} -h \${height} \${filename} --codec=vp8 \ |
| -o ./\$basename.webm \ |
| |
| EoF |
| } |
| |
| h264encoder() { |
| # Generate the encoder file. Note that the fingerprint on this is the ID of the |
| # encoder instance, so don't make changes here lightly! |
| cat << EoF > $TEMPFILE |
| #!/bin/sh |
| # Generated encoder for encoding H.264 at $rate kbps. |
| # Command line format: ./encoder <filename> <framerate> <width> <height> |
| # The script must be executed within the directory it's placed in. |
| filename=\$1 |
| frame_rate=\$2 |
| width=\$3 |
| height=\$4 |
| rate=$rate |
| |
| basename=\$(basename \$filename .yuv) |
| |
| ../../bin/x264 --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 ./\$basename.mkv \$filename |
| EoF |
| } |
| |
| encoder() { |
| if [ $CODEC == "vp8" ]; then |
| vp8encoder |
| else |
| h264encoder |
| fi |
| } |
| |
| measurer() { |
| cat << EoF > $1/measurer |
| #!/bin/sh |
| # Generated measurer for an encode of $CODEC at target rate $rate kbps. |
| # Command line format: measurer filename framerate width heigth |
| # |
| set -e |
| |
| filename=\$1 |
| frame_rate=\$2 |
| width=\$3 |
| height=\$4 |
| extension=$(extension) |
| |
| basename=\$(basename \$filename .yuv) |
| |
| rm -f tempyuvfile.yuv |
| encoded_rate=( \$(../../bin/ffmpeg -i ./\$basename.\$extension \ |
| tempyuvfile.yuv 2>&1 | awk '/bitrate:/ { print \$6 }') ) |
| psnr=\$(../../bin/psnr \$filename tempyuvfile.yuv \$width \$height 9999) |
| cat << EoF2 > \$basename.results |
| target_rate=$rate |
| encoded_rate=\$encoded_rate |
| psnr=\$psnr |
| EoF2 |
| EoF |
| } |
| |
| for (( rate=200; rate<=1600; rate+=200 )); do |
| for CODEC in "vp8" "h264"; do |
| if [ ! -d $CODEC ]; then |
| mkdir $CODEC |
| fi |
| encoder |
| DIR=$(md5sum $TEMPFILE | cut -b1-12) |
| if [ $list == 1 ]; then |
| # List the dirs that would be generated |
| echo $CODEC/$DIR |
| else |
| # Move the encoder file into the workdir for this encoding. |
| if [ ! -d $CODEC/$DIR ]; then |
| mkdir $CODEC/$DIR |
| fi |
| mv $TEMPFILE $CODEC/$DIR/encoder |
| measurer $CODEC/$DIR |
| chmod +x $CODEC/$DIR/encoder |
| chmod +x $CODEC/$DIR/measurer |
| fi |
| done |
| done |
| |