path: root/src/main/python/shapes/shapes.py

from PyQt5 import QtCore, QtWidgets
from PyQt5.QtCore import (QEvent, QFile, QIODevice, QMimeData, QPointF, QRect,
                          QRectF, QSizeF, Qt)
from PyQt5.QtGui import (QBrush, QColor, QCursor, QDrag, QFont, QImage,
                         QPainter, QPainterPath, QPen, QTransform, QTextCursor)
from PyQt5.QtSvg import QGraphicsSvgItem, QSvgRenderer
from PyQt5.QtWidgets import (QGraphicsColorizeEffect, QGraphicsEllipseItem,
                             QGraphicsItem, QGraphicsPathItem,
                             QGraphicsProxyWidget, QGraphicsSceneHoverEvent,
                             QLineEdit, QMenu, QGraphicsTextItem)

from .line import Line, findIndex
from utils.app import fileImporter

from utils.app import fileImporter

class ItemLabel(QGraphicsTextItem):
    def __init__(self, pos, parent=None):
        super().__init__(parent=parent)
        self.setPlainText("abc")
        self.setFlags(QGraphicsItem.ItemIsMovable |
                      QGraphicsItem.ItemIsSelectable |
                      QGraphicsItem.ItemIsFocusable)
        self.setFlag(QGraphicsItem.ItemSendsGeometryChanges, True)
        self.setTextInteractionFlags(Qt.NoTextInteraction)
        self.setPos(self.parentItem().boundingRect().bottomLeft())

    def mouseDoubleClickEvent(self, event):
        self.setTextInteractionFlags(Qt.TextEditorInteraction)
        self.setFocus()
        super(ItemLabel, self).mouseDoubleClickEvent(event)

    def focusOutEvent(self, event):
        super(ItemLabel, self).focusOutEvent(event)
        self.setTextInteractionFlags(Qt.NoTextInteraction)


class GripItem(QGraphicsPathItem):
    """
    Extends QGraphicsPathItem to create the structure of the Grabbable points for resizing shapes and connecting lines.
    Takes two parameters, reference item (On which the grip items are to appear) and the path of the item
    """

    def __init__(self, annotation_item, path, parent=None):
        """
        Extends PyQt5's QGraphicsPathItem to create the general structure of the Grabbable points for resizing shapes.
        """
        QGraphicsPathItem.__init__(self, parent)
        self.m_annotation_item = annotation_item
        # set path of item
        self.setPath(path)
        self.setAcceptHoverEvents(True)
        self.setCursor(QCursor(Qt.PointingHandCursor))


    def mouseReleaseEvent(self, event):
        """
        Automatically deselects grip item on mouse release
        """
        self.setSelected(False)
        super(GripItem, self).mouseReleaseEvent(event)


