From b5645c779a3e3c4ca758400cd9a718a7bc0c2bc6 Mon Sep 17 00:00:00 2001
From: chai <chaifix@163.com>
Date: Mon, 20 Sep 2021 22:44:08 +0800
Subject: *command buffer

---
 Assets/Scripts/Unit/LensEffect/LensEffectBase.cs | 101 +++++++++++++++++++++++
 1 file changed, 101 insertions(+)
 create mode 100644 Assets/Scripts/Unit/LensEffect/LensEffectBase.cs

(limited to 'Assets/Scripts/Unit/LensEffect/LensEffectBase.cs')

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
-- 
cgit v1.1-26-g67d0