78 lines
2.0 KiB
C#
78 lines
2.0 KiB
C#
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System;
|
|
using UnityEngine;
|
|
using System.Text;
|
|
using UnityEngine.UI;
|
|
|
|
namespace Coffee.UIEffects
|
|
{
|
|
public class MaterialCache
|
|
{
|
|
static Dictionary<Hash128, MaterialEntry> materialMap = new Dictionary<Hash128, MaterialEntry>();
|
|
|
|
private class MaterialEntry
|
|
{
|
|
public Material material;
|
|
public int referenceCount;
|
|
|
|
public void Release()
|
|
{
|
|
if (material)
|
|
{
|
|
UnityEngine.Object.DestroyImmediate(material, false);
|
|
}
|
|
|
|
material = null;
|
|
}
|
|
}
|
|
|
|
#if UNITY_EDITOR
|
|
[UnityEditor.InitializeOnLoadMethod]
|
|
private static void ClearCache()
|
|
{
|
|
foreach (var entry in materialMap.Values)
|
|
{
|
|
entry.Release();
|
|
}
|
|
|
|
materialMap.Clear();
|
|
}
|
|
#endif
|
|
|
|
public static Material Register(Material baseMaterial, Hash128 hash,
|
|
System.Action<Material, Graphic> onModifyMaterial, Graphic graphic)
|
|
{
|
|
if (!hash.isValid) return null;
|
|
|
|
MaterialEntry entry;
|
|
if (!materialMap.TryGetValue(hash, out entry))
|
|
{
|
|
entry = new MaterialEntry()
|
|
{
|
|
material = new Material(baseMaterial)
|
|
{
|
|
hideFlags = HideFlags.HideAndDontSave,
|
|
},
|
|
};
|
|
|
|
onModifyMaterial(entry.material, graphic);
|
|
materialMap.Add(hash, entry);
|
|
}
|
|
|
|
entry.referenceCount++;
|
|
return entry.material;
|
|
}
|
|
|
|
public static void Unregister(Hash128 hash)
|
|
{
|
|
MaterialEntry entry;
|
|
if (!hash.isValid || !materialMap.TryGetValue(hash, out entry)) return;
|
|
if (--entry.referenceCount > 0) return;
|
|
|
|
entry.Release();
|
|
materialMap.Remove(hash);
|
|
}
|
|
}
|
|
}
|