You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
397 lines
14 KiB
397 lines
14 KiB
from argparse import Namespace
|
|
import logging
|
|
import io
|
|
import requests
|
|
|
|
from esphome import core, automation
|
|
from esphome.components import display, font, time
|
|
import esphome.components.image as espImage
|
|
import esphome.config_validation as cv
|
|
import esphome.codegen as cg
|
|
from esphome.const import CONF_BLUE, CONF_GREEN, CONF_RED, CONF_RESIZE, CONF_FILE, CONF_ID, CONF_BRIGHTNESS, CONF_RAW_DATA_ID, CONF_TIME, CONF_TRIGGER_ID
|
|
from esphome.core import CORE, HexInt
|
|
from esphome.cpp_generator import RawExpression
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
DEPENDENCIES = ["display", "light", "api"]
|
|
AUTO_LOAD = ["ehmtxv2"]
|
|
IMAGE_TYPE_RGB565 = 4
|
|
MAXFRAMES = 110
|
|
MAXICONS = 90
|
|
ICONWIDTH = 8
|
|
ICONHEIGHT = 8
|
|
ICONBUFFERSIZE = ICONWIDTH * ICONHEIGHT * 4
|
|
SVG_ICONSTART = '<svg width="80px" height="80px" viewBox="0 0 80 80">'
|
|
SVG_FULL_SCREEN_START = '<svg width="320px" height="80px" viewBox="0 0 320 80">'
|
|
SVG_END = "</svg>"
|
|
|
|
logging.warning(f"")
|
|
logging.warning(f"This is a beta version of https://github.com/lubeda/EspHoMaTriXv2")
|
|
logging.warning(f"")
|
|
|
|
def rgb565_svg(x,y,r,g,b):
|
|
return f"<rect style=\"fill:rgb({(r << 3) | (r >> 2)},{(g << 2) | (g >> 4)},{(b << 3) | (b >> 2)});\" x=\"{x*10}\" y=\"{y*10}\" width=\"10\" height=\"10\"/>"
|
|
|
|
ehmtx_ns = cg.esphome_ns.namespace("esphome")
|
|
EHMTX_ = ehmtx_ns.class_("EHMTX", cg.Component)
|
|
Icons_ = ehmtx_ns.class_("EHMTX_Icon")
|
|
|
|
NextScreenTrigger = ehmtx_ns.class_(
|
|
"EHMTXNextScreenTrigger", automation.Trigger.template(cg.std_string)
|
|
)
|
|
|
|
IconErrorTrigger = ehmtx_ns.class_(
|
|
"EHMTXIconErrorTrigger", automation.Trigger.template(cg.std_string)
|
|
)
|
|
|
|
ExpiredScreenTrigger = ehmtx_ns.class_(
|
|
"EHMTXExpiredScreenTrigger", automation.Trigger.template(cg.std_string)
|
|
)
|
|
|
|
NextClockTrigger = ehmtx_ns.class_(
|
|
"EHMTXNextClockTrigger", automation.Trigger.template(cg.std_string)
|
|
)
|
|
|
|
AddScreenTrigger = ehmtx_ns.class_(
|
|
"EHMTXAddScreenTrigger", automation.Trigger.template(cg.std_string)
|
|
)
|
|
|
|
CONF_URL = "url"
|
|
CONF_FLAG = "flag"
|
|
CONF_CLOCKINTERVAL = "clock_interval"
|
|
CONF_TIMECOMPONENT = "time_component"
|
|
CONF_LAMEID = "lameid"
|
|
CONF_LIFETIME = "lifetime"
|
|
CONF_ICONS = "icons"
|
|
CONF_SHOWDOW = "show_dow"
|
|
CONF_SHOWDATE = "show_date"
|
|
CONF_FRAMEDURATION = "frame_duration"
|
|
CONF_SCROLLCOUNT = "scroll_count"
|
|
CONF_MATRIXCOMPONENT = "matrix_component"
|
|
CONF_HTML = "icons2html"
|
|
CONF_SCROLLINTERVAL = "scroll_interval"
|
|
CONF_RAINBOWINTERVAL = "rainbow_interval"
|
|
CONF_FRAMEINTERVAL = "frame_interval"
|
|
CONF_DEFAULT_FONT_ID = "default_font_id"
|
|
CONF_DEFAULT_FONT = "default_font"
|
|
CONF_DEFAULT_FONT_XOFFSET = "default_font_xoffset"
|
|
CONF_DEFAULT_FONT_YOFFSET = "default_font_yoffset"
|
|
CONF_special_FONT_ID = "special_font_id"
|
|
CONF_special_FONT_XOFFSET = "special_font_xoffset"
|
|
CONF_special_FONT_YOFFSET = "special_font_yoffset"
|
|
CONF_PINGPONG = "pingpong"
|
|
CONF_TIME_FORMAT = "time_format"
|
|
CONF_DATE_FORMAT = "date_format"
|
|
CONF_ON_NEXT_SCREEN = "on_next_screen"
|
|
CONF_ON_NEXT_CLOCK = "on_next_clock"
|
|
CONF_ON_ICON_ERROR = "on_icon_error"
|
|
CONF_ON_ADD_SCREEN = "on_add_screen"
|
|
CONF_ON_EXPIRED_SCREEN= "on_expired_screen"
|
|
CONF_SHOW_SECONDS = "show_seconds"
|
|
CONF_WEEK_START_MONDAY = "week_start_monday"
|
|
CONF_ICON = "icon_name"
|
|
CONF_TEXT = "text"
|
|
|
|
|
|
EHMTX_SCHEMA = cv.Schema({
|
|
cv.Required(CONF_ID): cv.declare_id(EHMTX_),
|
|
cv.Required(CONF_TIMECOMPONENT): cv.use_id(time),
|
|
cv.Required(CONF_MATRIXCOMPONENT): cv.use_id(display),
|
|
cv.Required(CONF_DEFAULT_FONT_ID): cv.use_id(font),
|
|
cv.Required(CONF_special_FONT_ID): cv.use_id(font),
|
|
cv.Optional(
|
|
CONF_CLOCKINTERVAL, default="0"
|
|
): cv.templatable(cv.positive_int),
|
|
cv.Optional(
|
|
CONF_HTML, default=False
|
|
): cv.boolean,
|
|
cv.Optional(
|
|
CONF_SHOW_SECONDS, default=False
|
|
): cv.boolean,
|
|
cv.Optional(
|
|
CONF_SHOWDATE, default=True
|
|
): cv.boolean,
|
|
cv.Optional(
|
|
CONF_WEEK_START_MONDAY, default=True
|
|
): cv.boolean,
|
|
cv.Optional(
|
|
CONF_SHOWDOW, default=True
|
|
): cv.boolean,
|
|
cv.Optional(
|
|
CONF_SHOWDATE, default=True
|
|
): cv.boolean,
|
|
cv.Optional(
|
|
CONF_TIME_FORMAT, default="%H:%M"
|
|
): cv.string,
|
|
cv.Optional(
|
|
CONF_DATE_FORMAT, default="%d.%m."
|
|
): cv.string,
|
|
cv.Optional(
|
|
CONF_DEFAULT_FONT_XOFFSET, default="1"
|
|
): cv.templatable(cv.int_range(min=-32, max=32)),
|
|
cv.Optional(
|
|
CONF_DEFAULT_FONT_YOFFSET, default="6"
|
|
): cv.templatable(cv.int_range(min=-32, max=32)),
|
|
cv.Optional(
|
|
CONF_special_FONT_XOFFSET, default="1"
|
|
): cv.templatable(cv.int_range(min=-32, max=32)),
|
|
cv.Optional(
|
|
CONF_special_FONT_YOFFSET, default="6"
|
|
): cv.templatable(cv.int_range(min=-32, max=32)),
|
|
cv.Optional(CONF_SCROLLINTERVAL, default="80"
|
|
): cv.templatable(cv.positive_int),
|
|
cv.Optional(CONF_RAINBOWINTERVAL, default="32"
|
|
): cv.templatable(cv.positive_int),
|
|
cv.Optional(CONF_SCROLLCOUNT, default="2"
|
|
): cv.templatable(cv.positive_int),
|
|
cv.Optional(
|
|
CONF_FRAMEINTERVAL, default="192"
|
|
): cv.templatable(cv.positive_int),
|
|
cv.Optional(CONF_BRIGHTNESS, default=80): cv.templatable(cv.int_range(min=0, max=255)),
|
|
cv.Optional(CONF_ON_NEXT_SCREEN): automation.validate_automation(
|
|
{
|
|
cv.GenerateID(CONF_TRIGGER_ID): cv.declare_id(NextScreenTrigger),
|
|
}
|
|
),
|
|
cv.Optional(CONF_ON_ICON_ERROR): automation.validate_automation(
|
|
{
|
|
cv.GenerateID(CONF_TRIGGER_ID): cv.declare_id(IconErrorTrigger),
|
|
}
|
|
),
|
|
cv.Optional(CONF_ON_ADD_SCREEN): automation.validate_automation(
|
|
{
|
|
cv.GenerateID(CONF_TRIGGER_ID): cv.declare_id(AddScreenTrigger),
|
|
}
|
|
),
|
|
cv.Optional(CONF_ON_EXPIRED_SCREEN): automation.validate_automation(
|
|
{
|
|
cv.GenerateID(CONF_TRIGGER_ID): cv.declare_id(NextScreenTrigger),
|
|
}
|
|
),
|
|
cv.Optional(CONF_ON_NEXT_CLOCK): automation.validate_automation(
|
|
{
|
|
cv.GenerateID(CONF_TRIGGER_ID): cv.declare_id(NextClockTrigger),
|
|
}
|
|
),
|
|
cv.Optional(CONF_ON_EXPIRED_SCREEN): automation.validate_automation(
|
|
{
|
|
cv.GenerateID(CONF_TRIGGER_ID): cv.declare_id(ExpiredScreenTrigger),
|
|
}
|
|
),
|
|
cv.Required(CONF_ICONS): cv.All(
|
|
cv.ensure_list(
|
|
{
|
|
cv.Required(CONF_ID): cv.declare_id(Icons_),
|
|
|
|
cv.Exclusive(CONF_FILE,"uri"): cv.file_,
|
|
cv.Exclusive(CONF_URL,"uri"): cv.url,
|
|
cv.Exclusive(CONF_LAMEID,"uri"): cv.string,
|
|
cv.Optional(CONF_RESIZE): cv.dimensions,
|
|
cv.Optional(
|
|
CONF_FRAMEDURATION, default="0"
|
|
): cv.templatable(cv.positive_int),
|
|
cv.Optional(
|
|
CONF_PINGPONG, default=False
|
|
): cv.boolean,
|
|
cv.GenerateID(CONF_RAW_DATA_ID): cv.declare_id(cg.uint8),
|
|
}
|
|
),
|
|
cv.Length(max=MAXICONS),
|
|
)})
|
|
|
|
CONFIG_SCHEMA = cv.All(font.validate_pillow_installed, EHMTX_SCHEMA)
|
|
|
|
CODEOWNERS = ["@lubeda"]
|
|
|
|
async def to_code(config):
|
|
|
|
from PIL import Image, ImageSequence
|
|
|
|
def openImageFile(path):
|
|
try:
|
|
return Image.open(path)
|
|
except Exception as e:
|
|
raise core.EsphomeError(f" ICONS: Could not load image file {path}: {e}")
|
|
|
|
def thumbnails(frames):
|
|
for frame in frames:
|
|
thumbnail = frame.copy()
|
|
thumbnail.thumbnail((32,8), Image.ANTIALIAS)
|
|
yield thumbnail
|
|
|
|
var = cg.new_Pvariable(config[CONF_ID])
|
|
|
|
logging.info(f"Preparing icons, this may take some seconds.")
|
|
|
|
html_string = F"<HTML><HEAD><TITLE>{CORE.config_path}</TITLE></HEAD>"
|
|
html_string += '''\
|
|
<STYLE>
|
|
svg { padding-top: 2x; padding-right: 2px; padding-bottom: 2px; padding-left: 2px; }
|
|
</STYLE><BODY>\
|
|
'''
|
|
|
|
yaml_string= ""
|
|
|
|
for conf in config[CONF_ICONS]:
|
|
|
|
if CONF_FILE in conf:
|
|
path = CORE.relative_config_path(conf[CONF_FILE])
|
|
try:
|
|
image = openImageFile(path)
|
|
except Exception as e:
|
|
raise core.EsphomeError(f" ICONS: Could not load image file {path}: {e}")
|
|
elif CONF_LAMEID in conf:
|
|
r = requests.get("https://developer.lametric.com/content/apps/icon_thumbs/" + conf[CONF_LAMEID], timeout=4.0)
|
|
if r.status_code != requests.codes.ok:
|
|
raise core.EsphomeError(f" ICONS: Could not download image file {conf[CONF_LAMEID]}: {conf[CONF_ID]}")
|
|
image = Image.open(io.BytesIO(r.content))
|
|
elif CONF_URL in conf:
|
|
r = requests.get(conf[CONF_URL], timeout=4.0)
|
|
if r.status_code != requests.codes.ok:
|
|
raise core.EsphomeError(f" ICONS: Could not download image file {conf[CONF_URL]}: {conf[CONF_ID]}")
|
|
image = Image.open(io.BytesIO(r.content))
|
|
|
|
width, height = image.size
|
|
|
|
if CONF_RESIZE in conf:
|
|
new_width_max, new_height_max = conf[CONF_RESIZE]
|
|
ratio = min(new_width_max / width, new_height_max / height)
|
|
width, height = int(width * ratio), int(height * ratio)
|
|
|
|
if hasattr(image, 'n_frames'):
|
|
frames = min(image.n_frames, MAXFRAMES)
|
|
else:
|
|
frames = 1
|
|
|
|
if ((width != 4*ICONWIDTH) or (width != ICONWIDTH)) and (height != ICONHEIGHT):
|
|
logging.warning(f" icon wrong size valid 8x8 or 8x32: {conf[CONF_ID]} skipped!")
|
|
else:
|
|
if (conf[CONF_FRAMEDURATION] == 0):
|
|
try:
|
|
duration = image.info['duration']
|
|
except:
|
|
duration = config[CONF_FRAMEINTERVAL]
|
|
else:
|
|
duration = conf[CONF_FRAMEDURATION]
|
|
|
|
html_string += F"<BR><B>{conf[CONF_ID]}</B> - ({duration} ms):<BR>"
|
|
yaml_string += F"\"{conf[CONF_ID]}\","
|
|
pos = 0
|
|
frameIndex = 0
|
|
html_string += f"<DIV ID={conf[CONF_ID]}>"
|
|
data = [0 for _ in range(ICONBUFFERSIZE * 2 * frames)]
|
|
for frameIndex in range(frames):
|
|
|
|
image.seek(frameIndex)
|
|
frame = image.convert("RGB")
|
|
if CONF_RESIZE in conf:
|
|
frame = frame.resize([width, height])
|
|
|
|
pixels = list(frame.getdata())
|
|
|
|
|
|
# width, height = image.size
|
|
if width == 8:
|
|
html_string += SVG_ICONSTART
|
|
else:
|
|
html_string += SVG_FULL_SCREEN_START
|
|
i = 0
|
|
for pix in pixels:
|
|
R = pix[0] >> 3
|
|
G = pix[1] >> 2
|
|
B = pix[2] >> 3
|
|
x = (i % width)
|
|
y = i//width
|
|
i +=1
|
|
rgb = (R << 11) | (G << 5) | B
|
|
html_string += rgb565_svg(x,y,R,G,B)
|
|
data[pos] = rgb >> 8
|
|
pos += 1
|
|
data[pos] = rgb & 255
|
|
pos += 1
|
|
html_string += SVG_END
|
|
html_string += f"</DIV>"
|
|
|
|
rhs = [HexInt(x) for x in data]
|
|
|
|
prog_arr = cg.progmem_array(conf[CONF_RAW_DATA_ID], rhs)
|
|
|
|
cg.new_Pvariable(
|
|
conf[CONF_ID],
|
|
prog_arr,
|
|
width,
|
|
height,
|
|
frames,
|
|
espImage.IMAGE_TYPE["RGB565"],
|
|
str(conf[CONF_ID]),
|
|
conf[CONF_PINGPONG],
|
|
duration,
|
|
)
|
|
|
|
cg.add(var.add_icon(RawExpression(str(conf[CONF_ID]))))
|
|
|
|
html_string += "</BODY></HTML>"
|
|
|
|
if config[CONF_HTML]:
|
|
try:
|
|
htmlfn = CORE.config_path.replace(".yaml","") + ".html"
|
|
with open(htmlfn, 'w') as f:
|
|
f.truncate()
|
|
f.write(html_string)
|
|
f.close()
|
|
logging.info(f"EsphoMaTrix: wrote html-file with icon preview: {htmlfn}")
|
|
|
|
except:
|
|
logging.warning(f"EsphoMaTrix: Error writing HTML file: {htmlfn}")
|
|
|
|
logging.info("List of icons for e.g. blueprint:\n\n\r["+yaml_string+"]\n")
|
|
|
|
disp = await cg.get_variable(config[CONF_MATRIXCOMPONENT])
|
|
cg.add(var.set_display(disp))
|
|
|
|
f = await cg.get_variable(config[CONF_DEFAULT_FONT_ID])
|
|
cg.add(var.set_default_font(f))
|
|
|
|
f = await cg.get_variable(config[CONF_special_FONT_ID])
|
|
cg.add(var.set_special_font(f))
|
|
|
|
ehmtxtime = await cg.get_variable(config[CONF_TIMECOMPONENT])
|
|
cg.add(var.set_clock(ehmtxtime))
|
|
|
|
cg.add(var.set_brightness(config[CONF_BRIGHTNESS]))
|
|
cg.add(var.set_scroll_interval(config[CONF_SCROLLINTERVAL]))
|
|
cg.add(var.set_rainbow_interval(config[CONF_SCROLLINTERVAL]))
|
|
cg.add(var.set_scroll_count(config[CONF_SCROLLCOUNT]))
|
|
cg.add(var.set_frame_interval(config[CONF_FRAMEINTERVAL]))
|
|
cg.add(var.set_week_start(config[CONF_WEEK_START_MONDAY]))
|
|
cg.add(var.set_clock_interval(config[CONF_CLOCKINTERVAL]))
|
|
cg.add(var.set_time_format(config[CONF_TIME_FORMAT]))
|
|
cg.add(var.set_date_format(config[CONF_DATE_FORMAT]))
|
|
cg.add(var.set_show_day_of_week(config[CONF_SHOWDOW]))
|
|
cg.add(var.set_show_date(config[CONF_SHOWDATE]))
|
|
cg.add(var.set_show_seconds(config[CONF_SHOW_SECONDS]))
|
|
cg.add(var.set_default_font_offset(config[CONF_DEFAULT_FONT_XOFFSET], config[CONF_DEFAULT_FONT_YOFFSET] ))
|
|
cg.add(var.set_special_font_offset(config[CONF_special_FONT_XOFFSET], config[CONF_special_FONT_YOFFSET] ))
|
|
for conf in config.get(CONF_ON_NEXT_SCREEN, []):
|
|
trigger = cg.new_Pvariable(conf[CONF_TRIGGER_ID], var)
|
|
await automation.build_automation(trigger, [(cg.std_string, "icon"), (cg.std_string, "text")], conf)
|
|
|
|
for conf in config.get(CONF_ON_NEXT_CLOCK, []):
|
|
trigger = cg.new_Pvariable(conf[CONF_TRIGGER_ID], var)
|
|
await automation.build_automation(trigger, [], conf)
|
|
|
|
for conf in config.get(CONF_ON_EXPIRED_SCREEN, []):
|
|
trigger = cg.new_Pvariable(conf[CONF_TRIGGER_ID], var)
|
|
await automation.build_automation(trigger, [(cg.std_string, "icon"), (cg.std_string, "text")] , conf)
|
|
|
|
for conf in config.get(CONF_ON_ICON_ERROR, []):
|
|
trigger = cg.new_Pvariable(conf[CONF_TRIGGER_ID], var)
|
|
await automation.build_automation(trigger, [(cg.std_string, "icon")] , conf)
|
|
|
|
for conf in config.get(CONF_ON_ADD_SCREEN, []):
|
|
trigger = cg.new_Pvariable(conf[CONF_TRIGGER_ID], var)
|
|
await automation.build_automation(trigger, [(cg.std_string, "icon"), (cg.uint8 , "mode")] , conf)
|
|
|
|
await cg.register_component(var, config)
|