class="kw">package com.hypixel.hytale.server.npc.systems;

class="kw">import com.hypixel.hytale.component.ArchetypeChunk;
class="kw">import com.hypixel.hytale.component.CommandBuffer;
class="kw">import com.hypixel.hytale.component.ComponentType;
class="kw">import com.hypixel.hytale.component.Ref;
class="kw">import com.hypixel.hytale.component.Store;
class="kw">import com.hypixel.hytale.component.dependency.Dependency;
class="kw">import com.hypixel.hytale.component.dependency.Order;
class="kw">import com.hypixel.hytale.component.dependency.SystemDependency;
class="kw">import com.hypixel.hytale.component.query.Query;
class="kw">import com.hypixel.hytale.logger.HytaleLogger;
class="kw">import com.hypixel.hytale.server.core.entity.movement.MovementStatesComponent;
class="kw">import com.hypixel.hytale.server.core.modules.physics.component.Velocity;
class="kw">import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
class="kw">import com.hypixel.hytale.server.npc.NPCPlugin;
class="kw">import com.hypixel.hytale.server.npc.entities.NPCEntity;
class="kw">import java.util.Objects;
class="kw">import java.util.Set;
class="kw">import javax.annotation.Nonnull;

class="kw">public class MovementStatesSystem class="kw">extends SteppableTickingSystem {
   @Nonnull
   class="kw">private class="kw">final ComponentType<EntityStore, NPCEntity> npcComponentType;
   @Nonnull
   class="kw">private class="kw">final ComponentType<EntityStore, Velocity> velocityComponentType;
   @Nonnull
   class="kw">private class="kw">final ComponentType<EntityStore, MovementStatesComponent> movementStatesComponentType;
   @Nonnull
   class="kw">private class="kw">final Set<Dependency<EntityStore>> dependencies = Set.of(new SystemDependency<>(Order.AFTER, ComputeVelocitySystem.class));
   @Nonnull
   class="kw">private class="kw">final Query<EntityStore> query;

   class="kw">public MovementStatesSystem(
      @Nonnull ComponentType<EntityStore, NPCEntity> var1,
      @Nonnull ComponentType<EntityStore, Velocity> var2,
      @Nonnull ComponentType<EntityStore, MovementStatesComponent> var3
   ) {
      this.npcComponentType = var1;
      this.velocityComponentType = var2;
      this.movementStatesComponentType = var3;
      this.query = Query.and(var1, var2, var3);
   }

   @Nonnull
   @Override
   class="kw">public Set<Dependency<EntityStore>> getDependencies() {
      class="kw">return this.dependencies;
   }

   @Override
   class="kw">public boolean isParallel(int var1, int var2) {
      class="kw">return false;
   }

   @Override
   class="kw">public void steppedTick(
      float var1, int var2, @Nonnull ArchetypeChunk<EntityStore> var3, @Nonnull Store<EntityStore> var4, @Nonnull CommandBuffer<EntityStore> var5
   ) {
      NPCEntity var6 = var3.getComponent(var2, this.npcComponentType);
      assert var6 != null;
      Velocity var7 = var3.getComponent(var2, this.velocityComponentType);
      assert var7 != null;
      MovementStatesComponent var8 = var3.getComponent(var2, this.movementStatesComponentType);
      assert var8 != null;

      try {
         Ref var9 = var3.getReferenceTo(var2);
         if (Objects.equals(var6.getRoleName(), "Empty_Role")) {
            class="kw">return;
         }

         var6.getRole().updateMovementState(var9, var8.getMovementStates(), var7.getVelocity(), var5);
      } catch (Exception var10) {
         ((HytaleLogger.Api)((HytaleLogger.Api)NPCPlugin.get().getLogger().atSevere()).withCause(var10))
            .log(
               "Failed to update movement states for %s, Archetype: %s, Spawn config index: %d: ",
               var6.getRoleName(),
               var3.getArchetype(),
               var6.getSpawnConfiguration()
            );
      }
   }

   @Nonnull
   @Override
   class="kw">public Query<EntityStore> getQuery() {
      class="kw">return this.query;
   }
}