Add extra materials baking

Author: vanifatovvlad

Runtime/Scripts/MeshAnimationAsset.cs

@@ -22,6 +22,11 @@ public class MeshAnimationAsset : ScriptableObject
         [ListDrawerSettings(Expanded = true, ShowPaging = false, AlwaysAddDefaultValue = true)]
         internal AnimationClip[] animationClips = new AnimationClip[0];
 
+        [Required]
+        [SerializeField]
+        [ListDrawerSettings(ShowPaging = false, AlwaysAddDefaultValue = true)]
+        internal List<string> extraMaterials = new List<string>();
+
         [DisableIf("@true")]
         [SerializeField]
         internal Texture2D bakedTexture = default;
@@ -30,11 +35,23 @@ public class MeshAnimationAsset : ScriptableObject
         [SerializeField]
         internal Material bakedMaterial = default;
 
+        [TableList(ShowPaging = false)]
+        [DisableIf("@true")]
+        [SerializeField]
+        internal List<ExtraMaterialData> extraMaterialData = new List<ExtraMaterialData>();
+
         [TableList(AlwaysExpanded = true, ShowPaging = false)]
         [DisableIf("@true")]
         [SerializeField]
         internal List<AnimationData> animationData = new List<AnimationData>();
 
+        [Serializable]
+        internal class ExtraMaterialData
+        {
+            public string name;
+            public Material material;
+        }
+
         [Serializable]
         internal class AnimationData
         {

Runtime/Scripts/MeshAnimationBaker.cs

@@ -3,6 +3,7 @@ namespace CodeWriter.MeshAnimation
 {
     using System;
     using System.Linq;
+    using System.Collections.Generic;
     using JetBrains.Annotations;
     using UnityEditor;
     using UnityEngine;
@@ -23,6 +24,11 @@ public static void Clear([NotNull] MeshAnimationAsset asset)
 
             DestroyObject(ref asset.bakedMaterial);
             DestroyObject(ref asset.bakedTexture);
+            
+            foreach (var data in asset.extraMaterialData) {
+                DestroyObject(ref data.material);
+            }
+            
             SaveAsset(asset);
         }
 
@@ -49,6 +55,7 @@ public static void Bake([NotNull] MeshAnimationAsset asset)
                 if (!aborted)
                 {
                     CreateMaterial(asset);
+                    CreateExtraMaterials(asset);
                     BakeAnimations(asset);
                 }
 
@@ -84,6 +91,30 @@ private static void CreateMaterial(MeshAnimationAsset asset)
             }
         }
 
+        private static void CreateExtraMaterials(MeshAnimationAsset asset)
+        {
+            foreach (var materialName in asset.extraMaterials)
+            {
+                var data = asset.extraMaterialData.Find(it => it.name == materialName);
+                if (data == null)
+                {
+                    data = new MeshAnimationAsset.ExtraMaterialData
+                    {
+                        name = materialName
+                    };
+                    asset.extraMaterialData.Add(data);
+                }
+
+                if (data.material == null)
+                {
+                    data.material = new Material(asset.shader) {name = $"{asset.name}_{materialName} Material"};
+                    AssetDatabase.AddObjectToAsset(data.material, asset);
+                }
+
+                data.material.shader = asset.shader;
+            }
+        }
+
         private static void CreateTexture(MeshAnimationAsset asset, out bool aborted)
         {
             aborted = false;
@@ -231,9 +262,17 @@ private static void BakeAnimations(MeshAnimationAsset asset)
             Object.DestroyImmediate(bakeObject);
             Object.DestroyImmediate(bakeMesh);
 
-            asset.bakedMaterial.SetTexture(AnimTextureProp, asset.bakedTexture);
-            asset.bakedMaterial.SetVector(AnimationMulProp, boundMax - boundMin);
-            asset.bakedMaterial.SetVector(AnimationAddProp, boundMin);
+            var materials = new HashSet<Material> { asset.bakedMaterial };
+            foreach (var data in asset.extraMaterialData)
+            {
+                materials.Add(data.material);
+            }
+
+            foreach (var material in materials) {
+                material.SetTexture(AnimTextureProp, asset.bakedTexture);
+                material.SetVector(AnimationMulProp, boundMax - boundMin);
+                material.SetVector(AnimationAddProp, boundMin);
+            }
         }
 
         private static void CaptureMeshToTexture(Texture2D texture, Mesh mesh, Vector3 min, Vector3 max, int line)