Simplify getting colormap sizes

QuLogic · QuLogic · commit 8551a497641c · 2024-02-15T15:44:13.000-05:00
All the special cases other than Greys are just the size of the colormap
themselves, but we want to drop that special case as well. Also,
colormaps are at most 256 elements long, so using `n=512` is probably
not necessary.
diff --git a/scripts/colormaps.py b/scripts/colormaps.py
@@ -35,36 +35,14 @@
          'terrain', 'ocean', 'gist_earth', 'cubehelix', 'rainbow'
          )
 
-for cmap in cmaps:
-    n = 512
+for name in cmaps:
+    # The maximum number of segments in a cmap is 256, and for anything smaller,
+    # the cmap will map as a suitably discrete set of colours.
+    Z = np.linspace(0, 1, 256).reshape(1, 256)
 
-    if cmap in ['tab10']: n = 10
-    if cmap in ['tab20', 'tab20b', 'tab20c']: n = 20
-    if cmap in ['Pastel1', 'Set1']: n = 9
-    if cmap in ['Pastel2', 'Accent', 'Dark2', 'Set2']: n = 8
-    if cmap in ['Set3']: n = 12
-    if cmap in ['Greys']: n = 11
-    Z = np.linspace(0, 1, n).reshape(1, n)
-
-    ax.imshow(Z, extent=[xmin, xmax, ymin, ymax], cmap=plt.get_cmap(cmap))
+    ax.imshow(Z, extent=[xmin, xmax, ymin, ymax], cmap=name)
     ax.set_xlim(xmin, xmax), ax.set_xticks([])
     ax.set_ylim(ymin, ymax), ax.set_yticks([])
 
-    """
-    if cmap == "tab10":
-        x = np.linspace(xmin, xmax, 11, endpoint=True) + 0.5*(xmax-xmin)/11
-        for i in range(10):
-            ax.text(x[i], (ymin+ymax)/2, "C%d"%i, color="white", zorder=10,
-                    family = "Source Pro Serif", size=10, ha="center", va="center")
-
-    if cmap == "Greys":
-        x = np.linspace(xmin, xmax, 12, endpoint=True) + 0.5*(xmax-xmin)/12
-        for i in range(11):
-            color = "%.1f"%(i/10)
-            text = "%.1f"%(1-i/10)
-            ax.text(x[i], (ymin+ymax)/2, text, color=color, zorder=10,
-                    family = "Source Pro Serif", size=10, ha="center", va="center")
-    """
-
-    fig.savefig(ROOT_DIR / f"figures/colormap-{cmap}.pdf")
+    fig.savefig(ROOT_DIR / f"figures/colormap-{name}.pdf")
     ax.clear()
