bec/bec_widgets
0
0
Fork 0
mirror of https://github.com/bec-project/bec_widgets.git synced 2025-07-14 03:31:50 +02:00
Code Activity

refactor: BECFigure, BECPlotBase changed back to pyqtgraph classes inheritance

Browse Source
This commit is contained in:
wyzula-jan
2024-02-14 18:27:59 +01:00
parent 9ef331c272
commit 7768e594b5
2 changed files with 23 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
bec_widgets/widgets/figure/figure.py
View File

@ -24,44 +24,37 @@ class FigureConfig(ConnectionConfig):
)
class BECFigure(BECConnector):
class BECFigure(BECConnector, pg.GraphicsLayoutWidget):
def __init__(
self,
parent: Optional[QWidget] = None,
config: Optional[FigureConfig] = None,
client=None,
gui_id: Optional[str] = None,
graphics_layout_widget: Optional[pg.GraphicsLayoutWidget] = None,
):
if config is None:
config = FigureConfig(widget_class=self.__class__.__name__)
else:
self.config = config
super().__init__(client=client, config=config, gui_id=gui_id)
# pg.GraphicsLayoutWidget.__init__(self, parent) #in case of inheritance
self.glw = (
graphics_layout_widget
if graphics_layout_widget
else pg.GraphicsLayoutWidget(parent=parent)
)
pg.GraphicsLayoutWidget.__init__(self, parent) # in case of inheritance
self.widgets = {}
# TODO just testing adding plot
self.add_widget("widget_1", row=0, col=0, title="Plot 1")
self.widgets["widget_1"].plot(
self.widgets["widget_1"].plot_data(
np.linspace(0, 10, 100), np.sin(np.linspace(0, 10, 100)), label="sin(x)"
)
def show(self): # TODO check if useful for anything
self.window = QMainWindow()
self.window.setCentralWidget(self.glw)
self.window.show()
def close(self): # TODO check if useful for anything
if hasattr(self, "window"):
self.window.close()
# def show(self): # TODO check if useful for anything
# self.window = QMainWindow()
# self.window.setCentralWidget(self)
# self.window.show()
#
# def close(self): # TODO check if useful for anything
# if hasattr(self, "window"):
# self.window.close()
def add_widget(self, widget_id: str = None, row: int = None, col: int = None, **kwargs):
# Generate unique widget_id if not provided
@ -78,13 +71,12 @@ class BECFigure(BECConnector):
parent_figure_id=self.gui_id, widget_class="BECPlotBase", gui_id=widget_id, **kwargs
)
plot_item = pg.PlotItem()
widget = BECPlotBase(plot_item=plot_item, config=widget_config)
widget = BECPlotBase(config=widget_config)
# Check if position is occupied
if row is not None and col is not None:
print("adding plot")
if self.glw.getItem(row, col):
if self.getItem(row, col):
print(
f"Position at row {row} and column {col} is already occupied."
) # TODO change to raise error
@ -94,14 +86,14 @@ class BECFigure(BECConnector):
widget_config.column = col
# Add widget to the figure
self.glw.addItem(widget.plt, row=row, col=col)
self.addItem(widget, row=row, col=col)
else:
row, col = self._find_next_empty_position()
widget_config.row = row
widget_config.column = col
# Add widget to the figure
self.glw.addItem(widget.plt, row=row, col=col)
self.addItem(widget, row=row, col=col)
# Saving config for future referencing
self.config.widgets[widget_id] = widget_config
@ -130,7 +122,7 @@ class BECFigure(BECConnector):
Returns:
BECPlotBase: the widget at the given coordinates
"""
widget = self.glw.getItem(row, col)
widget = self.getItem(row, col)
if widget is None:
raise KeyError(f"No widget at coordinates ({row}, {col})")
return widget
@ -151,7 +143,7 @@ class BECFigure(BECConnector):
def _find_next_empty_position(self):
"""Find the next empty position (new row) in the figure."""
row, col = 0, 0
while self.glw.getItem(row, col):
while self.getItem(row, col):
row += 1
return row, col
@ -207,13 +199,8 @@ class DebugWindow(QWidget):
def _init_ui(self):
# Plotting window
self.glw_1_layout = QVBoxLayout(self.glw) # Create a new QVBoxLayout
graphics_layout_widget = (
pg.GraphicsLayoutWidget()
) # create a new pg.GraphicsLayoutWidget instance
self.figure = BECFigure(
graphics_layout_widget=graphics_layout_widget, parent=self
) # Create a new BECDeviceMonitor
self.glw_1_layout.addWidget(self.figure.glw) # Add BECDeviceMonitor to the layout
self.figure = BECFigure(parent=self) # Create a new BECDeviceMonitor
self.glw_1_layout.addWidget(self.figure) # Add BECDeviceMonitor to the layout
self.console_layout = QVBoxLayout(self.widget_console)
self.console = JupyterConsoleWidget()

9
bec_widgets/widgets/plots/plot_base.py
View File

@ -33,7 +33,7 @@ class WidgetConfig(ConnectionConfig):
)
class BECPlotBase(BECConnector): # , pg.PlotItem):
class BECPlotBase(BECConnector, pg.PlotItem):
USER_ACCESS = [
"set",
"set_title",
@ -49,7 +49,6 @@ class BECPlotBase(BECConnector): # , pg.PlotItem):
def __init__(
self,
parent: Optional[QWidget] = None, # TODO decide if needed for this class
plot_item: Optional[pg.PlotItem] = None,
config: Optional[WidgetConfig] = None,
client=None,
gui_id: Optional[str] = None,
@ -57,7 +56,7 @@ class BECPlotBase(BECConnector): # , pg.PlotItem):
if config is None:
config = WidgetConfig(widget_class=self.__class__.__name__)
super().__init__(client=client, config=config, gui_id=gui_id)
self.plt = plot_item if plot_item else pg.PlotItem(parent=parent)
pg.PlotItem.__init__(self, parent)
def set(self, **kwargs) -> None:
"""
@ -176,7 +175,7 @@ class BECPlotBase(BECConnector): # , pg.PlotItem):
self.config.axis.x_grid = x
self.config.axis.y_grid = y
def plot(self, data_x: list | np.ndarray, data_y: list | np.ndarray, label: str = None):
def plot_data(self, data_x: list | np.ndarray, data_y: list | np.ndarray, label: str = None):
"""
Plot custom data on the plot widget. These data are not saved in config.
Args:
@ -184,4 +183,4 @@ class BECPlotBase(BECConnector): # , pg.PlotItem):
data_y(list|np.ndarray): y-axis data
label(str): label of the plot
"""
self.plt.plot(data_x, data_y, name=label)
self.plot(data_x, data_y, name=label)