image_decoder.h

Go to the documentation of this file.

#pragma once
#include "esphome/core/color.h"
 
namespace esphome::runtime_image {
 
enum DecodeError : int {
  DECODE_ERROR_INVALID_TYPE = -1,
  DECODE_ERROR_UNSUPPORTED_FORMAT = -2,
  DECODE_ERROR_OUT_OF_MEMORY = -3,
};
 
class RuntimeImage;
 
class ImageDecoder {
 public:
  ImageDecoder(RuntimeImage *image) : image_(image) {}
  virtual ~ImageDecoder() = default;
 
  virtual int prepare(size_t expected_size) {
    this->expected_size_ = expected_size;
    return 0;
  }
 
  virtual int decode(uint8_t *buffer, size_t size) = 0;
 
  bool set_size(int width, int height);
 
  void draw(int x, int y, int w, int h, const Color &color);
 
  virtual bool is_finished() const {
    if (this->expected_size_ > 0) {
      return this->decoded_bytes_ >= this->expected_size_;
    }
    // If size is unknown, derived classes should override this
    return false;
  }
 
 protected:
  RuntimeImage *image_;
  size_t expected_size_ = 0;  // Expected data size (0 if unknown)
  size_t decoded_bytes_ = 0;  // Bytes processed so far
  double x_scale_ = 1.0;
  double y_scale_ = 1.0;
};
 
}  // namespace esphome::runtime_image