OniSplit/Motoko/TextureImporter3.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.Xml;

namespace Oni.Motoko
{
    using Imaging;
    using Metadata;

    internal class TextureImporter3
    {
        private readonly string outputDirPath;
        private readonly Dictionary<string, TextureImporterOptions> textures = new Dictionary<string, TextureImporterOptions>(StringComparer.Ordinal);
        private TextureFormat defaultFormat = TextureFormat.BGR;
        private TextureFormat defaultSquareFormat = TextureFormat.BGR;
        private TextureFormat defaultAlphaFormat = TextureFormat.RGBA;
        private int maxSize = 512;

        public TextureImporter3(string outputDirPath)
        {
            this.outputDirPath = outputDirPath;
        }

        public TextureFormat DefaultFormat
        {
            get
            {
                return defaultFormat;
            }
            set
            {
                defaultFormat = value;
                defaultSquareFormat = value;
            }
        }

        public TextureFormat DefaultAlphaFormat
        {
            get { return defaultAlphaFormat; }
            set { defaultAlphaFormat = value; }
        }

        public int MaxSize
        {
            get { return maxSize; }
            set { maxSize = value; }
        }

        public string AddMaterial(Dae.Material material)
        {
            if (material == null || material.Effect == null)
                return null;

            var texture = material.Effect.Textures.FirstOrDefault(t => t.Channel == Dae.EffectTextureChannel.Diffuse);

            if (texture == null)
                return null;

            var sampler = texture.Sampler;

            if (sampler == null || sampler.Surface == null || sampler.Surface.InitFrom == null)
                return null;

            var filePath = Path.GetFullPath(sampler.Surface.InitFrom.FilePath);

            if (!File.Exists(filePath))
                return null;

            var options = GetOptions(Path.GetFileNameWithoutExtension(filePath), filePath);

            return options.Name;
        }

        public TextureImporterOptions AddMaterial(Akira.Material material)
        {
            return GetOptions(material.Name, material.ImageFilePath);
        }

        private TextureImporterOptions GetOptions(string name, string filePath)
        {
            TextureImporterOptions options;

            if (!textures.TryGetValue(name, out options))
            {
                options = new TextureImporterOptions
                {
                    Name = name,
                    Images = new[] { filePath }
                };

                textures.Add(name, options);
            }

            return options;
        }

        public TextureImporterOptions GetOptions(string name, bool create)
        {
            TextureImporterOptions options;

            if (!textures.TryGetValue(name, out options) && create)
            {
                options = new TextureImporterOptions
                {
                    Name = name
                };

                textures.Add(name, options);
            }

            return options;
        }

        public void ReadOptions(XmlReader xml, string basePath)
        {
            var options = GetOptions(xml.GetAttribute("Name"), true);
            var images = new List<string>();

            xml.ReadStartElement("Texture");

            while (xml.IsStartElement())
            {
                switch (xml.LocalName)
                {
                    case "Width":
                        options.Width = xml.ReadElementContentAsInt();
                        break;
                    case "Height":
                        options.Height = xml.ReadElementContentAsInt();
                        break;
                    case "Format":
                        options.Format = TextureImporter.ParseTextureFormat(xml.ReadElementContentAsString());
                        break;
                    case "Flags":
                        options.Flags = xml.ReadElementContentAsEnum<TextureFlags>();
                        break;
                    case "GunkFlags":
                        options.GunkFlags = xml.ReadElementContentAsEnum<Akira.GunkFlags>();
                        break;
                    case "EnvMap":
                        options.EnvironmentMap = xml.ReadElementContentAsString();
                        break;
                    case "Speed":
                        options.Speed = xml.ReadElementContentAsInt();
                        break;
                    case "Image":
                        images.Add(Path.Combine(basePath, xml.ReadElementContentAsString()));
                        break;
                    default:
                        Console.Error.WriteLine("Unknown texture option {0}", xml.LocalName);
                        xml.Skip();
                        break;
                }
            }

            xml.ReadEndElement();

            options.Images = images.ToArray();
        }

