view color_convert.py @ 20:24a574108365

more protocols; bugs in setColor
author drewp@bigasterisk.com
date Mon, 29 Jan 2024 11:52:43 -0800
parents 61d4ccecfed8
children b8201490c731
line wrap: on
line source

import logging
from dataclasses import dataclass

from color import Color

log = logging.getLogger('conv')


@dataclass(frozen=True)
class DeviceColor:
    """neutral representation of the adjusted color that we send to a device"""
    r: float = 0
    g: float = 0
    b: float = 0
    w: float = 0
    x: float = 0
    y: float = 0
    cw: float = 0
    ww: float = 0
    brightness: float = 0

    def summary(self) -> dict:
        return dict([(k, round(v, 3)) for k, v in self.__dict__.items() if v > 0])

# fix this to send what z2m likes
def zbConv(c: Color) -> DeviceColor:
    return DeviceColor(r=c.r, g=c.g, b=c.b, brightness=max(c.r, c.g, c.b))

def oneWhiteConv(c: Color) -> DeviceColor:
    return DeviceColor(r=c.r, g=c.g, b=c.b, w=max(c.r, c.g, c.b))

def twoWhitesConv(c: Color) -> DeviceColor:
    return DeviceColor(r=c.r, g=c.g, b=c.b, cw=max(c.r, c.g, c.b))

def relayConv(c: Color) -> DeviceColor:
    return DeviceColor(brightness=1 if (max(c.r, c.g, c.b) > 0) else 0)