mirror of https://github.com/esphome/esphome.git
58 lines
1.6 KiB
Python
58 lines
1.6 KiB
Python
from esphome import config_validation as cv
|
|
from esphome import codegen as cg
|
|
from esphome.const import CONF_BLUE, CONF_GREEN, CONF_ID, CONF_RED, CONF_WHITE
|
|
|
|
ColorStruct = cg.esphome_ns.struct("Color")
|
|
|
|
MULTI_CONF = True
|
|
|
|
CONF_RED_INT = "red_int"
|
|
CONF_GREEN_INT = "green_int"
|
|
CONF_BLUE_INT = "blue_int"
|
|
CONF_WHITE_INT = "white_int"
|
|
|
|
CONFIG_SCHEMA = cv.Schema(
|
|
{
|
|
cv.Required(CONF_ID): cv.declare_id(ColorStruct),
|
|
cv.Exclusive(CONF_RED, "red"): cv.percentage,
|
|
cv.Exclusive(CONF_RED_INT, "red"): cv.uint8_t,
|
|
cv.Exclusive(CONF_GREEN, "green"): cv.percentage,
|
|
cv.Exclusive(CONF_GREEN_INT, "green"): cv.uint8_t,
|
|
cv.Exclusive(CONF_BLUE, "blue"): cv.percentage,
|
|
cv.Exclusive(CONF_BLUE_INT, "blue"): cv.uint8_t,
|
|
cv.Exclusive(CONF_WHITE, "white"): cv.percentage,
|
|
cv.Exclusive(CONF_WHITE_INT, "white"): cv.uint8_t,
|
|
}
|
|
).extend(cv.COMPONENT_SCHEMA)
|
|
|
|
|
|
def to_code(config):
|
|
r = 0
|
|
if CONF_RED in config:
|
|
r = int(config[CONF_RED] * 255)
|
|
elif CONF_RED_INT in config:
|
|
r = config[CONF_RED_INT]
|
|
|
|
g = 0
|
|
if CONF_GREEN in config:
|
|
g = int(config[CONF_GREEN] * 255)
|
|
elif CONF_GREEN_INT in config:
|
|
g = config[CONF_GREEN_INT]
|
|
|
|
b = 0
|
|
if CONF_BLUE in config:
|
|
b = int(config[CONF_BLUE] * 255)
|
|
elif CONF_BLUE_INT in config:
|
|
b = config[CONF_BLUE_INT]
|
|
|
|
w = 0
|
|
if CONF_WHITE in config:
|
|
w = int(config[CONF_WHITE] * 255)
|
|
elif CONF_WHITE_INT in config:
|
|
w = config[CONF_WHITE_INT]
|
|
|
|
cg.variable(
|
|
config[CONF_ID],
|
|
cg.StructInitializer(ColorStruct, ("r", r), ("g", g), ("b", b), ("w", w)),
|
|
)
|