const coral = @import("coral");

const std = @import("std");

const ona = @import("ona");

const ChromaticAberration = extern struct {
	effect_magnitude: f32,
	padding: [12]u8 = undefined,
};

const CRT = extern struct {
	width: f32,
	height: f32,
	time: f32,
	padding: [4]u8 = undefined,
};

const Actors = struct {
	instances: coral.stack.Sequential(@Vector(2, f32)) = .{.allocator = coral.heap.allocator},
	body_texture: ona.gfx.Texture = .default,
	render_texture: ona.gfx.Texture = .default,
	ca_effect: ona.gfx.Effect = .default,
	crt_effect: ona.gfx.Effect = .default,
	staging_texture: ona.gfx.Texture = .default,
};

const Player = struct {
	move_x: ona.act.Axis = .{.keys = .{.a, .d}},
	move_y: ona.act.Axis = .{.keys = .{.w, .s}},
};

pub fn main() !void {
	try ona.start_app(setup, .{
		.tick_rate = 60,
		.execution = .{.thread_share = 0.1},
	});
}

fn load(config: ona.Write(ona.gfx.Config), actors: ona.Write(Actors), assets: ona.Write(ona.gfx.Assets)) !void {
	config.res.width, config.res.height = .{1280, 720};
	actors.res.body_texture = try assets.res.load_texture_file(coral.files.bundle, "actor.bmp");

	actors.res.render_texture = try assets.res.load_texture(.{
		.format = .rgba8,

		.access = .{
			.render = .{
				.width = config.res.width,
				.height = config.res.height,
			},
		},
	});

	actors.res.ca_effect = try assets.res.load_effect_file(coral.files.bundle, "./ca.frag.spv");
	actors.res.crt_effect = try assets.res.load_effect_file(coral.files.bundle, "./crt.frag.spv");

	try actors.res.instances.push_grow(.{0, 0});
}

fn exit(actors: ona.Write(Actors)) void {
	actors.res.instances.deinit();
}

fn render(commands: ona.gfx.Commands, actors: ona.Write(Actors), app: ona.Read(ona.App)) !void {
	try commands.set_target(.{
		.texture = actors.res.render_texture,
		.clear_color = ona.gfx.colors.black,
		.clear_depth = 0,
		.clear_stencil = 0,
	});

	try commands.draw_texture(.{
		.texture = .default,

		.transform = .{
			.origin = .{1280 / 2, 720 / 2},
			.xbasis = .{1280, 0},
			.ybasis = .{0, 720},
		},
	});

	try commands.set_effect(.{
		.effect = actors.res.crt_effect,

		.properties = std.mem.asBytes(&CRT{
			.width = 1280,
			.height = 720,
			.time = @floatCast(app.res.elapsed_time),
		}),
	});

	try commands.set_target(.{
		.texture = .backbuffer,
		.clear_color = null,
		.clear_depth = null,
		.clear_stencil = null,
	});

	try commands.draw_texture(.{
		.texture = actors.res.render_texture,

		.transform = .{
			.origin = .{1280 / 2, 720 / 2},
			.xbasis = .{1280, 0},
			.ybasis = .{0, 720},
		},
	});
}

fn update(player: ona.Read(Player), actors: ona.Write(Actors), mapping: ona.Read(ona.act.Mapping)) !void {
	actors.res.instances.values[0] += .{
		mapping.res.axis_strength(player.res.move_x) * 10,
		mapping.res.axis_strength(player.res.move_y) * 10,
	};
}

fn setup(world: *ona.World, events: ona.App.Events) !void {
	try world.set_state(Actors{});
	try world.set_state(Player{});

	try world.on_event(events.load, ona.system_fn(load), .{.label = "load"});
	try world.on_event(events.update, ona.system_fn(update), .{.label = "update"});
	try world.on_event(events.exit, ona.system_fn(exit), .{.label = "exit"});
	try world.on_event(events.render, ona.system_fn(render), .{.label = "render actors"});
}