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 SymbolSvgItem import SymbolSvgItem

from EngineeringConnectorItem import QEngineeringConnectorItem

from EngineeringTextItem import QEngineeringTextItem

from UserInputAttribute import UserInputAttribute

equipColumnList = ['UID', 'ITEM_NO', 'SERVICE', 'NO_REQ', 'FLUID', 'DESC_OF_PART', 'OPERATION_CONDITION_TEMP', 'OPERATION_CONDITION_PRESS', 'DESIGN_CONDITION_TEMP', 'DESIGN_CONDITION_PRESS', 'MATERIAL', 'WEIGHT', 'POWER', 'INSULATION', 'PNID_NO', 'REV']

class QEngineeringEquipmentItem(SymbolSvgItem):

    clicked = pyqtSignal(QGraphicsSvgItem)

    ZVALUE = 10

    '''

    '''

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

        SymbolSvgItem.__init__(self, path, uid)

        self.setZValue(QEngineeringEquipmentItem.ZVALUE)

    '''

        @brief  connect attribute

        @author humkyung

        @date   2018.05.03

    '''

    def connectAttribute(self, attributes):

        from QEngineeringTagNoTextItem import QEngineeringTagNoTextItem

        self.attrs.clear()

        rect = self.sceneBoundingRect()

        attrs = [attr for attr in attributes if type(attr) is QEngineeringTagNoTextItem]

        # check if text locates inside equipment

        for attr in attrs:

            if rect.contains(attr.center()):

                self.attrs.append(attr)

                attrs.remove(attr)

                break

        if not self.attrs:

            minDist = None

            selected = None

            # get nearest text from equipment

            for attr in attrs:

                dx = attr.center().x() - rect.center().x()

                dy = attr.center().y() - rect.center().y()

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

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

                    minDist = dist

                    selected = attr

            if selected is not None: self.attrs.append(selected)

    '''

        @brief      generate xml code for equipment

        @author     humkyung

        @date       2018.05.09

    '''

    def toXml(self):

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

        try:

            node = SymbolSvgItem.toXml(self)

            self.toXmlAsAttribute(node)

        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 node 

    '''

        @brief  generate xml code for attribute

        @author humkyung

        @date   2018.05.03

    '''

    def toXmlAsAttribute(self, parentNode):

        for attr in self.attrs:

            parentNode.append(attr.toXml(self, None))

    '''

        @brief      save Line Data

        @author     kyouho

        @date       2018.08.14

    '''

    def saveEquipData(self):

        try:

            from AppDocData import AppDocData

            docData = AppDocData.instance()

            docData.setEquipmentDataList([self.getEquipmentDataList()])

        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 equip Data List

        @author     kyouho

        @date       2018.08.14

    '''

    def getEquipmentDataList(self):

        dataList = []

        try:

            from AppDocData import AppDocData

            global equipColumnList

            docData = AppDocData.instance()

            attrs = self.attrs

            for index in range(len(equipColumnList)):

                dataList.append('')

            dataList[0] = str(self.uid)

            dataList[14] = docData.imgName

            for attr in attrs:

                attrInfo = docData.getSymbolAttributeByUID(attr.attribute)

                attrName = attrInfo[0]

                if equipColumnList.count(attrName):

                    colIndex = equipColumnList.index(attrName)

                    if type(attr) is UserInputAttribute:

                        dataList[colIndex] = attr.text if attr.text is not None else ''

                    elif type(attr) is QEngineeringTextItem:

                        dataList[colIndex] = attr.text()

                    else:

                        dataList[colIndex] = attr.uid

        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 dataList