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, QFontMetricsF

    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, QFontMetricsF

    except ImportError:

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

from EngineeringPolylineItem import QEngineeringPolylineItem

from GraphicsBoundingBoxItem import QGraphicsBoundingBoxItem

from AppDocData import *

from EngineeringAbstractItem import QEngineeringAbstractItem

from TextInfo import TextInfo

class QEngineeringTextItem(QGraphicsTextItem, QEngineeringAbstractItem):

    HIGHLIGHT = '#BC4438'

    ZVALUE = 100

    '''

        @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 uuid.UUID(uid, version=4)

        self.type = 'TEXT'

        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.lineNoDelimiter = '!-!'

        self.attribute = ''

        self.transfer = Transfer()

        self.setZValue(QEngineeringTextItem.ZVALUE)

    def __str__(self):

        """ return string represent uuid """

        return str(self.uid)

    '''

        @brief      Get owner

        @author     Jeongwoo

        @date       2018.05.17

    '''

    @property

    def owner(self):

        import uuid

        # find owner with uid

        if self._owner is not None and type(self._owner) is uuid.UUID:

            matches = [x for x in self.scene().items() if hasattr(x, 'uid') and str(x.uid) == str(self._owner)]

            if matches: self._owner = matches[0]

            return matches[0] if matches else None

        # up to here

        if type(self._owner) is not uuid.UUID and type(self._owner) is not str:

            return self._owner

        else:

            self._owner = None

            return None

    '''

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

    def boundingRect(self):

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

            return QRectF(0, 0, self.size[1], self.size[0])

        else:

            return QRectF(0, 0, self.size[0], self.size[1])

    '''

        @brief      hover event

        @authro     humkyung

        @date       

    '''

    def hoverEnterEvent(self, event):

        self.highlight(True)

    def hoverLeaveEvent(self, event):

        self.highlight(False)

    '''

        @brief      set highlight

        @author     kyouho

        @date       2018.08.27

    '''

    def highlight(self, flag):

        self.hover = flag

        if flag:

            if self._savedColor is None:

                self._savedColor = self.getColor()

            self.setColor(QEngineeringTextItem.HIGHLIGHT)

        elif hasattr(self, '_savedColor'):

            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 self.isSelected() and 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):

        from EngineeringLineNoTextItem import QEngineeringLineNoTextItem

        self.setColor(self.getColor())

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

            rect = QRectF(0, 0, self.size[1], self.size[0])

        else:

            rect = QRectF(0, 0, self.size[0], self.size[1])

        painter.setFont(self.font())

        color = self.defaultTextColor()

        #color = QColor('#151485')

        painter.setPen(QPen(color))

        painter.drawText(rect, Qt.AlignCenter, self.text())

        if self.isSelected():

            self.drawFocusRect(painter)

        if type(self) is QEngineeringLineNoTextItem and self.isSelected():

            for run in self.runs:

                for item in run.items:

                    item.setSelected(True)

        #elif type(self) is QEngineeringLineNoTextItem:

        #    for run in self.runs:

        #        for item in run.items:

        #            item.setSelected(False)

    '''

        @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, isDataList=False):

        from TextItemFactory import TextItemFactory

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

            from OcrResultDialog import QOcrResultDialog

            dialog = QOcrResultDialog(None, AppDocData.instance().getCurrentPidSource().getQImageOnRect(

                QRect(self.loc[0] - 3, self.loc[1] - 3, self.size[0] + 6, self.size[1] + 6)), 

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

            (isAccept, textInfoList) = dialog.showDialog()

            if isAccept:

                scene = self.scene()

                textInfo = textInfoList[0]

                x = textInfo.getX()

                y = textInfo.getY()

                angle = textInfo.getAngle()

                text = textInfo.getText()

                width = textInfo.getW()

                height = textInfo.getH()

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

                if item is not None:

                    item.loc = (x, y)

                    item.size = (width, height)

                    item.angle = angle

                    item.area = self.area

                    item.addTextItemToScene(scene)

                    item.transfer.onRemoved.connect(scene.parent().parent().parent().itemRemoved)

                    self.transfer.onRemoved.emit(self)

                    if isDataList:

                        return item

                else:

                    message = 'error occured({}) in {}:{}'.format('텍스트 생성에 실패했습니다.', sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno)

                    self.addMessage.emit(MessageType.Normal, message)

    '''

        @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 if fontSize == -1 else fontSize)

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

                y_factor = height / QFontMetricsF(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 if fontSize == -1 else fontSize)

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

                y_factor = height / QFontMetricsF(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 if fontSize == -1 else fontSize)

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

                y_factor = height / QFontMetricsF(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)

                transform.scale(sx, 1)

            self.setTransform(transform)

            scene.addItem(self)

        except Exception as ex:

            from App import App

            from AppDocData import MessageType

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

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

    '''

        @brief  get connected items

        @author humkyung

        @date   2018.04.23

        @history    2018.11.22      euisung     fix note road

    '''

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

        import uuid

        from TextItemFactory import TextItemFactory

        from AppDocData import AppDocData

        from EngineeringNoteItem import QEngineeringNoteItem

        from QEngineeringSizeTextItem import QEngineeringSizeTextItem

        from EngineeringValveOperCodeTextItem import QEngineeringValveOperCodeTextItem

        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

            #attributeValue = node.find('ATTRIBUTEVALUE')

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

            textInfo = TextInfo(value, x, y, width, height, angle)

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

            if item is not None:

                    item.loc = (x, y)

                    item.size = (width, height)

                    item.angle = angle

            '''

            if name == 'NOTE':

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

                if item is not None:

                    item.loc = (x, y)

                    item.size = (width, height)

                    item.angle = angle

                    item.setPlainText(value)

                    item.setToolTip('{} = {}'.format(item.type, value))

            elif name == 'SIZE':

                item = QEngineeringSizeTextItem()

                if item is not None:

                    item.loc = (x, y)

                    item.size = (width, height)

                    item.angle = angle

                    item.setPlainText(value)

                    item.setToolTip('{} = {}'.format(item.type, value))

            elif name == 'VALVE OPER CODE':

                item = QEngineeringValveOperCodeTextItem()

                if item is not None:

                    item.loc = (x, y)

                    item.size = (width, height)

                    item.angle = angle

                    item.setPlainText(value)

                    item.setToolTip('{} = {}'.format(item.type, value))

            else:

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

                if item is not None:

                    item.loc = (x, y)

                    item.size = (width, height)

                    item.angle = angle

            '''

            # set uid and owner of item

            if item is not None:

                item.uid = uuid.UUID(node.find('UID').text)

                item.setVisible(False)

                if node.find('OWNER') is not None and node.find('OWNER').text != 'None':

                    item._owner = uuid.UUID(node.find('OWNER').text, version=4)

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

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

        from EngineeringLineItem import QEngineeringLineItem

        from SymbolSvgItem import SymbolSvgItem

        try:

            node = Element('ATTRIBUTE')

            uidNode = Element('UID')

            uidNode.text = str(self.uid)

            node.append(uidNode)

            # write owner's uid to xml

            ownerNode = Element('OWNER')

            if self.owner is not None:

                ownerNode.text = str(self.owner)

            else:

                ownerNode.text = 'None'

            node.append(ownerNode)

            # up to here

            attributeValueNode = Element('ATTRIBUTEVALUE')

            attributeValueNode.text = self.attribute

            node.append(attributeValueNode)

            nameNode = Element('NAME')

            nameNode.text = self.type

            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:

            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 None

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

        if QColor.isValidColor(color):

            c = QColor()

            c.setNamedColor(color)

            self.setDefaultTextColor(c)

            self.update()

    def toSql(self):

        """

        convert instrument data to sql query for title block now on

        """

        from AppDocData import AppDocData

        appDocData = AppDocData.instance()

        titleBlockProps = appDocData.getTitleBlockProperties()

        for titleBlockProp in titleBlockProps:

            if self.area == titleBlockProp[0]:

                cols = ['UID', 'Drawing_UID', 'TitleBlockProperties_UID', 'VALUE']

                values = ['?','?','?','?']

                param = [str(self.uid), appDocData.activeDrawing.name, self.area, self.text()]

                sql = 'insert or replace into TitleBlockValues values({})'.format(','.join(values))

                return (sql, tuple(param))

        return None

'''

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