From ef739e3c4d66007fb9c2ced195edb539eb92f3a4 Mon Sep 17 00:00:00 2001
From: chai <chaifix@163.com>
Date: Tue, 3 Nov 2020 19:05:55 +0800
Subject: *misc

---
 Assets/Scripts/Avatar/AbilitySystem.cs | 19 +++++++++++++++++--
 1 file changed, 17 insertions(+), 2 deletions(-)

(limited to 'Assets/Scripts/Avatar/AbilitySystem.cs')

diff --git a/Assets/Scripts/Avatar/AbilitySystem.cs b/Assets/Scripts/Avatar/AbilitySystem.cs
index 024e73cb..3e2ab838 100644
--- a/Assets/Scripts/Avatar/AbilitySystem.cs
+++ b/Assets/Scripts/Avatar/AbilitySystem.cs
@@ -22,6 +22,8 @@ public class AbilitySystem
 
 	private List<AbilityBase> m_Abilities = new List<AbilityBase>();
 
+    private UberAbility m_UberAbility;
+
 	public AbilitySystem()
 	{
 	}
@@ -38,6 +40,11 @@ public class AbilitySystem
 		m_Currrent.OnEnter();
 	}
 
+    public void SetUberAbility(UberAbility ability)
+    {
+        m_UberAbility = ability;
+    }
+
 	public void AddAbility(AbilityBase ability)
 	{
 		m_Abilities.Add(ability);
@@ -49,7 +56,11 @@ public class AbilitySystem
 		{
 			m_Currrent.OnUpdate();
 		}
-	}
+        if(m_UberAbility != null)
+        {
+            m_UberAbility.OnUpdate();
+        }
+    }
 
 	public void OnLateUpdate()
 	{
@@ -57,7 +68,11 @@ public class AbilitySystem
 		{
 			m_Currrent.OnLateUpdate();
 		}
-	}
+        if(m_UberAbility != null)
+        {
+            m_UberAbility.OnLateUpdate();
+        }
+    }
 
     public void OnHit(PhysicsCollisionInfo info)
     {
-- 
cgit v1.1-26-g67d0