HYTOS

import os.path

import AbstractCommand

try:

    from PyQt5.QtCore import *

    from PyQt5.QtGui import *

    from PyQt5.QtWidgets import *

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):

    """

    ConnectionPointCommand class

    """

    onSuccess = pyqtSignal(QGraphicsItem)

    def __init__(self, imageViewer, connectionPointLineEdit, index=-1):

        super(ConnectionPointCommand, self).__init__(imageViewer)

        self.name = 'ConnectionPoint'

        self.connectionPointLineEdit = connectionPointLineEdit

        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.connectionPointLineEdit.setText("")

        self.index = index

    '''

        @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 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

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

                                                             self.index)

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

                    self.connectionPointLineEdit.setText(text)

                    self.index += 1

                    self.onSuccess.emit(conn)

                else:

                    QMessageBox.about(self.imageViewer, self.tr('Notice'),

                                      self.tr('Please select inside of the symbol.'))

                QGraphicsView.mousePressEvent(self.imageViewer, event)

        elif 'mouseMoveEvent' == param[0]:

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

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

            else:

                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, index=-1):

        from EngineeringConnectorItem import QEngineeringConnectorItem

        connector = QEngineeringConnectorItem(index=index)

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

        imageViewer.scene().addItem(connector)

        return 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 undo(self):

        pass

    def redo(self):

        pass