| OLD | NEW |
| 1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
| 2 # Copyright (c) 2013 The WebRTC project authors. All Rights Reserved. | 2 # Copyright (c) 2013 The WebRTC project authors. All Rights Reserved. |
| 3 # | 3 # |
| 4 # Use of this source code is governed by a BSD-style license | 4 # Use of this source code is governed by a BSD-style license |
| 5 # that can be found in the LICENSE file in the root of the source | 5 # that can be found in the LICENSE file in the root of the source |
| 6 # tree. An additional intellectual property rights grant can be found | 6 # tree. An additional intellectual property rights grant can be found |
| 7 # in the file PATENTS. All contributing project authors may | 7 # in the file PATENTS. All contributing project authors may |
| 8 # be found in the AUTHORS file in the root of the source tree. | 8 # be found in the AUTHORS file in the root of the source tree. |
| 9 | 9 |
| 10 import optparse | |
| 11 import os | 10 import os |
| 12 import shutil | |
| 13 import subprocess | |
| 14 import sys | |
| 15 import tempfile | |
| 16 | 11 |
| 12 # This is just a forwarding script during the transition of renaming tools to |
| 13 # rtc_tools. To avoid breaking Chromium tests. |
| 17 | 14 |
| 18 SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__)) | 15 SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__)) |
| 19 | 16 RTC_TOOLS_DIR = os.path.abspath(os.path.join(SCRIPT_DIR, os.pardir, |
| 20 # Chrome browsertests will throw away stderr; avoid that output gets lost. | 17 'rtc_tools')) |
| 21 sys.stderr = sys.stdout | 18 os.system(os.path.join(RTC_TOOLS_DIR, 'compare_videos.py')) |
| 22 | |
| 23 | |
| 24 def _ParseArgs(): | |
| 25 """Registers the command-line options.""" | |
| 26 usage = 'usage: %prog [options]' | |
| 27 parser = optparse.OptionParser(usage=usage) | |
| 28 | |
| 29 parser.add_option('--label', type='string', default='MY_TEST', | |
| 30 help=('Label of the test, used to identify different ' | |
| 31 'tests. Default: %default')) | |
| 32 parser.add_option('--ref_video', type='string', | |
| 33 help='Reference video to compare with (YUV).') | |
| 34 parser.add_option('--test_video', type='string', | |
| 35 help=('Test video to be compared with the reference ' | |
| 36 'video (YUV).')) | |
| 37 parser.add_option('--frame_analyzer', type='string', | |
| 38 help='Path to the frame analyzer executable.') | |
| 39 parser.add_option('--barcode_decoder', type='string', | |
| 40 help=('Path to the barcode decoder script. By default, we ' | |
| 41 'will assume we can find it in barcode_tools/' | |
| 42 'relative to this directory.')) | |
| 43 parser.add_option('--ffmpeg_path', type='string', | |
| 44 help=('The path to where the ffmpeg executable is located. ' | |
| 45 'If omitted, it will be assumed to be present in the ' | |
| 46 'PATH with the name ffmpeg[.exe].')) | |
| 47 parser.add_option('--zxing_path', type='string', | |
| 48 help=('The path to where the zxing executable is located. ' | |
| 49 'If omitted, it will be assumed to be present in the ' | |
| 50 'PATH with the name zxing[.exe].')) | |
| 51 parser.add_option('--stats_file_ref', type='string', default='stats_ref.txt', | |
| 52 help=('Path to the temporary stats file to be created and ' | |
| 53 'used for the reference video file. ' | |
| 54 'Default: %default')) | |
| 55 parser.add_option('--stats_file_test', type='string', | |
| 56 default='stats_test.txt', | |
| 57 help=('Path to the temporary stats file to be created and ' | |
| 58 'used for the test video file. Default: %default')) | |
| 59 parser.add_option('--stats_file', type='string', | |
| 60 help=('DEPRECATED')) | |
| 61 parser.add_option('--yuv_frame_width', type='int', default=640, | |
| 62 help='Width of the YUV file\'s frames. Default: %default') | |
| 63 parser.add_option('--yuv_frame_height', type='int', default=480, | |
| 64 help='Height of the YUV file\'s frames. Default: %default') | |
| 65 options, _ = parser.parse_args() | |
| 66 | |
| 67 if options.stats_file: | |
| 68 options.stats_file_test = options.stats_file | |
| 69 print ('WARNING: Using deprecated switch --stats_file. ' | |
| 70 'The new flag is --stats_file_test.') | |
| 71 | |
| 72 if not options.ref_video: | |
| 73 parser.error('You must provide a path to the reference video!') | |
| 74 if not os.path.exists(options.ref_video): | |
| 75 parser.error('Cannot find the reference video at %s' % options.ref_video) | |
| 76 | |
| 77 if not options.test_video: | |
| 78 parser.error('You must provide a path to the test video!') | |
| 79 if not os.path.exists(options.test_video): | |
| 80 parser.error('Cannot find the test video at %s' % options.test_video) | |
| 81 | |
| 82 if not options.frame_analyzer: | |
| 83 parser.error('You must provide the path to the frame analyzer executable!') | |
| 84 if not os.path.exists(options.frame_analyzer): | |
| 85 parser.error('Cannot find frame analyzer executable at %s!' % | |
| 86 options.frame_analyzer) | |
| 87 return options | |
| 88 | |
| 89 def _DevNull(): | |
| 90 """On Windows, sometimes the inherited stdin handle from the parent process | |
| 91 fails. Workaround this by passing null to stdin to the subprocesses commands. | |
| 92 This function can be used to create the null file handler. | |
| 93 """ | |
| 94 return open(os.devnull, 'r') | |
| 95 | |
| 96 def DecodeBarcodesInVideo(options, path_to_decoder, video, stat_file): | |
| 97 # Run barcode decoder on the test video to identify frame numbers. | |
| 98 png_working_directory = tempfile.mkdtemp() | |
| 99 cmd = [ | |
| 100 sys.executable, | |
| 101 path_to_decoder, | |
| 102 '--yuv_file=%s' % video, | |
| 103 '--yuv_frame_width=%d' % options.yuv_frame_width, | |
| 104 '--yuv_frame_height=%d' % options.yuv_frame_height, | |
| 105 '--stats_file=%s' % stat_file, | |
| 106 '--png_working_dir=%s' % png_working_directory, | |
| 107 ] | |
| 108 if options.zxing_path: | |
| 109 cmd.append('--zxing_path=%s' % options.zxing_path) | |
| 110 if options.ffmpeg_path: | |
| 111 cmd.append('--ffmpeg_path=%s' % options.ffmpeg_path) | |
| 112 | |
| 113 | |
| 114 barcode_decoder = subprocess.Popen(cmd, stdin=_DevNull(), | |
| 115 stdout=sys.stdout, stderr=sys.stderr) | |
| 116 barcode_decoder.wait() | |
| 117 | |
| 118 shutil.rmtree(png_working_directory) | |
| 119 if barcode_decoder.returncode != 0: | |
| 120 print 'Failed to run barcode decoder script.' | |
| 121 return 1 | |
| 122 return 0 | |
| 123 | |
| 124 def main(): | |
| 125 """The main function. | |
| 126 | |
| 127 A simple invocation is: | |
| 128 ./webrtc/tools/barcode_tools/compare_videos.py | |
| 129 --ref_video=<path_and_name_of_reference_video> | |
| 130 --test_video=<path_and_name_of_test_video> | |
| 131 --frame_analyzer=<path_and_name_of_the_frame_analyzer_executable> | |
| 132 | |
| 133 Notice that the prerequisites for barcode_decoder.py also applies to this | |
| 134 script. The means the following executables have to be available in the PATH: | |
| 135 * zxing | |
| 136 * ffmpeg | |
| 137 """ | |
| 138 options = _ParseArgs() | |
| 139 | |
| 140 if options.barcode_decoder: | |
| 141 path_to_decoder = options.barcode_decoder | |
| 142 else: | |
| 143 path_to_decoder = os.path.join(SCRIPT_DIR, 'barcode_tools', | |
| 144 'barcode_decoder.py') | |
| 145 | |
| 146 if DecodeBarcodesInVideo(options, path_to_decoder, | |
| 147 options.ref_video, options.stats_file_ref) != 0: | |
| 148 return 1 | |
| 149 if DecodeBarcodesInVideo(options, path_to_decoder, | |
| 150 options.test_video, options.stats_file_test) != 0: | |
| 151 return 1 | |
| 152 | |
| 153 # Run frame analyzer to compare the videos and print output. | |
| 154 cmd = [ | |
| 155 options.frame_analyzer, | |
| 156 '--label=%s' % options.label, | |
| 157 '--reference_file=%s' % options.ref_video, | |
| 158 '--test_file=%s' % options.test_video, | |
| 159 '--stats_file_ref=%s' % options.stats_file_ref, | |
| 160 '--stats_file_test=%s' % options.stats_file_test, | |
| 161 '--width=%d' % options.yuv_frame_width, | |
| 162 '--height=%d' % options.yuv_frame_height, | |
| 163 ] | |
| 164 frame_analyzer = subprocess.Popen(cmd, stdin=_DevNull(), | |
| 165 stdout=sys.stdout, stderr=sys.stderr) | |
| 166 frame_analyzer.wait() | |
| 167 if frame_analyzer.returncode != 0: | |
| 168 print 'Failed to run frame analyzer.' | |
| 169 return 1 | |
| 170 | |
| 171 return 0 | |
| 172 | |
| 173 if __name__ == '__main__': | |
| 174 sys.exit(main()) | |
| OLD | NEW |