/*
Crafter®.Graphics
Copyright (C) 2025 Catcrafts®
catcrafts.net

This library is free software; you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public
License version 3.0 as published by the Free Software Foundation;

This library is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
Lesser General Public License for more details.

You should have received a copy of the GNU Lesser General Public
License along with this library; if not, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA
*/

module;
#define STB_IMAGE_IMPLEMENTATION
#include "../lib/stb_image.h"
#include "../lib/stb_truetype.h"
export module Crafter.Graphics:RenderingElement;
import std;
import :Transform;
import :Font;
import :Types;
import :Image;
import :Window;

export namespace Crafter {
    enum class TextAlignment {
        Left,
        Center,
        Right
    };

    enum class TextOverflowMode {
        Clip,
        Wrap
    };

    enum class TextScaleMode {
        None,
        Font, 
        Element,
        Buffer
    };

    struct RenderElementScalingOwning {
        std::vector<Pixel_BU8_GU8_RU8_AU8> scalingBuffer;
        std::uint32_t bufferWidth;
        std::uint32_t bufferHeight;
        bool bufferUpdated = true;
        RenderElementScalingOwning() = default;
        RenderElementScalingOwning(std::uint32_t bufferWidth, std::uint32_t bufferHeight) : scalingBuffer(bufferWidth*bufferHeight), bufferWidth(bufferWidth), bufferHeight(bufferHeight) {
            
        }
    };

    struct RenderElementScalingNonOwning {
        Pixel_BU8_GU8_RU8_AU8* scalingBuffer;
        std::uint32_t bufferWidth;
        std::uint32_t bufferHeight;
        bool bufferUpdated = true;
        RenderElementScalingNonOwning() = default;
        RenderElementScalingNonOwning(Pixel_BU8_GU8_RU8_AU8* scalingBuffer, std::uint32_t bufferWidth, std::uint32_t bufferHeight) : scalingBuffer(scalingBuffer), bufferWidth(bufferWidth), bufferHeight(bufferHeight) {
            
        }
    };

    struct RenderElementRotating {
        std::uint32_t rotation;
        bool rotationUpdated = true;
        RenderElementRotating() = default;
        RenderElementRotating(std::uint32_t rotation) : rotation(rotation) {

        }
    };


    struct EmptyScalingBase {};
    struct EmptyRotatingBase {};

    template<bool Scaling, bool Owning>
    using ScalingBase =
        std::conditional_t<
            Scaling,
            std::conditional_t<Owning,
                RenderElementScalingOwning,
                RenderElementScalingNonOwning>,
            EmptyScalingBase
        >;

    template<bool Rotating>
    using RotatingBase =
        std::conditional_t<
            Rotating,
            RenderElementRotating,
            EmptyRotatingBase
        >;

    class RenderingElementBase : public Transform  {
    public:
        std::vector<Pixel_BU8_GU8_RU8_AU8> buffer;
        OpaqueType opaque;
        RenderingElementBase(Anchor anchor) : Transform(anchor) {
            scaled.width = 0;
        }
        RenderingElementBase(Anchor anchor, OpaqueType opaque) : Transform(anchor), opaque(opaque) {
            scaled.width = 0;
        }
    };

    template<bool Scaling, bool Owning, bool Rotating> requires ((!Rotating || Scaling) && (!Owning || Scaling))
    class RenderingElement : public RenderingElementBase, public ScalingBase<Scaling, Owning>, public RotatingBase<Rotating> {
    public:        
        RenderingElement() = default;
        RenderingElement(Anchor anchor, OpaqueType opaque) : RenderingElementBase(anchor, opaque) {

        }
        RenderingElement(Anchor anchor, OpaqueType opaque, std::uint32_t rotation) requires(Rotating) : RenderingElementBase(anchor, opaque), RotatingBase<Rotating>(rotation) {

        }

        RenderingElement(Anchor anchor, const std::string_view imagePath) : RenderingElementBase(anchor) {
            LoadImage(imagePath);
        }
        RenderingElement(Anchor anchor, const std::string_view imagePath, std::uint32_t rotation) requires(Rotating) : RenderingElementBase(anchor), RotatingBase<Rotating>(rotation) {
            LoadImage(imagePath);
        }

