1 files changed, 104 insertions, 0 deletions
diff --git a/silx/gui/utils/_image.py b/silx/gui/utils/_image.py
new file mode 100644
index 0000000..260aac3
--- /dev/null
+++ b/silx/gui/utils/_image.py
@@ -0,0 +1,104 @@
+# coding: utf-8
+# /*##########################################################################
+#
+# Copyright (c) 2017-2018 European Synchrotron Radiation Facility
+#
+# 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:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+#
+# ###########################################################################*/
+"""This module provides convenient functions to use with Qt objects.
+
+It provides:
+- conversion between numpy and QImage:
+  :func:`convertArrayToQImage`, :func:`convertQImageToArray`
+"""
+
+from __future__ import division
+
+
+__authors__ = ["T. Vincent"]
+__license__ = "MIT"
+__date__ = "16/01/2017"
+
+
+import sys
+import numpy
+
+from .. import qt
+
+
+def convertArrayToQImage(image):
+    """Convert an array-like RGB888 image to a QImage.
+
+    The created QImage is using a copy of the array data.
+
+    Limitation: Only supports RGB888 format.
+
+    :param image: Array-like image data
+    :type image: numpy.ndarray of uint8 of dimension HxWx3
+    :return: Corresponding Qt image
+    :rtype: QImage
+    """
+    # Possible extension: add a format argument to support more formats
+
+    image = numpy.array(image, copy=False, order='C', dtype=numpy.uint8)
+
+    height, width, depth = image.shape
+    assert depth == 3
+
+    qimage = qt.QImage(
+        image.data,
+        width,
+        height,
+        image.strides[0],  # bytesPerLine
+        qt.QImage.Format_RGB888)
+
+    return qimage.copy()  # Making a copy of the image and its data
+
+
+def convertQImageToArray(image):
+    """Convert a RGB888 QImage to a numpy array.
+
+    Limitation: Only supports RGB888 format.
+    If QImage is not RGB888 it gets converted to this format.
+
+    :param QImage: The QImage to convert.
+    :return: The image array
+    :rtype: numpy.ndarray of uint8 of shape HxWx3
+    """
+    # Possible extension: avoid conversion to support more formats
+
+    if image.format() != qt.QImage.Format_RGB888:
+        # Convert to RGB888 if needed
+        image = image.convertToFormat(qt.QImage.Format_RGB888)
+
+    ptr = image.bits()
+    if qt.BINDING not in ('PySide', 'PySide2'):
+        ptr.setsize(image.byteCount())
+        if qt.BINDING == 'PyQt4' and sys.version_info[0] == 2:
+            ptr = ptr.asstring()
+    elif sys.version_info[0] == 3:  # PySide with Python3
+        ptr = ptr.tobytes()
+
+    array = numpy.fromstring(ptr, dtype=numpy.uint8)
+
+    # Lines are 32 bits aligned: remove padding bytes
+    array = array.reshape(image.height(), -1)[:, :image.width() * 3]
+    array.shape = image.height(), image.width(), 3
+    return array