| #!/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, |
| os.pardir)) |
| |
| |
| def _RunCommand(argv, cwd=SRC_DIR, **kwargs): |
| logging.info('Running %r', argv) |
| subprocess.check_call(argv, cwd=cwd, **kwargs) |
| |
| |
| def _ParseArgs(): |
| parser = argparse.ArgumentParser(description='Start loopback video analysis.') |
| 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() |
| |
| 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']) |
| |
| tools_dir = os.path.join(SRC_DIR, 'tools-webrtc') |
| toolchain_dir = os.path.join(tools_dir, 'video_quality_toolchain') |
| |
| # Download ffmpeg and zxing. |
| download_script = os.path.join(tools_dir, 'download_tools.py') |
| _RunCommand([sys.executable, download_script, toolchain_dir]) |
| |
| # 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 ConvertVideo(input_video, output_video): |
| _RunCommand([ffmpeg_path, '-y', '-i', input_video, output_video]) |
| |
| ConvertVideo(test_video, test_video_yuv) |
| |
| reference_video = os.path.join(SRC_DIR, |
| 'resources', 'reference_video_640x360_30fps.y4m') |
| |
| reference_video_yuv = os.path.join(temp_dir, |
| 'reference_video_640x360_30fps.yuv') |
| |
| ConvertVideo(reference_video, reference_video_yuv) |
| |
| # Run compare script. |
| compare_script = os.path.join(SRC_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()) |
| |