mirror of https://github.com/esphome/esphome.git
47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
import esphome.codegen as cg
|
|
import esphome.config_validation as cv
|
|
from esphome.components import light, power_supply
|
|
from esphome.const import CONF_OUTPUT_ID, CONF_NUM_LEDS, CONF_RGB_ORDER, CONF_MAX_REFRESH_RATE, \
|
|
CONF_POWER_SUPPLY
|
|
from esphome.core import coroutine
|
|
|
|
fastled_base_ns = cg.esphome_ns.namespace('fastled_base')
|
|
FastLEDLightOutput = fastled_base_ns.class_('FastLEDLightOutput', cg.Component,
|
|
light.AddressableLight)
|
|
|
|
RGB_ORDERS = [
|
|
'RGB',
|
|
'RBG',
|
|
'GRB',
|
|
'GBR',
|
|
'BRG',
|
|
'BGR',
|
|
]
|
|
|
|
BASE_SCHEMA = light.ADDRESSABLE_LIGHT_SCHEMA.extend({
|
|
cv.GenerateID(CONF_OUTPUT_ID): cv.declare_id(FastLEDLightOutput),
|
|
|
|
cv.Required(CONF_NUM_LEDS): cv.positive_not_null_int,
|
|
cv.Optional(CONF_RGB_ORDER): cv.one_of(*RGB_ORDERS, upper=True),
|
|
cv.Optional(CONF_MAX_REFRESH_RATE): cv.positive_time_period_microseconds,
|
|
|
|
cv.Optional(CONF_POWER_SUPPLY): cv.use_id(power_supply.PowerSupply),
|
|
}).extend(cv.COMPONENT_SCHEMA)
|
|
|
|
|
|
@coroutine
|
|
def new_fastled_light(config):
|
|
var = cg.new_Pvariable(config[CONF_OUTPUT_ID])
|
|
yield cg.register_component(var, config)
|
|
|
|
if CONF_MAX_REFRESH_RATE in config:
|
|
cg.add(var.set_max_refresh_rate(config[CONF_MAX_REFRESH_RATE]))
|
|
|
|
if CONF_POWER_SUPPLY in config:
|
|
var_ = yield cg.get_variable(config[CONF_POWER_SUPPLY])
|
|
cg.add(var.set_power_supply(var_))
|
|
|
|
yield light.register_light(var, config)
|
|
cg.add_library('FastLED', '3.2.0')
|
|
yield var
|