vpxenc.sh 9.69 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17
#!/bin/sh
##
##  Copyright (c) 2014 The WebM project authors. All Rights Reserved.
##
##  Use of this source code is governed by a BSD-style license
##  that can be found in the LICENSE file in the root of the source
##  tree. An additional intellectual property rights grant can be found
##  in the file PATENTS.  All contributing project authors may
##  be found in the AUTHORS file in the root of the source tree.
##
##  This file tests vpxenc using hantro_collage_w352h288.yuv as input. To add
##  new tests to this file, do the following:
##    1. Write a shell function (this is your test).
##    2. Add the function to vpxenc_tests (on a new line).
##
. $(dirname $0)/tools_common.sh

18
readonly TEST_FRAMES=10
19 20 21 22

# Environment check: Make sure input is available.
vpxenc_verify_environment() {
  if [ ! -e "${YUV_RAW_INPUT}" ]; then
23
    elog "The file ${YUV_RAW_INPUT##*/} must exist in LIBVPX_TEST_DATA_PATH."
24 25
    return 1
  fi
26 27 28 29 30 31 32
  if [ "$(vpxenc_can_encode_vp9)" = "yes" ]; then
    if [ ! -e "${Y4M_NOSQ_PAR_INPUT}" ]; then
      elog "The file ${Y4M_NOSQ_PAR_INPUT##*/} must exist in"
      elog "LIBVPX_TEST_DATA_PATH."
      return 1
    fi
  fi
33 34 35 36
  if [ -z "$(vpx_tool_path vpxenc)" ]; then
    elog "vpxenc not found. It must exist in LIBVPX_BIN_PATH or its parent."
    return 1
  fi
37 38 39
}

vpxenc_can_encode_vp8() {
40
  if [ "$(vp8_encode_available)" = "yes" ]; then
41 42 43 44 45
    echo yes
  fi
}

vpxenc_can_encode_vp9() {
46
  if [ "$(vp9_encode_available)" = "yes" ]; then
47 48 49 50
    echo yes
  fi
}

51 52 53 54 55 56 57 58
# Echo vpxenc command line parameters allowing use of
# hantro_collage_w352h288.yuv as input.
yuv_input_hantro_collage() {
  echo ""${YUV_RAW_INPUT}"
       --width="${YUV_RAW_INPUT_WIDTH}"
       --height="${YUV_RAW_INPUT_HEIGHT}""
}

59 60 61 62
y4m_input_non_square_par() {
  echo ""${Y4M_NOSQ_PAR_INPUT}""
}

63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88
# Echo default vpxenc real time encoding params. $1 is the codec, which defaults
# to vp8 if unspecified.
vpxenc_rt_params() {
  local readonly codec="${1:-vp8}"
  echo "--codec=${codec}
    --buf-initial-sz=500
    --buf-optimal-sz=600
    --buf-sz=1000
    --cpu-used=-5
    --end-usage=cbr
    --error-resilient=1
    --kf-max-dist=90000
    --lag-in-frames=0
    --max-intra-rate=300
    --max-q=56
    --min-q=2
    --noise-sensitivity=0
    --overshoot-pct=50
    --passes=1
    --profile=0
    --resize-allowed=0
    --rt
    --static-thresh=0
    --undershoot-pct=50"
}

89 90 91 92 93
# Wrapper function for running vpxenc with pipe input. Requires that
# LIBVPX_BIN_PATH points to the directory containing vpxenc. $1 is used as the
# input file path and shifted away. All remaining parameters are passed through
# to vpxenc.
vpxenc_pipe() {
94
  local readonly encoder="$(vpx_tool_path vpxenc)"
95 96
  local readonly input="$1"
  shift
97 98 99
  cat "${input}" | eval "${VPX_TEST_PREFIX}" "${encoder}" - \
    --test-decode=fatal \
    "$@" ${devnull}
100
}
101

102 103 104 105
# Wrapper function for running vpxenc. Requires that LIBVPX_BIN_PATH points to
# the directory containing vpxenc. $1 one is used as the input file path and
# shifted away. All remaining parameters are passed through to vpxenc.
vpxenc() {
106
  local readonly encoder="$(vpx_tool_path vpxenc)"
107
  local readonly input="$1"
108
  shift
109
  eval "${VPX_TEST_PREFIX}" "${encoder}" "${input}" \
110 111
    --test-decode=fatal \
    "$@" ${devnull}
112 113
}

114 115
vpxenc_vp8_ivf() {
  if [ "$(vpxenc_can_encode_vp8)" = "yes" ]; then
116
    local readonly output="${VPX_TEST_OUTPUT_DIR}/vp8.ivf"
117
    vpxenc $(yuv_input_hantro_collage) \
118
      --codec=vp8 \
119 120
      --limit="${TEST_FRAMES}" \
      --ivf \
121
      --output="${output}"
122 123 124 125 126

    if [ ! -e "${output}" ]; then
      elog "Output file does not exist."
      return 1
    fi
127 128 129
  fi
}

