You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
139 lines
5.3 KiB
139 lines
5.3 KiB
#!/usr/bin/env python3
|
|
#
|
|
# This script was used to work out how to map the RGB/brightness input
|
|
# data to GPIO pinout voltages.
|
|
|
|
maxV = 3.27
|
|
|
|
requirements = [
|
|
# Input # Expected output
|
|
(0, 0, 0, 1.00, False, None, None, None),
|
|
(255, 255, 255, 0.00, False, None, None, None),
|
|
(255, 0, 0, 0.01, True, 2.95, maxV, maxV),
|
|
(255, 0, 0, 1.00, True, 1.82, 3.27, 3.27),
|
|
(255, 255, 0, 0.01, True, 3.04, 2.86, 3.17),
|
|
(255, 255, 0, 1.00, True, 2.56, 1.77, 3.15),
|
|
(255, 255, 255, 0.01, True, 3.04, 2.86, 3.07),
|
|
(255, 255, 255, 1.00, True, 2.56, 1.77, 2.73),
|
|
(0, 255, 0, 0.01, True, 3.27, 2.95, 3.27),
|
|
(0, 255, 0, 1.00, True, 3.27, 1.82, 3.27),
|
|
(0, 255, 255, 0.01, True, 3.13, 2.86, 3.07),
|
|
(0, 255, 255, 1.00, True, 2.97, 1.76, 2.71),
|
|
(0, 0, 255, 0.01, True, 3.27, 3.27, 2.95),
|
|
(0, 0, 255, 1.00, True, 3.27, 3.27, 1.82),
|
|
(255, 0, 255, 0.01, True, 2.88, 3.12, 2.86),
|
|
(255, 0, 255, 1.00, True, 1.85, 2.92, 1.77),
|
|
(255, 128, 0, 0.01, True, 2.86, 2.93, 3.17),
|
|
(255, 128, 0, 1.00, True, 1.77, 2.10, 3.15),
|
|
(255, 0 ,128, 0.01, True, 2.86, 3.12, 3.10),
|
|
(255, 0 ,128, 1.00, True, 1.77, 2.94, 2.84),
|
|
(255, 128, 128, 0.01, True, 2.86, 2.93, 3.10),
|
|
(255, 128, 128, 1.00, True, 1.77, 2.10, 2.86),
|
|
(128, 255, 0, 0.01, True, 3.11, 2.86, 3.17),
|
|
(128, 255, 0, 1.00, True, 2.87, 1.76, 3.15),
|
|
(255, 128, 255, 0.01, True, 2.86, 2.93, 2.87),
|
|
(255, 128, 255, 1.00, True, 1.76, 2.06, 1.80),
|
|
]
|
|
|
|
def get_red(r, g, b, brightness):
|
|
if r == 0:
|
|
if g == 1 and b == 1:
|
|
return round(3.13 - (brightness * 100 - 1) * (3.13 - 2.97)/99, 2)
|
|
else:
|
|
return maxV;
|
|
if g == 1:
|
|
if r == 1:
|
|
return round((3.04 - (brightness * 100 - 1) * (3.04 - 2.56)/99), 2)
|
|
elif r > 0:
|
|
return round((3.11 - (brightness * 100 - 1) * (3.11 - 2.87)/99), 2)
|
|
elif g > 0:
|
|
if b == 1:
|
|
return round((2.86 - (brightness * 100 - 1) * (2.86 - 1.76)/99), 2)
|
|
else:
|
|
return round((2.86 - (brightness * 100 - 1) * (2.86 - 1.77)/99), 2)
|
|
elif b == 1:
|
|
return round((2.88 - (brightness * 100 - 1) * (2.88 - 1.85)/99), 2)
|
|
elif b > 0:
|
|
return round((2.86 - (brightness * 100 - 1) * (2.86 - 1.77)/99), 2)
|
|
else:
|
|
return round((2.95 - (brightness * 100 - 1) * (2.95 - 1.82)/99), 2)
|
|
raise Exception("Shouldn't get here")
|
|
|
|
def get_green(r, g, b, brightness):
|
|
if g == 0:
|
|
if r == 1 and b == 1:
|
|
return round(3.12 - (brightness * 100 - 1) * (3.12 - 2.92)/99, 2)
|
|
elif r == 1 and b > 0:
|
|
return round(3.12 - (brightness * 100 - 1) * (3.12 - 2.94)/99, 2)
|
|
else:
|
|
return maxV;
|
|
if r == 1 and g == 1:
|
|
return round((2.86 - (brightness * 100 - 1) * (2.86 - 1.77)/99), 2)
|
|
elif r == 1:
|
|
if b == 1:
|
|
return round((2.93 - (brightness * 100 - 1) * (2.93 - 2.06)/99), 2)
|
|
else:
|
|
return round((2.93 - (brightness * 100 - 1) * (2.93 - 2.10)/99), 2)
|
|
elif r > 0:
|
|
return round((2.86 - (brightness * 100 - 1) * (2.86 - 1.76)/99), 2)
|
|
elif r == 0 and b == 1:
|
|
return round((2.86 - (brightness * 100 - 1) * (2.86 - 1.76)/99), 2)
|
|
else:
|
|
return round((2.95 - (brightness * 100 - 1) * (2.95 - 1.82)/99), 2)
|
|
raise Exception("Shouldn't get here")
|
|
|
|
def get_blue(r, g, b, brightness):
|
|
if b == 0:
|
|
if r > 0 and g > 0:
|
|
return round((3.17 - (brightness * 100 - 1) * (3.17 - 3.15)/99), 2)
|
|
else:
|
|
return maxV;
|
|
if r == 1 and g == 1:
|
|
return round((3.07 - (brightness * 100 - 1) * (3.07 - 2.73)/99), 2)
|
|
elif r == 1 and g == 0:
|
|
if b == 1:
|
|
return round((2.86 - (brightness * 100 - 1) * (2.86 - 1.77)/99), 2)
|
|
elif b > 0:
|
|
return round((3.10 - (brightness * 100 - 1) * (3.10 - 2.84)/99), 2)
|
|
elif r == 1 and g > 0:
|
|
if b == 1:
|
|
return round((2.87 - (brightness * 100 - 1) * (2.87 - 1.80)/99), 2)
|
|
else:
|
|
return round((3.10 - (brightness * 100 - 1) * (3.10 - 2.86)/99), 2)
|
|
elif r == 0 and g == 1:
|
|
return round((3.07 - (brightness * 100 - 1) * (3.07 - 2.71)/99), 2)
|
|
else:
|
|
return round((2.95 - (brightness * 100 - 1) * (2.95 - 1.82)/99), 2)
|
|
raise Exception("Shouldn't get here")
|
|
|
|
def make_rgb_fractions(r, g, b):
|
|
""" These ought to be 0 to 1 on input already, but it won't
|
|
hurt to make sure they are."""
|
|
m = max(r, g, b)
|
|
r = r/m
|
|
g = g/m
|
|
b = b/m
|
|
return (r, g, b)
|
|
|
|
|
|
for r,g,b,brightness,stateExpected, rVexpected, gVexpected, bVexpected in requirements:
|
|
print(f"IN: {r}, {g}, {b} at {brightness} / ", end="")
|
|
print(f"EXP: ({stateExpected}) {rVexpected},{gVexpected},{bVexpected} ", end="")
|
|
|
|
state = True
|
|
if r + g + b == 0 or brightness == 0:
|
|
state = False
|
|
rV = None
|
|
gV = None
|
|
bV = None
|
|
else:
|
|
r, g, b = make_rgb_fractions(r, g, b)
|
|
rV = get_red(r, g, b, brightness)
|
|
gV = get_green(r, g, b, brightness)
|
|
bV = get_blue(r, g, b, brightness)
|
|
print(f"OUT: ({state}) {rV},{gV},{bV} ", end="")
|
|
if rV == rVexpected and gV == gVexpected and bV == bVexpected and state == stateExpected:
|
|
print(" [PASS]");
|
|
else:
|
|
print()
|
|
|