#include <cstdio>
#include <string>

#include "Canvas.h"
#include "util.h"
#include "Museum.h"

using namespace std;

Canvas::Canvas(Museum & museum) : museum(museum) { }

Tile & Canvas::get_tile(unsigned x, unsigned y) {
	return *this->tiles[this->pos_to_index(x, y)];
}

void Canvas::set_tile(TileData data) {
	size_t index = this->pos_to_index(data.x, data.y);
	if (this->tiles[index] != nullptr)
		delete this->tiles[index];
	this->tiles[index] = new Tile(this->museum, data);
}

size_t Canvas::pos_to_index(unsigned x, unsigned y) {
	size_t index = y * this->data.columns + x;
	return index;
}

void Canvas::update() {
	this->update_steps();
	this->update_tiles();
}

void Canvas::set_data(CanvasData data) {
	this->data = data;
	this->tiles.resize(this->data.rows * this->data.columns);
	for (size_t y = 0; y < this->data.rows; y++) {
		for (size_t x = 0; x < this->data.columns; x++) {
			if (this->tiles[this->pos_to_index(x, y)] != nullptr)
				continue;
			this->set_tile({
				.x = static_cast<unsigned int>(x),
				.y = static_cast<unsigned int>(y),
			});
		}
	}
}

Canvas::~Canvas() {
	for (size_t i = 0; i < this->tiles.size(); i++) {
		if (this->tiles[i] == nullptr) continue;
		delete this->tiles[i];
		this->tiles[i] = nullptr;
	}
}

string Canvas::to_string(bool truecolor) {
	string out = "";

	for (size_t y = 0; y < this->data.rows; y++) {
		for (size_t x = 0; x < this->data.columns; x++) {
			Tile & tile = this->get_tile(x, y);
			string type_str = tile.data.type;
			if (type_str.length() == 0) type_str = ".";

			if (truecolor)
				out += stringf("\e[38;2;0;0;0;48;2;%d;%d;%dm",
						tile.color.red, tile.color.green, tile.color.blue);

			out += stringf("%-2s ", type_str.c_str());
		}
		if (truecolor) out += "\e[0m";
		out += "\n";
	}

	return out;
}

void Canvas::update_steps() {
	for (Artist * artist : this->museum.people.get_artists()) {
		if (artist->step == false) continue;
		artist->step = false;

		this->get_tile(artist->data.x, artist->data.y).behavior->step(artist);
	}
}

void Canvas::update_tiles() {
	for (Tile * tile : this->tiles) {
		if (tile == nullptr) continue;
		tile->update();
	}
}