Added takeSnapshot() functionality

- Version 4.2.0
- PyDev project interpreter set to python 2.7
- Added screenshot examples
- Added PIL dependency for snapshot Images

Author: dtmilano

AndroidViewClient/.pydevproject

@@ -2,7 +2,7 @@
 <?eclipse-pydev version="1.0"?>
 
 <pydev_project>
-<pydev_property name="org.python.pydev.PYTHON_PROJECT_INTERPRETER">Default</pydev_property>
+<pydev_property name="org.python.pydev.PYTHON_PROJECT_INTERPRETER">python 2.7</pydev_property>
 <pydev_property name="org.python.pydev.PYTHON_PROJECT_VERSION">python 2.7</pydev_property>
 <pydev_pathproperty name="org.python.pydev.PROJECT_SOURCE_PATH">
 <path>/AndroidViewClient/src</path>

AndroidViewClient/examples/screenshot-monkeyrunner.py

@@ -0,0 +1,21 @@
+#! /usr/bin/env monkeyrunner
+'''
+Copyright (C) 2012  Diego Torres Milano
+Created on Set 5, 2013
+
+@author: diego
+'''
+
+
+import sys
+import os
+
+from com.android.monkeyrunner import MonkeyRunner, MonkeyDevice
+
+if len(sys.argv) < 2:
+    print >> sys.stderr, "usage: %s filename.png [serialno]" % sys.argv[0]
+    sys.exit(1)
+
+filename = sys.argv.pop(1)
+device = MonkeyRunner.waitForConnection()
+device.takeSnapshot().writeToFile(filename, 'PNG')

AndroidViewClient/examples/screenshot.py

@@ -0,0 +1,20 @@
+#! /usr/bin/env python
+'''
+Copyright (C) 2012  Diego Torres Milano
+Created on Aug 31, 2013
+
+@author: diego
+'''
+
+
+import sys
+import os
+
+from com.dtmilano.android.viewclient import ViewClient
+
+if len(sys.argv) < 2:
+    sys.exit("usage: %s filename.png [serialno]" % sys.argv[0])
+
+filename = sys.argv.pop(1)
+device, serialno = ViewClient.connectToDeviceOrExit(verbose=False)
+device.takeSnapshot().save(filename, 'PNG')

AndroidViewClient/src/com/dtmilano/android/adb/adbclient.py

@@ -1,8 +1,24 @@
 '''
+Copyright (C) 2012-2013  Diego Torres Milano
 Created on Dec 1, 2012
 
-@author: diego
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+
+@author: Diego Torres Milano
 '''
+
+__version__ = '4.2.0'
+
 import sys
 import socket
 import time
@@ -104,7 +120,7 @@ def __send(self, msg, checkok=True, reconnect=False):
                 self.__setTransport()
         else:
             self.checkConnected()
-        self.socket.send('%04x%s' % (len(msg), msg))
+        self.socket.send('%04X%s' % (len(msg), msg))
         if checkok:
             self.__checkOk()
         if reconnect:
@@ -113,13 +129,23 @@ def __send(self, msg, checkok=True, reconnect=False):
             self.__connect()
             self.__setTransport()
 
-    def __receive(self):
+    def __receive(self, nob=None):
         if DEBUG:
             print >> sys.stderr, "__receive()"
         self.checkConnected()
-        n = int(self.socket.recv(4), 16)
-        recv = self.socket.recv(n)
-        return recv
+        if nob is None:
+            nob = int(self.socket.recv(4), 16)
+        if DEBUG:
+            print >> sys.stderr, "    __receive: receiving", nob, "bytes"
+        recv = bytearray()
+        nr = 0
+        while nr < nob:
+            chunk = self.socket.recv(min((nob-nr), 4096))
+            recv.extend(chunk)
+            nr += len(chunk)
+        if DEBUG:
+            print >> sys.stderr, "    __receive: returning len=", len(recv)
+        return str(recv)
 
     def __checkOk(self):
         if DEBUG:
@@ -221,7 +247,8 @@ def shell(self, cmd=None):
             return sout
 
     def __getRestrictedScreen(self):
-        ''' Gets mRestrictedScreen values from dumpsys. This is a method to obtain display dimensions '''
+        ''' Gets C{mRestrictedScreen} values from dumpsys. This is a method to obtain display dimensions '''
+        
         rsRE = re.compile('\s*mRestrictedScreen=\((?P<x>\d+),(?P<y>\d+)\) (?P<w>\d+)x(?P<h>\d+)')
         for line in self.shell('dumpsys window').splitlines():
             m = rsRE.match(line)
@@ -278,6 +305,34 @@ def startActivity(self, component=None, flags=None, uri=None):
             cmd += ' %s' % uri
         self.shell(cmd)
 
+    def takeSnapshot(self, reconnect=False):
+        try:
+            from PIL import Image
+        except:
+            raise Exception("You have to install PIL to use takeSnapshot()")
+        self.__send('framebuffer:', checkok=True, reconnect=False)
+        import struct
+        #case 1: // version
+        #           return 12; // bpp, size, width, height, 4*(length, offset)
+        received = self.__receive(1 * 4 + 12 * 4)
+        (version, bpp, size, width, height, roffset, rlen, boffset, blen, goffset, glen, aoffset, alen) = struct.unpack('<' + 'L' * 13, received)
+        mode = [None]*4
+        mode[roffset/rlen] = 'R'
+        mode[boffset/blen] = 'B'
+        mode[goffset/glen] = 'G'
+        mode[aoffset/alen] = 'A'
+        mode = ''.join(mode)
+        if DEBUG:
+            print >> sys.stderr, "    takeSnapshot:", (version, bpp, size, width, height, roffset, rlen, boffset, blen, goffset, blen, aoffset, alen)
+        self.__send('\0', checkok=False, reconnect=False)
+        if DEBUG:
+            print >> sys.stderr, "    takeSnapshot: reading %d bytes" % (size)
+        received = self.__receive(size)
+        if reconnect:
+            self.__connect()
+            self.__setTransport()
+        return Image.frombuffer(mode, (width, height), received, 'raw', mode, 0, 1)
+    
     def touch(self, x, y, eventType=DOWN_AND_UP):
         self.shell('input tap %d %d' % (x, y))
 

AndroidViewClient/src/com/dtmilano/android/viewclient.py

@@ -18,7 +18,7 @@
 @author: Diego Torres Milano
 '''
 
-__version__ = '4.1.0'
+__version__ = '4.2.0'
 
 import sys
 import subprocess

AndroidViewClient/tools/culebra

@@ -18,7 +18,7 @@ ___________________/  /__/  /__/  /__/  /________________________________
 
 '''
 
-__version__ = '4.1.0'
+__version__ = '4.2.0'
 
 import re
 import sys

AndroidViewClient/tools/dump

@@ -6,7 +6,7 @@ Created on Feb 3, 2012
 @author: diego
 '''
 
-__version__ = '4.1.0'
+__version__ = '4.2.0'
 
 import sys
 import os