Merge pull request #73 from mhearne-usgs/colormap

Added an option to color palette class to return 8-bit RGBA arrays; r…
usgs · Apr 18, 2018 · a814677 · a814677
2 parents c31519c + b46ac15
commit a814677
Show file tree

Hide file tree

Showing 3 changed files with 173 additions and 261 deletions.
diff --git a/impactutils/colors/cpalette.py b/impactutils/colors/cpalette.py
@@ -313,6 +313,7 @@ def getDataColor(self, value, color_format='mlab'):
                 - '255' Return a 4 element tuple of (R,G,B,A) with integer
                   values betwen 0 and 255.
                 - 'hex' Return an HTML-style hex color specification (#RRGGBB).
+                - 'array' Return an RGBA array of the same 1 or 2D dimensions as value.
 
         Returns:
             The color value associated with the input data value.
@@ -327,6 +328,9 @@ def getDataColor(self, value, color_format='mlab'):
         elif color_format == '255':
             color255 = tuple([int(c * 255) for c in color])
             return color255
+        elif color_format == 'array':
+            rgba = np.uint8(color*255)
+            return rgba
         elif color_format == 'hex':
             color255 = [int(c * 255) for c in color]
             hexcolor = ('#%02x%02x%02x'