view color_convert.py @ 23:7d9a056e29fe

esp rgbw output; cleanup
author drewp@bigasterisk.com
date Mon, 29 Jan 2024 23:41:27 -0800
parents b8201490c731
children ecbbf76318bb
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
    temp: 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 ikeaWhiteConv(c: Color) -> DeviceColor:
    return DeviceColor(brightness=max(c.r, c.g, c.b), temp=454)


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 brightnessConv(c: Color) -> DeviceColor:
    return DeviceColor(brightness=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), ww=0)


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