From 2d76b153bf7fbe90a93e50919e520657157d8691 Mon Sep 17 00:00:00 2001
From: Jesse Hills <3060199+jesserockz@users.noreply.github.com>
Date: Mon, 3 Mar 2025 07:47:09 +1300
Subject: [PATCH] [online_image] Allocate pngle manually to potentially use
 psram

---
 esphome/components/online_image/png_image.cpp | 18 ++++++++++++++++++
 esphome/components/online_image/png_image.h   |  8 +++++---
 2 files changed, 23 insertions(+), 3 deletions(-)

diff --git a/esphome/components/online_image/png_image.cpp b/esphome/components/online_image/png_image.cpp
index fc5fb554bf..54ff1fa6c3 100644
--- a/esphome/components/online_image/png_image.cpp
+++ b/esphome/components/online_image/png_image.cpp
@@ -40,6 +40,24 @@ static void draw_callback(pngle_t *pngle, uint32_t x, uint32_t y, uint32_t w, ui
   decoder->draw(x, y, w, h, color);
 }
 
+PngDecoder::PngDecoder(OnlineImage *image) : ImageDecoder(image) {
+  {
+    pngle_t *pngle = this->allocator_.allocate(1);
+    if (!pngle) {
+      ESP_LOGE(TAG, "Failed to allocate memory for PNGLE engine!");
+      return;
+    }
+    this->pngle_ = pngle;
+  }
+}
+
+PngDecoder::~PngDecoder() {
+  if (this->pngle_) {
+    pngle_reset(this->pngle_);
+    this->allocator_.deallocate(this->pngle_, 1);
+  }
+}
+
 int PngDecoder::prepare(size_t download_size) {
   ImageDecoder::prepare(download_size);
   if (!this->pngle_) {
diff --git a/esphome/components/online_image/png_image.h b/esphome/components/online_image/png_image.h
index 39f445c588..46519f8ef4 100644
--- a/esphome/components/online_image/png_image.h
+++ b/esphome/components/online_image/png_image.h
@@ -1,7 +1,8 @@
 #pragma once
 
-#include "image_decoder.h"
 #include "esphome/core/defines.h"
+#include "esphome/core/helpers.h"
+#include "image_decoder.h"
 #ifdef USE_ONLINE_IMAGE_PNG_SUPPORT
 #include <pngle.h>
 
@@ -18,13 +19,14 @@ class PngDecoder : public ImageDecoder {
    *
    * @param display The image to decode the stream into.
    */
-  PngDecoder(OnlineImage *image) : ImageDecoder(image), pngle_(pngle_new()) {}
-  ~PngDecoder() override { pngle_destroy(this->pngle_); }
+  PngDecoder(OnlineImage *image);
+  ~PngDecoder() override;
 
   int prepare(size_t download_size) override;
   int HOT decode(uint8_t *buffer, size_t size) override;
 
  protected:
+  RAMAllocator<pngle_t> allocator_;
   pngle_t *pngle_;
 };