        RenderingElement(Anchor anchor, const std::string_view imagePath, OpaqueType opaque) : RenderingElementBase(anchor, opaque) {
            LoadImageNoOpaqueCheck(imagePath);
        }
        RenderingElement(Anchor anchor, const std::string_view imagePath, OpaqueType opaque, std::uint32_t rotation) requires(Rotating) : RenderingElementBase(anchor, opaque), RotatingBase<Rotating>(rotation) {
            LoadImageNoOpaqueCheck(imagePath);
        }

        RenderingElement(Anchor anchor, OpaqueType opaque, std::uint32_t bufferWidth, std::uint32_t bufferHeight, Pixel_BU8_GU8_RU8_AU8* scalingBuffer) requires(Scaling && !Owning) : RenderingElementBase(anchor, opaque), ScalingBase<Scaling, Owning>(bufferWidth, bufferHeight, scalingBuffer) {

        }
        RenderingElement(Anchor anchor, OpaqueType opaque, std::uint32_t bufferWidth, std::uint32_t bufferHeight, Pixel_BU8_GU8_RU8_AU8* scalingBuffer, std::uint32_t rotation) requires(Scaling && !Owning && Rotating) : RenderingElementBase(anchor, opaque), ScalingBase<Scaling, Owning>(bufferWidth, bufferHeight, scalingBuffer), RotatingBase<Rotating>(rotation) {

        }

        RenderingElement(Anchor anchor, OpaqueType opaque, Image& image) requires(Scaling && !Owning) : RenderingElementBase(anchor, opaque), ScalingBase<Scaling, Owning>(image.buffer.data(), image.width, image.height) {

        }
        RenderingElement(Anchor anchor, OpaqueType opaque, Image& image, std::uint32_t rotation) requires(Scaling && !Owning && Rotating) : RenderingElementBase(anchor, opaque), ScalingBase<Scaling, Owning>(image.buffer.data(), image.width, image.height), RotatingBase<Rotating>(rotation) {

        }

        RenderingElement(Anchor anchor, Image& image) requires(Scaling && !Owning) : RenderingElementBase(anchor, image.opaque), ScalingBase<Scaling, Owning>(image.buffer.data(), image.width, image.height) {

        }
        RenderingElement(Anchor anchor, Image& image, std::uint32_t rotation) requires(Scaling && !Owning && Rotating) : RenderingElementBase(anchor, image.opaque), ScalingBase<Scaling, Owning>(image.buffer.data(), image.width, image.height), RotatingBase<Rotating>(rotation) {

        }

        RenderingElement(Anchor anchor, OpaqueType opaque, std::uint32_t bufferWidth, std::uint32_t bufferHeight) requires(Owning) : RenderingElementBase(anchor, opaque), ScalingBase<Scaling, Owning>(bufferWidth, bufferHeight) {

        }
        RenderingElement(Anchor anchor, OpaqueType opaque, std::uint32_t bufferWidth, std::uint32_t bufferHeight, std::uint32_t rotation) requires(Owning && Rotating) : RenderingElementBase(anchor, opaque), ScalingBase<Scaling, Owning>(bufferWidth, bufferHeight) , RotatingBase<Rotating>(rotation) {

        }

        RenderingElement(RenderingElement&) = delete;
        RenderingElement& operator=(RenderingElement&) = delete;

        void ScaleNearestNeighbor() requires(Scaling) {
            for (std::uint32_t y = 0; y < scaled.height; y++) {
                std::uint32_t srcY = y * ScalingBase<true, Owning>::bufferHeight / scaled.height;
                for (std::uint32_t x = 0; x < scaled.width; x++) {
                    std::uint32_t srcX = x * ScalingBase<true, Owning>::bufferWidth / scaled.width;
                    buffer[y * scaled.width + x] = ScalingBase<true, Owning>::scalingBuffer[srcY * ScalingBase<true, Owning>::bufferWidth + srcX];
                }
            }
        }

