HYTOS

import os.path

import AbstractCommand

try:

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

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

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

except ImportError:

    try:

        from PyQt4.QtCore import Qt, QPoint, QPointF, QRectF, pyqtSignal, QT_VERSION_STR, QMimeData

        from PyQt4.QtGui import QGraphicsView, QGraphicsScene, QImage, QPixmap, QPainterPath, QFileDialog, QColor, QPen, QBrush, QCursor

    except ImportError:

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

class ConnectionPointCommand(AbstractCommand.AbstractCommand):

    onSuccess = pyqtSignal()

    def __init__(self, imageViewer, connectionPointLineEdit, connectionPointListWidget):

        super(ConnectionPointCommand, self).__init__(imageViewer)

        self.name = 'ConnectionPoint' 

        self.connectionPointLineEdit = connectionPointLineEdit

        self.connectionPointListWidget = connectionPointListWidget

        self.listItems = self.connectionPointListWidget.items(QMimeData())

        self.imageViewer.setCursor(QCursor(Qt.CrossCursor))

        self.imageWidth = self.imageViewer.image().width()

        self.imageHeight = self.imageViewer.image().height()

        self.initMinPoint = QPointF(self.imageWidth * -1, self.imageHeight * -1)

        #self.showGuideline(self.initMinPoint)

        self.connectionPointLineEdit.setText("")

    '''

        @brief      pan image by left click and drag

        @history    2016.06.12  Jeongwoo    Make new point (newScenePos)

    '''

    def execute(self, param):

        from EngineeringOriginItem import QEngineeringOriginItem

        from EngineeringGuidelineItem import QEngineeringGuidelineItem

        event = param[1]

        scenePos = param[2]

        newScenePos = QPointF(round(scenePos.x()), round(scenePos.y()))

        items = [item for item in self.imageViewer.scene.items(scenePos) \

            if type(item) is QEngineeringOriginItem or type(item) is QEngineeringGuidelineItem]

        originItem = None

        if items and type(items[0]) is QEngineeringGuidelineItem:

            originItem = items[0].parentItem()

        elif items[0] and type(items[0]) is QEngineeringOriginItem:

            originItem = items[0]

        if originItem is not None:

            center = originItem.rect().center() 

            dx = scenePos.x() - center.x()

            dy = scenePos.y() - center.y()

            if abs(dx) < 1:

                newScenePos.setX(center.x())

            if abs(dy) < 1:

                newScenePos.setY(center.y())

        if 'mousePressEvent' == param[0]:

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

                if self.isOnImage(newScenePos.x(), newScenePos.y()):

                    ## drawCircle Method is static

                    ConnectionPointCommand.drawCircle(self.imageViewer, newScenePos.x(), newScenePos.y())

                    text = "{},{}".format(newScenePos.x(), newScenePos.y())

                    self.connectionPointLineEdit.setText(text)

                    self.connectionPointListWidget.addItem(text)

                    self.onSuccess.emit()

                else:

                    QMessageBox.about(self.imageViewer, "알림", "심볼 내부를 선택해주세요.")

                QGraphicsView.mousePressEvent(self.imageViewer, event)

        elif 'mouseMoveEvent' == param[0]:

            if self.isOnImage(newScenePos.x(), newScenePos.y()):

                #self.showGuideline(newScenePos)

                self.connectionPointLineEdit.setText("{},{}".format(newScenePos.x(), newScenePos.y()))

            else:

                #self.showGuideline(self.initMinPoint)

                self.connectionPointLineEdit.setText("")

            pass

        self.isTreated = False

    '''

        @history    2018.06.11  Jeongwoo    Shorten method to add Ellipse

                    2018.06.12  Jeongwoo    Modify RectF's coords

                    humkyung 2018.08.28 add connector item instead of QGraphicsEllipse

    '''

    @staticmethod

    def drawCircle(imageViewer, x, y):

        from EngineeringConnectorItem import QEngineeringConnectorItem

        connector = QEngineeringConnectorItem()

        connector.setPos((float(x), float(y)))

        imageViewer.scene.addItem(connector)

    '''

        @history    2018.06.11  Jeongwoo    Make area bigger (width/height 1px)

                    2018.06.12  Jeongwoo    Make area smaller (width/height 1px)

    '''

    def isOnImage(self, x, y):

        if (x >= 0 and x <= self.imageWidth) and (y >= 0 and y <= self.imageHeight):

            return True

        else:

            return False

    '''

    def showGuideline(self, pos):

        self.imageViewer.crosshairPos = pos

        self.imageViewer.scene.update(QRectF(0, 0, self.imageWidth, self.imageHeight))

    '''

    def undo(self):

        pass

    def redo(self):

        pass