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 |