#include "app.hpp"
#include "video.hpp"
extern "C" {
#include "plat.h"
#include "memory.h"
}
#include <string.h>
#include <math.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, 1.0f, 1.0f,
	-0.5f,  0.5f, 0.0f, 1.0f, 0.0f, 0.0f, 1.0f,
	 0.0f, -0.5f, 0.0f, 0.0f, 1.0f, 0.5f, 0.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;
}

static Sampler_Id create_clamped_linear(Device* dev) {
	Sampler_State s{};
	s.min = Filter_Mode::linear;
	s.mag = Filter_Mode::linear;
	s.address_u = Address_Mode::repeat;
	s.address_v = Address_Mode::repeat;
	return dev->create_sampler(s);
}

struct Config_Buffer {
	float offset[2];
};

int main() {
	Arena video_arena;
	Arena asset_arena;
	Asset_Arena assets;
	Device* dev;
	Shader* shader;
	Texture* texture;
	Texture* texture2;
	Buffer_Id vbo, cbuf;
	Sampler_Id clamped_linear;
	C2* app = App::create<C2>("c2");
	void* per_frame;
	int frame = 0;
	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");
	texture2 = (Texture*)assets.load("kita.tex");
	cbuf = dev->create_buffer(
		sizeof verts,
		Buffer_Flags::constant_buffer |
		Buffer_Flags::cpu_readwrite
	);
	per_frame = heap_alloc(
		dev->heap,
		per_frame_memory_size
	);
	clamped_linear = create_clamped_linear(dev);
	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();

		{
			void* mem;
			mem = dev->map_buffer(cbuf, 0, sizeof(Config_Buffer));
			Config_Buffer* c = (Config_Buffer*)mem;
			c->offset[0] = sinf((float)frame / 10.0f);
			c->offset[1] = cosf((float)frame / 30.0f);
			dev->unmap_buffer(cbuf);
		}

		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);
		pb.texture(
			shader->descriptor_binding("colour_texture"),
			frame % 2? texture->id: texture2->id,
			clamped_linear
		);
		pb.cbuffer(
			shader->descriptor_binding("config_buffer"),
			cbuf
		);
		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);

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