Welding/TraceDemo
9
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: добавил отображение региона для большой диаграммы

Browse Source
This commit is contained in:
Андрей Скирченко 2024-11-16 14:25:21 +03:00
parent 7954ef2501
commit 327e32b8e1
1 changed files with 45 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
src/gui/widgets/plot.py
View File

@ -21,7 +21,6 @@ class PlotWidget(BasePlotWidget):
stage: str, stage: str,
times: pd.Series, times: pd.Series,
dataframe: pd.DataFrame) -> Optional[pg.LinearRegionItem]: dataframe: pd.DataFrame) -> Optional[pg.LinearRegionItem]:
# TODO: сделать фабрику регионов: вертикальный, горизонтальный...
stage_diff = np.diff(dataframe[stage]) stage_diff = np.diff(dataframe[stage])
start_index = np.where(stage_diff == 1)[0] start_index = np.where(stage_diff == 1)[0]
finish_index = np.where(stage_diff == -1)[0] finish_index = np.where(stage_diff == -1)[0]
@ -34,6 +33,14 @@ class PlotWidget(BasePlotWidget):
return region return region
return None return None
@staticmethod
def _init_plot_widget(title: str) -> (pg.PlotWidget, pg.LegendItem):
plot_widget = pg.PlotWidget(title=title)
plot_widget.showGrid(x=True, y=True)
legend = pg.LegendItem((80, 60), offset=(70, 20))
legend.setParentItem(plot_widget.graphicsItem())
return plot_widget, legend
def get_stage_info(self, def get_stage_info(self,
stage: str, stage: str,
dataframe: pd.DataFrame, dataframe: pd.DataFrame,
@ -61,19 +68,17 @@ class PlotWidget(BasePlotWidget):
dataframe_headers = dataframe.columns.tolist() dataframe_headers = dataframe.columns.tolist()
for channel, description in self._plt_channels.items(): for channel, description in self._plt_channels.items():
plot_widget = pg.PlotWidget(title=channel) plot_widget, legend = self._init_plot_widget(title=channel)
plot_widget.showGrid(x=True, y=True)
legend = pg.LegendItem((80, 60), offset=(70, 20))
legend.setParentItem(plot_widget.graphicsItem())
settings = description["Settings"]
settings = description["Settings"]
if settings["stages"] and all([stage in dataframe_headers for stage in self._stages]): if settings["stages"] and all([stage in dataframe_headers for stage in self._stages]):
for stage in self._stages: for stage in self._stages:
region = self._create_stage_region(stage, time_axis, dataframe) region = self._create_stage_region(stage, time_axis, dataframe)
if region: if region:
plot_widget.addItem(region) plot_widget.addItem(region)
if settings["zoom"] and max(time_axis) < 5.0: if settings["zoom"]:
if max(time_axis) < 5.0:
stages = [self.get_stage_info("Welding", stages = [self.get_stage_info("Welding",
dataframe, dataframe,
signal["name"]) for signal in description["Signals"]] signal["name"]) for signal in description["Signals"]]
@ -99,6 +104,18 @@ class PlotWidget(BasePlotWidget):
plot_widget.setYRange(mean - 260, mean + 260) plot_widget.setYRange(mean - 260, mean + 260)
plot_widget.setInteractive(False) plot_widget.setInteractive(False)
else:
max_value = min([max(dataframe[signal["name"]]) for signal in description["Signals"]])
region = pg.LinearRegionItem([max_value - max_value * 0.015,
max_value + max_value * 0.015],
movable=False,
orientation="horizontal")
region.setBrush(pg.mkBrush([0, 250, 0, 100]))
plot_widget.setYRange(max_value - 200, max_value + 200)
plot_widget.setXRange(3.5, 4.5)
plot_widget.addItem(region)
plot_widget.setInteractive(False)
for signal in description["Signals"]: for signal in description["Signals"]:
if signal["name"] in dataframe_headers: if signal["name"] in dataframe_headers: