HYTOS

# coding: utf-8

"""

    This is area configuration module

"""

import os

import sys

from PyQt5.QtCore import *

from PyQt5.QtGui import *

from PyQt5.QtWidgets import *

from functools import reduce, partial

from AppDocData import AppDocData

import BMSetting_UI

class AttributeListModel(QStandardItemModel):

    def __init__(self, columListAll, parent=None, *args):

        QStandardItemModel.__init__(self, parent, *args)

        self.columListAll = columListAll

        for symbol in self.columListAll:

            items = [QStandardItem(symbol)]

            items[0].setEditable(False)

            self.appendRow(items)

        headers = [QStandardItem("Name"), QStandardItem("State")]

        for idx, header in enumerate(headers):

            header.setTextAlignment(Qt.AlignCenter)

            self.setHorizontalHeaderItem(idx, header)

class SymbolListModel(QStandardItemModel):

    def __init__(self, parent=None, *args):

        QStandardItemModel.__init__(self, parent, *args)

        self.SymbolList = [ "DRAIN CUP", "EOL", "FLANGE", "FLANGE BLIND", "ORIFICE", "SCREW CAP", \

            "3WAY", "4WAY", "BALL", "BUTTERFLY", "CHECK", "GATE", "GLOBE", "NEEDLE", \

            "CONTROL(BALL)", "CONTROL(BUTTERFLY)", "CONTROL(CHECK)", "CONTROL(GATE)", "CONTROL(GLOBE)", \

            "MOV(BALL)", "MOV(BUTTERFLY)", "MOV(CHECK)", "MOV(GATE)", "MOV(GLOBE)", "MAGNETIC FLOW METER", "MASS FLOW METER", "ULTRASONIC FLOW METER", "VORTEX FLOW METER", "OPC", \

            "GOOSE NECK VENT", "IP", "PG", "PP", "PSV", "RE", "SAMPLE CONNECTION", "SB", "SPACER", "STRAINER", "TG", "HOSE CONNECTION", "OD" ]

        for symbol in self.SymbolList:

            items = [QStandardItem(symbol)]

            items[0].setEditable(False)

            self.appendRow(items)

        headers = [QStandardItem("Name")]

        for idx, header in enumerate(headers):

            header.setTextAlignment(Qt.AlignCenter)

            self.setHorizontalHeaderItem(idx, header)

class SymbolMappingModel(QStandardItemModel):

    """This is SymbolMapping Model class"""

    def __init__(self):

        """constructor"""

        QStandardItemModel.__init__(self)

        app_doc_data = AppDocData.instance()

        project = app_doc_data.getCurrentProject()

        if project is not None:

            self.clear()

            self.load_symbol_info()

            headers = [QStandardItem("Name"), QStandardItem("State")]

            for idx, header in enumerate(headers):

                header.setTextAlignment(Qt.AlignCenter)

                self.setHorizontalHeaderItem(idx, header)

    def on_symbol_state_changed(self):

        pass

    def load_symbol_info(self):

        """load symbol information and display it on tree view"""

        try:

            app_doc_data = AppDocData.instance()

            symbolTypeList = app_doc_data.getSymbolTypeList()

            for row, symbolType in enumerate(symbolTypeList):

                items = [QStandardItem(symbolType[2]), QStandardItem('')]

                items[0].setData(symbolType, Qt.UserRole)

                items[0].setEditable(False)

                items[0].setSelectable(False)

                items[1].setEditable(False)

                items[1].setSelectable(False)

                symbolList = app_doc_data.getSymbolListByType('UID', symbolType[0])

                for symbol in symbolList:

                    childs = [QStandardItem(symbol.getName()), QStandardItem('')]

                    childs[0].setData(symbol.getUid(), Qt.UserRole)

                    childs[0].setEditable(False)

                    _, svg = app_doc_data.read_symbol_shape(symbol.sName)

                    if svg:

                        symbol.pixmap = QPixmap()

                        symbol.pixmap.loadFromData(svg if isinstance(svg, bytes) else svg.encode())

                        icon = QIcon(symbol.pixmap)

                        childs[0].setIcon(icon)

                        childs[0].svgFilePath = None  # save svg file path

                    else:

                        svgPath = symbol.getSvgFileFullPath()

                        symbol.pixmap = QPixmap(svgPath)

                        icon = QIcon(symbol.pixmap)

                        childs[0].setIcon(icon)

                        childs[0].svgFilePath = svgPath  # save svg file path

                    items[0].appendRow(childs)

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

                self.appendRow(items)

        except Exception as ex:

            from App import App

            from AppDocData import MessageType

            message = f'error occurred({ex}) in {sys.exc_info()[-1].tb_frame.f_code.co_filename}:' \

                      f'{sys.exc_info()[-1].tb_lineno}'

            App.mainWnd().addMessage.emit(MessageType.Error, message)

