PyBlocks/node_graphics_node.py at main · YashwanthProfile/PyBlocks · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
from PyQt5.QtWidgets import QGraphicsItem, QGraphicsTextItem, QGraphicsProxyWidget
from PyQt5.QtCore import QRectF, Qt
from PyQt5.QtGui import QFont, QPainter, QPainterPath, QPen, QColor, QBrush


class customgraphics_node(QGraphicsItem):
    def __init__(self, node, parent=None):
        super().__init__(parent)
        self.node           = node
        self.content        = self.node.content


        self._title_color   = Qt.white
        self._title_font    = QFont("Airal", 10)

        self.width          = 180
        self.height         = 240
        self.edge_size      = 10 # 10px
        self.title_height   = 24 # 24 px
        self._padding       = 4.0


        self._pen_default   = QPen(QColor("#7F000000"))
        self._pen_selected  = QPen(QColor("#FFFFA637"))

        self._brush_title       = QBrush(QColor("#FF313131"))
        self._brush_background  = QBrush(QColor("#E3212121"))

        # init title
        self.initTitle()
        self.title = self.node.title

        # init sockets
        self.initSockets()

        # init content
        self.initContent()

        self.initUI()

    @property
    def title(self): return self._title

    @title.setter
    def title(self, value):
        self._title = value
        self.title_item.setPlainText(self._title)


    def boundingRect(self):
        return QRectF(
            0,
            0,
            self.width,
            self.height
        ).normalized()


    def initUI(self):
        self.setFlag(QGraphicsItem.ItemIsSelectable)
        self.setFlag(QGraphicsItem.ItemIsMovable)

    def initTitle(self):
        self.title_item = QGraphicsTextItem(self)
        self.title_item.setDefaultTextColor(self._title_color)
        self.title_item.setFont(self._title_font)
        self.title_item.setPos(self._padding, 0)
        self.title_item.setTextWidth(
            self.width
            - 2 * self._padding
        )


    def initContent(self):
        self.graphicContent = QGraphicsProxyWidget(self)
        self.content.setGeometry(self.edge_size, self.title_height+self.edge_size,
                                 self.width-2*self.edge_size, self.height-2*self.edge_size-self.title_height)

        self.graphicContent.setWidget(self.content)


    def initSockets(self):
        pass


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

        # Paint title
        path_title = QPainterPath()
        path_title.setFillRule(Qt.WindingFill)
        path_title.addRoundedRect(0,0,self.width, self.title_height, self.edge_size, self.edge_size)
        path_title.addRect(0, self.title_height-self.edge_size, self.edge_size, self.edge_size)
        path_title.addRect(self.width-self.edge_size, self.title_height-self.edge_size, self.edge_size, self.edge_size)

        painter.setPen(Qt.NoPen)
        painter.setBrush(self._brush_title)
        painter.drawPath(path_title.simplified())


        # Paint content
        path_content = QPainterPath()
        path_content.setFillRule(Qt.WindingFill)
        path_content.addRoundedRect(0, self.title_height, self.width, self.height-self.title_height, self.edge_size, self.edge_size)
        path_content.addRect(0, self.title_height, self.edge_size, self.edge_size)
        path_content.addRect(self.width-self.edge_size, self.title_height, self.edge_size, self.edge_size)

        painter.setPen(Qt.NoPen)
        painter.setBrush(self._brush_background)
        painter.drawPath(path_content.simplified())

        # Paint outline
        path_outline = QPainterPath()
        path_outline.addRoundedRect(0, 0, self.width, self.height, self.edge_size, self.edge_size)
        painter.setPen(self._pen_default if not self.isSelected() else self._pen_selected)
        painter.setBrush(Qt.NoBrush)
        painter.drawPath(path_outline.simplified())