mirror of
https://github.com/esphome/esphome.git
synced 2024-12-11 07:54:54 +01:00
130 lines
3.8 KiB
Python
130 lines
3.8 KiB
Python
import esphome.codegen as cg
|
|
import esphome.config_validation as cv
|
|
from esphome import core, automation
|
|
from esphome.automation import maybe_simple_id
|
|
from esphome.const import CONF_ID, CONF_LAMBDA, CONF_PAGES, CONF_ROTATION
|
|
from esphome.core import coroutine, coroutine_with_priority
|
|
|
|
IS_PLATFORM_COMPONENT = True
|
|
|
|
display_ns = cg.esphome_ns.namespace("display")
|
|
DisplayBuffer = display_ns.class_("DisplayBuffer")
|
|
DisplayPage = display_ns.class_("DisplayPage")
|
|
DisplayPagePtr = DisplayPage.operator("ptr")
|
|
DisplayBufferRef = DisplayBuffer.operator("ref")
|
|
DisplayPageShowAction = display_ns.class_("DisplayPageShowAction", automation.Action)
|
|
DisplayPageShowNextAction = display_ns.class_(
|
|
"DisplayPageShowNextAction", automation.Action
|
|
)
|
|
DisplayPageShowPrevAction = display_ns.class_(
|
|
"DisplayPageShowPrevAction", automation.Action
|
|
)
|
|
|
|
DISPLAY_ROTATIONS = {
|
|
0: display_ns.DISPLAY_ROTATION_0_DEGREES,
|
|
90: display_ns.DISPLAY_ROTATION_90_DEGREES,
|
|
180: display_ns.DISPLAY_ROTATION_180_DEGREES,
|
|
270: display_ns.DISPLAY_ROTATION_270_DEGREES,
|
|
}
|
|
|
|
|
|
def validate_rotation(value):
|
|
value = cv.string(value)
|
|
if value.endswith("°"):
|
|
value = value[:-1]
|
|
return cv.enum(DISPLAY_ROTATIONS, int=True)(value)
|
|
|
|
|
|
BASIC_DISPLAY_SCHEMA = cv.Schema(
|
|
{
|
|
cv.Optional(CONF_LAMBDA): cv.lambda_,
|
|
}
|
|
)
|
|
|
|
FULL_DISPLAY_SCHEMA = BASIC_DISPLAY_SCHEMA.extend(
|
|
{
|
|
cv.Optional(CONF_ROTATION): validate_rotation,
|
|
cv.Optional(CONF_PAGES): cv.All(
|
|
cv.ensure_list(
|
|
{
|
|
cv.GenerateID(): cv.declare_id(DisplayPage),
|
|
cv.Required(CONF_LAMBDA): cv.lambda_,
|
|
}
|
|
),
|
|
cv.Length(min=1),
|
|
),
|
|
}
|
|
)
|
|
|
|
|
|
@coroutine
|
|
def setup_display_core_(var, config):
|
|
if CONF_ROTATION in config:
|
|
cg.add(var.set_rotation(DISPLAY_ROTATIONS[config[CONF_ROTATION]]))
|
|
if CONF_PAGES in config:
|
|
pages = []
|
|
for conf in config[CONF_PAGES]:
|
|
lambda_ = yield cg.process_lambda(
|
|
conf[CONF_LAMBDA], [(DisplayBufferRef, "it")], return_type=cg.void
|
|
)
|
|
page = cg.new_Pvariable(conf[CONF_ID], lambda_)
|
|
pages.append(page)
|
|
cg.add(var.set_pages(pages))
|
|
|
|
|
|
@coroutine
|
|
def register_display(var, config):
|
|
yield setup_display_core_(var, config)
|
|
|
|
|
|
@automation.register_action(
|
|
"display.page.show",
|
|
DisplayPageShowAction,
|
|
maybe_simple_id(
|
|
{
|
|
cv.Required(CONF_ID): cv.templatable(cv.use_id(DisplayPage)),
|
|
}
|
|
),
|
|
)
|
|
def display_page_show_to_code(config, action_id, template_arg, args):
|
|
var = cg.new_Pvariable(action_id, template_arg)
|
|
if isinstance(config[CONF_ID], core.Lambda):
|
|
template_ = yield cg.templatable(config[CONF_ID], args, DisplayPagePtr)
|
|
cg.add(var.set_page(template_))
|
|
else:
|
|
paren = yield cg.get_variable(config[CONF_ID])
|
|
cg.add(var.set_page(paren))
|
|
yield var
|
|
|
|
|
|
@automation.register_action(
|
|
"display.page.show_next",
|
|
DisplayPageShowNextAction,
|
|
maybe_simple_id(
|
|
{
|
|
cv.Required(CONF_ID): cv.templatable(cv.use_id(DisplayBuffer)),
|
|
}
|
|
),
|
|
)
|
|
def display_page_show_next_to_code(config, action_id, template_arg, args):
|
|
paren = yield cg.get_variable(config[CONF_ID])
|
|
yield cg.new_Pvariable(action_id, template_arg, paren)
|
|
|
|
|
|
@automation.register_action(
|
|
"display.page.show_previous",
|
|
DisplayPageShowPrevAction,
|
|
maybe_simple_id(
|
|
{
|
|
cv.Required(CONF_ID): cv.templatable(cv.use_id(DisplayBuffer)),
|
|
}
|
|
),
|
|
)
|
|
def display_page_show_previous_to_code(config, action_id, template_arg, args):
|
|
paren = yield cg.get_variable(config[CONF_ID])
|
|
yield cg.new_Pvariable(action_id, template_arg, paren)
|
|
|
|
|
|
@coroutine_with_priority(100.0)
|
|
def to_code(config):
|
|
cg.add_global(display_ns.using)
|