HYTOS

try:

    from PyQt5.QtCore import *

    from PyQt5.QtGui import *

    from PyQt5.QtWidgets import *

except ImportError:

    try:

        from PyQt4.QtCore import *

        from PyQt4.QtGui import *

    except ImportError:

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

from AppDocData import AppDocData

import os

import sys

import math

from SymbolSvgItem import SymbolSvgItem

from QEngineeringLineNoTextItem import QEngineeringLineNoTextItem

from enum import Enum

'''

    @brief      ItemType

    @author     Jeongwoo

    @date       2018.04.27

    @history    2018.05.10  Jeongwoo    Add LINE_NO

'''

class ItemType(Enum):

    SYMBOL = 1

    NOTE = 2

    LINE_NO = 3

class QResultPropertyTableWidget(QTableWidget):

    def __init__(self):

        QTableWidget.__init__(self)

        self.symData = None

        self.initResultPropertyTableWidget()

    '''

        @brief      Initialize TableWidget

        @author     Jeongwoo

        @date       18.04.13

        @history    .

    '''

    def initResultPropertyTableWidget(self):

        self.setColumnCount(2)

        self.setRowCount(13)

        self.horizontalHeader().hide()

        self.verticalHeader().hide()

        self.horizontalHeader().setStretchLastSection(True)

        self.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)

        self.setEditTriggers(QAbstractItemView.NoEditTriggers)

    '''

        @brief      Slot to accept item click event

        @author     Jeongwoo

        @date       18.04.13

        @history    humkyung 2018.04.17 check if given symbol type is SymbolSvgItem

    '''

    @pyqtSlot(SymbolSvgItem)

    def getClickedSymbol(self, symbol):

        if issubclass(type(symbol), SymbolSvgItem):

            self.symData = symbol

            self.symbolChanged()

    '''

        @brief      Slot to accept Note item click event

        @author     Jeongwoo

        @date       18.04.27

    '''

    @pyqtSlot(str, list)

    def getClickedNote(self, noteNoStr, noteContentsList):

        self.noteChanged(noteNoStr, noteContentsList)

    '''

        @brief      Slot to accept Line No Item Click event

        @author     Jeongwoo

        @date       18.05.10

    '''

    @pyqtSlot(QEngineeringLineNoTextItem)

    def getClickedLineNo(self, lineNoItem):

        self.lineNoChanged(lineNoItem)

    '''

        @brief      Reset table with new SymbolSvgItem

        @author     Jeongwoo

        @date       18.04.13

        @history    .

    '''

    def symbolChanged(self):

        self.initTitleCell(ItemType.SYMBOL)

        self.initContentsCell()

    '''

        @brief      Reset table with note info

        @author     Jeongwoo

        @date       18.04.27

    '''

    def noteChanged(self, noteNoStr, noteContentsList):

        self.initTitleCell(ItemType.NOTE)

        self.initNoteCell(noteNoStr, noteContentsList)

    '''

        @brief      Reset table with line no item

        @author     Jeongwoo

        @date       18.05.10

    '''

    def lineNoChanged(self, lineNoItem):

        self.initTitleCell(ItemType.LINE_NO)

        self.initLineNoCell(lineNoItem)

    '''

        @brief      Initialize Title Cell

        @author     Jeongwoo

        @date       18.04.13

        @history    2018.04.27 Jeongwoo Add if-statement by ItemType

                    2018.05.10 Jeongwoo Add if-statement LINE_NO ItemType

    '''

    def initTitleCell(self, type):

        self.clear()

        if type == ItemType.SYMBOL:

            self.setItem(0,     0,      QTableWidgetItem("심볼명"))

            self.setItem(1,     0,      QTableWidgetItem("타입"))

            self.setItem(2,     0,      QTableWidgetItem("각도"))

        elif type == ItemType.NOTE:

            self.setItem(0,     0,      QTableWidgetItem("노트번호"))

            self.setItem(1,     0,      QTableWidgetItem("노트내용"))

        elif type == ItemType.LINE_NO:

            '''

            DO NOTHING / initLineNoCell

            '''

        for index in range(self.rowCount()):

            item = self.item(index, 0)

            if item is not None:

                item.setBackground(QColor(220, 220, 220))

    '''

        @brief      Initialize Contents Cell

        @author     Jeongwoo

        @date       18.04.13

        @history    humkyung 2018.06.14 display symbol attributes 

    '''

    def initContentsCell(self):

        from QEngineeringInstrumentItem import QEngineeringInstrumentItem

        if self.symData is not None:

            self.setItem(0,     1,      QTableWidgetItem(self.symData.name))

            self.setItem(1,     1,      QTableWidgetItem(self.symData.type))

            self.setItem(2,     1,      QTableWidgetItem(str(round(math.degrees(self.symData.angle)))))

            # display attributes of symbol

            attrs = self.symData.getAttributes()

            if attrs is not None:

                attrItems = list(attrs.items())

                for index in range(len(attrItems)):

                    key = attrItems[index][0]

                    value = attrItems[index][1]

                    keyItem = QTableWidgetItem(key)

                    keyItem.setBackground(QColor(220, 220, 220))

                    self.setItem(1 + index, 0, keyItem)

                    self.setItem(1 + index, 1, QTableWidgetItem(value))

            # up to here

        else:

            for index in range(self.rowCount()):

                item = self.item(index, 1)

                if item is not None:

                    item.setText("")

    '''

        @brief      Initialize Note Contents Cell

        @author     Jeongwoo

        @date       18.04.27

    '''

    def initNoteCell(self, noteNoStr, noteContentsList):

        self.setItem(0,     1,      QTableWidgetItem(noteNoStr))

        for index in range(len(noteContentsList)):

            self.setItem(1+index,     1,      QTableWidgetItem(noteContentsList[index]))

    '''

        @brief      Initialize Line No Contents Cell

        @author     Jeongwoo

        @date       18.05.10

    '''

    def initLineNoCell(self, lineNoItem):

        lineNoTitleItem = QTableWidgetItem("Line No.")

        lineNoTitleItem.setBackground(QColor(220, 220, 220))

        self.setItem(0, 0, lineNoTitleItem)

        self.setItem(0, 1, QTableWidgetItem(lineNoItem.text()))

        attrs = lineNoItem.getLineNoAttributes()

        if attrs is not None:

            attrItems = list(attrs.items())

            for index in range(len(attrItems)):

                key = attrItems[index][0]

                value = attrItems[index][1]

                keyItem = QTableWidgetItem(key)

                keyItem.setBackground(QColor(220, 220, 220))

                self.setItem(1 + index, 0, keyItem)

                self.setItem(1 + index, 1, QTableWidgetItem(value))