Source code for mpl_toolkits.axes_grid1.axes_size

"""
Provides classes of simple units that will be used with AxesDivider
class (or others) to determine the size of each axes. The unit
classes define `get_size` method that returns a tuple of two floats,
meaning relative and absolute sizes, respectively.

Note that this class is nothing more than a simple tuple of two
floats. Take a look at the Divider class to see how these two
values are used.
"""

from numbers import Number

from matplotlib import cbook
from matplotlib.axes import Axes


class _Base:

    def __rmul__(self, other):
        return Fraction(other, self)

    def __add__(self, other):
        if isinstance(other, _Base):
            return Add(self, other)
        else:
            return Add(self, Fixed(other))


[docs]class Add(_Base):
[docs] def __init__(self, a, b): self._a = a self._b = b
[docs] def get_size(self, renderer): a_rel_size, a_abs_size = self._a.get_size(renderer) b_rel_size, b_abs_size = self._b.get_size(renderer) return a_rel_size + b_rel_size, a_abs_size + b_abs_size
[docs]class AddList(_Base):
[docs] def __init__(self, add_list): self._list = add_list
[docs] def get_size(self, renderer): sum_rel_size = sum([a.get_size(renderer)[0] for a in self._list]) sum_abs_size = sum([a.get_size(renderer)[1] for a in self._list]) return sum_rel_size, sum_abs_size
[docs]class Fixed(_Base): """ Simple fixed size with absolute part = *fixed_size* and relative part = 0. """
[docs] def __init__(self, fixed_size): cbook._check_isinstance(Number, fixed_size=fixed_size) self.fixed_size = fixed_size
[docs] def get_size(self, renderer): rel_size = 0. abs_size = self.fixed_size return rel_size, abs_size
[docs]class Scaled(_Base): """ Simple scaled(?) size with absolute part = 0 and relative part = *scalable_size*. """
[docs] def __init__(self, scalable_size): self._scalable_size = scalable_size
[docs] def get_size(self, renderer): rel_size = self._scalable_size abs_size = 0. return rel_size, abs_size
Scalable = Scaled def _get_axes_aspect(ax): aspect = ax.get_aspect() if aspect == "auto": aspect = 1. return aspect
[docs]class AxesX(_Base): """ Scaled size whose relative part corresponds to the data width of the *axes* multiplied by the *aspect*. """
[docs] def __init__(self, axes, aspect=1., ref_ax=None): self._axes = axes self._aspect = aspect if aspect == "axes" and ref_ax is None: raise ValueError("ref_ax must be set when aspect='axes'") self._ref_ax = ref_ax
[docs] def get_size(self, renderer): l1, l2 = self._axes.get_xlim() if self._aspect == "axes": ref_aspect = _get_axes_aspect(self._ref_ax) aspect = ref_aspect / _get_axes_aspect(self._axes) else: aspect = self._aspect rel_size = abs(l2-l1)*aspect abs_size = 0. return rel_size, abs_size
[docs]class AxesY(_Base): """ Scaled size whose relative part corresponds to the data height of the *axes* multiplied by the *aspect*. """
[docs] def __init__(self, axes, aspect=1., ref_ax=None): self._axes = axes self._aspect = aspect if aspect == "axes" and ref_ax is None: raise ValueError("ref_ax must be set when aspect='axes'") self._ref_ax = ref_ax
[docs] def get_size(self, renderer): l1, l2 = self._axes.get_ylim() if self._aspect == "axes": ref_aspect = _get_axes_aspect(self._ref_ax) aspect = _get_axes_aspect(self._axes) else: aspect = self._aspect rel_size = abs(l2-l1)*aspect abs_size = 0. return rel_size, abs_size
[docs]class MaxExtent(_Base): """ Size whose absolute part is either the largest width or the largest height of the given *artist_list*. """
[docs] def __init__(self, artist_list, w_or_h): self._artist_list = artist_list cbook._check_in_list(["width", "height"], w_or_h=w_or_h) self._w_or_h = w_or_h
[docs] def add_artist(self, a): self._artist_list.append(a)
[docs] def get_size(self, renderer): rel_size = 0. extent_list = [ getattr(a.get_window_extent(renderer), self._w_or_h) / a.figure.dpi for a in self._artist_list] abs_size = max(extent_list, default=0) return rel_size, abs_size
[docs]class MaxWidth(MaxExtent): """ Size whose absolute part is the largest width of the given *artist_list*. """
[docs] def __init__(self, artist_list): super().__init__(artist_list, "width")
[docs]class MaxHeight(MaxExtent): """ Size whose absolute part is the largest height of the given *artist_list*. """
[docs] def __init__(self, artist_list): super().__init__(artist_list, "height")
[docs]class Fraction(_Base): """ An instance whose size is a *fraction* of the *ref_size*. >>> s = Fraction(0.3, AxesX(ax)) """
[docs] def __init__(self, fraction, ref_size): cbook._check_isinstance(Number, fraction=fraction) self._fraction_ref = ref_size self._fraction = fraction
[docs] def get_size(self, renderer): if self._fraction_ref is None: return self._fraction, 0. else: r, a = self._fraction_ref.get_size(renderer) rel_size = r*self._fraction abs_size = a*self._fraction return rel_size, abs_size
[docs]class Padded(_Base): """ Return a instance where the absolute part of *size* is increase by the amount of *pad*. """
[docs] def __init__(self, size, pad): self._size = size self._pad = pad
[docs] def get_size(self, renderer): r, a = self._size.get_size(renderer) rel_size = r abs_size = a + self._pad return rel_size, abs_size
[docs]def from_any(size, fraction_ref=None): """ Create a Fixed unit when the first argument is a float, or a Fraction unit if that is a string that ends with %. The second argument is only meaningful when Fraction unit is created. >>> a = Size.from_any(1.2) # => Size.Fixed(1.2) >>> Size.from_any("50%", a) # => Size.Fraction(0.5, a) """ if isinstance(size, Number): return Fixed(size) elif isinstance(size, str): if size[-1] == "%": return Fraction(float(size[:-1]) / 100, fraction_ref) raise ValueError("Unknown format")
[docs]class SizeFromFunc(_Base):
[docs] def __init__(self, func): self._func = func
[docs] def get_size(self, renderer): rel_size = 0. bb = self._func(renderer) dpi = renderer.points_to_pixels(72.) abs_size = bb/dpi return rel_size, abs_size
[docs]class GetExtentHelper: _get_func_map = { "left": lambda self, axes_bbox: axes_bbox.xmin - self.xmin, "right": lambda self, axes_bbox: self.xmax - axes_bbox.xmax, "bottom": lambda self, axes_bbox: axes_bbox.ymin - self.ymin, "top": lambda self, axes_bbox: self.ymax - axes_bbox.ymax, }
[docs] def __init__(self, ax, direction): cbook._check_in_list(self._get_func_map, direction=direction) self._ax_list = [ax] if isinstance(ax, Axes) else ax self._direction = direction
[docs] def __call__(self, renderer): get_func = self._get_func_map[self._direction] vl = [get_func(ax.get_tightbbox(renderer, call_axes_locator=False), ax.bbox) for ax in self._ax_list] return max(vl)