/*
 * Copyright 2026 Rive
 *
 * Use of this source code is governed by a BSD-style license that can be
 * found in the LICENSE file.
 */

#include "gm.hpp"
#include "gmutils.hpp"
#include "common/testing_window.hpp"
#include "rive/renderer.hpp"
#include "rive/renderer/render_context.hpp"
#include "rive/renderer/render_context_impl.hpp"

#include "assets/nomoon.png.hpp"

using namespace rivegm;
using namespace rive;

// Renders the same scene twice, side by side: the LEFT cell through
// fully-featured ubershader pipelines and the RIGHT cell through specialized
// pipelines (via a render pass break between the two). The cells must match:
// a few LSB of rounding difference is expected on translucent content (the
// premultiplied and unmultiplied paint paths round differently), but any
// visible or structural divergence is an ubershader bug. The golden pins
// both halves, so a regression in either flavor diffs immediately.
//
// The scene is chosen to stress where the two pipeline flavors are most
// likely to diverge: ubershaders compile ENABLE_ADVANCED_BLEND, which flips
// the entire paint path to unmultiplied colors. So the scene leans on
// content whose premultiplied vs unmultiplied evaluation differs in rounding
// and codepath:
//   - translucent gradients fading to very low alpha (premultiply rounding),
//   - an image drawn at low opacity (the unmultiplied path round-trips image
//     texels through unmultiply/premultiply),
//   - an advanced blend (multiply) overlapping srcOver content (forces the
//     dst-read machinery alongside the plain path),
//   - thin strokes over the gradient (AA edge rounding).
//
// On backends that can't switch shader compilation mode at runtime, both
// cells render specialized and trivially match.
constexpr float CELL = 300.f;

static void draw_scene(Renderer* renderer, float ox)
{
    Factory* factory = TestingWindow::Get()->factory();
    constexpr float stops[2] = {0.f, 1.f};

    renderer->save();
    renderer->translate(ox, 0);

    // Opaque horizontal gradient base.
    AABB base = {20.f, 20.f, 280.f, 280.f};
    ColorInt baseColors[2] = {0xff2060c0, 0xffc06020};
    Paint basePaint;
    basePaint->shader(factory->makeLinearGradient(base.left(),
                                                  0,
                                                  base.right(),
                                                  0,
                                                  baseColors,
                                                  stops,
                                                  2));
    renderer->drawPath(PathBuilder::Rect(base), basePaint);

    // Translucent vertical gradient fading to near-zero alpha: the
    // premultiplied path rounds rgb*a per fragment, the unmultiplied path
    // premultiplies after coverage; low-alpha texels expose the difference.
    AABB overlay = {50.f, 50.f, 250.f, 250.f};
    ColorInt overlayColors[2] = {0xccff2020, 0x0dffff20};
    Paint overlayPaint;
    overlayPaint->shader(factory->makeLinearGradient(0,
                                                     overlay.top(),
                                                     0,
                                                     overlay.bottom(),
                                                     overlayColors,
                                                     stops,
                                                     2));
    renderer->drawPath(PathBuilder::Rect(overlay), overlayPaint);

    // Multiply square overlapping both gradients: advanced blend content in
    // the same scene as plain srcOver draws.
    Paint multiplyPaint;
    multiplyPaint->color(0xff90b0d0);
    multiplyPaint->blendMode(BlendMode::multiply);
    renderer->drawPath(PathBuilder::Rect({140.f, 80.f, 260.f, 200.f}),
                       multiplyPaint);

    // Image at low opacity: in the unmultiplied path, image texels round-trip
    // through unmultiply_rgb and a later premultiply.
    auto img = LoadImage(assets::nomoon_png());
    if (img != nullptr)
    {
        renderer->save();
        renderer->translate(60.f, 120.f);
        float scale = 140.f / std::max(img->width(), img->height());
        renderer->scale(scale, scale);
        renderer->drawImage(img.get(),
                            ImageSampler::LinearClamp(),
                            BlendMode::srcOver,
                            .35f);
        renderer->restore();
    }

    // Thin translucent strokes over everything: AA edge rounding.
    for (int i = 0; i < 3; ++i)
    {
        Paint strokePaint;
        strokePaint->style(RenderPaintStyle::stroke);
        strokePaint->thickness(1.5f);
        strokePaint->color(0x8020ff80);
        renderer->drawPath(PathBuilder::Circle(150.f, 150.f, 60.f + 25.f * i),
                           strokePaint);
    }

    renderer->restore();
}

class uber_parity_GM : public rivegm::GM
{
public:
    uber_parity_GM() : GM(600, 300) {}
    ColorInt clearColor() const override { return 0xff404040; }

    void updateFrameOptions(TestingWindow::FrameOptions* options) const override
    {
        // Makes GM::run restore the process-wide mode after endFrame(); the
        // per-half modes below override it in between.
        options->shaderCompilationMode =
            rive::gpu::ShaderCompilationMode::onlyUbershaders;
    }

    void onDraw(rive::Renderer* renderer) override
    {
        gpu::RenderContext* renderContext =
            TestingWindow::Get()->renderContext();
        if (renderContext == nullptr)
        {
            // No runtime mode switching on this backend; both cells render
            // the same way and trivially match.
            draw_scene(renderer, 0);
            draw_scene(renderer, CELL);
            return;
        }

        gpu::RenderContext::FrameDescriptor frameDescriptor =
            renderContext->frameDescriptor();
        frameDescriptor.loadAction = gpu::LoadAction::preserveRenderTarget;

        for (int i = 0; i < 2; ++i)
        {
            // Flushes the previous half (and initially the clear) under the
            // mode that was live when its draws were recorded.
            TestingWindow::Get()->flushPLSContext();
            renderContext->impl()->testingOnly_setShaderCompilationMode(
                i == 0 ? gpu::ShaderCompilationMode::onlyUbershaders
                       : gpu::ShaderCompilationMode::alwaysSynchronous);
            renderContext->beginFrame(frameDescriptor);
            draw_scene(renderer, i * CELL);
        }
        // endFrame() flushes the right half under alwaysSynchronous, then
        // GM::run restores the process-wide mode.
    }
};
GMREGISTER(uber_parity, return new uber_parity_GM)