class VisualizationPanel(Panel):
""" Switch visualizations and adjust their settings """
def __init__(self,
name,
default_position='right',
parent=None,
folded=False):
"""
All of the panel constructors follow the same format so that the construction can be automated.
:param name: Title of the panel and the key for accessing it
:param default_position: 'bottom', 'right'...
:param parent: self.main
"""
Panel.__init__(self, name, default_position, parent, folded)
inner = QtWidgets.QWidget()
inner.preferred_size = QtCore.QSize(200, 70)
inner.setMaximumWidth(220)
inner.setMaximumHeight(80)
inner.sizeHint = self.sizeHint
layout = QtWidgets.QVBoxLayout()
hlayout = QtWidgets.QHBoxLayout()
self.selector = SelectionBox(self)
self.selector.add_items([('%s (%s)' % (key, item.shortcut), key)
for key, item in VISUALIZATIONS.items()])
self.ui_manager.connect_element_to_action(self.selector,
'set_visualization')
hlayout.addWidget(self.selector)
self.toggle_options = PanelButton(pixmap=qt_prefs.settings_pixmap,
parent=self,
size=20)
self.toggle_options.setFixedSize(26, 26)
self.toggle_options.setCheckable(True)
ctrl.ui.connect_element_to_action(
self.toggle_options, 'toggle_panel_VisualizationOptionsPanel')
hlayout.addWidget(self.toggle_options, 1, QtCore.Qt.AlignRight)
layout.addLayout(hlayout)
hlayout = QtWidgets.QHBoxLayout()
w = 36
b1 = PanelButton(pixmap=qt_prefs.shape_icon_plain,
parent=self,
size=24)
b1.setFixedWidth(w)
b1.setCheckable(True)
ctrl.ui.connect_element_to_action(b1, 'set_no_frame_node_shape')
hlayout.addWidget(b1)
b2 = PanelButton(pixmap=qt_prefs.shape_icon_scope,
parent=self,
size=24)
b2.setFixedWidth(w)
b2.setCheckable(True)
ctrl.ui.connect_element_to_action(b2, 'set_scopebox_node_shape')
hlayout.addWidget(b2)
b3 = PanelButton(pixmap=qt_prefs.shape_icon_brackets,
parent=self,
size=24)
b3.setFixedWidth(w)
b3.setCheckable(True)
ctrl.ui.connect_element_to_action(b3, 'set_bracketed_node_shape')
hlayout.addWidget(b3)
b4 = PanelButton(pixmap=qt_prefs.shape_icon_box, parent=self, size=24)
b4.setFixedWidth(w)
b4.setCheckable(True)
ctrl.ui.connect_element_to_action(b4, 'set_box_node_shape')
hlayout.addWidget(b4)
b5 = PanelButton(pixmap=qt_prefs.shape_icon_card, parent=self, size=24)
b5.setFixedWidth(w)
b5.setCheckable(True)
ctrl.ui.connect_element_to_action(b5, 'set_card_node_shape')
hlayout.addWidget(b5)
layout.addLayout(hlayout)
#layout.setContentsMargins(0, 0, 0, 0)
inner.setLayout(layout)
self.watchlist = ['visualization']
self.preferred_size = inner.preferred_size
self.setWidget(inner)
inner.setAutoFillBackground(True)
self.finish_init()
def showEvent(self, event):
""" Panel may have missed signals to update its contents when it was hidden: update all
that signals would update.
:param event:
:return:
"""
# ??
super().showEvent(event)
def watch_alerted(self, obj, signal, field_name, value):
""" Receives alerts from signals that this object has chosen to listen. These signals
are declared in 'self.watchlist'.
This method will try to sort out the received signals and act accordingly.
:param obj: the object causing the alarm
:param signal: identifier for type of the alarm
:param field_name: name of the field of the object causing the alarm
:param value: value given to the field
:return:
"""
if signal == 'visualization':
if value and 'name' in value:
index = list(VISUALIZATIONS.keys()).index(value['name'])
self.selector.setCurrentIndex(index)
def sizeHint(self):
#print("VisualizationPanel asking for sizeHint, ", self.preferred_size)
return self.preferred_size
class VisualizationPanel(Panel):
""" Switch visualizations and adjust their settings """
def __init__(self, name, default_position='right', parent=None, folded=False):
"""
All of the panel constructors follow the same format so that the construction can be automated.
:param name: Title of the panel and the key for accessing it
:param default_position: 'bottom', 'right'...
:param parent: self.main
"""
Panel.__init__(self, name, default_position, parent, folded)
inner = QtWidgets.QWidget()
inner.preferred_size = QtCore.QSize(200, 70)
inner.setMaximumWidth(220)
inner.setMaximumHeight(80)
inner.sizeHint = self.sizeHint
layout = QtWidgets.QVBoxLayout()
hlayout = QtWidgets.QHBoxLayout()
self.selector = SelectionBox(self)
self.selector.add_items([('%s (%s)' % (key, item.shortcut), key) for key, item in
VISUALIZATIONS.items()])
self.ui_manager.connect_element_to_action(self.selector, 'set_visualization')
hlayout.addWidget(self.selector)
self.toggle_options = PanelButton(pixmap=qt_prefs.settings_pixmap,
parent=self, size=20)
self.toggle_options.setFixedSize(26, 26)
self.toggle_options.setCheckable(True)
ctrl.ui.connect_element_to_action(self.toggle_options,
'toggle_panel_VisualizationOptionsPanel')
hlayout.addWidget(self.toggle_options, 1, QtCore.Qt.AlignRight)
layout.addLayout(hlayout)
hlayout = QtWidgets.QHBoxLayout()
w = 36
b1 = PanelButton(pixmap=qt_prefs.shape_icon_plain, parent=self, size=24)
b1.setFixedWidth(w)
b1.setCheckable(True)
ctrl.ui.connect_element_to_action(b1, 'set_no_frame_node_shape')
hlayout.addWidget(b1)
b2 = PanelButton(pixmap=qt_prefs.shape_icon_scope, parent=self, size=24)
b2.setFixedWidth(w)
b2.setCheckable(True)
ctrl.ui.connect_element_to_action(b2, 'set_scopebox_node_shape')
hlayout.addWidget(b2)
b3 = PanelButton(pixmap=qt_prefs.shape_icon_brackets, parent=self, size=24)
b3.setFixedWidth(w)
b3.setCheckable(True)
ctrl.ui.connect_element_to_action(b3, 'set_bracketed_node_shape')
hlayout.addWidget(b3)
b4 = PanelButton(pixmap=qt_prefs.shape_icon_box, parent=self, size=24)
b4.setFixedWidth(w)
b4.setCheckable(True)
ctrl.ui.connect_element_to_action(b4, 'set_box_node_shape')
hlayout.addWidget(b4)
b5 = PanelButton(pixmap=qt_prefs.shape_icon_card, parent=self, size=24)
b5.setFixedWidth(w)
b5.setCheckable(True)
ctrl.ui.connect_element_to_action(b5, 'set_card_node_shape')
hlayout.addWidget(b5)
layout.addLayout(hlayout)
#layout.setContentsMargins(0, 0, 0, 0)
inner.setLayout(layout)
self.watchlist = ['visualization']
self.preferred_size = inner.preferred_size
self.setWidget(inner)
inner.setAutoFillBackground(True)
self.finish_init()
def showEvent(self, event):
""" Panel may have missed signals to update its contents when it was hidden: update all
that signals would update.
:param event:
:return:
"""
# ??
super().showEvent(event)
def watch_alerted(self, obj, signal, field_name, value):
""" Receives alerts from signals that this object has chosen to listen. These signals
are declared in 'self.watchlist'.
This method will try to sort out the received signals and act accordingly.
:param obj: the object causing the alarm
:param signal: identifier for type of the alarm
:param field_name: name of the field of the object causing the alarm
:param value: value given to the field
:return:
"""
if signal == 'visualization':
if value and 'name' in value:
index = list(VISUALIZATIONS.keys()).index(value['name'])
self.selector.setCurrentIndex(index)
def sizeHint(self):
#print("VisualizationPanel asking for sizeHint, ", self.preferred_size)
return self.preferred_size
class NewElementEmbed(UIEmbed):
def __init__(self, parent):
UIEmbed.__init__(self, parent, None, 'Create new node')
self.marker = None
self.guess_mode = True
layout = QtWidgets.QVBoxLayout()
layout.addLayout(self.top_row_layout)
hlayout = box_row(layout)
ui = self.ui_manager
self.new_arrow_button = icon_text_button(ui, hlayout, self, '', '',
" &Arrow", 'new_arrow', size=QtCore.QSize(48, 20),
draw_method=arrow)
self.divider_button = icon_text_button(ui, hlayout, self, '', '',
" &Divider", 'new_divider',
size=QtCore.QSize(48, 20), draw_method=divider)
self.new_arrow_button.setFlat(False)
self.divider_button.setFlat(False)
self.new_arrow_button.hide()
self.divider_button.hide()
tt = 'Text for new node'
smaller_font = qt_prefs.get_font(g.MAIN_FONT)
big_font = QtGui.QFont(smaller_font)
big_font.setPointSize(big_font.pointSize() * 2)
self.input_line_edit = ExpandingLineEdit(self,
tip=tt,
big_font=big_font,
smaller_font=smaller_font,
prefill='label',
on_edit=self.guess_type_for_input)
layout.addWidget(self.input_line_edit)
hlayout = QtWidgets.QHBoxLayout()
self.node_type_selector = SelectionBox(self)
self.node_type_selector.currentIndexChanged.connect(self.changed_node_type)
self.node_types = [('Guess from input', g.GUESS_FROM_INPUT)]
for key in classes.node_types_order:
# we have dedicated buttons for arrows and dividers
#if key not in (g.ARROW, g.DIVIDER):
node_class = classes.nodes.get(key, None)
if (not node_class) or node_class.is_syntactic and not ctrl.free_drawing_mode:
continue
self.node_types.append(('New %s' % node_class.display_name[0].lower(), key))
self.node_types.append(('New arrow', g.ARROW))
#self.node_types.append(('New divider', g.DIVIDER))
self.node_type_selector.add_items(self.node_types)
hlayout.addWidget(self.node_type_selector)
hlayout.addStretch(0)
self.enter_button = QtWidgets.QPushButton("Create ↩") # U+21A9 ↩
ui.connect_element_to_action(self.enter_button, 'create_new_node_from_text')
hlayout.addWidget(self.enter_button)
layout.addLayout(hlayout)
self.setLayout(layout)
self.assumed_width = 200
self.assumed_height = 117
@property
def graphic_item(self):
return self.marker
def mouseMoveEvent(self, event):
self.move(self.mapToParent(event.pos()) - self._drag_diff)
if self.marker:
self.marker.update_position()
self.marker.set_dragged(True)
QtWidgets.QWidget.mouseMoveEvent(self, event)
def guess_type_for_input(self, text):
if not self.guess_mode:
return
key = guess_node_type(text)
if key == 100:
self.top_title.setText('Create new tree')
else:
self.top_title.setText('Create new ' + classes.node_info[key]['name'].lower())
def changed_node_type(self, index=-1):
if index == 0:
self.guess_mode = True
elif index > 0:
title_text = self.node_type_selector.itemText(index)
self.top_title.setText('Create ' + title_text.lower())
self.guess_mode = False
def focus_to_main(self):
self.input_line_edit.setFocus(QtCore.Qt.PopupFocusReason)
def marker_dragged(self):
if self.guess_mode:
self.set_node_type(g.ARROW)
self.guess_mode = False
def get_marker_points(self):
p1 = self.marker.pos()
p2 = self.marker.mapToScene(self.marker.end_point)
return p1, p2
def set_node_type(self, value):
self.node_type_selector.setCurrentIndex(self.node_type_selector.findData(value, role=256))
def resizeEvent(self, event):
self.limited_update_position()
QtWidgets.QWidget.resizeEvent(self, event)
def limited_update_position(self):
""" Updates the position but tries to minimize the movement of left top
corner. This is called on resize, to ensure that all of the embed is in visible area.
:return:
"""
view = self.parent()
eg = self.geometry()
ex, ey, ew, eh = eg.getRect()
top_left = view.mapToScene(ex, ey)
bottom_right = view.mapToScene(QtCore.QPoint(ex + ew, ey + eh))
size_in_scene = bottom_right - top_left
x = top_left.x()
y = top_left.y()
w = size_in_scene.x()
h = size_in_scene.y()
vr = view.mapToScene(view.geometry()).boundingRect()
view_left, view_top, view_right, view_bottom = vr.getRect()
view_right += view_left
view_bottom += view_top
w_overlap = (x + w) - view_right
h_overlap = (y + h) - view_bottom
move = False
if w_overlap > 0:
x = x - w_overlap - 12
move = True
if h_overlap > 0:
y = y - h_overlap - 8
move = True
if move:
new_pos = QtCore.QPoint(x, y)
self.move(view.mapFromScene(new_pos))
self.updateGeometry()