class SizeGripItem(GripItem):
    """
    Extends grip items for vertical and horizontal directions, with hover events and directional changes
    """

    def __init__(self, annotation_item, index, direction=Qt.Horizontal, parent=None):
        self.width = self.height = 0
        if direction is Qt.Horizontal:
            self.height = annotation_item.boundingRect().height()
        else:
            self.width = annotation_item.boundingRect().width()

        path = QPainterPath()
        path.addRect(QRectF(-self.width / 2, -self.height / 2, self.width, self.height))
        super(SizeGripItem, self).__init__(annotation_item, path=path, parent=parent)
        self.setFlag(QGraphicsItem.ItemIsSelectable, True)
        self.setFlag(QGraphicsItem.ItemIsMovable, True)
        self.setFlag(QGraphicsItem.ItemSendsGeometryChanges, True)
        self.setPen(QPen(QColor("black"), -1))
        self.setZValue(2)
        self._direction = direction
        self.m_index = index

    @property
    def direction(self):
        """
        property that returns the current intended resize direction of the grip item object
        """
        return self._direction

    def paint(self, painter, option, widget):
        if self.isSelected() and not self.parentItem().isSelected():
            self.parentItem().setSelected(True)
            self.parentItem().setFlag(QGraphicsSvgItem.ItemIsMovable, False)
        super().paint(painter, option, widget)

    def updatePath(self):
        """updates path of size grip item
        """
        if self._direction is Qt.Horizontal:
            self.height = self.parentItem().boundingRect().height()
        else:
            self.width = self.parentItem().boundingRect().width()
        path = QPainterPath()
        path.addRect(QRectF(-self.width / 2, -self.height / 2, self.width, self.height))
        self.setPath(path)

    def updatePosition(self):
        """updates position of grip items
        """
        self.updatePath()
        pos = self.point(self.m_index)
        self.setEnabled(False)
        self.setPos(pos)
        self.setEnabled(True)

    def point(self, index):
        """
        yields a list of positions of grip items in a node item
        """
        width = self.parentItem().boundingRect().width()
        height = self.parentItem().boundingRect().height()
        if 0 <= index < 4:
            return [
                QPointF(0, -height / 2),
                QPointF(-width / 2, 0),
                QPointF(0, height / 2),
                QPointF(width / 2, 0)
            ][index]

    def hoverEnterEvent(self, event):
        """
        Changes cursor to horizontal resize or vertical resize depending on the direction of the grip item on mouse enter
        """
        # self.setPen(QPen(QColor("black"), 2))
        # self.setBrush(QColor("red"))
        if self._direction == Qt.Horizontal:
            self.setCursor(QCursor(Qt.SizeHorCursor))
        else:
            self.setCursor(QCursor(Qt.SizeVerCursor))
        super(SizeGripItem, self).hoverEnterEvent(event)

    def hoverLeaveEvent(self, event):
        """
        reverts cursor to default on mouse leave
        """
        # self.setPen(QPen(Qt.transparent))
        # self.setBrush(Qt.transparent)
        self.setCursor(QCursor(Qt.ArrowCursor))
        super(SizeGripItem, self).hoverLeaveEvent(event)

    def itemChange(self, change, value):
        """
        Moves position of grip item on resize
        """
        if change == QGraphicsItem.ItemPositionChange and self.isEnabled():
            p = QPointF(self.pos())
            if self.direction == Qt.Horizontal:
                p.setX(value.x())
            elif self.direction == Qt.Vertical:
                p.setY(value.y())
            # Find change in positions
            movement = p - self.pos()
            # Set transform to oppose change in transformation due to parent
            transform = QTransform()
            transform.translate(-movement.x() / 2, -movement.y() / 2)
            self.setTransform(transform, True)
            self.parentItem().resize(self.m_index, movement)
            return p
        return super(SizeGripItem, self).itemChange(change, value)

    def mouseReleaseEvent(self, event):
        super(SizeGripItem, self).mouseReleaseEvent(event)
        # Reset transform and update position
        self.resetTransform()
        self.updatePosition()
        # Make parent item move able
        self.parentItem().setFlag(QGraphicsSvgItem.ItemIsMovable, True)
        # If needed to reset transform of parent set it's position accordingly
        # self.parentItem().setPos(self.parentItem().x() + self.parentItem().transform().dx(), self.parentItem().y() + self.parentItem().transform().dy())
        # self.parentItem().resetTransform()


