8400246fce
Now that these scripts are called from browser tests, we need to print everything on stdout since the tests will throw away stderr when invoking programs. I chose to assign sys.stderr to sys.stdout. Otherwise I would have missed stuff like parser.error, which print to stderr. The error message will get improved because the old code did not catch the case when the binary was missing, which lead to a very confusing error when that was the case. This gets fixed now. BUG= R=kjellander@webrtc.org Review URL: https://webrtc-codereview.appspot.com/1886004 git-svn-id: http://webrtc.googlecode.com/svn/trunk@4416 4adac7df-926f-26a2-2b94-8c16560cd09d
113 lines
4.1 KiB
Python
Executable File
113 lines
4.1 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# Copyright (c) 2013 The WebRTC 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.
|
|
|
|
import optparse
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
|
|
|
|
SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
# Chrome browsertests will throw away stderr; avoid that output gets lost.
|
|
sys.stderr = sys.stdout
|
|
|
|
|
|
def _ParseArgs():
|
|
"""Registers the command-line options."""
|
|
usage = 'usage: %prog [options]'
|
|
parser = optparse.OptionParser(usage=usage)
|
|
|
|
parser.add_option('--ref_video', type='string',
|
|
help='Reference video to compare with (YUV).')
|
|
parser.add_option('--test_video', type='string',
|
|
help=('Test video to be comared with the reference '
|
|
'video (YUV).'))
|
|
parser.add_option('--frame_analyzer', type='string',
|
|
help='Path to the frame analyzer executable.')
|
|
parser.add_option('--stats_file', type='string', default='stats.txt',
|
|
help=('Path to the temporary stats file to be created and '
|
|
'used. Default: %default'))
|
|
parser.add_option('--yuv_frame_width', type='int', default=640,
|
|
help='Width of the YUV file\'s frames. Default: %default')
|
|
parser.add_option('--yuv_frame_height', type='int', default=480,
|
|
help='Height of the YUV file\'s frames. Default: %default')
|
|
options, _args = parser.parse_args()
|
|
|
|
if not options.ref_video:
|
|
parser.error('You must provide a path to the reference video!')
|
|
if not os.path.exists(options.ref_video):
|
|
parser.error('Cannot find the reference video at %s' % options.ref_video)
|
|
|
|
if not options.test_video:
|
|
parser.error('You must provide a path to the test video!')
|
|
if not os.path.exists(options.test_video):
|
|
parser.error('Cannot find the test video at %s' % options.test_video)
|
|
|
|
if not options.frame_analyzer:
|
|
parser.error('You must provide the path to the frame analyzer executable!')
|
|
if not os.path.exists(options.frame_analyzer):
|
|
parser.error('Cannot find frame analyzer executable at %s!' %
|
|
options.frame_analyzer)
|
|
return options
|
|
|
|
|
|
def main():
|
|
"""The main function.
|
|
|
|
A simple invocation is:
|
|
./webrtc/tools/barcode_tools/compare_videos.py
|
|
--ref_video=<path_and_name_of_reference_video>
|
|
--test_video=<path_and_name_of_test_video>
|
|
--frame_analyzer=<path_and_name_of_the_frame_analyzer_executable>
|
|
|
|
Notice that the prerequisites for barcode_decoder.py also applies to this
|
|
script. The means the following executables have to be available in the PATH:
|
|
* zxing
|
|
* ffmpeg
|
|
"""
|
|
options = _ParseArgs()
|
|
|
|
# Run barcode decoder on the test video to identify frame numbers.
|
|
path_to_decoder = os.path.join(SCRIPT_DIR, 'barcode_tools',
|
|
'barcode_decoder.py')
|
|
cmd = [
|
|
sys.executable,
|
|
path_to_decoder,
|
|
'--yuv_file=%s' % options.test_video,
|
|
'--yuv_frame_width=%d' % options.yuv_frame_width,
|
|
'--yuv_frame_height=%d' % options.yuv_frame_height,
|
|
'--stats_file=%s' % options.stats_file,
|
|
]
|
|
barcode_decoder = subprocess.Popen(cmd, stdout=sys.stdout, stderr=sys.stderr)
|
|
barcode_decoder.wait()
|
|
if barcode_decoder.returncode != 0:
|
|
print 'Failed to run barcode decoder script.'
|
|
return 1
|
|
|
|
# Run frame analyzer to compare the videos and print output.
|
|
cmd = [
|
|
options.frame_analyzer,
|
|
'--reference_file=%s' % options.ref_video,
|
|
'--test_file=%s' % options.test_video,
|
|
'--stats_file=%s' % options.stats_file,
|
|
'--width=%d' % options.yuv_frame_width,
|
|
'--height=%d' % options.yuv_frame_height,
|
|
]
|
|
frame_analyzer = subprocess.Popen(cmd, stdout=sys.stdout, stderr=sys.stderr)
|
|
frame_analyzer.wait()
|
|
if frame_analyzer.returncode != 0:
|
|
print 'Failed to run frame analyzer.'
|
|
return 1
|
|
|
|
return 0
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main())
|