#include "app.hpp"
#include "video.hpp"
extern "C" {
#include "plat.h"
#include "memory.h"
}
#include <string.h>

#define video_arena_size (1024 * 1024 * 16)
#define asset_arena_size (1024 * 1024 * 4)
#define per_frame_memory_size (1024 * 1024)

static float verts[] = {
	 0.5f,  0.5f, 1.0f, 0.0f, 0.0f,
	-0.5f,  0.5f, 0.0f, 1.0f, 0.0f,
	 0.0f, -0.5f, 0.0f, 0.0f, 1.0f
};

struct C2 : public App {
	Device* dev;
	void on_resize() override {
		dev->on_resize();
	}
};

static Buffer_Id upload_verts(Device* dev) {
	Buffer_Id stage = dev->create_buffer(
		sizeof verts,
		Buffer_Flags::copy_src |
		Buffer_Flags::cpu_readwrite
	);
	Buffer_Id vbo = dev->create_buffer(
		sizeof verts,
		Buffer_Flags::vertex_buffer |
		Buffer_Flags::copy_dst
	);
	{
		void* mem;
		mem = dev->map_buffer(stage, 0, sizeof verts);
		memcpy(mem, verts, sizeof verts);
		dev->unmap_buffer(stage);
	}
	{
		Context& ctx = dev->acquire();
		ctx.copy(vbo, stage);
		dev->submit(ctx);
	}
	dev->destroy_buffer(stage);
	return vbo;
}

int main() {
	Arena video_arena;
	Arena asset_arena;
	Asset_Arena assets;
	Device* dev;
	Shader* shader;
	Texture* texture;
	Buffer_Id vbo;
	C2* app = App::create<C2>("c2");
	void* per_frame;
	app->running = 1;
	init_arena(
		&video_arena,
		arena_alloc(app->arena, video_arena_size),
		video_arena_size
	);
	init_arena(
		&asset_arena,
		arena_alloc(app->arena, asset_arena_size),
		asset_arena_size
	);
	assets.init(&asset_arena, "pack");
	dev = Device::create(&video_arena, app); 
	app->dev = dev;
	shader = (Shader*)assets.load("triangle.csh");
	texture = (Texture*)assets.load("22.tex");
	per_frame = heap_alloc(
		dev->heap,
		per_frame_memory_size
	);
	assert(per_frame != 0);
	uint8_t r = 0;
	vbo = upload_verts(dev);
	while (app->running) {
		Arena frame_arena;
		init_arena(&frame_arena, per_frame, per_frame_memory_size);

		app->begin();
		dev->begin_frame();

		Pipeline_Builder pb(&frame_arena);
		pb.begin_rp();
		pb.rp_target(dev->get_backbuffer(), { r, 0x00, 0xff, 0xff });
		Render_Pass& pass = pb.build_rp();

		pb.begin();
		pb.shader(shader->id);
		pb.vertex_format(shader->vf);
		Pipeline& pip = pb.build();

		Vertex_Buffer_Binding binding[] = {{
			.id = vbo,
			.offset = 0,
			.target = shader->binding_index("verts")
		}, {}};

		Draw draw{};
		draw.verts = binding;
		draw.vertex_count = 3;
		draw.instance_count = 1;

		dev->get_ctx().submit(draw, pip, pass);
	/*	dev->get_ctx().submit(*dev, pass);*/

		r += 10;
		dev->present();
		app->end();
	}
	assets.destroy();
	dev->destroy_buffer(vbo);
	dev->destroy();
	app->destroy();
}