| 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())
|
| -
|
|
|