class LineGripItem(GripItem):
    """Extends grip items for connecting lines , with hover events and mouse events
    """
    circle = QPainterPath()
    circle.addEllipse(QRectF(-10, -10, 20, 20))

    def __init__(self, annotation_item, index, location, parent=None):
        self.path = LineGripItem.circle
        super(LineGripItem, self).__init__(annotation_item, path=self.path, parent=parent)
        self.m_index = index
        self.m_location = location
        self.line = None
        # stores current line which is in process
        self.tempLine = None
        # keep previous hovered item when line drawing in process
        self.previousHoveredItem = None
        self.setFlag(QGraphicsItem.ItemIsSelectable, True)
        self.setPen(QPen(QColor("black"), -1))

    def point(self, index):
        """
        yields a list of positions of grip items in a node item
        """
        width = self.parentItem().boundingRect().width()
        height = self.parentItem().boundingRect().height()
        if 0 <= index < 4:
            return [
                QPointF(0, -height / 2),
                QPointF(-width / 2, 0),
                QPointF(0, height / 2),
                QPointF(width / 2, 0)
            ][index]

    def updatePosition(self):
        pos = self.point(self.m_index)
        self.setEnabled(False)
        self.setPos(pos)
        self.setEnabled(True)
        if self.line:
            self.line.updateLine()

    def mousePressEvent(self, mouseEvent):
        """Handle all mouse press for this item
        """
        if mouseEvent.button() != Qt.LeftButton:
            return
        # initialize a line and add on scene
        if self.line and not self.line.scene():
            self.line = None

        if not self.line:
            startPoint = endPoint = self.parentItem().mapToScene(self.pos())
            self.tempLine = Line(startPoint, endPoint)
            self.scene().addItem(self.tempLine)
        super().mousePressEvent(mouseEvent)

    def mouseMoveEvent(self, mouseEvent):
        """Handle all mouse move for this item
        """
        # if line get started then update it's end point
        if self.tempLine:
            endPoint = mouseEvent.scenePos()
            self.tempLine.updateLine(endPoint=endPoint)

        item = self.scene().itemAt(mouseEvent.scenePos().x(), mouseEvent.scenePos().y(),
                                   self.parentItem().transform())

        if self.previousHoveredItem and item != self.previousHoveredItem and \
                item not in self.previousHoveredItem.lineGripItems:
            self.previousHoveredItem.hideGripItem()
        super().mouseMoveEvent(mouseEvent)

        if type(item) == NodeItem:
            self.previousHoveredItem = item
            item.showGripItem()

    def mouseReleaseEvent(self, mouseEvent):
        """Handle all mouse release for this item"""
        super().mouseReleaseEvent(mouseEvent)
        # set final position of line
        if self.tempLine:
            items = self.scene().items(QPointF(mouseEvent.scenePos().x(), mouseEvent.scenePos().y()))
            for item in items:
                if type(item) == LineGripItem and item != self:
                    if item.line and not item.line.scene():
                        item.line = None
                    if item.line:
                        break
                    self.tempLine.setStartGripItem(self)
                    self.tempLine.setEndGripItem(item)
                    endPoint = item.parentItem().mapToScene(item.pos())
                    self.tempLine.updateLine(endPoint=endPoint)
                    self.line = self.tempLine
                    item.line = self.tempLine
                    break
                elif type(item) == Line and item != self.tempLine:
                    self.tempLine.setStartGripItem(self)
                    endPoint = mouseEvent.scenePos()
                    self.tempLine.refLine = item
                    self.tempLine.refIndex = findIndex(item, endPoint)
                    self.tempLine.updateLine(endPoint=endPoint)
                    item.midLines.append(self.tempLine)
                    self.line = self.tempLine
                    break
            self.scene().removeItem(self.tempLine)
            if self.line:
                self.scene().addItemPlus(self.line)

        self.tempLine = None
        self.previousHoveredItem = None

    def show(self):
        """ shows line grip item
        """
        self.setPen(QPen(QColor("black"), 2))
        self.setBrush(QColor("red"))

    def hide(self):
        """ hides line grip item
        """
        if (self.parentItem().isSelected() or self.isSelected()) is False:
            self.setPen(QPen(Qt.transparent))
            self.setBrush(Qt.transparent)


