/*
   Copyright (C) 2001-2006, William Joseph.
   All Rights Reserved.

   This file is part of GtkRadiant.

   GtkRadiant 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.

   GtkRadiant 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 GtkRadiant; if not, write to the Free Software
   Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 */

#include "entity.h"

#include "ifilter.h"
#include "selectable.h"
#include "namespace.h"

#include "scenelib.h"
#include "entitylib.h"
#include "eclasslib.h"
#include "pivot.h"

#include "targetable.h"
#include "uniquenames.h"
#include "namekeys.h"
#include "stream/stringstream.h"
#include "filters.h"


#include "miscmodel.h"
#include "prop_dynamic.h"
#include "light.h"
#include "group.h"
#include "eclassmodel.h"
#include "generic.h"
#include "doom3group.h"


EGameType g_gameType;

inline scene::Node &entity_for_eclass(EntityClass *eclass)
{
	if ( classname_equal(eclass->name(), "prop_dynamic" ) ) {
		return New_PropDynamic(eclass);
	} if ( classname_equal(eclass->name(), "prop_static" ) ) {
		return New_PropStatic(eclass);
	} else if (classname_equal(eclass->name(), "light")
	           || classname_equal(eclass->name(), "lightJunior")) {
		return New_Light(eclass);
	} else if (classname_equal(eclass->name(), "env_sound")
	           || classname_equal(eclass->name(), "lightJunior")) {
		return New_Light(eclass);
	}
	if (!eclass->fixedsize) {
		/*if (g_gameType == eGameTypeDoom3) {
		        return New_Doom3Group(eclass);
		   } else {*/
		return New_Group(eclass);
		/*}*/
	} else if (!string_empty(eclass->modelpath())) {
		return New_EclassModel(eclass);
	} else {
		return New_GenericEntity(eclass);
	}
}

void Entity_setName(Entity &entity, const char *name)
{
	entity.setKeyValue("name", name);
}

typedef ReferenceCaller<Entity, void (const char *), Entity_setName> EntitySetNameCaller;

inline Namespaced *Node_getNamespaced(scene::Node &node)
{
	return NodeTypeCast<Namespaced>::cast(node);
}

inline scene::Node &node_for_eclass(EntityClass *eclass)
{
	scene::Node &node = entity_for_eclass(eclass);
	Node_getEntity(node)->setKeyValue("classname", eclass->name());

	/*if (string_not_empty(eclass->name())
	    && !string_equal(eclass->name(), "worldspawn")
	    && !string_equal(eclass->name(), "UNKNOWN_CLASS")) {
	    char buffer[1024];
	    strcpy(buffer, eclass->name());
	    strcat(buffer, "_1");
	    GlobalNamespace().makeUnique(buffer, EntitySetNameCaller(*Node_getEntity(node)));
	   }*/

	Namespaced *namespaced = Node_getNamespaced(node);
	if (namespaced != 0) {
		namespaced->setNamespace(GlobalNamespace());
	}

	return node;
}

EntityCreator::KeyValueChangedFunc EntityKeyValues::m_entityKeyValueChanged = 0;
EntityCreator::KeyValueChangedFunc KeyValue::m_entityKeyValueChanged = 0;
Counter *EntityKeyValues::m_counter = 0;

bool g_showNames = false;
bool g_showAngles = false;
bool g_newLightDraw = true;
bool g_lightRadii = true;

class ConnectEntities {
public:
Entity *m_e1;
Entity *m_e2;
int m_index;

ConnectEntities(Entity *e1, Entity *e2, int index) : m_e1(e1), m_e2(e2), m_index(index)
{
}

const char *keyname()
{
	StringOutputStream key(16);
	if (m_index <= 0) {
		return "target";
	}
	if (m_index == 1) {
		return "killtarget";
	}
	key << "target" << m_index;
	return key.c_str();
}

void connect(const char *name)
{
	m_e1->setKeyValue(keyname(), name);
	m_e2->setKeyValue("targetname", name);
}

typedef MemberCaller<ConnectEntities, void (const char *), &ConnectEntities::connect> ConnectCaller;
};

inline Entity *ScenePath_getEntity(const scene::Path &path)
{
	Entity *entity = Node_getEntity(path.top());
	if (entity == 0) {
		entity = Node_getEntity(path.parent());
	}
	return entity;
}

class Quake3EntityCreator : public EntityCreator {
public:
scene::Node &createEntity(EntityClass *eclass)
{
	return node_for_eclass(eclass);
}

void setKeyValueChangedFunc(KeyValueChangedFunc func)
{
	EntityKeyValues::setKeyValueChangedFunc(func);
}

void setCounter(Counter *counter)
{
	EntityKeyValues::setCounter(counter);
}

void connectEntities(const scene::Path &path, const scene::Path &targetPath, int index)
{
	Entity *e1 = ScenePath_getEntity(path);
	Entity *e2 = ScenePath_getEntity(targetPath);

	if (e1 == 0 || e2 == 0) {
		globalErrorStream() << "entityConnectSelected: both of the selected instances must be an entity\n";
		return;
	}

	if (e1 == e2) {
		globalErrorStream()
		        << "entityConnectSelected: the selected instances must not both be from the same entity\n";
		return;
	}


	UndoableCommand undo("entityConnectSelected");

	if (g_gameType == eGameTypeDoom3) {
		StringOutputStream key(16);
		if (index >= 0) {
			key << "target";
			if (index != 0) {
				key << index;
			}
			e1->setKeyValue(key.c_str(), e2->getKeyValue("name"));
			key.clear();
		} else {
			for (unsigned int i = 0;; ++i) {
				key << "target";
				if (i != 0) {
					key << i;
				}
				const char *value = e1->getKeyValue(key.c_str());
				if (string_empty(value)) {
					e1->setKeyValue(key.c_str(), e2->getKeyValue("name"));
					break;
				}
				key.clear();
			}
		}
	} else {
		ConnectEntities connector(e1, e2, index);
		const char *value = e2->getKeyValue("targetname");
		if (!string_empty(value)) {
			connector.connect(value);
		} else {
			const char *type = e2->getKeyValue("classname");
			if (string_empty(type)) {
				type = "t";
			}
			StringOutputStream key(64);
			key << type << "1";
			GlobalNamespace().makeUnique(key.c_str(), ConnectEntities::ConnectCaller(connector));
		}
	}

	SceneChangeNotify();
}

void setLightRadii(bool lightRadii)
{
	g_lightRadii = lightRadii;
}

bool getLightRadii() const
{
	return g_lightRadii;
}

void setShowNames(bool showNames)
{
	g_showNames = showNames;
}

bool getShowNames()
{
	return g_showNames;
}

void setShowAngles(bool showAngles)
{
	g_showAngles = showAngles;
}

bool getShowAngles()
{
	return g_showAngles;
}

void printStatistics() const
{
	StringPool_analyse(EntityKeyValues::getPool());
}
};

