Line plot: Customizable labels

Author: VesnaT

Orange/widgets/visualize/owlineplot.py

@@ -27,6 +27,8 @@
 from Orange.widgets.utils.widgetpreview import WidgetPreview
 from Orange.widgets.utils.state_summary import format_summary_details
 from Orange.widgets.visualize.owdistributions import LegendItem
+from Orange.widgets.visualize.utils.customizableplot import FontUpdater, \
+    BaseParameterSetter as Setter
 from Orange.widgets.widget import OWWidget, Input, Output, Msg
 
 
@@ -93,6 +95,23 @@ class LinePlotStyle:
 
 
 class LinePlotAxisItem(pg.AxisItem):
+    def generateDrawSpecs(self, p):
+        if self.tickFont:
+            p.setFont(self.tickFont)
+        return super().generateDrawSpecs(p)
+
+    def _updateMaxTextSize(self, x):
+        if self.orientation in ['left', 'right']:
+            self.textWidth = x
+            if self.style['autoExpandTextSpace'] is True:
+                self._updateWidth()
+        else:
+            self.textHeight = x
+            if self.style['autoExpandTextSpace'] is True:
+                self._updateHeight()
+
+
+class BottomAxisItem(LinePlotAxisItem):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
         self._ticks = {}
@@ -103,6 +122,9 @@ def set_ticks(self, ticks):
     def tickStrings(self, values, scale, spacing):
         return [self._ticks.get(v * scale, "") for v in values]
 
+    def boundingRect(self):
+        return super().boundingRect().adjusted(0, 0, 0, 10)
+
 
 class LinePlotViewBox(ViewBox):
     selection_changed = Signal(np.ndarray)
@@ -183,16 +205,56 @@ def reset(self):
         self._graph_state = SELECT
 
 
-class LinePlotGraph(pg.PlotWidget):
+class ParameterSetter(Setter):
+    initial_settings = {Setter.LABELS_BOX: {
+        Setter.FONT_FAMILY_LABEL: FontUpdater.FONT_FAMILY_SETTING,
+        Setter.AXIS_TICKS_LABEL: FontUpdater.FONT_SETTING,
+        Setter.LEGEND_LABEL: FontUpdater.FONT_SETTING,
+    }}
+
+    def __init__(self):
+        self.legend_settings = {}
+
+        def update_axes(**settings):
+            FontUpdater.update_axes_ticks_font(self.axis_items, **settings)
+
+        def update_legend(**settings):
+            self.legend_settings.update(**settings)
+            FontUpdater.update_legend_font(self.legend_items, **settings)
+
+        def update_font_family(**settings):
+            update_axes(**settings)
+            update_legend(**settings)
+
+        self.setters = {self.LABELS_BOX: {
+            self.FONT_FAMILY_LABEL: update_font_family,
+            self.AXIS_TICKS_LABEL: update_axes,
+            self.LEGEND_LABEL: update_legend,
+        }}
+
+    @property
+    def axis_items(self):
+        return [value["item"] for value in self.getPlotItem().axes.values()]
+
+    @property
+    def legend_items(self):
+        return self.legend.items
+
+
+# Customizable plot widget
+class LinePlotGraph(pg.PlotWidget, ParameterSetter):
     def __init__(self, parent):
-        self.bottom_axis = LinePlotAxisItem(orientation="bottom")
+        self.bottom_axis = BottomAxisItem(orientation="bottom")
+        left_item = LinePlotAxisItem(orientation="left")
         super().__init__(parent, viewBox=LinePlotViewBox(),
                          background="w", enableMenu=False,
-                         axisItems={"bottom": self.bottom_axis})
+                         axisItems={"bottom": self.bottom_axis,
+                                    "left": left_item})
         self.view_box = self.getViewBox()
         self.selection = set()
         self.legend = self._create_legend(((1, 0), (1, 0)))
         self.getPlotItem().buttonsHidden = True
+        self.getPlotItem().setContentsMargins(5, 0, 0, 10)
         self.setRenderHint(QPainter.Antialiasing, True)
 
     def _create_legend(self, anchor):
@@ -211,6 +273,8 @@ def update_legend(self, variable):
                 dots = pg.ScatterPlotItem(pen=c, brush=c, size=10, shape="s")
                 self.legend.addItem(dots, escape(name))
             self.legend.show()
+        FontUpdater.update_legend_font(self.legend_items,
+                                       **self.legend_settings)
 
     def select(self, indices):
         keys = QApplication.keyboardModifiers()
@@ -439,6 +503,7 @@ class Outputs:
     selection = Setting(None, schema_only=True)
 
     graph_name = "graph.plotItem"
+    initial_visual_settings = LinePlotGraph.initial_settings
 
     class Error(OWWidget.Error):
         not_enough_attrs = Msg("Need at least one continuous feature.")
@@ -771,6 +836,9 @@ def clear(self):
     def __in(obj, collection):
         return collection is not None and obj in collection
 
+    def set_visual_settings(self, *args):
+        self.graph.set_parameter(*args)
+
 
 if __name__ == "__main__":
     data = Table("brown-selected")

Orange/widgets/visualize/tests/test_owlineplot.py

@@ -306,6 +306,12 @@ def test_summary(self):
         self.assertEqual(info._StateInfo__output_summary.brief, "")
         self.assertEqual(info._StateInfo__output_summary.details, no_output)
 
+    def test_scalable_axis_item(self):
+        from pyqtgraph import __version__
+        # When upgraded to 0.11.1 check if resizing AxisItem font size works
+        # and LinePlotAxisItem functionality can be removed.
+        self.assertLess(__version__, "0.11.1")
+
 
 class TestSegmentsIntersection(unittest.TestCase):
     def test_ccw(self):