/*
 * This software is licensed under the terms of the MIT License.
 * See COPYING for further information.
 * ---
 * Copyright (c) 2011-2024, Lukas Weber <laochailan@web.de>.
 * Copyright (c) 2012-2024, Andrei Alexeyev <akari@taisei-project.org>.
 */

#include "elly.h"

#include "draw.h"
#include "spells/spells.h"

#include "stagedraw.h"
#include "util/glm.h"
#include "util/graphics.h"

TASK(elly_animate_colors, { BoxedBoss boss; }) {
	auto boss = TASK_BIND(ARGS.boss);

	for(int t = 0;; ++t, YIELD) {
		boss->glowcolor = *HSL(t/120.0, 1.0, 0.25);
		boss->shadowcolor = *HSLA_MUL_ALPHA((t+20)/120.0, 1.0, 0.25, 0.5);
	}
}

Boss *stage6_spawn_elly(cmplx pos) {
	Boss *b = create_boss("Elly", "elly", pos);
	boss_set_portrait(b, "elly", NULL, "normal");
	INVOKE_TASK(elly_animate_colors, ENT_BOX(b));
	return b;
}

static void scythe_particles(EllyScythe *s, real oldangle) {
	if(s->scale == 0) {
		return;
	}

	PARTICLE(
		.sprite_ptr = res_sprite("stage6/scythe"),
		.pos = s->pos+I*6*sin((global.frames - 0.5)/25.0),
		.draw_rule = pdraw_timeout_fade(0.75, 0),
		.timeout = 4,
		.angle = (s->angle + oldangle) * 0.5,
		.scale = s->scale,
		.flags = PFLAG_REQUIREDPARTICLE | PFLAG_MANUALANGLE,
		.layer = LAYER_PARTICLE_MID,
		.opacity = s->opacity,
	);

	PARTICLE(
		.sprite_ptr = res_sprite("stage6/scythe"),
		.pos = s->pos+I*6*sin(global.frames/25.0),
		.draw_rule = pdraw_timeout_fade(0.75, 0),
		.timeout = 4,
		.angle = s->angle,
		.scale = s->scale,
		.flags = PFLAG_REQUIREDPARTICLE | PFLAG_MANUALANGLE,
		.layer = LAYER_PARTICLE_MID,
		.opacity = s->opacity,
	);

	RNG_ARRAY(rand, 2);
	PARTICLE(
		.sprite = "smoothdot",
		.pos = s->pos + 100 * s->scale * vrng_real(rand[0]) * vrng_dir(rand[1]),
		.color = RGBA(1.0, 0.1, 1.0, 0.0),
		.draw_rule = pdraw_timeout_scalefade(0.4, 2, 1, 0),
		.timeout = 60,
		.flags = PFLAG_REQUIREDPARTICLE,
		.layer = LAYER_PARTICLE_LOW,
		.opacity = s->opacity,
		.scale = s->scale / 0.7,
	);
}

INLINE real normalize_angle(real a) {
	// [0 .. tau)
	return a - M_TAU * floorf(a / M_TAU);
}

TASK(scythe_update, { BoxedEllyScythe scythe; }) {
	EllyScythe *scythe = TASK_BIND(ARGS.scythe);

	real spin = 0;
	real oldangle;
	real return_speed_cap = 0;
	bool was_return_mode = false;

	for(;;) {
		oldangle = scythe->angle;
		YIELD;

		approach_asymptotic_p(&spin, scythe->spin, 0.1, 1e-4);
		real spin_sign = sign(spin);
		real spin_abs = fabs(spin);

		bool return_mode = scythe->spin == 0;

		if(was_return_mode && !return_mode) {
			return_speed_cap = 0;
		}

		if(spin_abs > return_speed_cap) {
			return_speed_cap = spin_abs;
		}

		if(return_mode) {
			real return_target = normalize_angle(scythe->resting_angle);
			real scythe_angle = normalize_angle(scythe->angle);
			real delta;

			if(spin_sign > 0) {
				if(return_target < scythe_angle) {
					return_target += M_TAU;
				}

				delta = return_target - scythe_angle;
			} else {
				if(scythe_angle < return_target) {
					return_target -= M_TAU;
				}

				delta = scythe_angle - return_target;
			}

			spin_abs = min(return_speed_cap, delta * 0.1);
		}

		was_return_mode = return_mode;
		scythe->angle += spin_sign * spin_abs;

		move_update(&scythe->pos, &scythe->move);
		scythe_particles(scythe, oldangle);
	}
}