class BMSettingDialog(QDialog):

    def __init__(self, parent, columListAll, columnOrder):

        QDialog.__init__(self, parent)

        self.ui = BMSetting_UI.Ui_BMSettingDialog()

        self.ui.setupUi(self)

        self.isAccepted = False

        self.symbol_mapping = {}

        self.columListAll = columListAll

        self.columnOrder = columnOrder

        self.symbolListModel = SymbolListModel()

        self.ui.tableViewSymbolList.setModel(self.symbolListModel)

        self.symbolMappingModel = SymbolMappingModel()

        self.symbolMappingModel.invisibleRootItem()

        self.ui.treeViewSymbolMapping.setModel(self.symbolMappingModel)

        self.ui.treeViewSymbolMapping.expandAll()

        self.add_check_box()

        self.load_symbol_mapping()

        self.attributeListModel = AttributeListModel(self.columListAll)

        self.ui.tableViewAttributeList.setModel(self.attributeListModel)

        self.ui.tableViewSymbolList.horizontalHeader().setStretchLastSection(True)

        self.ui.treeViewSymbolMapping.header().setStretchLastSection(True)

        self.ui.tableViewAttributeList.horizontalHeader().setStretchLastSection(True)

        self.ui.tableViewSymbolList.setSizeAdjustPolicy(QAbstractScrollArea.AdjustToContents)

        self.ui.treeViewSymbolMapping.setSizeAdjustPolicy(QAbstractScrollArea.AdjustToContents)

        self.ui.tableViewAttributeList.setSizeAdjustPolicy(QAbstractScrollArea.AdjustToContents)

        self.ui.treeViewSymbolMapping.resizeColumnToContents(0)

        self.ui.treeViewSymbolMapping.setColumnWidth(1, 50)

        self.ui.tableViewSymbolList.resizeColumnsToContents()

        self.ui.tableViewAttributeList.resizeColumnsToContents()

        self.ui.tableViewAttributeList.setColumnWidth(1, 50)

        self.ui.tableViewSymbolList.clicked.connect(self.on_symbol_changed)

        self.ui.buttonBox.accepted.connect(self.accept)

        self.ui.buttonBox.rejected.connect(self.reject)

        self.on_symbol_changed(self.symbolListModel.index(0, 0))

    def load_symbol_mapping(self):

        app_doc_data = AppDocData.instance()

        for symbol in self.symbolListModel.SymbolList:

            self.symbol_mapping[symbol] = []

        configs = app_doc_data.getConfigs(section='BM Symbol Mapping')

        for config in configs:

            self.symbol_mapping[config.key] = config.value.split(',')

    def add_check_box(self):

        for row in range(self.symbolMappingModel.rowCount()):

            parent_index = self.symbolMappingModel.index(row, 0)

            child_count = self.symbolMappingModel.rowCount(parent_index)

            for child_row in range(child_count):

                checkBox = QCheckBox()

                index = self.symbolMappingModel.index(child_row, 1, parent_index)

                symbol_index = self.symbolMappingModel.index(child_row, 0, parent_index)

                symbol_id = self.symbolMappingModel.itemFromIndex(symbol_index).data(Qt.UserRole)

                checkBox.stateChanged.connect(partial(self.on_checkBox_changed, str(symbol_id)))

                self.ui.treeViewSymbolMapping.setIndexWidget(index, checkBox)

    def on_checkBox_changed(self, symbol_id, checkState):

        index = self.ui.tableViewSymbolList.currentIndex()

        selected_symbol = self.symbolListModel.itemFromIndex(index)

        if checkState is int(Qt.Checked) and not symbol_id in self.symbol_mapping[selected_symbol.text()]:

            self.symbol_mapping[selected_symbol.text()].append(symbol_id)

        elif checkState is int(Qt.Unchecked) and symbol_id in self.symbol_mapping[selected_symbol.text()]:

            self.symbol_mapping[selected_symbol.text()].remove(symbol_id)

    def on_symbol_changed(self, index):

        selected_symbol = self.symbolListModel.itemFromIndex(index)

        for row in range(self.symbolMappingModel.rowCount()):

            category_index = self.symbolMappingModel.index(row, 0)

            child_count = self.symbolMappingModel.rowCount(category_index)

            for child_row in range(child_count):

                child_index = self.symbolMappingModel.index(child_row, 0, category_index)

                id2_symbol_item = self.symbolMappingModel.itemFromIndex(child_index)

                id2_symbol_uid = str(id2_symbol_item.data(Qt.UserRole))

                checkBox_index = self.symbolMappingModel.index(child_row, 1, category_index)

                checkBox = self.ui.treeViewSymbolMapping.indexWidget(checkBox_index)

                checkBox.setEnabled(True)

                if id2_symbol_uid in self.symbol_mapping[selected_symbol.text()]:

                    checkBox.setCheckState(Qt.Checked)

                else:

                    checkBox.setCheckState(Qt.Unchecked)

                    matches = [key for key, value in self.symbol_mapping.items() if key != id2_symbol_item.text() and id2_symbol_uid in value]

                    if matches:

                        checkBox.setEnabled(False) 

    def reject(self):

        QDialog.reject(self)

    def accept(self):

        try:

            self.isAccepted = False

            QDialog.accept(self)

        except Exception as ex:

            from App import App

            from AppDocData import MessageType

            message = 'error occurred({}) 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)

    def showDialog(self):

        self.setWindowFlags(self.windowFlags() & ~Qt.WindowCloseButtonHint & ~Qt.WindowContextHelpButtonHint)

        self.exec_()

        return self.isAccepted