mirror of https://github.com/esphome/esphome.git
62 lines
2.0 KiB
Python
62 lines
2.0 KiB
Python
import esphome.codegen as cg
|
|
import esphome.config_validation as cv
|
|
from esphome.components import output
|
|
from esphome.const import CONF_ID, CONF_LAMBDA, CONF_OUTPUTS, CONF_TYPE, CONF_BINARY
|
|
from .. import custom_ns
|
|
|
|
CustomBinaryOutputConstructor = custom_ns.class_("CustomBinaryOutputConstructor")
|
|
CustomFloatOutputConstructor = custom_ns.class_("CustomFloatOutputConstructor")
|
|
|
|
CONF_FLOAT = "float"
|
|
|
|
CONFIG_SCHEMA = cv.typed_schema(
|
|
{
|
|
CONF_BINARY: cv.Schema(
|
|
{
|
|
cv.GenerateID(): cv.declare_id(CustomBinaryOutputConstructor),
|
|
cv.Required(CONF_LAMBDA): cv.returning_lambda,
|
|
cv.Required(CONF_OUTPUTS): cv.ensure_list(
|
|
output.BINARY_OUTPUT_SCHEMA.extend(
|
|
{
|
|
cv.GenerateID(): cv.declare_id(output.BinaryOutput),
|
|
}
|
|
)
|
|
),
|
|
}
|
|
),
|
|
CONF_FLOAT: cv.Schema(
|
|
{
|
|
cv.GenerateID(): cv.declare_id(CustomFloatOutputConstructor),
|
|
cv.Required(CONF_LAMBDA): cv.returning_lambda,
|
|
cv.Required(CONF_OUTPUTS): cv.ensure_list(
|
|
output.FLOAT_OUTPUT_SCHEMA.extend(
|
|
{
|
|
cv.GenerateID(): cv.declare_id(output.FloatOutput),
|
|
}
|
|
)
|
|
),
|
|
}
|
|
),
|
|
},
|
|
lower=True,
|
|
)
|
|
|
|
|
|
def to_code(config):
|
|
type = config[CONF_TYPE]
|
|
if type == "binary":
|
|
ret_type = output.BinaryOutputPtr
|
|
klass = CustomBinaryOutputConstructor
|
|
else:
|
|
ret_type = output.FloatOutputPtr
|
|
klass = CustomFloatOutputConstructor
|
|
template_ = yield cg.process_lambda(
|
|
config[CONF_LAMBDA], [], return_type=cg.std_vector.template(ret_type)
|
|
)
|
|
|
|
rhs = klass(template_)
|
|
custom = cg.variable(config[CONF_ID], rhs)
|
|
for i, conf in enumerate(config[CONF_OUTPUTS]):
|
|
out = cg.Pvariable(conf[CONF_ID], custom.get_output(i))
|
|
yield output.register_output(out, conf)
|