static void stage6_init_elly_scythe(EllyScythe *scythe, cmplx pos) {
	scythe->pos = pos;
	scythe->scale = 0.7;
	scythe->resting_angle = -M_PI/2;
	scythe->angle = scythe->resting_angle;
	scythe->opacity = 1.0f;
	INVOKE_TASK(scythe_update, ENT_BOX(scythe));
}

void stage6_despawn_elly_scythe(EllyScythe *scythe) {
	coevent_signal_once(&scythe->events.despawned);
}

TASK(reset_scythe, { BoxedEllyScythe scythe; }) {
	EllyScythe *s = TASK_BIND(ARGS.scythe);
	s->resting_angle = -M_PI/2;
	s->move = move_from_towards(s->pos, ELLY_DEFAULT_POS + ELLY_SCYTHE_RESTING_OFS, 0.1);
	s->spin = 0;
}

Boss *stage6_elly_init_scythe_attack(ScytheAttackTaskArgs *args) {
	Boss *b = INIT_BOSS_ATTACK(&args->base);

	if(ENT_UNBOX(args->scythe) == NULL) {
		args->scythe = ENT_BOX(stage6_host_elly_scythe(b->pos));
	}

	INVOKE_TASK_AFTER(
		&args->base.attack->events.finished,
		reset_scythe,
		args->scythe
	);

	return b;
}

EllyScythe *stage6_host_elly_scythe(cmplx pos) {
	EllyScythe *scythe = TASK_HOST_ENT(EllyScythe);
	TASK_HOST_EVENTS(scythe->events);
	stage6_init_elly_scythe(scythe, pos);
	return scythe;
}

static void baryons_connector_particles(cmplx a, cmplx b, BoxedProjectileArray *particles) {
	Sprite *spr = res_sprite("stage6/baryon_connector");

	cmplx connector_pos = (a + b)/2;
	real connector_angle = carg(a - b) + M_PI/2;
	cmplx connector_scale = (cabs(a - b) - 70) / spr->w + I * 20 / spr->h;
	ENT_ARRAY_ADD(particles, PARTICLE(
		.sprite_ptr = spr,
		.pos = connector_pos,
		.draw_rule = pdraw_timeout_fade(0.5, 0),
		.timeout = 2,
		.angle = connector_angle,
		.flags = PFLAG_REQUIREDPARTICLE | PFLAG_MANUALANGLE,
		.scale = connector_scale
	));
}

static void baryons_particles(EllyBaryons *baryons, BoxedProjectileArray *particles) {
	if(baryons->scale == 0) {
		return;
	}

	ENT_ARRAY_ADD(particles, PARTICLE(
		.sprite_ptr = res_sprite("stage6/baryon_center"),
		.pos = baryons->center_pos,
		.draw_rule = pdraw_timeout_fade(0.75, 0),
		.timeout = 4,
		.angle = baryons->center_angle,
		.scale = baryons->scale,
		.flags = PFLAG_REQUIREDPARTICLE | PFLAG_MANUALANGLE,
	));

	ENT_ARRAY_ADD(particles, PARTICLE(
		.sprite_ptr = res_sprite("stage6/baryon"),
		.pos = baryons->center_pos,
		.angle = M_TAU/12,
		.draw_rule = pdraw_timeout_fade(0.75, 0),
		.timeout = 4,
		.flags = PFLAG_REQUIREDPARTICLE | PFLAG_MANUALANGLE,
	));


	for(int i = 0; i < NUM_BARYONS; i++) {
		cmplx pos = baryons->poss[i];
		cmplx pos_next = baryons->poss[(i+1) % NUM_BARYONS];

		ENT_ARRAY_ADD(particles, PARTICLE(
			.sprite_ptr = res_sprite("stage6/baryon"),
			.pos = pos,
			.draw_rule = pdraw_timeout_fade(0.75, 0),
			.timeout = 4,
			.scale = baryons->scale,
			.angle = baryons->angles[i]+M_TAU/12,
			.flags = PFLAG_REQUIREDPARTICLE | PFLAG_MANUALANGLE,
		));

		baryons_connector_particles(pos, pos_next, particles);
	}

	baryons_connector_particles(baryons->center_pos, baryons->poss[0], particles);
	baryons_connector_particles(baryons->center_pos, baryons->poss[3], particles);
}