        void ScaleRotating() requires(Scaling) {
            const double rad = (static_cast<double>(RotatingBase<true>::rotation) / static_cast<double>(std::numeric_limits<std::uint32_t>::max())) * 2.0 * std::numbers::pi;

            const std::uint32_t dstWidth  = scaled.width;
            const std::uint32_t dstHeight = scaled.height;

            const double c2 = std::abs(std::cos(rad));
            const double s2 = std::abs(std::sin(rad));

            const double rotatedWidth  = dstWidth  * c2 + dstHeight * s2;
            const double rotatedHeight = dstWidth  * s2 + dstHeight * c2;

            const std::uint32_t diffX = static_cast<std::uint32_t>(std::ceil((rotatedWidth  - dstWidth)  * 0.5));
            const std::uint32_t diffY = static_cast<std::uint32_t>(std::ceil((rotatedHeight - dstHeight) * 0.5));

            scaled.width += diffX + diffX;
            scaled.height += diffY + diffY;

            scaled.x -= diffX;
            scaled.y -= diffY;

            buffer.clear();
            buffer.resize(scaled.width * scaled.height);

            // Destination center
            const double dstCx = (static_cast<double>(scaled.width)  - 1.0) * 0.5;
            const double dstCy = (static_cast<double>(scaled.height) - 1.0) * 0.5;

            // Source center
            const double srcCx = (static_cast<double>(ScalingBase<true, Owning>::bufferWidth)  - 1.0) * 0.5;
            const double srcCy = (static_cast<double>(ScalingBase<true, Owning>::bufferHeight) - 1.0) * 0.5;

            const double c = std::cos(rad);
            const double s = std::sin(rad);

            // Scale factors (destination → source)
            const double scaleX = static_cast<double>(ScalingBase<true, Owning>::bufferWidth)  / dstWidth;
            const double scaleY = static_cast<double>(ScalingBase<true, Owning>::bufferHeight) / dstHeight;

            for (std::uint32_t yB = 0; yB < scaled.height; ++yB) {
                for (std::uint32_t xB = 0; xB < scaled.width; ++xB) {

                    // ---- Destination pixel relative to center ----
                    const double dx = (static_cast<double>(xB) - dstCx) * scaleX;
                    const double dy = (static_cast<double>(yB) - dstCy) * scaleY;

                    // ---- Inverse rotation ----
                    const double sx =  (c * dx - s * dy) + srcCx;
                    const double sy =  (s * dx + c * dy) + srcCy;

                    // ---- Nearest neighbour sampling ----
                    const std::int32_t srcX = static_cast<std::int32_t>(std::round(sx));
                    const std::int32_t srcY = static_cast<std::int32_t>(std::round(sy));

                    if (srcX >= 0 && srcX < ScalingBase<true, Owning>::bufferWidth && srcY >= 0 && srcY < ScalingBase<true, Owning>::bufferHeight) {
                        buffer[yB * scaled.width + xB] = ScalingBase<true, Owning>::scalingBuffer[srcY * ScalingBase<true, Owning>::bufferWidth + srcX];
                    }
                }
            }
        }
        
