From 8c91d9f7d9436dc5d36bb874eaf166c2ca84066b Mon Sep 17 00:00:00 2001
From: Jens Pitkanen <jens@neon.moe>
Date: Fri, 23 Aug 2019 00:21:12 +0300
Subject: [PATCH] Fix stompy api for animations

---
 Assets/Scripts/AI/Behaviours/Stomp.cs | 3 ++-
 Assets/Scripts/Enemies/Stompy.cs      | 6 ++++--
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/Assets/Scripts/AI/Behaviours/Stomp.cs b/Assets/Scripts/AI/Behaviours/Stomp.cs
index 8416a4d..31aa1fe 100644
--- a/Assets/Scripts/AI/Behaviours/Stomp.cs
+++ b/Assets/Scripts/AI/Behaviours/Stomp.cs
@@ -21,7 +21,8 @@ namespace Saltosion.OneWeapon.AI.Behaviours {
         public float LoweredStompCooldown;
         public float LoweredStompCooldownRadius;
 
-        public bool Stomping { get { return State == StompState.PreparingStomp || State == StompState.RecoveringFromStomp; } }
+        public bool PreparingStomp { get { return State == StompState.PreparingStomp; } }
+        public bool Stomping { get { return State == StompState.RecoveringFromStomp; } }
         public bool CoolingDown { get { return State == StompState.CoolingDown; } }
 
         private float CurrentCooldown;
diff --git a/Assets/Scripts/Enemies/Stompy.cs b/Assets/Scripts/Enemies/Stompy.cs
index 9f90104..48ace1d 100644
--- a/Assets/Scripts/Enemies/Stompy.cs
+++ b/Assets/Scripts/Enemies/Stompy.cs
@@ -45,8 +45,10 @@ namespace Saltosion.OneWeapon.Enemies {
             }
 
             /* TODO: Implement animations for Stompy
-            if (StompBehaviour.Stomping) {
-                Anim.Play("Attack");
+            if (StompBehaviour.PreparingStomp) {
+                Anim.Play("Prepare");
+            } else if (StompBehaviour.Stomping) {
+                Anim.Play("Stomp");
             } else {
                 Anim.Play("Walk");
             }