Feat material unused property remove

Editor/Inspector/TraceAndOptimizeEditor.cs

@@ -84,4 +84,4 @@ protected override void OnInspectorGUIInner()
             serializedObject.ApplyModifiedProperties();
         }
     }
-}
+}

Editor/OptimizerPlugin.cs

@@ -1,4 +1,4 @@
-using System;
+using System;
 using Anatawa12.AvatarOptimizer.ndmf;
 using nadena.dev.ndmf;
 using nadena.dev.ndmf.builtin;
@@ -71,6 +71,7 @@ protected override void Configure()
                         .Then.Run(Processors.TraceAndOptimizes.AutoMergeSkinnedMesh.Instance)
                         .Then.Run(Processors.TraceAndOptimizes.FindUnusedObjects.Instance)
                         .Then.Run(Processors.TraceAndOptimizes.ConfigureRemoveZeroSizedPolygon.Instance)
+                        .Then.Run(Processors.TraceAndOptimizes.MaterialUnusedPropertyRemove.Instance)
                         .Then.Run(Processors.MergeBoneProcessor.Instance)
                         .Then.Run(Processors.RemoveZeroSizedPolygonProcessor.Instance)
                         .Then.Run(Processors.AnimatorOptimizer.RemoveInvalidProperties.Instance)

Editor/Processors/TraceAndOptimize/MaterialUnusedPropertyRemove.cs

@@ -0,0 +1,91 @@
+using System.Linq;
+using nadena.dev.ndmf;
+using UnityEditor;
+using UnityEngine;
+
+namespace Anatawa12.AvatarOptimizer.Processors.TraceAndOptimizes
+{
+    internal class MaterialUnusedPropertyRemove : TraceAndOptimizePass<MaterialUnusedPropertyRemove>
+    {
+        public override string DisplayName => "T&O: MaterialUnusedPropertyRemove";
+        protected override void Execute(BuildContext context, TraceAndOptimizeState state)
+        {
+            if (!state.RemoveUnusedObjects) { return; }
+            if (state.SkipRemoveMaterialUnusedProperties) { return; }
+
+            var renderers = context.GetComponents<Renderer>();
+            var swapDict = renderers.SelectMany(i => i.sharedMaterials)
+                .Distinct().Where(i => i != null)
+                .ToDictionary(i => i, MaterialCleaning);
+
+            void SwapMaterialArray(Material[] matArray)
+            {
+                for (var i = 0; matArray.Length > i; i += 1)
+                {
+                    if (matArray[i] == null) { continue; }
+                    matArray[i] = swapDict[matArray[i]];
+                }
+            }
+
+            foreach (var renderer in renderers)
+            {
+                if (renderer is SkinnedMeshRenderer smr)
+                {
+                    var meshInfo = context.GetMeshInfoFor(smr);
+                    foreach (var subMesh in meshInfo.SubMeshes)
+                        SwapMaterialArray(subMesh.SharedMaterials);
+                }
+                else
+                {
+                    var matArray = renderer.sharedMaterials;
+                    SwapMaterialArray(matArray);
+                    renderer.sharedMaterials = matArray;
+                }
+            }
+
+            foreach (var matKv in swapDict) ObjectRegistry.RegisterReplacedObject(matKv.Key, matKv.Value);
+        }
+
+        static Material MaterialCleaning(Material i)
+        {
+            var mat = UnityEngine.Object.Instantiate(i);
+            mat.name = i.name + "&AAO_MATERIAL_UNUSED_PROPERTIES_REMOLDED";
+            RemoveUnusedProperties(mat);
+            return mat;
+        }
+
+        //MIT License
+        //Copyright (c) 2020-2021 lilxyzw
+        //https://github.com/lilxyzw/lilToon/blob/b96470d3dd9092b840052578048b2307fe6d8786/Assets/lilToon/Editor/lilMaterialUtils.cs#L658-L686
+        //
+        //https://light11.hatenadiary.com/entry/2018/12/04/224253
+        public static void RemoveUnusedProperties(Material material)
+        {
+            var so = new SerializedObject(material);
+            so.Update();
+            var savedProps = so.FindProperty("m_SavedProperties");
+
+            var texs = savedProps.FindPropertyRelative("m_TexEnvs");
+            DeleteUnused(ref texs, material);
+
+            var floats = savedProps.FindPropertyRelative("m_Floats");
+            DeleteUnused(ref floats, material);
+
+            var colors = savedProps.FindPropertyRelative("m_Colors");
+            DeleteUnused(ref colors, material);
+
+            so.ApplyModifiedProperties();
+        }
+
+        public static void DeleteUnused(ref SerializedProperty props, Material material)
+        {
+            for (int i = props.arraySize - 1; i >= 0; i--)
+            {
+                if (!material.HasProperty(props.GetArrayElementAtIndex(i).FindPropertyRelative("first").stringValue))
+                {
+                    props.DeleteArrayElementAtIndex(i);
+                }
+            }
+        }
+    }
+}

Internal/TraceAndOptimizeBase/TraceAndOptimizeProcessor.cs

@@ -34,6 +34,7 @@ public class TraceAndOptimizeState
         public bool SkipMergeMaterialAnimatingSkinnedMesh;
         public bool SkipMergeMaterials;
         public bool SkipRemoveEmptySubMesh;
+        public bool SkipRemoveMaterialUnusedProperties;
 
         public Dictionary<SkinnedMeshRenderer, HashSet<string>> PreserveBlendShapes =
             new Dictionary<SkinnedMeshRenderer, HashSet<string>>();
@@ -68,6 +69,7 @@ internal void Initialize(TraceAndOptimize config)
             SkipMergeMaterialAnimatingSkinnedMesh = config.advancedSettings.skipMergeMaterialAnimatingSkinnedMesh;
             SkipMergeMaterials = config.advancedSettings.skipMergeMaterials;
             SkipRemoveEmptySubMesh = config.advancedSettings.skipRemoveEmptySubMesh;
+            SkipRemoveMaterialUnusedProperties = config.advancedSettings.skipRemoveMaterialUnusedProperties;
 
             Enabled = true;
         }

Runtime/TraceAndOptimize.cs

@@ -125,6 +125,8 @@ internal struct AdvancedSettings
             public bool skipMergeMaterials;
             [ToggleLeft]
             public bool skipRemoveEmptySubMesh;
+            [ToggleLeft]
+            public bool skipRemoveMaterialUnusedProperties;
         }
     }
 }