Added additional_children parameter for Formatoption

Chilipp · Chilipp · commit a408329ccff4 · 2017-12-15T15:13:06.000+01:00
diff --git a/CHANGELOG.rst b/CHANGELOG.rst
@@ -23,6 +23,9 @@ Added
 * Added the ``gridweights``, ``fldmean``, ``fldstd`` and ``fldpctl`` methods
   to the ``psy`` DataArray accessor to calculate weighted means, standard
   deviations and percentiles over the spatial dimensions (x- and y).
+* Added the ``additional_children`` and ``additional_dependencies`` parameters
+  to the Formatoption intialization. These parameters can be used to provide
+  additional children for a formatoption for one plotter class
 
 
 v1.0.0
diff --git a/psyplot/plotter.py b/psyplot/plotter.py
@@ -403,7 +403,9 @@ def value2pickle(self):
 
     @docstrings.get_sectionsf('Formatoption')
     @dedent
-    def __init__(self, key, plotter=None, index_in_list=None, **kwargs):
+    def __init__(self, key, plotter=None, index_in_list=None,
+                 additional_children=[], additional_dependencies=[],
+                 **kwargs):
         """
         Parameters
         ----------
@@ -415,6 +417,11 @@ def __init__(self, key, plotter=None, index_in_list=None, **kwargs):
         index_in_list: int or None
             The index that shall be used if the data is a
             :class:`psyplot.InteractiveList`
+        additional_children: list or str
+            Additional children to use (see the :attr:`children` attribute)
+        additional_dependencies: list or str
+            Additional dependencies to use (see the :attr:`dependencies`
+            attribute)
         ``**kwargs``
             Further keywords may be used to specify different names for
             children, dependencies and connection formatoptions that match the
@@ -426,6 +433,10 @@ def __init__(self, key, plotter=None, index_in_list=None, **kwargs):
         self.plotter = plotter
         self.index_in_list = index_in_list
         self.shared = set()
+        self.additional_children = additional_children
+        self.additional_dependencies = additional_dependencies
+        self.children = self.children + additional_children
+        self.dependencies = self.dependencies + additional_dependencies
         self._child_mapping = dict(zip(*tee(chain(
             self.children, self.dependencies, self.connections,
             self.parents), 2)))
@@ -455,7 +466,10 @@ def __get__(self, instance, owner):
             return getattr(instance, '_' + self.key)
         except AttributeError:
             fmto = self.__class__(
-                self.key, instance, self.index_in_list, **self.init_kwargs)
+                self.key, instance, self.index_in_list,
+                additional_children=self.additional_children,
+                additional_dependencies=self.additional_dependencies,
+                **self.init_kwargs)
             setattr(instance, '_' + self.key, fmto)
             return fmto
 
