def __init__(self, parent=None):
super(NodeViewer, self).__init__(parent)
if parent is not None:
self.setWindowFlags(QtCore.Qt.Window)
scene_pos = (SCENE_AREA / 2) * -1
self.setScene(NodeScene(self))
self.setSceneRect(scene_pos, scene_pos, SCENE_AREA, SCENE_AREA)
self.setRenderHint(QtGui.QPainter.Antialiasing, True)
self.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
self.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
self.setViewportUpdateMode(QtWidgets.QGraphicsView.FullViewportUpdate)
self.setAcceptDrops(True)
self.resize(1000, 800)
self._pipe_layout = PIPE_LAYOUT_CURVED
self._detached_port = None
self._start_port = None
self._origin_pos = None
self._previous_pos = QtCore.QPoint(self.width(), self.height())
self._prev_selection = []
self._node_positions = {}
self._rubber_band = QtWidgets.QRubberBand(
QtWidgets.QRubberBand.Rectangle, self)
self._LIVE_PIPE = LivePipe()
self._LIVE_PIPE.setVisible(False)
self.scene().addItem(self._LIVE_PIPE)
self._SLICER_PIPE = SlicerPipe()
self._SLICER_PIPE.setVisible(False)
self.scene().addItem(self._SLICER_PIPE)
self._undo_stack = QtWidgets.QUndoStack(self)
self._context_menu = QtWidgets.QMenu('main', self)
self._context_menu.setStyleSheet(STYLE_QMENU)
self._search_widget = TabSearchWidget(self)
self._search_widget.search_submitted.connect(self._on_search_submitted)
# workaround fix for shortcuts from the non-native menu actions
# don't seem to trigger so we create a hidden menu bar.
menu_bar = QtWidgets.QMenuBar(self)
menu_bar.setNativeMenuBar(False)
# shortcuts don't work with "setVisibility(False)".
menu_bar.resize(0, 0)
menu_bar.addMenu(self._context_menu)
self.acyclic = True
self.LMB_state = False
self.RMB_state = False
self.MMB_state = False
self.ALT_state = False
self.CTRL_state = False
self.SHIFT_state = False
def __init__(self, parent=None):
super(NodeViewer, self).__init__(parent)
self.setScene(NodeScene(self))
self.setRenderHint(QtGui.QPainter.Antialiasing, True)
self.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
self.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
self.setViewportUpdateMode(QtWidgets.QGraphicsView.FullViewportUpdate)
self.setCacheMode(QtWidgets.QGraphicsView.CacheBackground)
self.setOptimizationFlag(
QtWidgets.QGraphicsView.DontAdjustForAntialiasing)
self.setAcceptDrops(True)
self.resize(850, 800)
self._scene_range = QtCore.QRectF(0, 0,
self.size().width(),
self.size().height())
self._update_scene()
self._last_size = self.size()
self._pipe_layout = PIPE_LAYOUT_CURVED
self._detached_port = None
self._start_port = None
self._origin_pos = None
self._previous_pos = QtCore.QPoint(self.width(), self.height())
self._prev_selection_nodes = []
self._prev_selection_pipes = []
self._node_positions = {}
self._rubber_band = QtWidgets.QRubberBand(
QtWidgets.QRubberBand.Rectangle, self)
self._LIVE_PIPE = LivePipe()
self._LIVE_PIPE.setVisible(False)
self.scene().addItem(self._LIVE_PIPE)
self._SLICER_PIPE = SlicerPipe()
self._SLICER_PIPE.setVisible(False)
self.scene().addItem(self._SLICER_PIPE)
self._undo_stack = QtWidgets.QUndoStack(self)
self._search_widget = TabSearchMenuWidget(self)
self._search_widget.search_submitted.connect(self._on_search_submitted)
# workaround fix for shortcuts from the non-native menu actions
# don't seem to trigger so we create a hidden menu bar.
menu_bar = QtWidgets.QMenuBar(self)
menu_bar.setNativeMenuBar(False)
# shortcuts don't work with "setVisibility(False)".
menu_bar.setMaximumWidth(0)
self._ctx_menu = BaseMenu('NodeGraph', self)
self._ctx_node_menu = BaseMenu('Nodes', self)
menu_bar.addMenu(self._ctx_menu)
menu_bar.addMenu(self._ctx_node_menu)
self._ctx_node_menu.setDisabled(True)
self.acyclic = True
self.LMB_state = False
self.RMB_state = False
self.MMB_state = False
self.ALT_state = False
self.CTRL_state = False
self.SHIFT_state = False
self.COLLIDING_state = False
def setupMenuBar(self, graph: NodeGraph):
rootMenu = graph.context_menu()
fileMenu = rootMenu.add_menu('&File')
editMenu = rootMenu.add_menu('&Edit')
# create "File" menu.
fileMenu.add_command('Open Graph...',
lambda: actions._open_session(graph),
QtGui.QKeySequence.Open)
fileMenu.add_command('Export Graph As...',
lambda: actions._save_session_as(graph),
'Alt+Shift+s')
fileMenu.add_command('Clear', lambda: actions._clear_session(graph))
fileMenu.add_separator()
fileMenu.add_command('Zoom In', lambda: actions._zoom_in(graph), '=')
fileMenu.add_command('Zoom Out', lambda: actions._zoom_out(graph), '-')
fileMenu.add_command('Reset Zoom', graph.reset_zoom, 'h')
# create "Edit" menu.
undo_actn = graph.undo_stack().createUndoAction(
graph.viewer(), '&Undo')
if LooseVersion(QtCore.qVersion()) >= LooseVersion('5.10'):
undo_actn.setShortcutVisibleInContextMenu(True)
undo_actn.setShortcuts(QtGui.QKeySequence.Undo)
editMenu.qmenu.addAction(undo_actn)
redo_actn = graph.undo_stack().createRedoAction(
graph.viewer(), '&Redo')
if LooseVersion(QtCore.qVersion()) >= LooseVersion('5.10'):
redo_actn.setShortcutVisibleInContextMenu(True)
redo_actn.setShortcuts(QtGui.QKeySequence.Redo)
editMenu.qmenu.addAction(redo_actn)
editMenu.add_separator()
editMenu.add_command('Clear Undo History',
lambda: actions._clear_undo(graph))
editMenu.add_separator()
editMenu.add_command('Copy', graph.copy_nodes, QtGui.QKeySequence.Copy)
editMenu.add_command('Paste', graph.paste_nodes,
QtGui.QKeySequence.Paste)
editMenu.add_command(
'Delete', lambda: graph.delete_nodes(graph.selected_nodes()),
QtGui.QKeySequence.Delete)
editMenu.add_separator()
editMenu.add_command('Select all', graph.select_all, 'Ctrl+A')
editMenu.add_command('Deselect all', graph.clear_selection,
'Ctrl+Shift+A')
editMenu.add_command(
'Enable/Disable',
lambda: graph.disable_nodes(graph.selected_nodes()), 'd')
editMenu.add_command(
'Duplicate', lambda: graph.duplicate_nodes(graph.selected_nodes()),
'Alt+c')
editMenu.add_command('Center Selection', graph.fit_to_selection, 'f')
editMenu.add_separator()
menuBar = QtWidgets.QMenuBar()
sessionMenu = QtWidgets.QMenu("Session")
menuBar.addMenu(fileMenu.qmenu)
menuBar.addMenu(editMenu.qmenu)
menuBar.addMenu(sessionMenu)
self.saveAction = QtWidgets.QAction("Save")
self.saveAction.setShortcut(QtGui.QKeySequence.Save)
self.saveAction.triggered.connect(self.onSave)
self.saveAsAction = QtWidgets.QAction("Save As...")
self.saveAsAction.setShortcut("Ctrl+Shift+S")
self.saveAsAction.triggered.connect(self.onSaveAs)
self.loadAction = QtWidgets.QAction("Load")
self.loadAction.triggered.connect(self.onLoad)
self.runAction = QtWidgets.QAction("Run")
self.runAction.triggered.connect(self.onRun)
self.runAction.setShortcut(QtGui.QKeySequence("R"))
self.openInCode = QtWidgets.QAction("Show Code In Visual Studio Code")
self.openInCode.triggered.connect(self.onOpenInVisualStudioCode)
self.openInCode.setShortcut(QtGui.QKeySequence("Q"))
sessionMenu.addAction(self.saveAction)
sessionMenu.addAction(self.saveAsAction)
sessionMenu.addAction(self.loadAction)
sessionMenu.addAction(self.runAction)
sessionMenu.addAction(self.openInCode)
self.sessionMenu = sessionMenu
self.viewMenu = QtWidgets.QMenu("View")
menuBar.addMenu(self.viewMenu)
self.window.verticalLayout.insertWidget(0, menuBar)
