Index: webrtc/tools/test_stubbed_loopback_video.py |
diff --git a/webrtc/tools/test_stubbed_loopback_video.py b/webrtc/tools/test_stubbed_loopback_video.py |
deleted file mode 100755 |
index 5f9e2bfac91485ccd1131d9cb81c26e1790c7aea..0000000000000000000000000000000000000000 |
--- a/webrtc/tools/test_stubbed_loopback_video.py |
+++ /dev/null |
@@ -1,135 +0,0 @@ |
-#!/usr/bin/env python |
-# Copyright (c) 2017 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. |
- |
-""" |
-This script is the wrapper that starts a loopback call with stubbed video in |
-and out. It then analyses the video quality of the output video against the |
-reference input video. |
- |
-It expect to be given the webrtc output build directory as the first argument |
-all other arguments are optional. |
- |
-It assumes you have a Android device plugged in. |
-""" |
- |
-import argparse |
-import logging |
-import os |
-import shutil |
-import subprocess |
-import sys |
-import tempfile |
- |
- |
-SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__)) |
-SRC_DIR = os.path.normpath(os.path.join(SCRIPT_DIR, os.pardir, os.pardir)) |
- |
- |
-def _RunCommand(argv, **kwargs): |
- logging.info('Running %r', argv) |
- subprocess.check_call(argv, **kwargs) |
- |
- |
-def _ParseArgs(): |
- parser = argparse.ArgumentParser(description='Start loopback video analysis.') |
- parser.add_argument('--source_dir', default=SRC_DIR, |
- help='The path to the WebRTC source directory. Default: %default.') |
- parser.add_argument('build_dir_android', |
- help='The path to the build directory for Android.') |
- parser.add_argument('--build_dir_x86', |
- help='The path to the build directory for building locally.') |
- parser.add_argument('--temp_dir', |
- help='A temporary directory to put the output.') |
- |
- args = parser.parse_args() |
- return args |
- |
- |
-def main(): |
- logging.basicConfig(level=logging.INFO) |
- |
- args = _ParseArgs() |
- |
- source_dir = args.source_dir |
- build_dir_android = args.build_dir_android |
- build_dir_x86 = args.build_dir_x86 |
- temp_dir = args.temp_dir |
- if not temp_dir: |
- temp_dir = tempfile.mkdtemp() |
- else: |
- if not os.path.exists(temp_dir): |
- os.makedirs(temp_dir) |
- |
- if not build_dir_x86: |
- build_dir_x86 = os.path.join(temp_dir, 'LocalBuild') |
- _RunCommand(['gn', 'gen', build_dir_x86]) |
- _RunCommand(['ninja', '-C', build_dir_x86, 'frame_analyzer']) |
- |
- toolchain_dir = os.path.join(source_dir, 'tools-webrtc', |
- 'video_quality_toolchain') |
- |
- # Download ffmpeg and zxing. |
- download_script = os.path.join(toolchain_dir, 'download.py') |
- _RunCommand([sys.executable, download_script]) |
- |
- # Run the Espresso code. |
- test_script = os.path.join(build_dir_android, |
- 'bin', 'run_AppRTCMobileTestStubbedVideoIO') |
- _RunCommand([sys.executable, test_script]) |
- |
- # Pull the output video. |
- test_video = os.path.join(temp_dir, 'test_video.y4m') |
- _RunCommand(['adb', 'pull', '/sdcard/output.y4m', test_video]) |
- |
- test_video_yuv = os.path.join(temp_dir, 'test_video.yuv') |
- |
- ffmpeg_path = os.path.join(toolchain_dir, 'linux', 'ffmpeg') |
- |
- def convert_video(input_video, output_video): |
- _RunCommand([ffmpeg_path, '-y', '-i', input_video, output_video]) |
- |
- convert_video(test_video, test_video_yuv) |
- |
- reference_video = os.path.join(source_dir, |
- 'resources', 'reference_video_640x360_30fps.y4m') |
- |
- reference_video_yuv = os.path.join(temp_dir, |
- 'reference_video_640x360_30fps.yuv') |
- |
- convert_video(reference_video, reference_video_yuv) |
- |
- # Run compare script. |
- compare_script = os.path.join(source_dir, 'webrtc', 'tools', |
- 'compare_videos.py') |
- zxing_path = os.path.join(toolchain_dir, 'linux', 'zxing') |
- |
- # The frame_analyzer binary should be built for local computer and not for |
- # Android |
- frame_analyzer = os.path.join(build_dir_x86, 'frame_analyzer') |
- |
- frame_width = 640 |
- frame_height = 360 |
- |
- stats_file_ref = os.path.join(temp_dir, 'stats_ref.txt') |
- stats_file_test = os.path.join(temp_dir, 'stats_test.txt') |
- |
- _RunCommand([ |
- sys.executable, compare_script, '--ref_video', reference_video_yuv, |
- '--test_video', test_video_yuv, '--yuv_frame_width', str(frame_width), |
- '--yuv_frame_height', str(frame_height), |
- '--stats_file_ref', stats_file_ref, |
- '--stats_file_test', stats_file_test, '--frame_analyzer', frame_analyzer, |
- '--ffmpeg_path', ffmpeg_path, '--zxing_path', zxing_path]) |
- |
- shutil.rmtree(temp_dir) |
- |
- |
-if __name__ == '__main__': |
- sys.exit(main()) |
- |