From 494d3f727786ff70753692d8de019d9600a007e1 Mon Sep 17 00:00:00 2001
From: chai <chaifix@163.com>
Date: Tue, 3 Aug 2021 19:33:43 +0800
Subject: *misc

---
 Assets/Scripts/Unit/Action/WaitForActionReachEnd.cs | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

(limited to 'Assets/Scripts/Unit/Action/WaitForActionReachEnd.cs')

diff --git a/Assets/Scripts/Unit/Action/WaitForActionReachEnd.cs b/Assets/Scripts/Unit/Action/WaitForActionReachEnd.cs
index 3f5d20d4..ab755b64 100644
--- a/Assets/Scripts/Unit/Action/WaitForActionReachEnd.cs
+++ b/Assets/Scripts/Unit/Action/WaitForActionReachEnd.cs
@@ -6,17 +6,19 @@ using UnityEngine;
 public class WaitForActionReachEnd : IEnumerator
 {
 	UnitAnimation m_UnitAnimation;
-
-	public WaitForActionReachEnd(UnitAnimation unitAnim)
+    UnitAnimation.ELayer m_Layer;
+
+    public WaitForActionReachEnd(UnitAnimation unitAnim, UnitAnimation.ELayer layer = UnitAnimation.ELayer.Basic)
 	{
 		m_UnitAnimation = unitAnim;
-	}
+        m_Layer = layer;
+    }
 
 	public object Current => null;
 
 	public bool MoveNext()
 	{
-        var stateInfo = m_UnitAnimation.stateInfo;
+        var stateInfo = m_UnitAnimation.layers[(int)m_Layer].stateInfo;
         float normalTime = stateInfo.normalizedTime;
         LogHelper.Log(stateInfo.loop.ToString());
 		return normalTime < 1f;
-- 
cgit v1.1-26-g67d0