add matplotlib and simple gradient colormaps

copy_from_colorcet.py

@@ -1,18 +1,15 @@
 import colorcet
 import re
+from matplotlib import colormaps
 
 
 def parse_cmaps() -> list[str]:
-    pat = re.compile(r"(\s|\[)(\d)(,\s|])")
     s = []
     for name in colorcet.all_original_names():
         cmap = getattr(colorcet, name)
-        if isinstance(cmap, list) and len(cmap) == 256 and isinstance(cmap[0], list):
-            c = f"{cmap}"
-            c = pat.sub(r"\1\2.0\3", c)
-            c = pat.sub(r"\1\2.0\3", c)
-            c = pat.sub(r"\1\2.0\3", c)
-            s.append(f"    \"{name}\" => {c},\n")
+        if isinstance(cmap, list) and isinstance(cmap[0], list):
+            c = f"&{[[float(i) for i in j] for j in cmap]}"
+            s.append(f"    \"{name.lower()}\" => {c},\n")
     return sorted(s)
 
 
@@ -21,13 +18,24 @@ def parse_aliases() -> list[str]:
     s = []
     for name in colorcet.all_original_names():
         for alias in pat.split(colorcet.get_aliases(name)):
-            s.append(f"    \"{alias}\" => \"{name}\",\n")
-    return sorted(s)
+            s.append(f"    \"{alias.lower()}\" => \"{name.lower()}\",\n")
+    return s
+
+
+def parse_mpl_cmaps() -> list[str]:
+    s = []
+    for name in list(colormaps):
+        if not name.lower().startswith("cet") and not name.lower().endswith("_r"):
+            cmap = colormaps[name]
+            cmap = cmap(range(cmap.N))[:, :3].tolist()
+            c = f"&{[[float(i) for i in j] for j in cmap]}"
+            s.append(f"    \"{name.lower()}\" => {c},\n")
+    return s
 
 
 def main() -> None:
     """extract colormaps from the python package colorcet, run cargo fmt after"""
-    cmaps = parse_cmaps()
+    cmaps = parse_cmaps() + parse_mpl_cmaps()
     aliases = parse_aliases()
 
     with open("src/colormaps.rs", "w") as f:
@@ -35,15 +43,15 @@ def main() -> None:
         f.write("\n")
         f.write("/// unique colormaps\n")
         f.write("#[allow(clippy::approx_constant)]\n")
-        f.write("pub static COLOR_MAPS: phf::Map<&'static str, [[f64; 3]; 256]> = phf_map! {\n")
-        for i in cmaps:
+        f.write("pub static COLOR_MAPS: phf::Map<&'static str, &[[f64; 3]]> = phf_map! {\n")
+        for i in sorted(cmaps):
             f.write(i)
         f.write("};\n")
         f.write("\n")
         f.write("/// aliases to colormaps\n")
         f.write("#[allow(clippy::approx_constant)]\n")
         f.write("pub static ALIASES: phf::Map<&'static str, &'static str> = phf_map! {\n")
-        for i in aliases:
+        for i in sorted(aliases):
             f.write(i)
         f.write("};")