OniSplit/Akira/AkiraDaeReader.cs

using System;
using System.Collections.Generic;
using System.IO;
using Oni.Imaging;

namespace Oni.Akira
{
    internal class AkiraDaeReader
    {
        #region Private data
        private readonly PolygonMesh mesh;
        private readonly List<Vector3> positions;
        private readonly Dictionary<Vector3, int> uniquePositions;
        private readonly List<Vector3> normals;
        private readonly List<Vector2> texCoords;
        private readonly Dictionary<Dae.Material, Material> materialMap;
        private readonly Dictionary<string, Material> materialFileMap;
        private readonly Stack<Matrix> nodeTransformStack;

        private Dae.Scene scene;
        private Dictionary<string, AkiraDaeNodeProperties> properties;
        private Matrix nodeTransform;
        private string nodeName;
        #endregion

        public static PolygonMesh Read(IEnumerable<string> filePaths)
        {
            var reader = new AkiraDaeReader();
            var properties = new Dictionary<string, AkiraDaeNodeProperties>();

            foreach (var filePath in filePaths)
                reader.ReadScene(Dae.Reader.ReadFile(filePath), properties);

            return reader.mesh;
        }

        public AkiraDaeReader()
        {
            mesh = new PolygonMesh(new MaterialLibrary());

            positions = mesh.Points;
            uniquePositions = new Dictionary<Vector3, int>();
            texCoords = mesh.TexCoords;
            normals = mesh.Normals;

            materialMap = new Dictionary<Dae.Material, Material>();
            materialFileMap = new Dictionary<string, Material>(StringComparer.OrdinalIgnoreCase);

            nodeTransformStack = new Stack<Matrix>();
            nodeTransform = Matrix.Identity;
        }

        public PolygonMesh Mesh => mesh;

        public void ReadScene(Dae.Scene scene, Dictionary<string, AkiraDaeNodeProperties> properties)
        {
            this.scene = scene;
            this.properties = properties;

            AkiraDaeNodeProperties sceneProperties;
            properties.TryGetValue(scene.Id, out sceneProperties);

            foreach (var node in scene.Nodes)
                ReadNode(node, sceneProperties);
        }

        private void ReadNode(Dae.Node node, AkiraDaeNodeProperties parentNodeProperties)
        {
            AkiraDaeNodeProperties nodeProperties;

            if (!properties.TryGetValue(node.Id, out nodeProperties))
                nodeProperties = parentNodeProperties;
            else if (nodeProperties.HasPhysics)
                return;

            nodeTransformStack.Push(nodeTransform);

            foreach (var transform in node.Transforms)
                nodeTransform = transform.ToMatrix() * nodeTransform;

            nodeName = node.Name;

            foreach (var geometryInstance in node.GeometryInstances)
                ReadGeometryInstance(node, nodeProperties, geometryInstance);

            foreach (var child in node.Nodes)
                ReadNode(child, nodeProperties);

            nodeTransform = nodeTransformStack.Pop();
        }

        private void ReadGeometryInstance(Dae.Node node, AkiraDaeNodeProperties nodeProperties, Dae.GeometryInstance instance)
        {
            foreach (var primitives in instance.Target.Primitives)
            {
                if (primitives.PrimitiveType != Dae.MeshPrimitiveType.Polygons)
                {
                    Console.Error.WriteLine("Unsupported primitive type '{0}' found in geometry '{1}', ignoring.", primitives.PrimitiveType, instance.Name);
                    continue;
                }

                ReadPolygonPrimitives(node, nodeProperties, primitives, instance.Materials.Find(m => m.Symbol == primitives.MaterialSymbol));
            }
        }

