GxEPD2-esphome/components/gxepd2/display.py

"""ESPHome display platform driving a GxEPD2 panel via a native chunked
refresh engine (see cycle_engine.h) instead of GxEPD2's own C++ classes.

Panels are selected via `model:`, each backed by a hand-written (later,
extracted) EpdPanelDescriptor - see panel_213_z98c.h / panel_426_gdeq0426t82.h
/ panel_583_gdeq0583t31.h. catalog.json/vendor/arduino_compat stay in the
tree as reference material for panels not yet ported to this engine.
"""

import esphome.codegen as cg
import esphome.config_validation as cv
from esphome import pins
from esphome.components import display, spi
from esphome.const import (
    CONF_BUSY_PIN,
    CONF_CS_PIN,
    CONF_DC_PIN,
    CONF_ID,
    CONF_LAMBDA,
    CONF_MODEL,
    CONF_RESET_PIN,
    CONF_SPI_ID,
    Framework,
)
from esphome.types import ConfigType

CODEOWNERS = ["@zib"]
DEPENDENCIES = ["spi"]

gxepd2_ns = cg.esphome_ns.namespace("gxepd2")
GxEPD2Display = gxepd2_ns.class_("GxEPD2Display", display.DisplayBuffer)

# model name -> C++ descriptor global (EpdPanelDescriptor) declared in the
# matching panel_*.h header, which gxepd2_display.h includes unconditionally
# - so any of these names can be selected without extra include wiring here.
MODELS = {
    "GxEPD2_213_Z98C": "PANEL_213_Z98C",
    "GxEPD2_426_GDEQ0426T82": "PANEL_426_GDEQ0426T82",
    "GxEPD2_583_GDEQ0583T31": "PANEL_583_GDEQ0583T31",
}

CONFIG_SCHEMA = cv.All(
    display.FULL_DISPLAY_SCHEMA.extend(
        {
            cv.GenerateID(): cv.declare_id(GxEPD2Display),
            cv.GenerateID(CONF_SPI_ID): cv.use_id(spi.SPIComponent),
            cv.Required(CONF_MODEL): cv.one_of(*MODELS),
            cv.Required(CONF_CS_PIN): pins.gpio_output_pin_schema,
            cv.Required(CONF_DC_PIN): pins.gpio_output_pin_schema,
            cv.Optional(CONF_RESET_PIN): pins.gpio_output_pin_schema,
            cv.Optional(CONF_BUSY_PIN): pins.gpio_input_pin_schema,
        }
    ),
    cv.only_with_framework(Framework.ESP_IDF),
)


async def to_code(config: ConfigType) -> None:
    var = cg.new_Pvariable(config[CONF_ID])

    cg.add(var.set_descriptor(cg.RawExpression(f"&esphome::gxepd2::{MODELS[config[CONF_MODEL]]}")))
    cg.add(var.set_cs_pin(await cg.gpio_pin_expression(config[CONF_CS_PIN])))
    cg.add(var.set_dc_pin(await cg.gpio_pin_expression(config[CONF_DC_PIN])))
    if reset_pin := config.get(CONF_RESET_PIN):
        cg.add(var.set_reset_pin(await cg.gpio_pin_expression(reset_pin)))
    if busy_pin := config.get(CONF_BUSY_PIN):
        cg.add(var.set_busy_pin(await cg.gpio_pin_expression(busy_pin)))

    spi_parent = await cg.get_variable(config[CONF_SPI_ID])
    cg.add(var.set_spi_parent(spi_parent))

    # display.register_display() only wires up `pages:` - a bare top-level
    # `lambda:` (what our CONFIG_SCHEMA actually supports) has to be
    # processed and attached via set_writer() by the platform itself, same
    # as every other display component's to_code() (see e.g. epaper_spi's
    # display.py). Missing this meant the configured lambda never ran at
    # all - the only thing ever drawn was auto_clear's default
    # fill(COLOR_OFF), which is why every real-hardware run came out solid
    # black regardless of any protocol-level change.
    if CONF_LAMBDA in config:
        lambda_ = await cg.process_lambda(config[CONF_LAMBDA], [(display.DisplayRef, "it")], return_type=cg.void)
        cg.add(var.set_writer(lambda_))

    await display.register_display(var, config)