Add handling of different color spaces

This commit is contained in:
sqozz 2024-03-02 02:01:13 +01:00
parent 82e2ceb841
commit 16ee0c9063
2 changed files with 42 additions and 13 deletions

View File

@ -3,12 +3,15 @@ import vu1
API = vu1.DialServer("http://localhost:5340", "my_secret_api_key")
dials = API.dials
# set image and value of specific dial by uid
# set color, image and value of specific dial by uid
dial = dials[dials.index("my_dial_uid")]
dial.image = "image_pack/gpu-load.png"
dial.value = 50
dial.color = vu1.VUColor(100, 100, 0) # as expected by the API (color between 0-100)
dial.color = vu1.RGBColor(1.0, 1.0, 0.0) # can be used to have colors as float
dial.color = vu1.HSVColor(0.16666666666666669, 1.0, 1.0) # same color as above (yellow) but written in HSV color space - userful for e.g. gradients
# set backlight to white on all available dials
for dial in dials:
dial.color = vu1.Color(100, 100, 100) #full white
dial.color = vu1.VUColor(100, 100, 100) # full white

46
vu1.py
View File

@ -72,7 +72,7 @@ class Dial():
self._fw_version = s["fw_version"]
self._hw_version = s["hw_version"]
self._protocol_version = s["protocol_version"]
self._color = Color(*s["backlight"].values())
self._color = VUColor(*s["backlight"].values())
self._image = s["image_file"]
@property
@ -137,15 +137,41 @@ class Dial():
file = {'imgfile': (open(new_image, 'rb'))}
self._server._req(f"{self.uid}/image/set", file, post=True)
class Color():
def __init__(self, red, green, blue):
self.red = max(0, min(100, red))
self.green = max(0, min(100, green))
self.blue = max(0, min(100, blue))
def hsv(self):
return colorsys.rgb_to_hsv(self.red, self.green, self.blue)
class VUColor():
def __init__(self, red=0, green=0, blue=0):
self._red = max(0, min(100, red))
self._green = max(0, min(100, green))
self._blue = max(0, min(100, blue))
def __iter__(self):
for attr in ["red", "green", "blue"]:
yield (attr, self.__dict__[attr])
yield (attr, self.__dict__[f"_{attr}"])
def to_rgb(self):
return RGBColor(self._red/100, self._green/100, self._blue/100)
def to_hsv(self):
return self.to_rgb().to_hsv()
class RGBColor(VUColor):
def __init__(self, red=0.0, green=0.0, blue=0.0):
self.red = max(0.0, min(1.0, red))
self.green = max(0.0, min(1.0, green))
self.blue = max(0.0, min(1.0, blue))
super().__init__(int(red*100), int(green*100), int(blue*100))
def to_hsv(self):
h, s, v = colorsys.rgb_to_hsv(self.red, self.green, self.blue)
return HSVColor(h, s, v)
class HSVColor(VUColor):
def __init__(self, hue=0.0, saturation=0.0, value=0.0):
self.hue = max(0.0, min(1.0, hue))
self.saturation = max(0.0, min(1.0, saturation))
self.value = max(0.0, min(1.0, value))
r, g, b = colorsys.hsv_to_rgb(self.hue, self.saturation, self.value)
super().__init__(int(r*100), int(g*100), int(b*100))
def to_rgb(self):
r, g, b = colorsys.hsv_to_rgb(self.hue, self.saturation, self.value)
return RGBColor(r, g, b)