#include "game.hpp"

namespace game {

unit_t::unit_t(game::state_t *game_, unit_t::type_t type_) : game::entity_t(game_, ET_UNIT)
{
	type = type_;
}

void unit_t::render_to(render::state_t *render)
{
	if (selected == selection_cookie)
		render->render_hlrect(render_bounds, sf::Color::Blue);

	if (!dead && health < max_health)
	{
		rectf_t bar;
		float frac;
		sf::Color color;

		bar[0][0] = (render_bounds[0][0] + render_bounds[1][0]) / 2;
		bar[0][1] = render_bounds[0][1] - 0.05f;
		bar[0][0] -= 0.3f;
		bar[1][0] = bar[0][0] + 0.6f;
		bar[1][1] = bar[0][1] + 0.05f;
		render->render_rect(bar, sf::Color::Black);

		frac = (float)health / max_health;

		if (frac < 0.3f)
			color = sf::Color::Red;
		else if (frac < 0.75f)
			color = sf::Color::Yellow;
		else
			color = sf::Color::Green;

		bar[1][0] = lerp(bar[0][0], bar[1][0], frac);
		render->render_rect(bar, color);
	}

	if (move.moving && debug_draw_paths)
		render->debug_path(&move.path);

	if (!dead && say_time + 5.0 > game->now) {
		v2f_t text_pos;
		float height;

		text_pos = render_bounds[0] + v2f_t(render_bounds.dim(0) / 2, -render_bounds.dim(1) * 0.1);
		height = size.dim_min() * 0.20f;
		render->render_text(text_pos, height, say_text,
				    render::ALIGN_CENTER_BOTTOM,
				    sf::Color::White);
	}
}

void unit_t::say(std::string str)
{
	say_text = str;
	say_time = game->now;
	game->interface->print(name + ": " + str);
}

bool unit_t::keep_moving(double speed)
{
	float time;
	bool rv = true;

	if (!move.moving)
		return true;

	if (move.blocked && game->now < move.next_attempt)
		return true;

	time = game->dt * speed;

	while (time > 0.0f) {
		v2f_t delta, next, x_new;
		world::cmodel_t cmodel_next;

		if (!move.path.size()) {
			move.moving = false;
			break;
		}

		next = *move.path.begin();
		delta = next - x;

		if (delta.len() != 0.0f)
			move.angle = delta.angle();

		if (delta.len() >= time) {
			x_new = x + delta * time / delta.len();
			time -= delta.len();
		} else {
			x_new = next;
			time -= delta.len();
			move.path.pop_front();
		}

		cmodel_next.bounds = size + x_new;
		cmodel_next.cflags = cmodel.cflags;
		if (!world->test_rect(&cmodel_next, this)) {
			x = x_new;
			cmodel = cmodel_next;
			continue;
		}

		if (move.attempts_left) {
			move.blocked = true;
			move.attempts_left--;
			move.next_attempt = game->now + 0.2f;
		} else {
			if ((x - move.dst).len() > 1.5f)
				rv = false;
			move.moving = false;
		}
		break;
	}

	place(world, x);
	return rv;
}

bool unit_t::start_moving(v2f_t dst, world::cflags_t cflags)
{
	world::cmodel_t rep;

	if (!world) {
		printf("unit_t::start_moving: entity is not linked\n");
		return false;
	}

	move.dst = dst;
	move.path.clear();

	rep.cflags = cflags;
	rep.bounds = cmodel.bounds;

	if (!world->find_path(x, move.dst, &rep, this, &move.path)) {
		move.moving = false;
		return false;
	}

	move.moving = true;

	move.blocked = false;
	move.attempts_left = 10;
	move.next_attempt = -INFINITY;
	return true;
}

void unit_t::damage(int points, unit_t *attacker)
{
	fx_blood_t *blood;

	blood = new fx_blood_t(game, x, type == UNIT_SPIDER);
	blood->place(&game->world);

	health -= points;
	if (health < 0)
		die(attacker);
}

void unit_t::die(unit_t *killer)
{
	game->interface->print(name + " " + text::get(text::UNIT_DEATH) + ".");
	dead = true;
	death_time = game->now;
	cmodel.cflags = 0;
	on_death();
}

void unit_t::try_attack(unit_t *target)
{
	std::stringstream ss;
	size_t hit_roll;
	size_t dmg_roll;

	ss << name << " " << text::get(text::UNIT_ATTACK) << " " << target->name << ": ";

	hit_roll = game->roll(&cs.hit_roll);
	ss << hit_roll << " vs " << target->cs.armor_class;

	if (hit_roll == 1 || hit_roll < target->cs.armor_class) {
		ss << " (" << text::get((hit_roll == 1 ? text::UNIT_CRITICAL_MISS : text::UNIT_MISS)) << ")";
		game->interface->print(ss.str());
		return;
	}

	dmg_roll = game->roll(&cs.damage_roll);

	if (hit_roll < 20) {
		ss << ", " << text::get(text::UNIT_DAMAGE) << ": ";
		ss << dmg_roll - cs.damage_roll.bonus << " + " << cs.damage_roll.bonus << " = " << dmg_roll;
	} else {
		ss << " (" << text::get(text::UNIT_CRITICAL_HIT) << ")" << ", " << text::get(text::UNIT_DAMAGE) << ": ";
		ss << "(" << dmg_roll - cs.damage_roll.bonus << " + " << cs.damage_roll.bonus << ") x 2 = " << dmg_roll * 2;
		dmg_roll *= 2;
	}

	game->interface->print(ss.str());
	target->damage(dmg_roll, this);
}

static unit_t *find_target(world::world_t *world, v2f_t x, float r,
                           unit_t::type_t type)
{
	rectf_t rect;
	float nearest_dist = INFINITY;
	unit_t *nearest = NULL;

	rect[0] = x - v2f_t(r, r);
	rect[1] = x + v2f_t(r, r);

	for (world::entity_t *ent : world->get_entities(rect, -1)) {
		unit_t *unit;
		float dist;

		if (ent->type != ET_UNIT)
			continue;

		unit = (unit_t*)ent;

		if (unit->type != type)
			continue;

		if (unit->dead)
			continue;

		dist = (unit->x - x).len();
		if (dist < nearest_dist) {
			nearest_dist = dist;
			nearest = unit;
		}
	}

	return nearest;
}

unit_soldier_t::unit_soldier_t(game::state_t *game) : unit_t(game, UNIT_SOLDIER)
{
	always_awake = true;
	friendly = true;
	controllable = true;

	cmodel.cflags = CF_BODY;
	health = max_health = 20;
	size[0] = v2f_t(-0.4f, -0.4f);
	size[1] = v2f_t(+0.4f, +0.4f);
	render_size[0] = v2f_t(-0.5f, -1.0f);
	render_size[1] = v2f_t(+0.5f, +0.5f);
	name = text::get(text::UNIT_NAME_SOLDIER);

	cs.armor_class = 10;
	cs.hit_roll = roll_params_t(20);
	cs.damage_roll = roll_params_t(4);
}

void unit_soldier_t::on_think(void)
{
	if (dead)
		return;

	game->wake_everything(x, 10);

	if (game->now > next_targetting) {
		unit_t *target;

		target = find_target(world, x, 10.0f, UNIT_SPIDER);
		if (target) {
			last_target_time = game->now;
			last_target_x = target->x;

			if (last_attack + 1.5 < game->now) {
				world::trace_t trace;
				fx_tracer_t *tracer;

				trace = world->trace(x, target->x, CF_SOLID);
				if (!trace.hit) {
					tracer = new fx_tracer_t(game, x, target->x);
					tracer->place(&game->world);

					last_attack = game->now;
					try_attack(target);
				}
			}

			//start_moving(target->x, CF_SOLID);
		}

		next_targetting = game->now + 0.2;
	}

	if (!keep_moving(4.0))
		say(text::get(text::SAY_BLOCKED));
}

void unit_soldier_t::on_death(void)
{
	render_size[0] = v2f_t(-0.75f, -0.5f);
	render_size[1] = v2f_t(+0.75f, +0.5f);
	render_layer = -1;
	cmodel.cflags = CF_BACKGROUND;
	place(world, x);
}

void unit_soldier_t::render_to(render::state_t *render)
{
	if (!dead) {
		render::oriented_sprite_t *legs, *body;
		float body_angle;

		if (move.moving && !move.blocked)
			legs = &assets::soldier.legs_walking;
		else
			legs = &assets::soldier.legs_idle;

		if (last_target_time + 3 > game->now) {
			if (last_attack + 0.1 > game->now)
				body = &assets::soldier.body_firing;
			else
				body = &assets::soldier.body_aiming;

			body_angle = (last_target_x - x).angle();

		} else {
			body = &assets::soldier.body_idle;
			body_angle = move.angle;
		}

		render->render(game->now * 10, legs, render_bounds, move.angle);
		render->render(game->now * 10, body, render_bounds, body_angle);
		render->render(game->now * 10, &assets::soldier.head_idle, render_bounds, body_angle);
	} else
		render->render(game->now * 10, &assets::soldier.dead, render_bounds);

	unit_t::render_to(render);
}

unit_spider_t::unit_spider_t(game::state_t *game) : unit_t(game, UNIT_SPIDER)
{
	cmodel.cflags = CF_BODY_SMALL;
	health = max_health = 4;
	size[0] = v2f_t(-0.2f, -0.3f);
	size[1] = v2f_t(+0.2f, +0.3f);
	render_size[0] = v2f_t(-0.3f, -0.3f);
	render_size[1] = v2f_t(+0.3f, +0.3f);
	name = text::get(text::UNIT_NAME_SPIDER);

	cs.armor_class = 5;
	cs.hit_roll = roll_params_t(20);
	cs.damage_roll = roll_params_t(3, 6, 1);
}

void unit_spider_t::on_think(void)
{
	if (dead)
		return;

	if (game->now > next_targetting) {
		unit_t *target;

		target = find_target(world, x, 10.0f, UNIT_SOLDIER);
		if (target) {
			if (last_attack + 0.5 < game->now) {
				world::trace_t trace;

				if ((x - target->x).len() < 1.0f) {
					trace = world->trace(x, target->x, CF_SOLID);
					if (!trace.hit)
						try_attack(target);

					last_attack = game->now;
				}
			}

			start_moving(target->x, CF_SOLID | CF_WATER);
		}

		next_targetting = game->now + 0.2;
	}

	keep_moving(7.0);
}

void unit_spider_t::on_wake(void)
{
	next_targetting = game->now;
}

void unit_spider_t::on_death(void)
{
	render_layer = -1;
	cmodel.cflags = CF_BACKGROUND;
}

void unit_spider_t::render_to(render::state_t *render)
{
	bool moving;

	moving = move.moving && !move.blocked;

	if (!dead)
		render->render(game->now * 20, (moving ? &assets::spider.walking :
		               &assets::spider.idle), render_bounds, move.angle);
	else
		render->render(game->now * 20, &assets::spider.dead, render_bounds);

	unit_t::render_to(render);
}

} // namespace game