class EditLinksDialog(QDialog):
"""
A dialog for editing links.
>>> dlg = EditLinksDialog()
>>> dlg.setNodes(source_node, sink_node)
>>> dlg.setLinks([(source_node.output_channel("Data"),
... sink_node.input_channel("Data"))])
>>> if dlg.exec_() == EditLinksDialog.Accepted:
... new_links = dlg.links()
...
"""
def __init__(self, parent=None, **kwargs):
# type: (Optional[QWidget], Any) -> None
super().__init__(parent, **kwargs)
self.setModal(True)
self.__setupUi()
def __setupUi(self):
layout = QVBoxLayout()
# Scene with the link editor.
self.scene = LinksEditScene()
self.view = QGraphicsView(self.scene)
self.view.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
self.view.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
self.view.setRenderHint(QPainter.Antialiasing)
self.scene.editWidget.geometryChanged.connect(self.__onGeometryChanged)
# Ok/Cancel/Clear All buttons.
buttons = QDialogButtonBox(QDialogButtonBox.Ok |
QDialogButtonBox.Cancel |
QDialogButtonBox.Reset,
Qt.Horizontal)
clear_button = buttons.button(QDialogButtonBox.Reset)
clear_button.setText(self.tr("Clear All"))
buttons.accepted.connect(self.accept)
buttons.rejected.connect(self.reject)
clear_button.clicked.connect(self.scene.editWidget.clearLinks)
layout.addWidget(self.view)
layout.addWidget(buttons)
self.setLayout(layout)
layout.setSizeConstraint(QVBoxLayout.SetFixedSize)
self.setSizeGripEnabled(False)
def setNodes(self, source_node, sink_node):
# type: (SchemeNode, SchemeNode) -> None
"""
Set the source/sink nodes (:class:`.SchemeNode` instances)
between which to edit the links.
.. note:: This should be called before :func:`setLinks`.
"""
self.scene.editWidget.setNodes(source_node, sink_node)
def setLinks(self, links):
# type: (List[IOPair]) -> None
"""
Set a list of links to display between the source and sink
nodes. The `links` is a list of (`OutputSignal`, `InputSignal`)
tuples where the first element is an output signal of the source
node and the second an input signal of the sink node.
"""
self.scene.editWidget.setLinks(links)
def links(self):
# type: () -> List[IOPair]
"""
Return the links between the source and sink node.
"""
return self.scene.editWidget.links()
def __onGeometryChanged(self):
size = self.scene.editWidget.size()
left, top, right, bottom = self.getContentsMargins()
self.view.setFixedSize(size.toSize() + \
QSize(left + right + 4, top + bottom + 4))
self.view.setSceneRect(self.scene.editWidget.geometry())
class EditLinksDialog(QDialog):
"""
A dialog for editing links.
>>> dlg = EditLinksDialog()
>>> dlg.setNodes(file_node, test_learners_node)
>>> dlg.setLinks([(file_node.output_channel("Data"),
... (test_learners_node.input_channel("Data")])
>>> if dlg.exec_() == EditLinksDialog.Accpeted:
... new_links = dlg.links()
...
"""
def __init__(self, *args, **kwargs):
QDialog.__init__(self, *args, **kwargs)
self.setModal(True)
self.__setupUi()
def __setupUi(self):
layout = QVBoxLayout()
# Scene with the link editor.
self.scene = LinksEditScene()
self.view = QGraphicsView(self.scene)
self.view.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
self.view.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
self.view.setRenderHint(QPainter.Antialiasing)
self.scene.editWidget.geometryChanged.connect(self.__onGeometryChanged)
# Ok/Cancel/Clear All buttons.
buttons = QDialogButtonBox(QDialogButtonBox.Ok |
QDialogButtonBox.Cancel |
QDialogButtonBox.Reset,
Qt.Horizontal)
clear_button = buttons.button(QDialogButtonBox.Reset)
clear_button.setText(self.tr("Clear All"))
buttons.accepted.connect(self.accept)
buttons.rejected.connect(self.reject)
clear_button.clicked.connect(self.scene.editWidget.clearLinks)
layout.addWidget(self.view)
layout.addWidget(buttons)
self.setLayout(layout)
layout.setSizeConstraint(QVBoxLayout.SetFixedSize)
self.setSizeGripEnabled(False)
def setNodes(self, source_node, sink_node):
"""
Set the source/sink nodes (:class:`.SchemeNode` instances)
between which to edit the links.
.. note:: This should be called before :func:`setLinks`.
"""
self.scene.editWidget.setNodes(source_node, sink_node)
def setLinks(self, links):
"""
Set a list of links to display between the source and sink
nodes. The `links` is a list of (`OutputSignal`, `InputSignal`)
tuples where the first element is an output signal of the source
node and the second an input signal of the sink node.
"""
self.scene.editWidget.setLinks(links)
def links(self):
"""
Return the links between the source and sink node.
"""
return self.scene.editWidget.links()
def __onGeometryChanged(self):
size = self.scene.editWidget.size()
left, top, right, bottom = self.getContentsMargins()
self.view.setFixedSize(size.toSize() + \
QSize(left + right + 4, top + bottom + 4))
