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

fix: исправил падение приложения в случае отсутствия меток начала и конца этапа

Browse Source
This commit is contained in:
Андрей Скирченко 2024-11-15 11:36:05 +03:00
parent 7683e4395c
commit 34b8b49624
1 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,3 +1,5 @@
from typing import Optional
import pandas as pd
from PyQt5.QtWidgets import QWidget, QVBoxLayout
import pyqtgraph as pg
@ -11,15 +13,17 @@ class PlotWidget(BasePlotWidget):
def _create_stage_region(self,
stage: str,
times: pd.Series,
dataframe: pd.DataFrame) -> pg.LinearRegionItem:
dataframe: pd.DataFrame) -> Optional[pg.LinearRegionItem]:
stage_diff = np.diff(dataframe[stage])
start_index = np.where(stage_diff == 1)[0]
finish_index = np.where(stage_diff == -1)[0]
start_timestamp = times[start_index[0]]
finish_timestamp = times[finish_index[0]]
region = pg.LinearRegionItem([start_timestamp, finish_timestamp], movable=False)
region.setBrush(pg.mkBrush(self._stage_colors[stage]))
return region
if start_index:
start_timestamp = times[start_index[0]]
finish_timestamp = times[finish_index[0]] if finish_index else times[len(times) - 1]
region = pg.LinearRegionItem([start_timestamp, finish_timestamp], movable=False)
region.setBrush(pg.mkBrush(self._stage_colors[stage]))
return region
return None
def _build_widget(self, dataframe: pd.DataFrame) -> QWidget:
widget = QWidget()
@ -38,7 +42,8 @@ class PlotWidget(BasePlotWidget):
if settings["stages"] and all([stage in dataframe_headers for stage in self._stages]):
for stage in self._stages:
region = self._create_stage_region(stage, time_axis, dataframe)
plot_widget.addItem(region)
if region:
plot_widget.addItem(region)
for signal in description["Signals"]:
if signal["name"] in dataframe_headers: