worldspawn/src/eclass_xml.cpp

/*
   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
 */

///\file
///\brief EntityClass plugin that supports the .ent xml entity-definition format.
///
/// the .ent xml format expresses entity-definitions.
///
/// <!-- defines an entity which cannot have brushes grouped with it -->
/// <point name="[name of entity type]" colour="[RGB floating-point colour shown in editor]"
///   box="[minXYZ maxXYZ floating point bounding-box]" model="[model path]">
///   <!-- attribute definitions go here -->
/// </point>
///
/// <!-- defines an entity which can have brushes grouped with it -->
/// <group name="[name of entity type]" colour="[RGB floating-point colour shown in editor]">
///   <!-- attribute definitions go here -->
/// </group>
///
///
/// the attributes of an entity type are defined like this:
///
///   <[name of attribute type]
///     key="[entity key name]"
///     name="[name shown in gui]"
///     value="[default entity key value]"
///   >[comment text shown in gui]</[name of attribute type]>
///
/// each attribute type has a specialised attribute-editor GUI
///
/// currently-supported attribute types:
///
/// string          a string
/// array           an array of strings - value is a semi-colon-delimited string
/// integer         an integer value
/// boolean         an integer - shows as a checkbox - true = non-zero
/// integer2        two integer values
/// integer3        three integer values
/// real3           three floating-point values
/// angle           specialisation of real - Yaw angle
/// direction       specialisation of real - Yaw angle, -1 = down, -2 = up
/// angles          specialisation of real3 - Pitch Yaw Roll
/// color           specialisation of real3 - RGB floating-point colour
/// target          a string that uniquely identifies an entity or group of entities
/// targetname      a string that uniquely identifies an entity or group of entities
/// sound           the VFS path to a sound file
/// texture         the VFS path to a texture file or a shader name
/// model           the VFS path to a model file
/// skin            the VFS path to a skin file
///
///
/// flag attributes define a flag in the "spawnflags" key:
///
///   <flag
///     key="[flag name]"
///     name="[name shown in gui]"
///     bit="[bit-index in spawnflags]"
///   >[comment text shown in gui]<flag>
///
/// the default value for a flag bit is always 0.
///
///
/// List attributes have a set of valid values.
/// Create new list attribute types like this:
///
/// <list name="[name of list attribute type]">
///   <item name="[first name shown in menu]" value="[entity key value]"/>
///   <item name="[second name shown in menu]" value="[entity key value]"/>
/// </list>
///
/// these can then be used as attribute types.
///
///
/// An attribute definition should specify a default value that corresponds
/// with the default value given by the game. If the default value is not
/// specified in the attribute definition, it is assumed to be an empty string.
///
/// If the currently-selected entity in Radiant does not specify a value for
/// the key of an attribute, the default value from the attribute-definition
/// will be displayed in the attribute-editor and used when visualising the
/// entity in the preview windows. E.g. the Doom3 "light" entity has a
/// "light_radius" key. Light entities without a "light_radius" key are
/// displayed in Doom3 with a radius of 300. The default value for the
/// "light_radius" attribute definition should be specified as "300 300 300".
///




#include "eclass_xml.h"

#include "ieclass.h"
#include "irender.h"
#include "ifilesystem.h"
#include "iarchive.h"

#include "xml/xmlparser.h"
#include "generic/object.h"
#include "generic/reference.h"
#include "stream/stringstream.h"
#include "stream/textfilestream.h"
#include "os/path.h"
#include "eclasslib.h"
#include "modulesystem/moduleregistry.h"
#include "stringio.h"

#define PARSE_ERROR(elementName, name) makeQuoted( elementName ) << " is not a valid child of " << makeQuoted( name )

class IgnoreBreaks {
public:
const char *m_first;
const char *m_last;

IgnoreBreaks(const char *first, const char *last) : m_first(first), m_last(last)
{
}
};

