def showImage(self, image):
scene = self._ui._graphicsView.scene()
items = scene.items()
for item in items:
scene.removeItem(item)
item = PixmapItem(image)
item.setMoveEventHandler(self._onMouseHoverImage)
scene.addItem(item)
def createScene(self):
self.scene = Scene(self.window)
self.ui.view.setScene(self.scene)
QObject.connect(self.scene, SIGNAL("selectToolRequested()"), self.slotSelectToolRequested)
QObject.connect(self.scene, SIGNAL("selectionChanged()"), self.slotSelectionChanged)
self.pixmapItem = PixmapItem()
self.pixmapItem.setZValue(-1)
self.scene.addItem(self.pixmapItem)
class Controller(QObject):
def __init__(self, parent):
QObject.__init__(self, parent)
self.window = KMainWindow()
self.ui = Ui_MainWindow()
self.ui.setupUi(self.window)
self.createActions()
self.createDragMeWidget()
self.createScene()
self.createToolBox()
self.window.resize(700, 500)
def createScene(self):
self.scene = Scene(self.window)
self.ui.view.setScene(self.scene)
QObject.connect(self.scene, SIGNAL("selectToolRequested()"), self.slotSelectToolRequested)
QObject.connect(self.scene, SIGNAL("selectionChanged()"), self.slotSelectionChanged)
self.pixmapItem = PixmapItem()
self.pixmapItem.setZValue(-1)
self.scene.addItem(self.pixmapItem)
def createToolBox(self):
self.toolGroup = QActionGroup(self)
self.toolGroup.addAction(self.ui.actionSelect)
self.toolGroup.addAction(self.ui.actionBubble)
self.toolGroup.addAction(self.ui.actionLine)
QObject.connect(self.toolGroup, SIGNAL("triggered(QAction*)"), self.slotToolChanged)
self.ui.toolBar.addSeparator()
self.colorSelector = KColorButton()
self.colorSelector.setColor(self.scene.newShapeSettings.pen.color())
QObject.connect(self.colorSelector, SIGNAL("changed(const QColor&)"), self.slotColorChanged)
self.ui.toolBar.addWidget(self.colorSelector)
self.thicknessSelector = QSpinBox()
self.thicknessSelector.setMinimum(1)
self.thicknessSelector.setMaximum(16)
self.thicknessSelector.setValue(self.scene.newShapeSettings.pen.width())
QObject.connect(self.thicknessSelector, SIGNAL("valueChanged(int)"), self.slotThicknessChanged)
self.ui.toolBar.addWidget(self.thicknessSelector)
def createDragMeWidget(self):
dragMeWidget = DragWidget(self.tr("Drag Me"), self.window)
self.ui.mainToolBar.addWidget(dragMeWidget)
QObject.connect(dragMeWidget, SIGNAL("dragStarted()"), self.slotDragStarted)
def createActions(self):
actionOpen = KStandardAction.open(self.open, self)
actionSave = KStandardAction.save(self.save, self)
actionScreenshot = KAction(self.tr("Screenshot"), self)
actionScreenshot.setIcon(KIcon("camera-photo"))
QObject.connect(actionScreenshot, SIGNAL("triggered()"), self.grabScreenshot)
actionDelete = KAction(self)
actionDelete.setShortcut(Qt.Key_Delete)
QObject.connect(actionDelete, SIGNAL("triggered()"), self.deleteItems)
self.ui.mainToolBar.addAction(actionOpen)
self.ui.mainToolBar.addAction(actionSave)
self.ui.mainToolBar.addSeparator()
self.ui.mainToolBar.addAction(actionScreenshot)
self.window.addAction(actionDelete)
def slotDragStarted(self):
drag = QDrag(self.window)
mimeData = QMimeData()
variant = QVariant(self.imageFromScene())
mimeData.setImageData(variant)
drag.setMimeData(mimeData)
drag.start()
def selectedShapes(self):
return [self.scene.shapeForItem(x) for x in self.scene.selectedItems()]
def slotSelectToolRequested(self):
self.ui.actionSelect.setChecked(True)
self.slotToolChanged(self.ui.actionSelect)
def slotSelectionChanged(self):
shapes = self.selectedShapes()
color = None
thickness = None
if shapes:
color = shapes[0].settings.pen.color()
thickness = shapes[0].settings.pen.width()
for shape in shapes[1:]:
pen = shape.settings.pen
if color and pen.color() != color:
color = None
if thickness and pen.width() != thickness:
thickness = None
else:
color = self.scene.newShapeSettings.pen.color()
thickness = self.scene.newShapeSettings.pen.width()
if color:
self.colorSelector.setColor(color)
if thickness:
self.thicknessSelector.setValue(thickness)
def slotColorChanged(self, color):
shapes = self.selectedShapes()
if shapes:
for shape in shapes:
shape.settings.setColor(color)
else:
self.scene.newShapeSettings.setColor(color)
def slotThicknessChanged(self, thickness):
shapes = self.selectedShapes()
if shapes:
for shape in shapes:
shape.settings.setThickness(thickness)
else:
self.scene.newShapeSettings.setThickness(thickness)
def show(self):
self.window.show()
def open(self):
name = QFileDialog.getOpenFileName(self.window, self.tr("Open Image"))
if name.isEmpty():
return
self.load(name)
def grabScreenshot(self):
pos = self.window.pos()
self.window.hide()
try:
pix = grab.showDialog()
if pix is not None:
self.setPixmap(pix)
finally:
self.window.show()
self.window.move(pos)
KWindowSystem.forceActiveWindow(self.window.winId())
def save(self):
name = QFileDialog.getSaveFileName(self.window, self.tr("Save Image as"))
if name.isEmpty():
return
image = self.imageFromScene()
ok = image.save(name)
if not ok:
KMessageBox.error(self.window, self.tr("Failed to save image as %1").arg(name));
def imageFromScene(self):
# Hide elements we don't want to show
selection = self.scene.selectedItems()
self.scene.clearSelection()
self.pixmapItem.setHandlesVisible(False)
# Render
rect = self.scene.itemsBoundingRect()
image = QImage(int(rect.width()), int(rect.height()), QImage.Format_ARGB32)
image.fill(Qt.transparent)
painter = QPainter(image)
painter.setRenderHint(QPainter.Antialiasing)
self.scene.render(painter, QRectF(image.rect()), rect)
painter.end()
# Restore hidden elements
for item in selection:
item.setSelected(True)
self.pixmapItem.setHandlesVisible(True)
return image
def load(self, fileName):
self.setPixmap(QPixmap(fileName))
def setPixmap(self, pix):
self.pixmapItem.setPixmap(pix)
self.scene.setSceneRect(QRectF(pix.rect()))
def slotToolChanged(self, action):
toolFromAction = {
self.ui.actionBubble: AddBubbleTool,
self.ui.actionLine: AddLineTool,
}
klass = toolFromAction.get(action)
if klass:
tool = klass(self.scene)
else:
tool = None
self.scene.setTool(tool)
def deleteItems(self):
for shape in self.selectedShapes():
self.scene.removeShape(shape)