        private Material ReadMaterial(Dae.Material material)
        {
            if (material == null || material.Effect == null)
                return null;

            Material polygonMaterial;

            if (materialMap.TryGetValue(material, out polygonMaterial))
                return polygonMaterial;

            Dae.EffectSampler diffuseSampler = null;
            Dae.EffectSampler transparentSampler = null;

            foreach (var texture in material.Effect.Textures)
            {
                if (texture.Channel == Dae.EffectTextureChannel.Diffuse)
                    diffuseSampler = texture.Sampler;
                else if (texture.Channel == Dae.EffectTextureChannel.Transparent)
                    transparentSampler = texture.Sampler;
            }

            if (diffuseSampler == null || diffuseSampler.Surface == null || diffuseSampler.Surface.InitFrom == null)
            {
                //
                // this material doesn't have a diffuse texture
                //

                return null;
            }

            var image = diffuseSampler.Surface.InitFrom;

            if (materialFileMap.TryGetValue(image.FilePath, out polygonMaterial))
                return polygonMaterial;

            polygonMaterial = mesh.Materials.GetMaterial(Path.GetFileNameWithoutExtension(image.FilePath));
            polygonMaterial.ImageFilePath = image.FilePath;

            if (transparentSampler == diffuseSampler)
                polygonMaterial.Flags |= GunkFlags.Transparent | GunkFlags.NoOcclusion | GunkFlags.TwoSided;

            materialFileMap.Add(image.FilePath, polygonMaterial);
            materialMap.Add(material, polygonMaterial);

            return polygonMaterial;
        }

        private void ReadPolygonPrimitives(Dae.Node node, AkiraDaeNodeProperties nodeProperties, Dae.MeshPrimitives primitives, Dae.MaterialInstance materialInstance)
        {
            Material material = null;

            if (materialInstance != null)
                material = ReadMaterial(materialInstance.Target);

            if (material == null)
                material = mesh.Materials.NotFound;

            int[] positionIndices = null;
            int[] texCoordIndices = null;
            int[] normalIndices = null;
            Color[] colors = null;

            foreach (var input in primitives.Inputs)
            {
                switch (input.Semantic)
                {
                    case Dae.Semantic.Position:
                        positionIndices = ReadInputIndexed(input, positions, uniquePositions, PositionReader);
                        break;

                    case Dae.Semantic.TexCoord:
                        texCoordIndices = ReadInputIndexed(input, texCoords, Dae.Source.ReadTexCoord);
                        break;

                    case Dae.Semantic.Normal:
                        normalIndices = ReadInputIndexed(input, normals, Dae.Source.ReadVector3);
                        break;

                    case Dae.Semantic.Color:
                        colors = ReadInput(input, Dae.Source.ReadColor);
                        break;
                }
            }

            if (texCoordIndices == null)
                Console.Error.WriteLine("Geometry '{0}' does not contain texture coordinates.", nodeName);

            int startIndex = 0;
            int degeneratePolygonCount = 0;

            foreach (int vertexCount in primitives.VertexCounts)
            {
                var polygonPointIndices = new int[vertexCount];
                Array.Copy(positionIndices, startIndex, polygonPointIndices, 0, vertexCount);

                if (CheckDegenerate(positions, polygonPointIndices))
                {
                    degeneratePolygonCount++;
                    startIndex += vertexCount;
                    continue;
                }

                var polygon = new Polygon(mesh, polygonPointIndices)
                {
                    FileName = node.FileName,
                    ObjectName = node.Name,
                    Material = material
                };

                if (texCoordIndices != null)
                {
                    polygon.TexCoordIndices = new int[vertexCount];
                    Array.Copy(texCoordIndices, startIndex, polygon.TexCoordIndices, 0, vertexCount);
                }
                else
                {
                    polygon.TexCoordIndices = new int[vertexCount];
                }

                if (normalIndices != null)
                {
                    polygon.NormalIndices = new int[vertexCount];
                    Array.Copy(normalIndices, startIndex, polygon.NormalIndices, 0, vertexCount);
                }

                if (colors != null)
                {
                    polygon.Colors = new Color[vertexCount];
                    Array.Copy(colors, startIndex, polygon.Colors, 0, vertexCount);
                }

                startIndex += vertexCount;

                if (nodeProperties != null)
                {
                    polygon.ScriptId = nodeProperties.ScriptId;
                    polygon.Flags |= nodeProperties.GunkFlags;
                }

                if (material == mesh.Materials.Markers.Ghost)
                    mesh.Ghosts.Add(polygon);
                else if (material == mesh.Materials.Markers.DoorFrame)
                    mesh.Doors.Add(polygon);
                else if (material.Name.StartsWith("bnv_grid_", StringComparison.Ordinal))
                    mesh.Floors.Add(polygon);
                else
                    mesh.Polygons.Add(polygon);
            }

            if (degeneratePolygonCount > 0)
            {
                Console.Error.WriteLine("Ignoring {0} degenerate polygons", degeneratePolygonCount);
            }
        }

