def __init__(self, parent=None):
super().__init__(parent)
self.__groups = None
self.__profiles = None
self.data = None
self.data_subset = None
self.subset_selection = []
self.graph_variables = []
# Setup GUI
infobox = gui.widgetBox(self.controlArea, "Info")
self.infoLabel = gui.widgetLabel(infobox, "No data on input.")
displaybox = gui.widgetBox(self.controlArea, "Display")
radiobox = gui.radioButtons(displaybox, self, "display_index",
callback=self.__update_visibility)
gui.appendRadioButton(radiobox, "Range (min-max) with mean")
gui.appendRadioButton(radiobox, "Line plot")
gui.appendRadioButton(radiobox, "Mean")
gui.appendRadioButton(radiobox, "Line plot with mean")
showbox = gui.widgetBox(self.controlArea, "Show")
gui.checkBox(showbox, self, "display_quartiles", "Error bars",
callback=self.__update_visibility)
self.group_vars = DomainModel(
placeholder="None", separators=False,
valid_types=DiscreteVariable)
self.group_view = gui.listView(
self.controlArea, self, "group_var", box="Group by",
model=self.group_vars, callback=self.__group_var_changed)
self.group_view.setEnabled(False)
self.group_view.setMinimumSize(QSize(30, 100))
self.group_view.setSizePolicy(QSizePolicy.Expanding,
QSizePolicy.Ignored)
self.gui = OWPlotGUI(self)
self.box_zoom_select(self.controlArea)
gui.rubber(self.controlArea)
gui.auto_commit(self.controlArea, self, "auto_commit",
"Send Selection", "Send Automatically")
self.graph = LinePlotGraph(self)
self.graph.getPlotItem().buttonsHidden = True
self.graph.setRenderHint(QPainter.Antialiasing, True)
self.mainArea.layout().addWidget(self.graph)
self._legend = LegendItem()
self._legend.setParentItem(self.graph.getPlotItem().vb)
self._legend.hide()
self._legend.anchor((1, 0), (1, 0))
def _create_legend(self, anchor):
legend = LegendItem()
legend.setParentItem(self.view_box)
legend.restoreAnchor(anchor)
legend.hide()
return legend
def _create_legend(self, anchor):
legend = LegendItem()
legend.setParentItem(self.view_box)
legend.restoreAnchor(anchor)
legend.hide()
return legend
class OWLinePlot(OWWidget):
name = "Line Plot"
description = "Visualization of data profiles (e.g., time series)."
icon = "icons/LinePlot.svg"
priority = 1030
class Inputs:
data = Input("Data", Table, default=True)
data_subset = Input("Data Subset", Table)
class Outputs:
selected_data = Output("Selected Data", Table, default=True)
annotated_data = Output(ANNOTATED_DATA_SIGNAL_NAME, Table)
settingsHandler = settings.PerfectDomainContextHandler()
group_var = settings.ContextSetting(None)
display_index = settings.Setting(LinePlotDisplay.RANGE_WITH_MEAN)
display_quartiles = settings.Setting(False)
auto_commit = settings.Setting(True)
selection = settings.ContextSetting([])
class Information(OWWidget.Information):
not_enough_attrs = Msg("Need at least one continuous feature.")
def __init__(self, parent=None):
super().__init__(parent)
self.__groups = None
self.__profiles = None
self.data = None
self.data_subset = None
self.subset_selection = []
self.graph_variables = []
# Setup GUI
infobox = gui.widgetBox(self.controlArea, "Info")
self.infoLabel = gui.widgetLabel(infobox, "No data on input.")
displaybox = gui.widgetBox(self.controlArea, "Display")
radiobox = gui.radioButtons(displaybox, self, "display_index",
callback=self.__update_visibility)
gui.appendRadioButton(radiobox, "Range (min-max) with mean")
gui.appendRadioButton(radiobox, "Line plot")
gui.appendRadioButton(radiobox, "Mean")
gui.appendRadioButton(radiobox, "Line plot with mean")
showbox = gui.widgetBox(self.controlArea, "Show")
gui.checkBox(showbox, self, "display_quartiles", "Error bars",
callback=self.__update_visibility)
self.group_vars = DomainModel(
placeholder="None", separators=False,
valid_types=DiscreteVariable)
self.group_view = gui.listView(
self.controlArea, self, "group_var", box="Group by",
model=self.group_vars, callback=self.__group_var_changed)
self.group_view.setEnabled(False)
self.group_view.setMinimumSize(QSize(30, 100))
self.group_view.setSizePolicy(QSizePolicy.Expanding,
QSizePolicy.Ignored)
self.gui = OWPlotGUI(self)
self.box_zoom_select(self.controlArea)
gui.rubber(self.controlArea)
gui.auto_commit(self.controlArea, self, "auto_commit",
"Send Selection", "Send Automatically")
self.graph = LinePlotGraph(self)
self.graph.getPlotItem().buttonsHidden = True
self.graph.setRenderHint(QPainter.Antialiasing, True)
self.mainArea.layout().addWidget(self.graph)
self._legend = LegendItem()
self._legend.setParentItem(self.graph.getPlotItem().vb)
self._legend.hide()
self._legend.anchor((1, 0), (1, 0))
@property
def has_selection(self):
return bool(self.selection)
@property
def has_subset(self):
return bool(self.data_subset)
def box_zoom_select(self, parent):
g = self.gui
box_zoom_select = gui.vBox(parent, "Zoom/Select")
zoom_select_toolbar = g.zoom_select_toolbar(
box_zoom_select, nomargin=True,
buttons=[g.StateButtonsBegin, g.SimpleSelect, g.Pan, g.Zoom,
g.StateButtonsEnd, g.ZoomReset]
)
buttons = zoom_select_toolbar.buttons
buttons[g.SimpleSelect].clicked.connect(self.select_button_clicked)
buttons[g.Pan].clicked.connect(self.pan_button_clicked)
buttons[g.Zoom].clicked.connect(self.zoom_button_clicked)
buttons[g.ZoomReset].clicked.connect(self.reset_button_clicked)
return box_zoom_select
def select_button_clicked(self):
self.graph.state = SELECT
self.graph.getViewBox().setMouseMode(self.graph.getViewBox().RectMode)
def pan_button_clicked(self):
self.graph.state = PANNING
self.graph.getViewBox().setMouseMode(self.graph.getViewBox().PanMode)
def zoom_button_clicked(self):
self.graph.state = ZOOMING
self.graph.getViewBox().setMouseMode(self.graph.getViewBox().RectMode)
def reset_button_clicked(self):
self.graph.getViewBox().autoRange()
def selection_changed(self):
self.selection = list(self.graph.selection)
self.__update_visibility()
self.commit()
def sizeHint(self):
return QSize(1000, 562)
def clear(self):
"""
Clear/reset the widget state.
"""
self.__groups = None
self.__profiles = None
self.graph_variables = []
self.graph.reset()
self.infoLabel.setText("No data on input.")
self.group_vars.set_domain(None)
self.group_view.setEnabled(False)
self._legend.hide()
@Inputs.data
def set_data(self, data):
"""
Set the input profile dataset.
"""
self.closeContext()
self.clear()
self.clear_messages()
self.data = data
self.selection = []
if data is not None:
self.group_vars.set_domain(data.domain)
self.group_view.setEnabled(len(self.group_vars) > 1)
self.group_var = data.domain.class_var if \
data.domain.class_var and data.domain.class_var.is_discrete \
else None
self.infoLabel.setText("%i instances on input\n%i attributes" % (
len(data), len(data.domain.attributes)))
self.graph_variables = [var for var in data.domain.attributes
if var.is_continuous]
if len(self.graph_variables) < 1:
self.data = None
self.Information.not_enough_attrs()
self.commit()
return
self.openContext(data)
self._setup_plot()
self.commit()
@Inputs.data_subset
def set_data_subset(self, subset):
"""
Set the supplementary input subset dataset.
"""
self.data_subset = subset
if len(self.subset_selection):
self.graph.deselect_subset()
def handleNewSignals(self):
self.subset_selection = []
if self.data is not None and self.has_subset and \
len(self.graph_variables):
intersection = set(self.data.ids).intersection(
set(self.data_subset.ids))
self.subset_selection = intersection
if self.__profiles is not None:
self.graph.select_subset(self.subset_selection)
if self.data is not None:
self.__update_visibility()
def _setup_plot(self):
"""Setup the plot with new curve data."""
if self.data is None:
return
ticks = [[(i + 1, str(a)) for i, a in enumerate(self.graph_variables)]]
self.graph.getAxis('bottom').setTicks(ticks)
self.graph.getViewBox().enableAutoRange()
if self.display_index in (LinePlotDisplay.INSTANCES,
LinePlotDisplay.INSTANCES_WITH_MEAN):
self._plot_profiles()
self._plot_groups()
self.__update_visibility()
self._setup_legend()
def _setup_legend(self):
self._legend.clear()
self._legend.hide()
if self.group_var:
for index, name in enumerate(self.group_var.values):
color = self.__get_line_color(None, index)
self._legend.addItem(
ScatterPlotItem(pen=color, brush=color, size=10, shape="s"),
escape(name)
)
self._legend.show()
def _plot_profiles(self):
X = np.arange(1, len(self.graph_variables) + 1)
data = self.data[:, self.graph_variables]
self.__profiles = []
for index, inst in zip(range(len(self.data)), data):
color = self.__get_line_color(index)
profile = LinePlotItem(self, index, inst.id, X, inst.x, color)
profile.sigClicked.connect(self.graph.select_by_click)
self.graph.add_line_plot_item(profile)
self.__profiles.append(profile)
self.graph.finished_adding()
self.__select_data_instances()
def _plot_groups(self):
if self.__groups is not None:
for group in self.__groups:
if group is not None:
self.graph.getViewBox().removeItem(group.mean)
self.graph.getViewBox().removeItem(group.range)
self.graph.getViewBox().removeItem(group.error_bar)
self.__groups = []
X = np.arange(1, len(self.graph_variables) + 1)
if self.group_var is None:
self.__plot_group(X, self.data[:, self.graph_variables])
else:
class_col_data, _ = self.data.get_column_view(self.group_var)
group_indices = [np.flatnonzero(class_col_data == i)
for i in range(len(self.group_var.values))]
for index, indices in enumerate(group_indices):
if len(indices) == 0:
self.__groups.append(None)
else:
group_data = self.data[indices, self.graph_variables]
self.__plot_group(X, group_data, index)
def __plot_group(self, X, data, index=None):
p = QPen(self.__get_line_color(None, index), LinePlotStyle.MEAN_WIDTH)
p.setCosmetic(True)
mean = np.nanmean(data.X, axis=0)
mean_curve = self.get_mean_curve(X, mean, p)
range_curve = self.get_range_curve(X, data, p)
error_bar = self.get_error_bar(X, data, mean)
self.graph.addItem(mean_curve)
self.graph.addItem(range_curve)
self.graph.addItem(error_bar)
self.__groups.append(namespace(mean=mean_curve,
range=range_curve,
error_bar=error_bar))
@staticmethod
def get_mean_curve(X, mean, pen):
return pg.PlotDataItem(x=X, y=mean, pen=pen, antialias=True)
@staticmethod
def get_range_curve(X, data, pen):
color = QColor(pen.color())
color.setAlpha(LinePlotStyle.RANGE_ALPHA)
max_curve = pg.PlotDataItem(x=X, y=np.nanmax(data.X, axis=0))
min_curve = pg.PlotDataItem(x=X, y=np.nanmin(data.X, axis=0))
return pg.FillBetweenItem(max_curve, min_curve, brush=color)
@staticmethod
def get_error_bar(X, data, mean):
q1, q2, q3 = np.nanpercentile(data.X, [25, 50, 75], axis=0)
bottom = np.clip(mean - q1, 0, mean - q1)
top = np.clip(q3 - mean, 0, q3 - mean)
return pg.ErrorBarItem(x=X, y=mean, bottom=bottom, top=top, beam=0.01)
def __update_visibility(self):
self.__update_visibility_profiles()
self.__update_visibility_groups()
def __update_visibility_groups(self):
show_mean = self.display_index in (LinePlotDisplay.MEAN,
LinePlotDisplay.INSTANCES_WITH_MEAN,
LinePlotDisplay.RANGE_WITH_MEAN)
show_range = self.display_index == LinePlotDisplay.RANGE_WITH_MEAN
if self.__groups is not None:
for group in self.__groups:
if group is not None:
group.mean.setVisible(show_mean)
group.range.setVisible(show_range)
group.error_bar.setVisible(self.display_quartiles)
def __update_visibility_profiles(self):
show_selection = self.display_index == LinePlotDisplay.RANGE_WITH_MEAN
show_inst = self.display_index in (LinePlotDisplay.INSTANCES,
LinePlotDisplay.INSTANCES_WITH_MEAN)
if self.__profiles is None and (show_inst or show_selection) \
and self.data is not None:
self._plot_profiles()
self.graph.select_subset(self.subset_selection)
if self.__profiles is not None:
selection = set(chain(
self.selection,
self.data_subset.ids if self.has_subset else []
))
for profile in self.__profiles:
selected = profile.index in selection
profile.setVisible(show_inst or selected and show_selection)
profile.reset_pen()
def __group_var_changed(self):
if self.data is None or not len(self.graph_variables):
return
self.__color_profiles()
self._plot_groups()
self.__update_visibility()
self._setup_legend()
def __color_profiles(self):
if self.__profiles is not None:
for profile in self.__profiles:
profile.setColor(self.__get_line_color(profile.index))
def __select_data_instances(self):
if self.data is None or not len(self.data) or not self.has_selection:
return
if max(self.selection) >= len(self.data):
self.selection = []
self.graph.select(self.selection)
def __get_line_color(self, data_index=None, mean_index=None):
color = QColor(LinePlotStyle.DEFAULT_COLOR)
if self.group_var is not None:
if data_index is not None:
value = self.data[data_index][self.group_var]
if np.isnan(value):
return color
index = int(value) if data_index is not None else mean_index
color = LinePlotStyle()(len(self.group_var.values))[index]
return color.darker(110) if data_index is None else color
def commit(self):
selected = self.data[self.selection] \
if self.data is not None and self.has_selection else None
annotated = create_annotated_table(self.data, self.selection)
self.Outputs.selected_data.send(selected)
self.Outputs.annotated_data.send(annotated)