You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
igc2kmz/igc2kmz/color.py

103 lines
2.8 KiB
Python

# igc2kmz/color.py igc2kmz color functions
# Copyright (C) 2008 Tom Payne
#
# 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 3 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, see <http://www.gnu.org/licenses/>.
def hsl_to_rgba(hsl, a=1.0):
"""Convert a HSL tuple into a RGBA tuple."""
def h_to_value(p, q, t):
if t < 0.0:
t += 1.0
elif 1.0 < t:
t -= 1.0
if t < 1.0 / 6.0:
return p + 6.0 * (q - p) * t
elif t < 0.5:
return q
elif t < 2.0 / 3.0:
return p + 6.0 * (q - p) * (2.0 / 3.0 - t)
else:
return p
h, s, l = hsl
if s == 0:
return (l, l, l, a)
if l < 0.5:
q = l * (s + 1.0)
else:
q = l + s - l * s
p = 2.0 * l - q
r = h_to_value(p, q, h + 1.0 / 3.0)
g = h_to_value(p, q, h)
b = h_to_value(p, q, h - 1.0 / 3.0)
return (r, g, b, a)
def hsv_to_rgb(hsv):
"""Convert a HSV tuple into a RGBA tuple."""
h, s, v = hsv
hi = int(h)
f = h - hi
p = v * (1.0 - f)
q = v * (1.0 - f * s)
t = v * (1.0 - (1.0 - f) * s)
if hi == 0:
return (v, t, p)
elif hi == 1:
return (q, v, p)
elif hi == 2:
return (p, v, t)
elif hi == 3:
return (p, q, v)
elif hi == 4:
return (t, p, v)
else:
return (v, p, q)
def grayscale_gradient(value):
"""Return a gradient from black to white."""
if value < 0.0:
return (0.0, 0.0, 0.0, 1.0)
elif 1.0 <= value:
return (1.0, 1.0, 1.0, 1.0)
else:
return (value, value, value, 1.0)
def default_gradient(value):
"""Return a gradient from blue to green to red."""
if value < 0.0:
return hsl_to_rgba((2.0 / 3.0, 1.0, 0.5))
elif 1.0 <= value:
return hsl_to_rgba((0.0, 1.0, 0.5))
else:
h = 2.0 * (1.0 - value) / 3.0
return hsl_to_rgba((h, 1.0, 0.5))
def bilinear_gradient(value):
"""Return a bilinear gradient from blue to green to red."""
if value < 0.0:
h = 2.0 / 3.0
elif value < 0.5:
h = (6.0 - 4.0 * value) / 9.0
elif value == 0.5:
h = 1.0 / 3.0
elif value < 1.0:
h = (4.0 - 4.0 * value) / 9.0
else:
h = 0.0
return hsl_to_rgba((h, 1.0, 0.5))