Quake3EntityCreator g_Quake3EntityCreator;

EntityCreator &GetEntityCreator()
{
	return g_Quake3EntityCreator;
}


class filter_entity_classname : public EntityFilter {
const char *m_classname;
public:
filter_entity_classname(const char *classname) : m_classname(classname)
{
}

bool filter(const Entity &entity) const
{
	return string_equal(entity.getKeyValue("classname"), m_classname);
}
};

class filter_entity_classgroup : public EntityFilter {
const char *m_classgroup;
std::size_t m_length;
public:
filter_entity_classgroup(const char *classgroup) : m_classgroup(classgroup), m_length(string_length(m_classgroup))
{
}

bool filter(const Entity &entity) const
{
	return string_equal_n(entity.getKeyValue("classname"), m_classgroup, m_length);
}
};

/* this is such a bad hack because radiant sucks and I ran out of time */
class filter_entity_classtwo : public EntityFilter {
const char *m_classname1;
const char *m_classname2;
public:
filter_entity_classtwo(const char *classname1, const char *classname2) : m_classname1(classname1), m_classname2(classname2)
{
}

bool filter(const Entity &entity) const
{
	return string_equal(entity.getKeyValue("classname"), m_classname1) || string_equal(entity.getKeyValue("classname"), m_classname2);
}
};

filter_entity_classtwo g_filter_entity_world("worldspawn", "func_group");
filter_entity_classname g_filter_entity_func_group("func_group");
filter_entity_classname g_filter_entity_light("light");
filter_entity_classname g_filter_entity_prop_static("prop_static");
filter_entity_classname g_filter_entity_prop_dynamic("prop_dynamic");
filter_entity_classgroup g_filter_entity_trigger("trigger_");
filter_entity_classgroup g_filter_entity_path("path_");

class filter_entity_doom3model : public EntityFilter {
public:
bool filter(const Entity &entity) const
{
	return string_equal(entity.getKeyValue("classname"), "func_static")
	       && !string_equal(entity.getKeyValue("model"), entity.getKeyValue("name"));
}
};

filter_entity_doom3model g_filter_entity_doom3model;


void Entity_InitFilters()
{
	add_entity_filter(g_filter_entity_world, EXCLUDE_WORLD);
	add_entity_filter(g_filter_entity_func_group, EXCLUDE_GROUP);
	add_entity_filter(g_filter_entity_trigger, EXCLUDE_TRIGGERS);
	add_entity_filter(g_filter_entity_prop_static, EXCLUDE_MODELS);
	add_entity_filter(g_filter_entity_prop_dynamic, EXCLUDE_MODELS);
	add_entity_filter(g_filter_entity_doom3model, EXCLUDE_MODELS);
	add_entity_filter(g_filter_entity_light, EXCLUDE_LIGHTS);
	add_entity_filter(g_filter_entity_path, EXCLUDE_PATHS);
	add_entity_filter(g_filter_entity_world, EXCLUDE_ENT, true);
}


#include "preferencesystem.h"

void Entity_Construct(EGameType gameType)
{
	g_gameType = gameType;

	Static<KeyIsName>::instance().m_keyIsName = keyIsNameQuake3;
	Static<KeyIsName>::instance().m_nameKey = "targetname";


	GlobalPreferenceSystem().registerPreference("SI_ShowNames", make_property_string(g_showNames));
	GlobalPreferenceSystem().registerPreference("SI_ShowAngles", make_property_string(g_showAngles));
	GlobalPreferenceSystem().registerPreference("NewLightStyle", make_property_string(g_newLightDraw));
	GlobalPreferenceSystem().registerPreference("LightRadiuses", make_property_string(g_lightRadii));

	Entity_InitFilters();
	Light_Construct();
	PropStatic_construct();
	PropDynamic_construct();
	/* Doom3Group_construct();*/

	RenderablePivot::StaticShader::instance() = GlobalShaderCache().capture("$PIVOT");

	GlobalShaderCache().attachRenderable(StaticRenderableConnectionLines::instance());
}

void Entity_Destroy()
{
	GlobalShaderCache().detachRenderable(StaticRenderableConnectionLines::instance());

	GlobalShaderCache().release("$PIVOT");

	/*Doom3Group_destroy();*/
	PropStatic_destroy();
	PropDynamic_destroy();
	Light_Destroy();
}