Animate research and training

tonihele · Jul 5, 2024 · c45f5ba · c45f5ba
1 parent fc95a16
commit c45f5ba
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/toniarts/openkeeper/game/logic/CreatureViewSystem.java b/src/toniarts/openkeeper/game/logic/CreatureViewSystem.java
@@ -32,6 +32,7 @@
 import toniarts.openkeeper.game.component.Unconscious;
 import toniarts.openkeeper.game.controller.creature.CreatureState;
 import toniarts.openkeeper.game.task.TaskType;
+import static toniarts.openkeeper.game.task.TaskType.TRAIN;
 import toniarts.openkeeper.tools.convert.map.Creature;
 
 /**
@@ -161,6 +162,10 @@ private static Creature.AnimationType getAnimation(TaskType taskType) {
                 return Creature.AnimationType.SLEEPING;
             case DIG_TILE:
                 return Creature.AnimationType.MELEE_ATTACK;
+            case TRAIN:
+                return Creature.AnimationType.MELEE_ATTACK;
+            case RESEARCH:
+                return Creature.AnimationType.RESEARCHING;
             default:
                 return Creature.AnimationType.STAND_STILL;
         }