130 131 132 133
vpxenc_vp8_webm() {
  if [ "$(vpxenc_can_encode_vp8)" = "yes" ] && \
     [ "$(webm_io_available)" = "yes" ]; then
    local readonly output="${VPX_TEST_OUTPUT_DIR}/vp8.webm"
134
    vpxenc $(yuv_input_hantro_collage) \
135
      --codec=vp8 \
136
      --limit="${TEST_FRAMES}" \
137
      --output="${output}"
138 139 140 141 142 143 144 145

    if [ ! -e "${output}" ]; then
      elog "Output file does not exist."
      return 1
    fi
  fi
}

146 147 148 149
vpxenc_vp8_webm_rt() {
  if [ "$(vpxenc_can_encode_vp8)" = "yes" ] && \
     [ "$(webm_io_available)" = "yes" ]; then
    local readonly output="${VPX_TEST_OUTPUT_DIR}/vp8_rt.webm"
150 151 152
    vpxenc $(yuv_input_hantro_collage) \
      $(vpxenc_rt_params vp8) \
      --output="${output}"
153 154 155 156 157 158 159
    if [ ! -e "${output}" ]; then
      elog "Output file does not exist."
      return 1
    fi
  fi
}

160 161 162 163
vpxenc_vp8_webm_2pass() {
  if [ "$(vpxenc_can_encode_vp8)" = "yes" ] && \
     [ "$(webm_io_available)" = "yes" ]; then
    local readonly output="${VPX_TEST_OUTPUT_DIR}/vp8.webm"
164
    vpxenc $(yuv_input_hantro_collage) \
165
      --codec=vp8 \
166 167
      --limit="${TEST_FRAMES}" \
      --output="${output}" \
168
      --passes=2
169 170 171 172 173 174 175 176 177 178 179 180 181 182

    if [ ! -e "${output}" ]; then
      elog "Output file does not exist."
      return 1
    fi
  fi
}

vpxenc_vp8_webm_lag10_frames20() {
  if [ "$(vpxenc_can_encode_vp8)" = "yes" ] && \
     [ "$(webm_io_available)" = "yes" ]; then
    local readonly lag_total_frames=20
    local readonly lag_frames=10
    local readonly output="${VPX_TEST_OUTPUT_DIR}/vp8_lag10_frames20.webm"
183
    vpxenc $(yuv_input_hantro_collage) \
184
      --codec=vp8 \
185 186 187 188
      --limit="${lag_total_frames}" \
      --lag-in-frames="${lag_frames}" \
      --output="${output}" \
      --auto-alt-ref=1 \
189
      --passes=2
190 191 192 193 194 195 196 197

    if [ ! -e "${output}" ]; then
      elog "Output file does not exist."
      return 1
    fi
  fi
}

198
vpxenc_vp8_ivf_piped_input() {
199
  if [ "$(vpxenc_can_encode_vp8)" = "yes" ]; then
200
    local readonly output="${VPX_TEST_OUTPUT_DIR}/vp8_piped_input.ivf"
201 202 203 204 205
    vpxenc_pipe $(yuv_input_hantro_collage) \
      --codec=vp8 \
      --limit="${TEST_FRAMES}" \
      --ivf \
      --output="${output}"
206 207 208 209 210

    if [ ! -e "${output}" ]; then
      elog "Output file does not exist."
      return 1
    fi
211 212 213
  fi
}

214 215 216
vpxenc_vp9_ivf() {
  if [ "$(vpxenc_can_encode_vp9)" = "yes" ]; then
    local readonly output="${VPX_TEST_OUTPUT_DIR}/vp9.ivf"
217
    vpxenc $(yuv_input_hantro_collage) \
218
      --codec=vp9 \
219
      --limit="${TEST_FRAMES}" \
220
      --ivf \
221
      --output="${output}"
222 223 224 225 226

    if [ ! -e "${output}" ]; then
      elog "Output file does not exist."
      return 1
    fi
227 228 229
  fi
}

230 231 232 233
vpxenc_vp9_webm() {
  if [ "$(vpxenc_can_encode_vp9)" = "yes" ] && \
     [ "$(webm_io_available)" = "yes" ]; then
    local readonly output="${VPX_TEST_OUTPUT_DIR}/vp9.webm"
234
    vpxenc $(yuv_input_hantro_collage) \
235
      --codec=vp9 \
236
      --limit="${TEST_FRAMES}" \
237
      --output="${output}"
238 239 240 241 242

    if [ ! -e "${output}" ]; then
      elog "Output file does not exist."
      return 1
    fi
243 244 245
  fi
}

