HYTOS

# coding: utf-8

import sys

import os.path

try:

    from PyQt5.QtCore import Qt, QRectF, pyqtSignal, QT_VERSION_STR, QPointF

    from PyQt5.QtGui import QImage, QPixmap, QPainterPath, QPainter, QCursor, QPen, QBrush, QColor, QTransform

    from PyQt5.QtWidgets import QGraphicsView, QGraphicsScene, QFileDialog, QApplication

except ImportError:

    try:

        from PyQt4.QtCore import *

        from PyQt4.QtGui import *

    except ImportError:

        raise ImportError("ImageViewerQt: Requires PyQt5 or PyQt4.")

sys.path.insert(0, os.path.dirname(os.path.realpath(__file__)) + '\\Commands')

import DefaultCommand

sys.path.insert(0, os.path.dirname(os.path.realpath(__file__)) + '\\Shapes')

from EngineeringLineItem import QEngineeringLineItem

from SymbolSvgItem import SymbolSvgItem

__author__ = "Marcel Goldschen-Ohm <marcel.goldschen@gmail.com>"

__version__ = '0.9.0'

class QtImageViewer(QGraphicsView):

    """ PyQt image viewer widget for a QPixmap in a QGraphicsView scene with mouse zooming and panning.

    Displays a QImage or QPixmap (QImage is internally converted to a QPixmap).

    To display any other image format, you must first convert it to a QImage or QPixmap.

    Some useful image format conversion utilities:

        qimage2ndarray: NumPy ndarray <==> QImage    (https://github.com/hmeine/qimage2ndarray)

        ImageQt: PIL Image <==> QImage  (https://github.com/python-pillow/Pillow/blob/master/PIL/ImageQt.py)

    Mouse interaction:

        Left mouse button drag: Pan image.

        Right mouse button drag: Zoom box.

        Right mouse button doubleclick: Zoom to show entire image.

    """

    # Mouse button signals emit image scene (x, y) coordinates.

    # !!! For image (row, column) matrix indexing, row = y and column = x.

    leftMouseButtonPressed = pyqtSignal(float, float)

    rightMouseButtonPressed = pyqtSignal(float, float)

    leftMouseButtonMoved = pyqtSignal(float, float)

    rightMouseButtonMoved = pyqtSignal(float, float)

    leftMouseButtonReleased = pyqtSignal(float, float)

    rightMouseButtonReleased = pyqtSignal(float, float)

    leftMouseButtonDoubleClicked = pyqtSignal(float, float)

    rightMouseButtonDoubleClicked = pyqtSignal(float, float)

    #itemRemoved = pyqtSignal(QGraphicsItem)

    startPointChanged = pyqtSignal(float, float)

    '''

        @history    2018.06.27  Jeongwoo    Change zoom rule (Qt.KeepAspectRatioByExpanding → Qt.KeepAspectRatio)

    '''

    def __init__(self, mainWindow = None):

        QGraphicsView.__init__(self)

        self.mainWindow = mainWindow

        # Image is displayed as a QPixmap in a QGraphicsScene attached to this QGraphicsView.

        self.command = None

        self.scene = QGraphicsScene(self)

        self.setScene(self.scene)

        self.scene.setBackgroundBrush(Qt.gray)

        self.crosshairPos = None

        self.isPressCtrl = False

        self.scaleFactor = 1.0

        self.numScheduledScalings = 0

        self.isOriginalPointSelected = False

        # Store a local handle to the scene's current image pixmap.

        self._pixmapHandle = None

        # Image aspect ratio mode.

        # !!! ONLY applies to full image. Aspect ratio is always ignored when zooming.

        #   Qt.IgnoreAspectRatio: Scale image to fit viewport.

        #   Qt.KeepAspectRatio: Scale image to fit inside viewport, preserving aspect ratio.

        #   Qt.KeepAspectRatioByExpanding: Scale image to fill the viewport, preserving aspect ratio.

        self.aspectRatioMode = Qt.KeepAspectRatio

        # Scroll bar behaviour.

        #   Qt.ScrollBarAlwaysOff: Never shows a scroll bar.

        #   Qt.ScrollBarAlwaysOn: Always shows a scroll bar.

        #   Qt.ScrollBarAsNeeded: Shows a scroll bar only when zoomed.

        self.setHorizontalScrollBarPolicy(Qt.ScrollBarAsNeeded)

        self.setVerticalScrollBarPolicy(Qt.ScrollBarAsNeeded)

        # Stack of QRectF zoom boxes in scene coordinates.

        self.zoomStack = []

        self.setRenderHint(QPainter.Antialiasing)

        self.setAcceptDrops(True)  # enable drop

        # Flags for enabling/disabling mouse interaction.

        self.canZoom = True

        self.canPan = True

        self.setMouseTracking(True)

        self.command = None

        # set currentAttribute

        self.currentAttribute = ''

    '''

        @brief      Return Pixmap Handler

        @author     Jeongwoo

        @date       2018.06.11

    '''

    def getPixmapHandle(self):

        return self._pixmapHandle

    '''

        @brief      Use Default ImageViewer Command

        @author     Jeongwoo

        @date       18.04.10

        @history    .

    '''

    def useDefaultCommand(self):

        """ Use Default Command

        """

        self.command = DefaultCommand.DefaultCommand(self)

    def hasImage(self):

        """ Returns whether or not the scene contains an image pixmap.

        """

        return self._pixmapHandle is not None

    def clearImage(self):

        """ Removes the current image pixmap from the scene if it exists.

        """

        if self.hasImage():

            self.scene.removeItem(self._pixmapHandle)

            self._pixmapHandle = None

    def pixmap(self):

        """ Returns the scene's current image pixmap as a QPixmap, or else None if no image exists.

        :rtype: QPixmap | None

        """

        if self.hasImage():

            return self._pixmapHandle.pixmap()

        return None

    def image(self):

        """ Returns the scene's current image pixmap as a QImage, or else None if no image exists.

        :rtype: QImage | None

        """

        if self.hasImage():

            return self._pixmapHandle.pixmap().toImage()

        return None

    def setImage(self, image):

        """ Set the scene's current image pixmap to the input QImage or QPixmap.

        Raises a RuntimeError if the input image has type other than QImage or QPixmap.

        :type image: QImage | QPixmap

        """

        try:

            if type(image) is QPixmap:

                pixmap = image

            elif type(image) is QImage:

                pixmap = QPixmap.fromImage(image)

            else:

                raise RuntimeError("ImageViewer.setImage: Argument must be a QImage or QPixmap.")

            self.clearImage()

            self.scene.clear()

            if self.hasImage():

                self._pixmapHandle.setPixmap(pixmap)

            else:

                self._pixmapHandle = self.scene.addPixmap(pixmap)

            self.setSceneRect(QRectF(pixmap.rect()))  # Set scene size to image size.

            self.updateViewer()

        except Exception as ex:

            print('error occured({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno))

    '''

        @brief  open a image file selected by user

        @author 

        @date

    '''

    def loadImageFromFile(self, folder='', fileName=""):

        import cv2

        """ Load an image from file.

        Without any arguments, loadImageFromFile() will popup a file dialog to choose the image file.

        With a fileName argument, loadImageFromFile(fileName) will attempt to load the specified image file directly.

        """

        try:

            if len(fileName) == 0:

                options = QFileDialog.Options()

                options |= QFileDialog.DontUseNativeDialog

                if QT_VERSION_STR[0] == '4':

                    fileName = QFileDialog.getOpenFileName(self, "Open image file", os.getcwd() if folder == '' else folder, "Image files(*.png *.jpg)", options=options)

                elif QT_VERSION_STR[0] == '5':

                    fileName, dummy = QFileDialog.getOpenFileName(self, "Open image file", os.getcwd() if folder == '' else folder, "Image files(*.png *.jpg)", options=options)

            if len(fileName) and os.path.isfile(fileName):

                cvImg = cv2.cvtColor(cv2.imread(fileName), cv2.COLOR_BGR2GRAY)

                blur = cv2.GaussianBlur(cvImg, (5,5),0)

                cvImg = cv2.threshold(cvImg, 0, 255, cv2.THRESH_BINARY+cv2.THRESH_OTSU)[1]

                bytesPerLine = cvImg.shape[1]

                image = QImage(cvImg.data, cvImg.shape[1], cvImg.shape[0], bytesPerLine, QImage.Format_Indexed8)

                self.setImage(image)

        except Exception as ex:

            print('error occured({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno))

        return fileName

    '''

        @history    2018.06.27  Jeongwoo    Change zoom rule (Qt.KeepAspectRatioByExpanding → Qt.KeepAspectRatio)

    '''

    def updateViewer(self):

        """ Show current zoom (if showing entire image, apply current aspect ratio mode).

        """

        if not self.hasImage():

            return

        if len(self.zoomStack):# and self.sceneRect().contains(self.zoomStack[-1]):

            self.fitInView(self.zoomStack[-1], Qt.KeepAspectRatio)  # Show zoomed rect (ignore aspect ratio).

        else:

            self.zoomStack = []  # Clear the zoom stack (in case we got here because of an invalid zoom).

            self.fitInView(self.sceneRect(), self.aspectRatioMode)  # Show entire image (use current aspect ratio mode).

    def zoomImageInit(self):

        if self.hasImage():

            self.zoomStack = []

            self.updateViewer()

            self.setCursor(QCursor(Qt.ArrowCursor))

    '''

        @brief      Zoom in & out image

        @author     Jeongwoo

        @date       -

        @history    18.04.11    Jeongwoo    add parameter 'adjust' (@ref ResultTreeWidget.itemClickEvent(self, item, columnNo))

    '''

    def zoomImage(self, isZoomIn, event, adjust = 1):

        """ Zoom in & out

        """

        HALF_SIZE = 300

        clickPos = event.pos()

        scenePos1 = self.mapToScene(clickPos.x() - HALF_SIZE//adjust, clickPos.y() - HALF_SIZE//adjust)

        scenePos2 = self.mapToScene(clickPos.x() + HALF_SIZE//adjust, clickPos.y() + HALF_SIZE//adjust)

        if isZoomIn:

            zoomArea = QRectF(QPointF(scenePos1.x() if scenePos1.x() > 0 else 0, scenePos1.y() if scenePos1.y() > 0 else 0), QPointF(scenePos2.x(), scenePos2.y()))

            #self.fitInView(zoomArea, Qt.KeepAspectRatioByExpanding)

            viewBBox = self.zoomStack[-1] if len(self.zoomStack) else self.sceneRect()

            selectionBBox = zoomArea.intersected(viewBBox)

            self.scene.setSelectionArea(QPainterPath())  # Clear current selection area.

            if selectionBBox.width() > HALF_SIZE*2 and selectionBBox.height() > HALF_SIZE*2:

                if selectionBBox.isValid() and (selectionBBox != viewBBox):

                    self.zoomStack.append(selectionBBox)

                    self.updateViewer()

        else:

            self.scene.setSelectionArea(QPainterPath())  # Clear current selection area.

            if len(self.zoomStack):

                self.zoomStack.pop()

            self.updateViewer()

    def resizeEvent(self, event):

        """ Maintain current zoom on resize.

        """

        self.updateViewer()

    '''

        @brief  mouse move event

    '''

    def mouseMoveEvent(self, event):

        try:

            if self.command is not None:

                scenePos = self.mapToScene(event.pos())

                self.command.execute(['mouseMoveEvent', event, scenePos])

                if self.command.isTreated == True: return

        except Exception as ex:

            print('error occured({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno))

        QGraphicsView.mouseMoveEvent(self, event)

    '''

        @brief      

        @author     

        @date       

        @history    block clear selection when right mouse button is clicked

    '''

    def mousePressEvent(self, event):

        try:

            if self.command is not None:

                scenePos = self.mapToScene(event.pos())

                self.command.execute(['mousePressEvent', event, scenePos])

                if self.command.isTreated == True: return

        except Exception as ex:

            print('error occured({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno))

        if event.button() != Qt.RightButton:

            QGraphicsView.mousePressEvent(self, event)

    def mouseReleaseEvent(self, event):

        """ Stop mouse pan or zoom mode (apply zoom if valid).

        """

        try:

            if self.command is not None:

                scenePos = self.mapToScene(event.pos())

                instance = self.command.execute(['mouseReleaseEvent', event, scenePos])

                if instance is not None:

                    self.scene.addItem(instance)

                if self.command is not None and self.command.isTreated == True: 

                    # SelectAttribute 경우만 추가

                    if self.command.name == "SelectAttribute":

                        items = self.scene.selectedItems()

                        if self.command.resultItem is not None and items is not None and len(items) == 1:

                            attrItem = self.command.resultItem

                            self.command.resultItem.attribute = self.currentAttribute

                            # 기존 연결되있는 다른 SymbolItem의 Attr 제거

                            items[0].removeAttr(attrItem)

                            # add attr

                            items[0].attrs.append(attrItem)

                            #resultPropertyTable refresh

                            self.mainWindow.refreshResultPropertyTableWidget()

                    elif self.command.name == 'Default' and self.command.isCopy:

                        return

                    self.command = DefaultCommand.DefaultCommand(self)

                    cursor = QCursor(Qt.ArrowCursor)

                    QApplication.instance().setOverrideCursor(cursor)

                    return

        except Exception as ex:

            print('error occured({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno))

        QGraphicsView.mouseReleaseEvent(self, event)

        #self.mainWindow.refreshResultPropertyTableWidget()

    def mouseDoubleClickEvent(self, event):

        """ Show entire image.

        """

        scenePos = self.mapToScene(event.pos())

        if self.command is not None:

            instance = self.command.execute(['mouseDoubleClickEvent', event, scenePos])

            if self.command.isTreated == True: return

        if event.button() == Qt.LeftButton:

            self.leftMouseButtonDoubleClicked.emit(scenePos.x(), scenePos.y())

        elif event.button() == Qt.RightButton:

            if self.canZoom:

                self.zoomStack = []  # Clear zoom stack.

                self.updateViewer()

            self.rightMouseButtonDoubleClicked.emit(scenePos.x(), scenePos.y())

        QGraphicsView.mouseDoubleClickEvent(self, event)

    '''

        @brief      key press event

        @author     Jeongwoo

        @date       2018.??.??

        @history    send escape key event to command

    '''

    def keyPressEvent(self, event):

        try:

            if event.key() == Qt.Key_Control:

                self.isPressCtrl = True

            elif event.key() == Qt.Key_Delete:

                for item in self.scene.selectedItems():

                    item.transfer.onRemoved.emit(item)

            elif event.key() == Qt.Key_Escape:

                if self.command is not None:

                    self.command.execute(['keyPressEvent', event, []])

            else:

                if self.command is not None:

                    self.command.execute(['keyPressEvent', event, []])

            QGraphicsView.keyPressEvent(self, event)

        except Exception as ex:

            print('error occured({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno))

    '''

        @brief  key release event

        @author Jeongwoo

        @date   2018.??.??

    '''

    def keyReleaseEvent(self, event):

        if event.key() == Qt.Key_Control:

            self.isPressCtrl = False

        elif event.key() == Qt.Key_Delete:

            pass

        QGraphicsView.keyReleaseEvent(self, event)

    def wheelEvent(self, event):

        if self.isPressCtrl == True:

            if self.canZoom and self.hasImage():

                numDegrees = event.angleDelta() / 8

                if numDegrees is not None:

                    if numDegrees.y() > 0:

                        self.zoomImage(True, event)

                    elif numDegrees.y() < 0:

                        self.zoomImage(False, event)

                #print("Zoomable")

                #numDegrees = event.angleDelta().y() // 8

                #numSteps = numDegrees // 15

                #self.numScheduledScalings = self.numScheduledScalings + numSteps

                #if self.numScheduledScalings * numSteps < 0:

                #    self.numScheduledScalings = numSteps

                #self.scaleFactor = 1.0 + (self.numScheduledScalings / 300.0)

                #print("scaleFactor : " + str(self.scaleFactor))

                #self.scale(self.scaleFactor, self.scaleFactor)

        else:

            super().wheelEvent(event)

    def drawForeground(self, painter, rect):

        image = self.image()

        if image is not None:

            width = image.width()

            height = image.height()

            if self.crosshairPos is not None:

                pen = QPen()

                pen.setColor(QColor(180, 180, 180))

                pen.setStyle(Qt.DashLine)

                pen.setWidthF(0.3)

                painter.setPen(pen)

                painter.drawLine(self.crosshairPos.x(), 0, self.crosshairPos.x(), height)#Vertical

                painter.drawLine(0, self.crosshairPos.y(), width, self.crosshairPos.y())#Horizontal

            #else:

            #    painter.eraseRect(QRectF(0, 0, width, height))

    '''

        @brief      draw background

        @author     humkyung

        @date       2018.07.23

    '''        

    def drawBackground(self, painter, rect):

        QGraphicsView.drawBackground(self, painter, rect) 

        '''

        gridSize = max(int(rect.width() / 100), int(rect.height() / 100))

        left = int(rect.left()) - (int(rect.left()) % gridSize)

        top = int(rect.top()) - (int(rect.top()) % gridSize)

        pen = QPen()

        pen.setColor(Qt.black)

        pen.setWidthF(1.5)

        painter.setPen(pen)

        points = []

        for x in range(left, int(rect.right()), gridSize):

            for y in range(top, int(rect.bottom()), gridSize):

                painter.drawPoint(QPointF(x, y))

        '''

    '''

        @history    2018.06.11  Jeongwoo    Change method to manage guideline items

    '''

    GUIDELINE_ITEMS = []

    def showGuideline(self, isShow):

        image = self.image()

        width = image.width()

        height = image.height()

        pen = QPen()

        pen.setColor(QColor(180, 180, 180))

        pen.setStyle(Qt.DashLine)

        pen.setWidthF(0.5)

        if isShow:

            verticalLine = self.scene.addLine(width/2, 0, width/2, height, pen)

            horizontalLine = self.scene.addLine(0, height/2, width, height/2, pen)

            self.GUIDELINE_ITEMS.clear()

            self.GUIDELINE_ITEMS.append(verticalLine)

            self.GUIDELINE_ITEMS.append(horizontalLine)

        else:

            items = self.scene.items()

            for item in self.GUIDELINE_ITEMS:

                if item in items:

                    self.scene.removeItem(item)

            self.GUIDELINE_ITEMS.clear()

    '''

        @brief  drag enter event

        @author humkyung

        @date   2018.04.17

    '''

    def dragEnterEvent(self, event):

        event.acceptProposedAction()

    '''

        @brief  drag move event

        @author humkyung

        @date   2018.04.17

    '''

    def dragMoveEvent(self, event):

        event.acceptProposedAction()

    '''

        @brief  drop event

        @author humkyung

        @date   2018.04.17

        @history    2018.06.08  Jeongwoo    Add Paramter on SymbolSvgItem.buildItem()

    '''

    def dropEvent(self, event):

        from AppDocData import AppDocData

        import symbol

        scenePos = self.mapToScene(event.pos())

        svgFileName = event.mimeData().text()

        svg = self.createSymbolObject(svgFileName)

        self.matchSymbolToLine(svg, scenePos)

        event.acceptProposedAction()

    '''

        @brief  drop create Symbol

        @author kyouho

        @date   2018.07.27

    '''

    def createSymbolObject(self, svgFileName):

        from AppDocData import AppDocData

        import symbol

        symbol = AppDocData.instance().getSymbolByQuery('name', svgFileName)

        svgFilePath = os.path.join(AppDocData.instance().getCurrentProject().getSvgFilePath(), symbol.getType(), svgFileName+'.svg')

        svg = SymbolSvgItem.createItem(symbol.getType(), svgFilePath)

        connPts = None

        strConnPts = symbol.getConnectionPoint()

        if strConnPts is not None:

            connPts = [(float(x.split(',')[0]), float(x.split(',')[1])) for x in strConnPts.split('/')]

        svg.buildItem(svgFileName, symbol.getType(), 0, None, None, [float(x) for x in symbol.getOriginalPoint().split(',')], connPts, symbol.getBaseSymbol(), symbol.getAdditionalSymbol(), symbol.getHasInstrumentLabel())

        return svg

    '''

        @brief  match symbol to line

        @author kyouho

        @date   2018.07.27

    '''

    def matchSymbolToLine(self, svg, scenePos):

        matches = [item for item in self.scene.items() if (type(item) is QEngineeringLineItem) and (item.distanceTo((scenePos.x(), scenePos.y())) < 20)]

        if len(matches) == 1:

            matches[0].insertSymbol(svg, scenePos)

        else:

            transform = QTransform()

            transform.translate(scenePos.x() - svg.symbolOrigin[0], scenePos.y() - svg.symbolOrigin[1])

            svg.setTransform(transform)

            svg.loc = [scenePos.x() - svg.symbolOrigin[0], scenePos.y() - svg.symbolOrigin[1]]

            svg.size = [svg.boundingRect().width(), svg.boundingRect().height()]

            self.scene.addItem(svg)

        svg.transfer.onRemoved.connect(self.mainWindow.itemRemoved)

        self.scene.clearFocus()

        for item in self.scene.selectedItems():

            item.setSelected(False)

        self.setFocus()

        svg.setSelected(True)

        self.scene.setFocusItem(svg)

    '''

        @brief  find item by uid (SymbolSvgItem 기반, QEngineeringConnectorItem 제외, QEngineeringLineItem 포함)

        @author kyouho

        @date   2018.07.31

    '''

    def findItemByUid(self, uid):

        from EngineeringConnectorItem import QEngineeringConnectorItem

        items = [item for item in self.scene.items() if hasattr(item, 'uid')]

        for item in items:

            if item.uid == uid:

                return item

        return None

if __name__ == '__main__':

    import sys

    try:

        from PyQt5.QtWidgets import QApplication

    except ImportError:

        try:

            from PyQt4.QtGui import QApplication

        except ImportError:

            raise ImportError("ImageViewerQt: Requires PyQt5 or PyQt4.")

    print('Using Qt ' + QT_VERSION_STR)

    def handleLeftClick(x, y):

        row = int(y)

        column = int(x)

        print("Clicked on image pixel (row="+str(row)+", column="+str(column)+")")

    # Create the application.

    app = QApplication(sys.argv)

    # Create image viewer and load an image file to display.

    viewer = QtImageViewer(None)

    viewer.loadImageFromFile()  # Pops up file dialog.

    # Handle left mouse clicks with custom slot.

    viewer.leftMouseButtonPressed.connect(handleLeftClick)

    # Show viewer and run application.

    viewer.show()

    sys.exit(app.exec_())