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

from QEngineeringTextItem import QEngineeringTextItem

from QEngineeringNoteItem import QEngineeringNoteItem

from QEngineeringLineNoTextItem import QEngineeringLineNoTextItem

from AppDocData import AppDocData

import os

import re

class QResultTreeWidget(QTreeWidget):

    TREE_DATA_ROLE = Qt.ToolTipRole

    #Add Signal

    singleClicked = pyqtSignal(SymbolSvgItem)

    noteNoSingleClicked = pyqtSignal(str, list)

    lineNoSingleClicked = pyqtSignal(QEngineeringLineNoTextItem)

    '''

        @history    2018.05.11  Jeongwoo    Add Context Menu settings

    '''

    def __init__(self, imageViewer):

        QTreeWidget.__init__(self)

        self.itemClicked.connect(self.itemClickEvent)

        self.imageViewer = imageViewer

        self.scene = imageViewer.scene

        self.root = None

        self.setContextMenuPolicy(Qt.CustomContextMenu)

        self.customContextMenuRequested.connect(self.openContextMenu)

    '''

        @brief      Show Context Menu

        @author     Jeongwoo

        @date       18.05.11

    '''

    def openContextMenu(self, position):

        indexes = self.selectedIndexes()

        itemPosition = self.mapTo(self, position)

        item = self.itemAt(itemPosition)

        data = item.data(0, self.TREE_DATA_ROLE)

        if len(indexes) > 0 and data is not None and type(data) is QEngineeringLineNoTextItem:

            index = indexes[0]

            menu = QMenu()

            pickColorAction = QAction(self.tr("색상 설정"))

            pickColorAction.triggered.connect(lambda : self.pickColorClickEvent(item))

            menu.addAction(pickColorAction)

            menu.exec_(self.viewport().mapToGlobal(position))

    '''

        @brief      Pick Color for Line No

        @author     Jeongwoo

        @date       18.05.11

        @history    18.05.14    Jeongwoo    Change method to change color by changeTreeWidgetItemColorRecursively()

    '''

    def pickColorClickEvent(self, lineNoTreeWidgetItem):

        ''''''

        color = QColorDialog.getColor() # Dialog returns QColor

        if color is not None:

            lineNoTreeWidgetItem.setForeground(0, QBrush(color))

            lineNoTreeWidgetItem.setFont(0, lineNoTreeWidgetItem.font(0))

            lineNoItem = lineNoTreeWidgetItem.data(0, self.TREE_DATA_ROLE)

            lineNoItem.setColor(color.name())

            connectedItems = lineNoItem.getConnectedItems()

            if connectedItems is not None:

                for connectedItem in connectedItems:

                    connectedItem.setColor(color.name())

                self.changeTreeWidgetItemColorRecursively(lineNoTreeWidgetItem, color)

    '''

        @brief      Change Color recursively

        @author     Jeongwoo

        @date       18.05.14

    '''

    def changeTreeWidgetItemColorRecursively(self, item, color):

        for index in range(item.childCount()):

            child = item.child(index)

            child.setForeground(0, QBrush(color))

            child.setFont(0, child.font(0))

            self.changeTreeWidgetItemColorRecursively(child, color)

    '''

        @brief      Clear TreeWidget and Set Current PID

        @author     Jeongwoo

        @date       18.04.11

        @history    2018.04.26  Jeongwoo    Add Child [SYMBOLS, NOTES] into root item

                    2018.05.09  Jeongwoo    Change method to add default tree items

    '''

    def setCurrentPID(self, baseName):

        self.clear()

        self.root = QTreeWidgetItem(self, [os.path.splitext(baseName)[0]])

        self.root.addChild(QTreeWidgetItem(['EQUIPMENTS']))

        self.root.addChild(QTreeWidgetItem(['SYMBOLS']))

        self.root.addChild(QTreeWidgetItem(['NOTES']))

    '''

        @brief  create tree item for pipe run if need

        @author humkyung

        @date   2018.05.15

    '''

    def addPipeRunTreeItemIfNeed(self, lineNoTreeItem, pipeRun):

        for index in range(lineNoTreeItem.childCount()):

            treeItem = lineNoTreeItem.child(index)

            itemData = treeItem.data(0, self.TREE_DATA_ROLE)

            if itemData is not None and itemData == pipeRun.uid:

                return treeItem

        runItem = QTreeWidgetItem(['RUNS {}'.format(lineNoTreeItem.childCount()+1)])

        runItem.setData(0, self.TREE_DATA_ROLE, pipeRun.uid)

        lineNoTreeItem.addChild(runItem)

        return runItem

    '''

        @brief      add a tree item

        @author     humkyung

        @date       2018.04.17

        @history    humkyung 2018.04.19 treat a case parent is changed

                    Jeongwoo 2018.04.26 QEngineeringTextItem → QEngineeringLineNoTextItem

                                        Insert if-statement for QEngineeringNoteItem

                                        Add if-statement for QEngineeringLineNoTextItem

                    Jeongwoo 2018.05.11 Set color when SymbolSvgItem moved into new parent(Line No)

    '''

    def addTreeItem(self, parent, child):

        item = None

        if (not hasattr(child, 'treeItem')) or (child.treeItem is None):

            if issubclass(type(child), SymbolSvgItem):

                if AppDocData.instance().isEquipmentByType(child.type):

                    symbolsRootItem = self.findItems('EQUIPMENTS', Qt.MatchExactly|Qt.MatchRecursive, 0)

                    symbolsRootItem = symbolsRootItem[0]

                    item = QTreeWidgetItem(symbolsRootItem, [child.name])

                else:

                    symbolsRootItem = self.findItems('SYMBOLS', Qt.MatchExactly|Qt.MatchRecursive, 0)

                    symbolsRootItem = symbolsRootItem[0]

                    item = QTreeWidgetItem(symbolsRootItem, [child.name])

                iconPath = os.path.join(AppDocData.instance().getCurrentProject().getSvgFilePath(), child.type , child.name + ".svg")

                item.setIcon(0, QIcon(iconPath))

                item.setData(0, self.TREE_DATA_ROLE, child)

                child.treeItem = item

                symbolsRootItem.addChild(item)

            elif type(child) is QEngineeringLineNoTextItem:

                item = QTreeWidgetItem([child.text()])

                item.setFlags(item.flags() | Qt.ItemIsUserCheckable)

                item.setData(0, self.TREE_DATA_ROLE, child)

                item.setCheckState(0, Qt.Unchecked)

                child.treeItem = item

                self.root.insertChild(0, item)

            elif (type(child) is QEngineeringNoteItem):

                item = QTreeWidgetItem([child.text()])

                item.setData(0, self.TREE_DATA_ROLE, child)

                child.treeItem = item

                if parent is not None:

                    parent.addChild(item)

        elif issubclass(type(child), SymbolSvgItem):  # change item's parent

            foundItems = self.findItems(child.name, Qt.MatchExactly|Qt.MatchRecursive, 0)

            if foundItems is not None:

                for item in foundItems:

                    data = item.data(0, self.TREE_DATA_ROLE)

                    if data is not None and (data == child) and (parent is not item.parent()):

                        parentData = parent.data(0, self.TREE_DATA_ROLE)

                        if type(parentData) is QEngineeringLineNoTextItem:

                            for index in range(len(parentData.runs)):

                                runGroup = parentData.runs[index]

                                if data in runGroup.items:

                                    item.parent().removeChild(item) # remove item from original parent

                                    runItem = self.addPipeRunTreeItemIfNeed(parent, runGroup) #parent.child(index)

                                    brush = parent.foreground(0)

                                    item.setForeground(0, brush)

                                    item.data(0, self.TREE_DATA_ROLE).setColor(brush.color().name())

                                    item.setFont(0, item.font(0))

                                    runItem.addChild(item)

                                else:

                                    print(item.text(0) + " Not Found on " + parent.text(0))

                        #brush = parent.foreground(0)

                        #item.setForeground(0, brush)

                        #item.data(0, self.TREE_DATA_ROLE).setColor(brush.color().name())

                        #item.setFont(0, item.font(0))

                        #parent.addChild(item)

                        break

        elif type(child) is QEngineeringLineNoTextItem:

            foundItems = self.findItems(child.text(), Qt.MatchExactly|Qt.MatchRecursive, 0)

            if foundItems is not None:

                for item in foundItems:

                    data = item.data(0, self.TREE_DATA_ROLE)

                    if data is not None and (data == child):

                        #for index in range(len(data.runs)):

                        #    item.addChild(QTreeWidgetItem(['RUNS' + str(index+1)]))

                        return item

        return item

    '''

        @brief      Scene Changed Listener

        @author     Jeongwoo

        @date       18.04.11

        @history    Jeongwoo 2018.04.25 Add QEngineeringNoteItem to list for loop  / NOT QEngineeringTextItem

                    Jeongwoo 2018.04.26 Change changedSceneItems conditions in loop (QEngineeringTextItem → QEngineeringLineNoTextItem)

    '''

    lastSceneItems = None

    def sceneChanged(self, sceneItems):

        changedSceneItems = [item for item in sceneItems if ((issubclass(type(item), SymbolSvgItem)) or (type(item) is QEngineeringNoteItem) or (type(item) is QEngineeringLineNoTextItem)) 

                            and (not hasattr(item, 'treeItem') or item.treeItem is None)] # Sublist includes SymbolSvgItem

        self.initResultTreeWidget(changedSceneItems)

    '''

        @brief      Initialize TreeWidget

                    Tree Item Clear and Add Item

        @author     Jeongwoo

        @date       2018.04.11

        @history    2018.04.12 Jeongwoo    Declare self.TREE_DATA_ROLE for QTreeWidgetItem.data(column, role)

        @history    2018.04.17 humkyung NOT remove child of root

                    2018.04.25 Jeongwoo     Add QEngineeringNoteItem Child

                    2018.04.26 Jeongwoo     Change method to insert child

    '''

    def initResultTreeWidget(self, items):

        for item in items:

            if (type(item) is QEngineeringNoteItem):

                notesRootItem = self.findItems('NOTES', Qt.MatchExactly|Qt.MatchRecursive, 0)

                notesRootItem = notesRootItem[0]

                self.addTreeItem(notesRootItem, item)

            else:

                self.addTreeItem(self.root, item)

        notesRootItem = self.findItems('NOTES', Qt.MatchExactly|Qt.MatchRecursive, 0)

        if notesRootItem is not None and len(notesRootItem) > 0:

            notesRootItem[0].sortChildren(0, Qt.AscendingOrder)

        self.expandAll()

    '''

        @brief      TreeWidget Item click listener

        @author     Jeongwoo

        @date       18.04.11

        @history    18.04.12    Jeongwoo    Declare self.TREE_DATA_ROLE for QTreeWidgetItem.data(column, role)

                    18.04.13    Jeongwoo    Signal 'singleClicked' emit SymbolSvgItem

                                            CenterOn() with Symbol's center

                    18.04.25    Jeongwoo    Add QEngineeringNoteItem Click event

                    18.04.26    Jeongwoo    QEngineeringTextItem → QEngineeringLineNoTextItem

                    18.04.27    Jeongwoo    Find NOTE Contents when NOTE No. Clicked

                                            Move singClicked to if-statement type(itemData) is SymbolSvgItem:

                    18.05.10    Jeongwoo    Add lineNoSingleClicked emit

    '''

    lastClickedItem = None

    def itemClickEvent(self, item, columnNo, isSvgClick = False):

        print("Item " + item.text(0) + " clicked")

        itemData = item.data(0, self.TREE_DATA_ROLE)

        if self.lastClickedItem is not None:

            self.scene.removeItem(self.lastClickedItem)

        if itemData is not None: ## Not PID Name

            if issubclass(type(itemData), SymbolSvgItem):

                ## Draw rectangle on selected symbol

                rect = itemData.sceneBoundingRect()

                graphicItem = self.scene.addRect(rect.left(), rect.top(), rect.width(), rect.height(), QPen(QColor(255,255,0,100)), QBrush(QColor(255,255,0,100)))

                if isSvgClick == False:

                    self.imageViewer.centerOn(rect.center())

                ## Send new event to imageViewer's zoomImage Method

                self.imageViewer.zoomImage(True, QMouseEvent(QEvent.MouseButtonPress, self.imageViewer.mapFromScene(QPointF(rect.left(), rect.top())), Qt.LeftButton, Qt.LeftButton, Qt.NoModifier), 3)

                self.lastClickedItem = graphicItem

                self.singleClicked.emit(itemData)

            elif type(itemData) is QEngineeringLineNoTextItem:

                rect = itemData.sceneBoundingRect()

                graphicItem = self.scene.addRect(rect.left(), rect.top(), rect.width(), rect.height(), QPen(QColor(255,255,0,100)), QBrush(QColor(255,255,0,100)))

                if isSvgClick == False:

                    self.imageViewer.centerOn(rect.center())

                ## Send new event to imageViewer's zoomImage Method

                self.imageViewer.zoomImage(True, QMouseEvent(QEvent.MouseButtonPress, self.imageViewer.mapFromScene(QPointF(rect.left(), rect.top())), Qt.LeftButton, Qt.LeftButton, Qt.NoModifier), 3)

                self.lastClickedItem = graphicItem

                self.lineNoSingleClicked.emit(itemData)

            elif type(itemData) is QEngineeringNoteItem:

                rect = itemData.sceneBoundingRect()

                graphicItem = self.scene.addRect(rect.left(), rect.top(), rect.width(), rect.height(), QPen(QColor(255,255,0,100)), QBrush(QColor(255,255,0,100)))

                if isSvgClick == False:

                    self.imageViewer.centerOn(rect.center())

                ## Send new event to imageViewer's zoomImage Method

                self.imageViewer.zoomImage(True, QMouseEvent(QEvent.MouseButtonPress, self.imageViewer.mapFromScene(QPointF(rect.left(), rect.top())), Qt.LeftButton, Qt.LeftButton, Qt.NoModifier), 3)

                self.lastClickedItem = graphicItem

                #noteContentsList = self.findNoteContents(itemData.text())

                noteContentsList = itemData.findNoteContents(itemData.text())

                self.noteNoSingleClicked.emit(itemData.text(), noteContentsList)

    '''

        @brief  find item which has data is given item

        @author humkyung

        @date   2018.04.23

    '''

    def findItemByData(self, item):

        if issubclass(type(item), SymbolSvgItem):

            foundItems = self.findItems(item.name, Qt.MatchExactly|Qt.MatchRecursive, 0)

            for foundItem in foundItems:

                data = foundItem.data(0, self.TREE_DATA_ROLE)

                if data is not None and data is item:

                    return foundItem

        elif type(item) is QEngineeringLineNoTextItem:

            foundItems = self.findItems(item.text() , Qt.MatchExactly|Qt.MatchRecursive, 0)

            for foundItem in foundItems:

                data = foundItem.data(0, self.TREE_DATA_ROLE)

                if data is not None and data is item:

                    return foundItem

        return None

    '''

        @brief      Find QTreeWidgetItem by SymbolSvgItem object

        @author     Jeongwoo

        @date       18.04.11

        @history    18.04.12    Jeongwoo    Change Flags (Qt.MatchExactly → Qt.MatchExactly|Qt.MatchRecursive)

                                            Declare self.TREE_DATA_ROLE for QTreeWidgetItem.data(column, role)

    '''

    @pyqtSlot(SymbolSvgItem)

    def findItem(self, item):

        if issubclass(type(item), SymbolSvgItem):

            foundItems = self.findItems(item.name, Qt.MatchExactly|Qt.MatchRecursive, 0)

            if foundItems is not None:

                for fItem in foundItems:

                    data = fItem.data(0, self.TREE_DATA_ROLE)

                    if data is not None and data == item:

                        self.setCurrentItem(fItem)

                        self.itemClickEvent(fItem, 0, True)

                        return

        ## Not found

        msg = QMessageBox()

        msg.setIcon(QMessageBox.Warning)

        msg.setText('선택하신 심볼의 데이터를 찾을 수 없습니다.')

    '''

        @brief  remove given item

        @author humkyung

        @date   2018.04.23

    '''

    @pyqtSlot(QGraphicsItem)

    def itemRemoved(self, item):

        foundItem = self.findItemByData(item)

        if foundItem is not None:

            foundItem.parent().removeChild(foundItem)