#include "base/logging.h"
#include "input/input_state.h"
#include "ui/screen.h"

Screen::Screen() : screenManager_(0) { }
Screen::~Screen() { }

ScreenManager::ScreenManager() {
	currentScreen_ = 0;
	nextScreen_ = 0;
}

ScreenManager::~ScreenManager() {
	delete currentScreen_;
}

void ScreenManager::switchScreen(Screen *screen) {
	if (dialog_.size())
	{
		WLOG("Switching screens - dropping the whole dialog stack");
		while (dialog_.size())
			pop();
	}
	// Note that if a dialog is found, this will be a silent background switch that
	// will only become apparent if the dialog is closed. The previous screen will stick around
	// until that switch.
	// TODO: is this still true?
	if (nextScreen_ != 0) {
		FLOG("WTF? Already had a nextScreen_");
	}
	if (screen != currentScreen_) {
		nextScreen_ = screen;
		nextScreen_->setScreenManager(this);
	}
}

void ScreenManager::update(InputState &input) {
	if (dialog_.size()) {
		dialog_.back()->update(input);
		return;
	}

	if (nextScreen_) {
		ILOG("Screen switch!");
		Screen *temp = currentScreen_;
		currentScreen_ = nextScreen_;
		delete temp;
		temp = 0;
		nextScreen_ = 0;
	}

	if (currentScreen_) {
		currentScreen_->update(input);
	}
}

void ScreenManager::touch(int pointer, float x, float y, double time, TouchEvent event)
{
	if (dialog_.size()) {
		dialog_.back()->touch(pointer, x, y, time, event);
		return;
	}
	if (currentScreen_)
	{
		currentScreen_->touch(pointer, x, y, time, event);
	}
}

void ScreenManager::render() {
	if (dialog_.size()) {
		dialog_.back()->render();
		return;
	}
	if (currentScreen_) {
		currentScreen_->render();
	}
	else {
		ELOG("No current screen!");
	}
}

void ScreenManager::deviceLost()
{
	if (currentScreen_)
		currentScreen_->deviceLost();
	// Dialogs too? Nah, they should only use the standard UI texture anyway.
	// TODO: Change this when it becomes necessary.
}

Screen *ScreenManager::topScreen() {
	if (dialog_.size())
		return dialog_.back();
	else
		return currentScreen_;
}

void ScreenManager::shutdown() {
	if (nextScreen_) {
		delete nextScreen_;
		nextScreen_ = 0;
	}
	if (currentScreen_) {
		delete currentScreen_;
		currentScreen_ = 0;
	}
}

void ScreenManager::push(Screen *screen) {
	screen->setScreenManager(this);
	dialog_.push_back(screen);
}

void ScreenManager::pop() {
	if (dialog_.size()) {
		delete dialog_.back();
		dialog_.pop_back();
	} else {
		ELOG("Can't push when no dialog is shown");
	}
}

void ScreenManager::finishDialog(const Screen *dialog, DialogResult result)
{
	if (!dialog_.size()) {
		ELOG("Must be in a dialog to finishDialog");
		return;
	}
	Screen *dlg = dialog_.back();
	if (dialog != dialog_.back())
	{
		ELOG("Wrong dialog being finished!");
		return;
	}
	if (dialog_.size()) {
		dialog_.pop_back();
	}
	Screen *caller = topScreen();
	caller->dialogFinished(dialog, result);
	delete dialog;
}