mirror of
https://github.com/esphome/esphome.git
synced 2025-03-19 00:58:13 +00:00
233 lines
7.6 KiB
C++
233 lines
7.6 KiB
C++
#include "image.h"
|
|
|
|
#include "esphome/core/hal.h"
|
|
|
|
namespace esphome {
|
|
namespace image {
|
|
|
|
void Image::draw(int x, int y, display::Display *display, Color color_on, Color color_off) {
|
|
switch (type_) {
|
|
case IMAGE_TYPE_BINARY: {
|
|
for (int img_x = 0; img_x < width_; img_x++) {
|
|
for (int img_y = 0; img_y < height_; img_y++) {
|
|
if (this->get_binary_pixel_(img_x, img_y)) {
|
|
display->draw_pixel_at(x + img_x, y + img_y, color_on);
|
|
} else if (!this->transparency_) {
|
|
display->draw_pixel_at(x + img_x, y + img_y, color_off);
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
case IMAGE_TYPE_GRAYSCALE:
|
|
for (int img_x = 0; img_x < width_; img_x++) {
|
|
for (int img_y = 0; img_y < height_; img_y++) {
|
|
const uint32_t pos = (img_x + img_y * this->width_);
|
|
const uint8_t gray = progmem_read_byte(this->data_start_ + pos);
|
|
Color color = Color(gray, gray, gray, 0xFF);
|
|
switch (this->transparency_) {
|
|
case TRANSPARENCY_CHROMA_KEY:
|
|
if (gray == 1) {
|
|
continue; // skip drawing
|
|
}
|
|
break;
|
|
case TRANSPARENCY_ALPHA_CHANNEL: {
|
|
auto on = (float) gray / 255.0f;
|
|
auto off = 1.0f - on;
|
|
// blend color_on and color_off
|
|
color = Color(color_on.r * on + color_off.r * off, color_on.g * on + color_off.g * off,
|
|
color_on.b * on + color_off.b * off, 0xFF);
|
|
break;
|
|
}
|
|
default:
|
|
break;
|
|
}
|
|
display->draw_pixel_at(x + img_x, y + img_y, color);
|
|
}
|
|
}
|
|
break;
|
|
case IMAGE_TYPE_RGB565:
|
|
for (int img_x = 0; img_x < width_; img_x++) {
|
|
for (int img_y = 0; img_y < height_; img_y++) {
|
|
auto color = this->get_rgb565_pixel_(img_x, img_y);
|
|
if (color.w >= 0x80) {
|
|
display->draw_pixel_at(x + img_x, y + img_y, color);
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
case IMAGE_TYPE_RGB:
|
|
for (int img_x = 0; img_x < width_; img_x++) {
|
|
for (int img_y = 0; img_y < height_; img_y++) {
|
|
auto color = this->get_rgb_pixel_(img_x, img_y);
|
|
if (color.w >= 0x80) {
|
|
display->draw_pixel_at(x + img_x, y + img_y, color);
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
Color Image::get_pixel(int x, int y, const Color color_on, const Color color_off) const {
|
|
if (x < 0 || x >= this->width_ || y < 0 || y >= this->height_)
|
|
return color_off;
|
|
switch (this->type_) {
|
|
case IMAGE_TYPE_BINARY:
|
|
if (this->get_binary_pixel_(x, y))
|
|
return color_on;
|
|
return color_off;
|
|
case IMAGE_TYPE_GRAYSCALE:
|
|
return this->get_grayscale_pixel_(x, y);
|
|
case IMAGE_TYPE_RGB565:
|
|
return this->get_rgb565_pixel_(x, y);
|
|
case IMAGE_TYPE_RGB:
|
|
return this->get_rgb_pixel_(x, y);
|
|
default:
|
|
return color_off;
|
|
}
|
|
}
|
|
#ifdef USE_LVGL
|
|
lv_img_dsc_t *Image::get_lv_img_dsc() {
|
|
// lazily construct lvgl image_dsc.
|
|
if (this->dsc_.data != this->data_start_) {
|
|
this->dsc_.data = this->data_start_;
|
|
this->dsc_.header.always_zero = 0;
|
|
this->dsc_.header.reserved = 0;
|
|
this->dsc_.header.w = this->width_;
|
|
this->dsc_.header.h = this->height_;
|
|
this->dsc_.data_size = this->get_width_stride() * this->get_height();
|
|
switch (this->get_type()) {
|
|
case IMAGE_TYPE_BINARY:
|
|
this->dsc_.header.cf = LV_IMG_CF_ALPHA_1BIT;
|
|
break;
|
|
|
|
case IMAGE_TYPE_GRAYSCALE:
|
|
this->dsc_.header.cf = LV_IMG_CF_ALPHA_8BIT;
|
|
break;
|
|
|
|
case IMAGE_TYPE_RGB:
|
|
#if LV_COLOR_DEPTH == 32
|
|
switch (this->transparent_) {
|
|
case TRANSPARENCY_ALPHA_CHANNEL:
|
|
this->dsc_.header.cf = LV_IMG_CF_TRUE_COLOR_ALPHA;
|
|
break;
|
|
case TRANSPARENCY_CHROMA_KEY:
|
|
this->dsc_.header.cf = LV_IMG_CF_TRUE_COLOR_CHROMA_KEYED;
|
|
break;
|
|
default:
|
|
this->dsc_.header.cf = LV_IMG_CF_TRUE_COLOR;
|
|
break;
|
|
}
|
|
#else
|
|
this->dsc_.header.cf =
|
|
this->transparency_ == TRANSPARENCY_ALPHA_CHANNEL ? LV_IMG_CF_RGBA8888 : LV_IMG_CF_RGB888;
|
|
#endif
|
|
break;
|
|
|
|
case IMAGE_TYPE_RGB565:
|
|
#if LV_COLOR_DEPTH == 16
|
|
switch (this->transparency_) {
|
|
case TRANSPARENCY_ALPHA_CHANNEL:
|
|
this->dsc_.header.cf = LV_IMG_CF_TRUE_COLOR_ALPHA;
|
|
break;
|
|
case TRANSPARENCY_CHROMA_KEY:
|
|
this->dsc_.header.cf = LV_IMG_CF_TRUE_COLOR_CHROMA_KEYED;
|
|
break;
|
|
default:
|
|
this->dsc_.header.cf = LV_IMG_CF_TRUE_COLOR;
|
|
break;
|
|
}
|
|
#else
|
|
this->dsc_.header.cf = this->transparent_ == TRANSPARENCY_ALPHA_CHANNEL ? LV_IMG_CF_RGB565A8 : LV_IMG_CF_RGB565;
|
|
#endif
|
|
break;
|
|
}
|
|
}
|
|
return &this->dsc_;
|
|
}
|
|
#endif // USE_LVGL
|
|
|
|
bool Image::get_binary_pixel_(int x, int y) const {
|
|
const uint32_t width_8 = ((this->width_ + 7u) / 8u) * 8u;
|
|
const uint32_t pos = x + y * width_8;
|
|
return progmem_read_byte(this->data_start_ + (pos / 8u)) & (0x80 >> (pos % 8u));
|
|
}
|
|
Color Image::get_rgb_pixel_(int x, int y) const {
|
|
const uint32_t pos = (x + y * this->width_) * this->bpp_ / 8;
|
|
Color color = Color(progmem_read_byte(this->data_start_ + pos + 0), progmem_read_byte(this->data_start_ + pos + 1),
|
|
progmem_read_byte(this->data_start_ + pos + 2), 0xFF);
|
|
|
|
switch (this->transparency_) {
|
|
case TRANSPARENCY_CHROMA_KEY:
|
|
if (color.g == 1 && color.r == 0 && color.b == 0) {
|
|
// (0, 1, 0) has been defined as transparent color for non-alpha images.
|
|
color.w = 0;
|
|
}
|
|
break;
|
|
case TRANSPARENCY_ALPHA_CHANNEL:
|
|
color.w = progmem_read_byte(this->data_start_ + (pos + 3));
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
return color;
|
|
}
|
|
Color Image::get_rgb565_pixel_(int x, int y) const {
|
|
const uint8_t *pos = this->data_start_ + (x + y * this->width_) * this->bpp_ / 8;
|
|
uint16_t rgb565 = encode_uint16(progmem_read_byte(pos), progmem_read_byte(pos + 1));
|
|
auto r = (rgb565 & 0xF800) >> 11;
|
|
auto g = (rgb565 & 0x07E0) >> 5;
|
|
auto b = rgb565 & 0x001F;
|
|
auto a = 0xFF;
|
|
switch (this->transparency_) {
|
|
case TRANSPARENCY_ALPHA_CHANNEL:
|
|
a = progmem_read_byte(pos + 2);
|
|
break;
|
|
case TRANSPARENCY_CHROMA_KEY:
|
|
if (rgb565 == 0x0020)
|
|
a = 0;
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
return Color((r << 3) | (r >> 2), (g << 2) | (g >> 4), (b << 3) | (b >> 2), a);
|
|
}
|
|
|
|
Color Image::get_grayscale_pixel_(int x, int y) const {
|
|
const uint32_t pos = (x + y * this->width_);
|
|
const uint8_t gray = progmem_read_byte(this->data_start_ + pos);
|
|
switch (this->transparency_) {
|
|
case TRANSPARENCY_CHROMA_KEY:
|
|
if (gray == 1)
|
|
return Color(0, 0, 0, 0);
|
|
return Color(gray, gray, gray, 0xFF);
|
|
case TRANSPARENCY_ALPHA_CHANNEL:
|
|
return Color(0, 0, 0, gray);
|
|
default:
|
|
return Color(gray, gray, gray, 0xFF);
|
|
}
|
|
}
|
|
int Image::get_width() const { return this->width_; }
|
|
int Image::get_height() const { return this->height_; }
|
|
ImageType Image::get_type() const { return this->type_; }
|
|
Image::Image(const uint8_t *data_start, int width, int height, ImageType type, Transparency transparency)
|
|
: width_(width), height_(height), type_(type), data_start_(data_start), transparency_(transparency) {
|
|
switch (this->type_) {
|
|
case IMAGE_TYPE_BINARY:
|
|
this->bpp_ = 1;
|
|
break;
|
|
case IMAGE_TYPE_GRAYSCALE:
|
|
this->bpp_ = 8;
|
|
break;
|
|
case IMAGE_TYPE_RGB565:
|
|
this->bpp_ = transparency == TRANSPARENCY_ALPHA_CHANNEL ? 24 : 16;
|
|
break;
|
|
case IMAGE_TYPE_RGB:
|
|
this->bpp_ = this->transparency_ == TRANSPARENCY_ALPHA_CHANNEL ? 32 : 24;
|
|
break;
|
|
}
|
|
}
|
|
|
|
} // namespace image
|
|
} // namespace esphome
|