HYTOS

# coding: utf-8

""" This is Load command module """

import os.path

import sys, math

from enum import Enum

import uuid

from PyQt5.QtCore import *

from PyQt5.QtWidgets import *

from PyQt5.QtGui import *

from AppDocData import AppDocData

from AbstractCommand import AbstractCommand

from QtImageViewer import QtImageViewer

from EngineeringAbstractItem import QEngineeringAbstractItem

from EngineeringLineItem import QEngineeringLineItem

from SymbolSvgItem import SymbolSvgItem

from EngineeringEquipmentItem import QEngineeringEquipmentItem

from EngineeringInstrumentItem import QEngineeringInstrumentItem

from EngineeringVendorItem import QEngineeringVendorItem

class ReplaceInsertCommand(AbstractCommand):

    CONDITION_REPLACE = {'HasCon':['item.has_connection'], 'HasNoCon':['not item.has_connection'], 'ConLine':['False'], 'All':['True']}

    CONDITION_INSERT = {'HasCon':['conn.connetionItem'], 'HasNoCon':['not conn.connectedItem'], 'ConLine':['conn.connectedItem and type(conn.connectedItem) is QEngineeringLineItem', 'conn.connectedItem.is_piping(True)'], 'All':['True']}

    display_message = pyqtSignal(Enum, str)

    def __init__(self):

        super(ReplaceInsertCommand, self).__init__(None)

    def execute(self, scene, find_symbol, replace_symbol, replace_action, condition, custom=None, strict=False, auto=True):

        """replace or insert symbol"""

        from App import App

        from AppDocData import AppDocData

        from AppDocData import MessageType

        app_doc_data = AppDocData.instance()

        try:

            if condition == 'Custom':

                conditions = [text for text in custom.split('\n')]

            else:

                conditions = ReplaceInsertCommand.CONDITION_REPLACE[condition] if replace_action == 'Replace' else ReplaceInsertCommand.CONDITION_INSERT[condition]

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

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

                            else [item for item in scene.items()

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

            new_symbols = []

            if replace_action == 'Replace' 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 = QtImageViewer.createSymbolObject(replace_symbol)

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

                for item in new_symbols:

                    QtImageViewer.matchSymbolToLine(scene, item[0], item[1], angle=item[2], flip=item[3], strict=strict, auto=auto)

                    item[0].bind_close_items()

            elif replace_action == 'Insert' and find_symbol != 'Equipment Package':

                # insert

                if not symbol_items:

                    return

                for item in symbol_items:

                    for conn in item.connectors:

                        target = True

                        for condition in conditions:

                            if condition and not eval(condition):

                                target = False

                                break

                        if target:

                            scenePos = conn.sceneBoundingRect().center()

                            svg = QtImageViewer.createSymbolObject(replace_symbol)

                            new_symbols.append([svg, scenePos])

                for item in new_symbols:

                    QtImageViewer.matchSymbolToLine(scene, item[0], item[1], strict=strict, auto=auto)

            elif replace_action == 'Insert'  and find_symbol == 'Equipment Package':

                if not symbol_items:

                    return

                items = [item for item in scene.items() \

                                if (issubclass(type(item), SymbolSvgItem) or type(item) is QEngineeringLineItem) and item.has_connection]

                for package in symbol_items:

                    x, y = package.sceneBoundingRect().x(), package.sceneBoundingRect().y()

                    targets = []

                    for item in items:

                        for conn in item.connectors[:2]:

                            if conn.parentItem().has_connection and not conn.connectedItem and package.includes(conn, margin = 100):

                                matches = [target for target in targets if target.parentItem() is conn.parentItem()]

                                if not matches:

                                    targets.append(conn)

                                else:

                                    ox, oy = abs(x - matches[0].sceneBoundingRect().x()), abs(y - matches[0].sceneBoundingRect().y())

                                    nx, ny = abs(x - conn.sceneBoundingRect().x()), abs(y - conn.sceneBoundingRect().y())

                                    old_dist = math.sqrt(ox * ox + oy * oy)

                                    new_dist = math.sqrt(nx * nx + ny * ny)

                                    if old_dist > new_dist:

                                        targets.remove(matches[0])

                                        targets.append(conn)

                    for target in targets:

                        svg = QtImageViewer.createSymbolObject(replace_symbol)

                        QtImageViewer.matchSymbolToLine(scene, svg, target.sceneBoundingRect().center(), strict=strict, auto=auto)

                        for conn in svg.connectors:

                            if not conn.connectedItem:

                                conn.connect(package)

                                break

        except Exception as ex:

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

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

            self.display_message.emit(MessageType.Error, message)