from helper import unittest, PillowTestCase, hopper
from test_imageqt import PillowQtTestCase

from PIL import ImageQt, Image


if ImageQt.qt_is_installed:
    from PIL.ImageQt import QImage, QPixmap

    try:
        from PyQt5 import QtGui
        from PyQt5.QtWidgets import QWidget, QHBoxLayout, QLabel, QApplication
        QT_VERSION = 5
    except (ImportError, RuntimeError):
        try:
            from PyQt4 import QtGui
            from PyQt4.QtGui import QWidget, QHBoxLayout, QLabel, QApplication
            QT_VERSION = 4
        except (ImportError, RuntimeError):
            from PySide import QtGui
            from PySide.QtGui import QWidget, QHBoxLayout, QLabel, QApplication
            QT_VERSION = 4
            

class TestToQImage(PillowQtTestCase, PillowTestCase):

    def test_sanity(self):
        PillowQtTestCase.setUp(self)
        for mode in ('RGB', 'RGBA', 'L', 'P', '1'):
            src = hopper(mode)
            data = ImageQt.toqimage(src)

            self.assertIsInstance(data, QImage)
            self.assertFalse(data.isNull())

            # reload directly from the qimage
            rt = ImageQt.fromqimage(data)
            if mode in ('L', 'P', '1'):
                self.assert_image_equal(rt, src.convert('RGB'))
            else:
                self.assert_image_equal(rt, src)

            if mode == '1':
                # BW appears to not save correctly on QT4 and QT5
                # kicks out errors on console:
                # libpng warning: Invalid color type/bit depth combination in IHDR
                # libpng error: Invalid IHDR data
                continue
            
            # Test saving the file
            tempfile = self.tempfile('temp_{}.png'.format(mode))
            data.save(tempfile)

            # Check that it actually worked. 
            reloaded = Image.open(tempfile)
            # Gray images appear to come back in palette mode.
            # They're roughly equivalent
            if QT_VERSION == 4 and mode == 'L':
                src = src.convert('P')
            self.assert_image_equal(reloaded, src)


    def test_segfault(self):
        PillowQtTestCase.setUp(self)

        app = QApplication([])
        ex = Example()


if ImageQt.qt_is_installed:
    class Example(QWidget):

        def __init__(self):
            super(Example, self).__init__()

            img = hopper().resize((1000,1000))

            qimage = ImageQt.ImageQt(img)

            pixmap1 = QtGui.QPixmap.fromImage(qimage)

            hbox = QHBoxLayout(self)

            lbl = QLabel(self)
            # Segfault in the problem
            lbl.setPixmap(pixmap1.copy())




def main():
    app = QApplication(sys.argv)
    ex = Example()
    sys.exit(app.exec_())




if __name__ == '__main__':
    unittest.main()