mirror of
https://github.com/wassname/sloth.git
synced 2026-06-28 16:14:24 +08:00
Martin Baeuml
290 lines
10 KiB
Python
290 lines
10 KiB
Python
"""Dies ist das AnnotationScene module"""
|
|
from PyQt4.QtGui import *
|
|
from PyQt4.QtCore import *
|
|
from items import *
|
|
from annotations.model import TypeRole, ImageRole
|
|
import math
|
|
import okapy
|
|
from okapy.guiqt.utilities import toQImage
|
|
|
|
class AnnotationScene(QGraphicsScene):
|
|
"""Dies ist ein Test"""
|
|
|
|
# TODO signal itemadded
|
|
|
|
def __init__(self, items=None, inserters=None, parent=None):
|
|
super(AnnotationScene, self).__init__(parent)
|
|
|
|
self.model_ = None
|
|
self.mode_ = None
|
|
self.inserter_ = None
|
|
self.debug_ = True
|
|
self.message_ = ""
|
|
self.last_key_ = None
|
|
|
|
self.itemfactory_ = Factory(items)
|
|
self.inserterfactory_ = Factory(inserters)
|
|
|
|
self.setBackgroundBrush(Qt.darkGray)
|
|
|
|
self.setMode(None)
|
|
self.reset()
|
|
|
|
#
|
|
# getters/setters
|
|
#______________________________________________________________________________________________________
|
|
def model(self):
|
|
return self.model_
|
|
|
|
def setModel(self, model):
|
|
if model == self.model_:
|
|
# same model as the current one
|
|
# reset caches anyway, invalidate root
|
|
self.reset()
|
|
return
|
|
|
|
# disconnect old signals
|
|
if self.model_ is not None:
|
|
self.disconnect(self.model_, SIGNAL('dataChanged(QModelIndex,QModelIndex)'), self.dataChanged)
|
|
self.disconnect(self.model_, SIGNAL('rowsInserted(QModelIndex,int,int)'), self.rowsInserted)
|
|
self.disconnect(self.model_, SIGNAL('rowsAboutToBeRemoved(QModelIndex,int,int)'), self.rowsAboutToBeRemoved)
|
|
self.disconnect(self.model_, SIGNAL('rowsRemoved(QModelIndex,int,int)'), self.rowsRemoved)
|
|
self.disconnect(self.model_, SIGNAL('modelReset()'), self.reset)
|
|
self.model_ = model
|
|
|
|
# connect new signals
|
|
if self.model_ is not None:
|
|
self.connect(self.model_, SIGNAL('dataChanged(QModelIndex,QModelIndex)'), self.dataChanged)
|
|
self.connect(self.model_, SIGNAL('rowsInserted(QModelIndex,int,int)'), self.rowsInserted)
|
|
self.connect(self.model_, SIGNAL('rowsAboutToBeRemoved(QModelIndex,int,int)'), self.rowsAboutToBeRemoved)
|
|
self.connect(self.model_, SIGNAL('rowsRemoved(QModelIndex,int,int)'), self.rowsRemoved)
|
|
self.connect(self.model_, SIGNAL('modelReset()'), self.reset)
|
|
|
|
# reset caches, invalidate root
|
|
self.reset()
|
|
|
|
def root(self):
|
|
return self.root_
|
|
|
|
def setRoot(self, root):
|
|
"""
|
|
Set the index of the model which denotes the current image to be
|
|
displayed by the scene. This can be either the index to a frame in a
|
|
video, or to an image.
|
|
"""
|
|
self.root_ = root
|
|
self.clear()
|
|
if not root.isValid():
|
|
return
|
|
|
|
assert self.root_.model() == self.model_
|
|
self.image_ = self.root_.data(ImageRole).toPyObject()
|
|
self.pixmap_ = QPixmap(okapy.guiqt.toQImage(self.image_))
|
|
item = QGraphicsPixmapItem(self.pixmap_)
|
|
item.setZValue(-1)
|
|
self.setSceneRect(0, 0, self.pixmap_.width(), self.pixmap_.height())
|
|
self.addItem(item)
|
|
|
|
num_items = self.model_.rowCount(self.root_)
|
|
self.insertItems(0, num_items)
|
|
self.update()
|
|
|
|
def insertItems(self, first, last):
|
|
assert self.model_ is not None
|
|
assert self.root_.isValid()
|
|
|
|
# create a graphics item for each model index
|
|
for row in range(first, last+1):
|
|
child = self.root_.child(row, 0) # get index
|
|
_type = str(child.data(TypeRole).toPyObject()) # get type from index
|
|
item = self.itemfactory_.create(_type, child) # create graphics item from factory
|
|
if item is not None:
|
|
self.addItem(item)
|
|
|
|
def mode(self):
|
|
return self.mode_
|
|
|
|
def setMode(self, mode):
|
|
print "setMode :", mode
|
|
self.mode_ = mode
|
|
|
|
if self.mode_ == None:
|
|
self.inserter_ = None
|
|
return
|
|
|
|
inserter = self.inserterfactory_.create(self.mode_['type'], self)
|
|
if inserter is None:
|
|
raise InvalidArgumentException("Invalid mode")
|
|
|
|
self.inserter_ = inserter
|
|
self.inserter_.setMode(self.mode_)
|
|
|
|
#
|
|
# common methods
|
|
#______________________________________________________________________________________________________
|
|
def reset(self):
|
|
self.clear()
|
|
self.setRoot(QModelIndex())
|
|
self.clearMessage()
|
|
|
|
def addItem(self, item):
|
|
QGraphicsScene.addItem(self, item)
|
|
# TODO emit signal itemAdded
|
|
|
|
#
|
|
# mouse event handlers
|
|
#______________________________________________________________________________________________________
|
|
def mousePressEvent(self, event):
|
|
if self.debug_:
|
|
print "mousePressEvent", self.sceneRect().contains(event.scenePos()), event.scenePos()
|
|
if not self.sceneRect().contains(event.scenePos()):
|
|
# ignore events outside the scene rect
|
|
return
|
|
elif self.inserter_ is not None:
|
|
# insert mode
|
|
self.inserter_.mousePressEvent(event, self.root_)
|
|
else:
|
|
# selection mode
|
|
QGraphicsScene.mousePressEvent(self, event)
|
|
|
|
def mouseReleaseEvent(self, event):
|
|
if self.debug_:
|
|
print "mouseReleaseEvent", self.sceneRect().contains(event.scenePos()), event.scenePos()
|
|
if self.inserter_ is not None:
|
|
# insert mode
|
|
self.inserter_.mouseReleaseEvent(event, self.root_)
|
|
else:
|
|
# selection mode
|
|
QGraphicsScene.mouseReleaseEvent(self, event)
|
|
|
|
def mouseMoveEvent(self, event):
|
|
#if self.debug_:
|
|
# print "mouseMoveEvent", self.sceneRect().contains(event.scenePos()), event.scenePos()
|
|
if self.inserter_ is not None:
|
|
# insert mode
|
|
self.inserter_.mouseMoveEvent(event, self.root_)
|
|
else:
|
|
# selection mode
|
|
QGraphicsScene.mouseMoveEvent(self, event)
|
|
|
|
#
|
|
# key event handlers
|
|
#______________________________________________________________________________________________________
|
|
def keyPressEvent(self, event):
|
|
if self.debug_:
|
|
print "keyPressEvent", event
|
|
|
|
if self.model_ is None or not self.root_.isValid():
|
|
event.ignore()
|
|
return
|
|
|
|
if self.inserter_ is not None:
|
|
# insert mode
|
|
self.inserter_.keyPressEvent(event, self.root_)
|
|
else:
|
|
# selection mode
|
|
if event.key() == Qt.Key_Delete:
|
|
for item in self.selectedItems():
|
|
index = item.index()
|
|
index.model().removeAnnotation(index)
|
|
event.accept()
|
|
return
|
|
|
|
if ord('0') <= event.key() <= ord('9'):
|
|
if self.last_key_ is None:
|
|
self.last_key_ = event.key()
|
|
else:
|
|
id = chr(self.last_key_) + chr(event.key())
|
|
print "id=", id
|
|
for item in self.selectedItems():
|
|
index = item.index()
|
|
data = dict(index.data(DataRole).toPyObject().iteritems())
|
|
if data['type'] == 'rect':
|
|
data['id'] = id
|
|
index.model().setData(index, QVariant(data), DataRole)
|
|
self.last_key_ = None
|
|
|
|
event.ignore()
|
|
|
|
#
|
|
# slots for signals from the model
|
|
# this is the implemenation of the scene as a view of the model
|
|
#______________________________________________________________________________________________________
|
|
def dataChanged(self, indexFrom, indexTo):
|
|
if self.root_ != indexFrom.parent() or self.root_ != indexTo.parent():
|
|
return
|
|
|
|
for row in range(indexFrom.row(), indexTo.row()+1):
|
|
item = self.itemFromIndex(indexFrom.sibling(row, 0))
|
|
if item is not None:
|
|
item.dataChanged()
|
|
|
|
def rowsInserted(self, index, first, last):
|
|
if self.root_ != index:
|
|
return
|
|
|
|
self.insertItems(first, last)
|
|
|
|
|
|
def rowsAboutToBeRemoved(self, index, first, last):
|
|
if self.root_ != index:
|
|
return
|
|
|
|
for row in range(first, last+1):
|
|
item = self.itemFromIndex(index.child(row, 0))
|
|
if item is not None:
|
|
self.removeItem(item)
|
|
|
|
def rowsRemoved(self, index, first, last):
|
|
pass
|
|
|
|
def itemFromIndex(self, index):
|
|
index = index.model().mapToSource(index) # TODO: solve this somehow else
|
|
for item in self.items():
|
|
# some graphics items will not have an index method,
|
|
# we just skip these
|
|
if hasattr(item, 'index') and item.index() == index:
|
|
return item
|
|
return None
|
|
|
|
#
|
|
# message handling and displaying
|
|
#______________________________________________________________________________________________________
|
|
def setMessage(self, message):
|
|
if self.message_ is not None:
|
|
self.clearMessage()
|
|
|
|
if message is None or message == "":
|
|
return
|
|
|
|
# TODO don't use text item at all, just draw the text in drawForeground
|
|
self.message_ = message
|
|
self.message_text_item_ = QGraphicsSimpleTextItem(message)
|
|
self.message_text_item_.setPos(20, 20)
|
|
self.invalidate(QRectF(), QGraphicsScene.ForegroundLayer)
|
|
|
|
def clearMessage(self):
|
|
if self.message_ is not None:
|
|
self.message_text_item_ = None
|
|
self.message_ = None
|
|
self.invalidate(QRectF(), QGraphicsScene.ForegroundLayer)
|
|
|
|
def drawForeground(self, painter, rect):
|
|
QGraphicsScene.drawForeground(self, painter, rect)
|
|
|
|
if self.message_ is not None:
|
|
assert self.message_text_item_ is not None
|
|
|
|
painter.setTransform(QTransform())
|
|
painter.setBrush(QColor('lightGray'))
|
|
painter.setPen(QPen(QBrush(QColor('black')), 2))
|
|
|
|
br = self.message_text_item_.boundingRect()
|
|
|
|
painter.drawRoundedRect(QRectF(10, 10, br.width()+20, br.height()+20), 10.0, 10.0)
|
|
painter.setTransform(QTransform.fromTranslate(20, 20))
|
|
painter.setPen(QPen(QColor('black'), 1))
|
|
|
|
self.message_text_item_.paint(painter, QStyleOptionGraphicsItem(), None)
|
|
|