/*
 * Copyright 2022 Rive
 */

#include "gm.hpp"

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

using namespace rivegm;

void GM::run(const char* name, std::vector<uint8_t>* pixels)
{
    TestingWindow::FrameOptions frameOptions = {
        .name = name,
        .clearColor = clearColor(),
        .triangulationThresholds = DeterministicTriangulationThresholds,
    };
    updateFrameOptions(&frameOptions);

    rive::gpu::RenderContext* renderContext = nullptr;
    auto previousMode = rive::gpu::ShaderCompilationMode::standard;
    if (frameOptions.shaderCompilationMode !=
        rive::gpu::ShaderCompilationMode::standard)
    {
        renderContext = TestingWindow::Get()->renderContext();
        if (renderContext != nullptr)
        {
            previousMode =
                renderContext->impl()->testingOnly_setShaderCompilationMode(
                    rive::gpu::ShaderCompilationMode::onlyUbershaders);
        }
    }

    auto renderer = TestingWindow::Get()->beginFrame(frameOptions);
    draw(renderer.get());
    TestingWindow::Get()->endFrame(pixels);

    if ((frameOptions.shaderCompilationMode !=
         rive::gpu::ShaderCompilationMode::standard) &&
        renderContext != nullptr)
    {
        renderContext->impl()->testingOnly_setShaderCompilationMode(
            previousMode);
    }
}

void GM::draw(rive::Renderer* renderer)
{
    renderer->save();
    this->onDraw(renderer);
    renderer->restore();
}