HYTOS

#!/usr/bin/env/python3

# coding: utf-8

import sys

import os

import math

from PyQt5.QtGui import *

from PyQt5.QtCore import *

from PyQt5.QtSvg import *

from PyQt5.QtWidgets import (QApplication, QGraphicsItem)

class SymbolSvgItem(QGraphicsSvgItem):

    clicked = pyqtSignal(QGraphicsSvgItem)

    removed = pyqtSignal(QGraphicsSvgItem)

    '''

        @history    18.04.11    Jeongwoo    Add Variable (Name, Type)

    '''

    def __init__(self, path):

        import uuid

        super(SymbolSvgItem, self).__init__(path)

        self.setFlags(QGraphicsItem.ItemIsSelectable|QGraphicsItem.ItemIsFocusable)

                      #QGraphicsItem.ItemIsMovable)

        self.uid = uuid.uuid4() # generate UUID

        self.name = ''

        self.type = ''

        self.angle = 0

        self.loc = None

        self.size = None

        self.conns = []

        self.setAcceptHoverEvents(True)

        self.setAcceptedMouseButtons(Qt.LeftButton)

        self.setAcceptTouchEvents(True)

    '''

        @brief  return bounding box of symbol

        @author humkyung

        @date   2018.04.08

    '''

    def rect(self):

        return self.sceneBoundingRect()

    '''

        @brief  return true if line is able to connect symbol

        @author humkyung

        @date   2018.04.13

    '''

    def isConnectable(self, line, offset,  toler=10):

        for pt in self.connPts:

            dx = pt[0] - (line[0][0] + offset[0])

            dy = pt[1] - (line[0][1] + offset[1])

            if (math.sqrt(dx*dx + dy*dy) < toler): return True

            dx = pt[0] - (line[-1][0] + offset[0])

            dy = pt[1] - (line[-1][1] + offset[1])

            if (math.sqrt(dx*dx + dy*dy) < toler): return True

        return False

    '''

        @brief  connect line and symbol is able to be connected and return line

        @author humkyung

        @date   2018.04.16

    '''

    def connectIfPossible(self, line, toler=10):

        from shapely.geometry import Point

        res = []

        startPt = line.startPoint()

        endPt = line.endPoint()

        for i in range(len(self.connPts)):

            if len(self.conns) <= i: self.conns.append(None)

            if (self.conns[i] is None) and (Point(startPt[0], startPt[1]).distance(Point(self.connPts[i][0], self.connPts[i][1])) < toler):

                self.conns[i] = line

                res.append(line)

            if (self.conns[i] is None) and (Point(endPt[0], endPt[1]).distance(Point(self.connPts[i][0], self.connPts[i][1])) < toler):

                self.conns[i] = line

                res.append(line)

        return res

    '''

        @brief  return center of symbol

        @author humkyung

        @date   2018.04.08

    '''

    def center(self):

        return self.sceneBoundingRect().center()

    def hoverEnterEvent(self, event):

        cursor = QCursor( Qt.OpenHandCursor )

        QApplication.instance().setOverrideCursor( cursor )

        self.update()

    def hoverLeaveEvent(self, event):

        QApplication.instance().restoreOverrideCursor()

        self.update()

    def hoverMoveEvent(self, event):

        pass

    '''

        @brief      Mouse Press Event

        @author     Jeongwoo

        @date       18.04.11

        @history    .

    '''

    def mousePressEvent(self, event):

        self.clicked.emit(self)

    '''

        @brief  remove item when user press delete key

        @author humkyung

        @date   2018.04.23

    '''

    def keyPressEvent(self, event): 

        if event.key() == Qt.Key_Delete:

            for conn in self.conns:

                conn.removeSymbol(self)

                break

            self.removed.emit(self)

            self.scene().removeItem(self)

    '''

        @brief  generate xml code

        @author humkyung

        @date   2018.04.23

    '''

    def toXml(self):

        from xml.etree.ElementTree import Element, SubElement, dump, ElementTree

        try:

            node = Element('SYMBOL')

            uidNode = Element('UID')

            uidNode.text = str(self.uid)

            nameNode = Element('NAME')

            nameNode.text = self.name

            rect = self.sceneBoundingRect()

            locNode = Element('LOCATION')

            locNode.text = '{},{}'.format(rect.left(), rect.top())

            sizeNode = Element('SIZE')

            sizeNode.text = '{},{}'.format(rect.width(), rect.height())

            node.append(uidNode)

            node.append(nameNode)

            node.append(locNode)

            node.append(sizeNode)

        except Exception as ex:

            print('error occured({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno))

        return node 

    '''

        @brief  override paint(draw connection points)

        @author humkyung

        @date   2018.04.21

    '''

    def paint(self, painter, options=None, widget=None):

        pen = QPen(Qt.red, 10, Qt.SolidLine)

        painter.setPen(pen)

        painter.drawEllipse(self.center(), 5, 5)

        QGraphicsSvgItem.paint(self, painter, options, widget)

        if self.isUnderMouse() == True:

            pen = QPen(Qt.red, 10, Qt.SolidLine)

            painter.setPen(pen)

            painter.drawEllipse(self.center(), 5, 5)