Add ApplyPresetsInMaterial()

JasonMa0012 · Jan 29, 2024 · 331b192 · 331b192
1 parent f15f869
commit 331b192
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 9 deletions.
diff --git a/Editor/MetaDataHelper.cs b/Editor/MetaDataHelper.cs
@@ -380,16 +380,11 @@ public void BuildPerFrameData(Shader shader, Material material, MaterialProperty
 				var drawer = ReflectionHelper.GetPropertyDrawer(shader, prop, out decoratorDrawers);
 
 				// Get Presets
-				if (drawer != null)
+				if (drawer != null && drawer is IBasePresetDrawer)
 				{
-					if (drawer is IBasePresetDrawer)
-					{
-						var activePreset = (drawer as IBasePresetDrawer).GetActivePreset(prop, perShaderData.propertyDatas[prop.name].propertyPresetAsset);
-						if (activePreset != null)
-						{
-							activePresets.Add(new PersetDynamicData(activePreset, prop));
-						}
-					}
+					var activePreset = (drawer as IBasePresetDrawer).GetActivePreset(prop, perShaderData.propertyDatas[prop.name].propertyPresetAsset);
+					if (activePreset != null)
+						activePresets.Add(new PersetDynamicData(activePreset, prop));
 				}
 			}
 

diff --git a/Editor/PresetHelper.cs b/Editor/PresetHelper.cs
@@ -58,5 +58,25 @@ public static ShaderPropertyPreset GetPresetFile(string presetFileName)
 
 			return _loadedPresets[presetFileName];
 		}
+
+		// Call this after a material has modified in code
+		public static void ApplyPresetsInMaterial(Material material)
+		{
+			var props = MaterialEditor.GetMaterialProperties(new[] { material });
+			foreach (var prop in props)
+			{
+				List<MaterialPropertyDrawer> decoratorDrawers;
+				var drawer = ReflectionHelper.GetPropertyDrawer(material.shader, prop, out decoratorDrawers);
+
+				// Get Presets
+				if (drawer != null && drawer is IBasePresetDrawer)
+				{
+					var activePreset = (drawer as IBasePresetDrawer).GetActivePreset(prop, PresetHelper.GetPresetFile((drawer as PresetDrawer).presetFileName));
+					if (activePreset != null)
+						activePreset.ApplyToDefaultMaterial(material);
+				}
+
+			}
+		}
 	}
 }