Add script for comparing video quality

This script makes it easier to run a simple command line
comparison between a captured YUV file and a reference video.

BUG=none
TEST=command line invocation
R=phoglund@webrtc.org

Review URL: https://webrtc-codereview.appspot.com/1320007

git-svn-id: http://webrtc.googlecode.com/svn/trunk/webrtc@4022 4adac7df-926f-26a2-2b94-8c16560cd09d

3 files changed, 111 insertions, 2 deletions

diff --git a/tools/barcode_tools/barcode_decoder.py b/tools/barcode_tools/barcode_decoder.py
index 40f7c1c1..f74fe7e3 100755
--- a/tools/barcode_tools/barcode_decoder.py
+++ b/tools/barcode_tools/barcode_decoder.py
@@ -251,7 +251,7 @@ def _main():
   """The main function.
 
   A simple invocation is:
-  ./tools/barcode_tolls/barcode_decoder.py
+  ./webrtc/tools/barcode_tools/barcode_decoder.py
   --yuv_file=<path_and_name_of_overlaid_yuv_video>
   --yuv_frame_width=640 --yuv_frame_height=480
   --stats_file=<path_and_name_to_stats_file>
diff --git a/tools/barcode_tools/barcode_encoder.py b/tools/barcode_tools/barcode_encoder.py
index 8c3a5bf7..3a8f3549 100755
--- a/tools/barcode_tools/barcode_encoder.py
+++ b/tools/barcode_tools/barcode_encoder.py
@@ -324,7 +324,7 @@ def _main():
   """The main function.
 
   A simple invocation will be:
-  ./tools/barcode_tools/barcode_encoder.py --barcode_height=32
+  ./webrtc/tools/barcode_tools/barcode_encoder.py --barcode_height=32
   --base_frame_width=352 --base_frame_height=288
   --base_yuv=<path_and_name_of_base_file>
   --output_yuv=<path and name_of_output_file>
diff --git a/tools/compare_videos.py b/tools/compare_videos.py
new file mode 100755
index 00000000..4b21b23a
--- /dev/null
+++ b/tools/compare_videos.py
@@ -0,0 +1,109 @@
+#!/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__))
+
+
+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 >> sys.stderr, '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.ref_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 >> sys.stderr, 'Failed to run frame analyzer.'
+    return 1
+
+  return 0
+
+if __name__ == '__main__':
+  sys.exit(main())