1 files changed, 101 insertions, 0 deletions
diff --git a/Assets/Scripts/Unit/LensEffect/LensEffectBase.cs b/Assets/Scripts/Unit/LensEffect/LensEffectBase.cs
new file mode 100644
index 00000000..08e20217
--- /dev/null
+++ b/Assets/Scripts/Unit/LensEffect/LensEffectBase.cs
@@ -0,0 +1,101 @@
+using System.Collections;
+using System.Collections.Generic;
+using UnityEngine;
+using UnityEngine.Rendering;
+
+public class MaterialEntry
+{
+    public string shader;
+    public Material material;
+}
+
+public abstract class LensEffectBase : MonoBehaviour
+{
+    public enum EStage
+    {
+        Before ,
+        Iterate , 
+        After , 
+        Finished ,
+    }
+
+    public abstract ERenderingEvent renderingEvents { get; }
+
+    protected static Dictionary<string/*shaderName*/, List<MaterialEntry>> m_MaterailPool = new Dictionary<string, List<MaterialEntry>>();
+
+    private List<MaterialEntry> m_InUseMaterials = new List<MaterialEntry>();
+
+    private static MaterialEntry _ClaimMaterial(string shader)
+    {
+        List<MaterialEntry> mats;
+        if (m_MaterailPool.TryGetValue(shader, out mats) && mats.Count > 0)
+        {
+            MaterialEntry me = mats[mats.Count - 1];
+            mats.RemoveAt(mats.Count - 1);
+            return me;
+        }
+        Material mat = new Material(Shader.Find(shader));
+        MaterialEntry entry = new MaterialEntry();
+        entry.material = mat;
+        entry.shader = shader;
+        return entry;
+    }
+
+    private static void _ReleaseMaterial(MaterialEntry entry)
+    {
+        List<MaterialEntry> mats;
+        if (!m_MaterailPool.TryGetValue(entry.shader, out mats))
+        {
+            mats = new List<MaterialEntry>();
+            m_MaterailPool.Add(entry.shader, mats);
+        }
+        mats.Add(entry);
+    }
+
+    protected MaterialEntry ClaimMaterial(string shader)
+    {
+        MaterialEntry entry = _ClaimMaterial(shader);
+        m_InUseMaterials.Add(entry);
+        return entry;
+    }
+
+    protected void ReleaseMaterial(MaterialEntry entry)
+    {
+        m_InUseMaterials.Remove(entry);
+        _ReleaseMaterial(entry);
+    }
+
+    protected void ReleaseAllInUsedMaterials()
+    {
+        for(int i = 0; i < m_InUseMaterials.Count; ++i)
+        {
+            _ReleaseMaterial(m_InUseMaterials[i]);
+        }
+        m_InUseMaterials.Clear();
+    }
+
+    public virtual void BeforeDepthTexture(EStage stage, CommandBuffer cb, BodyPartRenderer renderer = null) { }
+    public virtual void AfterDepthTexture(EStage stage, CommandBuffer cb, BodyPartRenderer renderer = null) { }
+
+    public virtual void BeforeDepthNormalsTexture(EStage stage, CommandBuffer cb, BodyPartRenderer renderer = null) { }
+    public virtual void AfterDepthNormalsTexture(EStage stage, CommandBuffer cb, BodyPartRenderer renderer = null) { }
+
+    public virtual void BeforeForwardOpaque(EStage stage, CommandBuffer cb, BodyPartRenderer renderer = null) { }
+    public virtual void AfterForwardOpaque(EStage stage, CommandBuffer cb, BodyPartRenderer renderer = null) { }
+
+    public virtual void BeforeImageEffectsOpaque(EStage stage, CommandBuffer cb, BodyPartRenderer renderer = null) { }
+    public virtual void AfterImageEffectsOpaque(EStage stage, CommandBuffer cb, BodyPartRenderer renderer = null) { }
+
+    public virtual void BeforeSkybox(EStage stage, CommandBuffer cb, BodyPartRenderer renderer = null) { }
+    public virtual void AfterSkybox(EStage stage, CommandBuffer cb, BodyPartRenderer renderer = null) { }
+
+    public virtual void BeforeForwardAlpha(EStage stage, CommandBuffer cb, BodyPartRenderer renderer = null) { }
+    public virtual void AfterForwardAlpha(EStage stage, CommandBuffer cb, BodyPartRenderer renderer = null) { }
+
+    public virtual void BeforeImageEffects(EStage stage, CommandBuffer cb, BodyPartRenderer renderer = null) { }
+    public virtual void AfterImageEffects(EStage stage, CommandBuffer cb, BodyPartRenderer renderer = null) { }
+
+    public virtual void BeforeEverything(EStage stage, CommandBuffer cb, BodyPartRenderer renderer = null) { }
+    public virtual void AfterEverything(EStage stage, CommandBuffer cb, BodyPartRenderer renderer = null) { }
+
+}
+\ No newline at end of file