Merge pull request #1808 from astrofrog/screenshot

Added a new GlueApplication.screenshot method

CHANGES.md

@@ -7,6 +7,9 @@ v0.14.0 (unreleased)
 * Change 'Export Session' dialog to offer to save with relative paths to data
   by default instead of absolute paths. [#1803]
 
+* Added a new method ``screenshot`` on ``GlueApplication`` to save a
+  screenshot of the current view. [#1808]
+
 * Show the active subset in the toolbar. [#1797]
 
 * Refactored the viewer class base classes:

glue/app/qt/application.py

@@ -1353,6 +1353,17 @@ def _choose_merge(data, others, suggested_label):
 
         return None, None
 
+    def screenshot(self, filename):
+        """
+        Save a screenshot of the current application window to a file.
+        """
+        image = QtGui.QImage(self.size(), QtGui.QImage.Format_RGB32)
+        painter = QtGui.QPainter(image)
+        flags = self.IgnoreMask | self.DrawWindowBackground | self.DrawChildren
+        self.render(painter, QtCore.QPoint(), QtGui.QRegion(), flags)
+        image.save(filename)
+        painter.end()
+
     def __gluestate__(self, context):
         state = super(GlueApplication, self).__gluestate__(context)
         state['tab_names'] = self.tab_names

glue/app/qt/tests/test_application.py

@@ -2,6 +2,7 @@
 
 from __future__ import absolute_import, division, print_function
 
+import os
 import sys
 
 import pytest
@@ -460,6 +461,12 @@ def test_deselect_tool_on_viewer_change(self):
 
         app.close()
 
+    def test_screenshot(self, tmpdir):
+        filename = tmpdir.join('screenshot.png').strpath
+        app = GlueApplication()
+        app.screenshot(filename)
+        assert os.path.exists(filename)
+
 
 def test_logger_close():