class NodeItem(QGraphicsSvgItem):
    """
        Extends PyQt5's QGraphicsSvgItem to create the basic structure of shapes with given unit operation type
    """
    # set a common renderer for all svg
    renderer = QSvgRenderer(fileImporter(f'svg/ellipse.svg'))

    def __init__(self, unitOperationType, parent=None):
        QGraphicsSvgItem.__init__(self, parent)
        self.m_type = unitOperationType
        self.id = None
        # self.m_renderer = QSvgRenderer("svg/" + unitOperationType + ".svg")
        # self.m_renderer = QSvgRenderer(fileImporter(f'svg/{unitOperationType}.svg'))
        self.m_renderer = QSvgRenderer(fileImporter(f'svg/ellipse.svg'))
        # self.m_renderer = QSvgRenderer(resourceManager.get_resource(f'toolbar/{unitOperationType}.svg'))
        self.setSharedRenderer(self.m_renderer)
        # set initial size of item
        self.width = 100
        self.height = 100
        self.rect = QRectF(-self.width / 2, -self.height / 2, self.width, self.height)
        # set graphical settings for this item
        self.setFlags(QGraphicsSvgItem.ItemIsMovable |
                      QGraphicsSvgItem.ItemIsSelectable |
                      QGraphicsSvgItem.ItemSendsGeometryChanges)
        self.setAcceptHoverEvents(True)
        self.setZValue(2)
        # grip items connected to this item
        self.lineGripItems = []
        self.sizeGripItems = []
        self.label = None

    def boundingRect(self):
        """Overrides QGraphicsSvgItem's boundingRect() virtual public function and
        returns a valid bounding
        """
        return self.rect

    def paint(self, painter, option, widget):
        """
            Paints the contents of an item in local coordinates.
            :param painter: QPainter instance
            :param option: QStyleOptionGraphicsItem instance
            :param widget: QWidget instance
        """
        if not self.m_renderer:
            QGraphicsSvgItem.paint(self, painter, option, widget)
        elif self.id:
            self.m_renderer.render(painter,self.id, self.boundingRect())
        else:
            self.m_renderer.render(painter, self.boundingRect())
        if self.isSelected():
            self.showGripItem()

    def resize(self, index, movement):
        """Move grip item with changing rect of node item
        """
        self.prepareGeometryChange()
        if index in [0, 1]:
            self.width -= movement.x()
            self.height -= movement.y()
        else:
            self.width += movement.x()
            self.height += movement.y()

        self.rect = QRectF(-self.width / 2, -self.height / 2, self.width, self.height)
        transform = QTransform()
        transform.translate(movement.x() / 2, movement.y() / 2)
        self.setTransform(transform, True)
        self.updateSizeGripItem([index])

    def addGripItem(self):
        """adds grip items
        """
        if self.scene():
            # add grip items for connecting lines
            for i, (location) in enumerate(
                    (
                            "top",
                            "left",
                            "bottom",
                            "right"
                    )
            ):
                item = LineGripItem(self, i, location, parent=self)
                self.lineGripItems.append(item)
            # add grip for resize it
            for i, (direction) in enumerate(
                    (
                            Qt.Vertical,
                            Qt.Horizontal,
                            Qt.Vertical,
                            Qt.Horizontal,
                    )
            ):
                item = SizeGripItem(self, i, direction, parent=self)
                self.sizeGripItems.append(item)

    def updateLineGripItem(self, index_no_updates=None):
        """
        updates line grip items
        """
        # index_no_updates = index_no_updates or []
        for item in self.lineGripItems:
            item.updatePosition()

    def updateSizeGripItem(self, index_no_updates=None):
        """
        updates size grip items
        """
        index_no_updates = index_no_updates or []
        for i, item in enumerate(self.sizeGripItems):
            if i not in index_no_updates:
                item.updatePosition()

    def itemChange(self, change, value):
        """Overloads and extends QGraphicsSvgItem to also update grip items
        """
        if change == QGraphicsItem.ItemSelectedHasChanged:
            if value is True:
                self.showGripItem()
            else:
                self.hideGripItem()
            return
        if change == QGraphicsItem.ItemTransformHasChanged:
            self.updateLineGripItem()
            return
        if change == QGraphicsItem.ItemPositionHasChanged:
            self.updateLineGripItem()
            self.updateSizeGripItem()
            return
        if change == QGraphicsItem.ItemSceneHasChanged:
            self.addGripItem()
            self.updateLineGripItem()
            self.updateSizeGripItem()
            return
        return super(NodeItem, self).itemChange(change, value)

    def removeFromCanvas(self):
        """This function is used to remove item from canvas
        :return:
        """
        for item in self.lineGripItems:
            item.removeConnectedLines()
        for item in self.sizeGripItems:
            item.removeFromCanvas()
        self.scene().removeItem(self)

    def hoverEnterEvent(self, event):
        """defines shape highlighting on Mouse Over
        """
        self.showGripItem()
        super(NodeItem, self).hoverEnterEvent(event)

    def hoverLeaveEvent(self, event):
        """defines shape highlighting on Mouse Leave
        """
        self.hideGripItem()
        super(NodeItem, self).hoverLeaveEvent(event)

    def showGripItem(self):
        """shows grip items of svg item
        """
        for item in self.lineGripItems:
            item.setPen(QPen(QColor("black"), 2))
            item.setBrush(QColor("red"))
        for item in self.sizeGripItems:
            item.setPen(QPen(QColor("black"), 2))

    def hideGripItem(self):
        """hide grip items of svg item
        """
        for item in self.lineGripItems:
            if item.isSelected() is False:
                item.setPen(QPen(Qt.transparent))
                item.setBrush(Qt.transparent)
        for item in self.sizeGripItems:
            item.setPen(QPen(Qt.transparent))
            item.setBrush(Qt.transparent)

    def contextMenuEvent(self, event):
        """Pop up menu
        :return:
        """
        contextMenu = QMenu()
        addLableAction = contextMenu.addAction("add Label")
        # addLableAction.triggered.connect(self.addLabel)
        action = contextMenu.exec_(event.screenPos())
        if action == addLableAction:
            self.label = ItemLabel(event.scenePos(), self)

