android: Add screen recording tool

build/android/pylib/android_commands.py

 # 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.
 
 """Provides an interface to communicate with the device via the adb command.
 
 Assumes adb binary is currently on system path.
 """
 
 import collections
 import datetime
 import logging
 import os
 import re
 import shlex
 import signal
 import subprocess
 import sys
 import tempfile
 import time
 
 import cmd_helper
 import constants
+import screenshot
 try:
   from pylib import pexpect
 except:
   pexpect = None
 
 sys.path.append(os.path.join(
     constants.DIR_SOURCE_ROOT, 'third_party', 'android_testrunner'))
 import adb_interface
 import am_instrument_parser
 import errors
@@ skipping 1537 matching lines @@
     except ValueError:
       if IsDeviceAttached(self._device):
         raise errors.DeviceUnresponsiveError('Device may be offline.')
 
       return False
 
   def TakeScreenshot(self, host_file):
     """Saves a screenshot image to |host_file| on the host.
 
     Args:
-      host_file: Absolute path to the image file to store on the host.
+      host_file: Absolute path to the image file to store on the host or None to
+                 use an autogenerated file name.
+
+    Returns:
+      Resulting host file name of the screenshot.
     """
-    host_dir = os.path.dirname(host_file)
-    if not os.path.exists(host_dir):
-      os.makedirs(host_dir)
-    device_file = '%s/screenshot.png' % self.GetExternalStorage()
-    self.RunShellCommand('/system/bin/screencap -p %s' % device_file)
-    self.PullFileFromDevice(device_file, host_file)
+    return screenshot.TakeScreenshot(self, host_file)
 
   def PullFileFromDevice(self, device_file, host_file):
     """Download |device_file| on the device from to |host_file| on the host.
 
     Args:
       device_file: Absolute path to the file to retrieve from the device.
       host_file: Absolute path to the file to store on the host.
     """
     assert self._adb.Pull(device_file, host_file)
     assert os.path.exists(host_file)
@@ skipping 63 matching lines @@
   """
   def __init__(self, output):
     self._output = output
 
   def write(self, data):
     data = data.replace('\r\r\n', '\n')
     self._output.write(data)
 
   def flush(self):
     self._output.flush()