Line plot: Customizable labels

Author: VesnaT

Orange/widgets/visualize/owlineplot.py

@@ -1,3 +1,4 @@
+from collections import OrderedDict
 from xml.sax.saxutils import escape
 
 import numpy as np
@@ -27,6 +28,11 @@
 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 (
+    BaseSetter, AxisFontUpdater, LegendFontUpdater, Customizable
+)
+from Orange.widgets.visualize.utils.customizableplot import FontUpdater, \
+    BaseParameterSetter as Setter
 from Orange.widgets.widget import OWWidget, Input, Output, Msg
 
 
@@ -183,7 +189,47 @@ def reset(self):
         self._graph_state = SELECT
 
 
-class LinePlotGraph(pg.PlotWidget):
+class ParameterSetter(Setter):
+    initial_settings = OrderedDict({
+        Setter.LABELS_BOX:
+            OrderedDict({name: values for name, values in (
+                (Setter.FONT_FAMILY_LABEL, FontUpdater.FONT_FAMILY_SETTING),
+                (Setter.AXIS_TICKS_LABEL, FontUpdater.AXIS_TICKS_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")
         super().__init__(parent, viewBox=LinePlotViewBox(),
@@ -211,6 +257,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 +487,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 +820,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")