#include "renderer_gl/command_plan.h"
#include "test_harness.h"

#include <string_view>

namespace {

void maps_render_pass_and_state_commands(pp::tests::Harness& h)
{
    const auto begin = pp::renderer::gl::plan_recorded_render_command(pp::renderer::RecordedRenderCommand {
        .kind = pp::renderer::RecordedRenderCommandKind::begin_render_pass,
        .target_desc = pp::renderer::TextureDesc {
            .extent = pp::renderer::Extent2D { .width = 64U, .height = 32U },
            .format = pp::renderer::TextureFormat::rgba8,
        },
        .clear_color_enabled = true,
        .clear_color = pp::renderer::ClearColor { .r = 0.25F, .g = 0.5F, .b = 0.75F, .a = 1.0F },
        .clear_depth_enabled = true,
        .clear_depth = 0.875F,
        .clear_stencil_enabled = true,
        .clear_stencil = 3U,
    });
    const auto viewport = pp::renderer::gl::plan_recorded_render_command(pp::renderer::RecordedRenderCommand {
        .kind = pp::renderer::RecordedRenderCommandKind::set_viewport,
        .viewport = pp::renderer::Viewport { .x = 2, .y = 4, .width = 32U, .height = 16U },
    });
    const auto scissor = pp::renderer::gl::plan_recorded_render_command(pp::renderer::RecordedRenderCommand {
        .kind = pp::renderer::RecordedRenderCommandKind::set_scissor,
        .scissor = pp::renderer::ScissorRect { .enabled = true, .x = 3, .y = 5, .width = 8U, .height = 6U },
    });
    const auto blend = pp::renderer::gl::plan_recorded_render_command(pp::renderer::RecordedRenderCommand {
        .kind = pp::renderer::RecordedRenderCommandKind::set_blend_state,
        .blend_state = pp::renderer::BlendState {
            .enabled = true,
            .source_color = pp::renderer::BlendFactor::source_alpha,
            .destination_color = pp::renderer::BlendFactor::one_minus_source_alpha,
        },
    });
    const auto depth = pp::renderer::gl::plan_recorded_render_command(pp::renderer::RecordedRenderCommand {
        .kind = pp::renderer::RecordedRenderCommandKind::set_depth_state,
        .depth_state = pp::renderer::DepthState {
            .test_enabled = true,
            .write_enabled = true,
            .compare = pp::renderer::CompareOp::greater,
        },
    });

    PP_EXPECT(h, begin.supported);
    PP_EXPECT(h, begin.kind == pp::renderer::gl::OpenGlPlannedCommandKind::begin_render_pass);
    PP_EXPECT(h, begin.clear_mask == 0x00004500U);
    PP_EXPECT(h, begin.clear_values.color[2] == 0.75F);
    PP_EXPECT(h, begin.clear_values.depth == 0.875F);
    PP_EXPECT(h, begin.clear_values.stencil == 3U);
    PP_EXPECT(h, begin.texture_format.internal_format == 0x8058U);
    PP_EXPECT(h, !begin.requires_render_pass);
    PP_EXPECT(h, viewport.supported);
    PP_EXPECT(h, viewport.requires_render_pass);
    PP_EXPECT(h, viewport.viewport.x == 2);
    PP_EXPECT(h, viewport.viewport.width == 32);
    PP_EXPECT(h, scissor.supported);
    PP_EXPECT(h, scissor.scissor.enabled == 1U);
    PP_EXPECT(h, scissor.scissor.height == 6);
    PP_EXPECT(h, blend.supported);
    PP_EXPECT(h, blend.blend.enabled == 1U);
    PP_EXPECT(h, blend.blend.source_color_factor == 0x0302U);
    PP_EXPECT(h, depth.supported);
    PP_EXPECT(h, depth.depth.test_enabled == 1U);
    PP_EXPECT(h, depth.depth.compare_function == 0x0204U);
}

void maps_binding_draw_and_blit_commands(pp::tests::Harness& h)
{
    const auto texture = pp::renderer::gl::plan_recorded_render_command(pp::renderer::RecordedRenderCommand {
        .kind = pp::renderer::RecordedRenderCommandKind::bind_texture,
        .texture_desc = pp::renderer::TextureDesc {
            .extent = pp::renderer::Extent2D { .width = 16U, .height = 16U },
            .format = pp::renderer::TextureFormat::depth24_stencil8,
        },
    });
    const auto sampler = pp::renderer::gl::plan_recorded_render_command(pp::renderer::RecordedRenderCommand {
        .kind = pp::renderer::RecordedRenderCommandKind::bind_sampler,
        .sampler_desc = pp::renderer::SamplerDesc {
            .min_filter = pp::renderer::SamplerFilter::nearest,
            .mag_filter = pp::renderer::SamplerFilter::linear,
            .mip_filter = pp::renderer::SamplerFilter::linear,
            .address_u = pp::renderer::SamplerAddressMode::repeat,
            .address_v = pp::renderer::SamplerAddressMode::mirrored_repeat,
            .address_w = pp::renderer::SamplerAddressMode::clamp_to_border,
        },
    });
    const auto mesh = pp::renderer::gl::plan_recorded_render_command(pp::renderer::RecordedRenderCommand {
        .kind = pp::renderer::RecordedRenderCommandKind::bind_mesh,
        .mesh_desc = pp::renderer::MeshDesc { .topology = pp::renderer::PrimitiveTopology::triangle_strip },
    });
    const auto draw = pp::renderer::gl::plan_recorded_render_command(pp::renderer::RecordedRenderCommand {
        .kind = pp::renderer::RecordedRenderCommandKind::draw,
        .mesh_desc = pp::renderer::MeshDesc { .topology = pp::renderer::PrimitiveTopology::lines },
        .draw_desc = pp::renderer::DrawDesc { .vertex_count = 4U, .index_count = 2U },
    });
    const auto blit = pp::renderer::gl::plan_recorded_render_command(pp::renderer::RecordedRenderCommand {
        .kind = pp::renderer::RecordedRenderCommandKind::blit_render_target,
        .blit_filter = pp::renderer::BlitFilter::linear,
    });

    PP_EXPECT(h, texture.supported);
    PP_EXPECT(h, texture.requires_render_pass);
    PP_EXPECT(h, texture.texture_format.internal_format == 0x88F0U);
    PP_EXPECT(h, sampler.supported);
    PP_EXPECT(h, sampler.sampler.min_filter == 0x2702U);
    PP_EXPECT(h, sampler.sampler.wrap_t == 0x8370U);
    PP_EXPECT(h, mesh.supported);
    PP_EXPECT(h, mesh.primitive_mode == 0x0005U);
    PP_EXPECT(h, draw.supported);
    PP_EXPECT(h, draw.primitive_mode == 0x0001U);
    PP_EXPECT(h, draw.draw_vertex_count == 4U);
    PP_EXPECT(h, draw.draw_index_count == 2U);
    PP_EXPECT(h, blit.supported);
    PP_EXPECT(h, !blit.requires_render_pass);
    PP_EXPECT(h, blit.blit_filter.value == 0x2601U);
}

void rejects_unsupported_command_tokens(pp::tests::Harness& h)
{
    const auto bad_blend = pp::renderer::gl::plan_recorded_render_command(pp::renderer::RecordedRenderCommand {
        .kind = pp::renderer::RecordedRenderCommandKind::set_blend_state,
        .blend_state = pp::renderer::BlendState {
            .source_color = static_cast<pp::renderer::BlendFactor>(255U),
        },
    });
    const auto bad_depth = pp::renderer::gl::plan_recorded_render_command(pp::renderer::RecordedRenderCommand {
        .kind = pp::renderer::RecordedRenderCommandKind::set_depth_state,
        .depth_state = pp::renderer::DepthState {
            .compare = static_cast<pp::renderer::CompareOp>(255U),
        },
    });
    const auto bad_sampler = pp::renderer::gl::plan_recorded_render_command(pp::renderer::RecordedRenderCommand {
        .kind = pp::renderer::RecordedRenderCommandKind::bind_sampler,
        .sampler_desc = pp::renderer::SamplerDesc {
            .address_u = static_cast<pp::renderer::SamplerAddressMode>(255U),
        },
    });
    const auto bad_mesh = pp::renderer::gl::plan_recorded_render_command(pp::renderer::RecordedRenderCommand {
        .kind = pp::renderer::RecordedRenderCommandKind::draw,
        .mesh_desc = pp::renderer::MeshDesc {
            .topology = static_cast<pp::renderer::PrimitiveTopology>(255U),
        },
    });
    const auto bad_blit = pp::renderer::gl::plan_recorded_render_command(pp::renderer::RecordedRenderCommand {
        .kind = pp::renderer::RecordedRenderCommandKind::blit_render_target,
        .blit_filter = static_cast<pp::renderer::BlitFilter>(255U),
    });
    const auto unknown = pp::renderer::gl::plan_recorded_render_command(pp::renderer::RecordedRenderCommand {
        .kind = static_cast<pp::renderer::RecordedRenderCommandKind>(255U),
    });

    PP_EXPECT(h, !bad_blend.supported);
    PP_EXPECT(h, bad_blend.blend.source_color_factor == 0U);
    PP_EXPECT(h, !bad_depth.supported);
    PP_EXPECT(h, bad_depth.depth.compare_function == 0U);
    PP_EXPECT(h, !bad_sampler.supported);
    PP_EXPECT(h, bad_sampler.sampler.wrap_s == 0U);
    PP_EXPECT(h, !bad_mesh.supported);
    PP_EXPECT(h, bad_mesh.primitive_mode == 0U);
    PP_EXPECT(h, !bad_blit.supported);
    PP_EXPECT(h, bad_blit.blit_filter.value == 0U);
    PP_EXPECT(h, !unknown.supported);
    PP_EXPECT(h, unknown.kind == pp::renderer::gl::OpenGlPlannedCommandKind::unknown);
}

void names_planned_command_kinds(pp::tests::Harness& h)
{
    PP_EXPECT(h, pp::renderer::gl::planned_command_kind_name(
                     pp::renderer::gl::OpenGlPlannedCommandKind::begin_render_pass)
                     == std::string_view("begin_render_pass"));
    PP_EXPECT(h, pp::renderer::gl::planned_command_kind_name(
                     pp::renderer::gl::OpenGlPlannedCommandKind::passthrough)
                     == std::string_view("passthrough"));
    PP_EXPECT(h, pp::renderer::gl::planned_command_kind_name(
                     static_cast<pp::renderer::gl::OpenGlPlannedCommandKind>(255U))
                     == std::string_view("unknown"));
}

}

int main()
{
    pp::tests::Harness harness;
    harness.run("maps_render_pass_and_state_commands", maps_render_pass_and_state_commands);
    harness.run("maps_binding_draw_and_blit_commands", maps_binding_draw_and_blit_commands);
    harness.run("rejects_unsupported_command_tokens", rejects_unsupported_command_tokens);
    harness.run("names_planned_command_kinds", names_planned_command_kinds);
    return harness.finish();
}