# classes of pfd-symbols
class AirBlownCooler(NodeItem):
    def __init__(self):
        super(AirBlownCooler, self).__init__("AirBlownCooler", parent=None)


class Bag(NodeItem):
    def __init__(self):
        super(Bag, self).__init__("Bag", parent=None)


class Boiler(NodeItem):
    def __init__(self):
        super(Boiler, self).__init__("Boiler", parent=None)


class Breaker(NodeItem):
    def __init__(self):
        super(Breaker, self).__init__("Breaker", parent=None)


class BriquettingMachine(NodeItem):
    def __init__(self):
        super(BriquettingMachine, self).__init__("BriquettingMachine", parent=None)


class Centrifugal(NodeItem):
    def __init__(self):
        super(Centrifugal, self).__init__("Centrifugal", parent=None)


class CentrifugalCompressor(NodeItem):
    def __init__(self):
        super(CentrifugalCompressor, self).__init__("CentrifugalCompressor", parent=None)


class Centrifugalpump(NodeItem):
    def __init__(self):
        super(Centrifugalpump, self).__init__("Centrifugalpump", parent=None)


class CentrifugalPump2(NodeItem):
    def __init__(self):
        super(CentrifugalPump2, self).__init__("CentrifugalPump2", parent=None)


class CentrifugalPump3(NodeItem):
    def __init__(self):
        super(CentrifugalPump3, self).__init__("CentrifugalPump3", parent=None)


class Column(NodeItem):
    def __init__(self):
        super(Column, self).__init__("Column", parent=None)


class Compressor(NodeItem):
    def __init__(self):
        super(Compressor, self).__init__("Compressor", parent=None)


class CompressorSilencers(NodeItem):
    def __init__(self):
        super(CompressorSilencers, self).__init__("CompressorSilencers", parent=None)


class Condenser(NodeItem):
    def __init__(self):
        super(Condenser, self).__init__("Condenser", parent=None)


class Cooler(NodeItem):
    def __init__(self):
        super(Cooler, self).__init__("Cooler", parent=None)


class CoolingTower3(NodeItem):
    def __init__(self):
        super(CoolingTower3, self).__init__("CoolingTower3", parent=None)


class CoolingTwoer2(NodeItem):
    def __init__(self):
        super(CoolingTwoer2, self).__init__("CoolingTwoer2", parent=None)


class Crusher(NodeItem):
    def __init__(self):
        super(Crusher, self).__init__("Crusher", parent=None)


class DoublePipeHeat(NodeItem):
    def __init__(self):
        super(DoublePipeHeat, self).__init__("DoublePipeHeat", parent=None)


class ExtractorHood(NodeItem):
    def __init__(self):
        super(ExtractorHood, self).__init__("ExtractorHood", parent=None)


class FiredHeater(NodeItem):
    def __init__(self):
        super(FiredHeater, self).__init__("FiredHeater", parent=None)


class ForcedDraftCooling(NodeItem):
    def __init__(self):
        super(ForcedDraftCooling, self).__init__("ForcedDraftCooling", parent=None)


class Furnace(NodeItem):
    def __init__(self):
        super(Furnace, self).__init__("Furnace", parent=None)


class GasBottle(NodeItem):
    def __init__(self):
        super(GasBottle, self).__init__("GasBottle", parent=None)


class HalfPipeMixingVessel(NodeItem):
    def __init__(self):
        super(HalfPipeMixingVessel, self).__init__("HalfPipeMixingVessel", parent=None)


class Heater(NodeItem):
    def __init__(self):
        super(Heater, self).__init__("Heater", parent=None)


