Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(92)

Unified Diff: webrtc/tools/compare_videos.py

Issue 2965593002: Move webrtc/{tools => rtc_tools} (Closed)
Patch Set: Adding back root changes Created 3 years, 6 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « webrtc/tools/class_usage.sh ('k') | webrtc/tools/converter/converter.h » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: webrtc/tools/compare_videos.py
diff --git a/webrtc/tools/compare_videos.py b/webrtc/tools/compare_videos.py
index c4a70c20fed23085c86a1344d4b8ddde95f1af63..fcfbc382405504d6047f387fb228024fa99083a8 100755
--- a/webrtc/tools/compare_videos.py
+++ b/webrtc/tools/compare_videos.py
@@ -7,168 +7,12 @@
# 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 shutil
-import subprocess
-import sys
-import tempfile
+# This is just a forwarding script during the transition of renaming tools to
+# rtc_tools. To avoid breaking Chromium tests.
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('--label', type='string', default='MY_TEST',
- help=('Label of the test, used to identify different '
- 'tests. Default: %default'))
- 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 compared with the reference '
- 'video (YUV).'))
- parser.add_option('--frame_analyzer', type='string',
- help='Path to the frame analyzer executable.')
- parser.add_option('--barcode_decoder', type='string',
- help=('Path to the barcode decoder script. By default, we '
- 'will assume we can find it in barcode_tools/'
- 'relative to this directory.'))
- parser.add_option('--ffmpeg_path', type='string',
- help=('The path to where the ffmpeg executable is located. '
- 'If omitted, it will be assumed to be present in the '
- 'PATH with the name ffmpeg[.exe].'))
- parser.add_option('--zxing_path', type='string',
- help=('The path to where the zxing executable is located. '
- 'If omitted, it will be assumed to be present in the '
- 'PATH with the name zxing[.exe].'))
- parser.add_option('--stats_file_ref', type='string', default='stats_ref.txt',
- help=('Path to the temporary stats file to be created and '
- 'used for the reference video file. '
- 'Default: %default'))
- parser.add_option('--stats_file_test', type='string',
- default='stats_test.txt',
- help=('Path to the temporary stats file to be created and '
- 'used for the test video file. Default: %default'))
- parser.add_option('--stats_file', type='string',
- help=('DEPRECATED'))
- 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, _ = parser.parse_args()
-
- if options.stats_file:
- options.stats_file_test = options.stats_file
- print ('WARNING: Using deprecated switch --stats_file. '
- 'The new flag is --stats_file_test.')
-
- 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 _DevNull():
- """On Windows, sometimes the inherited stdin handle from the parent process
- fails. Workaround this by passing null to stdin to the subprocesses commands.
- This function can be used to create the null file handler.
- """
- return open(os.devnull, 'r')
-
-def DecodeBarcodesInVideo(options, path_to_decoder, video, stat_file):
- # Run barcode decoder on the test video to identify frame numbers.
- png_working_directory = tempfile.mkdtemp()
- cmd = [
- sys.executable,
- path_to_decoder,
- '--yuv_file=%s' % video,
- '--yuv_frame_width=%d' % options.yuv_frame_width,
- '--yuv_frame_height=%d' % options.yuv_frame_height,
- '--stats_file=%s' % stat_file,
- '--png_working_dir=%s' % png_working_directory,
- ]
- if options.zxing_path:
- cmd.append('--zxing_path=%s' % options.zxing_path)
- if options.ffmpeg_path:
- cmd.append('--ffmpeg_path=%s' % options.ffmpeg_path)
-
-
- barcode_decoder = subprocess.Popen(cmd, stdin=_DevNull(),
- stdout=sys.stdout, stderr=sys.stderr)
- barcode_decoder.wait()
-
- shutil.rmtree(png_working_directory)
- if barcode_decoder.returncode != 0:
- print 'Failed to run barcode decoder script.'
- return 1
- return 0
-
-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()
-
- if options.barcode_decoder:
- path_to_decoder = options.barcode_decoder
- else:
- path_to_decoder = os.path.join(SCRIPT_DIR, 'barcode_tools',
- 'barcode_decoder.py')
-
- if DecodeBarcodesInVideo(options, path_to_decoder,
- options.ref_video, options.stats_file_ref) != 0:
- return 1
- if DecodeBarcodesInVideo(options, path_to_decoder,
- options.test_video, options.stats_file_test) != 0:
- return 1
-
- # Run frame analyzer to compare the videos and print output.
- cmd = [
- options.frame_analyzer,
- '--label=%s' % options.label,
- '--reference_file=%s' % options.ref_video,
- '--test_file=%s' % options.test_video,
- '--stats_file_ref=%s' % options.stats_file_ref,
- '--stats_file_test=%s' % options.stats_file_test,
- '--width=%d' % options.yuv_frame_width,
- '--height=%d' % options.yuv_frame_height,
- ]
- frame_analyzer = subprocess.Popen(cmd, stdin=_DevNull(),
- 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())
+RTC_TOOLS_DIR = os.path.abspath(os.path.join(SCRIPT_DIR, os.pardir,
+ 'rtc_tools'))
+os.system(os.path.join(RTC_TOOLS_DIR, 'compare_videos.py'))
« no previous file with comments | « webrtc/tools/class_usage.sh ('k') | webrtc/tools/converter/converter.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698