2341d510d3
Co-authored-by: pre-commit-ci-lite[bot] <117423508+pre-commit-ci-lite[bot]@users.noreply.github.com> Co-authored-by: Jonathan Swoboda <154711427+swoboda1337@users.noreply.github.com>
124 lines
3.8 KiB
Python
124 lines
3.8 KiB
Python
from esphome import automation
|
|
import esphome.codegen as cg
|
|
import esphome.config_validation as cv
|
|
from esphome.const import CONF_ID
|
|
from esphome.core import ID
|
|
|
|
from .defines import (
|
|
CONF_STYLE_DEFINITIONS,
|
|
CONF_THEME,
|
|
CONF_TOP_LAYER,
|
|
LValidator,
|
|
literal,
|
|
)
|
|
from .helpers import add_lv_use
|
|
from .lvcode import LambdaContext, LocalVariable, lv
|
|
from .schemas import ALL_STYLES, FULL_STYLE_SCHEMA, remap_property
|
|
from .types import ObjUpdateAction, lv_obj_t, lv_style_t
|
|
from .widgets import (
|
|
Widget,
|
|
add_widgets,
|
|
collect_parts,
|
|
set_obj_properties,
|
|
theme_widget_map,
|
|
wait_for_widgets,
|
|
)
|
|
from .widgets.obj import obj_spec
|
|
|
|
|
|
def has_style_props(config) -> bool:
|
|
return any(prop in config for prop in ALL_STYLES)
|
|
|
|
|
|
async def style_set(svar, style):
|
|
for prop, validator in ALL_STYLES.items():
|
|
if (value := style.get(prop)) is not None:
|
|
if isinstance(validator, LValidator):
|
|
value = await validator.process(value)
|
|
if isinstance(value, list):
|
|
value = "|".join(value)
|
|
lv.call(f"style_set_{remap_property(prop)}", svar, literal(value))
|
|
|
|
|
|
async def create_style(id_name, style=None):
|
|
style_id = ID(id_name, True, lv_style_t)
|
|
svar = cg.new_Pvariable(style_id)
|
|
lv.style_init(svar)
|
|
if style:
|
|
await style_set(svar, style)
|
|
return svar
|
|
|
|
|
|
class LVStyle:
|
|
"""
|
|
A class to lazily create a named style
|
|
"""
|
|
|
|
named_styles = {}
|
|
|
|
def __init__(self, id_name, style=None):
|
|
self.id_name = id_name
|
|
self.style = style
|
|
self._style_var = None
|
|
|
|
async def get_var(self):
|
|
if self._style_var is None:
|
|
self._style_var = await create_style(self.id_name + "_style", self.style)
|
|
return self._style_var
|
|
|
|
@classmethod
|
|
def get_style(cls, id_name):
|
|
return cls.named_styles.setdefault(id_name, LVStyle(id_name))
|
|
|
|
|
|
async def styles_to_code(config):
|
|
"""Convert styles to C__ code."""
|
|
for style in config.get(CONF_STYLE_DEFINITIONS, ()):
|
|
await create_style(style[CONF_ID].id, style)
|
|
|
|
|
|
@automation.register_action(
|
|
"lvgl.style.update",
|
|
ObjUpdateAction,
|
|
FULL_STYLE_SCHEMA.extend(
|
|
{
|
|
cv.Required(CONF_ID): cv.use_id(lv_style_t),
|
|
}
|
|
),
|
|
synchronous=True,
|
|
)
|
|
async def style_update_to_code(config, action_id, template_arg, args):
|
|
await wait_for_widgets()
|
|
style = await cg.get_variable(config[CONF_ID])
|
|
async with LambdaContext(parameters=args, where=action_id) as context:
|
|
await style_set(style, config)
|
|
|
|
return cg.new_Pvariable(action_id, template_arg, await context.get_lambda())
|
|
|
|
|
|
async def theme_to_code(config):
|
|
if theme := config.get(CONF_THEME):
|
|
add_lv_use(CONF_THEME)
|
|
for w_name, style in theme.items():
|
|
# Work around Python 3.10 bug with nested async comprehensions
|
|
# With Python 3.11 this could be simplified
|
|
# TODO: Now that we require Python 3.11+, this can be updated to use nested comprehensions
|
|
styles = {}
|
|
for part, states in collect_parts(style).items():
|
|
styles[part] = {
|
|
state: await create_style(
|
|
"_lv_theme_style_" + w_name + "_" + part + "_" + state, props
|
|
)
|
|
for state, props in states.items()
|
|
}
|
|
theme_widget_map[w_name] = styles
|
|
|
|
|
|
async def add_top_layer(lv_component, config):
|
|
top_layer = lv.disp_get_layer_top(lv_component.var.get_disp())
|
|
if top_conf := config.get(CONF_TOP_LAYER):
|
|
with LocalVariable("top_layer", lv_obj_t, top_layer) as top_layer_obj:
|
|
top_w = Widget(top_layer_obj, obj_spec, top_conf)
|
|
await set_obj_properties(top_w, top_conf)
|
|
await add_widgets(top_w, top_conf)
|