#include <mbgl/gl/headless_backend.hpp>
#include <mbgl/gl/headless_display.hpp>
#include <mbgl/gl/context.hpp>
#include <mbgl/renderer/backend_scope.hpp>

#include <cassert>
#include <stdexcept>
#include <type_traits>

namespace mbgl {

class HeadlessBackend::View {
public:
    View(gl::Context& context, Size size)
        : color(context.createRenderbuffer<gl::RenderbufferType::RGBA>(size)),
          depthStencil(context.createRenderbuffer<gl::RenderbufferType::DepthStencil>(size)),
          framebuffer(context.createFramebuffer(color, depthStencil)) {
    }

    gl::Renderbuffer<gl::RenderbufferType::RGBA> color;
    gl::Renderbuffer<gl::RenderbufferType::DepthStencil> depthStencil;
    gl::Framebuffer framebuffer;
};

HeadlessBackend::HeadlessBackend(Size size_)
    : size(size_) {
}

HeadlessBackend::~HeadlessBackend() {
    BackendScope guard { *this };
    view.reset();
    context.reset();
}

void HeadlessBackend::activate() {
    active = true;

    if (!hasContext()) {
        if (!hasDisplay()) {
            throw std::runtime_error("Display is not set");
        }
        createContext();
    }

    assert(hasContext());
    impl->activateContext();
}

void HeadlessBackend::deactivate() {
    assert(hasContext());
    impl->deactivateContext();
    active = false;
}

void HeadlessBackend::bind() {
    gl::Context& context_ = getContext();

    if (!view) {
        view = std::make_unique<View>(context_, size);
    }

    context_.bindFramebuffer = view->framebuffer.framebuffer;
    context_.scissorTest = false;
    context_.viewport = { 0, 0, size };
}

void HeadlessBackend::updateAssumedState() {
    // no-op
}

void HeadlessBackend::setSize(Size size_) {
    size = size_;
    view.reset();
}

PremultipliedImage HeadlessBackend::readStillImage() {
    return getContext().readFramebuffer<PremultipliedImage>(size);
}

} // namespace mbgl