static void baryons_bg_feedback(Stage6DrawData *draw_data, int t) {
	stage_draw_begin_noshake();
	r_state_push();

	r_shader("baryon_feedback");
	r_uniform_vec2("blur_resolution", 0.5*VIEWPORT_W, 0.5*VIEWPORT_H);
	r_uniform_float("hue_shift", 0);
	r_uniform_float("time", t/60.0);

	r_framebuffer(draw_data->baryon.aux_fb);
	r_blend(BLEND_NONE);
	r_uniform_vec2("blur_direction", 1, 0);
	r_uniform_float("hue_shift", 0.01);
	r_color4(0.95, 0.88, 0.9, 0.5);

	draw_framebuffer_tex(draw_data->baryon.fbpair.front, VIEWPORT_W, VIEWPORT_H);

	fbpair_swap(&draw_data->baryon.fbpair);

	r_framebuffer(draw_data->baryon.fbpair.back);
	r_uniform_vec2("blur_direction", 0, 1);
	r_color4(1, 1, 1, 1);
	draw_framebuffer_tex(draw_data->baryon.aux_fb, VIEWPORT_W, VIEWPORT_H);

	r_state_pop();
	stage_draw_end_noshake();
}

static void baryons_bg_blend(Stage6DrawData *draw_data) {
	stage_draw_begin_noshake();
	r_state_push();

	r_blend(BLEND_PREMUL_ALPHA);
	r_shader_standard();
	fbpair_swap(&draw_data->baryon.fbpair);
	r_color4(0.7, 0.7, 0.7, 0.7);
	draw_framebuffer_tex(draw_data->baryon.fbpair.front, VIEWPORT_W, VIEWPORT_H);

	r_state_pop();
	stage_draw_end_noshake();
}

static void baryons_bg_fill(Stage6DrawData *draw_data, int t, EllyBaryons *baryons, BoxedProjectileArray *particles) {
	if(!particles->size) {
		return;
	}

	r_state_push();
	r_color4(1, 1, 1, 1);
	r_framebuffer(draw_data->baryon.fbpair.front);

	ENT_ARRAY_FOREACH(particles, Projectile *p, {
		r_state_push();
		p->ent.draw_func(&p->entity_interface);
		r_state_pop();
	});

	if(baryons) {
		ShaderProgram *shader = res_shader("sprite_default");
		Sprite *sprite = res_sprite("part/myon");

		for(int i = 0; i < NUM_BARYONS; ++i) {
			r_draw_sprite(&(SpriteParams) {
				.sprite_ptr = sprite,
				.shader_ptr = shader,
				.color = RGBA(1, 0.2, 1.0, 0.7),
				.pos.as_cmplx = baryons->poss[i], //+10*frand()*cexp(2.0*I*M_PI*frand());
				.rotation.angle = (i - t) / 16.0, // frand()*M_PI*2,
				.scale.both = 2,
			});
		}

		r_flush_sprites();
	}

	r_state_pop();
}

TASK(baryons_particle_spawner, { BoxedEllyBaryons baryons; BoxedProjectileArray *particles; }) {
	EllyBaryons *baryons = TASK_BIND(ARGS.baryons);

	for (;;) {
		baryons_particles(baryons, ARGS.particles);
		ENT_ARRAY_COMPACT(ARGS.particles);
		YIELD;
	}
}

