def _roi_changed(self, obj, name, old, new):
if old is not None:
self._graphics.scene().removeItem(self.pixmapitem)
self.pixmapitem = None
self.pixmap = None
if new is not None:
self.pixmapitem = QGraphicsPixmapItem()
self._set_pixmap_from_roi(new)
self._graphics.scene().addItem(self.pixmapitem)
def __init__(self):
super(ArrayGraphicsView, self).__init__()
self.setViewportUpdateMode(self.FullViewportUpdate)
self.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
self.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
self.pixmapItem = QGraphicsPixmapItem()
self.setScene(QGraphicsScene())
self.scene().addItem(self.pixmapItem)
self.setBackgroundBrush(QBrush(Qt.black))
self.setMouseTracking(True)
def addUnit(self, unit):
global mapSize
unit.tile = self
self.units.append(unit)
img = QImage(unit.image).scaledToWidth(self.getImageRect().width())
if unit.owner:
rect = img.rect()
painter = QPainter(img)
painter.setPen(unit.owner.unitColor())
painter.drawEllipse(rect)
hpWidth = 20
greenWidth = unit.hp / float(unit.maxHp) * hpWidth
painter.fillRect(5, 5, greenWidth, 5, Qt.green)
painter.fillRect(5 + greenWidth, 5, hpWidth-greenWidth, 5, Qt.red)
painter.end()
image = QGraphicsPixmapItem(QPixmap(img), self)
if mapSize == 0:
mapSize = 1
image.setOffset(self.x + 12/(2*mapSize), self.y + 10/(2*mapSize))
self.unitImages.append(image)
class ROIDisplayItem(HasTraits):
roi = Instance(ROI)
selected = Bool(False)
slicer = Instance(Slicer)
pixmap = Instance(QPixmap, default=None)
pixmapitem = Instance(QGraphicsPixmapItem, default=None)
def __init__(self, graphics, slicer, **kwargs):
self._graphics = graphics
self.slicer = slicer
super(ROIDisplayItem, self).__init__(**kwargs)
def destroy(self):
self._graphics.scene().removeItem(self.pixmapitem)
@on_trait_change('roi')
def _roi_changed(self, obj, name, old, new):
if old is not None:
self._graphics.scene().removeItem(self.pixmapitem)
self.pixmapitem = None
self.pixmap = None
if new is not None:
self.pixmapitem = QGraphicsPixmapItem()
self._set_pixmap_from_roi(new)
self._graphics.scene().addItem(self.pixmapitem)
@on_trait_change('roi:updated,roi:visible,slicer:slc,selected')
def _roi_updated(self):
self._set_pixmap_from_roi(self.roi)
def _set_pixmap_from_roi(self, roi):
if roi.visible:
color = _display_color(roi.color, self.selected)
else:
color = QColor(Qt.transparent)
self.pixmap = _ndarray_to_arraypixmap(roi.mask[self.slicer.slc.view_slice], color.toTuple())
self.pixmapitem.setPixmap(self.pixmap)
self.pixmapitem.setZValue(_foreground_roi_z if self.selected else _background_roi_z)
def update_frame(self, img):
'''
update the image
:return:
'''
#if not self.queue.empty():
# frame = self.queue.get()
#img = frame['img']
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
size = self.video_view.size()
img = cv2.resize(img, (size.width() - 10, size.height() - 10))
height, width, bpc = img.shape
bpl = bpc * width
image = QImage(img.data, width, height, bpl, QImage.Format_RGB888)
pitem = QGraphicsPixmapItem(QPixmap.fromImage(image))
scene = QGraphicsScene()
scene.addItem(pitem)
self.video_view.setScene(scene)
def addUnit(self, unit):
unit.tile = self
self.units.append(unit)
image = QGraphicsPixmapItem(QPixmap(unit.image), self)
image.setOffset(self.x + 12, self.y + 10)
self.unitImages.append(image)
class ArrayGraphicsView(QGraphicsView):
'''ArrayGraphicsView is used for viewing a numpy array.
TODO:
* When image is loaded it should be zoomed to fit the window
* Maximum zoom should be based on pixel size
'''
mousemoved = Signal(object)
mousewheeled = Signal(object)
mousedoubleclicked = Signal(object)
mousepressed = Signal(object)
mousereleased = Signal(object)
mouse_entered = Signal(object)
mouse_left = Signal(object)
key_pressed = Signal(object)
def __init__(self):
super(ArrayGraphicsView, self).__init__()
self.setViewportUpdateMode(self.FullViewportUpdate)
self.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
self.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
self.pixmapItem = QGraphicsPixmapItem()
self.setScene(QGraphicsScene())
self.scene().addItem(self.pixmapItem)
self.setBackgroundBrush(QBrush(Qt.black))
self.setMouseTracking(True)
def mouseevent_to_item_coords(self, ev):
sp = self.mapToScene(ev.pos())
p = self.pixmapItem.mapFromScene(sp)
return (p.x(), p.y())
def enterEvent(self, ev):
super(ArrayGraphicsView, self).enterEvent(ev)
self.mouse_entered.emit(ev)
def leaveEvent(self, ev):
super(ArrayGraphicsView, self).leaveEvent(ev)
self.mouse_left.emit(ev)
def mouseReleaseEvent(self, ev):
super(ArrayGraphicsView, self).mouseReleaseEvent(ev)
self.mousereleased.emit(ev)
def mousePressEvent(self, ev):
super(ArrayGraphicsView, self).mousePressEvent(ev)
self.mousepressed.emit(ev)
def mouseMoveEvent(self, ev):
super(ArrayGraphicsView, self).mouseMoveEvent(ev)
self.mousemoved.emit(ev)
def mouseDoubleClickEvent(self, ev):
super(ArrayGraphicsView, self).mouseDoubleClickEvent(ev)
self.mousedoubleclicked.emit(ev)
def wheelEvent(self, ev):
self.mousewheeled.emit(ev)
def keyPressEvent(self, ev):
super(ArrayGraphicsView, self).keyPressEvent(ev)
self.key_pressed.emit(ev)
def setPixmap(self, pixmap):
'''Set the array to be viewed.
Args:
array (numpy array): the array to be viewed
This will remove the previous array but maintain the previous scaling
as well as the panned position.
'''
self.pixmap = pixmap
self.pixmapItem.setPixmap(self.pixmap)
# Constrain scene to be the boundary of the pixmap
pad = 5
r = self.pixmapItem.boundingRect()
r = QRectF(r.left()-pad,r.top()-pad,r.width()+2*pad,r.height()+2*pad)
self.setSceneRect(r)
def fitView(self):
self.fitInView(self.sceneRect(), Qt.KeepAspectRatio)