        private static bool CheckDegenerate(List<Vector3> positions, int[] positionIndices)
        {
            if (positionIndices.Length < 3)
                return true;

            var p0 = positions[positionIndices[0]];
            var p1 = positions[positionIndices[1]];

            for (int i = 2; i < positionIndices.Length; i++)
            {
                var p2 = positions[positionIndices[i]];

                var s0 = p0 - p1;
                var s1 = p2 - p1;

                Vector3 c;
                Vector3.Cross(ref s0, ref s1, out c);

                if (Math.Abs(c.LengthSquared()) < 0.0001f && Vector3.Dot(s0, s1) > 0.0f)
                    return true;

                p0 = p1;
                p1 = p2;
            }

            return false;
        }

        private static int[] ReadInputIndexed<T>(Dae.IndexedInput input, List<T> list, Func<Dae.Source, int, T> elementReader)
            where T : struct
        {
            var indices = new int[input.Indices.Count];

            for (int i = 0; i < input.Indices.Count; i++)
            {
                var v = elementReader(input.Source, input.Indices[i]);
                indices[i] = list.Count;
                list.Add(v);
            }

            return indices;
        }

        private static int[] ReadInputIndexed<T>(Dae.IndexedInput input, List<T> list, Dictionary<T, int> uniqueList, Func<Dae.Source, int, T> elementReader)
            where T : struct
        {
            var indices = new int[input.Indices.Count];

            for (int i = 0; i < input.Indices.Count; i++)
            {
                var v = elementReader(input.Source, input.Indices[i]);

                int index;

                if (!uniqueList.TryGetValue(v, out index))
                {
                    index = list.Count;
                    list.Add(v);
                    uniqueList.Add(v, index);
                }

                indices[i] = index;
            }

            return indices;
        }

        private static T[] ReadInput<T>(Dae.IndexedInput input, Func<Dae.Source, int, T> elementReader)
            where T : struct
        {
            var values = new T[input.Indices.Count];

            for (int i = 0; i < input.Indices.Count; i++)
                values[i] = elementReader(input.Source, input.Indices[i]);

            return values;
        }