TASK(baryons_visuals, { BoxedEllyBaryons baryons; }) {
	DECLARE_ENT_ARRAY(Projectile, particles, MAX_BARYON_PARTICLES);
	CoEvent *draw_event = &stage_get_draw_events()->background_drawn;
	Stage6DrawData *draw_data = stage6_get_draw_data();

	INVOKE_SUBTASK(baryons_particle_spawner, ARGS.baryons, &particles);

	int timeout = 60;
	int endtime = timeout;

	for(int t = 0; t < endtime; ++t) {
		WAIT_EVENT_OR_DIE(draw_event);
		EllyBaryons *baryons = ENT_UNBOX(ARGS.baryons);

		if(baryons != NULL) {
			endtime = t + timeout;
		}

		if(config_get_int(CONFIG_POSTPROCESS) < 1) {
			continue;
		}

		baryons_bg_fill(draw_data, t, baryons, &particles);
		baryons_bg_feedback(draw_data, t);
		baryons_bg_blend(draw_data);
	}
}

TASK(baryons_update, { BoxedEllyBaryons baryons; }) {
	EllyBaryons *baryons = TASK_BIND(ARGS.baryons);

	for (;;) {
		baryons->center_angle += 0.035;

		for(int i = 0; i < NUM_BARYONS; i++) {
			cmplx d = (baryons->target_poss[i] - baryons->poss[i]);
			if(cabs2(d) > 0) {
				baryons->poss[i] += baryons->relaxation_rate * (d / sqrt(cabs(d)) + 0.01*d);
			}
		}

		YIELD;
	}
}

TASK(baryons_quick_intro, { BoxedEllyBaryons baryons; }) {
	EllyBaryons *baryons = TASK_BIND(ARGS.baryons);

	int duration = 30;
	for(int t = 0; t < duration; t++) {
		baryons->scale = glm_ease_quad_in(t / (float) duration);
		YIELD;
	}
}

cmplx stage6_elly_baryon_default_offset(int i) {
	return 150 * cdir(M_TAU / NUM_BARYONS * i);
}

void stage6_init_elly_baryons(BoxedEllyBaryons baryons, BoxedBoss boss) {
	Boss *b = NOT_NULL(ENT_UNBOX(boss));
	EllyBaryons *eb = NOT_NULL(ENT_UNBOX(baryons));
	for(int i = 0; i < NUM_BARYONS; i++) {
		eb->target_poss[i] = b->pos + stage6_elly_baryon_default_offset(i);
		eb->poss[i] = b->pos;
	}
	eb->center_pos = b->pos;
	eb->scale = 0;
	eb->relaxation_rate = 0.5;
	INVOKE_TASK(baryons_update, baryons);
	INVOKE_TASK(baryons_visuals, baryons);
}

TASK(reset_baryons, { BoxedBoss boss; BoxedEllyBaryons baryons; }) {
	auto baryons = TASK_BIND(ARGS.baryons);
	auto boss = NOT_NULL(ENT_UNBOX(ARGS.boss));

	for(int i = 0; i < NUM_BARYONS; ++i) {
		baryons->target_poss[i] = boss->pos + stage6_elly_baryon_default_offset(i);
	}
}

Boss *stage6_elly_init_baryons_attack(BaryonsAttackTaskArgs *args) {
	if(ENT_UNBOX(args->baryons) == NULL) {
		args->baryons = ENT_BOX(stage6_host_elly_baryons(args->base.boss));
		stage6_init_elly_baryons(args->baryons, args->base.boss);
		INVOKE_TASK(baryons_quick_intro, args->baryons);
	}

	auto boss = INIT_BOSS_ATTACK(&args->base);

	INVOKE_TASK_AFTER(
		&args->base.attack->events.finished,
		reset_baryons,
		ENT_BOX(boss), args->baryons
	);

	return boss;
}

EllyBaryons *stage6_host_elly_baryons(BoxedBoss boss) {
	EllyBaryons *baryons = TASK_HOST_ENT(EllyBaryons);
	TASK_HOST_EVENTS(baryons->events);
	return baryons;
}

void elly_clap(Boss *b, int claptime) {
	aniplayer_queue(&b->ani, "clapyohands", 1);
	aniplayer_queue_frames(&b->ani, "holdyohands", claptime);
	aniplayer_queue(&b->ani, "unclapyohands", 1);
	aniplayer_queue(&b->ani, "main", 0);
}

