def plot_data(self, data):
if self.scene_manager is None:
self.scene_manager = SceneManager(self)
self.scene_manager.plot_data = data
self.scene_manager.init_scene()
self.setScene(self.scene_manager.scene)
self.scene_manager.show_full_scene()
def dialog_ui(self) -> QDialog:
if self.__dialog_ui is None:
dir_name = os.path.dirname(
os.readlink(__file__)) if os.path.islink(
__file__) else os.path.dirname(__file__)
logging.getLogger().setLevel(logging.WARNING)
self.__dialog_ui = uic.loadUi(
os.path.realpath(
os.path.join(dir_name, "insert_sine_dialog.ui")))
logging.getLogger().setLevel(logger.level)
self.__dialog_ui.setAttribute(Qt.WA_DeleteOnClose)
self.__dialog_ui.setModal(True)
self.__dialog_ui.doubleSpinBoxAmplitude.setValue(self.__amplitude)
self.__dialog_ui.doubleSpinBoxFrequency.setValue(self.__frequency)
self.__dialog_ui.doubleSpinBoxPhase.setValue(self.__phase)
self.__dialog_ui.doubleSpinBoxSampleRate.setValue(
self.__sample_rate)
self.__dialog_ui.doubleSpinBoxNSamples.setValue(self.__num_samples)
self.__dialog_ui.lineEditTime.setValidator(
QRegExpValidator(
QRegExp(r"[0-9]+([nmµ]?|([\.,][0-9]{1,3}[nmµ]?))?$")))
scene_manager = SceneManager(self.dialog_ui.graphicsViewSineWave)
self.__dialog_ui.graphicsViewSineWave.scene_manager = scene_manager
self.insert_indicator = scene_manager.scene.addRect(
0, -2, 0, 4, QPen(QColor(Qt.transparent), 0),
QBrush(self.INSERT_INDICATOR_COLOR))
self.insert_indicator.stackBefore(
scene_manager.scene.selection_area)
self.set_time()
return self.__dialog_ui
class ZoomableGraphicView(SelectableGraphicView):
MINIMUM_VIEW_WIDTH = 300
# argument is x zoom factor
# if argument is -1, then show_full_scene was triggered during zoom
zoomed = pyqtSignal(float)
def __init__(self, parent=None):
super().__init__(parent)
self.context_menu_position = None # type: QPoint
self.scene_type = 0
self.zoom_in_action = QAction(self.tr("Zoom in"), self)
self.zoom_in_action.setShortcut(QKeySequence.ZoomIn)
self.zoom_in_action.triggered.connect(self.on_zoom_in_action_triggered)
self.zoom_in_action.setShortcutContext(Qt.WidgetWithChildrenShortcut)
self.zoom_in_action.setIcon(QIcon.fromTheme("zoom-in"))
self.addAction(self.zoom_in_action)
self.zoom_out_action = QAction(self.tr("Zoom out"), self)
self.zoom_out_action.setShortcut(QKeySequence.ZoomOut)
self.zoom_out_action.triggered.connect(
self.on_zoom_out_action_triggered)
self.zoom_out_action.setShortcutContext(Qt.WidgetWithChildrenShortcut)
self.zoom_out_action.setIcon(QIcon.fromTheme("zoom-out"))
self.addAction(self.zoom_out_action)
self.zoom_original_action = QAction(self.tr("Zoom original"), self)
self.zoom_original_action.setShortcut(QKeySequence(Qt.CTRL + Qt.Key_0))
self.zoom_original_action.triggered.connect(
self.on_zoom_original_action_triggered)
self.zoom_original_action.setShortcutContext(
Qt.WidgetWithChildrenShortcut)
self.zoom_original_action.setIcon(QIcon.fromTheme("zoom-original"))
self.addAction(self.zoom_original_action)
self.redraw_timer = QTimer()
self.redraw_timer.setSingleShot(True)
self.redraw_timer.timeout.connect(self.redraw_view)
self.zoomed.connect(self.on_signal_zoomed)
self.scene_y_min = float("nan") # NaN = AutoDetect
self.scene_y_max = float("nan") # NaN = AutoDetect
self.scene_x_zoom_stretch = 1
@property
def y_center(self):
try:
if self.scene_type == 0:
# Normal scene
return 0
else:
return -self.signal.qad_center
except Exception as e:
logger.error(
"Could not access y_center property: {0}. Falling back to 0".
format(e))
return 0
def create_context_menu(self):
menu = QMenu()
self._add_zoom_actions_to_menu(menu)
return menu
def contextMenuEvent(self, event: QContextMenuEvent):
self.context_menu_position = event.pos()
menu = self.create_context_menu()
menu.exec_(self.mapToGlobal(event.pos()))
self.context_menu_position = None
def _add_zoom_actions_to_menu(self, menu: QMenu):
menu.addAction(self.zoom_in_action)
menu.addAction(self.zoom_out_action)
if self.something_is_selected:
zoom_action = menu.addAction(self.tr("Zoom selection"))
zoom_action.setIcon(QIcon.fromTheme("zoom-fit-best"))
zoom_action.triggered.connect(self.on_zoom_action_triggered)
menu.addSeparator()
def scrollContentsBy(self, dx: int, dy: int):
try:
super().scrollContentsBy(dx, dy)
self.redraw_timer.start(0)
except RuntimeError as e:
logger.warning("Graphic View already closed: " + str(e))
def zoom(self, factor, zoom_to_mouse_cursor=True, cursor_pos=None):
if factor > 1 and self.view_rect().width(
) / factor < self.MINIMUM_VIEW_WIDTH:
factor = self.view_rect().width() / self.MINIMUM_VIEW_WIDTH
if zoom_to_mouse_cursor:
pos = self.mapFromGlobal(
QCursor.pos()) if cursor_pos is None else cursor_pos
else:
pos = None
old_pos = self.mapToScene(pos) if pos is not None else None
show_full = False
if self.view_rect().width() / factor > self.sceneRect().width():
self.show_full_scene()
factor = 1
show_full = True
self.scale(factor, 1)
if show_full:
self.zoomed.emit(-1)
else:
self.zoomed.emit(factor)
if pos is not None:
move = self.mapToScene(pos) - old_pos
self.translate(move.x(), 0)
def wheelEvent(self, event: QWheelEvent):
zoom_factor = 1.001**event.angleDelta().y()
self.zoom(zoom_factor, cursor_pos=event.pos())
def resizeEvent(self, event):
if self.sceneRect().width() == 0:
return
self.auto_fit_view()
def auto_fit_view(self):
h_tar = self.sceneRect().height()
h_view = self.view_rect().height()
if abs(h_tar) > 0:
self.scale(1, h_view / h_tar)
self.centerOn(self.view_rect().x() + self.view_rect().width() / 2,
self.y_center)
def show_full_scene(self, reinitialize=False):
y_factor = self.transform().m22()
self.resetTransform()
# Use full self.width() here to enable show_full_scene when view_rect not yet set e.g. in Record Signal Dialog
x_factor = self.width() / (
self.sceneRect().width() *
self.scene_x_zoom_stretch) if self.sceneRect().width() else 1
self.scale(x_factor, y_factor)
self.centerOn(0, self.y_center)
self.redraw_view(reinitialize)
def zoom_to_selection(self, start: int, end: int):
if start == end:
return
x_factor = self.view_rect().width() / (end - start)
self.zoom(x_factor, zoom_to_mouse_cursor=False)
self.centerOn(start + (end - start) / 2, self.y_center)
def plot_data(self, data):
if self.scene_manager is None:
self.scene_manager = SceneManager(self)
self.scene_manager.minimum = self.scene_y_min
self.scene_manager.maximum = self.scene_y_max
self.scene_manager.plot_data = data
self.scene_manager.init_scene()
self.setScene(self.scene_manager.scene)
self.scene_manager.show_full_scene()
def redraw_view(self, reinitialize=False):
if self.scene_manager is not None:
self.scene_manager.scene_type = self.scene_type
if reinitialize:
self.scene_manager.init_scene()
vr = self.view_rect()
start, end = vr.x(), vr.x() + vr.width()
self.scene_manager.show_scene_section(
start, end, *self._get_sub_path_ranges_and_colors(start, end))
def _get_sub_path_ranges_and_colors(self, start: float, end: float):
# Overwritten in Epic Graphic View
return None, None
def eliminate(self):
self.redraw_timer.stop()
super().eliminate()
@pyqtSlot()
def on_signal_zoomed(self):
self.redraw_timer.start(30)
@pyqtSlot()
def on_zoom_in_action_triggered(self):
self.zoom(1.1)
@pyqtSlot()
def on_zoom_out_action_triggered(self):
self.zoom(0.9)
@pyqtSlot()
def on_zoom_original_action_triggered(self):
self.show_full_scene(reinitialize=False)
self.zoomed.emit(-1)
@pyqtSlot()
def on_zoom_action_triggered(self):
self.zoom_to_selection(self.selection_area.start,
self.selection_area.end)