| Index: build/android/pylib/screenshot.py
|
| diff --git a/build/android/pylib/screenshot.py b/build/android/pylib/screenshot.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..0fcc59048ec425aab0d7ba5124cdeeacf44e9c36
|
| --- /dev/null
|
| +++ b/build/android/pylib/screenshot.py
|
| @@ -0,0 +1,99 @@
|
| +# Copyright 2013 The Chromium Authors. All rights reserved.
|
| +# Use of this source code is governed by a BSD-style license that can be
|
| +# found in the LICENSE file.
|
| +
|
| +import logging
|
| +import os
|
| +import tempfile
|
| +import time
|
| +
|
| +from pylib import cmd_helper
|
| +from pylib import device_signal
|
| +from pylib.device import device_errors
|
| +
|
| +# TODO(jbudorick) Remove once telemetry gets switched over.
|
| +import pylib.android_commands
|
| +import pylib.device.device_utils
|
| +
|
| +
|
| +class VideoRecorder(object):
|
| + """Records a screen capture video from an Android Device (KitKat or newer).
|
| +
|
| + Args:
|
| + device: DeviceUtils instance.
|
| + host_file: Path to the video file to store on the host.
|
| + megabits_per_second: Video bitrate in megabits per second. Allowed range
|
| + from 0.1 to 100 mbps.
|
| + size: Video frame size tuple (width, height) or None to use the device
|
| + default.
|
| + rotate: If True, the video will be rotated 90 degrees.
|
| + """
|
| + def __init__(self, device, megabits_per_second=4, size=None,
|
| + rotate=False):
|
| + # TODO(jbudorick) Remove once telemetry gets switched over.
|
| + if isinstance(device, pylib.android_commands.AndroidCommands):
|
| + device = pylib.device.device_utils.DeviceUtils(device)
|
| + self._device = device
|
| + self._device_file = (
|
| + '%s/screen-recording.mp4' % device.GetExternalStoragePath())
|
| + self._recorder = None
|
| + self._recorder_stdout = None
|
| + self._is_started = False
|
| +
|
| + self._args = ['adb']
|
| + if str(self._device):
|
| + self._args += ['-s', str(self._device)]
|
| + self._args += ['shell', 'screenrecord', '--verbose']
|
| + self._args += ['--bit-rate', str(megabits_per_second * 1000 * 1000)]
|
| + if size:
|
| + self._args += ['--size', '%dx%d' % size]
|
| + if rotate:
|
| + self._args += ['--rotate']
|
| + self._args += [self._device_file]
|
| +
|
| + def Start(self):
|
| + """Start recording video."""
|
| + self._recorder_stdout = tempfile.mkstemp()[1]
|
| + self._recorder = cmd_helper.Popen(
|
| + self._args, stdout=open(self._recorder_stdout, 'w'))
|
| + if not self._device.GetPids('screenrecord'):
|
| + raise RuntimeError('Recording failed. Is your device running Android '
|
| + 'KitKat or later?')
|
| +
|
| + def IsStarted(self):
|
| + if not self._is_started:
|
| + for line in open(self._recorder_stdout):
|
| + self._is_started = line.startswith('Content area is ')
|
| + if self._is_started:
|
| + break
|
| + return self._is_started
|
| +
|
| + def Stop(self):
|
| + """Stop recording video."""
|
| + os.remove(self._recorder_stdout)
|
| + self._is_started = False
|
| + if not self._recorder:
|
| + return
|
| + if not self._device.KillAll('screenrecord', signum=device_signal.SIGINT,
|
| + quiet=True):
|
| + logging.warning('Nothing to kill: screenrecord was not running')
|
| + self._recorder.wait()
|
| +
|
| + def Pull(self, host_file=None):
|
| + """Pull resulting video file from the device.
|
| +
|
| + Args:
|
| + host_file: Path to the video file to store on the host.
|
| + Returns:
|
| + Output video file name on the host.
|
| + """
|
| + # TODO(jbudorick): Merge filename generation with the logic for doing so in
|
| + # DeviceUtils.
|
| + host_file_name = (
|
| + host_file
|
| + or 'screen-recording-%s.mp4' % time.strftime('%Y%m%dT%H%M%S',
|
| + time.localtime()))
|
| + host_file_name = os.path.abspath(host_file_name)
|
| + self._device.PullFile(self._device_file, host_file_name)
|
| + self._device.RunShellCommand('rm -f "%s"' % self._device_file)
|
| + return host_file_name
|
|
|