        private Vector3 PositionReader(Dae.Source source, int index)
        {
            Vector3 p = Dae.Source.ReadVector3(source, index);
            Vector3 r;
            Vector3.Transform(ref p, ref nodeTransform, out r);
            return r;
        }
    }
}
Revision:	1114
Committed:	Wed Jan 22 14:08:57 2020 UTC (5 years, 9 months ago) by iritscen
File size:	12602 byte(s)
Log Message:	Adding OniSplit source code (v0.9.99.0). Many thanks to Neo for all his work over the years.
#	Content
1	using System;
2	using System.Collections.Generic;
3	using System.IO;
4	using Oni.Imaging;
5
6	namespace Oni.Akira
7	{
8	internal class AkiraDaeReader
9	{
10	#region Private data
11	private readonly PolygonMesh mesh;
12	private readonly List<Vector3> positions;
13	private readonly Dictionary<Vector3, int> uniquePositions;
14	private readonly List<Vector3> normals;
15	private readonly List<Vector2> texCoords;
16	private readonly Dictionary<Dae.Material, Material> materialMap;
17	private readonly Dictionary<string, Material> materialFileMap;
18	private readonly Stack<Matrix> nodeTransformStack;
19
20	private Dae.Scene scene;
21	private Dictionary<string, AkiraDaeNodeProperties> properties;
22	private Matrix nodeTransform;
23	private string nodeName;
24	#endregion
25
26	public static PolygonMesh Read(IEnumerable<string> filePaths)
27	{
28	var reader = new AkiraDaeReader();
29	var properties = new Dictionary<string, AkiraDaeNodeProperties>();
30
31	foreach (var filePath in filePaths)
32	reader.ReadScene(Dae.Reader.ReadFile(filePath), properties);
33
34	return reader.mesh;
35	}
36
37	public AkiraDaeReader()
38	{
39	mesh = new PolygonMesh(new MaterialLibrary());
40
41	positions = mesh.Points;
42	uniquePositions = new Dictionary<Vector3, int>();
43	texCoords = mesh.TexCoords;
44	normals = mesh.Normals;
45
46	materialMap = new Dictionary<Dae.Material, Material>();
47	materialFileMap = new Dictionary<string, Material>(StringComparer.OrdinalIgnoreCase);
48
49	nodeTransformStack = new Stack<Matrix>();
50	nodeTransform = Matrix.Identity;
51	}
52
53	public PolygonMesh Mesh => mesh;
54
55	public void ReadScene(Dae.Scene scene, Dictionary<string, AkiraDaeNodeProperties> properties)
56	{
57	this.scene = scene;
58	this.properties = properties;
59
60	AkiraDaeNodeProperties sceneProperties;
61	properties.TryGetValue(scene.Id, out sceneProperties);
62
63	foreach (var node in scene.Nodes)
64	ReadNode(node, sceneProperties);
65	}
66
67	private void ReadNode(Dae.Node node, AkiraDaeNodeProperties parentNodeProperties)
68	{
69	AkiraDaeNodeProperties nodeProperties;
70
71	if (!properties.TryGetValue(node.Id, out nodeProperties))
72	nodeProperties = parentNodeProperties;
73	else if (nodeProperties.HasPhysics)
74	return;
75
76	nodeTransformStack.Push(nodeTransform);
77
78	foreach (var transform in node.Transforms)
79	nodeTransform = transform.ToMatrix() * nodeTransform;
80
81	nodeName = node.Name;
82
83	foreach (var geometryInstance in node.GeometryInstances)
84	ReadGeometryInstance(node, nodeProperties, geometryInstance);
85
86	foreach (var child in node.Nodes)
87	ReadNode(child, nodeProperties);
88
89	nodeTransform = nodeTransformStack.Pop();
90	}
91
92	private void ReadGeometryInstance(Dae.Node node, AkiraDaeNodeProperties nodeProperties, Dae.GeometryInstance instance)
93	{
94	foreach (var primitives in instance.Target.Primitives)
95	{
96	if (primitives.PrimitiveType != Dae.MeshPrimitiveType.Polygons)
97	{
98	Console.Error.WriteLine("Unsupported primitive type '{0}' found in geometry '{1}', ignoring.", primitives.PrimitiveType, instance.Name);
99	continue;
100	}
101
102	ReadPolygonPrimitives(node, nodeProperties, primitives, instance.Materials.Find(m => m.Symbol == primitives.MaterialSymbol));
103	}
104	}
105
106	private Material ReadMaterial(Dae.Material material)
107	{
108	if (material == null \|\| material.Effect == null)
109	return null;
110
111	Material polygonMaterial;
112
113	if (materialMap.TryGetValue(material, out polygonMaterial))
114	return polygonMaterial;
115
116	Dae.EffectSampler diffuseSampler = null;
117	Dae.EffectSampler transparentSampler = null;
118
119	foreach (var texture in material.Effect.Textures)
120	{
121	if (texture.Channel == Dae.EffectTextureChannel.Diffuse)
122	diffuseSampler = texture.Sampler;
123	else if (texture.Channel == Dae.EffectTextureChannel.Transparent)
124	transparentSampler = texture.Sampler;
125	}
126
127	if (diffuseSampler == null \|\| diffuseSampler.Surface == null \|\| diffuseSampler.Surface.InitFrom == null)
128	{
129	//
130	// this material doesn't have a diffuse texture
131	//
132
133	return null;
134	}
135
136	var image = diffuseSampler.Surface.InitFrom;
137
138	if (materialFileMap.TryGetValue(image.FilePath, out polygonMaterial))
139	return polygonMaterial;
140
141	polygonMaterial = mesh.Materials.GetMaterial(Path.GetFileNameWithoutExtension(image.FilePath));
142	polygonMaterial.ImageFilePath = image.FilePath;
143
144	if (transparentSampler == diffuseSampler)
145	polygonMaterial.Flags \|= GunkFlags.Transparent \| GunkFlags.NoOcclusion \| GunkFlags.TwoSided;
146
147	materialFileMap.Add(image.FilePath, polygonMaterial);
148	materialMap.Add(material, polygonMaterial);
149
150	return polygonMaterial;
151	}
152
153	private void ReadPolygonPrimitives(Dae.Node node, AkiraDaeNodeProperties nodeProperties, Dae.MeshPrimitives primitives, Dae.MaterialInstance materialInstance)
154	{
155	Material material = null;
156
157	if (materialInstance != null)
158	material = ReadMaterial(materialInstance.Target);
159
160	if (material == null)
161	material = mesh.Materials.NotFound;
162
163	int[] positionIndices = null;
164	int[] texCoordIndices = null;
165	int[] normalIndices = null;
166	Color[] colors = null;
167
168	foreach (var input in primitives.Inputs)
169	{
170	switch (input.Semantic)
171	{
172	case Dae.Semantic.Position:
173	positionIndices = ReadInputIndexed(input, positions, uniquePositions, PositionReader);
174	break;
175
176	case Dae.Semantic.TexCoord:
177	texCoordIndices = ReadInputIndexed(input, texCoords, Dae.Source.ReadTexCoord);
178	break;
179
180	case Dae.Semantic.Normal:
181	normalIndices = ReadInputIndexed(input, normals, Dae.Source.ReadVector3);
182	break;
183
184	case Dae.Semantic.Color:
185	colors = ReadInput(input, Dae.Source.ReadColor);
186	break;
187	}
188	}
189
190	if (texCoordIndices == null)
191	Console.Error.WriteLine("Geometry '{0}' does not contain texture coordinates.", nodeName);
192
193	int startIndex = 0;
194	int degeneratePolygonCount = 0;
195
196	foreach (int vertexCount in primitives.VertexCounts)
197	{
198	var polygonPointIndices = new int[vertexCount];
199	Array.Copy(positionIndices, startIndex, polygonPointIndices, 0, vertexCount);
200
201	if (CheckDegenerate(positions, polygonPointIndices))
202	{
203	degeneratePolygonCount++;
204	startIndex += vertexCount;
205	continue;
206	}
207
208	var polygon = new Polygon(mesh, polygonPointIndices)
209	{
210	FileName = node.FileName,
211	ObjectName = node.Name,
212	Material = material
213	};
214
215	if (texCoordIndices != null)
216	{
217	polygon.TexCoordIndices = new int[vertexCount];
218	Array.Copy(texCoordIndices, startIndex, polygon.TexCoordIndices, 0, vertexCount);
219	}
220	else
221	{
222	polygon.TexCoordIndices = new int[vertexCount];
223	}
224
225	if (normalIndices != null)
226	{
227	polygon.NormalIndices = new int[vertexCount];
228	Array.Copy(normalIndices, startIndex, polygon.NormalIndices, 0, vertexCount);
229	}
230
231	if (colors != null)
232	{
233	polygon.Colors = new Color[vertexCount];
234	Array.Copy(colors, startIndex, polygon.Colors, 0, vertexCount);
235	}
236
237	startIndex += vertexCount;
238
239	if (nodeProperties != null)
240	{
241	polygon.ScriptId = nodeProperties.ScriptId;
242	polygon.Flags \|= nodeProperties.GunkFlags;
243	}
244
245	if (material == mesh.Materials.Markers.Ghost)
246	mesh.Ghosts.Add(polygon);
247	else if (material == mesh.Materials.Markers.DoorFrame)
248	mesh.Doors.Add(polygon);
249	else if (material.Name.StartsWith("bnv_grid_", StringComparison.Ordinal))
250	mesh.Floors.Add(polygon);
251	else
252	mesh.Polygons.Add(polygon);
253	}
254
255	if (degeneratePolygonCount > 0)
256	{
257	Console.Error.WriteLine("Ignoring {0} degenerate polygons", degeneratePolygonCount);
258	}
259	}
260
261	private static bool CheckDegenerate(List<Vector3> positions, int[] positionIndices)
262	{
263	if (positionIndices.Length < 3)
264	return true;
265
266	var p0 = positions[positionIndices[0]];
267	var p1 = positions[positionIndices[1]];
268
269	for (int i = 2; i < positionIndices.Length; i++)
270	{
271	var p2 = positions[positionIndices[i]];
272
273	var s0 = p0 - p1;
274	var s1 = p2 - p1;
275
276	Vector3 c;
277	Vector3.Cross(ref s0, ref s1, out c);
278
279	if (Math.Abs(c.LengthSquared()) < 0.0001f && Vector3.Dot(s0, s1) > 0.0f)
280	return true;
281
282	p0 = p1;
283	p1 = p2;
284	}
285
286	return false;
287	}
288
289	private static int[] ReadInputIndexed<T>(Dae.IndexedInput input, List<T> list, Func<Dae.Source, int, T> elementReader)
290	where T : struct
291	{
292	var indices = new int[input.Indices.Count];
293
294	for (int i = 0; i < input.Indices.Count; i++)
295	{
296	var v = elementReader(input.Source, input.Indices[i]);
297	indices[i] = list.Count;
298	list.Add(v);
299	}
300
301	return indices;
302	}
303
304	private static int[] ReadInputIndexed<T>(Dae.IndexedInput input, List<T> list, Dictionary<T, int> uniqueList, Func<Dae.Source, int, T> elementReader)
305	where T : struct
306	{
307	var indices = new int[input.Indices.Count];
308
309	for (int i = 0; i < input.Indices.Count; i++)
310	{
311	var v = elementReader(input.Source, input.Indices[i]);
312
313	int index;
314
315	if (!uniqueList.TryGetValue(v, out index))
316	{
317	index = list.Count;
318	list.Add(v);
319	uniqueList.Add(v, index);
320	}
321
322	indices[i] = index;
323	}
324
325	return indices;
326	}
327
328	private static T[] ReadInput<T>(Dae.IndexedInput input, Func<Dae.Source, int, T> elementReader)
329	where T : struct
330	{
331	var values = new T[input.Indices.Count];
332
333	for (int i = 0; i < input.Indices.Count; i++)
334	values[i] = elementReader(input.Source, input.Indices[i]);
335
336	return values;
337	}
338
339	private Vector3 PositionReader(Dae.Source source, int index)
340	{
341	Vector3 p = Dae.Source.ReadVector3(source, index);
342	Vector3 r;
343	Vector3.Transform(ref p, ref nodeTransform, out r);
344	return r;
345	}
346	}
347	}