HYTOS

# coding: utf-8

import sys

import os

import math

from PyQt5.QtGui import *

from PyQt5.QtCore import *

from PyQt5.QtSvg import *

from PyQt5.QtWidgets import (QApplication, QGraphicsItem)

from AppDocData import *

from SymbolSvgItem import SymbolSvgItem

from SymbolAttr import SymbolAttr

from UserInputAttribute import UserInputAttribute

from EngineeringConnectorItem import QEngineeringConnectorItem

class QEngineeringSpecBreakItem(SymbolSvgItem):

    """

    This is engineering specbreak item class

    """

    ZVALUE = 20

    clicked = pyqtSignal(QGraphicsSvgItem)

    '''

    '''

    def __init__(self, path, uid=None, flip=None):

        SymbolSvgItem.__init__(self, path, uid, flip)

        attr = SymbolAttr()

        attr.Attribute = 'UpStream'

        attr.AttributeType = 'Comp Item'

        self.attrs[attr] = None

        attr = SymbolAttr()

        attr.Attribute = 'DownStream'

        attr.AttributeType = 'Comp Item'

        self.attrs[attr] = None

        #attr = SymbolAttr()

        #attr.Attribute = 'SetSpecs'

        #attr.AttributeType = 'Link Dialog'

        #self.attrs[attr] = 'Double Click'

        '''

        attr = SymbolAttr()

        attr.Attribute = 'Spec'

        attr.AttributeType = 'FluidCode'

        self.attrs[attr] = ['P', 'OWS']

        attr = SymbolAttr()

        attr.Attribute = 'Spec'

        attr.AttributeType = 'InsulationPurpose'

        self.attrs[attr] = ['N', 'Ic']

        '''

    '''

        @brief      get attribute

        @author     humkyung

        @date       2018.06.14

        @history    kyouho  2018.07.18  Add only attr QEngineeringTextItem

                    euisung 2019.01.15  return self.attrs

    '''

    def getAttributes(self):

        """

            attributes MUST have UpStream, DownStream attribute

        """

        return self.attrs

    '''

        @brief      generate xml code for nozzle

        @author     humkyung

        @date       2018.07.19

    '''

    def toXml(self):

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

        from EngineeringTextItem import QEngineeringTextItem

        from EngineeringSpecBreakItem import QEngineeringSpecBreakItem

        from SymbolAttr import SymbolAttr

        try:

            node = Element('SYMBOL')

            uidNode = Element('UID')

            uidNode.text = str(self.uid)

            node.append(uidNode)

            nameNode = Element('NAME')

            nameNode.text = self.name

            node.append(nameNode)

            typeNode = Element('TYPE')

            typeNode.text = self.type

            node.append(typeNode)

            # write owner's uid to xml

            if self.owner is not None:

                ownerNode = Element('OWNER')

                ownerNode.text = str(self.owner.uid)

                node.append(ownerNode)

            # up to here

            originNode = Element('ORIGINALPOINT')

            originNode.text = '{},{}'.format(self.origin[0], self.origin[1])

            node.append(originNode)

            connectorsNode = Element('CONNECTORS')

            for connector in self.connectors:

                connectorNode = Element('CONNECTOR')

                connectedItemNode = Element('CONNECTEDITEM')

                connectedItemNode.text = str(connector.connectedItem.uid) if connector.connectedItem is not None else 'None'

                connectPointNode = Element('CONNECTPOINT')

                connectPointNode.text = str(connector.connectPoint[0]) + ',' + str(connector.connectPoint[1])

                sceneConnectPointNode = Element('SCENECONNECTPOINT')

                sceneConnectPointNode.text = str(connector.sceneConnectPoint[0]) + ',' + str(connector.sceneConnectPoint[1])

                connectorNode.append(connectedItemNode)

                connectorNode.append(connectPointNode)

                connectorNode.append(sceneConnectPointNode)

                connectorsNode.append(connectorNode)

            node.append(connectorsNode)

            connectionNode = Element('CONNECTIONPOINT')

            connection_point = []

            if self.connectors is not None:

                for connector in self.connectors:

                    connection_point.append(repr(connector))

            connectionNode.text = '/'.join(connection_point)

            node.append(connectionNode)

            rect = self.sceneBoundingRect()

            locNode = Element('LOCATION')

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

            node.append(locNode)

            sizeNode = Element('SIZE')

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

            node.append(sizeNode)

            angleNode = Element('ANGLE')

            angleNode.text = str(self.angle)

            node.append(angleNode)

            parentSymbolNode = Element('PARENT')

            parentSymbolNode.text = str(self.parentSymbol)

            node.append(parentSymbolNode)

            childSymbolNode = Element('CHILD')

            childSymbolNode.text = str(self.childSymbol)

            node.append(childSymbolNode)

            hasInstrumentLabelNode = Element('HASINSTRUMENTLABEL')

            hasInstrumentLabelNode.text = str(self.hasInstrumentLabel)

            node.append(hasInstrumentLabelNode)

            areaNode = Element('AREA')

            areaNode.text = self.area

            node.append(areaNode)

            attributesNode = Element('SYMBOLATTRIBUTES')

            _attrs = list(self.getAttributes())

            for attr in _attrs:

                if type(attr) is SymbolAttr:

                    _node = attr.toXml()

                    if attr.AttributeType != 'Spec':

                        _node.text = self.attrs[attr]

                    else:

                        _node.text = str(self.attrs[attr][0]) + ',' + str(self.attrs[attr][1])

                    attributesNode.append(_node)

            node.append(attributesNode)

            currentPointModeIndexNode = Element('CURRENTPOINTMODEINDEX')

            currentPointModeIndexNode.text = str(self.currentPointModeIndex)

            node.append(currentPointModeIndexNode)

        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 node