d103f4ba4a
R=kjellander@webrtc.org, mandermo@webrtc.org TBR=perkj@webrtc.org BUG=chromium:685222 NOTRY=True Review-Url: https://codereview.webrtc.org/2695713002 Cr-Commit-Position: refs/heads/master@{#16647}
134 lines
4.1 KiB
Python
Executable File
134 lines
4.1 KiB
Python
Executable File
#!/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'])
|
|
|
|
toolchain_dir = os.path.join(SRC_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(SRC_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(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())
|
|
|