        public void Write()
        {
            Parallel.ForEach(textures.Values, options =>
            {
                if (options.Images.Length > 0)
                {
                    var writer = new TexImporter(this, options);
                    writer.Import();
                    writer.Write(outputDirPath);
                }
            });

            Console.WriteLine("Imported {0} textures", textures.Count);
        }

        private class TexImporter : Importer
        {
            private readonly TextureImporter3 importer;
            private readonly TextureImporterOptions options;

            public TexImporter(TextureImporter3 importer, TextureImporterOptions options)
            {
                this.importer = importer;
                this.options = options;

                BeginImport();
            }

            public void Import()
            {
                var surfaces = new List<Surface>();

                foreach (string imageFilePath in options.Images)
                    surfaces.Add(TextureUtils.LoadImage(imageFilePath));

                if (surfaces.Count == 0)
                    throw new InvalidDataException("No images found. A texture must have at least one image.");

                TextureFormat format;

                if (options.Format != null)
                {
                    format = options.Format.Value;
                }
                else
                {
                    var surface = surfaces[0];

                    if (surface.HasTransparentPixels())
                        format = importer.defaultAlphaFormat;
                    else if (surface.Width % 4 == 0 && surface.Height % 4 == 0)
                        format = importer.defaultSquareFormat;
                    else
                        format = importer.defaultFormat;
                }

                int imageWidth = 0;
                int imageHeight = 0;

                foreach (var surface in surfaces)
                {
                    if (imageWidth == 0)
                        imageWidth = surface.Width;
                    else if (imageWidth != surface.Width)
                        throw new NotSupportedException("All animation frames must have the same size.");

                    if (imageHeight == 0)
                        imageHeight = surface.Height;
                    else if (imageHeight != surface.Height)
                        throw new NotSupportedException("All animation frames must have the same size.");
                }

                int width = options.Width;
                int height = options.Height;

                if (width == 0)
                    width = imageWidth;
                else if (width > imageWidth)
                    throw new NotSupportedException("Cannot upscale images.");

                if (height == 0)
                    height = imageHeight;
                else if (height > imageHeight)
                    throw new NotSupportedException("Cannot upscale images.");

                if (width > importer.maxSize || height > importer.maxSize)
                {
                    if (width > height)
                    {
                        height = importer.maxSize * height / width;
                        width = importer.maxSize;
                    }
                    else
                    {
                        width = importer.maxSize * width / height;
                        height = importer.maxSize;
                    }
                }

                width = TextureUtils.RoundToPowerOf2(width);
                height = TextureUtils.RoundToPowerOf2(height);

                if (width != imageWidth || height != imageHeight)
                {
                    for (int i = 0; i < surfaces.Count; i++)
                        surfaces[i] = surfaces[i].Resize(width, height);
                }

                var flags = options.Flags | TextureFlags.HasMipMaps;
                flags &= ~(TextureFlags.HasEnvMap | TextureFlags.SwapBytes);
                var envMapName = options.EnvironmentMap;

                if (format != TextureFormat.RGBA)
                    flags |= TextureFlags.SwapBytes;

                if (!string.IsNullOrEmpty(envMapName))
                    flags |= TextureFlags.HasEnvMap;

                var name = options.Name;
                int speed = options.Speed;

                for (int i = 0; i < surfaces.Count; i++)
                {
                    var descriptor = CreateInstance(TemplateTag.TXMP, i == 0 ? name : null);

                    using (var writer = descriptor.OpenWrite(128))
                    {
                        writer.Write((int)flags);
                        writer.WriteUInt16(width);
                        writer.WriteUInt16(height);
                        writer.Write((int)format);

                        if (i == 0 && surfaces.Count > 1)
                            writer.WriteInstanceId(surfaces.Count);
                        else
                            writer.Write(0);

                        if (!string.IsNullOrEmpty(envMapName))
                            writer.WriteInstanceId(surfaces.Count + ((surfaces.Count > 1) ? 1 : 0));
                        else
                            writer.Write(0);

                        writer.Write(RawWriter.Align32());
                        writer.Skip(12);

                        var mainSurface = surfaces[i];

                        var levels = new List<Surface>(16);
                        levels.Add(mainSurface);

                        if ((flags & TextureFlags.HasMipMaps) != 0)
                        {
                            int mipWidth = width;
                            int mipHeight = height;

                            var surface = mainSurface;

                            while (mipWidth > 1 || mipHeight > 1)
                            {
                                mipWidth = Math.Max(mipWidth >> 1, 1);
                                mipHeight = Math.Max(mipHeight >> 1, 1);

                                surface = surface.Resize(mipWidth, mipHeight);

                                levels.Add(surface);
                            }
                        }

                        foreach (var level in levels)
                        {
                            var surface = level.Convert(format.ToSurfaceFormat());

                            RawWriter.Write(surface.Data);
                        }
                    }
                }

                if (surfaces.Count > 1)
                {
                    var txan = CreateInstance(TemplateTag.TXAN);

                    using (var writer = txan.OpenWrite(12))
                    {
                        writer.WriteInt16(speed);
                        writer.WriteInt16(speed);
                        writer.Write(0);
                        writer.Write(surfaces.Count);
                        writer.Write(0);

                        for (int i = 1; i < surfaces.Count; i++)
                            writer.WriteInstanceId(i);
                    }
                }

                if (!string.IsNullOrEmpty(envMapName))
                    CreateInstance(TemplateTag.TXMP, envMapName);
            }
        }
    }
}
Revision:	1114
Committed:	Wed Jan 22 14:08:57 2020 UTC (5 years, 8 months ago) by iritscen
File size:	12721 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 System.Xml;
5
6	namespace Oni.Motoko
7	{
8	using Imaging;
9	using Metadata;
10
11	internal class TextureImporter3
12	{
13	private readonly string outputDirPath;
14	private readonly Dictionary<string, TextureImporterOptions> textures = new Dictionary<string, TextureImporterOptions>(StringComparer.Ordinal);
15	private TextureFormat defaultFormat = TextureFormat.BGR;
16	private TextureFormat defaultSquareFormat = TextureFormat.BGR;
17	private TextureFormat defaultAlphaFormat = TextureFormat.RGBA;
18	private int maxSize = 512;
19
20	public TextureImporter3(string outputDirPath)
21	{
22	this.outputDirPath = outputDirPath;
23	}
24
25	public TextureFormat DefaultFormat
26	{
27	get
28	{
29	return defaultFormat;
30	}
31	set
32	{
33	defaultFormat = value;
34	defaultSquareFormat = value;
35	}
36	}
37
38	public TextureFormat DefaultAlphaFormat
39	{
40	get { return defaultAlphaFormat; }
41	set { defaultAlphaFormat = value; }
42	}
43
44	public int MaxSize
45	{
46	get { return maxSize; }
47	set { maxSize = value; }
48	}
49
50	public string AddMaterial(Dae.Material material)
51	{
52	if (material == null \|\| material.Effect == null)
53	return null;
54
55	var texture = material.Effect.Textures.FirstOrDefault(t => t.Channel == Dae.EffectTextureChannel.Diffuse);
56
57	if (texture == null)
58	return null;
59
60	var sampler = texture.Sampler;
61
62	if (sampler == null \|\| sampler.Surface == null \|\| sampler.Surface.InitFrom == null)
63	return null;
64
65	var filePath = Path.GetFullPath(sampler.Surface.InitFrom.FilePath);
66
67	if (!File.Exists(filePath))
68	return null;
69
70	var options = GetOptions(Path.GetFileNameWithoutExtension(filePath), filePath);
71
72	return options.Name;
73	}
74
75	public TextureImporterOptions AddMaterial(Akira.Material material)
76	{
77	return GetOptions(material.Name, material.ImageFilePath);
78	}
79
80	private TextureImporterOptions GetOptions(string name, string filePath)
81	{
82	TextureImporterOptions options;
83
84	if (!textures.TryGetValue(name, out options))
85	{
86	options = new TextureImporterOptions
87	{
88	Name = name,
89	Images = new[] { filePath }
90	};
91
92	textures.Add(name, options);
93	}
94
95	return options;
96	}
97
98	public TextureImporterOptions GetOptions(string name, bool create)
99	{
100	TextureImporterOptions options;
101
102	if (!textures.TryGetValue(name, out options) && create)
103	{
104	options = new TextureImporterOptions
105	{
106	Name = name
107	};
108
109	textures.Add(name, options);
110	}
111
112	return options;
113	}
114
115	public void ReadOptions(XmlReader xml, string basePath)
116	{
117	var options = GetOptions(xml.GetAttribute("Name"), true);
118	var images = new List<string>();
119
120	xml.ReadStartElement("Texture");
121
122	while (xml.IsStartElement())
123	{
124	switch (xml.LocalName)
125	{
126	case "Width":
127	options.Width = xml.ReadElementContentAsInt();
128	break;
129	case "Height":
130	options.Height = xml.ReadElementContentAsInt();
131	break;
132	case "Format":
133	options.Format = TextureImporter.ParseTextureFormat(xml.ReadElementContentAsString());
134	break;
135	case "Flags":
136	options.Flags = xml.ReadElementContentAsEnum<TextureFlags>();
137	break;
138	case "GunkFlags":
139	options.GunkFlags = xml.ReadElementContentAsEnum<Akira.GunkFlags>();
140	break;
141	case "EnvMap":
142	options.EnvironmentMap = xml.ReadElementContentAsString();
143	break;
144	case "Speed":
145	options.Speed = xml.ReadElementContentAsInt();
146	break;
147	case "Image":
148	images.Add(Path.Combine(basePath, xml.ReadElementContentAsString()));
149	break;
150	default:
151	Console.Error.WriteLine("Unknown texture option {0}", xml.LocalName);
152	xml.Skip();
153	break;
154	}
155	}
156
157	xml.ReadEndElement();
158
159	options.Images = images.ToArray();
160	}
161
162	public void Write()
163	{
164	Parallel.ForEach(textures.Values, options =>
165	{
166	if (options.Images.Length > 0)
167	{
168	var writer = new TexImporter(this, options);
169	writer.Import();
170	writer.Write(outputDirPath);
171	}
172	});
173
174	Console.WriteLine("Imported {0} textures", textures.Count);
175	}
176
177	private class TexImporter : Importer
178	{
179	private readonly TextureImporter3 importer;
180	private readonly TextureImporterOptions options;
181
182	public TexImporter(TextureImporter3 importer, TextureImporterOptions options)
183	{
184	this.importer = importer;
185	this.options = options;
186
187	BeginImport();
188	}
189
190	public void Import()
191	{
192	var surfaces = new List<Surface>();
193
194	foreach (string imageFilePath in options.Images)
195	surfaces.Add(TextureUtils.LoadImage(imageFilePath));
196
197	if (surfaces.Count == 0)
198	throw new InvalidDataException("No images found. A texture must have at least one image.");
199
200	TextureFormat format;
201
202	if (options.Format != null)
203	{
204	format = options.Format.Value;
205	}
206	else
207	{
208	var surface = surfaces[0];
209
210	if (surface.HasTransparentPixels())
211	format = importer.defaultAlphaFormat;
212	else if (surface.Width % 4 == 0 && surface.Height % 4 == 0)
213	format = importer.defaultSquareFormat;
214	else
215	format = importer.defaultFormat;
216	}
217
218	int imageWidth = 0;
219	int imageHeight = 0;
220
221	foreach (var surface in surfaces)
222	{
223	if (imageWidth == 0)
224	imageWidth = surface.Width;
225	else if (imageWidth != surface.Width)
226	throw new NotSupportedException("All animation frames must have the same size.");
227
228	if (imageHeight == 0)
229	imageHeight = surface.Height;
230	else if (imageHeight != surface.Height)
231	throw new NotSupportedException("All animation frames must have the same size.");
232	}
233
234	int width = options.Width;
235	int height = options.Height;
236
237	if (width == 0)
238	width = imageWidth;
239	else if (width > imageWidth)
240	throw new NotSupportedException("Cannot upscale images.");
241
242	if (height == 0)
243	height = imageHeight;
244	else if (height > imageHeight)
245	throw new NotSupportedException("Cannot upscale images.");
246
247	if (width > importer.maxSize \|\| height > importer.maxSize)
248	{
249	if (width > height)
250	{
251	height = importer.maxSize * height / width;
252	width = importer.maxSize;
253	}
254	else
255	{
256	width = importer.maxSize * width / height;
257	height = importer.maxSize;
258	}
259	}
260
261	width = TextureUtils.RoundToPowerOf2(width);
262	height = TextureUtils.RoundToPowerOf2(height);
263
264	if (width != imageWidth \|\| height != imageHeight)
265	{
266	for (int i = 0; i < surfaces.Count; i++)
267	surfaces[i] = surfaces[i].Resize(width, height);
268	}
269
270	var flags = options.Flags \| TextureFlags.HasMipMaps;
271	flags &= ~(TextureFlags.HasEnvMap \| TextureFlags.SwapBytes);
272	var envMapName = options.EnvironmentMap;
273
274	if (format != TextureFormat.RGBA)
275	flags \|= TextureFlags.SwapBytes;
276
277	if (!string.IsNullOrEmpty(envMapName))
278	flags \|= TextureFlags.HasEnvMap;
279
280	var name = options.Name;
281	int speed = options.Speed;
282
283	for (int i = 0; i < surfaces.Count; i++)
284	{
285	var descriptor = CreateInstance(TemplateTag.TXMP, i == 0 ? name : null);
286
287	using (var writer = descriptor.OpenWrite(128))
288	{
289	writer.Write((int)flags);
290	writer.WriteUInt16(width);
291	writer.WriteUInt16(height);
292	writer.Write((int)format);
293
294	if (i == 0 && surfaces.Count > 1)
295	writer.WriteInstanceId(surfaces.Count);
296	else
297	writer.Write(0);
298
299	if (!string.IsNullOrEmpty(envMapName))
300	writer.WriteInstanceId(surfaces.Count + ((surfaces.Count > 1) ? 1 : 0));
301	else
302	writer.Write(0);
303
304	writer.Write(RawWriter.Align32());
305	writer.Skip(12);
306
307	var mainSurface = surfaces[i];
308
309	var levels = new List<Surface>(16);
310	levels.Add(mainSurface);
311
312	if ((flags & TextureFlags.HasMipMaps) != 0)
313	{
314	int mipWidth = width;
315	int mipHeight = height;
316
317	var surface = mainSurface;
318
319	while (mipWidth > 1 \|\| mipHeight > 1)
320	{
321	mipWidth = Math.Max(mipWidth >> 1, 1);
322	mipHeight = Math.Max(mipHeight >> 1, 1);
323
324	surface = surface.Resize(mipWidth, mipHeight);
325
326	levels.Add(surface);
327	}
328	}
329
330	foreach (var level in levels)
331	{
332	var surface = level.Convert(format.ToSurfaceFormat());
333
334	RawWriter.Write(surface.Data);
335	}
336	}
337	}
338
339	if (surfaces.Count > 1)
340	{
341	var txan = CreateInstance(TemplateTag.TXAN);
342
343	using (var writer = txan.OpenWrite(12))
344	{
345	writer.WriteInt16(speed);
346	writer.WriteInt16(speed);
347	writer.Write(0);
348	writer.Write(surfaces.Count);
349	writer.Write(0);
350
351	for (int i = 1; i < surfaces.Count; i++)
352	writer.WriteInstanceId(i);
353	}
354	}
355
356	if (!string.IsNullOrEmpty(envMapName))
357	CreateInstance(TemplateTag.TXMP, envMapName);
358	}
359	}
360	}
361	}