HYTOS

# coding: utf-8

import os.path

import copy

import sys

try:

    from PyQt5.QtCore import *

    from PyQt5.QtGui import *

    from PyQt5.QtWidgets import *

except ImportError:

    try:

        from PyQt4.QtCore import *

        from PyQt4.QtGui import *

    except ImportError:

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

from EngineeringPolylineItem import QEngineeringPolylineItem

from GraphicsBoundingBoxItem import QGraphicsBoundingBoxItem

from AppDocData import *

from EngineeringAbstractItem import QEngineeringAbstractItem

class QEngineeringUnknownItem(QEngineeringPolylineItem, QEngineeringAbstractItem):

    HIGHLIGHT = '#BC4438'

    ZVALUE = 60

    '''

        @history    2018.06.18  Jeongwoo    Add variable [transfer] for pyqtSignal

    '''

    def __init__(self, pts, lineIndicator, isVH=None, otherLine=None, parent=None):

        import uuid

        QEngineeringPolylineItem.__init__(self, parent)

        QEngineeringAbstractItem.__init__(self)

        self.uid = uuid.uuid4() # generate UUID

        self.setFlags(QGraphicsItem.ItemIsSelectable|QGraphicsItem.ItemIsFocusable)

        self.setAcceptHoverEvents(True)

        self.setAcceptTouchEvents(True)

        for pt in pts:

            self._pol.append(QPointF(pt[0], pt[1]))

        self._pol.append(QPointF(pts[0][0], pts[0][1])) # close path

        self.lineIndicator = lineIndicator

        self.isVH = isVH

        self.otherLine = otherLine

        self.buildItem()

    '''

        @build  build path

        @author humkyung

        @date   2018.07.06

    '''

    def buildItem(self):

        super(QEngineeringUnknownItem, self).buildItem()

        self.setZValue(QEngineeringUnknownItem.ZVALUE)

        if self.lineIndicator == 'Match':

            self.setPen(QPen(Qt.blue, 1, Qt.SolidLine))

            self.setBrush(QBrush(QColor(0, 0, 255, 255), Qt.SolidPattern))

        else:

            self.setPen(QPen(Qt.red, 1, Qt.DashDotLine))

            self.setBrush(QBrush(QColor(255, 255, 0, 127)))

    def hoverEnterEvent(self, event):

        self._savedColor = self.getColor()

        self.setColor(QEngineeringUnknownItem.HIGHLIGHT)

        self.update()

    def hoverLeaveEvent(self, event):

        self.setColor(self._savedColor)

        self.update()

    def hoverMoveEvent(self, event):

        pass

    '''

        @brief      remove item when user press delete key

        @author     humkyung

        @date       2018.06.11

        @history    2018.06.18  Jeongwoo    Call deleteUnknownItemFromScene method

    '''

    def keyPressEvent(self, event): 

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

            self.deleteUnknownItemFromScene()

    '''

        @brief      show context menu

        @author     humkyung

        @date       2018.08.08

    '''

    def contextMenuEvent(self, event):

        if self.isSelected():

            menu = QMenu()

            registerSymbolAction = QAction("심볼 등록")

            registerSymbolAction.triggered.connect(lambda: self.onRegisterSymbolEvent())

            menu.addAction(registerSymbolAction)

            recognizeTextAction = QAction("텍스트 인식")

            recognizeTextAction.triggered.connect(lambda: self.onRecognizeTextEvent())

            menu.addAction(recognizeTextAction)

            menu.exec_(event.screenPos())

    '''

        @brief      change unknown item's color

        @author     humkyung

        @date       2018.08.13

    '''

    def setColor(self, color):

        self._color = color

        c = QColor()

        c.setNamedColor(color)

        if self.lineIndicator == 'Match':

            c.setAlpha(255)

        else:

            c.setAlpha(177)

        self.setBrush(QBrush(c))

        self.update()

    '''

        @brief      return unknown item's color

        @author     humkyung

        @date       2018.08.13

    '''

    def getColor(self):

        return self.brush().color().name()

    '''

        @brief  override paint(draw connection points)

        @author humkyung

        @date   2018.08.08

    '''

    def paint(self, painter, options=None, widget=None):

        painter.setClipRect(options.exposedRect)

        QEngineeringPolylineItem.paint(self, painter, options, widget)

        if self.isSelected():

            painter.fillPath(self.path(), QBrush(QColor(255, 0, 0, 127)))

    '''

        @brief      show symbol editor

        @author     humkyung

        @date       2018.08.08

    '''

    def onRegisterSymbolEvent(self):

        from App import App

        try:

            selected = [item for item in self.scene().items() if item.isSelected() and type(item) is QEngineeringUnknownItem]

            rect = QRectF()

            for item in selected:

                rect = rect.united(item.sceneBoundingRect())

            App.mainWnd().onAreaSelected(round(rect.left()), round(rect.top()), round(rect.width()), round(rect.height()))

        except Exception as ex:

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

            App.mainWnd().addMessage.emit(MessageType.Error, message)

    '''

        @brief      show text recognition dialog

        @author     humkyung

        @date       2018.08.08

    '''

    def onRecognizeTextEvent(self):

        from App import App

        try:

            selected = [item for item in self.scene().items() if item.isSelected() and type(item) is QEngineeringUnknownItem]

            rect = QRectF()

            for item in selected:

                rect = rect.united(item.sceneBoundingRect())

            App.mainWnd().onRecognizeText(round(rect.left()), round(rect.top()), round(rect.width()), round(rect.height()))

        except Exception as ex:

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

            App.mainWnd().addMessage.emit(MessageType.Error, message)

    '''

        @brief  parse xml for item

        @author humkyung

        @date   2018.06.20

    '''

    @staticmethod 

    def fromXml(node):

        from AppDocData import AppDocData

        item = None

        try:

            location = [float(x) for x in node.find('LOCATION').text.split(',')]

            size = [float(x) for x in node.find('SIZE').text.split(',')]

            points = [[int(float(coord)) for coord in pt.split(',')] for pt in node.find('POINTS').text.split('\n')]

            line_indicator_node = node.find('LINEINDICATOR')

            if line_indicator_node is not None:

                lineIndicator = line_indicator_node.text.split('#')

                if lineIndicator[0] == 'True':

                    item = QEngineeringUnknownItem(points, 'True', lineIndicator[1], [round(float(lineIndicator[2])), round(float(lineIndicator[3])), round(float(lineIndicator[4])), round(float(lineIndicator[5]))])

                elif lineIndicator[0] == 'Match':

                    item = QEngineeringUnknownItem(points, 'Match', lineIndicator[1], [round(float(lineIndicator[2])), round(float(lineIndicator[3])), round(float(lineIndicator[4])), round(float(lineIndicator[5]))])

                else:

                    item = QEngineeringUnknownItem(points, 'False')

            else:

                item = QEngineeringUnknownItem(points, 'False')

            item.setVisible(False)

            if node.find('AREA') is None:

                appDocData = AppDocData.instance()

                minx = min([coord[0] for coord in points])

                maxx = max([coord[0] for coord in points])

                miny = min([coord[1] for coord in points])

                maxy = max([coord[1] for coord in points])

                for area in appDocData.getAreaList():

                    if area.contains([minx, miny, maxx, maxy]):

                        item.area = area.name

                        break

            else:

                item.area = node.find('AREA').text

        except Exception as ex:

            from App import App

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

            App.mainWnd().addMessage.emit(MessageType.Error, message)

        return item

    '''

        @brief      generate xml code

        @author     humkyung

        @date       2018.06.11

        @history    2018.06.14  Jeongwoo    Add LOCATION, SIZE info

    '''

    def toXml(self, owner = None, name='UNKNOWN'):

        from xml.etree.ElementTree import Element, SubElement, dump, ElementTree

        from EngineeringLineItem import QEngineeringLineItem

        from SymbolSvgItem import SymbolSvgItem

        try:

            node = Element('UNKNOWN')

            rect = self.sceneBoundingRect()

            locNode = Element('LOCATION')

            locNode.text = '{},{}'.format(rect.left(), rect.top())

            node.append(locNode)

            sizeNode = Element('SIZE')

            sizeNode.text = '{},{}'.format(rect.width(), rect.height())

            node.append(sizeNode)

            areaNode = Element('AREA')

            areaNode.text = self.area

            node.append(areaNode)

            ptNode = Element('POINTS')

            points = None

            for pt in self._pol:

                if points is None:

                    points = '{},{}'.format(int(pt.x()), int(pt.y())) 

                else:

                    points += '\n{},{}'.format(int(pt.x()), int(pt.y()))

            ptNode.text = points

            node.append(ptNode)

            lINode = Element('LINEINDICATOR')

            if self.lineIndicator == 'True':

                lINode.text = 'True' + '#' + self.isVH + '#' + str(self.otherLine[0]) + '#' + str(self.otherLine[1]) + '#' + str(self.otherLine[2]) + '#' + str(self.otherLine[3])

            elif self.lineIndicator == 'Match':

                lINode.text = 'Match' + '#' + self.isVH + '#' + str(self.otherLine[0]) + '#' + str(self.otherLine[1]) + '#' + str(self.otherLine[2]) + '#' + str(self.otherLine[3])

            else:

                lINode.text = 'False' + '#' + 'None' + '#' + 'None'

            node.append(lINode)

        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 str(self.uid)

        return node 

    '''

        @brief      Add to Scene

        @author     Jeongwoo

        @date       2018.06.14

    '''

    def addUnknownItemToScene(self, scene):

        self.setZValue(1.0)

        scene.addItem(self)

    '''

        @brief      Delete from Scene

        @author     Jeongwoo

        @date       2018.06.14

        @history    2018.06.18  Jeongwoo    Add calling removed signal

    '''

    def deleteUnknownItemFromScene(self):

        self.transfer.onRemoved.emit(self)

        self.scene().removeItem(self)

    def boundingRectOnScene(self):

        rect = self.sceneBoundingRect()

        return rect