class Escena(QWidget):
def __init__(self):
QWidget.__init__(self)
self.resize(QDesktopWidget().availableGeometry(self).size() * 0.6)
self.scene = QGraphicsScene()
self.scene.setSceneRect(-3000, -4000, 6000, 8000)
self.view = QGraphicsView(self.scene)
self.view.resize(self.size())
self.view.fitInView(self.scene.sceneRect(), Qt.KeepAspectRatio)
self.view.setParent(self)
self.view.setTransformationAnchor(QGraphicsView.NoAnchor)
self.view.setResizeAnchor(QGraphicsView.NoAnchor)
self.view.scale(3, -3)
self.lines = []
self.show()
def wheelEvent(self, event):
zoomInFactor = 1.15
zoomOutFactor = 1 / zoomInFactor
# Zoom
if event.delta() > 0:
zoomFactor = zoomInFactor
else:
zoomFactor = zoomOutFactor
self.view.scale(zoomFactor, zoomFactor)
def resizeEvent(self, event):
self.view.resize(self.size())
def draw(self, comps, cluster):
colors = QColor.colorNames()
# for co in sample:
# print(co)
# x, _, z, _ = co['world']
# self.scene.addEllipse(x-50,z-50,100,100, pen=QPen(QColor('orange'), 5))
# x,_,z,_ = sample[-1]['world']
# self.scene.addRect(x,z,60,60, pen=QPen(QColor('red'), 100))
# x,_,z,_ = sample[0]['world']
# self.scene.addRect(x,z,60,60, pen=QPen(QColor('green'), 100))
# for co in comps:
# if sample[co[-1]]['timestamp']-sample[co[0]]['timestamp'] > 200 and len(co)> 4:
# color = colors[random.randint(0, len(colors)-1)]
# for c in co:
# x, _, z, _ = sample[c]['world']
# self.scene.addEllipse(x-15,z-15,30,30, pen=QPen(QColor(color), 100), brush=QBrush(color=QColor(color)))
if cluster[-1]['timestamp'] - cluster[0]['timestamp'] > 200 and len(
cluster) > 4:
color = colors[random.randint(0, len(colors) - 1)]
for sample in cluster:
x, _, z, _ = sample['world']
self.scene.addEllipse(x - 15,
z - 15,
30,
30,
pen=QPen(QColor(color), 100),
brush=QBrush(color=QColor(color)))
def drawTrack(self, clusters):
colors = QColor.colorNames()
# for line in self.lines:
# self.scene.removeItem(line)
self.scene.clear()
for cluster in clusters:
color = colors[random.randint(0, len(colors) - 1)]
for t in cluster:
self.scene.addLine(t[0][0],
t[0][1],
t[1][0],
t[1][1],
pen=QPen(QColor(color), 60))