        void UpdatePosition(Window& window, ScaleData oldScale) {
            if constexpr(Scaling && !Rotating) {
                if(oldScale.width != scaled.width || oldScale.height != scaled.height) {
                    buffer.resize(scaled.width * scaled.height);
                    ScaleNearestNeighbor();
                    window.AddDirtyRect(oldScale);
                    window.AddDirtyRect(scaled);
                } else if(oldScale.x != scaled.x || oldScale.y != scaled.y) {
                    window.AddDirtyRect(oldScale);
                    window.AddDirtyRect(scaled);
                    if(ScalingBase<true, Owning>::bufferUpdated) {
                        ScaleNearestNeighbor();
                        ScalingBase<true, Owning>::bufferUpdated = false;
                    }
                } else if(ScalingBase<true, Owning>::bufferUpdated) {
                    ScaleNearestNeighbor();
                    ScalingBase<true, Owning>::bufferUpdated = false;
                }
            } else if constexpr(Rotating) {
                if(oldScale.width != scaled.width || oldScale.height != scaled.height) {
                    buffer.resize(scaled.width * scaled.height);
                    ScaleRotating();
                    window.AddDirtyRect(oldScale);
                    window.AddDirtyRect(scaled);
                } else if(oldScale.x != scaled.x || oldScale.y != scaled.y) {
                    window.AddDirtyRect(oldScale);
                    window.AddDirtyRect(scaled);
                    if(ScalingBase<true, Owning>::bufferUpdated || RotatingBase<true>::rotationUpdated) {
                        ScaleRotating();
                        ScalingBase<true, Owning>::bufferUpdated = false;
                        RotatingBase<true>::rotationUpdated = false;
                    }
                } else if(ScalingBase<true, Owning>::bufferUpdated || RotatingBase<true>::rotationUpdated) {
                    ScaleRotating();
                    ScalingBase<true, Owning>::bufferUpdated = false;
                    RotatingBase<true>::rotationUpdated = false;
                }
            } else {
                if(oldScale.width != scaled.width || oldScale.height != scaled.height) {
                    buffer.resize(scaled.width * scaled.height);
                    window.AddDirtyRect(oldScale);
                    window.AddDirtyRect(scaled);
                }
                if(oldScale.x != scaled.x || oldScale.y != scaled.y) {
                    window.AddDirtyRect(oldScale);
                    window.AddDirtyRect(scaled);
                }
            }
        }

        void UpdatePosition(Window& window) override {
            ScaleData oldScale = scaled;
            window.ScaleElement(*this);
            UpdatePosition(window, oldScale);
            for(Transform* child : children) {
                child->UpdatePosition(window, *this);
            }
        }

        void UpdatePosition(Window& window, Transform& parent) override {
            ScaleData oldScale = scaled;
            window.ScaleElement(*this, parent);
            UpdatePosition(window, oldScale);
            for(Transform* child : children) {
                child->UpdatePosition(window, *this);
            }
}

