Added TriangleIterator (experimental)

Former-commit-id: b94f4d4a26a92f4acecf4c53291db80756feb289
2013-01-04 16:27:22 +01:00
parent 1f6cac042b
commit d4030f666b
2 changed files with 212 additions and 0 deletions
--- a/src/Nazara/Utility/TriangleIterator.cpp
+++ b/src/Nazara/Utility/TriangleIterator.cpp
@@ -0,0 +1,166 @@
+// Copyright (C) 2012 Jérôme Leclercq
+// This file is part of the "Nazara Engine - Utility module"
+// For conditions of distribution and use, see copyright notice in Config.hpp
+
+#include <Nazara/Utility/TriangleIterator.hpp>
+#include <Nazara/Utility/SubMesh.hpp>
+#include <Nazara/Utility/Debug.hpp>
+
+NzTriangleIterator::NzTriangleIterator(NzSubMesh* subMesh, nzBufferAccess access) :
+m_primitiveType(subMesh->GetPrimitiveType()),
+m_indexMapper(subMesh->GetIndexBuffer(), nzBufferAccess_ReadOnly),
+m_vertexMapper(subMesh)
+{
+	NazaraUnused(access);
+
+	m_currentIndex = 3;
+	m_triangleIndices[0] = m_indexMapper.Get(0);
+	m_triangleIndices[1] = m_indexMapper.Get(1);
+	m_triangleIndices[2] = m_indexMapper.Get(2);
+
+	const NzIndexBuffer* indexBuffer = m_indexMapper.GetBuffer();
+	if (indexBuffer)
+		m_indexCount = indexBuffer->GetIndexCount();
+	else
+		m_indexCount = subMesh->GetVertexBuffer()->GetVertexCount();
+}
+
+bool NzTriangleIterator::Advance()
+{
+	if (m_currentIndex >= m_indexCount)
+	{
+		///FIXME: Que faire si Advance() est appelé deux fois
+		m_indexMapper.Unmap();
+		m_vertexMapper.Unmap();
+		return false;
+	}
+
+	switch (m_primitiveType)
+	{
+		case nzPrimitiveType_TriangleFan:
+			m_triangleIndices[1] = m_indexMapper.Get(m_currentIndex++);
+			m_triangleIndices[2] = m_indexMapper.Get(m_currentIndex++);
+			break;
+
+		case nzPrimitiveType_TriangleList:
+			m_triangleIndices[0] = m_indexMapper.Get(m_currentIndex++);
+			m_triangleIndices[1] = m_indexMapper.Get(m_currentIndex++);
+			m_triangleIndices[2] = m_indexMapper.Get(m_currentIndex++);
+			break;
+
+		case nzPrimitiveType_TriangleStrip:
+			m_triangleIndices[2] = m_indexMapper.Get(m_currentIndex++);
+			m_triangleIndices[1] = m_triangleIndices[2];
+			m_triangleIndices[0] = m_triangleIndices[1];
+			break;
+
+		default:
+			return false;
+	}
+
+	return true;
+}
+
+NzVector3f NzTriangleIterator::GetNormal(unsigned int i) const
+{
+	#if NAZARA_UTILITY_SAFE
+	if (i > 2)
+	{
+		NazaraError("Index out of range: (" + NzString::Number(i) + " > 2)");
+		return NzVector3f();
+	}
+	#endif
+
+	return m_vertexMapper.GetNormal(m_triangleIndices[i]);
+}
+
+NzVector3f NzTriangleIterator::GetPosition(unsigned int i) const
+{
+	#if NAZARA_UTILITY_SAFE
+	if (i > 2)
+	{
+		NazaraError("Index out of range: (" + NzString::Number(i) + " > 2)");
+		return NzVector3f();
+	}
+	#endif
+
+	return m_vertexMapper.GetPosition(m_triangleIndices[i]);
+}
+
+NzVector3f NzTriangleIterator::GetTangent(unsigned int i) const
+{
+	#if NAZARA_UTILITY_SAFE
+	if (i > 2)
+	{
+		NazaraError("Index out of range: (" + NzString::Number(i) + " > 2)");
+		return NzVector3f();
+	}
+	#endif
+
+	return m_vertexMapper.GetTangent(m_triangleIndices[i]);
+}
+
+NzVector2f NzTriangleIterator::GetTexCoords(unsigned int i) const
+{
+	#if NAZARA_UTILITY_SAFE
+	if (i > 2)
+	{
+		NazaraError("Index out of range: (" + NzString::Number(i) + " > 2)");
+		return NzVector2f();
+	}
+	#endif
+
+	return m_vertexMapper.GetTexCoords(m_triangleIndices[i]);
+}
+
+void NzTriangleIterator::SetNormal(unsigned int i, const NzVector3f& normal)
+{
+	#if NAZARA_UTILITY_SAFE
+	if (i > 2)
+	{
+		NazaraError("Index out of range: (" + NzString::Number(i) + " > 2)");
+		return;
+	}
+	#endif
+
+	m_vertexMapper.SetNormal(m_triangleIndices[i], normal);
+}
+
+void NzTriangleIterator::SetPosition(unsigned int i, const NzVector3f& position)
+{
+	#if NAZARA_UTILITY_SAFE
+	if (i > 2)
+	{
+		NazaraError("Index out of range: (" + NzString::Number(i) + " > 2)");
+		return;
+	}
+	#endif
+
+	m_vertexMapper.SetPosition(m_triangleIndices[i], position);
+}
+
+void NzTriangleIterator::SetTangent(unsigned int i, const NzVector3f& tangent)
+{
+	#if NAZARA_UTILITY_SAFE
+	if (i > 2)
+	{
+		NazaraError("Index out of range: (" + NzString::Number(i) + " > 2)");
+		return;
+	}
+	#endif
+
+	m_vertexMapper.SetTangent(m_triangleIndices[i], tangent);
+}
+
+void NzTriangleIterator::SetTexCoords(unsigned int i, const NzVector2f& texCoords)
+{
+	#if NAZARA_UTILITY_SAFE
+	if (i > 2)
+	{
+		NazaraError("Index out of range: (" + NzString::Number(i) + " > 2)");
+		return;
+	}
+	#endif
+
+	m_vertexMapper.SetTexCoords(m_triangleIndices[i], texCoords);
+}