HYTOS

# coding: utf-8

""" This is place stream line module """

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, QRectF, pyqtSignal, QT_VERSION_STR

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

    except ImportError:

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

class PlaceStreamlineCommand(AbstractCommand.AbstractCommand):

    """ This is place line class """

    onSuccess = pyqtSignal()

    onRejected = pyqtSignal(AbstractCommand.AbstractCommand)

    def __init__(self, imageViewer):

        super(PlaceStreamlineCommand, self).__init__(imageViewer)

        self.name = 'PlaceLine' 

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

        self._streamline = None

        self.connectorItems = []

    @property

    def streamline(self):

        """ getter of _streamline """

        return self._streamline

    '''

        @brief      reset command status

        @author     humkyung

        @date       2018.07.23

    '''

    def reset(self):

        self._streamline = None

    '''

        @brief      place a line

        @author     humkyung

        @date       2018.07.23

    '''

    def execute(self, param):

        import shapely

        from EngineeringConnectorItem import QEngineeringConnectorItem

        from SymbolSvgItem import SymbolSvgItem

        from EngineeringStreamlineItem import QEngineeringStreamlineItem

        self.isTreated = False

        event = param[1]

        if 'mousePressEvent' == param[0] and event.button() == Qt.LeftButton:

            selected = self.imageViewer.scene.itemAt(param[2], QTransform())

            if self._streamline is None: 

                if selected is not None and type(selected) is QEngineeringConnectorItem:

                    self._streamline = QEngineeringStreamlineItem()

                    self._streamline._vertices.append(selected.center())

                    self.imageViewer.scene.addItem(self._streamline)

                    self.imageViewer.scene.setFocusItem(self._streamline)

                    self.connectorItems.append(selected)

            else:

                try: 

                    if selected is not None and type(selected) is QEngineeringConnectorItem:

                        self.connectorItems.append(selected)

                        self._streamline._vertices[-1] = selected.center()

                    #QGraphicsView.mouseReleaseEvent(self.imageViewer, event)

                    self._streamline.build_connectors(self.connectorItems)

                    self._streamline.update()

                    self.onSuccess.emit()

                    self.connectorItems.clear()

                finally:

                    pass

        elif 'mouseReleaseEvent' == param[0] and event.button() == Qt.RightButton:

            self.onRejected.emit(self)

        elif 'mouseMoveEvent' == param[0] and self._streamline is not None:

            self._streamline.onMouseMoved(event, param[2])

        elif 'keyPressEvent' == param[0]:

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

                self.onRejected.emit(self)

                self.isTreated = False

    def undo(self):

        pass

    def redo(self):

        pass