#include "generator.h"
#include "core/config/engine.h"
#include "core/math/math_funcs.h"
#include "core/object/class_db.h"
#include "macros.h"

void Generator::_bind_methods() {
	BIND_HPROPERTY(Variant::OBJECT, state, PROPERTY_HINT_RESOURCE_TYPE, "Sentence");
}

void Generator::initialise_state() {
	print_line("Filling state with unknown");
	for (int i{ 0 }; i < this->state->symbols.size(); ++i) {
		this->state->symbols.set(i, Terminals::Undefined);
	}
	print_line("Writing borders X");
	for (int i{ 0 }; i < this->state->size.x; ++i) {
		this->state->set_at({ i, 0 }, NonTerminals::Blocked);
		this->state->set_at({ i, this->state->size.y - 1 }, NonTerminals::Blocked);
	}
	print_line("Writing borders Y");
	for (int i{ 0 }; i < this->state->size.y; ++i) {
		this->state->set_at({ 0, i }, NonTerminals::Blocked);
		this->state->set_at({ this->state->size.x - 1, i }, NonTerminals::Blocked);
	}
	print_line("Writing path");
	int y{ (Math::abs((int)Math::rand()) % (this->state->size.y - 4)) + 2 };
	print_line("Path at y =", y);
	for (int i{ 1 }; i < this->state->size.x - 1; ++i) {
		this->state->set_at({ i, y }, Terminals::Path);
	}
	print_line("Writing start & goal");
	this->state->set_at({ 0, y }, Terminals::Start);
	this->state->set_at({ this->state->size.x - 1, y }, Terminals::Goal);
}

void Generator::ready() {
	initialise_state();
	print_line("STATE:");
	print_line(this->state->get_symbols_string());
	// TODO: repeat until terminal state
	this->rule->try_apply(this->state);
	print_line("STATE:");
	print_line(this->state->get_symbols_string());
}

void Generator::_notification(int what) {
	if (Engine::get_singleton()->is_editor_hint()) {
		return;
	}
	switch (what) {
		default:
			return;
		case NOTIFICATION_READY:
			ready();
		case NOTIFICATION_CHILD_ORDER_CHANGED:
			for (Variant child : get_children()) {
				if (Rule * rule{ cast_to<Rule>(child) }) {
					this->rule = rule;
					return;
				}
			}
			return;
	}
}