colorwheel generator

abm/data/metaprotocol/experiments/scripts/VisualFlocking/circular_colorwheel.py

@@ -0,0 +1,34 @@
+import matplotlib.pyplot as plt
+import numpy as np
+from matplotlib import cm
+import matplotlib as mpl
+import colorcet as cc
+
+# choosing a colormap CET_C8
+# https://colorcet.holoviz.org/user_guide/Continuous.html
+# https://colorcet.holoviz.org/user_guide/Discrete.html
+# https://colorcet.holoviz.org/user_guide/PerceptuallyUniform.html
+cmap = cc.cm['CET_CBC1']
+
+fig = plt.figure()
+
+display_axes = fig.add_axes([0.1,0.1,0.8,0.8], projection='polar')
+display_axes._direction = 2*np.pi ## This is a nasty hack - using the hidden field to
+                                  ## multiply the values such that 1 become 2*pi
+                                  ## this field is supposed to take values 1 or -1 only!!
+
+norm = mpl.colors.Normalize(0.0, 2*np.pi)
+
+# Plot the colorbar onto the polar axis
+# note - use orientation horizontal so that the gradient goes around
+# the wheel rather than centre out
+quant_steps = 2056
+cb = mpl.colorbar.ColorbarBase(display_axes, cmap=cmap,
+                                   norm=norm,
+                                   orientation='horizontal')
+
+# aesthetics - get rid of border and axis labels
+cb.outline.set_visible(False)
+display_axes.set_axis_off()
+display_axes.set_rlim([-1,1])
+plt.show() # Replace with plt.savefig if you want to save a file