        void LoadImage(const std::string_view imagePath) {
            std::filesystem::path abs = std::filesystem::absolute(imagePath);
            int xSize;
            int ySize;
            unsigned char* bgData = stbi_load(abs.string().c_str(), &xSize, &ySize, nullptr, 4);

            if constexpr(Scaling && !Owning) {
                ScalingBase<true, false>::bufferUpdated = true;
            } else if constexpr(Scaling && Owning) {
                ScalingBase<true, true>::bufferWidth = xSize;
                ScalingBase<true, true>::bufferHeight = ySize;
                ScalingBase<true, true>::bufferUpdated = true;
                ScalingBase<true, Owning>::scalingBuffer.resize(xSize*ySize);
            } else {
                buffer.resize(xSize*ySize);
            }

            opaque = OpaqueType::FullyOpaque;

            if constexpr(Scaling) {
                for(std::uint32_t x = 0; x < xSize; x++) {
                    for(std::uint32_t y = 0; y < ySize; y++) {
                        std::uint32_t idx = (x*ySize+y)*4;
                        ScalingBase<true, Owning>::scalingBuffer[x*ySize+y].r = bgData[idx];
                        ScalingBase<true, Owning>::scalingBuffer[x*ySize+y].g = bgData[idx+1];
                        ScalingBase<true, Owning>::scalingBuffer[x*ySize+y].b = bgData[idx+2];
                        ScalingBase<true, Owning>::scalingBuffer[x*ySize+y].a = bgData[idx+3];
                    }
                }

                for(std::uint32_t i = 0; i < xSize*ySize; i++) {
                    if(ScalingBase<true, Owning>::scalingBuffer[i].a != 255) {
                        opaque = OpaqueType::SemiOpaque;
                        for(std::uint32_t i2 = 0; i2 < xSize*ySize; i2++) {
                            if(ScalingBase<true, Owning>::scalingBuffer[i2].a != 0 && ScalingBase<true, Owning>::scalingBuffer[i2].a != 255) {
                                opaque = OpaqueType::Transparent;
                                return;
                            }
                        }
                        return;
                    }
                }
            } else {
                for(std::uint32_t x = 0; x < xSize; x++) {
                    for(std::uint32_t y = 0; y < ySize; y++) {
                        std::uint32_t idx = (x*ySize+y)*4;
                        buffer[x*ySize+y].r = bgData[idx];
                        buffer[x*ySize+y].g = bgData[idx+1];
                        buffer[x*ySize+y].b = bgData[idx+2];
                        buffer[x*ySize+y].a = bgData[idx+3];
                    }
                }

                for(std::uint32_t i = 0; i < xSize*ySize; i++) {
                    if(buffer[i].a != 255) {
                        opaque = OpaqueType::SemiOpaque;
                        for(std::uint32_t i2 = 0; i2 < xSize*ySize; i2++) {
                            if(buffer[i2].a != 0 && buffer[i2].a != 255) {
                                opaque = OpaqueType::Transparent;
                                return;
                            }
                        }
                        return;
                    }
                }
            }
        }
        void LoadImageNoOpaqueCheck(const std::string_view imagePath) {
            std::filesystem::path abs = std::filesystem::absolute(imagePath);
            int xSize;
            int ySize;
            unsigned char* bgData = stbi_load(abs.string().c_str(), &xSize, &ySize, nullptr, 4);


            if constexpr(Scaling && !Owning) {
                ScalingBase<true, false>::bufferUpdated = true;
            } else if constexpr(Scaling && Owning) {
                ScalingBase<true, true>::bufferWidth = xSize;
                ScalingBase<true, true>::bufferHeight = ySize;
                ScalingBase<true, true>::bufferUpdated = true;
                ScalingBase<true, Owning>::scalingBuffer.resize(xSize*ySize);
            } else {
                buffer.resize(xSize*ySize);
            }


            if constexpr(Scaling) {
                for(std::uint32_t x = 0; x < xSize; x++) {
                    for(std::uint32_t y = 0; y < ySize; y++) {
                        std::uint32_t idx = (x*ySize+y)*4;
                        ScalingBase<true, Owning>::scalingBuffer[x*ySize+y].r = bgData[idx];
                        ScalingBase<true, Owning>::scalingBuffer[x*ySize+y].g = bgData[idx+1];
                        ScalingBase<true, Owning>::scalingBuffer[x*ySize+y].b = bgData[idx+2];
                        ScalingBase<true, Owning>::scalingBuffer[x*ySize+y].a = bgData[idx+3];
                    }
                }
            } else {
                for(std::uint32_t x = 0; x < xSize; x++) {
                    for(std::uint32_t y = 0; y < ySize; y++) {
                        std::uint32_t idx = (x*ySize+y)*4;
                        buffer[x*ySize+y].r = bgData[idx];
                        buffer[x*ySize+y].g = bgData[idx+1];
                        buffer[x*ySize+y].b = bgData[idx+2];
                        buffer[x*ySize+y].a = bgData[idx+3];
                    }
                }
            }
        }
        std::vector<std::string_view> ResizeText(Window& window, const std::string_view text, float size, Font& font, TextOverflowMode overflowMode = TextOverflowMode::Clip, TextScaleMode scaleMode = TextScaleMode::None, Transform* parent = nullptr) {
            float scale = stbtt_ScaleForPixelHeight(&font.font, size);
            int baseline = (int)(font.ascent * scale);
            
            std::vector<std::string_view> lines;
            std::string_view remaining = text;

            std::uint32_t lineHeight = (font.ascent - font.descent) * scale;
            
            if(overflowMode == TextOverflowMode::Clip) {
                while (!remaining.empty()) {
                    // Find next newline or end of string
                    auto newlinePos = remaining.find('\n');
                    if (newlinePos != std::string_view::npos) {
                        lines.emplace_back(remaining.substr(0, newlinePos));
                        remaining = remaining.substr(newlinePos + 1);
                    } else {
                        lines.emplace_back(remaining);
                        break;
                    }
                }
                std::uint32_t maxWidth = 0;

                for(const std::string_view line: lines) {
                    std::uint32_t lineWidth = 0;
                    for (const char c : line) {
                        int advance, lsb;
                        stbtt_GetCodepointHMetrics(&font.font, c, &advance, &lsb);
                        lineWidth += (int)(advance * scale);
                    }
                    if(lineWidth > maxWidth) {
                        maxWidth = lineWidth;
                    }
                }


                if(scaleMode == TextScaleMode::Element) {
                    std::int32_t logicalPerPixelY = anchor.height / scaled.height;
                    std::int32_t oldHeight = anchor.height;
                    std::int32_t logicalPerPixelX = anchor.width / scaled.width;
                    std::int32_t oldwidth = anchor.width;
                    anchor.height = lineHeight * logicalPerPixelY;
                    anchor.width = maxWidth * logicalPerPixelX;
                    if(oldHeight != anchor.height || oldwidth != anchor.width) {
                        if(parent) {
                            UpdatePosition(window, *parent);
                        } else {
                            UpdatePosition(window);
                        }
                    }
                } else if(scaleMode == TextScaleMode::Font) {
                    //todo
                } else if(scaleMode == TextScaleMode::Buffer) {
                    if constexpr(Scaling && Owning) {
                        std::uint32_t neededHeight = lines.size() * lineHeight;
                        if(neededHeight != ScalingBase<true, true>::bufferHeight || maxWidth != ScalingBase<true, true>::bufferWidth) {
                            ScalingBase<true, true>::bufferHeight = neededHeight;
                            ScalingBase<true, true>::bufferWidth = maxWidth;
                            ScalingBase<true, true>::bufferUpdated = true;
                            ScalingBase<true, Owning>::scalingBuffer.resize(neededHeight*maxWidth);
                        }
                    }
                } else {
                    if constexpr(Scaling) {
                        lines.resize(ScalingBase<true, Owning>::bufferHeight / lines.size());
                    } else {
                        lines.resize(scaled.height / lines.size());
                    }
                }
            } else {
                while (!remaining.empty()) {
                    std::string_view line;
                    auto newlinePos = remaining.find('\n');
                    if (newlinePos != std::string_view::npos) {
                        line = remaining.substr(0, newlinePos);
                        remaining = remaining.substr(newlinePos + 1);
                    } else {
                        line = remaining;
                        remaining = "";
                    }

                    std::uint32_t lineWidth = 0;
                    std::size_t lastWrapPos = 0; // position of last space that can be used to wrap
                    std::size_t startPos = 0;

                    for (std::size_t i = 0; i < line.size(); ++i) {
                        char c = line[i];

                        // get width of this character
                        int advance, lsb;
                        stbtt_GetCodepointHMetrics(&font.font, c, &advance, &lsb);
                        lineWidth += (std::uint32_t)(advance * scale);

                        // remember last space for wrapping
                        if (c == ' ') {
                            lastWrapPos = i;
                        }

                        // if line exceeds width, wrap
                        if (lineWidth > scaled.width) {
                            std::size_t wrapPos;
                            if (lastWrapPos > startPos) {
                                wrapPos = lastWrapPos; // wrap at last space
                            } else {
                                wrapPos = i; // no space, hard wrap
                            }

                            // push the line up to wrapPos
                            lines.push_back(line.substr(startPos, wrapPos - startPos));

                            // skip any spaces at the beginning of next line
                            startPos = wrapPos;
                            while (startPos < line.size() && line[startPos] == ' ') {
                                ++startPos;
                            }

                            // reset width and i
                            lineWidth = 0;
                            i = startPos - 1; // -1 because loop will increment i
                        }
                    }

                    // add the remaining part of the line
                    if (startPos < line.size()) {
                        lines.push_back(line.substr(startPos));
                    }
                }

                if(scaleMode == TextScaleMode::Element) {
                    std::int32_t logicalPerPixelY = anchor.height / scaled.height;
                    std::int32_t oldHeight = anchor.height;
                    anchor.height = lineHeight * logicalPerPixelY;
                    if(oldHeight != anchor.height) {
                        if(parent) {
                            UpdatePosition(window, *parent);
                        } else {
                            UpdatePosition(window);
                        }
                    }
                } else if(scaleMode == TextScaleMode::Font) {
                    //todo
                } else if(scaleMode == TextScaleMode::Buffer) {
                    if constexpr(Scaling && Owning) {
                        std::uint32_t neededHeight = lines.size() * lineHeight;
                        if(neededHeight != ScalingBase<true, true>::bufferHeight) {
                            ScalingBase<true, true>::bufferHeight = neededHeight;
                            ScalingBase<true, true>::bufferUpdated = true;
                            ScalingBase<true, Owning>::scalingBuffer.resize(neededHeight*ScalingBase<true, true>::bufferWidth);
                        }
                    }
                } else {
                    if constexpr(Scaling) {
                        lines.resize(ScalingBase<true, Owning>::bufferHeight / lines.size());
                    } else {
                        lines.resize(scaled.height / lines.size());
                    }
                }
            }

            return lines;
        }
        void RenderText(Window& window, std::span<const std::string_view> lines, float size, Pixel_BU8_GU8_RU8_AU8 color, Font& font, TextAlignment alignment = TextAlignment::Left, std::uint32_t offsetX = 0, std::uint32_t offsetY = 0) {     
            float scale = stbtt_ScaleForPixelHeight(&font.font, size);
            int baseline = (int)(font.ascent * scale);
            std::uint32_t lineHeight = (font.ascent - font.descent) * scale;
            std::uint32_t currentY = baseline;
            for(std::string_view line : lines) {

                std::uint32_t lineWidth = 0;
                for (const char c : line) {
                    int advance, lsb;
                    stbtt_GetCodepointHMetrics(&font.font, c, &advance, &lsb);
                    lineWidth += (int)(advance * scale);
                }

                std::uint32_t startX = 0;
                switch (alignment) {
                    case TextAlignment::Left:
                        startX = 0;
                        break;
                    case TextAlignment::Center:
                        startX = (scaled.width - lineWidth) / 2;
                        break;
                    case TextAlignment::Right:
                        startX = scaled.width - lineWidth;
                        break;
                }
                std::uint32_t x = startX;

                for (std::size_t i = 0; i < line.size(); ++i) {
                    int codepoint = line[i];
                    
                    int ax;
                    int lsb;
                    stbtt_GetCodepointHMetrics(&font.font, codepoint, &ax, &lsb);

                    int c_x1, c_y1, c_x2, c_y2;
                    stbtt_GetCodepointBitmapBox(&font.font, codepoint, scale, scale, &c_x1, &c_y1, &c_x2, &c_y2);

                    int w = c_x2 - c_x1;
                    int h = c_y2 - c_y1;

                    std::vector<unsigned char> bitmap(w * h);
                    stbtt_MakeCodepointBitmap(&font.font, bitmap.data(), w, h, w, scale, scale, codepoint);

                    // Only render characters that fit within the scaled bounds
                    for (int j = 0; j < h; j++) {
                        for (int i = 0; i < w; i++) {
                            int bufferX = x + i + c_x1 + offsetX;
                            int bufferY = currentY + j + c_y1 + offsetY;
                            
                            // Only draw pixels that are within our scaled buffer bounds
                            if constexpr(Scaling) {
                                if (bufferX >= 0 && bufferX < ScalingBase<true, Owning>::bufferWidth && bufferY >= 0 && bufferY < ScalingBase<true, Owning>::bufferHeight) {
                                    ScalingBase<true, Owning>::scalingBuffer[bufferY * ScalingBase<true, Owning>::bufferWidth + bufferX] = {color.r, color.g, color.b, bitmap[j * w + i]};
                                }
                            } else {
                                if (bufferX >= 0 && bufferX < (int)scaled.width && bufferY >= 0 && bufferY < (int)scaled.height) {
                                    buffer[bufferY * scaled.width + bufferX] = {color.r, color.g, color.b, bitmap[j * w + i]};
                                }
                            }
                        }
                    }

                    x += (int)(ax * scale);

                    if (i + 1 < line.size()) {
                        x += (int)stbtt_GetCodepointKernAdvance(&font.font, codepoint, line[i+1]);
                    }
                }
                currentY += lineHeight;
            }
        }
    };
}