Add example code for current logo

timhoffm · timhoffm · commit 96c8a5766a88 · 2019-03-31T11:45:56.000+02:00
diff --git a/examples/misc/logos2.py b/examples/misc/logos2.py
@@ -8,6 +8,156 @@
 Thanks to Tony Yu <tsyu80@gmail.com> for the logo design
 """
 
+import numpy as np
+import matplotlib as mpl
+import matplotlib.pyplot as plt
+import matplotlib.cm as cm
+from matplotlib.patches import Circle, Rectangle
+
+
+MPL_BLUE = '#11557c'
+
+
+def get_fontdict():
+    mgr = FontManager()
+    if 'Calibri' in mgr.findfont('Calibri:bold'):
+        return {'fontfamily': 'Calibri', 'fontweight': 'bold'}
+    if 'Carlito' in mgr.findfont('Carlito:bold'):
+        return {'fontfamily': 'Carlito', 'fontweight': 'bold'}
+    return None
+
+
+def create_icon_axes(fig, ax_position, lw_bars, lw_grid, lw_border, rgrid):
+    """
+    Create a polar axes containing the matplotlib radar plot.
+
+    Parameters
+    ----------
+    fig : matplotlib.figure.Figure
+        The figure to draw into.
+    ax_position : (float, float, float, float)
+        The position of the created Axes in figure coordinates as
+        (x, y, width, height).
+    lw_bars : float
+        The linewidth of the bars.
+    lw_grid : float
+        The linewidth of the grid.
+    lw_border : float
+        The linewidth of the Axes border.
+    rgrid : array-like
+        Positions of the radial grid.
+
+    Returns
+    -------
+    ax : matplotlib.axes.Axes
+        The created Axes.
+    """
+    with plt.rc_context({'axes.edgecolor': MPL_BLUE,
+                         'axes.linewidth': lw_border}):
+        ax = fig.add_axes(ax_position, projection='polar')
+        ax.set_axisbelow(True)
+
+        N = 7
+        arc = 2. * np.pi
+        theta = np.arange(0.0, arc, arc / N)
+        radii = np.array([2, 6, 8, 7, 4, 5, 8])
+        width = np.pi / 4 * np.array([0.4, 0.4, 0.6, 0.8, 0.2, 0.5, 0.3])
+        bars = ax.bar(theta, radii, width=width, bottom=0.0, align='edge',
+                      edgecolor='0.1', lw=lw_bars)
+        for r, bar in zip(radii, bars):
+            color = *cm.jet(r / 10.)[:3], 0.6  # color from jet with alpha=0.6
+            bar.set_facecolor(color)
+
+        ax.tick_params(labelbottom=False, labeltop=False,
+                       labelleft=False, labelright=False)
+
+        ax.grid(lw=lw_grid, ls=':', color='0.7')
+        ax.set_rmax(9)
+        ax.set_yticks(rgrid)
+
+        # the actual visible background - extends a bit beyond the axis
+        ax.add_patch(Rectangle((0, 0), arc, 9.58,
+                               facecolor='white', zorder=0,
+                               clip_on=False, in_layout=False))
+        return ax
+
+
+def make_logo(pixels, lw_bars, lw_grid, lw_border, rgrid, with_text=False):
+    dpi = 100
+    height = pixels / dpi
+    figsize = (5 * height, height) if with_text else (height, height)
+    fig = plt.figure(figsize=figsize, dpi=dpi)
+    fig.patch.set_alpha(0)
+
+    if with_text:
+        fig.text(0.5, 0.5, 'matplotlib', fontsize=pixels * 0.8,
+                 color=MPL_BLUE, ha='center', va='center',
+                 fontdict=get_fontdict(), zorder=-10)
+
+    ax_position = (.535, .1, .17, .8) if with_text else (.03, .03, .94, .94)
+    ax = create_icon_axes(fig, ax_position, lw_bars, lw_grid, lw_border, rgrid)
+
+    return fig, ax
+
+
+params_large = {
+    'pixels': 300,
+    'lw_bars': 2,
+    'lw_grid': 0.8,
+    'lw_border': 3,
+    'rgrid': [1, 3, 5, 7],
+}
+
+fig, ax = make_logo(**params_large, with_text=True)
+plt.savefig('logo_large_text.png')
+
+fig, ax = make_logo(**params_large)
+plt.savefig('logo_large.png')
+
+##############################################################################
+
+params_medium = {
+    'pixels': 64,
+    'lw_bars': 0.5,
+    'lw_grid': 0.5,
+    'lw_border': 1,
+    'rgrid': [3, 6],
+}
+
+make_logo(**params_medium)
+plt.savefig('logo_medium.png')
+
+##############################################################################
+
+params_small = {
+    'pixels': 32,
+    'lw_bars': 0.3,
+    'lw_grid': 0.3,
+    'lw_border': 0.3,
+    'rgrid': [5],
+}
+
+make_logo(**params_small)
+plt.savefig('logo_small.png')
+
+##############################################################################
+
+params_xsmall = {
+    'pixels': 16,
+    'lw_bars': 0.1,
+    'lw_grid': 0.3,
+    'lw_border': 0.3,
+    'rgrid': [],
+}
+
+make_logo(**params_xsmall)
+plt.savefig('logo_xsmall.png')
+
+##############################################################################
+#
+# Old stuff
+# """""""""
+
 import numpy as np
 import matplotlib as mpl
 import matplotlib.pyplot as plt
