hytos / DTI_PID / DTI_PID / Shapes / GraphicsBoundingBoxItem.py @ 2ffe1307
이력 | 보기 | 이력해설 | 다운로드 (9.05 KB)
| 1 | 40b4bfe5 | humkyung | import os.path |
|---|---|---|---|
| 2 | import copy |
||
| 3 | try:
|
||
| 4 | da8d9523 | humkyung | from PyQt5.QtCore import * |
| 5 | from PyQt5.QtGui import * |
||
| 6 | from PyQt5.QtWidgets import * |
||
| 7 | 40b4bfe5 | humkyung | except ImportError: |
| 8 | try:
|
||
| 9 | from PyQt4.QtCore import Qt, QRectF, pyqtSignal, QT_VERSION_STR |
||
| 10 | from PyQt4.QtGui import QGraphicsView, QGraphicsScene, QImage, QPixmap, QPainterPath, QFileDialog |
||
| 11 | except ImportError: |
||
| 12 | raise ImportError("ImageViewerQt: Requires PyQt5 or PyQt4.") |
||
| 13 | |||
| 14 | 3d6ff047 | humkyung | from EngineeringAbstractItem import QEngineeringAbstractItem |
| 15 | |||
| 16 | class QGraphicsBoundingBoxItem(QGraphicsRectItem, QEngineeringAbstractItem): |
||
| 17 | fd688617 | humkyung | onRemoved = pyqtSignal(QGraphicsItem) |
| 18 | |||
| 19 | 40b4bfe5 | humkyung | def __init__(self, x, y, width, height): |
| 20 | da8d9523 | humkyung | from EngineeringConnectorItem import QEngineeringConnectorItem |
| 21 | |||
| 22 | 40b4bfe5 | humkyung | QGraphicsRectItem.__init__(self, x, y, width, height)
|
| 23 | 3d6ff047 | humkyung | QEngineeringAbstractItem.__init__(self)
|
| 24 | |||
| 25 | 40b4bfe5 | humkyung | self.setAcceptHoverEvents(True) |
| 26 | 8fd1d96f | humkyung | self.setAcceptTouchEvents(True) |
| 27 | 40b4bfe5 | humkyung | self.isEntered = False |
| 28 | cd57bd35 | humkyung | self.center = self.rect().center() |
| 29 | aa48afcc | humkyung | self.angle = 0 |
| 30 | dcdd75a6 | humkyung | self.isSymbol = False |
| 31 | 4c612af8 | humkyung | self.isCreated = False |
| 32 | self._vertices = []
|
||
| 33 | da8d9523 | humkyung | |
| 34 | ## create connectors for corner
|
||
| 35 | connector = QEngineeringConnectorItem(self)
|
||
| 36 | connector.setPos((x, y)) |
||
| 37 | connector.setParentItem(self)
|
||
| 38 | connector.setZValue(self.zValue() + 1) |
||
| 39 | 992b5028 | gaqhf | connector.setFlags(QGraphicsItem.ItemIsSelectable|QGraphicsItem.ItemIsFocusable) |
| 40 | connector.setAcceptTouchEvents(True)
|
||
| 41 | connector.transfer.onPosChanged.connect(self.onConnectorPosChaned)
|
||
| 42 | da8d9523 | humkyung | self.connectors.append(connector)
|
| 43 | |||
| 44 | connector = QEngineeringConnectorItem(self)
|
||
| 45 | connector.setPos((x + width, y)) |
||
| 46 | connector.setParentItem(self)
|
||
| 47 | connector.setZValue(self.zValue() + 1) |
||
| 48 | 992b5028 | gaqhf | connector.setFlags(QGraphicsItem.ItemIsSelectable|QGraphicsItem.ItemIsFocusable) |
| 49 | connector.setAcceptTouchEvents(True)
|
||
| 50 | connector.transfer.onPosChanged.connect(self.onConnectorPosChaned)
|
||
| 51 | da8d9523 | humkyung | self.connectors.append(connector)
|
| 52 | |||
| 53 | connector = QEngineeringConnectorItem(self)
|
||
| 54 | connector.setPos((x + width, y + height)) |
||
| 55 | connector.setParentItem(self)
|
||
| 56 | connector.setZValue(self.zValue() + 1) |
||
| 57 | 992b5028 | gaqhf | connector.setFlags(QGraphicsItem.ItemIsSelectable|QGraphicsItem.ItemIsFocusable) |
| 58 | connector.setAcceptTouchEvents(True)
|
||
| 59 | connector.transfer.onPosChanged.connect(self.onConnectorPosChaned)
|
||
| 60 | da8d9523 | humkyung | self.connectors.append(connector)
|
| 61 | |||
| 62 | connector = QEngineeringConnectorItem(self)
|
||
| 63 | connector.setPos((x, y + height)) |
||
| 64 | connector.setParentItem(self)
|
||
| 65 | connector.setZValue(self.zValue() + 1) |
||
| 66 | 992b5028 | gaqhf | connector.setFlags(QGraphicsItem.ItemIsSelectable|QGraphicsItem.ItemIsFocusable) |
| 67 | connector.setAcceptTouchEvents(True)
|
||
| 68 | connector.transfer.onPosChanged.connect(self.onConnectorPosChaned)
|
||
| 69 | da8d9523 | humkyung | self.connectors.append(connector)
|
| 70 | ## up to here
|
||
| 71 | |||
| 72 | self.transfer = Transfer()
|
||
| 73 | |||
| 74 | '''
|
||
| 75 | @brief reshape rectangle
|
||
| 76 | @author humkyung
|
||
| 77 | @date 2018.08.29
|
||
| 78 | '''
|
||
| 79 | def onConnectorPosChaned(self, connector): |
||
| 80 | found = -1
|
||
| 81 | for i in range(len(self.connectors)): |
||
| 82 | if self.connectors[i] is connector: |
||
| 83 | found = i |
||
| 84 | break
|
||
| 85 | |||
| 86 | if found != -1: |
||
| 87 | other = (found + 2) % 4 |
||
| 88 | minx = min(self.connectors[found].center()[0], self.connectors[other].center()[0]) |
||
| 89 | miny = min(self.connectors[found].center()[1], self.connectors[other].center()[1]) |
||
| 90 | maxx = max(self.connectors[found].center()[0], self.connectors[other].center()[0]) |
||
| 91 | maxy = max(self.connectors[found].center()[1], self.connectors[other].center()[1]) |
||
| 92 | self.setRect(minx, miny, maxx - minx, maxy - miny)
|
||
| 93 | |||
| 94 | self.connectors[found].setPos((minx, miny))
|
||
| 95 | index = (found + 1) % 4 |
||
| 96 | self.connectors[index].setPos((minx, maxy))
|
||
| 97 | self.connectors[other].setPos((maxx, maxy))
|
||
| 98 | index = (found + 3) % 4 |
||
| 99 | self.connectors[index].setPos((maxx, miny))
|
||
| 100 | |||
| 101 | self.transfer.onSizeChanged.emit(self) |
||
| 102 | |||
| 103 | '''
|
||
| 104 | @brief set graphicsboundingbox's color
|
||
| 105 | @author humkyung
|
||
| 106 | @date 2018.08.29
|
||
| 107 | '''
|
||
| 108 | def setColor(self, color): |
||
| 109 | self._color = color
|
||
| 110 | c = QColor() |
||
| 111 | c.setNamedColor(color) |
||
| 112 | _pen = self.pen()
|
||
| 113 | _pen.setColor(c) |
||
| 114 | self.setPen(_pen)
|
||
| 115 | self.update()
|
||
| 116 | |||
| 117 | def getColor(self): |
||
| 118 | return self.pen().color().name() |
||
| 119 | 4c612af8 | humkyung | |
| 120 | 2ffe1307 | humkyung | """
|
| 121 | @brief override setRect
|
||
| 122 | @author humkyung
|
||
| 123 | @date 2018.09.12
|
||
| 124 | """
|
||
| 125 | def setRect(self, x, y, width, height): |
||
| 126 | QGraphicsRectItem.setRect(self, x, y, width, height)
|
||
| 127 | |||
| 128 | minx, miny, maxx, maxy = x, y, x + width, y + height |
||
| 129 | self.connectors[0].setPos((minx, miny)) |
||
| 130 | self.connectors[1].setPos((minx, maxy)) |
||
| 131 | self.connectors[2].setPos((maxx, maxy)) |
||
| 132 | self.connectors[3].setPos((maxx, miny)) |
||
| 133 | |||
| 134 | 4c612af8 | humkyung | '''
|
| 135 | @brief construct a bounding box item
|
||
| 136 | '''
|
||
| 137 | def process(self, param): |
||
| 138 | if ('mouseMoveEvent' == param[0]): |
||
| 139 | if len(self._vertices) == 1: |
||
| 140 | self._vertices.append(param[2]) |
||
| 141 | elif len(self._vertices) > 1: |
||
| 142 | self._vertices[1] = param[2] |
||
| 143 | |||
| 144 | if len(self._vertices) == 2: |
||
| 145 | x = min(self._vertices[0].x(), self._vertices[1].x()) |
||
| 146 | y = min(self._vertices[0].y(), self._vertices[1].y()) |
||
| 147 | dx = abs(self._vertices[0].x() - self._vertices[1].x()) |
||
| 148 | dy = abs(self._vertices[0].y() - self._vertices[1].y()) |
||
| 149 | self.setRect(x, y, dx, dy)
|
||
| 150 | elif ('mouseReleaseEvent' == param[0]) and (param[1].button() == Qt.LeftButton): |
||
| 151 | if len(self._vertices) == 0: |
||
| 152 | self._vertices.append(param[2]) |
||
| 153 | elif len(self._vertices) == 1: |
||
| 154 | self._vertices.append(param[2]) |
||
| 155 | elif len(self._vertices) == 2: |
||
| 156 | x = min(self._vertices[0].x(), self._vertices[1].x()) |
||
| 157 | y = min(self._vertices[0].y(), self._vertices[1].y()) |
||
| 158 | dx = abs(self._vertices[0].x() - self._vertices[1].x()) |
||
| 159 | dy = abs(self._vertices[0].y() - self._vertices[1].y()) |
||
| 160 | self.setRect(x, y, dx, dy)
|
||
| 161 | self.isCreated = True |
||
| 162 | 8fd1d96f | humkyung | '''
|
| 163 | @brief reshape while mouse drag
|
||
| 164 | @author humkyung
|
||
| 165 | @date 2018.08.18
|
||
| 166 | '''
|
||
| 167 | def mouseMoveEvent(self, event): |
||
| 168 | if event.button() == Qt.LeftButton:
|
||
| 169 | scenePos = event.scenePos() |
||
| 170 | |||
| 171 | if len(self._vertices) == 1: |
||
| 172 | self._vertices.append(scenePos)
|
||
| 173 | elif len(self._vertices) > 1: |
||
| 174 | self._vertices[1] = scenePos |
||
| 175 | |||
| 176 | if len(self._vertices) == 2: |
||
| 177 | x = min(self._vertices[0].x(), self._vertices[1].x()) |
||
| 178 | y = min(self._vertices[0].y(), self._vertices[1].y()) |
||
| 179 | dx = abs(self._vertices[0].x() - self._vertices[1].x()) |
||
| 180 | dy = abs(self._vertices[0].y() - self._vertices[1].y()) |
||
| 181 | self.setRect(x, y, dx, dy)
|
||
| 182 | |||
| 183 | QGraphicsRectItem.mouseMoveEvent(self, event)
|
||
| 184 | |||
| 185 | '''
|
||
| 186 | @brief
|
||
| 187 | @author humkyung
|
||
| 188 | @date 2018.08.18
|
||
| 189 | '''
|
||
| 190 | def mouseReleaseEvent(self, event): |
||
| 191 | if event.button() == Qt.LeftButton:
|
||
| 192 | scenePos = event.scenePos() |
||
| 193 | |||
| 194 | if len(self._vertices) == 0: |
||
| 195 | self._vertices.append(scenePos)
|
||
| 196 | elif len(self._vertices) == 1: |
||
| 197 | self._vertices.append(scenePos)
|
||
| 198 | elif len(self._vertices) == 2: |
||
| 199 | x = min(self._vertices[0].x(), self._vertices[1].x()) |
||
| 200 | y = min(self._vertices[0].y(), self._vertices[1].y()) |
||
| 201 | dx = abs(self._vertices[0].x() - self._vertices[1].x()) |
||
| 202 | dy = abs(self._vertices[0].y() - self._vertices[1].y()) |
||
| 203 | self.setRect(x, y, dx, dy)
|
||
| 204 | self.isCreated = True |
||
| 205 | |||
| 206 | QGraphicsRectItem.mouseReleaseEvent(self, event)
|
||
| 207 | 40b4bfe5 | humkyung | |
| 208 | '''
|
||
| 209 | @brief clone an object
|
||
| 210 | '''
|
||
| 211 | def clone(self): |
||
| 212 | clone = QGraphicsBoundingBoxItem(self.rect().x(), self.rect().y(), self.rect().width(), self.rect().height()) |
||
| 213 | clone.isCreated = True
|
||
| 214 | cd57bd35 | humkyung | clone.center = self.center
|
| 215 | 40b4bfe5 | humkyung | return clone
|
| 216 | |||
| 217 | 4c612af8 | humkyung | '''
|
| 218 | @brief initialize
|
||
| 219 | '''
|
||
| 220 | def init(self): |
||
| 221 | self._vertices = []
|
||
| 222 | self.isCreated = False |
||
| 223 | |||
| 224 | 40b4bfe5 | humkyung | def hoverEnterEvent(self, event): |
| 225 | self.isEntered = True |
||
| 226 | 3a4cc147 | humkyung | self.update()
|
| 227 | 40b4bfe5 | humkyung | |
| 228 | def hoverLeaveEvent(self, event): |
||
| 229 | self.isEntered = False |
||
| 230 | 3a4cc147 | humkyung | self.update()
|
| 231 | 40b4bfe5 | humkyung | |
| 232 | '''
|
||
| 233 | @brief override paint
|
||
| 234 | '''
|
||
| 235 | def paint(self, painter, options=None, widget=None): |
||
| 236 | QGraphicsRectItem.paint(self, painter, options, widget)
|
||
| 237 | |||
| 238 | if self.isEntered == True: |
||
| 239 | painter.setPen(self.pen())
|
||
| 240 | da8d9523 | humkyung | painter.drawEllipse(self.center, 5, 5) |
| 241 | |||
| 242 | '''
|
||
| 243 | @brief The class transfer pyqtSignal Event
|
||
| 244 | @author humkyung
|
||
| 245 | @date 2018.08.30
|
||
| 246 | '''
|
||
| 247 | class Transfer(QObject): |
||
| 248 | onSizeChanged = pyqtSignal(QGraphicsItem) |
||
| 249 | onRemoved = pyqtSignal(QGraphicsItem) |
||
| 250 | |||
| 251 | def __init__(self, parent = None): |
||
| 252 | QObject.__init__(self, parent) |
