HYTOS

# coding: utf-8

import os.path

import sys

import copy

try:

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

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

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

except ImportError:

    try:

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

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

    except ImportError:

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

from EngineeringPolylineItem import QEngineeringPolylineItem

from GraphicsBoundingBoxItem import QGraphicsBoundingBoxItem

from UserInputAttribute import UserInputAttribute

from OcrResultDialog import QOcrResultDialog

from AppDocData import AppDocData

from EngineeringTextItem import QEngineeringTextItem

lineColumnList = ['UID', 'LINE_SIZE', 'LINE_SYMBOL', 'LINE_NO', 'LINE_CLASS', 'LINE_ROUTING_FROM', 'LINE_ROUTING_TO', 'SERVICE_FLUID', 'SERVICE_DENSITY', 'SERVICE_STATE', 'OPERATION_CONDITION_TEMP', 'OPERATION_CONDITION_PRESS', 'DESIGN_CONDITION_TEMP', 'DESIGN_CONDITION_PRESS', 'TEST_CONDITION_TEMP', 'TEST_CONDITION_PRESS', 'INSUL_CODE', 'PAINT_CODE', 'NDE_CODE', 'PWHT', 'PNID_NO']

class QEngineeringLineNoTextItem(QEngineeringTextItem):

    '''

        @history    18.05.14    Jeongwoo    Add variable self.runs

                    humkyung 2018.07.09 add stream no

    '''

    def __init__(self, parent=None):

        QEngineeringTextItem.__init__(self, parent)

        self._runs = []

    '''

        @brief  getter of runs

        @author humkyung

        @date   2018.05.11

    '''

    @property

    def runs(self):

        return self._runs

    '''

        @brief  setter of runs

        @author humkyung

        @date   2018.05.11

    '''

    @runs.setter

    def runs(self, value):

        self._runs = value

    def getLongestTwoPoints(self, pts):

        import math

        res = [None, None]

        maxDistance = None

        for i in range(len(pts)):

            for j in range(i+1, len(pts)):

                dx = pts[i][0] - pts[j][0]

                dy = pts[i][1] - pts[j][1]

                dist = math.sqrt(dx*dx + dy*dy)

                if (maxDistance is None) or (maxDistance < dist):

                    maxDistance = dist

                    res[0] = pts[i]

                    res[1] = pts[j]

        return res

    '''

        @brief      set attribute

        @author     humkyung

        @date       2018.07.20

    '''

    def setAttribute(self, name, value):

        matches = [attr for attr in self._attrs if attr[0] == name]

        if len(matches) == 1:

            matches[0][1] = value

    '''

        @brief      get attribute

        @author     kyouho

        @date       2018.09.06

    '''

    def getAttributes(self):

        attrs = {}

        attrs['Line No'] = self.uid

        try:

            docData = AppDocData.instance()

            lineNoconfigs = docData.getConfigs('Line No', 'Configuration')

            from TextItemFactory import TextItemFactory

            item = TextItemFactory.instance().isLineNo(self.text())

            # Line No 부분

            if item[0]:

                result = item[1]

                configs = lineNoconfigs[0].value.split(self.delimiter)

                lineAttrs = docData.getLineProperties()

                for lineAttr in lineAttrs:

                    if lineAttr[3] == 'String':

                        find = False

                        for attr in self.attrs:

                            if type(attr) is UserInputAttribute and attr.attribute == lineAttr[0]:

                                find = True

                                break;

                        if not find:

                            newAttr = UserInputAttribute(lineAttr[0], '')

                            self._attrs.append(newAttr)

                        attrs[lineAttr[0]] = ''

                    else:

                        for i in range(len(configs)):

                            if lineAttr[0] == configs[i]:

                                attrs[lineAttr[0]] = result[i]

                                break

            for attr in self._attrs:

                if type(attr) is UserInputAttribute:

                    attrs[attr.attribute] = attr.text

            attrs['CONN'] = self.conns[0].uid if self.conns else ''

        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 attrs

    '''

        @brief      generate xml code

        @author     humkyung

        @date       2018.04.23

        @history    humkyung 2018.05.02 write symbol's attribute

                    humkyung 2018.05.16 write run information to xml

    '''

    def toXml(self):

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

        from EngineeringLineItem import QEngineeringLineItem

        from SymbolSvgItem import SymbolSvgItem

        try:

            docData = AppDocData.instance()

            configs = docData.getConfigs('Line No', 'Delimiter')

            delimiter = configs[0].value if 1 == len(configs) else '-'

            lineNoconfigs = docData.getConfigs('Line No', 'Configuration')

            node = Element('LINE_NO')

            uidNode = Element('UID')

            uidNode.text = str(self.uid)

            node.append(uidNode)

            textNode = Element('TEXT')

            textNode.text = self.text() 

            node.append(textNode)

            rect = self.sceneBoundingRect()

            locNode = Element('LOCATION')

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

            node.append(locNode)

            widthNode = Element('WIDTH')

            widthNode.text = str(rect.width())

            node.append(widthNode)

            heightNode = Element('HEIGHT')

            heightNode.text = str(rect.height())

            node.append(heightNode)

            angleNode = Element('ANGLE')

            angleNode.text = str(self.angle)

            node.append(angleNode)

            areaNode = Element('AREA')

            areaNode.text = self.area

            node.append(areaNode)

            for run in self.runs:

                node.append(run.toXml())

            for attr in self._attrs:

                if type(attr) is not UserInputAttribute:

                    attrNode = Element('ATTRIBUTE')

                    uidNode = Element('UID')

                    uidNode.text = str(self.uid)

                    attrNode.append(uidNode)

                    nameNode = Element('NAME')

                    nameNode.text = str(attr[0])

                    attrNode.append(nameNode)

                    valueNode = Element('VALUE')

                    valueNode.text = str(attr[1])

                    attrNode.append(valueNode)

                    node.append(attrNode) 

                else:

                    node.append(attr.toXml())

            if self.conns:

                connNode = Element('CONNLINE')

                connNode.text = self.conns[0].uid

                node.append(connNode)

        except Exception as ex:

            return str(self.uid)

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

        return node 

    '''

        @Override (QEngineeringTextItem)

        @brief  get connected items

        @author Jeongwoo

        @date   2018.05.15

    '''

    def getConnectedItems(self):

        visited = []

        try:

            if self.runs is not None:

                for run in self.runs:

                    items = run.items

                    if items is not None:

                        for item in items:

                            pool = []

                            pool.append(item)

                            while len(pool) > 0:

                                it = pool.pop()

                                visited.append(it)

                                for connector in it.connectors:

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

        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      save Line Data

        @author     kyouho

        @date       2018.08.14

    '''

    def saveLineData(self):

        try:

            docData = AppDocData.instance()

            docData.setLineDataList([self.getLineDataList()])

        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)

    '''

        @brief      return Line Data List

        @author     kyouho

        @date       2018.08.14

    '''

    def getLineDataList(self):

        dataList = []

        try:

            import uuid

            global lineColumnList

            docData = AppDocData.instance()

            attrs = self.getAttributes()

            attrs = attrs.items()

            for index in range(len(lineColumnList)):

                dataList.append('')

            dataList[20] = docData.imgName

            for attr in attrs:

                if type(attr) is not UserInputAttribute:

                    lineProp = docData.getLinePropertiesByUID(attr[0])

                    if lineProp:

                        attrName = lineProp[0][1].upper().replace(' ','')

                    else:

                        attrName = attr[0].upper().replace(' ','')

                    data = attr[1] if attr[1] is not None else ''

                    if attrName == 'NOMINALDIAMETER':

                        dataList[1] = data

                    elif attrName == 'FLUIDCODE':

                        dataList[2] = data

                        dataList[7] = data

                    elif attrName == 'TAGSEQNO':

                        pass

                    elif attrName == 'INSULATIONPURPOSE':

                        dataList[16] = data

                    elif attrName == 'STREAMNO':

                        pass

                    elif attrName == 'LINENO':

                        dataList[3] = data

                    elif attrName == 'PNIDNUMBER':

                        pass

                    elif attrName == 'PIPINGMATERIALSCLASS':

                        dataList[4] = data

                    elif attrName == '':

                        pass

                else:

                    typeUID = attr.attribute

                    value = attr.text

                    lineAttr = docData.getLinePropertiesByUID(typeUID)

                    for index in range(len(lineColumnList)):

                        if lineColumnList[index] == lineAttr[0][1]:

                            dataList[index] = value

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

        if dataList[0] == '':

            dataList[0] = str(uuid.uuid4())

        return dataList