diff options
author | Picca Frédéric-Emmanuel <picca@synchrotron-soleil.fr> | 2017-08-18 14:48:52 +0200 |
---|---|---|
committer | Picca Frédéric-Emmanuel <picca@synchrotron-soleil.fr> | 2017-08-18 14:48:52 +0200 |
commit | f7bdc2acff3c13a6d632c28c4569690ab106eed7 (patch) | |
tree | 9d67cdb7152ee4e711379e03fe0546c7c3b97303 /silx/gui/_utils.py |
Import Upstream version 0.5.0+dfsg
Diffstat (limited to 'silx/gui/_utils.py')
-rw-r--r-- | silx/gui/_utils.py | 102 |
1 files changed, 102 insertions, 0 deletions
diff --git a/silx/gui/_utils.py b/silx/gui/_utils.py new file mode 100644 index 0000000..e29141f --- /dev/null +++ b/silx/gui/_utils.py @@ -0,0 +1,102 @@ +# coding: utf-8 +# /*########################################################################## +# +# Copyright (c) 2017 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. +""" + +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 != 'PySide': + 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 |