From 37dbbf00ebd8a1e4e4110008cee64798bc48e838 Mon Sep 17 00:00:00 2001 From: Jan Grewe Date: Thu, 2 Sep 2021 13:52:59 +0200 Subject: [PATCH] [plot] separate plotting of dataarrays and tags --- nixview/ui/plotscreen.py | 19 ++++++++++++++++--- 1 file changed, 16 insertions(+), 3 deletions(-) diff --git a/nixview/ui/plotscreen.py b/nixview/ui/plotscreen.py index 00a60bd..912986c 100644 --- a/nixview/ui/plotscreen.py +++ b/nixview/ui/plotscreen.py @@ -1,4 +1,4 @@ -from nixview.util.enums import PlotterTypes +from nixview.util.enums import NodeType, PlotterTypes from PyQt5.QtWidgets import QGroupBox, QHBoxLayout, QLabel, QPushButton, QSizePolicy, QSlider, QVBoxLayout, QWidget from PyQt5.QtCore import pyqtSignal, Qt import matplotlib @@ -11,7 +11,6 @@ try: except ImportError as e: print("cannot import matplotlib, headless mode?", e) from matplotlib.widgets import Slider -from IPython import embed from nixview.util.file_handler import FileHandler from nixview.util.dataview import DataView @@ -554,10 +553,11 @@ class PlotScreen(QWidget): def on_view_changed(self): self._pan_slider.setEnabled(self.plotter.can_pan_horizontally) - def plot(self, item): + def plot_data_array(self, item): try: self._data_view = DataView(item, self._file_handler) except ValueError as e: + print("ping!") communicator.plot_error.emit("error in plotscreen.plot %s" % e) return if self._data_view is None: @@ -594,3 +594,16 @@ class PlotScreen(QWidget): self.plotter.plot() else: self._container.set_plotter(None) + + def plot_tag(self, item): + pass + + def plot(self, item): + if item.entity_type == NodeType.DataArray: + self.plot_data_array(item) + elif item.entity_type == NodeType.MultiTag or item.entity_type == NodeType.Tag: + self.plot_tag(item) + else: + communicator.plot_error.emit("error in plotscreen.plot cannot plot entity of type%s" % item.entity_type) + return + \ No newline at end of file