OWMosaicDisplay: data info displayed in the status bar

Author: aturanjanin

Orange/widgets/visualize/owmosaic.py

@@ -27,6 +27,8 @@
                                                  ANNOTATED_DATA_SIGNAL_NAME)
 from Orange.widgets.utils.itemmodels import DomainModel
 from Orange.widgets.utils.widgetpreview import WidgetPreview
+from Orange.widgets.utils.state_summary import (format_multiple_summaries,
+                                                format_summary_details)
 from Orange.widgets.visualize.utils import (
     CanvasText, CanvasRectangle, ViewWithPress, VizRankDialog)
 from Orange.widgets.visualize.utils.plotutils import wrap_legend_items
@@ -332,6 +334,9 @@ def __init__(self):
         self.canvas_view.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
         self.canvas_view.setRenderHint(QPainter.Antialiasing)
 
+        self.info.set_input_summary(self.info.NoInput)
+        self.info.set_output_summary(self.info.NoOutput)
+
         box = gui.vBox(self.controlArea, box=True)
         self.model_1 = DomainModel(
             order=DomainModel.MIXED, valid_types=DomainModel.PRIMITIVE)
@@ -467,10 +472,26 @@ def handleNewSignals(self):
             self.__pending_selection = None
         else:
             self.selection = set()
+        self._set_input_summary()
         self.set_color_data()
         self.update_graph()
         self.send_selection()
 
+    def _set_input_summary(self):
+        n_data = len(self.data) if self.data else 0
+        n_subset = len(self.subset_data) if self.subset_data else 0
+
+        summary, details, kwargs = self.info.NoInput, "", {}
+        if self.data or self.subset_data:
+            summary = f"{self.info.format_number(n_data)}, " \
+                      f"{self.info.format_number(n_subset)}"
+            kwargs = {"format": Qt.RichText}
+            details = format_multiple_summaries([
+                ("Data", self.data),
+                ("Data subset", self.subset_data)
+            ])
+        self.info.set_input_summary(summary, details, **kwargs)
+
     def clear_selection(self):
         self.selection = set()
         self.update_selection_rects()
@@ -516,6 +537,7 @@ def send_selection(self):
             self.Outputs.selected_data.send(None)
             self.Outputs.annotated_data.send(
                 create_annotated_table(self.data, []))
+            self.info.set_output_summary(self.info.NoOutput)
             return
         filters = []
         self.Warning.no_cont_selection_sql.clear()
@@ -537,6 +559,10 @@ def send_selection(self):
         sel_idx = [i for i, id in enumerate(self.data.ids) if id in idset]
         if self.discrete_data is not self.data:
             selection = self.data[sel_idx]
+
+        summary = len(selection) if selection else self.info.NoOutput
+        details = format_summary_details(selection) if selection else ""
+        self.info.set_output_summary(summary, details)
         self.Outputs.selected_data.send(selection)
         self.Outputs.annotated_data.send(
             create_annotated_table(self.data, sel_idx))
@@ -992,5 +1018,5 @@ def get_conditional_distribution(data, attrs):
 
 
 if __name__ == "__main__":  # pragma: no cover
-    data = Table("zoo")
-    WidgetPreview(OWMosaicDisplay).run(data, set_subset_data=data[::10])
+    dataset = Table("zoo")
+    WidgetPreview(OWMosaicDisplay).run(dataset, set_subset_data=dataset[::10])

Orange/widgets/visualize/tests/test_owmosaic.py

@@ -13,6 +13,8 @@
 from Orange.widgets.tests.base import WidgetTest, WidgetOutputsTestMixin
 from Orange.widgets.visualize.owmosaic import OWMosaicDisplay
 from Orange.widgets.tests.utils import simulate
+from Orange.widgets.utils.state_summary import (format_multiple_summaries,
+                                                format_summary_details)
 
 
 class TestOWMosaicDisplay(WidgetTest, WidgetOutputsTestMixin):
@@ -184,6 +186,54 @@ def test_selection_setting(self):
         self.assertEqual(w.selection, {1})
         self.assertIsNotNone(self.get_output(w.Outputs.selected_data, widget=w))
 
+    def test_summary(self):
+        """Check if status bar is updated when data is received"""
+        info = self.widget.info
+        data, subset = Table("zoo"), Table("zoo")[::10]
+        no_input, no_output = "No data on input", "No data on output"
+
+        self.send_signal(self.widget.Inputs.data, data)
+        data_list = [("Data", data), ("Data subset", None)]
+        summary, details = f"{len(data)}, 0", format_multiple_summaries(data_list)
+        self.assertEqual(info._StateInfo__input_summary.brief, summary)
+        self.assertEqual(info._StateInfo__input_summary.details, details)
+        self.assertEqual(info._StateInfo__output_summary.brief, "")
+        self.assertEqual(info._StateInfo__output_summary.details, no_output)
+
+        self._select_data()
+        output = self.get_output(self.widget.Outputs.selected_data)
+        summary, details = f"{len(output)}", format_summary_details(output)
+        self.assertEqual(info._StateInfo__output_summary.brief, summary)
+        self.assertEqual(info._StateInfo__output_summary.details, details)
+
+        self.send_signal(self.widget.Inputs.data_subset, subset)
+        data_list = [("Data", data), ("Data subset", subset)]
+        summary = f"{len(data)}, {len(subset)}"
+        details = format_multiple_summaries(data_list)
+        self.assertEqual(info._StateInfo__input_summary.brief, summary)
+        self.assertEqual(info._StateInfo__input_summary.details, details)
+        self.assertEqual(info._StateInfo__output_summary.brief, "")
+        self.assertEqual(info._StateInfo__output_summary.details, no_output)
+        self._select_data()
+        output = self.get_output(self.widget.Outputs.selected_data)
+        summary, details = f"{len(output)}", format_summary_details(output)
+        self.assertEqual(info._StateInfo__output_summary.brief, summary)
+        self.assertEqual(info._StateInfo__output_summary.details, details)
+
+        self.send_signal(self.widget.Inputs.data, None)
+        data_list = [("Data", None), ("Data subset", subset)]
+        summary, details = f"0, {len(subset)}", format_multiple_summaries(data_list)
+        self.assertEqual(info._StateInfo__input_summary.brief, summary)
+        self.assertEqual(info._StateInfo__input_summary.details, details)
+        self.assertEqual(info._StateInfo__output_summary.brief, "")
+        self.assertEqual(info._StateInfo__output_summary.details, no_output)
+
+        self.send_signal(self.widget.Inputs.data_subset, None)
+        self.assertEqual(info._StateInfo__input_summary.brief, "")
+        self.assertEqual(info._StateInfo__input_summary.details, no_input)
+        self.assertEqual(info._StateInfo__output_summary.brief, "")
+        self.assertEqual(info._StateInfo__output_summary.details, no_output)
+
 
 # Derive from WidgetTest to simplify creation of the Mosaic widget, although
 # we are actually testing the MosaicVizRank dialog and not the widget