support getting colormaps by their name

swharden · swharden · commit 21f588e2d675 · 2020-07-25T18:27:24.000-04:00
swharden#18
diff --git a/src/Spectrogram.Tests/ColormapExamples.cs b/src/Spectrogram.Tests/ColormapExamples.cs
@@ -32,5 +32,15 @@ public void Test_Make_CommonColormaps()
                 Debug.WriteLine($"![](dev/graphics/hal-{cmap.Name}.png)");
             }
         }
+
+        [Test]
+        public void Test_Colormaps_ByName()
+        {
+            string[] names = Colormap.GetColormapNames();
+            Console.WriteLine(string.Join(", ", names));
+
+            Colormap viridisCmap = Colormap.GetColormap("viridis");
+            Assert.AreEqual("Viridis", viridisCmap.Name);
+        }
     }
 }
diff --git a/src/Spectrogram/Colormap.cs b/src/Spectrogram/Colormap.cs
@@ -1,5 +1,6 @@
 ﻿using System;
 using System.Collections.Generic;
+using System.ComponentModel;
 using System.Drawing;
 using System.Linq;
 using System.Text;
@@ -46,6 +47,20 @@ public static Colormap[] GetColormaps()
             return ics.Select(x => new Colormap(x)).ToArray();
         }
 
+        public static string[] GetColormapNames()
+        {
+            return GetColormaps().Select(x => x.Name).ToArray();
+        }
+
+        public static Colormap GetColormap(string colormapName)
+        {
+            foreach (Colormap cmap in GetColormaps())
+                if (string.Equals(cmap.Name, colormapName, StringComparison.InvariantCultureIgnoreCase))
+                    return cmap;
+
+            throw new ArgumentException($"Colormap does not exist: {colormapName}");
+        }
+
         public (byte r, byte g, byte b) GetRGB(byte value)
         {
             return cmap.GetRGB(value);

Original file line number	Diff line number	Diff line change
`@@ -32,5 +32,15 @@ public void Test_Make_CommonColormaps()`
`32`	`32`	`Debug.WriteLine($"![](dev/graphics/hal-{cmap.Name}.png)");`
`33`	`33`	`}`
`34`	`34`	`}`
	`35`	`+`
	`36`	`+ [Test]`
	`37`	`+ public void Test_Colormaps_ByName()`
	`38`	`+ {`
	`39`	`+ string[] names = Colormap.GetColormapNames();`
	`40`	`+ Console.WriteLine(string.Join(", ", names));`
	`41`	`+`
	`42`	`+ Colormap viridisCmap = Colormap.GetColormap("viridis");`
	`43`	`+ Assert.AreEqual("Viridis", viridisCmap.Name);`
	`44`	`+ }`
`35`	`45`	`}`
`36`	`46`	`}`