Add example code for current logo

timhoffm · timhoffm · commit e6b37ade26a8 · 2019-01-12T21:00:45.000+01:00
diff --git a/examples/misc/logos2.py b/examples/misc/logos2.py
@@ -8,6 +8,87 @@
 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
+
+
+def make_logo(figsize, lw_bars, lw_grid, lw_border, rgrid):
+    mpl_blue = '#11557c'
+
+    with plt.rc_context(
+            {'axes.edgecolor': mpl_blue, 'axes.linewidth': lw_border}):
+        fig = plt.figure(figsize=figsize, dpi=100)
+        fig.patch.set_alpha(0)
+
+        ax = fig.add_subplot(111, 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,
+                      edgecolor=mpl_blue, lw=lw_bars)
+        for r, bar in zip(radii, bars):
+            bar.set_facecolor(cm.jet(r / 10.))
+            bar.set_alpha(0.6)
+
+        ax.tick_params(labelbottom=False, labeltop=False,
+                       labelleft=False, labelright=False)
+
+        ax.grid(lw=lw_grid, ls='-', color='0.7')
+        ax.set_rlim(top=9)
+        ax.set_yticks(rgrid)
+
+        fig.tight_layout(pad=0)
+        return fig, ax
+
+
+params_large = {
+    'figsize': (3, 3),
+    'lw_bars': 2,
+    'lw_grid': 0.8,
+    'lw_border': 3,
+    'rgrid': np.arange(1, 9, 2),
+}
+fig, ax = make_logo(**params_large)
+plt.savefig('logo_large.png')
+
+##############################################################################
+
+params_medium = {
+    'figsize': (1, 1),
+    'lw_bars': 0.5,
+    'lw_grid': 0.5,
+    'lw_border': 1,
+    'rgrid': [3, 6],
+}
+
+make_logo(**params_medium)
+plt.savefig('logo_small.png')
+
+##############################################################################
+
+params_xsmall = {
+    'figsize': (.32, .32),
+    'lw_bars': 0.3,
+    'lw_grid': 0.3,
+    'lw_border': 0.3,
+    'rgrid': [5],
+}
+
+make_logo(**params_xsmall)
+plt.savefig('logo_xsmall.png')
+
+##############################################################################
+#
+# Old stuff
+# """""""""
+
 import numpy as np
 import matplotlib as mpl
 import matplotlib.pyplot as plt
