mirror of https://github.com/esphome/esphome.git
100 lines
3.6 KiB
Python
100 lines
3.6 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)
|