From b781f9ecf36bcc6c614538ac427fa2eb8f1e29d2 Mon Sep 17 00:00:00 2001
From: chai <chaifix@163.com>
Date: Wed, 29 Sep 2021 20:26:12 +0800
Subject: *misc

---
 Assets/Scripts/Unit/Components/UnitLensEffect.cs | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

(limited to 'Assets/Scripts/Unit/Components/UnitLensEffect.cs')

diff --git a/Assets/Scripts/Unit/Components/UnitLensEffect.cs b/Assets/Scripts/Unit/Components/UnitLensEffect.cs
index 7758c943..a3c5d2c8 100644
--- a/Assets/Scripts/Unit/Components/UnitLensEffect.cs
+++ b/Assets/Scripts/Unit/Components/UnitLensEffect.cs
@@ -140,6 +140,8 @@ public class UnitLensEffect : UnitComponent
             }
         }
 
+        List<LensEffectBase> temp = ListPool<LensEffectBase>.Get();
+
         for (int i = 0; i < m_Effects.Count; ++i)
         {
             LensEffectBase eff = m_Effects[i];
@@ -147,8 +149,20 @@ public class UnitLensEffect : UnitComponent
             if (method == null)
                 continue;
             method.Invoke(eff, new object[] { });
+            if(eff.CanDestroy())
+            {
+                temp.Add(eff);
+            }
         }
 
+        for(int i = 0; i< temp.Count; ++i)
+        {
+            temp[i].OnDestroy();
+            m_Effects.Remove(temp[i]);
+        }
+
+        ListPool<LensEffectBase>.Release(temp);
+
         foreach (var cb in m_InUseCommandBuffers)
         {
             CameraEvent ce = cb.Key.ToCameraEvent();
-- 
cgit v1.1-26-g67d0