"""
Module that deals with color ramps
This product includes color specifications and designs developed by
Cynthia Brewer (http://colorbrewer.org/).
This file uses the minify function, available from
https://github.com/getify/JSON.minify
"""
# This file is part of 'TuiView' - a simple Raster viewer
# Copyright (C) 2012 Sam Gillingham
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
import os
import numpy
import json
from . import viewererrors
# environment variable we use to specify file
# to load extra ramps from
EXTRA_RAMP_VAR = 'TUIVIEW_EXTRA_RAMP'
HAVE_LOADED_EXTRA_RAMPS = False
RANDOM_NAME = 'RANDOM'
RANDOM_DISPLAY = 'Random'
# init our dictionary of data
# longest list of colors for each name
# from http://colorbrewer.org/
# generated by colorbrewer2py.py
RAMP = {}
RAMP['Accent'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Qualitative'}
RAMP['Accent']['description'] = {}
RAMP['Accent']['description']['red'] = '127 190 253 255 56 240 191 102'
RAMP['Accent']['description']['green'] = '201 174 192 255 108 2 91 102'
RAMP['Accent']['description']['blue'] = '127 212 134 153 176 127 23 102'
RAMP['Blues'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Sequential'}
RAMP['Blues']['description'] = {}
RAMP['Blues']['description']['red'] = '247 222 198 158 107 66 33 8 8'
RAMP['Blues']['description']['green'] = '251 235 219 202 174 146 113 81 48'
RAMP['Blues']['description']['blue'] = '255 247 239 225 214 198 181 156 107'
RAMP['BrBG'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Diverging'}
RAMP['BrBG']['description'] = {}
RAMP['BrBG']['description']['red'] = '84 140 191 223 246 245 199 128 53 1 0'
RAMP['BrBG']['description']['green'] = '48 81 129 194 232 245 234 205 151 102 60'
RAMP['BrBG']['description']['blue'] = '5 10 45 125 195 245 229 193 143 94 48'
RAMP['BuGn'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Sequential'}
RAMP['BuGn']['description'] = {}
RAMP['BuGn']['description']['red'] = '247 229 204 153 102 65 35 0 0'
RAMP['BuGn']['description']['green'] = '252 245 236 216 194 174 139 109 68'
RAMP['BuGn']['description']['blue'] = '253 249 230 201 164 118 69 44 27'
RAMP['BuPu'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Sequential'}
RAMP['BuPu']['description'] = {}
RAMP['BuPu']['description']['red'] = '247 224 191 158 140 140 136 129 77'
RAMP['BuPu']['description']['green'] = '252 236 211 188 150 107 65 15 0'
RAMP['BuPu']['description']['blue'] = '253 244 230 218 198 177 157 124 75'
RAMP['Dark2'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Qualitative'}
RAMP['Dark2']['description'] = {}
RAMP['Dark2']['description']['red'] = '27 217 117 231 102 230 166 102'
RAMP['Dark2']['description']['green'] = '158 95 112 41 166 171 118 102'
RAMP['Dark2']['description']['blue'] = '119 2 179 138 30 2 29 102'
RAMP['GnBu'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Sequential'}
RAMP['GnBu']['description'] = {}
RAMP['GnBu']['description']['red'] = '247 224 204 168 123 78 43 8 8'
RAMP['GnBu']['description']['green'] = '252 243 235 221 204 179 140 104 64'
RAMP['GnBu']['description']['blue'] = '240 219 197 181 196 211 190 172 129'
RAMP['Greens'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Sequential'}
RAMP['Greens']['description'] = {}
RAMP['Greens']['description']['red'] = '247 229 199 161 116 65 35 0 0'
RAMP['Greens']['description']['green'] = '252 245 233 217 196 171 139 109 68'
RAMP['Greens']['description']['blue'] = '245 224 192 155 118 93 69 44 27'
RAMP['Greys'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Sequential'}
RAMP['Greys']['description'] = {}
RAMP['Greys']['description']['red'] = '255 240 217 189 150 115 82 37 0'
RAMP['Greys']['description']['green'] = '255 240 217 189 150 115 82 37 0'
RAMP['Greys']['description']['blue'] = '255 240 217 189 150 115 82 37 0'
RAMP['OrRd'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Sequential'}
RAMP['OrRd']['description'] = {}
RAMP['OrRd']['description']['red'] = '255 254 253 253 252 239 215 179 127'
RAMP['OrRd']['description']['green'] = '247 232 212 187 141 101 48 0 0'
RAMP['OrRd']['description']['blue'] = '236 200 158 132 89 72 31 0 0'
RAMP['Oranges'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Sequential'}
RAMP['Oranges']['description'] = {}
RAMP['Oranges']['description']['red'] = '255 254 253 253 253 241 217 166 127'
RAMP['Oranges']['description']['green'] = '245 230 208 174 141 105 72 54 39'
RAMP['Oranges']['description']['blue'] = '235 206 162 107 60 19 1 3 4'
RAMP['PRGn'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Diverging'}
RAMP['PRGn']['description'] = {}
RAMP['PRGn']['description']['red'] = '64 118 153 194 231 247 217 166 90 27 0'
RAMP['PRGn']['description']['green'] = '0 42 112 165 212 247 240 219 174 120 68'
RAMP['PRGn']['description']['blue'] = '75 131 171 207 232 247 211 160 97 55 27'
RAMP['Paired'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Qualitative'}
RAMP['Paired']['description'] = {}
RAMP['Paired']['description']['red'] = '166 31 178 51 251 227 253 255 202 106 255 177'
RAMP['Paired']['description']['green'] = '206 120 223 160 154 26 191 127 178 61 255 89'
RAMP['Paired']['description']['blue'] = '227 180 138 44 153 28 111 0 214 154 153 40'
RAMP['Pastel1'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Qualitative'}
RAMP['Pastel1']['description'] = {}
RAMP['Pastel1']['description']['red'] = '251 179 204 222 254 255 229 253 242'
RAMP['Pastel1']['description']['green'] = '180 205 235 203 217 255 216 218 242'
RAMP['Pastel1']['description']['blue'] = '174 227 197 228 166 204 189 236 242'
RAMP['Pastel2'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Qualitative'}
RAMP['Pastel2']['description'] = {}
RAMP['Pastel2']['description']['red'] = '179 253 203 244 230 255 241 204'
RAMP['Pastel2']['description']['green'] = '226 205 213 202 245 242 226 204'
RAMP['Pastel2']['description']['blue'] = '205 172 232 228 201 174 204 204'
RAMP['PiYG'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Diverging'}
RAMP['PiYG']['description'] = {}
RAMP['PiYG']['description']['red'] = '142 197 222 241 253 247 230 184 127 77 39'
RAMP['PiYG']['description']['green'] = '1 27 119 182 224 247 245 225 188 146 100'
RAMP['PiYG']['description']['blue'] = '82 125 174 218 239 247 208 134 65 33 25'
RAMP['PuBuGn'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Sequential'}
RAMP['PuBuGn']['description'] = {}
RAMP['PuBuGn']['description']['red'] = '255 236 208 166 103 54 2 1 1'
RAMP['PuBuGn']['description']['green'] = '247 226 209 189 169 144 129 108 70'
RAMP['PuBuGn']['description']['blue'] = '251 240 230 219 207 192 138 89 54'
RAMP['PuBu'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Sequential'}
RAMP['PuBu']['description'] = {}
RAMP['PuBu']['description']['red'] = '255 236 208 166 116 54 5 4 2'
RAMP['PuBu']['description']['green'] = '247 231 209 189 169 144 112 90 56'
RAMP['PuBu']['description']['blue'] = '251 242 230 219 207 192 176 141 88'
RAMP['PuOr'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Diverging'}
RAMP['PuOr']['description'] = {}
RAMP['PuOr']['description']['red'] = '127 179 224 253 254 247 216 178 128 84 45'
RAMP['PuOr']['description']['green'] = '59 88 130 184 224 247 218 171 115 39 0'
RAMP['PuOr']['description']['blue'] = '8 6 20 99 182 247 235 210 172 136 75'
RAMP['PuRd'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Sequential'}
RAMP['PuRd']['description'] = {}
RAMP['PuRd']['description']['red'] = '247 231 212 201 223 231 206 152 103'
RAMP['PuRd']['description']['green'] = '244 225 185 148 101 41 18 0 0'
RAMP['PuRd']['description']['blue'] = '249 239 218 199 176 138 86 67 31'
RAMP['Purples'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Sequential'}
RAMP['Purples']['description'] = {}
RAMP['Purples']['description']['red'] = '252 239 218 188 158 128 106 84 63'
RAMP['Purples']['description']['green'] = '251 237 218 189 154 125 81 39 0'
RAMP['Purples']['description']['blue'] = '253 245 235 220 200 186 163 143 125'
RAMP['RdBu'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Diverging'}
RAMP['RdBu']['description'] = {}
RAMP['RdBu']['description']['red'] = '103 178 214 244 253 247 209 146 67 33 5'
RAMP['RdBu']['description']['green'] = '0 24 96 165 219 247 229 197 147 102 48'
RAMP['RdBu']['description']['blue'] = '31 43 77 130 199 247 240 222 195 172 97'
RAMP['RdGy'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Diverging'}
RAMP['RdGy']['description'] = {}
RAMP['RdGy']['description']['red'] = '103 178 214 244 253 255 224 186 135 77 26'
RAMP['RdGy']['description']['green'] = '0 24 96 165 219 255 224 186 135 77 26'
RAMP['RdGy']['description']['blue'] = '31 43 77 130 199 255 224 186 135 77 26'
RAMP['RdPu'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Sequential'}
RAMP['RdPu']['description'] = {}
RAMP['RdPu']['description']['red'] = '255 253 252 250 247 221 174 122 73'
RAMP['RdPu']['description']['green'] = '247 224 197 159 104 52 1 1 0'
RAMP['RdPu']['description']['blue'] = '243 221 192 181 161 151 126 119 106'
RAMP['RdYlBu'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Diverging'}
RAMP['RdYlBu']['description'] = {}
RAMP['RdYlBu']['description']['red'] = '165 215 244 253 254 255 224 171 116 69 49'
RAMP['RdYlBu']['description']['green'] = '0 48 109 174 224 255 243 217 173 117 54'
RAMP['RdYlBu']['description']['blue'] = '38 39 67 97 144 191 248 233 209 180 149'
RAMP['RdYlGn'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Diverging'}
RAMP['RdYlGn']['description'] = {}
RAMP['RdYlGn']['description']['red'] = '165 215 244 253 254 255 217 166 102 26 0'
RAMP['RdYlGn']['description']['green'] = '0 48 109 174 224 255 239 217 189 152 104'
RAMP['RdYlGn']['description']['blue'] = '38 39 67 97 139 191 139 106 99 80 55'
RAMP['Reds'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Sequential'}
RAMP['Reds']['description'] = {}
RAMP['Reds']['description']['red'] = '255 254 252 252 251 239 203 165 103'
RAMP['Reds']['description']['green'] = '245 224 187 146 106 59 24 15 0'
RAMP['Reds']['description']['blue'] = '240 210 161 114 74 44 29 21 13'
RAMP['Set1'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Qualitative'}
RAMP['Set1']['description'] = {}
RAMP['Set1']['description']['red'] = '228 55 77 152 255 255 166 247 153'
RAMP['Set1']['description']['green'] = '26 126 175 78 127 255 86 129 153'
RAMP['Set1']['description']['blue'] = '28 184 74 163 0 51 40 191 153'
RAMP['Set2'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Qualitative'}
RAMP['Set2']['description'] = {}
RAMP['Set2']['description']['red'] = '102 252 141 231 166 255 229 179'
RAMP['Set2']['description']['green'] = '194 141 160 138 216 217 196 179'
RAMP['Set2']['description']['blue'] = '165 98 203 195 84 47 148 179'
RAMP['Set3'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Qualitative'}
RAMP['Set3']['description'] = {}
RAMP['Set3']['description']['red'] = '141 255 190 251 128 253 179 252 217 188 204 255'
RAMP['Set3']['description']['green'] = '211 255 186 128 177 180 222 205 217 128 235 237'
RAMP['Set3']['description']['blue'] = '199 179 218 114 211 98 105 229 217 189 197 111'
RAMP['Spectral'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Diverging'}
RAMP['Spectral']['description'] = {}
RAMP['Spectral']['description']['red'] = '158 213 244 253 254 255 230 171 102 50 94'
RAMP['Spectral']['description']['green'] = '1 62 109 174 224 255 245 221 194 136 79'
RAMP['Spectral']['description']['blue'] = '66 79 67 97 139 191 152 164 165 189 162'
RAMP['YlGnBu'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Sequential'}
RAMP['YlGnBu']['description'] = {}
RAMP['YlGnBu']['description']['red'] = '255 237 199 127 65 29 34 37 8'
RAMP['YlGnBu']['description']['green'] = '255 248 233 205 182 145 94 52 29'
RAMP['YlGnBu']['description']['blue'] = '217 177 180 187 196 192 168 148 88'
RAMP['YlGn'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Sequential'}
RAMP['YlGn']['description'] = {}
RAMP['YlGn']['description']['red'] = '255 247 217 173 120 65 35 0 0'
RAMP['YlGn']['description']['green'] = '255 252 240 221 198 171 132 104 69'
RAMP['YlGn']['description']['blue'] = '229 185 163 142 121 93 67 55 41'
RAMP['YlOrBr'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Sequential'}
RAMP['YlOrBr']['description'] = {}
RAMP['YlOrBr']['description']['red'] = '255 255 254 254 254 236 204 153 102'
RAMP['YlOrBr']['description']['green'] = '255 247 227 196 153 112 76 52 37'
RAMP['YlOrBr']['description']['blue'] = '229 188 145 79 41 20 2 4 6'
RAMP['YlOrRd'] = {'author': 'Cynthia Brewer', 'comments': 'Colours from www.colorbrewer.org by Cynthia A. Brewer, Geography, Pennsylvania State University.', 'type': 'Sequential'}
RAMP['YlOrRd']['description'] = {}
RAMP['YlOrRd']['description']['red'] = '255 255 254 254 253 252 227 189 128'
RAMP['YlOrRd']['description']['green'] = '255 237 217 178 141 78 26 0 0'
RAMP['YlOrRd']['description']['blue'] = '204 160 118 76 60 42 28 38 38'
# Viridis palettes
RAMP['viridis'] = {'author': ' Stefan van der Walt and Nathaniel Smith', 'comments': '', 'type': 'Diverging'}
RAMP['viridis']['description'] = {}
RAMP['viridis']['description']['red'] = '68 72 67 56 45 37 30 43 81 133 194 253'
RAMP['viridis']['description']['green'] = '1 33 62 89 112 133 155 176 197 213 223 231'
RAMP['viridis']['description']['blue'] = '84 115 133 140 142 142 138 127 106 74 35 37'
RAMP['magma'] = {'author': ' Stefan van der Walt and Nathaniel Smith', 'comments': '', 'type': 'Diverging'}
RAMP['magma']['description'] = {}
RAMP['magma']['description']['red'] = '0 18 51 90 125 163 200 233 249 254 254 252'
RAMP['magma']['description']['green'] = '0 13 16 22 36 48 62 85 124 168 211 253'
RAMP['magma']['description']['blue'] = '4 50 104 126 130 126 115 98 93 115 149 191'
RAMP['plasma'] = {'author': ' Stefan van der Walt and Nathaniel Smith', 'comments': '', 'type': 'Diverging'}
RAMP['plasma']['description'] = {}
RAMP['plasma']['description']['red'] = '13 62 99 135 166 192 213 231 245 253 252 240'
RAMP['plasma']['description']['green'] = '8 4 0 7 32 58 84 111 140 173 210 249'
RAMP['plasma']['description']['blue'] = '135 156 167 166 152 131 110 90 70 50 37 33'
RAMP['inferno'] = {'author': ' Stefan van der Walt and Nathaniel Smith', 'comments': '', 'type': 'Diverging'}
RAMP['inferno']['description'] = {}
RAMP['inferno']['description']['red'] = '0 20 58 96 133 169 203 230 247 252 245 252'
RAMP['inferno']['description']['green'] = '0 11 9 19 33 46 65 93 131 173 219 255'
RAMP['inferno']['description']['blue'] = '4 53 99 110 107 94 73 47 17 18 75 164'
RAMP['cividis'] = {'author': ' Stefan van der Walt and Nathaniel Smith', 'comments': '', 'type': 'Diverging'}
RAMP['cividis']['description'] = {}
RAMP['cividis']['description']['red'] = '0 0 42 72 94 114 135 158 182 208 234 255'
RAMP['cividis']['description']['green'] = '32 48 64 82 98 115 132 150 169 190 211 234'
RAMP['cividis']['description']['blue'] = '77 111 108 107 110 116 121 119 113 103 87 70'
RAMP['rainbow'] = {'type': 'Sequential', 'description': {}}
RAMP['rainbow']['description']['red'] = '255 0 0'
RAMP['rainbow']['description']['green'] = '0 255 0'
RAMP['rainbow']['description']['blue'] = '0 0 255'
[docs]def HEXtoRGB(colorstring):
"""
Converts #RRGGBB to an (R, G, B) tuple,
from http://code.activestate.com/recipes/266466-html-colors-tofrom-rgb-tuples/
"""
colorstring = colorstring.strip()
if colorstring[0] == '#':
colorstring = colorstring[1:]
if len(colorstring) != 6:
msg = "Colour #%s is not in #RRGGBB format" % colorstring
raise viewererrors.ColorRampException(msg)
r, g, b = colorstring[:2], colorstring[2:4], colorstring[4:]
r, g, b = [int(n, 16) for n in (r, g, b)]
return (r, g, b)
[docs]def getRampsFromFile(fname):
"""
Read extra color ramps into our global RAMPS dictionary
from specified file
"""
from .minify_json import json_minify
# Read palette file
palettesFobj = open(fname, "r")
# Minify file contents
palsMinified = json_minify(palettesFobj.read())
# Loads palettes in a dict
pals = json.loads(palsMinified)
# For each palette that's been detected
for pal in pals:
# Check name is present and unique
if "name" in pal:
cur_name = pal["name"]
else:
# Quit - invalid colour scheme structure
msg = 'Invalid colour ramp structure'
raise viewererrors.ColorRampException(msg)
if cur_name in RAMP.keys():
# Quit - invalid colour scheme name
msg = 'Duplicated colour ramp name'
raise viewererrors.ColorRampException(msg)
# Check red, green and blue fields are present
if "description" in pal:
# Parsing description of the colour ramp
descriptionKeys = pal["description"].keys()
# Is it RGB coded?
isRgb = set(descriptionKeys).issubset(set(['red', 'green', 'blue']))
# Is it hex coded?
isHex = descriptionKeys == ['hex']
# If RGB is given it has priority
if isRgb:
reds = pal["description"]["red"]
greens = pal["description"]["green"]
blues = pal["description"]["blue"]
else:
# Else look for HEX
if isHex:
# Convert HEX to RGB
colrgb = numpy.array([HEXtoRGB(col) for col in pal["description"]["hex"].split()])
# Add RGB values to palette decription
reds = " ".join(map(str, colrgb[:, 0]))
greens = " ".join(map(str, colrgb[:, 1]))
blues = " ".join(map(str, colrgb[:, 2]))
else:
# Quit - invalid colour scheme
msg = 'Invalid colour ramp structure'
raise viewererrors.ColorRampException(msg)
else:
# Quit - invalid colour scheme
msg = 'Invalid colour ramp structure'
raise viewererrors.ColorRampException(msg)
# Other fields optional
# not sure what we'll be doing with this yet
if "author" in pal:
cur_author = pal["author"]
else:
cur_author = ''
if "comments" in pal:
cur_comments = pal["comments"]
else:
cur_comments = ''
if "type" in pal:
cur_type = pal["type"]
else:
cur_type = ''
# Assembling dictionary entry
RAMP[cur_name] = {'author': cur_author, 'comments': cur_comments, 'type': cur_type, 'description': {}}
# Add decsription fields
RAMP[cur_name]["description"]["red"] = reds
RAMP[cur_name]["description"]["green"] = greens
RAMP[cur_name]["description"]["blue"] = blues
[docs]def getRampsForDisplay():
"""
Returns a list of (name, displayname) tuples for
populating Combo boxes. Sorted by type
"""
# sort by stype
rampDict = {}
for key in RAMP:
rampType = RAMP[key]['type']
if rampType in rampDict:
rampDict[rampType].append(key)
else:
rampDict[rampType] = [key]
display = []
for rampType in rampDict:
for name in rampDict[rampType]:
title = "%s (%s)" % (name, rampType)
display.append((name, title))
# add our 'special' Random ramp
display.append((RANDOM_NAME, RANDOM_DISPLAY))
return display
[docs]def getLUTForRamp(code, name, lutsize):
"""
Returns an LUT for the specified color and name
"""
if name == RANDOM_NAME:
# just get some random lut data for this band
return numpy.random.randint(0, 255, size=lutsize, dtype=numpy.uint8)
else:
# interpolate the given coloramp over lutsize
# get the data as string
colstr = RAMP[name]['description'][code]
# turn it into a list of floats
# numpy.interp() needs floats
colList = [float(x) for x in colstr.split()]
# the x-values of the observations
# evenly spaced 0-255 with len(colList) obs
xobs = numpy.linspace(0, 255, len(colList))
# create an array from our list
yobs = numpy.array(colList)
# values to interpolate at 0-255
# same size as the lut
xinterp = numpy.linspace(0, 255, lutsize)
# do the interp
yinterp = numpy.interp(xinterp, xobs, yobs)
# return as 8 bit int
return yinterp.astype(numpy.uint8)