mirror of
https://github.com/vrm-c/UniVRM.git
synced 2026-05-07 13:47:06 -05:00
322 lines
12 KiB
C#
322 lines
12 KiB
C#
using System;
|
||
using System.Collections.Generic;
|
||
using UniGLTF.Extensions.VRMC_vrm;
|
||
using UnityEngine;
|
||
using VRM10.MToon10;
|
||
using Object = UnityEngine.Object;
|
||
|
||
namespace UniVRM10
|
||
{
|
||
///
|
||
/// Base + (A.Target - Base) * A.Weight + (B.Target - Base) * B.Weight ...
|
||
///
|
||
internal sealed class MaterialValueBindingMerger : IDisposable
|
||
{
|
||
private static readonly string COLOR_PROPERTY = MToon10Prop.BaseColorFactor.ToUnityShaderLabName();
|
||
private static readonly string EMISSION_COLOR_PROPERTY = MToon10Prop.EmissiveFactor.ToUnityShaderLabName();
|
||
private static readonly string RIM_COLOR_PROPERTY = MToon10Prop.ParametricRimColorFactor.ToUnityShaderLabName();
|
||
private static readonly string OUTLINE_COLOR_PROPERTY = MToon10Prop.OutlineColorFactor.ToUnityShaderLabName();
|
||
private static readonly string SHADE_COLOR_PROPERTY = MToon10Prop.ShadeColorFactor.ToUnityShaderLabName();
|
||
private static readonly string MATCAP_COLOR_PROPERTY = MToon10Prop.MatcapColorFactor.ToUnityShaderLabName();
|
||
|
||
private readonly HashSet<Material> _clonedMaterials = new();
|
||
|
||
public static string GetProperty(MaterialColorType bindType)
|
||
{
|
||
switch (bindType)
|
||
{
|
||
case MaterialColorType.color:
|
||
return COLOR_PROPERTY;
|
||
|
||
case MaterialColorType.emissionColor:
|
||
return EMISSION_COLOR_PROPERTY;
|
||
|
||
case MaterialColorType.shadeColor:
|
||
return SHADE_COLOR_PROPERTY;
|
||
|
||
case MaterialColorType.rimColor:
|
||
return RIM_COLOR_PROPERTY;
|
||
|
||
case MaterialColorType.outlineColor:
|
||
return OUTLINE_COLOR_PROPERTY;
|
||
|
||
case MaterialColorType.matcapColor:
|
||
return MATCAP_COLOR_PROPERTY;
|
||
}
|
||
|
||
throw new NotImplementedException();
|
||
}
|
||
|
||
#region MaterialMap
|
||
/// <summary>
|
||
/// MaterialValueBinding の対象になるマテリアルの情報を記録する
|
||
/// </summary>
|
||
Dictionary<string, PreviewMaterialItem> m_materialMap = new Dictionary<string, PreviewMaterialItem>();
|
||
|
||
void InitializeMaterialMap(Dictionary<ExpressionKey, VRM10Expression> clipMap, Transform root, bool isPrefabInstance)
|
||
{
|
||
var materialNameMap = new Dictionary<string, Material>();
|
||
foreach (var renderer in root.GetComponentsInChildren<Renderer>())
|
||
{
|
||
// VFXRendererなど、Materialが設定できないRendererが存在する
|
||
if (renderer is not SkinnedMeshRenderer && renderer is not MeshRenderer) continue;
|
||
|
||
if (isPrefabInstance)
|
||
{
|
||
// EditorImportされたPrefabのInstanceとして生成されている場合、Materialを複製する
|
||
var sharedMaterials = renderer.sharedMaterials;
|
||
var materials = renderer.materials;
|
||
for (var i = 0; i < materials.Length; i++)
|
||
{
|
||
var sharedMaterial = sharedMaterials[i];
|
||
var material = materials[i];
|
||
|
||
if (!sharedMaterial || !material) continue;
|
||
|
||
// 複製されたマテリアルはこのクラス内で破棄
|
||
if (sharedMaterial != material) _clonedMaterials.Add(material);
|
||
|
||
// 複製前の名前に揃え、それを記録しておく
|
||
// なお、Vrm10Runtimeのインスタンスが作られるより先にユーザーによってMaterialが複製されるパターンは想定しない
|
||
material.name = sharedMaterial.name;
|
||
materialNameMap.TryAdd(sharedMaterial.name, material);
|
||
}
|
||
}
|
||
else
|
||
{
|
||
// PrefabのInstanceでない（RuntimeImportされている）ならMaterialは複製しない
|
||
foreach (var material in renderer.sharedMaterials)
|
||
{
|
||
if (material)
|
||
{
|
||
materialNameMap.TryAdd(material.name, material);
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
foreach (var kv in clipMap)
|
||
{
|
||
foreach (var binding in kv.Value.MaterialColorBindings)
|
||
{
|
||
PreviewMaterialItem item;
|
||
if (!m_materialMap.TryGetValue(binding.MaterialName, out item))
|
||
{
|
||
if (!materialNameMap.TryGetValue(binding.MaterialName, out Material material))
|
||
{
|
||
// not found skip
|
||
continue;
|
||
}
|
||
item = new PreviewMaterialItem(material);
|
||
m_materialMap.Add(binding.MaterialName, item);
|
||
}
|
||
// color default value
|
||
var propName = GetProperty(binding.BindType);
|
||
if (!item.PropMap.ContainsKey(binding.BindType))
|
||
{
|
||
item.PropMap.Add(binding.BindType, new PropItem
|
||
{
|
||
Name = propName,
|
||
DefaultValues = item.Material.GetVector(propName),
|
||
});
|
||
}
|
||
}
|
||
|
||
foreach (var binding in kv.Value.MaterialUVBindings)
|
||
{
|
||
PreviewMaterialItem item;
|
||
if (!m_materialMap.TryGetValue(binding.MaterialName, out item))
|
||
{
|
||
if (!materialNameMap.TryGetValue(binding.MaterialName, out Material material))
|
||
{
|
||
// not found skip
|
||
continue;
|
||
}
|
||
item = new PreviewMaterialItem(material);
|
||
m_materialMap.Add(binding.MaterialName, item);
|
||
}
|
||
}
|
||
}
|
||
}
|
||
#endregion
|
||
|
||
#region Accumulate
|
||
struct DictionaryKeyMaterialValueBindingComparer : IEqualityComparer<MaterialColorBinding>
|
||
{
|
||
public bool Equals(MaterialColorBinding x, MaterialColorBinding y)
|
||
{
|
||
return x.TargetValue == y.TargetValue && x.MaterialName == y.MaterialName && x.BindType == y.BindType;
|
||
}
|
||
|
||
public int GetHashCode(MaterialColorBinding obj)
|
||
{
|
||
return obj.GetHashCode();
|
||
}
|
||
}
|
||
|
||
static DictionaryKeyMaterialValueBindingComparer comparer = new DictionaryKeyMaterialValueBindingComparer();
|
||
|
||
/// <summary>
|
||
/// MaterialValueの適用値を蓄積する
|
||
/// </summary>
|
||
/// <typeparam name="MaterialValueBinding"></typeparam>
|
||
/// <typeparam name="float"></typeparam>
|
||
/// <returns></returns>
|
||
Dictionary<MaterialColorBinding, float> m_materialColorMap = new Dictionary<MaterialColorBinding, float>(comparer);
|
||
|
||
/// <summary>
|
||
/// UV Scale/Offset
|
||
/// </summary>
|
||
Dictionary<string, Vector4> m_materialUVMap = new Dictionary<string, Vector4>();
|
||
|
||
public void AccumulateValue(VRM10Expression clip, float value)
|
||
{
|
||
// material color
|
||
foreach (var binding in clip.MaterialColorBindings)
|
||
{
|
||
float acc;
|
||
if (m_materialColorMap.TryGetValue(binding, out acc))
|
||
{
|
||
m_materialColorMap[binding] = acc + value;
|
||
}
|
||
else
|
||
{
|
||
m_materialColorMap[binding] = value;
|
||
}
|
||
}
|
||
|
||
// maetrial uv
|
||
foreach (var binding in clip.MaterialUVBindings)
|
||
{
|
||
if (m_materialMap.TryGetValue(binding.MaterialName, out PreviewMaterialItem item))
|
||
{
|
||
var delta = binding.ScalingOffset - item.DefaultUVScaleOffset;
|
||
|
||
Vector4 acc;
|
||
if (m_materialUVMap.TryGetValue(binding.MaterialName, out acc))
|
||
{
|
||
m_materialUVMap[binding.MaterialName] = acc + delta * value;
|
||
}
|
||
else
|
||
{
|
||
m_materialUVMap[binding.MaterialName] = item.DefaultUVScaleOffset + delta * value;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
struct MaterialTarget : IEquatable<MaterialTarget>
|
||
{
|
||
public string MaterialName;
|
||
public string ValueName;
|
||
|
||
public bool Equals(MaterialTarget other)
|
||
{
|
||
return MaterialName == other.MaterialName
|
||
&& ValueName == other.ValueName;
|
||
}
|
||
|
||
public override bool Equals(object obj)
|
||
{
|
||
if (obj is MaterialTarget)
|
||
{
|
||
return Equals((MaterialTarget)obj);
|
||
}
|
||
else
|
||
{
|
||
return false;
|
||
}
|
||
}
|
||
|
||
public override int GetHashCode()
|
||
{
|
||
if (MaterialName == null || ValueName == null)
|
||
{
|
||
return 0;
|
||
}
|
||
return MaterialName.GetHashCode() + ValueName.GetHashCode();
|
||
}
|
||
|
||
public static MaterialTarget Create(MaterialColorBinding binding)
|
||
{
|
||
return new MaterialTarget
|
||
{
|
||
MaterialName = binding.MaterialName,
|
||
ValueName = GetProperty(binding.BindType),
|
||
};
|
||
}
|
||
}
|
||
|
||
HashSet<MaterialTarget> m_used = new HashSet<MaterialTarget>();
|
||
public void Apply()
|
||
{
|
||
{
|
||
m_used.Clear();
|
||
foreach (var kv in m_materialColorMap)
|
||
{
|
||
var key = MaterialTarget.Create(kv.Key);
|
||
PreviewMaterialItem item;
|
||
if (m_materialMap.TryGetValue(key.MaterialName, out item))
|
||
{
|
||
// 初期値(コンストラクタで記録)
|
||
var initial = item.PropMap[kv.Key.BindType].DefaultValues;
|
||
if (!m_used.Contains(key))
|
||
{
|
||
//
|
||
// m_used に入っていない場合は、このフレームで初回の呼び出しになる。
|
||
// (Apply はフレームに一回呼ばれる想定)
|
||
// 初回は、値を初期値に戻す。
|
||
//
|
||
item.Material.SetColor(key.ValueName, initial);
|
||
m_used.Add(key);
|
||
}
|
||
|
||
// 現在値
|
||
var current = item.Material.GetVector(key.ValueName);
|
||
// 変化量
|
||
var value = (kv.Key.TargetValue - initial) * kv.Value;
|
||
// 適用
|
||
item.Material.SetColor(key.ValueName, current + value);
|
||
}
|
||
else
|
||
{
|
||
// エラー？
|
||
}
|
||
}
|
||
m_materialColorMap.Clear();
|
||
}
|
||
|
||
{
|
||
foreach (var kv in m_materialUVMap)
|
||
{
|
||
PreviewMaterialItem item;
|
||
if (m_materialMap.TryGetValue(kv.Key, out item))
|
||
{
|
||
//
|
||
// Standard and MToon use _MainTex_ST as uv0 scale/offset
|
||
//
|
||
item.Material.mainTextureScale = new Vector2(kv.Value.x, kv.Value.y);
|
||
item.Material.mainTextureOffset = new Vector2(kv.Value.z, kv.Value.w);
|
||
}
|
||
}
|
||
m_materialUVMap.Clear();
|
||
}
|
||
}
|
||
#endregion
|
||
|
||
public MaterialValueBindingMerger(Dictionary<ExpressionKey, VRM10Expression> clipMap, Transform root, bool isPrefabInstance)
|
||
{
|
||
InitializeMaterialMap(clipMap, root, isPrefabInstance);
|
||
}
|
||
|
||
public void Dispose()
|
||
{
|
||
foreach (var clonedMaterial in _clonedMaterials)
|
||
{
|
||
Object.Destroy(clonedMaterial);
|
||
}
|
||
_clonedMaterials.Clear();
|
||
}
|
||
}
|
||
}
|