| OLD | NEW |
|---|
| (Empty) |
| 1 #!/usr/bin/env python | |
| 2 # Copyright (c) 2017 The WebRTC project authors. All Rights Reserved. | |
| 3 # | |
| 4 # Use of this source code is governed by a BSD-style license | |
| 5 # that can be found in the LICENSE file in the root of the source | |
| 6 # tree. An additional intellectual property rights grant can be found | |
| 7 # in the file PATENTS. All contributing project authors may | |
| 8 # be found in the AUTHORS file in the root of the source tree. | |
| 9 | |
| 10 """ | |
| 11 This script is the wrapper that starts a loopback call with stubbed video in | |
| 12 and out. It then analyses the video quality of the output video against the | |
| 13 reference input video. | |
| 14 | |
| 15 It expect to be given the webrtc output build directory as the first argument | |
| 16 all other arguments are optional. | |
| 17 | |
| 18 It assumes you have a Android device plugged in. | |
| 19 """ | |
| 20 | |
| 21 import argparse | |
| 22 import logging | |
| 23 import os | |
| 24 import shutil | |
| 25 import subprocess | |
| 26 import sys | |
| 27 import tempfile | |
| 28 | |
| 29 | |
| 30 SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__)) | |
| 31 SRC_DIR = os.path.normpath(os.path.join(SCRIPT_DIR, os.pardir, os.pardir)) | |
| 32 | |
| 33 | |
| 34 def _RunCommand(argv, **kwargs): | |
| 35 logging.info('Running %r', argv) | |
| 36 subprocess.check_call(argv, **kwargs) | |
| 37 | |
| 38 | |
| 39 def _ParseArgs(): | |
| 40 parser = argparse.ArgumentParser(description='Start loopback video analysis.') | |
| 41 parser.add_argument('--source_dir', default=SRC_DIR, | |
| 42 help='The path to the WebRTC source directory. Default: %default.') | |
| 43 parser.add_argument('build_dir_android', | |
| 44 help='The path to the build directory for Android.') | |
| 45 parser.add_argument('--build_dir_x86', | |
| 46 help='The path to the build directory for building locally.') | |
| 47 parser.add_argument('--temp_dir', | |
| 48 help='A temporary directory to put the output.') | |
| 49 | |
| 50 args = parser.parse_args() | |
| 51 return args | |
| 52 | |
| 53 | |
| 54 def main(): | |
| 55 logging.basicConfig(level=logging.INFO) | |
| 56 | |
| 57 args = _ParseArgs() | |
| 58 | |
| 59 source_dir = args.source_dir | |
| 60 build_dir_android = args.build_dir_android | |
| 61 build_dir_x86 = args.build_dir_x86 | |
| 62 temp_dir = args.temp_dir | |
| 63 if not temp_dir: | |
| 64 temp_dir = tempfile.mkdtemp() | |
| 65 else: | |
| 66 if not os.path.exists(temp_dir): | |
| 67 os.makedirs(temp_dir) | |
| 68 | |
| 69 if not build_dir_x86: | |
| 70 build_dir_x86 = os.path.join(temp_dir, 'LocalBuild') | |
| 71 _RunCommand(['gn', 'gen', build_dir_x86]) | |
| 72 _RunCommand(['ninja', '-C', build_dir_x86, 'frame_analyzer']) | |
| 73 | |
| 74 toolchain_dir = os.path.join(source_dir, 'tools-webrtc', | |
| 75 'video_quality_toolchain') | |
| 76 | |
| 77 # Download ffmpeg and zxing. | |
| 78 download_script = os.path.join(toolchain_dir, 'download.py') | |
| 79 _RunCommand([sys.executable, download_script]) | |
| 80 | |
| 81 # Run the Espresso code. | |
| 82 test_script = os.path.join(build_dir_android, | |
| 83 'bin', 'run_AppRTCMobileTestStubbedVideoIO') | |
| 84 _RunCommand([sys.executable, test_script]) | |
| 85 | |
| 86 # Pull the output video. | |
| 87 test_video = os.path.join(temp_dir, 'test_video.y4m') | |
| 88 _RunCommand(['adb', 'pull', '/sdcard/output.y4m', test_video]) | |
| 89 | |
| 90 test_video_yuv = os.path.join(temp_dir, 'test_video.yuv') | |
| 91 | |
| 92 ffmpeg_path = os.path.join(toolchain_dir, 'linux', 'ffmpeg') | |
| 93 | |
| 94 def convert_video(input_video, output_video): | |
| 95 _RunCommand([ffmpeg_path, '-y', '-i', input_video, output_video]) | |
| 96 | |
| 97 convert_video(test_video, test_video_yuv) | |
| 98 | |
| 99 reference_video = os.path.join(source_dir, | |
| 100 'resources', 'reference_video_640x360_30fps.y4m') | |
| 101 | |
| 102 reference_video_yuv = os.path.join(temp_dir, | |
| 103 'reference_video_640x360_30fps.yuv') | |
| 104 | |
| 105 convert_video(reference_video, reference_video_yuv) | |
| 106 | |
| 107 # Run compare script. | |
| 108 compare_script = os.path.join(source_dir, 'webrtc', 'tools', | |
| 109 'compare_videos.py') | |
| 110 zxing_path = os.path.join(toolchain_dir, 'linux', 'zxing') | |
| 111 | |
| 112 # The frame_analyzer binary should be built for local computer and not for | |
| 113 # Android | |
| 114 frame_analyzer = os.path.join(build_dir_x86, 'frame_analyzer') | |
| 115 | |
| 116 frame_width = 640 | |
| 117 frame_height = 360 | |
| 118 | |
| 119 stats_file_ref = os.path.join(temp_dir, 'stats_ref.txt') | |
| 120 stats_file_test = os.path.join(temp_dir, 'stats_test.txt') | |
| 121 | |
| 122 _RunCommand([ | |
| 123 sys.executable, compare_script, '--ref_video', reference_video_yuv, | |
| 124 '--test_video', test_video_yuv, '--yuv_frame_width', str(frame_width), | |
| 125 '--yuv_frame_height', str(frame_height), | |
| 126 '--stats_file_ref', stats_file_ref, | |
| 127 '--stats_file_test', stats_file_test, '--frame_analyzer', frame_analyzer, | |
| 128 '--ffmpeg_path', ffmpeg_path, '--zxing_path', zxing_path]) | |
| 129 | |
| 130 shutil.rmtree(temp_dir) | |
| 131 | |
| 132 | |
| 133 if __name__ == '__main__': | |
| 134 sys.exit(main()) | |
| 135 | |
| OLD | NEW |
|---|
Chromium Code Reviews
Issue 2681833007:
Move Android video quality loopback script. (Closed)
