Windows: Capture all visible windows

Author: BoboTiG

CHANGELOG

@@ -2,11 +2,12 @@ History:
 
 <see Git checking messages for history>
 
-dev     2017/xx/xx
+dev     2018/xx/xx
       - removed support for Python 3.3
       - CI: build the documentation
       - doc: improvements and fixes (fix #37)
       - MSS: possibility to get the whole PNG raw bytes
+      - Windows: capture all visible windows (fix #28, #29)
 
 3.1.1   2017/11/27
       - MSS: add the 'mss' entry point

CHANGES.rst

@@ -1,45 +1,59 @@
-3.1.0 (2017-08-xx)
+3.1.2 (2018-01-xx)
 ==================
 
+tools.py
+--------
+- Changed signature of ``to_png(data, size, output)`` to ``to_png(data, size, output=None)``. If ``output`` is ``None``, the raw PNG bytes will be returned.
+
+
+3.1.1 (2017-11-27)
+==================
+
+__main__.py
+-----------
+- Added ``args`` argument to ``main()``
+
 base.py
 -------
-- Moved `ScreenShot` class to screenshot.py
+- Moved ``ScreenShot`` class to screenshot.py
 
 darwin.py
 ---------
-- Removed `get_infinity()` function
-- Removed `resize()` method. Use `_crop_width()` method instead.
+- Add ``CGPoint.__repr__()``
+- Add ``CGRect.__repr__()``
+- Add ``CGSize.__repr__()``
+- Removed ``get_infinity()`` function
 
 windows.py
 ----------
-- Added `scale_factor` property to `MSS` class
-- Added `scale()` method to `MSS` class
+- Added ``scale()`` method to ``MSS`` class
+- Added ``scale_factor`` property to ``MSS`` class
 
 
 3.0.0 (2017-07-06)
 ==================
 
 base.py
 -------
-- Added the `ScreenShot` class containing data for a given screen shot (support the Numpy array interface [`ScreenShot.__array_interface__`])
-- Add `shot()` method to `MSSBase`. It takes the same arguments as the `save()` method.
-- Renamed `get_pixels` to `grab`. It now returns a `ScreenShot` object.
-- Moved `to_png` method to `tools.py`. It is now a simple function.
-- Removed `enum_display_monitors()` method. Use `monitors` property instead.
-- Removed `monitors` attribute. Use `monitors` property instead.
-- Removed `width` attribute. Use `ScreenShot.size[0]` attribute or `ScreenShot.width` property instead.
-- Removed `height` attribute. Use `ScreenShot.size[1]` attribute or `ScreenShot.height` property instead.
-- Removed `image`. Use the `ScreenShot.raw` attribute or `ScreenShot.rgb` property instead.
-- Removed `bgra_to_rgb()` method. Use `ScreenShot.rgb` property instead.
+- Added the ``ScreenShot`` class containing data for a given screen shot (support the Numpy array interface [``ScreenShot.__array_interface__``])
+- Add ``shot()`` method to ``MSSBase``. It takes the same arguments as the ``save()`` method.
+- Renamed ``get_pixels`` to ``grab``. It now returns a ``ScreenShot`` object.
+- Moved ``to_png`` method to ``tools.py``. It is now a simple function.
+- Removed ``enum_display_monitors()`` method. Use ``monitors`` property instead.
+- Removed ``monitors`` attribute. Use ``monitors`` property instead.
+- Removed ``width`` attribute. Use ``ScreenShot.size[0]`` attribute or ``ScreenShot.width`` property instead.
+- Removed ``height`` attribute. Use ``ScreenShot.size[1]`` attribute or ``ScreenShot.height`` property instead.
+- Removed ``image``. Use the ``ScreenShot.raw`` attribute or ``ScreenShot.rgb`` property instead.
+- Removed ``bgra_to_rgb()`` method. Use ``ScreenShot.rgb`` property instead.
 
 darwin.py
 ---------
-- Removed `_crop_width()` method. Screen shots are now using the width setted by the OS (rounded to 16).
+- Removed ``_crop_width()`` method. Screen shots are now using the width setted by the OS (rounded to 16).
 
 exception.py
 ------------
-- Renamed `ScreenshotError` class to `ScreenShotError`
+- Renamed ``ScreenshotError`` class to ``ScreenShotError``
 
 tools.py
 --------
-- Changed signature of `to_png(data, monitor, output)` to `to_png(data, size, output)` where `size` is a `tuple(width, height)`
+- Changed signature of ``to_png(data, monitor, output)`` to ``to_png(data, size, output)`` where ``size`` is a ``tuple(width, height)``

LICENSE

@@ -1,5 +1,5 @@
 MIT License
-Copyright (c) 2016-2017, Mickaël 'Tiger-222' Schoentgen
+Copyright (c) 2016-2018, Mickaël 'Tiger-222' Schoentgen
 
 Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
 

docs/source/conf.py

@@ -23,7 +23,7 @@
 
 # General information about the project.
 project = 'Python MSS'
-copyright = "2013-2017, Mickaël 'Tiger-222' Schoentgen & contributors"
+copyright = "2013-2018, Mickaël 'Tiger-222' Schoentgen & contributors"
 author = 'Tiger-222'
 
 # The version info for the project you're documenting, acts as replacement for

docs/source/where.rst

@@ -6,8 +6,8 @@ This is a non exhaustive list where MSS is integrated or has inspired.
 Do not hesistate to `say Hello! <https://github.com/BoboTiG/python-mss/issues>`_ if you are using MSS too.
 
 
-Computer Vison
-==============
+AI, Computer Vison
+==================
 
 - `DoomPy <https://github.com/WNoxchi/DoomPy>`_ (Autonomous Anti-Demonic Combat Algorithms);
 - `Europilot <https://github.com/jsistla/eu-pilot>`_, a self-driving algorithm using Euro Truck Simulator (ETS2);

mss/__init__.py

@@ -16,7 +16,7 @@
 __version__ = '3.1.2'
 __author__ = "Mickaël 'Tiger-222' Schoentgen"
 __copyright__ = """
-    Copyright (c) 2013-2017, Mickaël 'Tiger-222' Schoentgen
+    Copyright (c) 2013-2018, Mickaël 'Tiger-222' Schoentgen
 
     Permission to use, copy, modify, and distribute this software and its
     documentation for any purpose and without fee or royalty is hereby

mss/__main__.py

@@ -40,16 +40,16 @@ def main(args=None):
     }
     if options.coordinates:
         try:
-            top, left, width, height = map(int, options.coordinates.split(','))
+            top, left, width, height = options.coordinates.split(',')
         except ValueError:
             print('Coordinates syntax: top, left, width, height')
             return 2
 
         kwargs['mon'] = {
-            'top': top,
-            'left': left,
-            'width': width,
-            'height': height,
+            'top': int(top),
+            'left': int(left),
+            'width': int(width),
+            'height': int(height),
         }
         if options.output == 'monitor-{mon}.png':
             kwargs['output'] = 'sct-{top}x{left}_{width}x{height}.png'

mss/tools.py

@@ -53,3 +53,5 @@ def to_png(data, size, output=None):
         fileh.write(b''.join(ihdr))
         fileh.write(b''.join(idat))
         fileh.write(b''.join(iend))
+
+    return None

mss/windows.py

@@ -15,6 +15,11 @@
 __all__ = ('MSS',)
 
 
+CAPTUREBLT = 0x40000000
+DIB_RGB_COLORS = 0
+SRCCOPY = 0x00CC0020
+
+
 class BITMAPINFOHEADER(ctypes.Structure):
     """ Information about the dimensions and color format of a DIB. """
 
@@ -196,10 +201,10 @@ def grab(self, monitor):
                                        monitor['width'], monitor['height'],
                                        srcdc,
                                        monitor['left'], monitor['top'],
-                                       0xcc0020)  # SRCCOPY
+                                       SRCCOPY | CAPTUREBLT)
 
             bits = ctypes.windll.gdi32.GetDIBits(
-                memdc, bmp, 0, monitor['height'], data, bmi, 0)
+                memdc, bmp, 0, monitor['height'], data, bmi, DIB_RGB_COLORS)
             if bits != monitor['height']:
                 del data
                 raise ScreenShotError('gdi32.GetDIBits() failed.', locals())