def add_human_by_pos(self, id, pos):
x, y = pos
human = QGraphicsEllipseItem(0, 0, 200, 200)
self._scene.addItem(human)
human.setBrush(QBrush(Qt.black, style=Qt.SolidPattern))
human_text = QGraphicsTextItem(str(pos))
font = QFont("Helvetica [Cronyx]", 40, QFont.Bold)
human_text.setFont(font)
human_text.setParentItem(human)
human.setPos(pos[0], pos[1])
self._humans[id] = human
human.setZValue(30)
def createItem(x, scene):
rectItem = QGraphicsRectItem(QRectF(x + 40, 40, 120, 120))
scene.addItem(rectItem)
rectItem.setPen(QPen(Qt.black))
rectItem.setBrush(Qt.gray)
innerRectItem = QGraphicsRectItem(QRectF(x + 50, 50, 45, 100), rectItem)
innerRectItem.setPen(QPen(Qt.black))
innerRectItem.setBrush(Qt.white)
# scene.addItem(innerRectItem)
ellipseItem = QGraphicsEllipseItem(QRectF(x + 105, 50, 45, 100), rectItem)
ellipseItem.setPen(QPen(Qt.black))
ellipseItem.setBrush(Qt.white)
# scene.addItem(ellipseItem)
return rectItem
rectItem = QGraphicsRectItem(QRectF(-25, 25, 200, 40))
rectItem.setPen(QPen(Qt.red, 3, Qt.DashDotLine))
rectItem.setBrush(Qt.gray)
scene.addItem(rectItem)
print(f'Rect Pos: {rectItem.pos()}')
textItem = QGraphicsSimpleTextItem("Foundation of Qt")
scene.addItem(textItem)
print(f'Text Pos: {textItem.pos()}')
textItem.setPos(50, 0)
print(f'Text Pos: {textItem.pos()}')
ellipseItem = QGraphicsEllipseItem(QRectF(170, 20, 100, 75))
ellipseItem.setPen(QPen(Qt.darkBlue))
ellipseItem.setBrush(Qt.blue)
scene.addItem(ellipseItem)
points = [
QPointF(10, 10),
QPointF(0, 90),
QPointF(40, 70),
QPointF(80, 110),
QPointF(70, 20)
]
polygonItem = QGraphicsPolygonItem(QPolygonF(points))
polygonItem.setPen(QPen(Qt.darkGreen))
polygonItem.setBrush(Qt.yellow)
scene.addItem(polygonItem)
class AbstractNode(AbstractGeneralGraphicsItemClass):
__metaclass__ = AbstractGeneralGraphicsItem_Meta
def __init__(self, graphWidget, state):
super(AbstractNode, self).__init__()
self.graph = weakref.ref(graphWidget)
self.isAcceptable = state.isAcceptable
self.isCurrent = state.isCurrent
self.newPos = QPointF()
self.setFlag(QGraphicsItem.ItemIsMovable)
self.setFlag(QGraphicsItem.ItemIsSelectable)
self.setFlag(QGraphicsItem.ItemSendsGeometryChanges)
self.setCacheMode(self.DeviceCoordinateCache)
self.setZValue(-1)
self.setAcceptHoverEvents(True)
self._edges = []
self.currentFill = self.fillQColor
self.border = QGraphicsEllipseItem(-self.radius, -self.radius,
self.radius * 2, self.radius * 2)
self.finalNodeBorder = QGraphicsEllipseItem(-self.radius + 5,
-self.radius + 5,
self.radius * 2 - 10,
self.radius * 2 - 10)
self.finalNodeBorder.mapToParent(self.border.pos())
if self.isAcceptable == False:
pen = self.finalNodeBorder.pen()
pen.setColor(QColor(25, 255, 5))
self.finalNodeBorder.update()
if self.isCurrent == True:
self.resolveFillColor()
self.labelBox = QGraphicsTextItem(self)
self.labelBox.setPos(-self.radius / 2, -self.radius / 2)
self.labelBox.setDefaultTextColor(QColor(0, 0, 0))
@property
@abstractmethod
def borderQColor(self):
pass
@property
@abstractmethod
def fillQColor(self):
pass
@property
@abstractmethod
def currentStateFillQColor(self):
pass
@property
@abstractmethod
def radius(self):
pass
@abstractmethod
def setLabel(self, value):
pass
def toggleCurrentState(self):
self.isCurrent = not self.isCurrent
self.resolveFillColor()
def resolveFillColor(self):
brush = QBrush()
if self.isCurrent == True:
brush.setColor(self.currentStateFillQColor)
brush.setStyle(Qt.SolidPattern)
self.currentFill = self.currentStateFillQColor
else:
brush.setColor(self.fillQColor)
brush.setStyle(Qt.SolidPattern)
self.currentFill = self.fillQColor
self.border.setBrush(brush)
self.update()
def toggleFinalNodeBorder(self):
self.isAcceptable = not self.isAcceptable
self.resolveFinalNodeBorder()
def resolveFinalNodeBorder(self):
pen = QPen()
if self.isAcceptable == True:
pen.setColor(self.borderQColor)
pen.setWidth(2)
else:
pen.setColor(self.currentFill)
pen.setWidth(1)
self.finalNodeBorder.setPen(pen)
self.update()
def addEdge(self, edge):
self._edges.append(weakref.ref(edge))
edge.adjust()
def getEdges(self):
return self._edges
def getEdgesForInput(self, input):
edges = []
for edge in self._edges:
if edge.onInput == input:
edges.append(edge)
return edges
def getEdgesFromSelf(self, input):
edges = []
for edge in self._edges:
if edge.onInput == input and edge.fromNode is self:
edges.append(edge)
return edges
def getEdgeToNode(self, toNode):
toReturn = None
for edge in self._edges:
if edge.toNode() is toNode:
toReturn = edge
return toReturn
def boundingRect(self):
adjust = 2.0
return QRectF(-self.radius - adjust, -self.radius - adjust,
(self.radius + adjust) * 2, (self.radius + adjust) * 2)
def shape(self):
path = QtGui.QPainterPath()
path.addEllipse(-self.radius, -self.radius, self.radius * 2,
self.radius * 2)
return path
def paint(self, painter, option, widget):
pen = QPen()
pen.setWidth(2)
pen.setColor(self.borderQColor)
brush = QBrush()
self.border.setPen(pen)
self.border.setBrush(brush)
self.resolveFillColor()
self.resolveFinalNodeBorder()
self.border.paint(painter, option, widget)
self.finalNodeBorder.paint(painter, option, widget)
self.labelBox.setTextWidth(35)
self.labelBox.setDefaultTextColor(QColor(0, 0, 0))
def itemChange(self, change, value):
if change == QGraphicsItem.ItemPositionChange:
for edge in self._edges:
edge().adjust()
return QGraphicsItem.itemChange(self, change, value)
def mousePressEvent(self, event):
self.update()
QGraphicsItem.mousePressEvent(self, event)
def mouseReleaseEvent(self, event):
QGraphicsItem.mouseReleaseEvent(self, event)
self.update()
def eventFilter(self, obj, event):
if event.type() == QEvent.GraphicsSceneMousePress:
if event.button() == Qt.MouseButton.LeftButton:
self.mouse_pressed = True
self.mouse_pressed_x, self.mouse_pressed_y = event.pos().x(
), event.pos().y()
if self.draw_ellipse:
ellipsis = QGraphicsEllipseItem(self.chart)
ellipsis.setZValue(12)
ellipsis.setBrush(QBrush(QColor(244, 67, 54, 50)))
ellipsis.setPen(QPen(Qt.transparent))
self.ellipses.append(ellipsis)
elif self.write_text:
for t in self.texts:
r = QRectF()
r.setTopLeft(t.pos())
r.setWidth(t.boundingRect().width())
r.setHeight(t.boundingRect().height())
if r.contains(self.mouse_pressed_x,
self.mouse_pressed_y):
return True
"""
The user clicked over an area where there is no text. So we create one.
"""
text = QGraphicsTextItem(self.chart)
text.setZValue(12)
text.setPos(
QPointF(self.mouse_pressed_x, self.mouse_pressed_y))
text.setPlainText("label")
text.setAcceptHoverEvents(True)
text.setTabChangesFocus(True)
text.setFlags(QGraphicsTextItem.ItemIsMovable)
text.installEventFilter(self.text_event_filter)
self.texts.append(text)
return True
elif event.button() == Qt.MouseButton.RightButton:
x, y = event.pos().x(), event.pos().y()
for e in self.ellipses:
if e.rect().contains(x, y):
e.hide()
self.ellipses.remove(e)
for t in self.texts:
r = QRectF()
r.setTopLeft(t.pos())
r.setWidth(t.boundingRect().width())
r.setHeight(t.boundingRect().height())
if r.contains(x, y):
t.hide()
self.texts.remove(t)
return True
return QObject.eventFilter(self, obj, event)
elif event.type() == QEvent.GraphicsSceneMouseRelease:
self.mouse_pressed = False
return True
elif event.type() == QEvent.GraphicsSceneMouseMove:
if self.mouse_pressed:
if self.draw_ellipse:
x, y = event.pos().x(), event.pos().y()
width = x - self.mouse_pressed_x
height = y - self.mouse_pressed_y
self.ellipses[-1].setRect(self.mouse_pressed_x,
self.mouse_pressed_y, width,
height)
return True
return QObject.eventFilter(self, obj, event)
return QObject.eventFilter(self, obj, event)