246 247 248 249
vpxenc_vp9_webm_rt() {
  if [ "$(vpxenc_can_encode_vp9)" = "yes" ] && \
     [ "$(webm_io_available)" = "yes" ]; then
    local readonly output="${VPX_TEST_OUTPUT_DIR}/vp9_rt.webm"
250 251 252
    vpxenc $(yuv_input_hantro_collage) \
      $(vpxenc_rt_params vp9) \
      --output="${output}"
253 254 255 256 257 258 259 260

    if [ ! -e "${output}" ]; then
      elog "Output file does not exist."
      return 1
    fi
  fi
}

261
vpxenc_vp9_webm_2pass() {
262 263 264
  if [ "$(vpxenc_can_encode_vp9)" = "yes" ] && \
     [ "$(webm_io_available)" = "yes" ]; then
    local readonly output="${VPX_TEST_OUTPUT_DIR}/vp9.webm"
265
    vpxenc $(yuv_input_hantro_collage) \
266
      --codec=vp9 \
267 268
      --limit="${TEST_FRAMES}" \
      --output="${output}" \
269
      --passes=2
270 271 272 273 274

    if [ ! -e "${output}" ]; then
      elog "Output file does not exist."
      return 1
    fi
275 276 277
  fi
}

278
vpxenc_vp9_ivf_lossless() {
279
  if [ "$(vpxenc_can_encode_vp9)" = "yes" ]; then
280
    local readonly output="${VPX_TEST_OUTPUT_DIR}/vp9_lossless.ivf"
281
    vpxenc $(yuv_input_hantro_collage) \
282
      --codec=vp9 \
283 284 285
      --limit="${TEST_FRAMES}" \
      --ivf \
      --output="${output}" \
286
      --lossless=1
287 288 289 290 291 292 293 294 295 296 297

    if [ ! -e "${output}" ]; then
      elog "Output file does not exist."
      return 1
    fi
  fi
}

vpxenc_vp9_ivf_minq0_maxq0() {
  if [ "$(vpxenc_can_encode_vp9)" = "yes" ]; then
    local readonly output="${VPX_TEST_OUTPUT_DIR}/vp9_lossless_minq0_maxq0.ivf"
298
    vpxenc $(yuv_input_hantro_collage) \
299
      --codec=vp9 \
300 301 302 303
      --limit="${TEST_FRAMES}" \
      --ivf \
      --output="${output}" \
      --min-q=0 \
304
      --max-q=0
305 306 307 308 309

    if [ ! -e "${output}" ]; then
      elog "Output file does not exist."
      return 1
    fi
310 311 312
  fi
}

313 314 315 316 317 318
vpxenc_vp9_webm_lag10_frames20() {
  if [ "$(vpxenc_can_encode_vp9)" = "yes" ] && \
     [ "$(webm_io_available)" = "yes" ]; then
    local readonly lag_total_frames=20
    local readonly lag_frames=10
    local readonly output="${VPX_TEST_OUTPUT_DIR}/vp9_lag10_frames20.webm"
319
    vpxenc $(yuv_input_hantro_collage) \
320
      --codec=vp9 \
321 322 323 324
      --limit="${lag_total_frames}" \
      --lag-in-frames="${lag_frames}" \
      --output="${output}" \
      --passes=2 \
325
      --auto-alt-ref=1
326 327 328 329 330 331 332 333

    if [ ! -e "${output}" ]; then
      elog "Output file does not exist."
      return 1
    fi
  fi
}

334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350
# TODO(fgalligan): Test that DisplayWidth is different than video width.
vpxenc_vp9_webm_non_square_par() {
  if [ "$(vpxenc_can_encode_vp9)" = "yes" ] && \
     [ "$(webm_io_available)" = "yes" ]; then
    local readonly output="${VPX_TEST_OUTPUT_DIR}/vp9_non_square_par.webm"
    vpxenc $(y4m_input_non_square_par) \
      --codec=vp9 \
      --limit="${TEST_FRAMES}" \
      --output="${output}"

    if [ ! -e "${output}" ]; then
      elog "Output file does not exist."
      return 1
    fi
  fi
}

351 352
vpxenc_tests="vpxenc_vp8_ivf
              vpxenc_vp8_webm
353
              vpxenc_vp8_webm_rt
354 355
              vpxenc_vp8_webm_2pass
              vpxenc_vp8_webm_lag10_frames20
356
              vpxenc_vp8_ivf_piped_input
357 358
              vpxenc_vp9_ivf
              vpxenc_vp9_webm
359
              vpxenc_vp9_webm_rt
360
              vpxenc_vp9_webm_2pass
361
              vpxenc_vp9_ivf_lossless
362
              vpxenc_vp9_ivf_minq0_maxq0
363 364
              vpxenc_vp9_webm_lag10_frames20
              vpxenc_vp9_webm_non_square_par"
365 366

run_tests vpxenc_verify_environment "${vpxenc_tests}"