HYTOS

# coding: utf-8

import os.path

import copy

import sys

try:

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

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

    from PyQt5.QtWidgets import QGraphicsView, QGraphicsScene, QFileDialog, QGraphicsItem, QAbstractGraphicsShapeItem, QGraphicsTextItem

except ImportError:

    try:

        from PyQt4.QtCore import Qt, QRectF, pyqtSignal, QRect, QObject, QT_VERSION_STR

        from PyQt4.QtGui import QGraphicsView, QGraphicsScene, QImage, QPixmap, QPainterPath, QFileDialog, QFont, QColor, QFontMetrics

    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 QEngineeringTextItem(QGraphicsTextItem, QEngineeringAbstractItem):

    HIGHLIGHT = '#BC4438'

    '''

        @history    2018.05.17  Jeongwoo    Add self._owner variable

    '''

    def __init__(self, uid=None, parent=None):

        import uuid

        QGraphicsTextItem.__init__(self, parent)

        QEngineeringAbstractItem.__init__(self)

        self.uid = uuid.uuid4() if uid is None else uid

        self.loc = None

        self.size = None

        self.angle = 0  # angle in radian

        self.conns = []

        self._owner = None

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

        self.setAcceptHoverEvents(True)

        self.setAcceptTouchEvents(True)

        self.setColor(self._color)

        self._savedColor = None        

        self.delimiter = '"'

        self.attribute = ''

        self.transfer = Transfer()

    '''

        @brief      Get owner

        @author     Jeongwoo

        @date       2018.05.17

    '''

    @property

    def owner(self):

        return self._owner

    '''

        @brief      Set owner

        @author     Jeongwoo

        @date       2018.05.17

        @history    2018.05.17  Jeongwoo    Add Calling setColor if self._owner is None or not

    '''

    @owner.setter

    def owner(self, value):

        self._owner = value

        if self._owner is None:

            self._color = self.DEFAULT_COLOR

        self.setColor(self._color)

    '''

        @brief  return text string

        @author humkyung

        @date   2018.04.16

    '''

    def text(self):

        return self.toPlainText()

    '''

        @brief  return center position of text

        @author humkyung

        @date   2018.04.16

    '''

    def center(self):

        return self.sceneBoundingRect().center()

    '''

        @brief      hover event

        @authro     humkyung

        @date       

    '''

    def hoverEnterEvent(self, event):

        self.setHightlight()

    def hoverLeaveEvent(self, event):

        self.unsetHightlight()

    '''

        @brief      set highlight

        @author     kyouho

        @date       2018.08.27

    '''

    def setHightlight(self):

        if self._savedColor is None:

            self._savedColor = self.getColor()

        self.setColor(QEngineeringTextItem.HIGHLIGHT)

        self.update()

    '''

        @brief      unset highlight

        @author     kyouho

        @date       2018.08.27

    '''

    def unsetHightlight(self):

        self.setColor(self._savedColor)

        self.update()

    def hoverMoveEvent(self, event):

        pass

    '''

        @brief      remove item when user press delete key

        @author     humkyung

        @date       2018.04.23

        @history    2018.05.25  Jeongwoo    Seperate delete item method

                    humkyung 2018.08.18 rotate text when user press 'R'

    '''

    def keyPressEvent(self, event): 

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

            self.deleteTextItemFromScene()

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

            #degree 0

            if 0 == self.angle:

                self.angle = 1.57

            #degree 90

            elif (1.57 == self.angle):

                self.angle = 3.14

            #degree 180

            elif 3.14 == self.angle:

                self.angle = 4.71

            #degree 270

            elif 4.71 == self.angle :

                self.angle = 0

            width = self.size[0]

            height = self.size[1]

            self.size = (height, width)

            self.rotate()

        QGraphicsTextItem.keyPressEvent(self, event)

    '''

        @brief  draw rect when item is selected

        @author humkyung

        @date   2018.07.08

    '''

    def drawFocusRect(self, painter):

        self.focuspen = QPen(Qt.DotLine)

        self.focuspen.setColor(Qt.black)

        self.focuspen.setWidthF(1.5)

        hilightColor = QColor(255, 0, 0, 127)

        painter.setBrush(QBrush(hilightColor))

        painter.setPen(self.focuspen)

        painter.drawRect(self.boundingRect())

    '''

        @brief  override paint(draw connection points)

        @author humkyung

        @date   2018.07.08

    '''

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

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

        if self.isSelected():

            self.drawFocusRect(painter)

    '''

        @brief      Delete text item

        @author     Jeongwoo

        @date       2018.05.25

    '''

    def deleteTextItemFromScene(self):

        self.transfer.onRemoved.emit(self)

        self.scene().removeItem(self)

    '''

        @brief      Return real item position

        @author     Jeongwoo

        @date       2018.05.25

    '''

    def boundingRectOnScene(self):

        rect = self.boundingRect()

        rect.moveTo(self.loc[0], self.loc[1])

        return rect

    '''

        @brief      Double click event, Show QOcrResultDialog

        @author     Jeongwoo

        @date       18.04.23

        @history    18.06.20    Jeongwoo    Resize QRect added 1

    '''

    def mouseDoubleClickEvent(self, event):

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

            from OcrResultDialog import QOcrResultDialog

            dialog = QOcrResultDialog(None, AppDocData.instance().getCurrentPidSource().getQImageOnRect(QRect(self.loc[0], self.loc[1], self.size[0]+1, self.size[1]))

                                                       , QRect(self.loc[0], self.loc[1], self.size[0]+1, self.size[1]), True, self.text())

            (isAccept, textInfoList) = dialog.showDialog()

            if isAccept:

                scene = self.scene()

                scene.removeItem(self)

                text = textInfoList[0].getText()

                self.setPlainText(text)

                self.setDefaultTextColor(Qt.blue)

                self.addTextItemToScene(scene)

    '''

        @brief      rotate text

        @author     humkyung

        @date       2018.08.18

    '''

    def rotate(self):

        sx = 1

        sy = 1

        width = self.size[0]

        height = self.size[1]

        x = self.loc[0]

        y = self.loc[1]

        transform = QTransform()

        if (1.57 == self.angle) or (4.71 == self.angle):

            rect = self.boundingRect()

            sx = width/rect.height()

            sy = height/rect.width()

            transform.translate(x, y)

            transform.translate(width*0.5, height*0.5)

            transform.scale(1, sy)

            transform.rotateRadians(-self.angle)

            transform.translate(-rect.width()*0.5, -rect.height()*0.5)

        elif 3.14 == self.angle:

            rect = self.boundingRect()

            sx = width/rect.width()

            sy = height/rect.height()

            transform.translate(x, y - round((rect.height()-height)*0.5))

            transform.scale(sx, 1)

            transform.rotateRadians(-self.angle)

            transform.translate(-width*0.5, -height*0.5)

        else:

            rect = self.boundingRect()

            sx = width/rect.width()

            sy = height/rect.height()

            #if '\n' not in text:

            transform.translate(x, y - round((rect.height()-height)*0.5))

            transform.scale(sx, 1)

        self.setTransform(transform)

        self.update()

    '''

        @brief      Put text on scene

        @author     Jeongwoo

        @date       18.04.23

        @history    humkyung 2018.06.30 apply font configuration

    '''

    def addTextItemToScene(self, scene):

        try:

            docData = AppDocData.instance()

            configs = docData.getConfigs('Text Style', 'Font Name')

            fontName = configs[0].value if configs else 'Arial'

            configs = docData.getConfigs('Text Style', 'Font Size')

            fontSize = int(configs[0].value) if configs else -1

            sx = 1

            sy = 1

            width = self.size[0]

            height = self.size[1]

            x = self.loc[0]

            y = self.loc[1]

            rect = None

            transform = QTransform()

            if (1.57 == self.angle) or (4.71 == self.angle):

                font = QFont(fontName, width*1.2 if fontSize == -1 else fontSize)

                x_factor = width / QFontMetrics(font).height()

                y_factor = height / QFontMetrics(font).width(self.text())

                factor = min(x_factor, y_factor)

                font.setPointSizeF(font.pointSizeF()*factor)

                self.setFont(font)

                rect = self.boundingRect()

                sx = width/rect.height()

                sy = height/rect.width()

                transform.translate(x, y)

                transform.translate(width*0.5, height*0.5)

                transform.scale(1, sy)

                transform.rotateRadians(-self.angle)

                transform.translate(-rect.width()*0.5, -rect.height()*0.5)

            elif 3.14 == self.angle:

                font = QFont(fontName, height*1.2 if fontSize == -1 else fontSize)

                x_factor = width / QFontMetrics(font).width(self.text())

                y_factor = height / QFontMetrics(font).height()

                factor = min(x_factor, y_factor)

                font.setPointSizeF(font.pointSizeF()*factor)

                self.setFont(font)

                rect = self.boundingRect()

                sx = width/rect.width()

                sy = height/rect.height()

                transform.translate(x, y - round((rect.height()-height)*0.5))

                transform.scale(sx, 1)

                transform.rotateRadians(-self.angle)

                transform.translate(-width*0.5, -height*0.5)

            else:

                font = QFont(fontName, height*1.2 if fontSize == -1 else fontSize)

                x_factor = width / QFontMetrics(font).width(self.text())

                y_factor = height / QFontMetrics(font).height()

                factor = min(x_factor, y_factor)

                font.setPointSizeF(font.pointSizeF()*factor)

                self.setFont(font)

                rect = self.boundingRect()

                sx = width/rect.width()

                sy = height/rect.height()

                #if '\n' not in text:

                transform.translate(x, y - round((rect.height()-height)*0.5))

                transform.scale(sx, 1)

            self.setTransform(transform)

            scene.addItem(self)

        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  get connected items

        @author humkyung

        @date   2018.04.23

    '''

    def getConnectedItems(self):

        visited = []

        try:

            if 1 == len(self.conns):

                # iterate connected items

                pool = []

                visited = []

                pool.append(self.conns[0])

                while len(pool) > 0:

                    it = pool.pop()

                    visited.append(it)

                    for conn in it.conns:

                        if (conn is not None) and (conn not in visited): pool.append(conn)

                # up to here

        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 visited

    '''

        @brief      parse xml code

        @author     humkyung

        @date       2018.09.15

    '''

    @staticmethod 

    def fromXml(node):

        from TextItemFactory import TextItemFactory

        from AppDocData import AppDocData

        from QEngineeringNoteItem import QEngineeringNoteItem

        item = None

        try:

            location = node.find('LOCATION').text if node.find('LOCATION') is not None else '0,0'

            x = float(location.split(',')[0])

            y = float(location.split(',')[1])

            width = float(node.find('WIDTH').text) if node.find('WIDTH') is not None else 0

            height = float(node.find('HEIGHT').text) if node.find('HEIGHT') is not None else 0

            angle = float(node.find('ANGLE').text) if node.find('ANGLE') is not None else 0

            value = node.find('VALUE').text

            uid = node.find('UID')

            attributeValue = node.find('ATTRIBUTEVALUE')

            name = node.find('NAME').text

            if name == 'TEXT':

                item = TextItemFactory.instance().createTextItem(value)

                if item is not None:

                    item.loc = (x, y)

                    item.size = (width, height)

                    item.angle = angle

                    item.setPlainText(value)

            elif name == 'NOTE':

                item = QEngineeringNoteItem()

                if item is not None:

                    item.loc = (x, y)

                    item.size = (width, height)

                    item.angle = angle

                    item.setPlainText(value)

            ## assign area

            if item is not None:

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

                    appDocData = AppDocData.instance()

                    for area in appDocData.getAreaList():

                        if area.contains([x, y]):

                            item.area = area.name

                            break

                else:

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

            ## up to here

        except Exception as ex:

            from App import App

            from AppDocData import MessageType

            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.04.23

        @history    humkyung 2018.04.27 move to QEngineeringLineNoTextItem

                    humkyung 2018.05.02 add name as parameter

                    Jeongwoo 2018.05.30 Change variable [owner] is nullable and Add/Modify attributes

    '''

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

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

        from EngineeringLineItem import QEngineeringLineItem

        from SymbolSvgItem import SymbolSvgItem

        try:

            node = Element('ATTRIBUTE')

            #if owner is not None:

            uidNode = Element('UID')

            uidNode.text = str(self.uid)

            node.append(uidNode)

            attributeValueNode = Element('ATTRIBUTEVALUE')

            attributeValueNode.text = self.attribute

            node.append(attributeValueNode)

            nameNode = Element('NAME')

            nameNode.text = name

            node.append(nameNode)

            locNode = Element('LOCATION')

            locNode.text = '{},{}'.format(self.loc[0], self.loc[1])

            node.append(locNode)

            valueNode = Element('VALUE')

            valueNode.text = self.text()

            node.append(valueNode)

            angleNode = Element('ANGLE')

            angleNode.text = str(self.angle)

            node.append(angleNode)

            widthNode = Element('WIDTH')

            widthNode.text = str(self.size[0])

            node.append(widthNode)

            heightNode = Element('HEIGHT')

            heightNode.text = str(self.size[1])

            node.append(heightNode)

            areaNode = Element('AREA')

            areaNode.text = self.area

            node.append(areaNode)

        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      Set Color. Override QEngineeringAbstractItem's

        @author     Jeongwoo

        @date       2018.05.11

        @history    humkyung 2018.05.13 update after change color

    '''

    def setColor(self, color):

        self._color = color

        c = QColor()

        c.setNamedColor(color)

        self.setDefaultTextColor(c)

        self.update()

    def getColor(self):

        return self._color

'''

    @brief      The class transfer pyqtSignal Event. Cause Subclass of QGraphicsRectItem can't use pyqtSignal

    @author     Jeongwoo

    @date       2018.06.18

'''

class Transfer(QObject):

    onRemoved = pyqtSignal(QGraphicsItem)

    def __init__(self, parent = None):

        QObject.__init__(self, parent)