Index: build/android/screenshot.py |
diff --git a/build/android/screenshot.py b/build/android/screenshot.py |
new file mode 100755 |
index 0000000000000000000000000000000000000000..097739fb62deb0667141e2fe16e9533f678757b0 |
--- /dev/null |
+++ b/build/android/screenshot.py |
@@ -0,0 +1,98 @@ |
+#!/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 screenshot |
+from pylib.device import device_errors |
+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 len(args) > 1: |
+ parser.error('Too many positional arguments.') |
+ host_file = args[0] if args else options.file |
+ |
+ if options.verbose: |
+ logging.getLogger().setLevel(logging.DEBUG) |
+ |
+ devices = device_utils.DeviceUtils.HealthyDevices() |
+ if options.device: |
+ device = next((d for d in devices if d == options.device), None) |
+ if not device: |
+ raise device_errors.DeviceUnreachableError(options.device) |
+ else: |
+ if len(devices) > 1: |
+ parser.error('Multiple devices are attached. ' |
+ 'Please specify device serial number with --device.') |
+ elif len(devices) == 1: |
+ device = devices[0] |
+ else: |
+ raise device_errors.NoDevicesError() |
+ |
+ if options.video: |
+ _CaptureVideo(device, host_file, options) |
+ else: |
+ _CaptureScreenshot(device, host_file) |
+ return 0 |
+ |
+ |
+if __name__ == '__main__': |
+ sys.exit(main()) |