assimp/code/OgreStructs.cpp

/*
Open Asset Import Library (assimp)
----------------------------------------------------------------------

Copyright (c) 2006-2012, assimp team
All rights reserved.

Redistribution and use of this software in source and binary forms, 
with or without modification, are permitted provided that the 
following conditions are met:

* Redistributions of source code must retain the above
  copyright notice, this list of conditions and the
  following disclaimer.

* Redistributions in binary form must reproduce the above
  copyright notice, this list of conditions and the
  following disclaimer in the documentation and/or other
  materials provided with the distribution.

* Neither the name of the assimp team, nor the names of its
  contributors may be used to endorse or promote products
  derived from this software without specific prior
  written permission of the assimp team.

THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 
"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 
OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 
LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 
THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

----------------------------------------------------------------------
*/

#ifndef ASSIMP_BUILD_NO_OGRE_IMPORTER

#include "OgreStructs.h"
#include "TinyFormatter.h"

namespace Assimp
{
namespace Ogre
{

// VertexElement

VertexElement::VertexElement() : 
	index(0),
	source(0),
	offset(0),
	type(VET_FLOAT1),
	semantic(VES_POSITION)
{
}

size_t VertexElement::Size() const
{
	return TypeSize(type);
}

size_t VertexElement::ComponentCount() const
{
	return ComponentCount(type);
}

size_t VertexElement::ComponentCount(Type type)
{
	switch(type)
	{
		case VET_COLOUR:
		case VET_COLOUR_ABGR:
		case VET_COLOUR_ARGB:
		case VET_FLOAT1:
		case VET_DOUBLE1:
		case VET_SHORT1:
		case VET_USHORT1:
		case VET_INT1:
		case VET_UINT1:
			return 1;
		case VET_FLOAT2:
		case VET_DOUBLE2:
		case VET_SHORT2:
		case VET_USHORT2:
		case VET_INT2:
		case VET_UINT2:
			return 2;
		case VET_FLOAT3:
		case VET_DOUBLE3:
		case VET_SHORT3:
		case VET_USHORT3:
		case VET_INT3:
		case VET_UINT3:
			return 3;
		case VET_FLOAT4:
		case VET_DOUBLE4:
		case VET_SHORT4:
		case VET_USHORT4:
		case VET_INT4:
		case VET_UINT4:
		case VET_UBYTE4:
			return 4;
	}
	return 0;
}

size_t VertexElement::TypeSize(Type type)
{
	switch(type)
	{
		case VET_COLOUR:
		case VET_COLOUR_ABGR:
		case VET_COLOUR_ARGB:
			return sizeof(unsigned int);
		case VET_FLOAT1:
			return sizeof(float);
		case VET_FLOAT2:
			return sizeof(float)*2;
		case VET_FLOAT3:
			return sizeof(float)*3;
		case VET_FLOAT4:
			return sizeof(float)*4;
		case VET_DOUBLE1:
			return sizeof(double);
		case VET_DOUBLE2:
			return sizeof(double)*2;
		case VET_DOUBLE3:
			return sizeof(double)*3;
		case VET_DOUBLE4:
			return sizeof(double)*4;
		case VET_SHORT1:
			return sizeof(short);
		case VET_SHORT2:
			return sizeof(short)*2;
		case VET_SHORT3:
			return sizeof(short)*3;
		case VET_SHORT4:
			return sizeof(short)*4;
		case VET_USHORT1:
			return sizeof(unsigned short);
		case VET_USHORT2:
			return sizeof(unsigned short)*2;
		case VET_USHORT3:
			return sizeof(unsigned short)*3;
		case VET_USHORT4:
			return sizeof(unsigned short)*4;
		case VET_INT1:
			return sizeof(int);
		case VET_INT2:
			return sizeof(int)*2;
		case VET_INT3:
			return sizeof(int)*3;
		case VET_INT4:
			return sizeof(int)*4;
		case VET_UINT1:
			return sizeof(unsigned int);
		case VET_UINT2:
			return sizeof(unsigned int)*2;
		case VET_UINT3:
			return sizeof(unsigned int)*3;
		case VET_UINT4:
			return sizeof(unsigned int)*4;
		case VET_UBYTE4:
			return sizeof(unsigned char)*4;
	}
	return 0;
}

std::string VertexElement::TypeToString()
{
	return TypeToString(type);
}

std::string VertexElement::TypeToString(Type type)
{
	switch(type)
	{
		case VET_COLOUR:		return "COLOUR";
		case VET_COLOUR_ABGR:	return "COLOUR_ABGR";
		case VET_COLOUR_ARGB:	return "COLOUR_ARGB";
		case VET_FLOAT1:		return "FLOAT1";
		case VET_FLOAT2:		return "FLOAT2";
		case VET_FLOAT3:		return "FLOAT3";
		case VET_FLOAT4:		return "FLOAT4";
		case VET_DOUBLE1:		return "DOUBLE1";
		case VET_DOUBLE2:		return "DOUBLE2";
		case VET_DOUBLE3:		return "DOUBLE3";
		case VET_DOUBLE4:		return "DOUBLE4";
		case VET_SHORT1:		return "SHORT1";
		case VET_SHORT2:		return "SHORT2";
		case VET_SHORT3:		return "SHORT3";
		case VET_SHORT4:		return "SHORT4";
		case VET_USHORT1:		return "USHORT1";
		case VET_USHORT2:		return "USHORT2";
		case VET_USHORT3:		return "USHORT3";
		case VET_USHORT4:		return "USHORT4";
		case VET_INT1:			return "INT1";
		case VET_INT2:			return "INT2";
		case VET_INT3:			return "INT3";
		case VET_INT4:			return "INT4";
		case VET_UINT1:			return "UINT1";
		case VET_UINT2:			return "UINT2";
		case VET_UINT3:			return "UINT3";
		case VET_UINT4:			return "UINT4";
		case VET_UBYTE4:		return "UBYTE4";
	}
	return "Uknown_VertexElement::Type";
}

std::string VertexElement::SemanticToString()
{
	return SemanticToString(semantic);
}

std::string VertexElement::SemanticToString(Semantic semantic)
{
	switch(semantic)
	{
		case VES_POSITION:				return "POSITION";
		case VES_BLEND_WEIGHTS:			return "BLEND_WEIGHTS";
		case VES_BLEND_INDICES:			return "BLEND_INDICES";
		case VES_NORMAL:				return "NORMAL";
		case VES_DIFFUSE:				return "DIFFUSE";
		case VES_SPECULAR:				return "SPECULAR";
		case VES_TEXTURE_COORDINATES:	return "TEXTURE_COORDINATES";
		case VES_BINORMAL:				return "BINORMAL";
		case VES_TANGENT:				return "TANGENT";
	}
	return "Uknown_VertexElement::Semantic";
}

// IVertexData

IVertexData::IVertexData() :
	count(0)
{
}

bool IVertexData::HasBoneAssignments() const
{
	return !boneAssignments.empty();
}

void IVertexData::AddVertexMapping(uint32_t oldIndex, uint32_t newIndex)
{
	BoneAssignmentsForVertex(oldIndex, newIndex, boneAssignmentsMap[newIndex]);
	vertexIndexMapping[oldIndex].push_back(newIndex);
}

void IVertexData::BoneAssignmentsForVertex(uint32_t currentIndex, uint32_t newIndex, VertexBoneAssignmentList &dest) const
{
	for (VertexBoneAssignmentList::const_iterator iter=boneAssignments.begin(), end=boneAssignments.end();
		iter!=end; ++iter)
	{
		if (iter->vertexIndex == currentIndex)
		{
			VertexBoneAssignment a = (*iter);
			a.vertexIndex = newIndex;
			dest.push_back(a);
		}
	}
}

AssimpVertexBoneWeightList IVertexData::AssimpBoneWeights(size_t vertices)
{
	AssimpVertexBoneWeightList weights;	
	for(size_t vi=0; vi<vertices; ++vi)
	{
		VertexBoneAssignmentList &vertexWeights = boneAssignmentsMap[vi];
		for (VertexBoneAssignmentList::const_iterator iter=vertexWeights.begin(), end=vertexWeights.end();
			iter!=end; ++iter)
		{
			std::vector<aiVertexWeight> &boneWeights = weights[iter->boneIndex];
			boneWeights.push_back(aiVertexWeight(vi, iter->weight));
		}
	}
	return weights;
}

std::set<uint16_t> IVertexData::ReferencedBonesByWeights() const
{
	std::set<uint16_t> referenced;
	for (VertexBoneAssignmentList::const_iterator iter=boneAssignments.begin(), end=boneAssignments.end();
		iter!=end; ++iter)
	{
		referenced.insert(iter->boneIndex);
	}
	return referenced;
}

// VertexData

VertexData::VertexData()
{
}

VertexData::~VertexData()
{
	Reset();
}

void VertexData::Reset()
{
	// Releases shared ptr memory streams.
	vertexBindings.clear();
	vertexElements.clear();
}

uint32_t VertexData::VertexSize(uint16_t source) const
{
	uint32_t size = 0;
	for(VertexElementList::const_iterator iter=vertexElements.begin(), end=vertexElements.end(); iter != end; ++iter)
	{
		if (iter->source == source)
			size += iter->Size();
	}
	return size;
}

MemoryStream *VertexData::VertexBuffer(uint16_t source)
{
	if (vertexBindings.find(source) != vertexBindings.end())
		return vertexBindings[source];
	return 0;
}

VertexElement *VertexData::GetVertexElement(VertexElement::Semantic semantic, uint16_t index)
{
	for(VertexElementList::iterator iter=vertexElements.begin(), end=vertexElements.end(); iter != end; ++iter)
	{
		VertexElement &element = (*iter);
		if (element.semantic == semantic && element.index == index)
			return &element;
	}
	return 0;
}

// VertexDataXml

VertexDataXml::VertexDataXml()
{
}

bool VertexDataXml::HasNormals() const
{
	return !normals.empty();
}

bool VertexDataXml::HasTangents() const
{
	return !tangents.empty();
}

bool VertexDataXml::HasUvs() const
{
	return !uvs.empty();
}

size_t VertexDataXml::NumUvs() const
{
	return uvs.size();
}

// IndexData

IndexData::IndexData() :
	count(0),
	faceCount(0),
	is32bit(false)
{
}

IndexData::~IndexData()
{
	Reset();	
}

void IndexData::Reset()
{
	// Release shared ptr memory stream.
	buffer.reset();
}

size_t IndexData::IndexSize() const
{
	return (is32bit ? sizeof(uint32_t) : sizeof(uint16_t));
}

size_t IndexData::FaceSize() const
{
	return IndexSize() * 3;
}

// Mesh

Mesh::Mesh() :
	sharedVertexData(0),
	skeleton(0),
	hasSkeletalAnimations(false)
{
}

Mesh::~Mesh()
{
	Reset();
}

void Mesh::Reset()
{
	OGRE_SAFE_DELETE(skeleton)
	OGRE_SAFE_DELETE(sharedVertexData)

	for(size_t i=0, len=subMeshes.size(); i<len; ++i) {
		OGRE_SAFE_DELETE(subMeshes[i])
	}
	subMeshes.clear();
	for(size_t i=0, len=animations.size(); i<len; ++i) {
		OGRE_SAFE_DELETE(animations[i])
	}
	animations.clear();
	for(size_t i=0, len=poses.size(); i<len; ++i) {
		OGRE_SAFE_DELETE(poses[i])
	}
	poses.clear();
}

size_t Mesh::NumSubMeshes() const
{
	return subMeshes.size();
}

SubMesh *Mesh::GetSubMesh(uint16_t index) const
{
	for(size_t i=0; i<subMeshes.size(); ++i)
		if (subMeshes[i]->index == index)
			return subMeshes[i];
	return 0;
}

void Mesh::ConvertToAssimpScene(aiScene* dest)
{
	// Setup
	dest->mNumMeshes = NumSubMeshes();
	dest->mMeshes = new aiMesh*[dest->mNumMeshes];

	// Create root node
	dest->mRootNode = new aiNode();
	dest->mRootNode->mNumMeshes = dest->mNumMeshes;
	dest->mRootNode->mMeshes = new unsigned int[dest->mRootNode->mNumMeshes];

	// Export meshes
	for(size_t i=0; i<dest->mNumMeshes; ++i)
	{
		dest->mMeshes[i] = subMeshes[i]->ConvertToAssimpMesh(this);
		dest->mRootNode->mMeshes[i] = i;
	}
}

// ISubMesh

ISubMesh::ISubMesh() :
	index(0),
	materialIndex(-1),
	usesSharedVertexData(false),
	operationType(OT_POINT_LIST)
{
}

// SubMesh

SubMesh::SubMesh() :
	vertexData(0),
	indexData(new IndexData())
{
}

SubMesh::~SubMesh()
{
	Reset();
}

void SubMesh::Reset()
{
	OGRE_SAFE_DELETE(vertexData)
	OGRE_SAFE_DELETE(indexData)
}

aiMesh *SubMesh::ConvertToAssimpMesh(Mesh *parent)
{
	if (operationType != OT_TRIANGLE_LIST) {
		throw DeadlyImportError(Formatter::format() << "Only mesh operation type OT_TRIANGLE_LIST is supported. Found " << operationType);
	}
		
	aiMesh *dest = new aiMesh();
	dest->mPrimitiveTypes = aiPrimitiveType_TRIANGLE;

	if (!name.empty())
		dest->mName = name;

	// Material index	
	if (materialIndex != -1)
		dest->mMaterialIndex = materialIndex;

	// Pick source vertex data from shader geometry or from internal geometry.
	VertexData *src = (!usesSharedVertexData ? vertexData : parent->sharedVertexData);

	VertexElement *positionsElement = src->GetVertexElement(VertexElement::VES_POSITION);
	VertexElement *normalsElement   = src->GetVertexElement(VertexElement::VES_NORMAL);
	VertexElement *uv1Element       = src->GetVertexElement(VertexElement::VES_TEXTURE_COORDINATES, 0);
	VertexElement *uv2Element       = src->GetVertexElement(VertexElement::VES_TEXTURE_COORDINATES, 1);

	// Sanity checks
	if (!positionsElement) {
		throw DeadlyImportError("Failed to import Ogre VertexElement::VES_POSITION. Mesh does not have vertex positions!");
	} else if (positionsElement->type != VertexElement::VET_FLOAT3) {
		throw DeadlyImportError("Ogre Mesh position vertex element type != VertexElement::VET_FLOAT3. This is not supported.");
	} else if (normalsElement && normalsElement->type != VertexElement::VET_FLOAT3) {
		throw DeadlyImportError("Ogre Mesh normal vertex element type != VertexElement::VET_FLOAT3. This is not supported.");
	}

	// Faces
	dest->mNumFaces = indexData->faceCount;
	dest->mFaces = new aiFace[dest->mNumFaces];

	// Assimp required unique vertices, we need to convert from Ogres shared indexing.
	size_t uniqueVertexCount = dest->mNumFaces * 3;
	dest->mNumVertices = uniqueVertexCount;
	dest->mVertices = new aiVector3D[dest->mNumVertices];

	// Source streams
	MemoryStream *positions      = src->VertexBuffer(positionsElement->source);
	MemoryStream *normals        = (normalsElement ? src->VertexBuffer(normalsElement->source) : 0);
	MemoryStream *uv1            = (uv1Element ? src->VertexBuffer(uv1Element->source) : 0);
	MemoryStream *uv2            = (uv2Element ? src->VertexBuffer(uv2Element->source) : 0);

	// Element size
	const size_t sizePosition    = positionsElement->Size();
	const size_t sizeNormal      = (normalsElement ? normalsElement->Size() : 0);
	const size_t sizeUv1         = (uv1Element ? uv1Element->Size() : 0);
	const size_t sizeUv2         = (uv2Element ? uv2Element->Size() : 0);

	// Vertex width
	const size_t vWidthPosition  = src->VertexSize(positionsElement->source);
	const size_t vWidthNormal    = (normalsElement ? src->VertexSize(normalsElement->source) : 0);
	const size_t vWidthUv1       = (uv1Element ? src->VertexSize(uv1Element->source) : 0);
	const size_t vWidthUv2       = (uv2Element ? src->VertexSize(uv2Element->source) : 0);
	
	bool boneAssignments = src->HasBoneAssignments();
	
	// Prepare normals
	if (normals) 
		dest->mNormals = new aiVector3D[dest->mNumVertices];

	// Prepare UVs, ignoring incompatible UVs.
	if (uv1)
	{
		if (uv1Element->type == VertexElement::VET_FLOAT2 || uv1Element->type == VertexElement::VET_FLOAT3)
		{
			dest->mNumUVComponents[0] = uv1Element->ComponentCount();
			dest->mTextureCoords[0] = new aiVector3D[dest->mNumVertices];	
		}
		else
		{
			DefaultLogger::get()->warn(Formatter::format() << "Ogre imported UV0 type " << uv1Element->TypeToString() << " is not compatible with Assimp. Ignoring UV.");
			uv1 = 0;
		}
	}
	if (uv2)
	{
		if (uv2Element->type == VertexElement::VET_FLOAT2 || uv2Element->type == VertexElement::VET_FLOAT3)
		{
			dest->mNumUVComponents[1] = uv2Element->ComponentCount();
			dest->mTextureCoords[1] = new aiVector3D[dest->mNumVertices];	
		}
		else
		{
			DefaultLogger::get()->warn(Formatter::format() << "Ogre imported UV0 type " << uv2Element->TypeToString() << " is not compatible with Assimp. Ignoring UV.");
			uv2 = 0;
		}
	}

	aiVector3D *uv1Dest = (uv1 ? dest->mTextureCoords[0] : 0);
	aiVector3D *uv2Dest = (uv2 ? dest->mTextureCoords[1] : 0);

	MemoryStream *faces = indexData->buffer.get();
	for (size_t fi=0, isize=indexData->IndexSize(), fsize=indexData->FaceSize(); 
		 fi<dest->mNumFaces; ++fi)
	{
		// Source Ogre face
		aiFace ogreFace;
		ogreFace.mNumIndices = 3;
		ogreFace.mIndices = new unsigned int[3];

		faces->Seek(fi * fsize, aiOrigin_SET);
		if (indexData->is32bit)
		{
			faces->Read(&ogreFace.mIndices[0], isize, 3);
		}
		else
		{
			uint16_t iout = 0;
			for (size_t ii=0; ii<3; ++ii)
			{
				faces->Read(&iout, isize, 1);
				ogreFace.mIndices[ii] = static_cast<unsigned int>(iout);
			}
		}

		// Destination Assimp face
		aiFace &face = dest->mFaces[fi];
		face.mNumIndices = 3;
		face.mIndices = new unsigned int[3];

		const size_t pos = fi * 3;
		for (size_t v=0; v<3; ++v)
		{
			const size_t newIndex = pos + v;

			// Write face index
			face.mIndices[v] = newIndex;

			// Ogres vertex index to ref into the source buffers.
			const size_t ogreVertexIndex = ogreFace.mIndices[v];
			src->AddVertexMapping(ogreVertexIndex, newIndex);
			
			// Position
			positions->Seek((vWidthPosition * ogreVertexIndex) + positionsElement->offset, aiOrigin_SET);
			positions->Read(&dest->mVertices[newIndex], sizePosition, 1);

			// Normal
			if (normals)
			{
				normals->Seek((vWidthNormal * ogreVertexIndex) + normalsElement->offset, aiOrigin_SET);
				normals->Read(&dest->mNormals[newIndex], sizeNormal, 1);
			}
			// UV0
			if (uv1 && uv1Dest)
			{
				uv1->Seek((vWidthUv1 * ogreVertexIndex) + uv1Element->offset, aiOrigin_SET);
				uv1->Read(&uv1Dest[newIndex], sizeUv1, 1);
			}
			// UV1
			if (uv2 && uv2Dest)
			{
				uv2->Seek((vWidthUv2 * ogreVertexIndex) + uv2Element->offset, aiOrigin_SET);
				uv2->Read(&uv2Dest[newIndex], sizeUv2, 1);
			}
		}
	}
	
	// Bones and bone weights
	if (parent->skeleton && boneAssignments)
	{
		AssimpVertexBoneWeightList weights = src->AssimpBoneWeights(dest->mNumVertices);
		std::set<uint16_t> referencedBones = src->ReferencedBonesByWeights();

		dest->mNumBones = referencedBones.size();
		dest->mBones = new aiBone*[dest->mNumBones];

		size_t assimpBoneIndex = 0;
		for(std::set<uint16_t>::const_iterator rbIter=referencedBones.begin(), rbEnd=referencedBones.end(); rbIter != rbEnd; ++rbIter, ++assimpBoneIndex)
		{
			Bone *bone = parent->skeleton->BoneById((*rbIter));
			dest->mBones[assimpBoneIndex] = bone->ConvertToAssimpBone(parent->skeleton, weights[bone->id]);
		}
	}

	return dest;
}

// MeshXml

MeshXml::MeshXml() :
	sharedVertexData(0),
	skeleton(0)
{
}

MeshXml::~MeshXml()
{
	Reset();
}

void MeshXml::Reset()
{
	OGRE_SAFE_DELETE(skeleton)
	OGRE_SAFE_DELETE(sharedVertexData)

	for(size_t i=0, len=subMeshes.size(); i<len; ++i) {
		OGRE_SAFE_DELETE(subMeshes[i])
	}
	subMeshes.clear();
}

size_t MeshXml::NumSubMeshes() const
{
	return subMeshes.size();
}

SubMeshXml *MeshXml::GetSubMesh(uint16_t index) const
{
	for(size_t i=0; i<subMeshes.size(); ++i)
		if (subMeshes[i]->index == index)
			return subMeshes[i];
	return 0;
}

void MeshXml::ConvertToAssimpScene(aiScene* dest)
{
	// Setup
	dest->mNumMeshes = NumSubMeshes();
	dest->mMeshes = new aiMesh*[dest->mNumMeshes];

	// Create root node
	dest->mRootNode = new aiNode();
	dest->mRootNode->mNumMeshes = dest->mNumMeshes;
	dest->mRootNode->mMeshes = new unsigned int[dest->mRootNode->mNumMeshes];

	// Export meshes
	for(size_t i=0; i<dest->mNumMeshes; ++i)
	{
		dest->mMeshes[i] = subMeshes[i]->ConvertToAssimpMesh(this);
		dest->mRootNode->mMeshes[i] = i;
	}
	
	// Export skeleton
	if (skeleton)
	{
		// Bones
		if (!skeleton->bones.empty())
		{
			BoneList rootBones = skeleton->RootBones();
			dest->mRootNode->mNumChildren = rootBones.size();
			dest->mRootNode->mChildren = new aiNode*[dest->mRootNode->mNumChildren];
			
			for(size_t i=0, len=rootBones.size(); i<len; ++i)
			{
				dest->mRootNode->mChildren[i] = rootBones[i]->ConvertToAssimpNode(skeleton, dest->mRootNode);
			}
		}

		// Animations
		if (!skeleton->animations.empty())
		{
			dest->mNumAnimations = skeleton->animations.size();
			dest->mAnimations = new aiAnimation*[dest->mNumAnimations];
			
			for(size_t i=0, len=skeleton->animations.size(); i<len; ++i)
			{
				dest->mAnimations[i] = skeleton->animations[i]->ConvertToAssimpAnimation();
			}
		}
	}
}

// SubMeshXml

SubMeshXml::SubMeshXml() :
	vertexData(0),
	indexData(new IndexDataXml())
{
}

SubMeshXml::~SubMeshXml()
{
	Reset();
}

void SubMeshXml::Reset()
{
	OGRE_SAFE_DELETE(indexData)
	OGRE_SAFE_DELETE(vertexData)
}

aiMesh *SubMeshXml::ConvertToAssimpMesh(MeshXml *parent)
{
	aiMesh *dest = new aiMesh();
	dest->mPrimitiveTypes = aiPrimitiveType_TRIANGLE;

	if (!name.empty())
		dest->mName = name;

	// Material index	
	if (materialIndex != -1)
		dest->mMaterialIndex = materialIndex;

	// Faces
	dest->mNumFaces = indexData->faceCount;
	dest->mFaces = new aiFace[dest->mNumFaces];

	// Assimp required unique vertices, we need to convert from Ogres shared indexing.
	size_t uniqueVertexCount = dest->mNumFaces * 3;
	dest->mNumVertices = uniqueVertexCount;
	dest->mVertices = new aiVector3D[dest->mNumVertices];

	VertexDataXml *src = (!usesSharedVertexData ? vertexData : parent->sharedVertexData);
	bool boneAssignments = src->HasBoneAssignments();
	bool normals = src->HasNormals();
	size_t uvs = src->NumUvs();
	
	// Prepare normals
	if (normals) 
		dest->mNormals = new aiVector3D[dest->mNumVertices];

	// Prepare UVs
	for(size_t uvi=0; uvi<uvs; ++uvi)
	{
		dest->mNumUVComponents[uvi] = 2;
		dest->mTextureCoords[uvi] = new aiVector3D[dest->mNumVertices];
	}

	for (size_t fi=0; fi<dest->mNumFaces; ++fi)
	{
		// Source Ogre face
		aiFace &ogreFace = indexData->faces[fi];

		// Destination Assimp face
		aiFace &face = dest->mFaces[fi];
		face.mNumIndices = 3;
		face.mIndices = new unsigned int[3];

		const size_t pos = fi * 3;
		for (size_t v=0; v<3; ++v)
		{
			const size_t newIndex = pos + v;

			// Write face index
			face.mIndices[v] = newIndex;

			// Ogres vertex index to ref into the source buffers.
			const size_t ogreVertexIndex = ogreFace.mIndices[v];
			src->AddVertexMapping(ogreVertexIndex, newIndex);
			
			// Position
			dest->mVertices[newIndex] = src->positions[ogreVertexIndex];

			// Normal
			if (normals)
				dest->mNormals[newIndex] = src->normals[ogreVertexIndex];
				
			// UVs
			for(size_t uvi=0; uvi<uvs; ++uvi)
			{
				aiVector3D *uvDest = dest->mTextureCoords[uvi];
				std::vector<aiVector3D> &uvSrc = src->uvs[uvi];
				uvDest[newIndex] = uvSrc[ogreVertexIndex];
			}
		}
	}
	
	// Bones and bone weights
	if (parent->skeleton && boneAssignments)
	{
		AssimpVertexBoneWeightList weights = src->AssimpBoneWeights(dest->mNumVertices);
		std::set<uint16_t> referencedBones = src->ReferencedBonesByWeights();

		dest->mNumBones = referencedBones.size();
		dest->mBones = new aiBone*[dest->mNumBones];
		
		size_t assimpBoneIndex = 0;
		for(std::set<uint16_t>::const_iterator rbIter=referencedBones.begin(), rbEnd=referencedBones.end(); rbIter != rbEnd; ++rbIter, ++assimpBoneIndex)
		{
			Bone *bone = parent->skeleton->BoneById((*rbIter));
			dest->mBones[assimpBoneIndex] = bone->ConvertToAssimpBone(parent->skeleton, weights[bone->id]);
		}
	}

	return dest;
}

// Animation

Animation::Animation(Skeleton *parent) :
	parentSkeleton(parent),
	parentMesh(0),
	length(0.0f),
	baseTime(-1.0f)
{
}

Animation::Animation(Mesh *parent) : 
	parentMesh(parent),
	parentSkeleton(0),
	length(0.0f),
	baseTime(-1.0f)
{
}

VertexData *Animation::AssociatedVertexData(VertexAnimationTrack *track) const
{
	if (!parentMesh)
		return 0;

	bool sharedGeom = (track->target == 0);
	if (sharedGeom)
		return parentMesh->sharedVertexData;
	else
		return parentMesh->GetSubMesh(track->target-1)->vertexData;
}

aiAnimation *Animation::ConvertToAssimpAnimation()
{
	aiAnimation *anim = new aiAnimation();
	anim->mName = name;
	anim->mDuration = static_cast<double>(length);
	anim->mTicksPerSecond = 1.0;

	// Tracks
	if (!tracks.empty())
	{
		anim->mNumChannels = tracks.size();
		anim->mChannels = new aiNodeAnim*[anim->mNumChannels];
		
		for(size_t i=0, len=tracks.size(); i<len; ++i)
		{
			anim->mChannels[i] = tracks[i].ConvertToAssimpAnimationNode(parentSkeleton);
		}
	}
	return anim;
}

// Skeleton

Skeleton::Skeleton()
{
}

Skeleton::~Skeleton()
{
	Reset();
}

void Skeleton::Reset()
{
	for(size_t i=0, len=bones.size(); i<len; ++i) {
		OGRE_SAFE_DELETE(bones[i])
	}
	bones.clear();
	for(size_t i=0, len=animations.size(); i<len; ++i) {
		OGRE_SAFE_DELETE(animations[i])
	}
	animations.clear();
}

BoneList Skeleton::RootBones() const
{
	BoneList rootBones;
	for(BoneList::const_iterator iter = bones.begin(); iter != bones.end(); ++iter)
	{
		if (!(*iter)->IsParented())
			rootBones.push_back((*iter));
	}
	return rootBones;
}

size_t Skeleton::NumRootBones() const
{
	size_t num = 0;
	for(BoneList::const_iterator iter = bones.begin(); iter != bones.end(); ++iter)
	{
		if (!(*iter)->IsParented())
			num++;
	}
	return num;
}

Bone *Skeleton::BoneByName(const std::string &name) const
{
	for(BoneList::const_iterator iter = bones.begin(); iter != bones.end(); ++iter)
	{
		if ((*iter)->name == name)
			return (*iter);
	}
	return 0;
}

Bone *Skeleton::BoneById(uint16_t id) const
{
	for(BoneList::const_iterator iter = bones.begin(); iter != bones.end(); ++iter)
	{
		if ((*iter)->id == id)
			return (*iter);
	}
	return 0;
}

// Bone

Bone::Bone() :
	id(0),
	parent(0),
	parentId(-1),
	rotationAngle(0.0f),
	scale(1.0f, 1.0f, 1.0f)
{
}

bool Bone::IsParented() const
{
	return (parentId != -1 && parent != 0);
}

uint16_t Bone::ParentId() const
{
	return static_cast<uint16_t>(parentId);
}

void Bone::AddChild(Bone *bone)
{
	if (!bone)
		return;
	if (bone->IsParented())
		throw DeadlyImportError("Attaching child Bone that is already parented: " + bone->name);
		
	bone->parent = this;
	bone->parentId = id;
	children.push_back(bone->id);
}

void Bone::CalculateWorldMatrixAndDefaultPose(Skeleton *skeleton)
{
	aiMatrix4x4 t0, t1;
	aiMatrix4x4 transform = aiMatrix4x4::Rotation(-rotationAngle, rotation, t1) * aiMatrix4x4::Translation(-position, t0);

	if (!IsParented())
		worldMatrix = transform;
	else
		worldMatrix = transform * parent->worldMatrix;

	aiMatrix4x4 t2, t3; /// @todo t0 and t1 could probably be reused here?
	defaultPose = aiMatrix4x4::Translation(position, t2) * aiMatrix4x4::Rotation(rotationAngle, rotation, t3);

	// Recursively for all children now that the parent matrix has been calculated.
	for (size_t i=0, len=children.size(); i<len; ++i)
	{
		Bone *child = skeleton->BoneById(children[i]);
		if (!child) {
			throw DeadlyImportError(Formatter::format() << "CalculateWorldMatrixAndDefaultPose: Failed to find child bone " << children[i] << " for parent " << id << " " << name);
		}
		child->CalculateWorldMatrixAndDefaultPose(skeleton);
	}
}

aiNode *Bone::ConvertToAssimpNode(Skeleton *skeleton, aiNode *parentNode)
{
	aiMatrix4x4 t0,t1;

	// Bone node
	aiNode* node = new aiNode(name);
	node->mParent = parentNode;
	node->mTransformation = defaultPose;

	// Children
	if (!children.empty())
	{
		node->mNumChildren = children.size();
		node->mChildren = new aiNode*[node->mNumChildren];

		for(size_t i=0, len=children.size(); i<len; ++i)
		{
			Bone *child = skeleton->BoneById(children[i]);
			if (!child) {
				throw DeadlyImportError(Formatter::format() << "ConvertToAssimpNode: Failed to find child bone " << children[i] << " for parent " << id << " " << name);
			}
			node->mChildren[i] = child->ConvertToAssimpNode(skeleton, node);
		}
	}
	return node;
}

aiBone *Bone::ConvertToAssimpBone(Skeleton *parent, const std::vector<aiVertexWeight> &boneWeights)
{
	aiBone *bone = new aiBone();
	bone->mName = name;
	bone->mOffsetMatrix = worldMatrix;
	
	if (!boneWeights.empty())
	{
		bone->mNumWeights = boneWeights.size();
		bone->mWeights = new aiVertexWeight[boneWeights.size()];
		memcpy(bone->mWeights, &boneWeights[0], boneWeights.size() * sizeof(aiVertexWeight));
	}

	return bone;
}

// VertexAnimationTrack

VertexAnimationTrack::VertexAnimationTrack() :
	target(0),
	type(VAT_NONE)
{
}

aiNodeAnim *VertexAnimationTrack::ConvertToAssimpAnimationNode(Skeleton *skeleton)
{
	if (boneName.empty() || type != VAT_TRANSFORM) {
		throw DeadlyImportError("VertexAnimationTrack::ConvertToAssimpAnimationNode: Cannot convert track that has no target bone name or is not type of VAT_TRANSFORM");
	}

	aiNodeAnim *nodeAnim = new aiNodeAnim();
	nodeAnim->mNodeName = boneName;
	
	Bone *bone = skeleton->BoneByName(boneName);
	if (!bone) {
		throw DeadlyImportError("VertexAnimationTrack::ConvertToAssimpAnimationNode: Failed to find bone " + boneName + " from parent Skeleton");
	}

	// Keyframes
	size_t numKeyframes = transformKeyFrames.size();

	nodeAnim->mPositionKeys = new aiVectorKey[numKeyframes];				
	nodeAnim->mRotationKeys = new aiQuatKey[numKeyframes];
	nodeAnim->mScalingKeys = new aiVectorKey[numKeyframes];
	nodeAnim->mNumPositionKeys = numKeyframes;
	nodeAnim->mNumRotationKeys = numKeyframes;
	nodeAnim->mNumScalingKeys  = numKeyframes;

	for(size_t kfi=0; kfi<numKeyframes; ++kfi)
	{
		const TransformKeyFrame &kfSource = transformKeyFrames[kfi];

		// Create a matrix to transform a vector from the bones 
		// default pose to the bone bones in this animation key
		aiMatrix4x4 t0, t1;
		aiMatrix4x4 keyBonePose =
			aiMatrix4x4::Translation(kfSource.position, t0) *
			aiMatrix4x4(kfSource.rotation.GetMatrix()) *
			aiMatrix4x4::Scaling(kfSource.scale, t1);

		// Calculate the complete transformation from world space to bone space
		aiMatrix4x4 finalTransform = bone->defaultPose * keyBonePose;

		aiVector3D kfPos; aiQuaternion kfRot; aiVector3D kfScale;
		finalTransform.Decompose(kfScale, kfRot, kfPos);

		double t = static_cast<double>(kfSource.timePos);
		nodeAnim->mPositionKeys[kfi].mTime = t;
		nodeAnim->mRotationKeys[kfi].mTime = t;
		nodeAnim->mScalingKeys[kfi].mTime = t;

		nodeAnim->mPositionKeys[kfi].mValue = kfPos;					
		nodeAnim->mRotationKeys[kfi].mValue = kfRot;
		nodeAnim->mScalingKeys[kfi].mValue = kfScale;
	}

	return nodeAnim;
}

} // Ogre
} // Assimp

#endif // ASSIMP_BUILD_NO_OGRE_IMPORTER