//
// C++ Implementation: InspectWidget
//
// Description:
//
//
// Author: Erik Ogenvik <erik@ogenvik.org>, (C) 2004
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software Foundation,
// Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.//
//
#include "InspectWidget.h"

#include "ColouredListItem.h"
#include "../GUIManager.h"
#include "domain/EmberEntity.h"
#include "../World.h"
#include "components/ogre/authoring/AuthoringManager.h"
#include "components/ogre/model/Model.h"
#include "components/ogre/EmberOgre.h"


#include "framework/Tokeniser.h"
#include "framework/ConsoleBackend.h"

#include <CEGUI/Window.h>
#include <CEGUI/widgets/Listbox.h>
#include <CEGUI/widgets/PushButton.h>

#include <Eris/TypeInfo.h>

#include <Atlas/PresentationBridge.h>
#include <components/ogre/model/ModelRepresentation.h>

namespace Ember::OgreView::Gui {

WidgetPluginCallback InspectWidget::registerWidget(GUIManager& guiManager) {

	struct State {
		std::shared_ptr<InspectWidget> widget;
	};
	auto state = std::make_shared<State>();

	auto connectFn = [state, &guiManager](World&) {
		state->widget = std::make_shared<Gui::InspectWidget>(guiManager);
	};
	auto conCreated = EmberOgre::getSingleton().EventWorldCreated.connect(connectFn);
	auto conDestroyed = EmberOgre::getSingleton().EventWorldDestroyed.connect([state]() { state->widget.reset(); });

	if (EmberOgre::getSingleton().getWorld()) {
		connectFn(*EmberOgre::getSingleton().getWorld());
	}

	//Just hold on to an instance.
	return [state, conCreated, conDestroyed]() mutable {
		conCreated.disconnect();
		conDestroyed.disconnect();
		state.reset();
	};

}

InspectWidget::InspectWidget(GUIManager& guiManager) :
		Widget(guiManager),
		Inspect("inspect", this, "Inspect an entity."),
		mChildList(nullptr),
		mInfo(nullptr),
		mCurrentEntity(nullptr),
		mChangedThisFrame(false) {
	guiManager.EventFrameStarted.connect(sigc::mem_fun(*this, &InspectWidget::frameStarted));
	loadMainSheet("InspectWidget.layout", "InspectWidget/");
	mMainWindow->setVisible(false);

	mChildList = dynamic_cast<CEGUI::Listbox*>(getWindow("ChildList"));
	mChildList->subscribeEvent(CEGUI::Listbox::EventMouseDoubleClick, CEGUI::Event::Subscriber(&InspectWidget::ChildList_MouseDoubleClick, this));

	mInfo = getWindow("EntityInfo");


	mGuiManager.EventEntityAction.connect(sigc::mem_fun(*this, &InspectWidget::handleAction));
	enableCloseButton();

	getWindow("ShowOgreBoundingBox")->subscribeEvent(CEGUI::PushButton::EventClicked, [this]() {
		if (mCurrentEntity) {
			mCurrentEntity->setVisualize("OgreBBox", !mCurrentEntity->getVisualize("OgreBBox"));
		}
		return true;
	});

	getWindow("ShowErisBoundingBox")->subscribeEvent(CEGUI::PushButton::EventClicked, [this]() {
		if (mCurrentEntity && EmberOgre::getSingleton().getWorld()) {
			if (EmberOgre::getSingleton().getWorld()->getAuthoringManager().hasSimpleEntityVisualization(*mCurrentEntity)) {
				EmberOgre::getSingleton().getWorld()->getAuthoringManager().hideSimpleEntityVisualization(*mCurrentEntity);
			} else {
				EmberOgre::getSingleton().getWorld()->getAuthoringManager().displaySimpleEntityVisualization(*mCurrentEntity);
			}
		}
		return true;
	});


	getWindow("ShowGeometry")->subscribeEvent(CEGUI::PushButton::EventClicked, [this]() {
		if (mCurrentEntity && EmberOgre::getSingleton().getWorld()) {
			if (EmberOgre::getSingleton().getWorld()->getAuthoringManager().hasGeometryVisualization(*mCurrentEntity)) {
				EmberOgre::getSingleton().getWorld()->getAuthoringManager().hideGeometryVisualization(*mCurrentEntity);
			} else {
				EmberOgre::getSingleton().getWorld()->getAuthoringManager().displayGeometryVisualization(*mCurrentEntity);
			}
		}
	});

	getWindow("ShowModel")->subscribeEvent(CEGUI::PushButton::EventClicked, [this]() {
		if (mCurrentEntity) {
			auto model = Model::ModelRepresentation::getModelForEntity(*mCurrentEntity);
			if (model) {
				model->setVisible(!model->getVisible());
			}
		}
	});

}

InspectWidget::~InspectWidget() = default;

void InspectWidget::entity_BeingDeleted() {
	disconnectFromEntity();
	mCurrentEntity = nullptr;
}

void InspectWidget::updateAttributeString() {
	std::stringstream ss;

	Atlas::PresentationBridge bridge(ss);
	bridge.setMaxItemsPerLevel(15);
	Atlas::Message::Encoder encoder(bridge);
	encoder.streamMessageElement(mCurrentEntity->getProperties());

	mAttributesString = ss.str();
}

void InspectWidget::runCommand(const std::string& command, const std::string& args) {
	if (Inspect == command && EmberOgre::getSingleton().getWorld()) {
		//the first argument must be a valid entity id
		Tokeniser tokeniser;
		tokeniser.initTokens(args);
		std::string entityId = tokeniser.nextToken();
		if (!entityId.empty()) {
			EmberEntity* entity = EmberOgre::getSingleton().getWorld()->getEmberEntity(entityId);
			if (entity != nullptr) {
				startInspecting(entity);
			}
		} else {
			ConsoleBackend::getSingletonPtr()->pushMessage("You must specify a valid entity id to inspect.", "error");
		}

	} else {
		Widget::runCommand(command, args);
	}

}

void InspectWidget::disconnectFromEntity() {
	mChangedConnection.disconnect();
	mChildAddedConnection.disconnect();
	mChildRemovedConnection.disconnect();
}

void InspectWidget::handleAction(const std::string& action, EmberEntity* entity) {

	if (action == "inspect") {
		startInspecting(entity);
	}
}

void InspectWidget::startInspecting(EmberEntity* entity) {
	disconnectFromEntity();

	mChangedThisFrame = true;

	show();

	mCurrentEntity = entity;
	updateAttributeString();
	showEntityInfo(entity);

	mChangedConnection = entity->Changed.connect(sigc::mem_fun(*this, &InspectWidget::entity_Changed));
	mChildAddedConnection = entity->ChildAdded.connect(sigc::mem_fun(*this, &InspectWidget::entity_ChildAdded));
	mChildRemovedConnection = entity->ChildRemoved.connect(sigc::mem_fun(*this, &InspectWidget::entity_ChildRemoved));
	mBeingDeletedConnection = entity->BeingDeleted.connect(sigc::mem_fun(*this, &InspectWidget::entity_BeingDeleted));


	fillChildrenList();

}

void InspectWidget::frameStarted(float) {
	if (mMainWindow->isVisible() && mCurrentEntity && mChangedThisFrame) {
		showEntityInfo(mCurrentEntity);
	}
}

void InspectWidget::showEntityInfo(EmberEntity* entity) {
	Eris::Entity* parent = entity->getLocation();
	std::stringstream ss;
	ss.precision(4);

	ss << "Name: " << entity->getName() << "\n";
	ss << "Id: " << entity->getId() << "\n";
	ss << "Parent: ";
	if (parent) {
		ss << parent->getName() << " (Id: " << parent->getId() << ")";
	} else {
		ss << "none";
	}
	ss << "\n";

	if (entity->getPredictedPos().isValid()) {
		ss << "PredPosition: " << entity->getPredictedPos() << "\n";
	}
	if (entity->getPosition().isValid()) {
		ss << "Position: " << entity->getPosition() << "\n";
	}
	const WFMath::Vector<3>& velocity = entity->getVelocity();
	if (velocity.isValid()) {
		ss << "Velocity: " << velocity << ": " << sqrt(velocity.sqrMag()) << "\n";
	}

	if (entity->getOrientation().isValid()) {
		ss << "Orientation: " << entity->getOrientation() << "\n";
	}
	if (entity->getBBox().isValid()) {
		ss << "Boundingbox: " << entity->getBBox() << "\n";
	}

	ss << "Type: " << entity->getType()->getName() << "\n";

	ss << "Attributes:\n";

	ss << mAttributesString;

	mInfo->setText(ss.str());
	mChangedThisFrame = false;

}

void InspectWidget::fillChildrenList() {
	size_t numberOfChildren = mCurrentEntity->numContained();
	mChildList->resetList();

	for (size_t i = 0; i < numberOfChildren; ++i) {
		Eris::Entity* child = mCurrentEntity->getContained(i);
		addChildToList(child);
	}

}

void InspectWidget::addChildToList(Eris::Entity* child) {
	CEGUI::String name(child->getType()->getName());
	if (child->getName().empty()) {
		name += " (" + child->getId() + ")";
	} else {
		name += " (" + child->getId() + " : " + child->getName() + ")";
	}
	CEGUI::ListboxItem* item = Gui::ColouredListItem::createColouredListItem(name);
	item->setUserData(child);
	mChildList->addItem(item);
}

void InspectWidget::entity_ChildAdded(Eris::Entity* entity) {
	addChildToList(entity);
}

void InspectWidget::entity_ChildRemoved(Eris::Entity* entity) {
	for (size_t i = 0; i < mChildList->getItemCount(); ++i) {
		CEGUI::ListboxItem* item = mChildList->getListboxItemFromIndex(i);
		if (item->getUserData() == entity) {
			mChildList->removeItem(item);
			break;
		}
	}
}

void InspectWidget::entity_Changed(const std::set<std::string>&) {
	updateAttributeString();
	mChangedThisFrame = true;
}


bool InspectWidget::ChildList_MouseDoubleClick(const CEGUI::EventArgs&) {
	//Inspect the child entity
	CEGUI::ListboxItem* item = mChildList->getFirstSelectedItem();
	if (item) {
		startInspecting(static_cast<EmberEntity*>(item->getUserData()));
	}

	return true;
}


}
