HYTOS

# coding: utf-8

""" This is text item edit dialog module """

import os

import sys

from PyQt5.QtCore import *

from PyQt5.QtGui import *

from PyQt5.QtWidgets import *

from AppDocData import AppDocData

import ReplaceSymbol_UI

from EngineeringTextItem import QEngineeringTextItem

from SymbolSvgItem import SymbolSvgItem

from EngineeringLineItem import QEngineeringLineItem

from EngineeringVendorItem import QEngineeringVendorItem

class QReplaceSymbolDialog(QDialog):

    """ This is symbol item replace and insert dialog class """

    CONDITION_REPLACE = {'radioButtonHasCon':['item.has_connection'], 'radioButtonHasNoCon':['not item.has_connection'], 'radioButtonConLine':['False'], 'radioButtonAll':['True']}

    CONDITION_INSERT = {'radioButtonHasCon':['conn.connetionItem'], 'radioButtonHasNoCon':['not conn.connectedItem'], 'radioButtonConLine':['conn.connectedItem and type(conn.connectedItem) is QEngineeringLineItem'], 'radioButtonAll':['True']}

    def __init__(self, parent):

        QDialog.__init__(self, parent)

        self.mainWindow = parent

        self.view = self.mainWindow.graphicsView

        self.ui = ReplaceSymbol_UI.Ui_ReplaceSymbolDialog()

        self.ui.setupUi(self)

        self.ui.pushButtonRun.clicked.connect(self.run)

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

        self.init_symbol_list()

    def init_symbol_list(self):

        app_doc_data = AppDocData.instance()

        symbol_list = app_doc_data.getTargetSymbolList(all=True)

        # for nozzle and equipment package

        self.ui.comboBoxFind.addItem('Equipment Package')

        for symbol in symbol_list:

            self.ui.comboBoxFind.addItem(symbol.getName())

            self.ui.comboBoxReplace.addItem(symbol.getName())

    def run(self):

        find_symbol = self.ui.comboBoxFind.currentText()

        replace_symbol = self.ui.comboBoxReplace.currentText()

        condition = self.ui.buttonGroup_2.checkedButton().objectName()

        if condition == 'radioButtonCustom':

            conditions = [text for text in self.ui.plainTextEdit.toPlainText().split('\n')]

        else:

            conditions = QReplaceSymbolDialog.CONDITION_REPLACE[condition] if self.ui.radioButtonReplace.isChecked() else QReplaceSymbolDialog.CONDITION_INSERT[condition]

        symbol_items = [item for item in self.view.scene().items()

                            if issubclass(type(item), SymbolSvgItem) and find_symbol == item.name] if find_symbol != 'Equipment Package'

                        else [item for item in self.view.scene().items()

                            if issubclass(type(item), QEngineeringVendorItem) and item.pack_type == 'Equipment Package']

        new_symbols = []

        if self.ui.radioButtonReplace.isChecked() and find_symbol != 'Equipment Package':

            # replace

            for item in reversed(symbol_items):

                for condition in conditions:

                    if condition and not eval(condition):

                        symbol_items.remove(item)

                        break

            if not symbol_items:

                    return

            for item in symbol_items:

                scenePos = item.mapToScene(item.transformOriginPoint())

                item.transfer.onRemoved.emit(item)

                svg = self.view.createSymbolObject(replace_symbol)

                new_symbols.append([svg, scenePos, item.angle, item.flip])

            for item in new_symbols:

                self.view.matchSymbolToLine(item[0], item[1], angle=item[2], flip=item[3])

                item[0].bind_close_items()

        elif not self.ui.radioButtonReplace.isChecked() and find_symbol != 'Equipment Package':

            # insert

            if not symbol_items:

                return

            for item in symbol_items:

                for conn in item.connectors:

                    for condition in conditions:

                        if condition and eval(condition):

                            scenePos = conn.sceneBoundingRect().center()

                            svg = self.view.createSymbolObject(replace_symbol)

                            new_symbols.append([svg, scenePos])

            for item in new_symbols:

                self.view.matchSymbolToLine(item[0], item[1])

        elif not self.ui.radioButtonReplace.isChecked() and find_symbol == 'Equipment Package':

            pass

    def reject(self):

        """close a dialog"""

        QDialog.reject(self)