source: orange/Orange/OrangeCanvas/document/schemeedit.py @ 11464:7b8484a00739

Revision 11464:7b8484a00739, 52.3 KB checked in by Ales Erjavec <ales.erjavec@…>, 12 months ago (diff)

Fixed CanvasScene clear/reset.

Disconnect all SchemeNodes/Links/Annotations from their corresponding
graphics items in the CanvasScene.

RevLine 
[11149]1"""
[11390]2====================
3Scheme Editor Widget
4====================
5
[11149]6
7"""
[11222]8
[11208]9import sys
[11151]10import logging
[11279]11import itertools
[11222]12
[11151]13from operator import attrgetter
[11236]14from contextlib import nested
[11264]15from urllib import urlencode
[11151]16
[11149]17from PyQt4.QtGui import (
[11201]18    QWidget, QVBoxLayout, QInputDialog, QMenu, QAction, QActionGroup,
19    QKeySequence, QUndoStack, QGraphicsItem, QGraphicsObject,
20    QGraphicsTextItem, QCursor, QFont, QPainter, QPixmap, QColor,
[11296]21    QIcon, QWhatsThisClickedEvent
[11149]22)
23
[11243]24from PyQt4.QtCore import (
[11296]25    Qt, QObject, QEvent, QSignalMapper, QRectF, QCoreApplication
[11243]26)
27
[11149]28from PyQt4.QtCore import pyqtProperty as Property, pyqtSignal as Signal
29
[11243]30from ..registry.qt import whats_this_helper
31from ..gui.quickhelp import QuickHelpTipEvent
[11208]32from ..gui.utils import message_information, disabled
[11390]33from ..scheme import scheme, SchemeNode, SchemeLink, BaseSchemeAnnotation
[11149]34from ..canvas.scene import CanvasScene
35from ..canvas.view import CanvasView
36from ..canvas import items
[11151]37from . import interactions
[11149]38from . import commands
[11163]39from . import quickmenu
[11149]40
[11390]41
[11151]42log = logging.getLogger(__name__)
43
44
45# TODO: Should this be moved to CanvasScene?
46class GraphicsSceneFocusEventListener(QGraphicsObject):
47
48    itemFocusedIn = Signal(QGraphicsItem)
49    itemFocusedOut = Signal(QGraphicsItem)
50
51    def __init__(self, parent=None):
52        QGraphicsObject.__init__(self, parent)
53        self.setFlag(QGraphicsItem.ItemHasNoContents)
54
55    def sceneEventFilter(self, obj, event):
56        if event.type() == QEvent.FocusIn and \
57                obj.flags() & QGraphicsItem.ItemIsFocusable:
58            obj.focusInEvent(event)
59            if obj.hasFocus():
60                self.itemFocusedIn.emit(obj)
61            return True
62        elif event.type() == QEvent.FocusOut:
63            obj.focusOutEvent(event)
64            if not obj.hasFocus():
65                self.itemFocusedOut.emit(obj)
66            return True
67
68        return QGraphicsObject.sceneEventFilter(self, obj, event)
69
70    def boundingRect(self):
71        return QRectF()
[11149]72
73
74class SchemeEditWidget(QWidget):
[11371]75    """
[11390]76    A widget for editing a :class:`~.scheme.Scheme` instance.
[11371]77
78    """
79    #: Undo command has become available/unavailable.
[11149]80    undoAvailable = Signal(bool)
[11371]81
82    #: Redo command has become available/unavailable.
[11149]83    redoAvailable = Signal(bool)
[11371]84
85    #: Document modified state has changed.
[11149]86    modificationChanged = Signal(bool)
[11371]87
88    #: Undo command was added to the undo stack.
[11149]89    undoCommandAdded = Signal()
[11371]90
91    #: Item selection has changed.
[11149]92    selectionChanged = Signal()
93
[11371]94    #: Document title has changed.
[11149]95    titleChanged = Signal(unicode)
96
[11371]97    #: Document path has changed.
[11219]98    pathChanged = Signal(unicode)
99
[11198]100    # Quick Menu triggers
101    (NoTriggers,
102     Clicked,
103     DoubleClicked,
104     SpaceKey,
105     AnyKey) = [0, 1, 2, 4, 8]
106
[11149]107    def __init__(self, parent=None, ):
108        QWidget.__init__(self, parent)
109
110        self.__modified = False
111        self.__registry = None
[11156]112        self.__scheme = None
[11219]113        self.__path = u""
[11198]114        self.__quickMenuTriggers = SchemeEditWidget.SpaceKey | \
115                                   SchemeEditWidget.DoubleClicked
116        self.__emptyClickButtons = 0
[11259]117        self.__channelNamesVisible = True
[11411]118        self.__nodeAnimationEnabled = True
[11198]119        self.__possibleSelectionHandler = None
[11151]120        self.__possibleMouseItemsMove = False
121        self.__itemsMoving = {}
[11154]122        self.__contextMenuTarget = None
[11163]123        self.__quickMenu = None
[11243]124        self.__quickTip = ""
[11151]125
[11198]126        self.__undoStack = QUndoStack(self)
127        self.__undoStack.cleanChanged[bool].connect(self.__onCleanChanged)
128
[11222]129        # OWBaseWidget properties when set to clean state
130        self.__cleanSettings = []
131
[11151]132        self.__editFinishedMapper = QSignalMapper(self)
133        self.__editFinishedMapper.mapped[QObject].connect(
134            self.__onEditingFinished
135        )
136
137        self.__annotationGeomChanged = QSignalMapper(self)
138
[11194]139        self.__setupActions()
[11149]140        self.__setupUi()
141
[11208]142        self.__editMenu = QMenu(self.tr("&Edit"), self)
143        self.__editMenu.addAction(self.__undoAction)
144        self.__editMenu.addAction(self.__redoAction)
145        self.__editMenu.addSeparator()
146        self.__editMenu.addAction(self.__selectAllAction)
147
148        self.__widgetMenu = QMenu(self.tr("&Widget"), self)
149        self.__widgetMenu.addAction(self.__openSelectedAction)
150        self.__widgetMenu.addSeparator()
[11225]151        self.__widgetMenu.addAction(self.__renameAction)
[11208]152        self.__widgetMenu.addAction(self.__removeSelectedAction)
153        self.__widgetMenu.addSeparator()
154        self.__widgetMenu.addAction(self.__helpAction)
155
156        self.__linkMenu = QMenu(self.tr("Link"), self)
[11194]157        self.__linkMenu.addAction(self.__linkEnableAction)
158        self.__linkMenu.addSeparator()
159        self.__linkMenu.addAction(self.__linkRemoveAction)
160        self.__linkMenu.addAction(self.__linkResetAction)
161
162    def __setupActions(self):
163
164        self.__zoomAction = \
165            QAction(self.tr("Zoom"), self,
[11208]166                    objectName="zoom-action",
[11194]167                    checkable=True,
168                    shortcut=QKeySequence.ZoomIn,
169                    toolTip=self.tr("Zoom in the scheme."),
[11373]170                    toggled=self.toggleZoom,
[11194]171                    )
172
173        self.__cleanUpAction = \
174            QAction(self.tr("Clean Up"), self,
[11208]175                    objectName="cleanup-action",
[11194]176                    toolTip=self.tr("Align widget to a grid."),
177                    triggered=self.alignToGrid,
178                    )
179
180        self.__newTextAnnotationAction = \
181            QAction(self.tr("Text"), self,
[11208]182                    objectName="new-text-action",
[11194]183                    toolTip=self.tr("Add a text annotation to the scheme."),
184                    checkable=True,
185                    toggled=self.__toggleNewTextAnnotation,
186                    )
187
[11201]188        # Create a font size menu for the new annotation action.
189        self.__fontMenu = QMenu("Font Size", self)
190        self.__fontActionGroup = group = \
191            QActionGroup(self, exclusive=True,
192                         triggered=self.__onFontSizeTriggered)
193
194        def font(size):
[11343]195            f = QFont(self.font())
196            f.setPixelSize(size)
197            return f
[11201]198
199        for size in [12, 14, 16, 18, 20, 22, 24]:
[11343]200            action = QAction("%ipx" % size, group,
[11201]201                             checkable=True,
202                             font=font(size))
203
204            self.__fontMenu.addAction(action)
205
206        group.actions()[2].setChecked(True)
207
208        self.__newTextAnnotationAction.setMenu(self.__fontMenu)
209
[11194]210        self.__newArrowAnnotationAction = \
211            QAction(self.tr("Arrow"), self,
[11208]212                    objectName="new-arrow-action",
[11194]213                    toolTip=self.tr("Add a arrow annotation to the scheme."),
214                    checkable=True,
215                    toggled=self.__toggleNewArrowAnnotation,
216                    )
217
[11201]218        # Create a color menu for the arrow annotation action
219        self.__arrowColorMenu = QMenu("Arrow Color",)
220        self.__arrowColorActionGroup = group = \
221            QActionGroup(self, exclusive=True,
222                         triggered=self.__onArrowColorTriggered)
223
224        def color_icon(color):
225            icon = QIcon()
226            for size in [16, 24, 32]:
227                pixmap = QPixmap(size, size)
228                pixmap.fill(QColor(0, 0, 0, 0))
229                p = QPainter(pixmap)
230                p.setRenderHint(QPainter.Antialiasing)
231                p.setBrush(color)
232                p.setPen(Qt.NoPen)
233                p.drawEllipse(1, 1, size - 2, size - 2)
234                p.end()
235                icon.addPixmap(pixmap)
236            return icon
237
238        for color in ["#000", "#C1272D", "#662D91", "#1F9CDF", "#39B54A"]:
239            icon = color_icon(QColor(color))
240            action = QAction(group, icon=icon, checkable=True,
241                             iconVisibleInMenu=True)
242            action.setData(color)
243            self.__arrowColorMenu.addAction(action)
244
245        group.actions()[1].setChecked(True)
246
247        self.__newArrowAnnotationAction.setMenu(self.__arrowColorMenu)
248
[11208]249        self.__undoAction = self.__undoStack.createUndoAction(self)
250        self.__undoAction.setShortcut(QKeySequence.Undo)
251        self.__undoAction.setObjectName("undo-action")
252
253        self.__redoAction = self.__undoStack.createRedoAction(self)
254        self.__redoAction.setShortcut(QKeySequence.Redo)
255        self.__redoAction.setObjectName("redo-action")
256
257        self.__selectAllAction = \
258            QAction(self.tr("Select all"), self,
259                    objectName="select-all-action",
260                    toolTip=self.tr("Select all items."),
261                    triggered=self.selectAll,
262                    shortcut=QKeySequence.SelectAll
263                    )
264
265        self.__openSelectedAction = \
266            QAction(self.tr("Open"), self,
267                    objectName="open-action",
268                    toolTip=self.tr("Open selected widget"),
269                    triggered=self.openSelected,
270                    enabled=False)
271
272        self.__removeSelectedAction = \
273            QAction(self.tr("Remove"), self,
274                    objectName="remove-selected",
275                    toolTip=self.tr("Remove selected items"),
276                    triggered=self.removeSelected,
277                    enabled=False
278                    )
279
280        shortcuts = [Qt.Key_Delete,
281                     Qt.ControlModifier + Qt.Key_Backspace]
282
283        if sys.platform == "darwin":
284            # Command Backspace should be the first
285            # (visible shortcut in the menu)
286            shortcuts.reverse()
287
288        self.__removeSelectedAction.setShortcuts(shortcuts)
289
290        self.__renameAction = \
291            QAction(self.tr("Rename"), self,
292                    objectName="rename-action",
293                    toolTip=self.tr("Rename selected widget"),
294                    triggered=self.__onRenameAction,
295                    shortcut=QKeySequence(Qt.Key_F2),
296                    enabled=False)
297
298        self.__helpAction = \
299            QAction(self.tr("Help"), self,
300                    objectName="help-action",
301                    toolTip=self.tr("Show widget help"),
302                    triggered=self.__onHelpAction,
[11341]303                    shortcut=QKeySequence("F1")
[11208]304                    )
305
[11154]306        self.__linkEnableAction = \
307            QAction(self.tr("Enabled"), self,
308                    objectName="link-enable-action",
[11194]309                    triggered=self.__toggleLinkEnabled,
[11154]310                    checkable=True,
311                    )
312
313        self.__linkRemoveAction = \
314            QAction(self.tr("Remove"), self,
315                    objectName="link-remove-action",
316                    triggered=self.__linkRemove,
317                    toolTip=self.tr("Remove link."),
318                    )
319
320        self.__linkResetAction = \
321            QAction(self.tr("Reset Signals"), self,
322                    objectName="link-reset-action",
323                    triggered=self.__linkReset,
324                    )
325
[11194]326        self.addActions([self.__newTextAnnotationAction,
327                         self.__newArrowAnnotationAction,
328                         self.__linkEnableAction,
329                         self.__linkRemoveAction,
330                         self.__linkResetAction])
[11154]331
[11236]332        # Actions which should be disabled while a multistep
333        # interaction is in progress.
334        self.__disruptiveActions = \
335                [self.__undoAction,
336                 self.__redoAction,
337                 self.__removeSelectedAction,
338                 self.__selectAllAction]
339
[11149]340    def __setupUi(self):
341        layout = QVBoxLayout()
342        layout.setContentsMargins(0, 0, 0, 0)
343        layout.setSpacing(0)
344
345        scene = CanvasScene()
[11259]346        scene.set_channel_names_visible(self.__channelNamesVisible)
[11411]347        scene.set_node_animation_enabled(self.__nodeAnimationEnabled)
[11343]348        scene.setFont(self.font())
[11259]349
[11149]350        view = CanvasView(scene)
[11159]351        view.setFrameStyle(CanvasView.NoFrame)
[11149]352        view.setRenderHint(QPainter.Antialiasing)
[11154]353        view.setContextMenuPolicy(Qt.CustomContextMenu)
354        view.customContextMenuRequested.connect(
355            self.__onCustomContextMenuRequested
356        )
357
[11149]358        self.__view = view
359        self.__scene = scene
360
[11151]361        self.__focusListener = GraphicsSceneFocusEventListener()
362        self.__focusListener.itemFocusedIn.connect(self.__onItemFocusedIn)
363        self.__focusListener.itemFocusedOut.connect(self.__onItemFocusedOut)
364        self.__scene.addItem(self.__focusListener)
365
366        self.__scene.selectionChanged.connect(
367            self.__onSelectionChanged
368        )
369
[11149]370        layout.addWidget(view)
371        self.setLayout(layout)
372
[11194]373    def toolbarActions(self):
[11371]374        """
375        Return a list of actions that can be inserted into a toolbar.
[11390]376        At the moment these are:
377
378            - 'Zoom' action
379            - 'Clean up' action (align to grid)
380            - 'New text annotation' action (with a size menu)
381            - 'New arrow annotation' action (with a color menu)
382
[11194]383        """
384        return [self.__zoomAction,
385                self.__cleanUpAction,
386                self.__newTextAnnotationAction,
387                self.__newArrowAnnotationAction]
388
[11208]389    def menuBarActions(self):
[11371]390        """
[11390]391        Return a list of actions that can be inserted into a `QMenuBar`.
[11208]392
393        """
394        return [self.__editMenu.menuAction(), self.__widgetMenu.menuAction()]
395
[11149]396    def isModified(self):
[11371]397        """
398        Is the document is a modified state.
[11222]399        """
400        return self.__modified or not self.__undoStack.isClean()
[11149]401
402    def setModified(self, modified):
[11371]403        """
404        Set the document modified state.
405        """
[11222]406        if self.__modified != modified:
407            self.__modified = modified
408
409        if not modified:
410            self.__cleanSettings = self.__scheme.widget_settings()
411            self.__undoStack.setClean()
[11149]412        else:
[11222]413            self.__cleanSettings = []
[11149]414
415    modified = Property(bool, fget=isModified, fset=setModified)
416
[11222]417    def isModifiedStrict(self):
[11371]418        """
419        Is the document modified. Run a strict check against all node
[11222]420        properties as they were at the time when the last call to
421        `setModified(True)` was made.
422
423        """
424        settingsChanged = self.__cleanSettings != \
425                          self.__scheme.widget_settings()
426
427        log.debug("Modified strict check (modified flag: %s, "
428                  "undo stack clean: %s, properties: %s)",
429                  self.__modified,
430                  self.__undoStack.isClean(),
431                  settingsChanged)
432
433        return self.isModified() or settingsChanged
434
[11198]435    def setQuickMenuTriggers(self, triggers):
[11371]436        """
437        Set quick menu trigger flags.
[11390]438
439        Flags can be a bitwise `or` of:
440
441            - `SchemeEditWidget.NoTrigeres`
442            - `SchemeEditWidget.Clicked`
443            - `SchemeEditWidget.DoubleClicked`
444            - `SchemeEditWidget.SpaceKey`
445            - `SchemeEditWidget.AnyKey`
446
[11198]447        """
448        if self.__quickMenuTriggers != triggers:
449            self.__quickMenuTriggers = triggers
450
[11451]451    def quickMenuTriggers(self):
[11371]452        """
453        Return quick menu trigger flags.
454        """
[11198]455        return self.__quickMenuTriggers
456
[11259]457    def setChannelNamesVisible(self, visible):
[11371]458        """
459        Set channel names visibility state. When enabled the links
460        in the view will have a source/sink channel names displayed over
461        them.
[11259]462
463        """
464        if self.__channelNamesVisible != visible:
465            self.__channelNamesVisible = visible
466            self.__scene.set_channel_names_visible(visible)
467
468    def channelNamesVisible(self):
[11371]469        """
470        Return the channel name visibility state.
[11259]471        """
472        return self.__channelNamesVisible
473
[11411]474    def setNodeAnimationEnabled(self, enabled):
475        """
476        Set the node item animation enabled state.
477        """
478        if self.__nodeAnimationEnabled != enabled:
479            self.__nodeAnimationEnabled = enabled
480            self.__scene.set_node_animation_enabled(enabled)
481
482    def nodeAnimationEnabled(self):
483        """
484        Return the node item animation enabled state.
485        """
486        return self.__nodeAnimationEnabled
487
[11149]488    def undoStack(self):
[11371]489        """
490        Return the undo stack.
[11149]491        """
492        return self.__undoStack
493
[11219]494    def setPath(self, path):
[11371]495        """
496        Set the path associated with the current scheme.
[11219]497
[11259]498        .. note:: Calling `setScheme` will invalidate the path (i.e. set it
499                  to an empty string)
[11219]500
501        """
502        if self.__path != path:
503            self.__path = unicode(path)
504            self.pathChanged.emit(self.__path)
505
506    def path(self):
[11371]507        """
508        Return the path associated with the scheme
[11219]509        """
510        return self.__path
511
[11149]512    def setScheme(self, scheme):
[11371]513        """
[11390]514        Set the :class:`~.scheme.Scheme` instance to display/edit.
[11371]515        """
[11149]516        if self.__scheme is not scheme:
[11156]517            if self.__scheme:
518                self.__scheme.title_changed.disconnect(self.titleChanged)
[11165]519                self.__scheme.node_added.disconnect(self.__onNodeAdded)
520                self.__scheme.node_removed.disconnect(self.__onNodeRemoved)
[11296]521                self.__scheme.removeEventFilter(self)
[11156]522
[11149]523            self.__scheme = scheme
[11151]524
[11219]525            self.setPath("")
526
[11156]527            if self.__scheme:
528                self.__scheme.title_changed.connect(self.titleChanged)
[11165]529                self.__scheme.node_added.connect(self.__onNodeAdded)
530                self.__scheme.node_removed.connect(self.__onNodeRemoved)
[11156]531                self.titleChanged.emit(scheme.title)
[11222]532                self.__cleanSettings = scheme.widget_settings()
533            else:
534                self.__cleanSettings = []
[11156]535
[11245]536            # Clear the current item selection in the scene so edit action
537            # states are updated accordingly.
538            self.__scene.clearSelection()
539
[11151]540            self.__annotationGeomChanged.deleteLater()
541            self.__annotationGeomChanged = QSignalMapper(self)
542
[11149]543            self.__undoStack.clear()
544
[11151]545            self.__focusListener.itemFocusedIn.disconnect(
546                self.__onItemFocusedIn
547            )
548            self.__focusListener.itemFocusedOut.disconnect(
549                self.__onItemFocusedOut
550            )
551
552            self.__scene.selectionChanged.disconnect(
553                self.__onSelectionChanged
554            )
555
[11149]556            self.__scene.removeEventFilter(self)
[11464]557
558            # Clear all items from the scene
559            self.__scene.clear_scene()
560
[11149]561            self.__scene.deleteLater()
562
563            self.__scene = CanvasScene()
564            self.__view.setScene(self.__scene)
[11259]565            self.__scene.set_channel_names_visible(self.__channelNamesVisible)
[11411]566            self.__scene.set_node_animation_enabled(
567                self.__nodeAnimationEnabled
568            )
569
[11343]570            self.__scene.setFont(self.font())
[11259]571
[11149]572            self.__scene.installEventFilter(self)
573
574            self.__scene.set_registry(self.__registry)
[11192]575
576            # Focus listener
577            self.__focusListener = GraphicsSceneFocusEventListener()
578            self.__focusListener.itemFocusedIn.connect(
579                self.__onItemFocusedIn
580            )
581            self.__focusListener.itemFocusedOut.connect(
582                self.__onItemFocusedOut
583            )
584            self.__scene.addItem(self.__focusListener)
[11149]585
[11151]586            self.__scene.selectionChanged.connect(
587                self.__onSelectionChanged
588            )
589
590            self.__scene.node_item_activated.connect(
591                self.__onNodeActivate
592            )
593
594            self.__scene.annotation_added.connect(
595                self.__onAnnotationAdded
596            )
597
598            self.__scene.annotation_removed.connect(
599                self.__onAnnotationRemoved
600            )
601
[11192]602            self.__scene.set_scheme(scheme)
[11151]603
[11266]604            if self.__scheme:
605                for node in self.__scheme.nodes:
606                    self.__onNodeAdded(node)
607
[11296]608                self.__scheme.installEventFilter(self)
609
[11149]610    def scheme(self):
[11371]611        """
[11390]612        Return the :class:`~.scheme.Scheme` edited by the widget.
[11221]613        """
[11149]614        return self.__scheme
615
616    def scene(self):
[11371]617        """
618        Return the :class:`QGraphicsScene` instance used to display the
619        current scheme.
620
[11221]621        """
[11149]622        return self.__scene
623
624    def view(self):
[11371]625        """
[11390]626        Return the :class:`QGraphicsView` instance used to display the
[11371]627        current scene.
628
[11221]629        """
[11149]630        return self.__view
631
632    def setRegistry(self, registry):
[11371]633        # Is this method necessary?
634        # It should be removed when the scene (items) is fixed
635        # so all information regarding the visual appearance is
636        # included in the node/widget description.
[11149]637        self.__registry = registry
638        if self.__scene:
639            self.__scene.set_registry(registry)
[11163]640            self.__quickMenu = None
641
642    def quickMenu(self):
[11371]643        """
[11390]644        Return a :class:`~.quickmenu.QuickMenu` popup menu instance for
645        new node creation.
646
[11163]647        """
648        if self.__quickMenu is None:
649            menu = quickmenu.QuickMenu(self)
650            if self.__registry is not None:
651                menu.setModel(self.__registry.model())
652            self.__quickMenu = menu
653        return self.__quickMenu
[11149]654
[11227]655    def setTitle(self, title):
[11279]656        """
657        Set the scheme title.
658        """
[11227]659        self.__undoStack.push(
660            commands.SetAttrCommand(self.__scheme, "title", title)
661        )
662
663    def setDescription(self, description):
[11279]664        """
665        Set the scheme description string.
666        """
[11227]667        self.__undoStack.push(
668            commands.SetAttrCommand(self.__scheme, "description", description)
669        )
670
[11149]671    def addNode(self, node):
[11279]672        """
[11390]673        Add a new node (:class:`.SchemeNode`) to the document.
[11149]674        """
675        command = commands.AddNodeCommand(self.__scheme, node)
676        self.__undoStack.push(command)
677
[11279]678    def createNewNode(self, description, title=None, position=None):
679        """
[11390]680        Create a new :class:`.SchemeNode` and add it to the document.
[11425]681        The new node is constructed using :func:`newNodeHelper` method.
[11151]682
[11149]683        """
[11279]684        node = self.newNodeHelper(description, title, position)
685        self.addNode(node)
[11151]686
[11279]687        return node
688
689    def newNodeHelper(self, description, title=None, position=None):
690        """
[11390]691        Return a new initialized :class:`.SchemeNode`. If `title`
692        and `position` are not supplied they are initialized to sensible
[11371]693        defaults.
[11279]694
695        """
696        if title is None:
697            title = self.enumerateTitle(description.name)
698
699        if position is None:
700            position = self.nextPosition()
701
[11390]702        return SchemeNode(description, title=title, position=position)
[11279]703
704    def enumerateTitle(self, title):
705        """
[11371]706        Enumerate a `title` string (i.e. add a number in parentheses) so
707        it is not equal to any node title in the current scheme.
[11279]708
709        """
710        curr_titles = set([node.title for node in self.scheme().nodes])
711        template = title + " ({0})"
712
713        enumerated = itertools.imap(template.format, itertools.count(1))
714        candidates = itertools.chain([title], enumerated)
715
716        seq = itertools.dropwhile(curr_titles.__contains__, candidates)
717        return next(seq)
718
719    def nextPosition(self):
720        """
[11390]721        Return the next default node position as a (x, y) tuple. This is
[11279]722        a position left of the last added node.
723
724        """
725        nodes = self.scheme().nodes
726        if nodes:
727            x, y = nodes[-1].position
728            position = (x + 150, y)
[11151]729        else:
[11279]730            position = (150, 150)
731        return position
[11149]732
733    def removeNode(self, node):
[11371]734        """
[11390]735        Remove a `node` (:class:`.SchemeNode`) from the scheme
[11221]736        """
[11149]737        command = commands.RemoveNodeCommand(self.__scheme, node)
738        self.__undoStack.push(command)
739
740    def renameNode(self, node, title):
[11371]741        """
[11390]742        Rename a `node` (:class:`.SchemeNode`) to `title`.
[11221]743        """
[11149]744        command = commands.RenameNodeCommand(self.__scheme, node, title)
745        self.__undoStack.push(command)
746
747    def addLink(self, link):
[11371]748        """
[11390]749        Add a `link` (:class:`.SchemeLink`) to the scheme.
[11221]750        """
[11149]751        command = commands.AddLinkCommand(self.__scheme, link)
752        self.__undoStack.push(command)
753
754    def removeLink(self, link):
[11371]755        """
[11390]756        Remove a link (:class:`.SchemeLink`) from the scheme.
[11221]757        """
[11149]758        command = commands.RemoveLinkCommand(self.__scheme, link)
759        self.__undoStack.push(command)
760
761    def addAnnotation(self, annotation):
[11371]762        """
[11390]763        Add `annotation` (:class:`.BaseSchemeAnnotation`) to the scheme
[11221]764        """
[11149]765        command = commands.AddAnnotationCommand(self.__scheme, annotation)
766        self.__undoStack.push(command)
767
768    def removeAnnotation(self, annotation):
[11371]769        """
[11390]770        Remove `annotation` (:class:`.BaseSchemeAnnotation`) from the scheme.
[11221]771        """
[11149]772        command = commands.RemoveAnnotationCommand(self.__scheme, annotation)
773        self.__undoStack.push(command)
774
775    def removeSelected(self):
[11371]776        """
777        Remove all selected items in the scheme.
[11221]778        """
[11149]779        selected = self.scene().selectedItems()
[11151]780        if not selected:
781            return
782
[11149]783        self.__undoStack.beginMacro(self.tr("Remove"))
784        for item in selected:
785            if isinstance(item, items.NodeItem):
786                node = self.scene().node_for_item(item)
787                self.__undoStack.push(
788                    commands.RemoveNodeCommand(self.__scheme, node)
789                )
790            elif isinstance(item, items.annotationitem.Annotation):
791                annot = self.scene().annotation_for_item(item)
792                self.__undoStack.push(
793                    commands.RemoveAnnotationCommand(self.__scheme, annot)
794                )
795        self.__undoStack.endMacro()
796
797    def selectAll(self):
[11371]798        """
799        Select all selectable items in the scheme.
[11221]800        """
[11149]801        for item in self.__scene.items():
802            if item.flags() & QGraphicsItem.ItemIsSelectable:
803                item.setSelected(True)
804
[11373]805    def toggleZoom(self, zoom):
[11371]806        """
807        Toggle view zoom. If `zoom` is True the scheme is displayed
808        scaled to 150%.
809
810        """
[11194]811        view = self.view()
812        if zoom:
813            view.scale(1.5, 1.5)
814        else:
815            view.resetTransform()
816
[11149]817    def alignToGrid(self):
[11151]818        """
[11371]819        Align nodes to a grid.
820        """
821        # TODO: The the current layout implementation is BAD (fix is urgent).
[11151]822        tile_size = 150
823        tiles = {}
824
825        nodes = sorted(self.scheme().nodes, key=attrgetter("position"))
826
827        if nodes:
828            self.__undoStack.beginMacro(self.tr("Align To Grid"))
829
830            for node in nodes:
831                x, y = node.position
832                x = int(round(float(x) / tile_size) * tile_size)
833                y = int(round(float(y) / tile_size) * tile_size)
834                while (x, y) in tiles:
835                    x += tile_size
836
837                self.__undoStack.push(
838                    commands.MoveNodeCommand(self.scheme(), node,
839                                             node.position, (x, y))
840                )
841
842                tiles[x, y] = node
843                self.__scene.item_for_node(node).setPos(x, y)
844
845            self.__undoStack.endMacro()
[11149]846
[11266]847    def focusNode(self):
[11371]848        """
[11390]849        Return the current focused :class:`.SchemeNode` or ``None`` if no
[11266]850        node has focus.
851
852        """
853        focus = self.__scene.focusItem()
854        node = None
855        if isinstance(focus, items.NodeItem):
856            try:
857                node = self.__scene.node_for_item(focus)
858            except KeyError:
859                # in case the node has been removed but the scene was not
860                # yet fully updated.
861                node = None
862        return node
863
[11149]864    def selectedNodes(self):
[11371]865        """
[11390]866        Return all selected :class:`.SchemeNode` items.
[11221]867        """
[11149]868        return map(self.scene().node_for_item,
869                   self.scene().selected_node_items())
870
[11221]871    def selectedAnnotations(self):
[11371]872        """
[11390]873        Return all selected :class:`.BaseSchemeAnnotation` items.
[11221]874        """
875        return map(self.scene().annotation_for_item,
876                   self.scene().selected_annotation_items())
877
[11149]878    def openSelected(self):
[11371]879        """
880        Open (show and raise) all widgets for the current selected nodes.
[11221]881        """
[11154]882        selected = self.scene().selected_node_items()
883        for item in selected:
884            self.__onNodeActivate(item)
[11149]885
886    def editNodeTitle(self, node):
[11371]887        """
888        Edit (rename) the `node`'s title. Opens an input dialog.
[11221]889        """
[11151]890        name, ok = QInputDialog.getText(
891                    self, self.tr("Rename"),
[11310]892                    unicode(self.tr("Enter a new name for the '%s' widget")) \
[11151]893                    % node.title,
894                    text=node.title
895                    )
896
897        if ok:
898            self.__undoStack.push(
899                commands.RenameNodeCommand(self.__scheme, node, node.title,
900                                           unicode(name))
901            )
[11149]902
903    def __onCleanChanged(self, clean):
[11156]904        if self.isWindowModified() != (not clean):
905            self.setWindowModified(not clean)
[11149]906            self.modificationChanged.emit(not clean)
907
[11343]908    def changeEvent(self, event):
909        if event.type() == QEvent.FontChange:
910            self.__updateFont()
911
912        QWidget.changeEvent(self, event)
913
[11149]914    def eventFilter(self, obj, event):
915        # Filter the scene's drag/drop events.
916        if obj is self.scene():
917            etype = event.type()
918            if  etype == QEvent.GraphicsSceneDragEnter or \
919                    etype == QEvent.GraphicsSceneDragMove:
920                mime_data = event.mimeData()
921                if mime_data.hasFormat(
922                        "application/vnv.orange-canvas.registry.qualified-name"
923                        ):
924                    event.acceptProposedAction()
925                return True
926            elif etype == QEvent.GraphicsSceneDrop:
927                data = event.mimeData()
928                qname = data.data(
929                    "application/vnv.orange-canvas.registry.qualified-name"
930                )
931                desc = self.__registry.widget(unicode(qname))
932                pos = event.scenePos()
[11279]933                self.createNewNode(desc, position=(pos.x(), pos.y()))
[11149]934                return True
935
936            elif etype == QEvent.GraphicsSceneMousePress:
937                return self.sceneMousePressEvent(event)
[11151]938            elif etype == QEvent.GraphicsSceneMouseMove:
939                return self.sceneMouseMoveEvent(event)
940            elif etype == QEvent.GraphicsSceneMouseRelease:
941                return self.sceneMouseReleaseEvent(event)
942            elif etype == QEvent.GraphicsSceneMouseDoubleClick:
943                return self.sceneMouseDoubleClickEvent(event)
[11389]944            elif etype == QEvent.KeyPress:
[11151]945                return self.sceneKeyPressEvent(event)
946            elif etype == QEvent.KeyRelease:
947                return self.sceneKeyReleaseEvent(event)
948            elif etype == QEvent.GraphicsSceneContextMenu:
949                return self.sceneContextMenuEvent(event)
[11149]950
[11296]951        elif obj is self.__scheme:
952            if event.type() == QEvent.WhatsThisClicked:
953                # Re post the event
954                self.__showHelpFor(event.href())
955
[11149]956        return QWidget.eventFilter(self, obj, event)
957
958    def sceneMousePressEvent(self, event):
959        scene = self.__scene
960        if scene.user_interaction_handler:
961            return False
962
963        pos = event.scenePos()
964
[11241]965        anchor_item = scene.item_at(pos, items.NodeAnchorItem,
966                                    buttons=Qt.LeftButton)
[11149]967        if anchor_item and event.button() == Qt.LeftButton:
968            # Start a new link starting at item
[11208]969            scene.clearSelection()
[11151]970            handler = interactions.NewLinkAction(self)
[11236]971            self._setUserInteractionHandler(handler)
[11149]972            return handler.mousePressEvent(event)
973
[11151]974        any_item = scene.item_at(pos)
975        if not any_item and event.button() == Qt.LeftButton:
[11198]976            self.__emptyClickButtons |= Qt.LeftButton
977            # Create a RectangleSelectionAction but do not set in on the scene
978            # just yet (instead wait for the mouse move event).
[11151]979            handler = interactions.RectangleSelectionAction(self)
[11198]980            rval = handler.mousePressEvent(event)
981            if rval == True:
982                self.__possibleSelectionHandler = handler
[11240]983            return rval
[11151]984
985        if any_item and event.button() == Qt.LeftButton:
986            self.__possibleMouseItemsMove = True
987            self.__itemsMoving.clear()
988            self.__scene.node_item_position_changed.connect(
989                self.__onNodePositionChanged
990            )
991            self.__annotationGeomChanged.mapped[QObject].connect(
992                self.__onAnnotationGeometryChanged
993            )
994
[11236]995            set_enabled_all(self.__disruptiveActions, False)
996
[11149]997        return False
998
[11151]999    def sceneMouseMoveEvent(self, event):
1000        scene = self.__scene
1001        if scene.user_interaction_handler:
1002            return False
[11149]1003
[11198]1004        if self.__emptyClickButtons & Qt.LeftButton and \
1005                event.buttons() & Qt.LeftButton and \
1006                self.__possibleSelectionHandler:
1007            # Set the RectangleSelection (initialized in mousePressEvent)
1008            # on the scene
1009            handler = self.__possibleSelectionHandler
[11236]1010            self._setUserInteractionHandler(handler)
[11198]1011            return handler.mouseMoveEvent(event)
1012
[11151]1013        return False
1014
1015    def sceneMouseReleaseEvent(self, event):
1016        if event.button() == Qt.LeftButton and self.__possibleMouseItemsMove:
1017            self.__possibleMouseItemsMove = False
1018            self.__scene.node_item_position_changed.disconnect(
1019                self.__onNodePositionChanged
1020            )
1021            self.__annotationGeomChanged.mapped[QObject].disconnect(
1022                self.__onAnnotationGeometryChanged
1023            )
1024
[11236]1025            set_enabled_all(self.__disruptiveActions, True)
1026
[11151]1027            if self.__itemsMoving:
1028                self.__scene.mouseReleaseEvent(event)
1029                stack = self.undoStack()
1030                stack.beginMacro(self.tr("Move"))
1031                for scheme_item, (old, new) in self.__itemsMoving.items():
[11390]1032                    if isinstance(scheme_item, SchemeNode):
[11151]1033                        command = commands.MoveNodeCommand(
1034                            self.scheme(), scheme_item, old, new
1035                        )
[11390]1036                    elif isinstance(scheme_item, BaseSchemeAnnotation):
[11151]1037                        command = commands.AnnotationGeometryChange(
1038                            self.scheme(), scheme_item, old, new
1039                        )
1040                    else:
1041                        continue
1042
1043                    stack.push(command)
1044                stack.endMacro()
1045
1046                self.__itemsMoving.clear()
1047                return True
[11198]1048
1049        if self.__emptyClickButtons & Qt.LeftButton and \
1050                event.button() & Qt.LeftButton:
1051            self.__emptyClickButtons &= ~Qt.LeftButton
1052
1053            if self.__quickMenuTriggers & SchemeEditWidget.Clicked and \
1054                    mouse_drag_distance(event, Qt.LeftButton) < 1:
1055                action = interactions.NewNodeAction(self)
[11236]1056
1057                with nested(disabled(self.__undoAction),
1058                            disabled(self.__redoAction)):
1059                    action.create_new(event.screenPos())
1060
[11198]1061                event.accept()
1062                return True
1063
[11151]1064        return False
[11149]1065
1066    def sceneMouseDoubleClickEvent(self, event):
[11151]1067        scene = self.__scene
1068        if scene.user_interaction_handler:
[11149]1069            return False
1070
1071        item = scene.item_at(event.scenePos())
[11198]1072        if not item and self.__quickMenuTriggers & \
1073                SchemeEditWidget.DoubleClicked:
[11149]1074            # Double click on an empty spot
[11236]1075            # Create a new node using QuickMenu
[11151]1076            action = interactions.NewNodeAction(self)
[11236]1077
1078            with nested(disabled(self.__undoAction),
1079                        disabled(self.__redoAction)):
1080                action.create_new(event.screenPos())
1081
[11149]1082            event.accept()
1083            return True
[11151]1084
[11241]1085        item = scene.item_at(event.scenePos(), items.LinkItem,
1086                             buttons=Qt.LeftButton)
1087
1088        if item is not None and event.button() == Qt.LeftButton:
[11154]1089            link = self.scene().link_for_item(item)
1090            action = interactions.EditNodeLinksAction(self, link.source_node,
1091                                                      link.sink_node)
1092            action.edit_links()
1093            event.accept()
1094            return True
1095
[11149]1096        return False
[11151]1097
1098    def sceneKeyPressEvent(self, event):
[11198]1099        scene = self.__scene
1100        if scene.user_interaction_handler:
1101            return False
1102
1103        # If a QGraphicsItem is in text editing mode, don't interrupt it
1104        focusItem = scene.focusItem()
1105        if focusItem and isinstance(focusItem, QGraphicsTextItem) and \
1106                focusItem.textInteractionFlags() & Qt.TextEditable:
1107            return False
1108
1109        # If the mouse is not over out view
1110        if not self.view().underMouse():
1111            return False
1112
[11208]1113        handler = None
[11198]1114        if (event.key() == Qt.Key_Space and \
1115                self.__quickMenuTriggers & SchemeEditWidget.SpaceKey):
[11208]1116            handler = interactions.NewNodeAction(self)
[11198]1117
[11208]1118        elif len(event.text()) and \
[11198]1119                self.__quickMenuTriggers & SchemeEditWidget.AnyKey:
[11208]1120            handler = interactions.NewNodeAction(self)
[11198]1121            # TODO: set the search text to event.text() and set focus on the
1122            # search line
[11208]1123
1124        if handler is not None:
[11236]1125            # Control + Backspace (remove widget action on Mac OSX) conflicts
1126            # with the 'Clear text' action in the search widget (there might
1127            # be selected items in the canvas), so we disable the
[11208]1128            # remove widget action so the text editing follows standard
1129            # 'look and feel'
[11236]1130            with nested(disabled(self.__removeSelectedAction),
1131                        disabled(self.__undoAction),
1132                        disabled(self.__redoAction)):
[11208]1133                handler.create_new(QCursor.pos())
1134
[11198]1135            event.accept()
1136            return True
1137
[11151]1138        return False
1139
1140    def sceneKeyReleaseEvent(self, event):
1141        return False
1142
1143    def sceneContextMenuEvent(self, event):
1144        return False
1145
[11236]1146    def _setUserInteractionHandler(self, handler):
[11371]1147        """
1148        Helper method for setting the user interaction handlers.
[11236]1149        """
1150        if self.__scene.user_interaction_handler:
[11244]1151            if isinstance(self.__scene.user_interaction_handler,
1152                          (interactions.ResizeArrowAnnotation,
1153                           interactions.ResizeTextAnnotation)):
1154                self.__scene.user_interaction_handler.commit()
1155
[11236]1156            self.__scene.user_interaction_handler.ended.disconnect(
1157                self.__onInteractionEnded
1158            )
[11244]1159
[11236]1160        if handler:
1161            handler.ended.connect(self.__onInteractionEnded)
1162            # Disable actions which could change the model
1163            set_enabled_all(self.__disruptiveActions, False)
1164
1165        self.__scene.set_user_interaction_handler(handler)
1166
1167    def __onInteractionEnded(self):
1168        self.sender().ended.disconnect(self.__onInteractionEnded)
1169        set_enabled_all(self.__disruptiveActions, True)
1170
[11151]1171    def __onSelectionChanged(self):
[11221]1172        nodes = self.selectedNodes()
1173        annotations = self.selectedAnnotations()
[11208]1174
[11221]1175        self.__openSelectedAction.setEnabled(bool(nodes))
1176        self.__removeSelectedAction.setEnabled(
1177            bool(nodes) or bool(annotations)
1178        )
[11208]1179
[11221]1180        self.__helpAction.setEnabled(len(nodes) == 1)
1181        self.__renameAction.setEnabled(len(nodes) == 1)
1182
1183        if len(nodes) > 1:
1184            self.__openSelectedAction.setText(self.tr("Open All"))
1185        else:
1186            self.__openSelectedAction.setText(self.tr("Open"))
1187
1188        if len(nodes) + len(annotations) > 1:
1189            self.__removeSelectedAction.setText(self.tr("Remove All"))
1190        else:
1191            self.__removeSelectedAction.setText(self.tr("Remove"))
1192
1193        if len(nodes) == 0:
[11208]1194            self.__openSelectedAction.setText(self.tr("Open"))
1195            self.__removeSelectedAction.setText(self.tr("Remove"))
1196
[11266]1197        focus = self.focusNode()
1198        if focus is not None:
1199            desc = focus.description
[11341]1200            tip = whats_this_helper(desc, include_more_link=True)
[11243]1201        else:
1202            tip = ""
1203
1204        if tip != self.__quickTip:
1205            self.__quickTip = tip
1206            ev = QuickHelpTipEvent("", self.__quickTip,
1207                                   priority=QuickHelpTipEvent.Permanent)
1208
1209            QCoreApplication.sendEvent(self, ev)
1210
[11165]1211    def __onNodeAdded(self, node):
1212        widget = self.__scheme.widget_for_node[node]
1213        widget.widgetStateChanged.connect(self.__onWidgetStateChanged)
1214
1215    def __onNodeRemoved(self, node):
1216        widget = self.__scheme.widget_for_node[node]
1217        widget.widgetStateChanged.disconnect(self.__onWidgetStateChanged)
1218
1219    def __onWidgetStateChanged(self, *args):
1220        widget = self.sender()
1221        self.scheme()
1222        widget_to_node = dict(reversed(item) for item in \
1223                              self.__scheme.widget_for_node.items())
1224        node = widget_to_node[widget]
1225        item = self.__scene.item_for_node(node)
1226
1227        info = widget.widgetStateToHtml(True, False, False)
1228        warning = widget.widgetStateToHtml(False, True, False)
1229        error = widget.widgetStateToHtml(False, False, True)
1230
1231        item.setInfoMessage(info or None)
1232        item.setWarningMessage(warning or None)
1233        item.setErrorMessage(error or None)
1234
[11151]1235    def __onNodeActivate(self, item):
1236        node = self.__scene.node_for_item(item)
1237        widget = self.scheme().widget_for_node[node]
1238        widget.show()
1239        widget.raise_()
1240
1241    def __onNodePositionChanged(self, item, pos):
1242        node = self.__scene.node_for_item(item)
1243        new = (pos.x(), pos.y())
1244        if node not in self.__itemsMoving:
1245            self.__itemsMoving[node] = (node.position, new)
1246        else:
1247            old, _ = self.__itemsMoving[node]
1248            self.__itemsMoving[node] = (old, new)
1249
1250    def __onAnnotationGeometryChanged(self, item):
1251        annot = self.scene().annotation_for_item(item)
1252        if annot not in self.__itemsMoving:
1253            self.__itemsMoving[annot] = (annot.geometry,
1254                                         geometry_from_annotation_item(item))
1255        else:
1256            old, _ = self.__itemsMoving[annot]
1257            self.__itemsMoving[annot] = (old,
1258                                         geometry_from_annotation_item(item))
1259
1260    def __onAnnotationAdded(self, item):
[11192]1261        log.debug("Annotation added (%r)", item)
[11151]1262        item.setFlag(QGraphicsItem.ItemIsSelectable)
[11192]1263        item.setFlag(QGraphicsItem.ItemIsMovable)
1264        item.setFlag(QGraphicsItem.ItemIsFocusable)
1265
1266        item.installSceneEventFilter(self.__focusListener)
1267
[11151]1268        if isinstance(item, items.ArrowAnnotation):
1269            pass
1270        elif isinstance(item, items.TextAnnotation):
[11192]1271            # Make the annotation editable.
1272            item.setTextInteractionFlags(Qt.TextEditorInteraction)
1273
[11151]1274            self.__editFinishedMapper.setMapping(item, item)
1275            item.editingFinished.connect(
1276                self.__editFinishedMapper.map
1277            )
[11192]1278
[11151]1279        self.__annotationGeomChanged.setMapping(item, item)
1280        item.geometryChanged.connect(
1281            self.__annotationGeomChanged.map
1282        )
1283
1284    def __onAnnotationRemoved(self, item):
[11192]1285        log.debug("Annotation removed (%r)", item)
[11151]1286        if isinstance(item, items.ArrowAnnotation):
1287            pass
1288        elif isinstance(item, items.TextAnnotation):
1289            item.editingFinished.disconnect(
1290                self.__editFinishedMapper.map
1291            )
[11192]1292
1293        item.removeSceneEventFilter(self.__focusListener)
1294
[11151]1295        self.__annotationGeomChanged.removeMappings(item)
1296        item.geometryChanged.disconnect(
1297            self.__annotationGeomChanged.map
1298        )
1299
1300    def __onItemFocusedIn(self, item):
[11371]1301        """
1302        Annotation item has gained focus.
[11192]1303        """
1304        if not self.__scene.user_interaction_handler:
1305            self.__startControlPointEdit(item)
[11151]1306
1307    def __onItemFocusedOut(self, item):
[11371]1308        """
1309        Annotation item lost focus.
[11192]1310        """
1311        self.__endControlPointEdit()
[11151]1312
1313    def __onEditingFinished(self, item):
[11371]1314        """
1315        Text annotation editing has finished.
[11192]1316        """
[11151]1317        annot = self.__scene.annotation_for_item(item)
1318        text = unicode(item.toPlainText())
1319        if annot.text != text:
1320            self.__undoStack.push(
1321                commands.TextChangeCommand(self.scheme(), annot,
1322                                           annot.text, text)
1323            )
1324
[11194]1325    def __toggleNewArrowAnnotation(self, checked):
1326        if self.__newTextAnnotationAction.isChecked():
[11371]1327            # Uncheck the text annotation action if needed.
[11194]1328            self.__newTextAnnotationAction.setChecked(not checked)
[11195]1329
1330        action = self.__newArrowAnnotationAction
1331
[11194]1332        if not checked:
[11371]1333            # The action was unchecked (canceled by the user)
[11194]1334            handler = self.__scene.user_interaction_handler
1335            if isinstance(handler, interactions.NewArrowAnnotation):
1336                # Cancel the interaction and restore the state
[11195]1337                handler.ended.disconnect(action.toggle)
1338                handler.cancel(interactions.UserInteraction.UserCancelReason)
[11194]1339                log.info("Canceled new arrow annotation")
1340
1341        else:
1342            handler = interactions.NewArrowAnnotation(self)
[11201]1343            checked = self.__arrowColorActionGroup.checkedAction()
1344            handler.setColor(checked.data().toPyObject())
1345
[11195]1346            handler.ended.connect(action.toggle)
1347
[11236]1348            self._setUserInteractionHandler(handler)
[11194]1349
[11201]1350    def __onFontSizeTriggered(self, action):
1351        if not self.__newTextAnnotationAction.isChecked():
[11371]1352            # When selecting from the (font size) menu the 'Text'
1353            # action does not get triggered automatically.
[11201]1354            self.__newTextAnnotationAction.trigger()
1355        else:
[11371]1356            # Update the preferred font on the interaction handler.
[11201]1357            handler = self.__scene.user_interaction_handler
1358            if isinstance(handler, interactions.NewTextAnnotation):
1359                handler.setFont(action.font())
1360
[11194]1361    def __toggleNewTextAnnotation(self, checked):
1362        if self.__newArrowAnnotationAction.isChecked():
[11371]1363            # Uncheck the arrow annotation if needed.
[11194]1364            self.__newArrowAnnotationAction.setChecked(not checked)
1365
[11195]1366        action = self.__newTextAnnotationAction
1367
[11194]1368        if not checked:
[11371]1369            # The action was unchecked (canceled by the user)
[11194]1370            handler = self.__scene.user_interaction_handler
1371            if isinstance(handler, interactions.NewTextAnnotation):
1372                # cancel the interaction and restore the state
[11195]1373                handler.ended.disconnect(action.toggle)
1374                handler.cancel(interactions.UserInteraction.UserCancelReason)
[11194]1375                log.info("Canceled new text annotation")
1376
1377        else:
1378            handler = interactions.NewTextAnnotation(self)
[11201]1379            checked = self.__fontActionGroup.checkedAction()
1380            handler.setFont(checked.font())
1381
[11195]1382            handler.ended.connect(action.toggle)
1383
[11236]1384            self._setUserInteractionHandler(handler)
[11194]1385
[11201]1386    def __onArrowColorTriggered(self, action):
1387        if not self.__newArrowAnnotationAction.isChecked():
[11371]1388            # When selecting from the (color) menu the 'Arrow'
1389            # action does not get triggered automatically.
[11201]1390            self.__newArrowAnnotationAction.trigger()
1391        else:
[11371]1392            # Update the preferred color on the interaction handler
[11201]1393            handler = self.__scene.user_interaction_handler
1394            if isinstance(handler, interactions.NewArrowAnnotation):
1395                handler.setColor(action.data().toPyObject())
1396
[11154]1397    def __onCustomContextMenuRequested(self, pos):
1398        scenePos = self.view().mapToScene(pos)
1399        globalPos = self.view().mapToGlobal(pos)
1400
1401        item = self.scene().item_at(scenePos, items.NodeItem)
1402        if item is not None:
[11208]1403            self.__widgetMenu.popup(globalPos)
[11154]1404            return
1405
[11241]1406        item = self.scene().item_at(scenePos, items.LinkItem,
1407                                    buttons=Qt.RightButton)
[11154]1408        if item is not None:
1409            link = self.scene().link_for_item(item)
1410            self.__linkEnableAction.setChecked(link.enabled)
1411            self.__contextMenuTarget = link
1412            self.__linkMenu.popup(globalPos)
1413            return
1414
[11208]1415    def __onRenameAction(self):
[11371]1416        """
1417        Rename was requested for the selected widget.
1418        """
[11208]1419        selected = self.selectedNodes()
1420        if len(selected) == 1:
1421            self.editNodeTitle(selected[0])
1422
1423    def __onHelpAction(self):
[11371]1424        """
1425        Help was requested for the selected widget.
[11264]1426        """
[11208]1427        nodes = self.selectedNodes()
1428        help_url = None
1429        if len(nodes) == 1:
1430            node = nodes[0]
1431            desc = node.description
1432
[11264]1433            help_url = "help://search?" + urlencode({"id": desc.id})
[11296]1434            self.__showHelpFor(help_url)
[11264]1435
[11296]1436    def __showHelpFor(self, help_url):
1437        """
1438        Show help for an "help" url.
1439        """
1440        # Notify the parent chain and let them respond
1441        ev = QWhatsThisClickedEvent(help_url)
1442        handled = QCoreApplication.sendEvent(self, ev)
[11264]1443
[11296]1444        if not handled:
1445            message_information(
1446                self.tr("Sorry there is no documentation available for "
1447                        "this widget."),
1448                parent=self)
[11208]1449
[11194]1450    def __toggleLinkEnabled(self, enabled):
[11371]1451        """
1452        Link 'enabled' state was toggled in the context menu.
[11192]1453        """
[11154]1454        if self.__contextMenuTarget:
1455            link = self.__contextMenuTarget
1456            command = commands.SetAttrCommand(
1457                link, "enabled", enabled, name=self.tr("Set enabled"),
1458            )
1459            self.__undoStack.push(command)
1460
1461    def __linkRemove(self):
[11371]1462        """
1463        Remove link was requested from the context menu.
[11192]1464        """
[11154]1465        if self.__contextMenuTarget:
1466            self.removeLink(self.__contextMenuTarget)
1467
1468    def __linkReset(self):
[11371]1469        """
1470        Link reset from the context menu was requested.
[11192]1471        """
[11154]1472        if self.__contextMenuTarget:
1473            link = self.__contextMenuTarget
1474            action = interactions.EditNodeLinksAction(
1475                self, link.source_node, link.sink_node
1476            )
1477            action.edit_links()
1478
[11192]1479    def __startControlPointEdit(self, item):
[11371]1480        """
1481        Start a control point edit interaction for `item`.
[11192]1482        """
1483        if isinstance(item, items.ArrowAnnotation):
1484            handler = interactions.ResizeArrowAnnotation(self)
1485        elif isinstance(item, items.TextAnnotation):
1486            handler = interactions.ResizeTextAnnotation(self)
1487        else:
1488            log.warning("Unknown annotation item type %r" % item)
1489            return
1490
1491        handler.editItem(item)
[11236]1492        self._setUserInteractionHandler(handler)
[11192]1493
1494        log.info("Control point editing started (%r)." % item)
1495
1496    def __endControlPointEdit(self):
[11371]1497        """
1498        End the current control point edit interaction.
[11192]1499        """
1500        handler = self.__scene.user_interaction_handler
1501        if isinstance(handler, (interactions.ResizeArrowAnnotation,
[11195]1502                                interactions.ResizeTextAnnotation)) and \
1503                not handler.isFinished() and not handler.isCanceled():
[11192]1504            handler.commit()
1505            handler.end()
1506
1507            log.info("Control point editing finished.")
1508
[11343]1509    def __updateFont(self):
[11371]1510        """
1511        Update the font for the "Text size' menu and the default font
1512        used in the `CanvasScene`.
1513
1514        """
[11343]1515        actions = self.__fontActionGroup.actions()
1516        font = self.font()
1517        for action in actions:
1518            size = action.font().pixelSize()
1519            action_font = QFont(font)
1520            action_font.setPixelSize(size)
1521            action.setFont(action_font)
1522
1523        if self.__scene:
1524            self.__scene.setFont(font)
1525
[11151]1526
1527def geometry_from_annotation_item(item):
1528    if isinstance(item, items.ArrowAnnotation):
1529        line = item.line()
1530        p1 = item.mapToScene(line.p1())
1531        p2 = item.mapToScene(line.p2())
1532        return ((p1.x(), p1.y()), (p2.x(), p2.y()))
1533    elif isinstance(item, items.TextAnnotation):
1534        geom = item.geometry()
1535        return (geom.x(), geom.y(), geom.width(), geom.height())
[11198]1536
1537
1538def mouse_drag_distance(event, button=Qt.LeftButton):
[11371]1539    """
1540    Return the (manhattan) distance between the mouse position
[11198]1541    when the `button` was pressed and the current mouse position.
1542
1543    """
1544    diff = (event.buttonDownScreenPos(button) - event.screenPos())
1545    return diff.manhattanLength()
[11236]1546
1547
1548def set_enabled_all(objects, enable):
[11371]1549    """
1550    Set `enabled` properties on all objects (objects with `setEnabled` method).
[11236]1551    """
1552    for obj in objects:
1553        obj.setEnabled(enable)
Note: See TracBrowser for help on using the repository browser.