template<typename TextOutputStreamType>
TextOutputStreamType &ostream_write(TextOutputStreamType &ostream, const IgnoreBreaks &ignoreBreaks)
{
	for (const char *i = ignoreBreaks.m_first; i != ignoreBreaks.m_last; ++i) {
		if (*i != '\n') {
			ostream << *i;
		}
	}
	return ostream;
}

namespace {

class TreeXMLImporter : public TextOutputStream {
public:
virtual TreeXMLImporter &pushElement(const XMLElement &element) = 0;

virtual void popElement(const char *name) = 0;
};

template<typename Type>
class Storage {
char m_storage[sizeof(Type)];
public:
Type &get()
{
	return *reinterpret_cast<Type *>( m_storage );
}

const Type &get() const
{
	return *reinterpret_cast<const Type *>( m_storage );
}
};

class BreakImporter : public TreeXMLImporter {
public:
BreakImporter(StringOutputStream &comment)
{
	comment << '\n';
}

static const char *name()
{
	return "n";
}

TreeXMLImporter &pushElement(const XMLElement &element)
{
	ERROR_MESSAGE(PARSE_ERROR(element.name(), name()));
	return *this;
}

void popElement(const char *elementName)
{
	ERROR_MESSAGE(PARSE_ERROR(elementName, name()));
}

std::size_t write(const char *data, std::size_t length)
{
	return length;
}
};

class AttributeImporter : public TreeXMLImporter {
StringOutputStream &m_comment;

public:
AttributeImporter(StringOutputStream &comment, EntityClass *entityClass, const XMLElement &element) : m_comment(
		comment)
{
	const char *type = element.name();
	const char *key = element.attribute("key");
	const char *name = element.attribute("name");
	const char *value = element.attribute("value");

	ASSERT_MESSAGE(!string_empty(key), "key attribute not specified");
	ASSERT_MESSAGE(!string_empty(name), "name attribute not specified");

	if (string_equal(type, "flag")) {
		std::size_t bit = atoi(element.attribute("bit"));
		ASSERT_MESSAGE(bit < MAX_FLAGS, "invalid flag bit");
		ASSERT_MESSAGE(string_empty(entityClass->flagnames[bit]), "non-unique flag bit");
		strcpy(entityClass->flagnames[bit], key);
	}

	m_comment << key;
	m_comment << " : ";

	EntityClass_insertAttribute(*entityClass, key, EntityClassAttribute(type, name, value));
}

~AttributeImporter()
{
}

TreeXMLImporter &pushElement(const XMLElement &element)
{
	ERROR_MESSAGE(PARSE_ERROR(element.name(), "attribute"));
	return *this;
}

void popElement(const char *elementName)
{
	ERROR_MESSAGE(PARSE_ERROR(elementName, "attribute"));
}

std::size_t write(const char *data, std::size_t length)
{
	return m_comment.write(data, length);
}
};

bool attributeSupported(const char *name)
{
	return string_equal(name, "real")
	       || string_equal(name, "integer")
	       || string_equal(name, "boolean")
	       || string_equal(name, "string")
	       || string_equal(name, "array")
	       || string_equal(name, "flag")
	       || string_equal(name, "real3")
	       || string_equal(name, "integer3")
	       || string_equal(name, "direction")
	       || string_equal(name, "angle")
	       || string_equal(name, "angles")
	       || string_equal(name, "color")
	       || string_equal(name, "target")
	       || string_equal(name, "targetname")
	       || string_equal(name, "sound")
	       || string_equal(name, "texture")
	       || string_equal(name, "model")
	       || string_equal(name, "skin")
	       || string_equal(name, "integer2");
}

typedef std::map<CopiedString, ListAttributeType> ListAttributeTypes;

bool listAttributeSupported(ListAttributeTypes &listTypes, const char *name)
{
	return listTypes.find(name) != listTypes.end();
}


class ClassImporter : public TreeXMLImporter {
EntityClassCollector &m_collector;
EntityClass *m_eclass;
StringOutputStream m_comment;
Storage<AttributeImporter> m_attribute;
ListAttributeTypes &m_listTypes;

public:
ClassImporter(EntityClassCollector &collector, ListAttributeTypes &listTypes, const XMLElement &element)
	: m_collector(collector), m_listTypes(listTypes)
{
	m_eclass = Eclass_Alloc();
	m_eclass->free = &Eclass_Free;

	const char *name = element.attribute("name");
	ASSERT_MESSAGE(!string_empty(name), "name attribute not specified for class");
	m_eclass->m_name = name;

	const char *color = element.attribute("color");
	ASSERT_MESSAGE(!string_empty(name), "color attribute not specified for class " << name);
	string_parse_vector3(color, m_eclass->color);
	eclass_capture_state(m_eclass);

	const char *model = element.attribute("model");
	if (!string_empty(model)) {
		StringOutputStream buffer(256);
		buffer << PathCleaned(model);
		m_eclass->m_modelpath = buffer.c_str();
	}

	const char *type = element.name();
	if (string_equal(type, "point")) {
		const char *box = element.attribute("box");
		ASSERT_MESSAGE(!string_empty(box), "box attribute not found for class " << name);
		m_eclass->fixedsize = true;
		string_parse_vector(box, &m_eclass->mins.x(), &m_eclass->mins.x() + 6);
	}
}

~ClassImporter()
{
	m_eclass->m_comments = m_comment.c_str();
	m_collector.insert(m_eclass);

	for (ListAttributeTypes::iterator i = m_listTypes.begin(); i != m_listTypes.end(); ++i) {
		m_collector.insert((*i).first.c_str(), (*i).second);
	}
}

static const char *name()
{
	return "class";
}

TreeXMLImporter &pushElement(const XMLElement &element)
{
	if (attributeSupported(element.name()) || listAttributeSupported(m_listTypes, element.name())) {
		constructor(m_attribute.get(), makeReference(m_comment), m_eclass, element);
		return m_attribute.get();
	} else {
		ERROR_MESSAGE(PARSE_ERROR(element.name(), name()));
		return *this;
	}
}

void popElement(const char *elementName)
{
	if (attributeSupported(elementName) || listAttributeSupported(m_listTypes, elementName)) {
		destructor(m_attribute.get());
	} else {
		ERROR_MESSAGE(PARSE_ERROR(elementName, name()));
	}
}

std::size_t write(const char *data, std::size_t length)
{
	return m_comment.write(data, length);
}
};

class ItemImporter : public TreeXMLImporter {
public:
ItemImporter(ListAttributeType &list, const XMLElement &element)
{
	const char *name = element.attribute("name");
	const char *value = element.attribute("value");
	list.push_back(name, value);
}

TreeXMLImporter &pushElement(const XMLElement &element)
{
	ERROR_MESSAGE(PARSE_ERROR(element.name(), "item"));
	return *this;
}

void popElement(const char *elementName)
{
	ERROR_MESSAGE(PARSE_ERROR(elementName, "item"));
}

std::size_t write(const char *data, std::size_t length)
{
	return length;
}
};

bool isItem(const char *name)
{
	return string_equal(name, "item");
}

class ListAttributeImporter : public TreeXMLImporter {
ListAttributeType *m_listType;
Storage<ItemImporter> m_item;
public:
ListAttributeImporter(ListAttributeTypes &listTypes, const XMLElement &element)
{
	const char *name = element.attribute("name");
	m_listType = &listTypes[name];
}

TreeXMLImporter &pushElement(const XMLElement &element)
{
	if (isItem(element.name())) {
		constructor(m_item.get(), makeReference(*m_listType), element);
		return m_item.get();
	} else {
		ERROR_MESSAGE(PARSE_ERROR(element.name(), "list"));
		return *this;
	}
}

void popElement(const char *elementName)
{
	if (isItem(elementName)) {
		destructor(m_item.get());
	} else {
		ERROR_MESSAGE(PARSE_ERROR(elementName, "list"));
	}
}

std::size_t write(const char *data, std::size_t length)
{
	return length;
}
};

bool classSupported(const char *name)
{
	return string_equal(name, "group")
	       || string_equal(name, "point");
}

bool listSupported(const char *name)
{
	return string_equal(name, "list");
}

class ClassesImporter : public TreeXMLImporter {
EntityClassCollector &m_collector;
Storage<ClassImporter> m_class;
Storage<ListAttributeImporter> m_list;
ListAttributeTypes m_listTypes;

public:
ClassesImporter(EntityClassCollector &collector) : m_collector(collector)
{
}

static const char *name()
{
	return "classes";
}

TreeXMLImporter &pushElement(const XMLElement &element)
{
	if (classSupported(element.name())) {
		constructor(m_class.get(), makeReference(m_collector), makeReference(m_listTypes), element);
		return m_class.get();
	} else if (listSupported(element.name())) {
		constructor(m_list.get(), makeReference(m_listTypes), element);
		return m_list.get();
	} else {
		ERROR_MESSAGE(PARSE_ERROR(element.name(), name()));
		return *this;
	}
}

void popElement(const char *elementName)
{
	if (classSupported(elementName)) {
		destructor(m_class.get());
	} else if (listSupported(elementName)) {
		destructor(m_list.get());
	} else {
		ERROR_MESSAGE(PARSE_ERROR(elementName, name()));
	}
}

std::size_t write(const char *data, std::size_t length)
{
	return length;
}
};

class EclassXMLImporter : public TreeXMLImporter {
EntityClassCollector &m_collector;
Storage<ClassesImporter> m_classes;

public:
EclassXMLImporter(EntityClassCollector &collector) : m_collector(collector)
{
}

static const char *name()
{
	return "classes";
}

TreeXMLImporter &pushElement(const XMLElement &element)
{
	if (string_equal(element.name(), ClassesImporter::name())) {
		constructor(m_classes.get(), makeReference(m_collector));
		return m_classes.get();
	} else {
		ERROR_MESSAGE(PARSE_ERROR(element.name(), name()));
		return *this;
	}
}

void popElement(const char *elementName)
{
	if (string_equal(elementName, ClassesImporter::name())) {
		destructor(m_classes.get());
	} else {
		ERROR_MESSAGE(PARSE_ERROR(elementName, name()));
	}
}

std::size_t write(const char *data, std::size_t length)
{
	return length;
}
};

class TreeXMLImporterStack : public XMLImporter {
std::vector<Reference<TreeXMLImporter> > m_importers;
public:
TreeXMLImporterStack(TreeXMLImporter &importer)
{
	m_importers.push_back(makeReference(importer));
}

void pushElement(const XMLElement &element)
{
	m_importers.push_back(makeReference(m_importers.back().get().pushElement(element)));
}

void popElement(const char *name)
{
	m_importers.pop_back();
	m_importers.back().get().popElement(name);
}

std::size_t write(const char *buffer, std::size_t length)
{
	return m_importers.back().get().write(buffer, length);
}
};


const char *GetExtension()
{
	return "ent";
}

void ScanFile(EntityClassCollector &collector, const char *filename)
{
	TextFileInputStream inputFile(filename);
	if (!inputFile.failed()) {
		XMLStreamParser parser(inputFile);

		EclassXMLImporter importer(collector);
		TreeXMLImporterStack stack(importer);
		parser.exportXML(stack);
	}
}


}

#include "modulesystem/singletonmodule.h"

class EntityClassXMLDependencies : public GlobalFileSystemModuleRef, public GlobalShaderCacheModuleRef {
};

class EclassXMLAPI {
EntityClassScanner m_eclassxml;
public:
typedef EntityClassScanner Type;

STRING_CONSTANT(Name, "xml");

EclassXMLAPI()
{
	m_eclassxml.scanFile = &ScanFile;
	m_eclassxml.getExtension = &GetExtension;
}

EntityClassScanner *getTable()
{
	return &m_eclassxml;
}
};

typedef SingletonModule<EclassXMLAPI, EntityClassXMLDependencies> EclassXMLModule;
typedef Static<EclassXMLModule> StaticEclassXMLModule;
StaticRegisterModule staticRegisterEclassXML(StaticEclassXMLModule::instance());