class HeatExchanger(NodeItem):
    def __init__(self):
        super(HeatExchanger, self).__init__("HeatExchanger", parent=None)


class HeatExchanger2(NodeItem):
    def __init__(self):
        super(HeatExchanger2, self).__init__("HeatExchanger2", parent=None)


class HorizontalVessel(NodeItem):
    def __init__(self):
        super(HorizontalVessel, self).__init__("HorizontalVessel", parent=None)


class InducedDraftCooling(NodeItem):
    def __init__(self):
        super(InducedDraftCooling, self).__init__("InducedDraftCooling", parent=None)


class JacketedMixingVessel(NodeItem):
    def __init__(self):
        super(JacketedMixingVessel, self).__init__("JacketedMixingVessel", parent=None)


class LiquidRingCompressor(NodeItem):
    def __init__(self):
        super(LiquidRingCompressor, self).__init__("LiquidRingCompressor", parent=None)


class Mixing(NodeItem):
    def __init__(self):
        super(Mixing, self).__init__("Mixing", parent=None)


class MixingReactor(NodeItem):
    def __init__(self):
        super(MixingReactor, self).__init__("MixingReactor", parent=None)


class OilBurner(NodeItem):
    def __init__(self):
        super(OilBurner, self).__init__("OilBurner", parent=None)


class OpenTank(NodeItem):
    def __init__(self):
        super(OpenTank, self).__init__("OpenTank", parent=None)


class ProportionalPump(NodeItem):
    def __init__(self):
        super(ProportionalPump, self).__init__("ProportionalPump", parent=None)


class Pump(NodeItem):
    def __init__(self):
        super(Pump, self).__init__("Pump", parent=None)


class Pump2(NodeItem):
    def __init__(self):
        super(Pump2, self).__init__("Pump2", parent=None)


class ReboilerHeatExchange(NodeItem):
    def __init__(self):
        super(ReboilerHeatExchange, self).__init__("ReboilerHeatExchange", parent=None)


class ReciprocatingCompressor(NodeItem):
    def __init__(self):
        super(ReciprocatingCompressor, self).__init__("ReciprocatingCompressor", parent=None)


class RotaryCompresor(NodeItem):
    def __init__(self):
        super(RotaryCompresor, self).__init__("RotaryCompresor", parent=None)


class RotaryGearPump(NodeItem):
    def __init__(self):
        super(RotaryGearPump, self).__init__("RotaryGearPump", parent=None)


class ScrewPump(NodeItem):
    def __init__(self):
        super(ScrewPump, self).__init__("ScrewPump", parent=None)


class SelectableCompressor(NodeItem):
    def __init__(self):
        super(SelectableCompressor, self).__init__("SelectableCompressor", parent=None)


class SelectableFan(NodeItem):
    def __init__(self):
        super(SelectableFan, self).__init__("SelectableFan", parent=None)


class SinglePassHeat(NodeItem):
    def __init__(self):
        super(SinglePassHeat, self).__init__("SinglePassHeat", parent=None)


class SpiralHeatExchanger(NodeItem):
    def __init__(self):
        super(SpiralHeatExchanger, self).__init__("SpiralHeatExchanger", parent=None)


class StraightTubersHeat(NodeItem):
    def __init__(self):
        super(StraightTubersHeat, self).__init__("StraightTubersHeat", parent=None)


class Tank(NodeItem):
    def __init__(self):
        super(Tank, self).__init__("Tank", parent=None)


class TurbinePump(NodeItem):
    def __init__(self):
        super(TurbinePump, self).__init__("TurbinePump", parent=None)


class UTubeHeatExchanger(NodeItem):
    def __init__(self):
        super(UTubeHeatExchanger, self).__init__("UTubeHeatExchanger", parent=None)


class VaccumPump(NodeItem):
    def __init__(self):
        super(VaccumPump, self).__init__("VaccumPump", parent=None)


class VerticalPump(NodeItem):
    def __init__(self):
        super(VerticalPump, self).__init__("VerticalPump", parent=None)


class VerticalVessel(NodeItem):
    def __init__(self):
        super(VerticalVessel, self).__init__("VerticalVessel", parent=None)


class WastewaterTreatment(NodeItem):
    def __init__(self):
        super(WastewaterTreatment, self).__init__("WastewaterTreatment", parent=None)