using System; using UnityEngine; using UnityEngine.Rendering; namespace UnityEditor { internal class LitGUI : ShaderGUI { public enum SurfaceType { Opaque, Transparent } public enum BlendMode { Lerp, Add, SoftAdd, Multiply, Premultiply } public enum SmoothnessMapChannel { MaskAlpha, AlbedoAlpha, } public enum EmissiveColorMode { UseEmissiveColor, UseEmissiveMask, } public enum DoubleSidedMode { None, DoubleSided, DoubleSidedLightingFlip, DoubleSidedLightingMirror, } public enum NormalMapSpace { TangentSpace, ObjectSpace, } public enum HeightmapMode { Parallax, Displacement, } private static class Styles { public static string OptionText = "Options"; public static string SurfaceTypeText = "Surface Type"; public static string BlendModeText = "Blend Mode"; public static GUIContent alphaCutoffEnableText = new GUIContent("Alpha Cutoff Enable", "Threshold for alpha cutoff"); public static GUIContent alphaCutoffText = new GUIContent("Alpha Cutoff", "Threshold for alpha cutoff"); public static GUIContent doubleSidedModeText = new GUIContent("Double Sided", "This will render the two face of the objects (disable backface culling)"); public static readonly string[] surfaceTypeNames = Enum.GetNames(typeof(SurfaceType)); public static readonly string[] blendModeNames = Enum.GetNames(typeof(BlendMode)); public static string InputsOptionsText = "Inputs options"; public static GUIContent smoothnessMapChannelText = new GUIContent("Smoothness Source", "Smoothness texture and channel"); public static GUIContent emissiveColorModeText = new GUIContent("Emissive Color Usage", "Use emissive color or emissive mask"); public static string InputsText = "Inputs"; public static string InputsMapText = ""; public static GUIContent baseColorText = new GUIContent("Base Color + Opacity", "Albedo (RGB) and Opacity (A)"); public static GUIContent baseColorSmoothnessText = new GUIContent("Base Color + Smoothness", "Albedo (RGB) and Smoothness (A)"); public static GUIContent metalicText = new GUIContent("Metalic", "Metalic scale factor"); public static GUIContent smoothnessText = new GUIContent("Smoothness", "Smoothness scale factor"); public static GUIContent maskMapESText = new GUIContent("Mask Map - M(R), AO(G), E(B), S(A)", "Mask map"); public static GUIContent maskMapEText = new GUIContent("Mask Map - M(R), AO(G), E(B)", "Mask map"); public static GUIContent maskMapText = new GUIContent("Mask Map - M(R), AO(G)", "Mask map"); public static GUIContent maskMapSText = new GUIContent("Mask Map - M(R), AO(G), S(A)", "Mask map"); public static GUIContent specularOcclusionMapText = new GUIContent("Specular Occlusion Map (RGBA)", "Specular Occlusion Map"); public static GUIContent normalMapText = new GUIContent("Normal Map", "Normal Map (BC5) - DXT5 for test"); public static GUIContent normalMapSpaceText = new GUIContent("Normal Map space", ""); public static GUIContent heightMapText = new GUIContent("Height Map", "Height Map"); public static GUIContent heightMapModeText = new GUIContent("Height Map Mode", ""); // public static GUIContent diffuseLightingMapText = new GUIContent("DiffuseLightingMap", "Lightmap/Lightprobe data (fill by system is not done"); public static GUIContent emissiveText = new GUIContent("Emissive Color", "Emissive"); public static GUIContent emissiveIntensityText = new GUIContent("Emissive Intensity", "Emissive"); } MaterialProperty surfaceType = null; MaterialProperty blendMode = null; MaterialProperty alphaCutoff = null; MaterialProperty alphaCutoffEnable = null; MaterialProperty doubleSidedMode = null; MaterialProperty smoothnessMapChannel = null; MaterialProperty emissiveColorMode = null; MaterialProperty baseColor = null; MaterialProperty baseColorMap = null; MaterialProperty metalic = null; MaterialProperty smoothness = null; MaterialProperty maskMap = null; MaterialProperty specularOcclusionMap = null; MaterialProperty normalMap = null; MaterialProperty normalMapSpace = null; MaterialProperty heightMap = null; MaterialProperty heightScale = null; MaterialProperty heightBias = null; MaterialProperty heightMapMode = null; // MaterialProperty diffuseLightingMap = null; MaterialProperty emissiveColor = null; MaterialProperty emissiveColorMap = null; MaterialProperty emissiveIntensity = null; // MaterialProperty subSurfaceRadius = null; // MaterialProperty subSurfaceRadiusMap = null; protected MaterialEditor m_MaterialEditor; protected const string kSurfaceType = "_SurfaceType"; protected const string kBlendMode = "_BlendMode"; protected const string kAlphaCutoff = "_AlphaCutoff"; protected const string kAlphaCutoffEnabled = "_AlphaCutoffEnable"; protected const string kDoubleSidedMode = "_DoubleSidedMode"; protected const string kSmoothnessTextureChannelProp = "_SmoothnessTextureChannel"; protected const string kEmissiveColorMode = "_EmissiveColorMode"; protected const string kNormalMapSpace = "_NormalMapSpace"; protected const string kHeightMapMode = "_HeightMapMode"; protected const string kNormalMap = "_NormalMap"; protected const string kMaskMap = "_MaskMap"; protected const string kspecularOcclusionMap = "_SpecularOcclusionMap"; protected const string kEmissiveColorMap = "_EmissiveColorMap"; protected const string kHeightMap = "_HeightMap"; public void FindOptionProperties(MaterialProperty[] props) { surfaceType = FindProperty(kSurfaceType, props); blendMode = FindProperty(kBlendMode, props); alphaCutoff = FindProperty(kAlphaCutoff, props); alphaCutoffEnable = FindProperty(kAlphaCutoffEnabled, props); doubleSidedMode = FindProperty(kDoubleSidedMode, props); smoothnessMapChannel = FindProperty(kSmoothnessTextureChannelProp, props); emissiveColorMode = FindProperty(kEmissiveColorMode, props); normalMapSpace = FindProperty(kNormalMapSpace, props); heightMapMode = FindProperty(kHeightMapMode, props); } public void FindInputProperties(MaterialProperty[] props) { baseColor = FindProperty("_BaseColor", props); baseColorMap = FindProperty("_BaseColorMap", props); metalic = FindProperty("_Metalic", props); smoothness = FindProperty("_Smoothness", props); maskMap = FindProperty(kMaskMap, props); specularOcclusionMap = FindProperty(kspecularOcclusionMap, props); normalMap = FindProperty(kNormalMap, props); heightMap = FindProperty(kHeightMap, props); heightScale = FindProperty("_HeightScale", props); heightBias = FindProperty("_HeightBias", props); // diffuseLightingMap = FindProperty("_DiffuseLightingMap", props); emissiveColor = FindProperty("_EmissiveColor", props); emissiveColorMap = FindProperty(kEmissiveColorMap, props); emissiveIntensity = FindProperty("_EmissiveIntensity", props); } public override void OnGUI(MaterialEditor materialEditor, MaterialProperty[] props) { FindOptionProperties(props); // MaterialProperties can be animated so we do not cache them but fetch them every event to ensure animated values are updated correctly FindInputProperties(props); m_MaterialEditor = materialEditor; Material material = materialEditor.target as Material; ShaderPropertiesGUI(material); } protected void ShaderOptionsGUI() { EditorGUI.indentLevel++; GUILayout.Label(Styles.OptionText, EditorStyles.boldLabel); SurfaceTypePopup(); if ((SurfaceType)surfaceType.floatValue == SurfaceType.Transparent) { BlendModePopup(); } m_MaterialEditor.ShaderProperty(alphaCutoffEnable, Styles.alphaCutoffEnableText.text); if (alphaCutoffEnable.floatValue == 1.0) { m_MaterialEditor.ShaderProperty(alphaCutoff, Styles.alphaCutoffText.text); } m_MaterialEditor.ShaderProperty(doubleSidedMode, Styles.doubleSidedModeText.text); EditorGUI.indentLevel--; } protected void ShaderInputOptionsGUI() { EditorGUI.indentLevel++; GUILayout.Label(Styles.InputsOptionsText, EditorStyles.boldLabel); m_MaterialEditor.ShaderProperty(smoothnessMapChannel, Styles.smoothnessMapChannelText.text); m_MaterialEditor.ShaderProperty(emissiveColorMode, Styles.emissiveColorModeText.text); m_MaterialEditor.ShaderProperty(normalMapSpace, Styles.normalMapSpaceText.text); m_MaterialEditor.ShaderProperty(heightMapMode, Styles.heightMapModeText.text); EditorGUI.indentLevel--; } protected void ShaderInputGUI() { EditorGUI.indentLevel++; bool smoothnessInAlbedoAlpha = (SmoothnessMapChannel)smoothnessMapChannel.floatValue == SmoothnessMapChannel.AlbedoAlpha; bool useEmissiveMask = (EmissiveColorMode)emissiveColorMode.floatValue == EmissiveColorMode.UseEmissiveMask; GUILayout.Label(Styles.InputsText, EditorStyles.boldLabel); m_MaterialEditor.TexturePropertySingleLine(smoothnessInAlbedoAlpha ? Styles.baseColorSmoothnessText : Styles.baseColorText, baseColorMap, baseColor); m_MaterialEditor.ShaderProperty(metalic, Styles.metalicText); m_MaterialEditor.ShaderProperty(smoothness, Styles.smoothnessText); if (smoothnessInAlbedoAlpha && useEmissiveMask) m_MaterialEditor.TexturePropertySingleLine(Styles.maskMapEText, maskMap); else if (smoothnessInAlbedoAlpha && !useEmissiveMask) m_MaterialEditor.TexturePropertySingleLine(Styles.maskMapText, maskMap); else if (!smoothnessInAlbedoAlpha && useEmissiveMask) m_MaterialEditor.TexturePropertySingleLine(Styles.maskMapESText, maskMap); else if (!smoothnessInAlbedoAlpha && !useEmissiveMask) m_MaterialEditor.TexturePropertySingleLine(Styles.maskMapSText, maskMap); m_MaterialEditor.TexturePropertySingleLine(Styles.specularOcclusionMapText, specularOcclusionMap); m_MaterialEditor.TexturePropertySingleLine(Styles.normalMapText, normalMap); m_MaterialEditor.TexturePropertySingleLine(Styles.heightMapText, heightMap, heightScale, heightBias); if (!useEmissiveMask) { m_MaterialEditor.TexturePropertySingleLine(Styles.emissiveText, emissiveColorMap, emissiveColor); } m_MaterialEditor.ShaderProperty(emissiveIntensity, Styles.emissiveIntensityText); EditorGUI.indentLevel--; } public void ShaderPropertiesGUI(Material material) { // Use default labelWidth EditorGUIUtility.labelWidth = 0f; // Detect any changes to the material EditorGUI.BeginChangeCheck(); { ShaderOptionsGUI(); EditorGUILayout.Space(); ShaderInputOptionsGUI(); EditorGUILayout.Space(); ShaderInputGUI(); } if (EditorGUI.EndChangeCheck()) { foreach (var obj in m_MaterialEditor.targets) SetupMaterial((Material)obj); } } // TODO: try to setup minimun value to fall back to standard shaders and reverse public override void AssignNewShaderToMaterial(Material material, Shader oldShader, Shader newShader) { base.AssignNewShaderToMaterial(material, oldShader, newShader); } void SurfaceTypePopup() { EditorGUI.showMixedValue = surfaceType.hasMixedValue; var mode = (SurfaceType)surfaceType.floatValue; EditorGUI.BeginChangeCheck(); mode = (SurfaceType)EditorGUILayout.Popup(Styles.SurfaceTypeText, (int)mode, Styles.surfaceTypeNames); if (EditorGUI.EndChangeCheck()) { m_MaterialEditor.RegisterPropertyChangeUndo("Surface Type"); surfaceType.floatValue = (float)mode; } EditorGUI.showMixedValue = false; } void BlendModePopup() { EditorGUI.showMixedValue = blendMode.hasMixedValue; var mode = (BlendMode)blendMode.floatValue; EditorGUI.BeginChangeCheck(); mode = (BlendMode)EditorGUILayout.Popup(Styles.BlendModeText, (int)mode, Styles.blendModeNames); if (EditorGUI.EndChangeCheck()) { m_MaterialEditor.RegisterPropertyChangeUndo("Blend Mode"); blendMode.floatValue = (float)mode; } EditorGUI.showMixedValue = false; } protected virtual void SetupKeywordsForInputMaps(Material material) { SetKeyword(material, "_NORMALMAP", material.GetTexture(kNormalMap)); SetKeyword(material, "_MASKMAP", material.GetTexture(kMaskMap)); SetKeyword(material, "_SPECULAROCCLUSIONMAP", material.GetTexture(kspecularOcclusionMap)); SetKeyword(material, "_EMISSIVE_COLOR_MAP", material.GetTexture(kEmissiveColorMap)); SetKeyword(material, "_HEIGHTMAP", material.GetTexture(kHeightMap)); } protected void SetupMaterial(Material material) { // Note: keywords must be based on Material value not on MaterialProperty due to multi-edit & material animation // (MaterialProperty value might come from renderer material property block) bool alphaTestEnable = material.GetFloat(kAlphaCutoffEnabled) == 1.0; SurfaceType surfaceType = (SurfaceType)material.GetFloat(kSurfaceType); BlendMode blendMode = (BlendMode)material.GetFloat(kBlendMode); DoubleSidedMode doubleSidedMode = (DoubleSidedMode)material.GetFloat(kDoubleSidedMode); if (surfaceType == SurfaceType.Opaque) { material.SetOverrideTag("RenderType", alphaTestEnable ? "TransparentCutout" : ""); material.SetInt("_SrcBlend", (int)UnityEngine.Rendering.BlendMode.One); material.SetInt("_DstBlend", (int)UnityEngine.Rendering.BlendMode.Zero); material.SetInt("_ZWrite", 1); material.renderQueue = alphaTestEnable ? (int)UnityEngine.Rendering.RenderQueue.AlphaTest : -1; } else { material.SetOverrideTag("RenderType", "Transparent"); material.SetInt("_ZWrite", 0); material.renderQueue = (int)UnityEngine.Rendering.RenderQueue.Transparent; switch (blendMode) { case BlendMode.Lerp: material.SetInt("_SrcBlend", (int)UnityEngine.Rendering.BlendMode.SrcAlpha); material.SetInt("_DstBlend", (int)UnityEngine.Rendering.BlendMode.OneMinusSrcAlpha); break; case BlendMode.Add: material.SetInt("_SrcBlend", (int)UnityEngine.Rendering.BlendMode.One); material.SetInt("_DstBlend", (int)UnityEngine.Rendering.BlendMode.One); break; case BlendMode.SoftAdd: material.SetInt("_SrcBlend", (int)UnityEngine.Rendering.BlendMode.OneMinusDstColor); material.SetInt("_DstBlend", (int)UnityEngine.Rendering.BlendMode.One); break; case BlendMode.Multiply: material.SetInt("_SrcBlend", (int)UnityEngine.Rendering.BlendMode.DstColor); material.SetInt("_DstBlend", (int)UnityEngine.Rendering.BlendMode.Zero); break; case BlendMode.Premultiply: material.SetInt("_SrcBlend", (int)UnityEngine.Rendering.BlendMode.One); material.SetInt("_DstBlend", (int)UnityEngine.Rendering.BlendMode.OneMinusSrcAlpha); break; } } if (doubleSidedMode == DoubleSidedMode.None) { material.SetInt("_CullMode", (int)UnityEngine.Rendering.CullMode.Back); } else { material.SetInt("_CullMode", (int)UnityEngine.Rendering.CullMode.Off); } if (doubleSidedMode == DoubleSidedMode.DoubleSidedLightingFlip) { material.EnableKeyword("_DOUBLESIDED_LIGHTING_FLIP"); material.DisableKeyword("_DOUBLESIDED_LIGHTING_MIRROR"); } else if (doubleSidedMode == DoubleSidedMode.DoubleSidedLightingMirror) { material.DisableKeyword("_DOUBLESIDED_LIGHTING_FLIP"); material.EnableKeyword("_DOUBLESIDED_LIGHTING_MIRROR"); } else { material.DisableKeyword("_DOUBLESIDED_LIGHTING_FLIP"); material.DisableKeyword("_DOUBLESIDED_LIGHTING_MIRROR"); } SetKeyword(material, "_ALPHATEST_ON", alphaTestEnable); SetKeyword(material, "_NORMALMAP_TANGENT_SPACE", (NormalMapSpace)material.GetFloat(kNormalMapSpace) == NormalMapSpace.TangentSpace); SetKeyword(material, "_SMOOTHNESS_TEXTURE_ALBEDO_CHANNEL_A", ((SmoothnessMapChannel)material.GetFloat(kSmoothnessTextureChannelProp)) == SmoothnessMapChannel.AlbedoAlpha); SetKeyword(material, "_EMISSIVE_COLOR", ((EmissiveColorMode)material.GetFloat(kEmissiveColorMode)) == EmissiveColorMode.UseEmissiveColor); SetKeyword(material, "_HEIGHTMAP_AS_DISPLACEMENT", (HeightmapMode)material.GetFloat(kHeightMapMode) == HeightmapMode.Displacement); SetupKeywordsForInputMaps(material); /* // Setup lightmap emissive flags MaterialGlobalIlluminationFlags flags = material.globalIlluminationFlags; if ((flags & (MaterialGlobalIlluminationFlags.BakedEmissive | MaterialGlobalIlluminationFlags.RealtimeEmissive)) != 0) { flags &= ~MaterialGlobalIlluminationFlags.EmissiveIsBlack; if (!shouldEmissionBeEnabled) flags |= MaterialGlobalIlluminationFlags.EmissiveIsBlack; material.globalIlluminationFlags = flags; } */ } static bool ShouldEmissionBeEnabled(Material mat, Color color) { //var realtimeEmission = (mat.globalIlluminationFlags & MaterialGlobalIlluminationFlags.RealtimeEmissive) > 0; //return color.maxColorComponent > 0.1f / 255.0f || realtimeEmission; return false; } bool HasValidEmissiveKeyword(Material material) { /* // Material animation might be out of sync with the material keyword. // So if the emission support is disabled on the material, but the property blocks have a value that requires it, then we need to show a warning. // (note: (Renderer MaterialPropertyBlock applies its values to emissionColorForRendering)) bool hasEmissionKeyword = material.IsKeywordEnabled ("_EMISSION"); if (!hasEmissionKeyword && ShouldEmissionBeEnabled (material, emissionColorForRendering.colorValue)) return false; else return true; */ return true; } protected void SetKeyword(Material m, string keyword, bool state) { if (state) m.EnableKeyword(keyword); else m.DisableKeyword(keyword); } } } // namespace UnityEditor