blob: fb1aee1d985aaedef0c47c22605328ff4ca969bd [file] [log] [blame]
#!/usr/bin/env python
# Copyright (c) 2012 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.
"""Takes a screenshot or a screen video capture from an Android device."""
import logging
import optparse
import os
import sys
from pylib import android_commands
from pylib import screenshot
from pylib.device import device_utils
def _PrintMessage(heading, eol='\n'):
sys.stdout.write('%s%s' % (heading, eol))
sys.stdout.flush()
def _CaptureScreenshot(device, host_file):
host_file = device.TakeScreenshot(host_file)
_PrintMessage('Screenshot written to %s' % os.path.abspath(host_file))
def _CaptureVideo(device, host_file, options):
size = tuple(map(int, options.size.split('x'))) if options.size else None
recorder = screenshot.VideoRecorder(device,
megabits_per_second=options.bitrate,
size=size,
rotate=options.rotate)
try:
recorder.Start()
_PrintMessage('Recording. Press Enter to stop...', eol='')
raw_input()
finally:
recorder.Stop()
host_file = recorder.Pull(host_file)
_PrintMessage('Video written to %s' % os.path.abspath(host_file))
def main():
# Parse options.
parser = optparse.OptionParser(description=__doc__,
usage='screenshot.py [options] [filename]')
parser.add_option('-d', '--device', metavar='ANDROID_DEVICE', help='Serial '
'number of Android device to use.', default=None)
parser.add_option('-f', '--file', help='Save result to file instead of '
'generating a timestamped file name.', metavar='FILE')
parser.add_option('-v', '--verbose', help='Verbose logging.',
action='store_true')
video_options = optparse.OptionGroup(parser, 'Video capture')
video_options.add_option('--video', help='Enable video capturing. Requires '
'Android KitKat or later', action='store_true')
video_options.add_option('-b', '--bitrate', help='Bitrate in megabits/s, '
'from 0.1 to 100 mbps, %default mbps by default.',
default=4, type='float')
video_options.add_option('-r', '--rotate', help='Rotate video by 90 degrees.',
default=False, action='store_true')
video_options.add_option('-s', '--size', metavar='WIDTHxHEIGHT',
help='Frame size to use instead of the device '
'screen size.', default=None)
parser.add_option_group(video_options)
(options, args) = parser.parse_args()
if options.verbose:
logging.getLogger().setLevel(logging.DEBUG)
if not options.device and len(android_commands.GetAttachedDevices()) > 1:
parser.error('Multiple devices are attached. '
'Please specify device serial number with --device.')
if len(args) > 1:
parser.error('Too many positional arguments.')
host_file = args[0] if args else options.file
device = device_utils.DeviceUtils(options.device)
if options.video:
_CaptureVideo(device, host_file, options)
else:
_CaptureScreenshot(device, host_file)
return 0
if __name__ == '__main__':
sys.exit(main())