-
Notifications
You must be signed in to change notification settings - Fork 14
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Deduplicated logic around generating materials between the low-level …
…and high-level GLTF code, and optimized how textures are converted into GLTF images.
- Loading branch information
1 parent
4e030db
commit 8d28824
Showing
7 changed files
with
188 additions
and
184 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
154 changes: 154 additions & 0 deletions
154
FinModelUtility/Fin/Fin/src/model/io/exporters/gltf/GltfMaterialManager.cs
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,154 @@ | ||
| using System; | ||
| using System.Collections.Concurrent; | ||
| using System.Collections.Generic; | ||
| using System.IO; | ||
| using System.Linq; | ||
| using System.Numerics; | ||
|
|
||
| using CommunityToolkit.HighPerformance.Helpers; | ||
|
|
||
| using fin.image; | ||
| using fin.model.util; | ||
| using fin.util.image; | ||
|
|
||
| using SharpGLTF.Materials; | ||
| using SharpGLTF.Memory; | ||
| using SharpGLTF.Schema2; | ||
|
|
||
| using AlphaMode = SharpGLTF.Materials.AlphaMode; | ||
|
|
||
| namespace fin.model.io.exporters.gltf { | ||
| public class GltfMaterialBuilder { | ||
| private readonly struct Fin2GltfImageConverter : IAction { | ||
| private readonly IImage[] finImages_; | ||
| private readonly IDictionary<IImage, MemoryImage> gltfImageByFinImage_; | ||
|
|
||
| public Fin2GltfImageConverter(IImage[] finImages, | ||
| IDictionary<IImage, MemoryImage> | ||
| gltfImageByFinImage) { | ||
| this.finImages_ = finImages; | ||
| this.gltfImageByFinImage_ = gltfImageByFinImage; | ||
| } | ||
|
|
||
| public void Invoke(int i) { | ||
| var finImage = this.finImages_[i]; | ||
|
|
||
| using var imageStream = new MemoryStream(); | ||
| finImage.ExportToStream(imageStream, LocalImageFormat.PNG); | ||
|
|
||
| this.gltfImageByFinImage_[finImage] = | ||
| new MemoryImage(imageStream.ToArray()); | ||
| } | ||
| } | ||
|
|
||
| public IDictionary<IMaterial, Material> GetMaterials( | ||
| ModelRoot gltfModelRoot, | ||
| IMaterialManager finMaterialManager) | ||
| => this.ConvertMaterials_(finMaterialManager) | ||
| .ToDictionary(tuple => tuple.Item1, | ||
| tuple => gltfModelRoot.CreateMaterial(tuple.Item2)); | ||
|
|
||
| public IDictionary<IMaterial, MaterialBuilder> GetMaterialBuilders( | ||
| IMaterialManager finMaterialManager) | ||
| => this.ConvertMaterials_(finMaterialManager) | ||
| .ToDictionary(tuple => tuple.Item1, tuple => tuple.Item2); | ||
|
|
||
| private IEnumerable<(IMaterial, MaterialBuilder)> ConvertMaterials_( | ||
| IMaterialManager finMaterialManager) { | ||
| var finImages = finMaterialManager.Textures | ||
| .Select(texture => texture.Image) | ||
| .Distinct() | ||
| .ToArray(); | ||
| var gltfImageByFinImage = new ConcurrentDictionary<IImage, MemoryImage>(); | ||
| ParallelHelper.For(0, | ||
| finImages.Length, | ||
| new Fin2GltfImageConverter( | ||
| finImages, | ||
| gltfImageByFinImage)); | ||
|
|
||
| // TODO: Update this if GLTF is ever extended... | ||
| return finMaterialManager.All.Select(finMaterial => { | ||
| var gltfMaterialBuilder | ||
| = new MaterialBuilder(finMaterial.Name) | ||
| .WithDoubleSide(finMaterial.CullingMode switch { | ||
| CullingMode.SHOW_FRONT_ONLY => false, | ||
| // Darn, guess we can't support this. | ||
| CullingMode.SHOW_BACK_ONLY => true, | ||
| CullingMode.SHOW_BOTH => true, | ||
| // Darn, guess we can't support this either. | ||
| CullingMode.SHOW_NEITHER => false, | ||
| _ => throw new ArgumentOutOfRangeException() | ||
| }) | ||
| .WithSpecularGlossinessShader() | ||
| .WithSpecularGlossiness(new Vector3(0), 0); | ||
|
|
||
| switch (finMaterial) { | ||
| case IStandardMaterial standardMaterial: { | ||
| var diffuseTexture = standardMaterial.DiffuseTexture; | ||
| if (diffuseTexture != null) { | ||
| gltfMaterialBuilder.UseChannel(KnownChannel.Diffuse) | ||
| .UseTexture(diffuseTexture, | ||
| gltfImageByFinImage[ | ||
| diffuseTexture.Image]); | ||
| } | ||
|
|
||
| var normalTexture = standardMaterial.NormalTexture; | ||
| if (normalTexture != null) { | ||
| gltfMaterialBuilder.UseChannel(KnownChannel.Normal) | ||
| .UseTexture(normalTexture, | ||
| gltfImageByFinImage | ||
| [normalTexture.Image]); | ||
| } | ||
|
|
||
| var emissiveTexture = standardMaterial.EmissiveTexture; | ||
| if (emissiveTexture != null) { | ||
| gltfMaterialBuilder.UseChannel(KnownChannel.Emissive) | ||
| .UseTexture(emissiveTexture, | ||
| gltfImageByFinImage[ | ||
| emissiveTexture.Image]); | ||
| } | ||
|
|
||
| /*var specularTexture = standardMaterial.SpecularTexture; | ||
| if (specularTexture != null) { | ||
| gltfMaterial.WithSpecularGlossiness( | ||
| GltfModelExporter.GetGltfImageFromFinTexture_( | ||
| specularTexture), new Vector3(.1f), .1f); | ||
| }*/ | ||
|
|
||
| var ambientOcclusionTexture = | ||
| standardMaterial.AmbientOcclusionTexture; | ||
| if (ambientOcclusionTexture != null) { | ||
| gltfMaterialBuilder | ||
| .UseChannel(KnownChannel.Occlusion) | ||
| .UseTexture(ambientOcclusionTexture, | ||
| gltfImageByFinImage[ | ||
| ambientOcclusionTexture.Image]); | ||
| } | ||
|
|
||
| break; | ||
| } | ||
| default: { | ||
| var texture = PrimaryTextureFinder.GetFor(finMaterial); | ||
| if (texture != null) { | ||
| var alphaMode = texture.TransparencyType switch { | ||
| ImageTransparencyType.OPAQUE => AlphaMode.OPAQUE, | ||
| ImageTransparencyType.MASK => AlphaMode.MASK, | ||
| ImageTransparencyType.TRANSPARENT => AlphaMode.BLEND, | ||
| _ => throw new ArgumentOutOfRangeException() | ||
| }; | ||
| gltfMaterialBuilder.WithAlpha(alphaMode); | ||
|
|
||
| gltfMaterialBuilder | ||
| .UseChannel(KnownChannel.Diffuse) | ||
| .UseTexture(texture, gltfImageByFinImage[texture.Image]); | ||
| } | ||
|
|
||
| break; | ||
| } | ||
| } | ||
|
|
||
| return (finMaterial, gltfMaterial: gltfMaterialBuilder); | ||
| }); | ||
| } | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.