123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308 |
- using System.IO;
- using UnityEditor.AssetImporters;
- using UnityEngine;
- using UnityEditor.Experimental.AssetImporters;
- namespace UnityEditor.Rendering.Universal
- {
- class FBXArnoldSurfaceMaterialDescriptionPreprocessor : AssetPostprocessor
- {
- static readonly uint k_Version = 2;
- static readonly int k_Order = 4;
- static readonly string k_ShaderPath = "Packages/com.unity.render-pipelines.universal/Runtime/Materials/ArnoldStandardSurface/ArnoldStandardSurface.shadergraph";
- static readonly string k_ShaderTransparentPath = "Packages/com.unity.render-pipelines.universal/Runtime/Materials/ArnoldStandardSurface/ArnoldStandardSurfaceTransparent.shadergraph";
- public override uint GetVersion()
- {
- return k_Version;
- }
- public override int GetPostprocessOrder()
- {
- return k_Order;
- }
- static bool IsMayaArnoldStandardSurfaceMaterial(MaterialDescription description)
- {
- float typeId;
- description.TryGetProperty("TypeId", out typeId);
- return typeId == 1138001;
- }
- static bool Is3DsMaxArnoldStandardSurfaceMaterial(MaterialDescription description)
- {
- float classIdA;
- float classIdB;
- description.TryGetProperty("ClassIDa", out classIdA);
- description.TryGetProperty("ClassIDb", out classIdB);
- return classIdA == 2121471519 && classIdB == 1660373836;
- }
- public void OnPreprocessMaterialDescription(MaterialDescription description, Material material,
- AnimationClip[] clips)
- {
- var lowerCasePath = Path.GetExtension(assetPath).ToLower();
- if (lowerCasePath == ".fbx")
- {
- if (IsMayaArnoldStandardSurfaceMaterial(description))
- CreateFromMayaArnoldStandardSurfaceMaterial(description, material, clips);
- else if (Is3DsMaxArnoldStandardSurfaceMaterial(description))
- CreateFrom3DsMaxArnoldStandardSurfaceMaterial(description, material, clips);
- }
- }
- void CreateFromMayaArnoldStandardSurfaceMaterial(MaterialDescription description, Material material,
- AnimationClip[] clips)
- {
- float floatProperty;
- Vector4 vectorProperty;
- TexturePropertyDescription textureProperty;
- Shader shader;
- float opacity = 1.0f;
- Vector4 opacityColor;
- TexturePropertyDescription opacityMap;
- description.TryGetProperty("opacity", out opacityColor);
- bool hasOpacityMap = description.TryGetProperty("opacity", out opacityMap);
- opacity = Mathf.Min(Mathf.Min(opacityColor.x, opacityColor.y), opacityColor.z);
- float transmission;
- description.TryGetProperty("transmission", out transmission);
- if (opacity == 1.0f && !hasOpacityMap)
- {
- opacity = 1.0f - transmission;
- }
- if (opacity < 1.0f || hasOpacityMap)
- {
- shader = AssetDatabase.LoadAssetAtPath<Shader>(k_ShaderTransparentPath);
- if (shader == null)
- return;
- material.shader = shader;
- if (hasOpacityMap)
- {
- material.SetTexture("_OPACITY_MAP", opacityMap.texture);
- material.SetFloat("_OPACITY", 1.0f);
- }
- else
- {
- material.SetFloat("_OPACITY", opacity);
- }
- }
- else
- {
- shader = AssetDatabase.LoadAssetAtPath<Shader>(k_ShaderPath);
- if (shader == null)
- return;
- material.shader = shader;
- }
-
- foreach (var clip in clips)
- {
- clip.ClearCurves();
- }
- description.TryGetProperty("base", out floatProperty);
- if (description.TryGetProperty("baseColor", out textureProperty))
- {
- SetMaterialTextureProperty("_BASE_COLOR_MAP", material, textureProperty);
- material.SetColor("_BASE_COLOR", Color.white * floatProperty);
- }
- else if (description.TryGetProperty("baseColor", out vectorProperty))
- {
- if (QualitySettings.activeColorSpace == ColorSpace.Gamma)
- {
- vectorProperty.x = Mathf.LinearToGammaSpace(vectorProperty.x);
- vectorProperty.y = Mathf.LinearToGammaSpace(vectorProperty.y);
- vectorProperty.z = Mathf.LinearToGammaSpace(vectorProperty.z);
- vectorProperty *= floatProperty;
- }
- material.SetColor("_BASE_COLOR", vectorProperty * floatProperty);
- }
- if (description.TryGetProperty("emission", out floatProperty) && floatProperty > 0.0f)
- {
- remapPropertyColorOrTexture(description, material, "emissionColor", "_EMISSION_COLOR", floatProperty);
- }
- remapPropertyFloatOrTexture(description, material, "metalness", "_METALNESS");
- description.TryGetProperty("specular", out floatProperty);
- remapPropertyColorOrTexture(description, material, "specularColor", "_SPECULAR_COLOR", floatProperty);
- remapPropertyFloatOrTexture(description, material, "specularRoughness", "_SPECULAR_ROUGHNESS");
- remapPropertyFloatOrTexture(description, material, "specularIOR", "_SPECULAR_IOR");
- remapPropertyTexture(description, material, "normalCamera", "_NORMAL_MAP");
- }
- void CreateFrom3DsMaxArnoldStandardSurfaceMaterial(MaterialDescription description, Material material,
- AnimationClip[] clips)
- {
- float floatProperty;
- Vector4 vectorProperty;
- TexturePropertyDescription textureProperty;
- var shader = AssetDatabase.LoadAssetAtPath<Shader>(k_ShaderPath);
- if (shader == null)
- return;
- material.shader = shader;
- foreach (var clip in clips)
- {
- clip.ClearCurves();
- }
- float opacity = 1.0f;
- Vector4 opacityColor;
- TexturePropertyDescription opacityMap;
- description.TryGetProperty("opacity", out opacityColor);
- bool hasOpacityMap = description.TryGetProperty("opacity", out opacityMap);
- opacity = Mathf.Min(Mathf.Min(opacityColor.x, opacityColor.y), opacityColor.z);
- if (opacity < 1.0f || hasOpacityMap)
- {
- if (hasOpacityMap)
- {
- material.SetTexture("_OPACITY_MAP", opacityMap.texture);
- material.SetFloat("_OPACITY", 1.0f);
- }
- else
- {
- material.SetFloat("_OPACITY", opacity);
- }
- }
- description.TryGetProperty("base", out floatProperty);
- if (description.TryGetProperty("base_color.shader", out textureProperty))
- {
- SetMaterialTextureProperty("_BASE_COLOR_MAP", material, textureProperty);
- material.SetColor("_BASE_COLOR", Color.white * floatProperty);
- }
- else if (description.TryGetProperty("base_color", out vectorProperty))
- {
- if (QualitySettings.activeColorSpace == ColorSpace.Gamma)
- {
- vectorProperty.x = Mathf.LinearToGammaSpace(vectorProperty.x);
- vectorProperty.y = Mathf.LinearToGammaSpace(vectorProperty.y);
- vectorProperty.z = Mathf.LinearToGammaSpace(vectorProperty.z);
- }
- material.SetColor("_BASE_COLOR", vectorProperty * floatProperty);
- }
- if (description.TryGetProperty("emission", out floatProperty) && floatProperty > 0.0f)
- {
- remapPropertyColorOrTexture3DsMax(description, material, "emission_color", "_EMISSION_COLOR",
- floatProperty);
- }
- remapPropertyFloatOrTexture3DsMax(description, material, "metalness", "_METALNESS");
- description.TryGetProperty("specular", out float specularFactor);
- remapPropertyColorOrTexture3DsMax(description, material, "specular_color", "_SPECULAR_COLOR",
- specularFactor);
- remapPropertyFloatOrTexture3DsMax(description, material, "specular_roughness", "_SPECULAR_ROUGHNESS");
- remapPropertyFloatOrTexture3DsMax(description, material, "specular_IOR", "_SPECULAR_IOR");
- remapPropertyTexture(description, material, "normal_camera", "_NORMAL_MAP");
- }
- static void SetMaterialTextureProperty(string propertyName, Material material,
- TexturePropertyDescription textureProperty)
- {
- material.SetTexture(propertyName, textureProperty.texture);
- material.SetTextureOffset(propertyName, textureProperty.offset);
- material.SetTextureScale(propertyName, textureProperty.scale);
- }
- static void remapPropertyFloat(MaterialDescription description, Material material, string inPropName,
- string outPropName)
- {
- if (description.TryGetProperty(inPropName, out float floatProperty))
- {
- material.SetFloat(outPropName, floatProperty);
- }
- }
- static void remapPropertyTexture(MaterialDescription description, Material material, string inPropName,
- string outPropName)
- {
- if (description.TryGetProperty(inPropName, out TexturePropertyDescription textureProperty))
- {
- material.SetTexture(outPropName, textureProperty.texture);
- }
- }
- static void remapPropertyColorOrTexture3DsMax(MaterialDescription description, Material material,
- string inPropName, string outPropName, float multiplier = 1.0f)
- {
- if (description.TryGetProperty(inPropName + ".shader", out TexturePropertyDescription textureProperty))
- {
- material.SetTexture(outPropName + "_MAP", textureProperty.texture);
- material.SetColor(outPropName, Color.white * multiplier);
- }
- else
- {
- description.TryGetProperty(inPropName, out Vector4 vectorProperty);
- material.SetColor(outPropName, vectorProperty * multiplier);
- }
- }
- static void remapPropertyFloatOrTexture3DsMax(MaterialDescription description, Material material,
- string inPropName, string outPropName)
- {
- if (description.TryGetProperty(inPropName, out TexturePropertyDescription textureProperty))
- {
- material.SetTexture(outPropName + "_MAP", textureProperty.texture);
- material.SetFloat(outPropName, 1.0f);
- }
- else
- {
- description.TryGetProperty(inPropName, out float floatProperty);
- material.SetFloat(outPropName, floatProperty);
- }
- }
- static void remapPropertyColorOrTexture(MaterialDescription description, Material material, string inPropName,
- string outPropName, float multiplier = 1.0f)
- {
- if (description.TryGetProperty(inPropName, out TexturePropertyDescription textureProperty))
- {
- material.SetTexture(outPropName + "_MAP", textureProperty.texture);
- material.SetColor(outPropName, Color.white * multiplier);
- }
- else
- {
- description.TryGetProperty(inPropName, out Vector4 vectorProperty);
- material.SetColor(outPropName, vectorProperty * multiplier);
- }
- }
- static void remapPropertyFloatOrTexture(MaterialDescription description, Material material, string inPropName,
- string outPropName)
- {
- if (description.TryGetProperty(inPropName, out TexturePropertyDescription textureProperty))
- {
- material.SetTexture(outPropName + "_MAP", textureProperty.texture);
- material.SetFloat(outPropName, 1.0f);
- }
- else
- {
- description.TryGetProperty(inPropName, out float floatProperty);
- material.SetFloat(outPropName, floatProperty);
- }
- }
- }
- }
|