Singularity/Library/PackageCache/com.unity.render-pipelines..../Editor/AssetPostProcessors/FBXMaterialDescriptionPrepr...
2024-05-06 11:45:45 -07:00

282 lines
12 KiB
C#

using System.IO;
using UnityEditor.AssetImporters;
using UnityEngine;
using UnityEngine.Rendering;
using UnityEngine.Rendering.Universal;
namespace UnityEditor.Rendering.Universal
{
class FBXMaterialDescriptionPreprocessor : AssetPostprocessor
{
static readonly uint k_Version = 2;
static readonly int k_Order = -980;
public override uint GetVersion()
{
return k_Version;
}
public override int GetPostprocessOrder()
{
return k_Order;
}
public void OnPreprocessMaterialDescription(MaterialDescription description, Material material, AnimationClip[] clips)
{
var pipelineAsset = GraphicsSettings.currentRenderPipeline;
if (!pipelineAsset || pipelineAsset.GetType() != typeof(UniversalRenderPipelineAsset))
return;
var lowerCaseExtension = Path.GetExtension(assetPath).ToLower();
if (lowerCaseExtension != ".fbx" && lowerCaseExtension != ".obj" && lowerCaseExtension != ".blend" && lowerCaseExtension != ".mb" && lowerCaseExtension != ".ma" && lowerCaseExtension != ".max")
return;
string path = AssetDatabase.GUIDToAssetPath(ShaderUtils.GetShaderGUID(ShaderPathID.Lit));
var shader = AssetDatabase.LoadAssetAtPath<Shader>(path);
if (shader == null)
return;
material.shader = shader;
Vector4 vectorProperty;
float floatProperty;
TexturePropertyDescription textureProperty;
bool isTransparent = false;
float opacity;
float transparencyFactor;
if (!description.TryGetProperty("Opacity", out opacity))
{
if (description.TryGetProperty("TransparencyFactor", out transparencyFactor))
{
opacity = transparencyFactor == 1.0f ? 1.0f : 1.0f - transparencyFactor;
}
if (opacity == 1.0f && description.TryGetProperty("TransparentColor", out vectorProperty))
{
opacity = vectorProperty.x == 1.0f ? 1.0f : 1.0f - vectorProperty.x;
}
}
if (opacity < 1.0f || (opacity == 1.0f && description.TryGetProperty("TransparentColor", out textureProperty)))
{
isTransparent = true;
}
else if (description.HasAnimationCurve("TransparencyFactor") || description.HasAnimationCurve("TransparentColor"))
{
isTransparent = true;
}
if (isTransparent)
{
material.SetFloat("_Mode", 3.0f);
material.SetOverrideTag("RenderType", "Transparent");
material.SetFloat("_SrcBlend", (float)UnityEngine.Rendering.BlendMode.One);
material.SetFloat("_DstBlend", (float)UnityEngine.Rendering.BlendMode.OneMinusSrcAlpha);
material.SetFloat("_ZWrite", 0.0f);
material.EnableKeyword("_ALPHAPREMULTIPLY_ON");
material.renderQueue = (int)UnityEngine.Rendering.RenderQueue.Transparent;
material.SetFloat("_Surface", 1.0f);
material.EnableKeyword("_SURFACE_TYPE_TRANSPARENT");
}
else
{
material.SetFloat("_Mode", 0.0f);
material.SetOverrideTag("RenderType", "");
material.SetFloat("_SrcBlend", (float)UnityEngine.Rendering.BlendMode.One);
material.SetFloat("_DstBlend", (float)UnityEngine.Rendering.BlendMode.Zero);
material.SetFloat("_ZWrite", 1.0f);
material.DisableKeyword("_ALPHATEST_ON");
material.DisableKeyword("_ALPHABLEND_ON");
material.DisableKeyword("_ALPHAPREMULTIPLY_ON");
material.renderQueue = -1;
material.SetFloat("_Surface", 0.0f);
material.DisableKeyword("_SURFACE_TYPE_TRANSPARENT");
}
if (description.TryGetProperty("DiffuseColor", out textureProperty) && textureProperty.texture != null)
{
Color diffuseColor = new Color(1.0f, 1.0f, 1.0f, 1.0f);
if (description.TryGetProperty("DiffuseFactor", out floatProperty))
diffuseColor *= floatProperty;
diffuseColor.a = opacity;
SetMaterialTextureProperty("_BaseMap", material, textureProperty);
material.SetColor("_BaseColor", diffuseColor);
}
else if (description.TryGetProperty("DiffuseColor", out vectorProperty))
{
Color diffuseColor = vectorProperty;
diffuseColor.a = opacity;
material.SetColor("_BaseColor", PlayerSettings.colorSpace == ColorSpace.Linear ? diffuseColor.gamma : diffuseColor);
}
if (description.TryGetProperty("Bump", out textureProperty))
{
SetMaterialTextureProperty("_BumpMap", material, textureProperty);
material.EnableKeyword("_NORMALMAP");
if (description.TryGetProperty("BumpFactor", out floatProperty))
material.SetFloat("_BumpScale", floatProperty);
}
else if (description.TryGetProperty("NormalMap", out textureProperty))
{
SetMaterialTextureProperty("_BumpMap", material, textureProperty);
material.EnableKeyword("_NORMALMAP");
if (description.TryGetProperty("BumpFactor", out floatProperty))
material.SetFloat("_BumpScale", floatProperty);
}
if (description.TryGetProperty("EmissiveColor", out textureProperty))
{
Color emissiveColor = new Color(1.0f, 1.0f, 1.0f, 1.0f);
material.SetColor("_EmissionColor", emissiveColor);
SetMaterialTextureProperty("_EmissionMap", material, textureProperty);
if (description.TryGetProperty("EmissiveFactor", out floatProperty) && floatProperty > 0.0f)
{
material.EnableKeyword("_EMISSION");
material.globalIlluminationFlags |= MaterialGlobalIlluminationFlags.RealtimeEmissive;
}
}
else if (
description.TryGetProperty("EmissiveColor", out vectorProperty) && vectorProperty.magnitude > vectorProperty.w
|| description.HasAnimationCurve("EmissiveColor.x"))
{
if (description.TryGetProperty("EmissiveFactor", out floatProperty))
vectorProperty *= floatProperty;
material.SetColor("_EmissionColor", vectorProperty);
if (floatProperty > 0.0f)
{
material.EnableKeyword("_EMISSION");
material.globalIlluminationFlags |= MaterialGlobalIlluminationFlags.RealtimeEmissive;
}
}
if (description.TryGetProperty("Shininess", out float shininess))
{
var glossiness = Mathf.Sqrt(shininess * 0.01f);
material.SetFloat("_Smoothness", glossiness);
}
else
material.SetFloat("_Smoothness", 0.0f);
if (PlayerSettings.colorSpace == ColorSpace.Linear)
RemapAndTransformColorCurves(description, clips, "DiffuseColor", "_BaseColor", ConvertFloatLinearToGamma);
else
RemapColorCurves(description, clips, "DiffuseColor", "_BaseColor");
RemapTransparencyCurves(description, clips);
RemapColorCurves(description, clips, "EmissiveColor", "_EmissionColor");
}
static void RemapTransparencyCurves(MaterialDescription description, AnimationClip[] clips)
{
// For some reason, Opacity is never animated, we have to use TransparencyFactor and TransparentColor
for (int i = 0; i < clips.Length; i++)
{
bool foundTransparencyCurve = false;
AnimationCurve curve;
if (description.TryGetAnimationCurve(clips[i].name, "TransparencyFactor", out curve))
{
ConvertKeys(curve, ConvertFloatOneMinus);
clips[i].SetCurve("", typeof(Material), "_BaseColor.a", curve);
foundTransparencyCurve = true;
}
else if (description.TryGetAnimationCurve(clips[i].name, "TransparentColor.x", out curve))
{
ConvertKeys(curve, ConvertFloatOneMinus);
clips[i].SetCurve("", typeof(Material), "_BaseColor.a", curve);
foundTransparencyCurve = true;
}
if (foundTransparencyCurve && !description.HasAnimationCurveInClip(clips[i].name, "DiffuseColor"))
{
Vector4 diffuseColor;
description.TryGetProperty("DiffuseColor", out diffuseColor);
clips[i].SetCurve("", typeof(Material), "_BaseColor.r", AnimationCurve.Constant(0.0f, 1.0f, diffuseColor.x));
clips[i].SetCurve("", typeof(Material), "_BaseColor.g", AnimationCurve.Constant(0.0f, 1.0f, diffuseColor.y));
clips[i].SetCurve("", typeof(Material), "_BaseColor.b", AnimationCurve.Constant(0.0f, 1.0f, diffuseColor.z));
}
}
}
static void RemapColorCurves(MaterialDescription description, AnimationClip[] clips, string originalPropertyName, string newPropertyName)
{
AnimationCurve curve;
for (int i = 0; i < clips.Length; i++)
{
if (description.TryGetAnimationCurve(clips[i].name, originalPropertyName + ".x", out curve))
{
clips[i].SetCurve("", typeof(Material), newPropertyName + ".r", curve);
}
if (description.TryGetAnimationCurve(clips[i].name, originalPropertyName + ".y", out curve))
{
clips[i].SetCurve("", typeof(Material), newPropertyName + ".g", curve);
}
if (description.TryGetAnimationCurve(clips[i].name, originalPropertyName + ".z", out curve))
{
clips[i].SetCurve("", typeof(Material), newPropertyName + ".b", curve);
}
}
}
static void RemapAndTransformColorCurves(MaterialDescription description, AnimationClip[] clips, string originalPropertyName, string newPropertyName, System.Func<float, float> converter)
{
AnimationCurve curve;
for (int i = 0; i < clips.Length; i++)
{
if (description.TryGetAnimationCurve(clips[i].name, originalPropertyName + ".x", out curve))
{
ConvertKeys(curve, converter);
clips[i].SetCurve("", typeof(Material), newPropertyName + ".r", curve);
}
if (description.TryGetAnimationCurve(clips[i].name, originalPropertyName + ".y", out curve))
{
ConvertKeys(curve, converter);
clips[i].SetCurve("", typeof(Material), newPropertyName + ".g", curve);
}
if (description.TryGetAnimationCurve(clips[i].name, originalPropertyName + ".z", out curve))
{
ConvertKeys(curve, converter);
clips[i].SetCurve("", typeof(Material), newPropertyName + ".b", curve);
}
}
}
static float ConvertFloatLinearToGamma(float value)
{
return Mathf.LinearToGammaSpace(value);
}
static float ConvertFloatOneMinus(float value)
{
return 1.0f - value;
}
static void ConvertKeys(AnimationCurve curve, System.Func<float, float> convertionDelegate)
{
Keyframe[] keyframes = curve.keys;
for (int i = 0; i < keyframes.Length; i++)
{
keyframes[i].value = convertionDelegate(keyframes[i].value);
}
curve.keys = keyframes;
}
static void SetMaterialTextureProperty(string propertyName, Material material, TexturePropertyDescription textureProperty)
{
material.SetTexture(propertyName, textureProperty.texture);
material.SetTextureOffset(propertyName, textureProperty.offset);
material.SetTextureScale(propertyName, textureProperty.scale);
}
}
}