MovementStatesSystems

package com.hypixel.hytale.server.core.entity.movement;

import com.hypixel.hytale.component.AddReason;
import com.hypixel.hytale.component.ArchetypeChunk;
import com.hypixel.hytale.component.CommandBuffer;
import com.hypixel.hytale.component.ComponentType;
import com.hypixel.hytale.component.Holder;
import com.hypixel.hytale.component.Ref;
import com.hypixel.hytale.component.RemoveReason;
import com.hypixel.hytale.component.Store;
import com.hypixel.hytale.component.SystemGroup;
import com.hypixel.hytale.component.query.Query;
import com.hypixel.hytale.component.system.HolderSystem;
import com.hypixel.hytale.component.system.RefSystem;
import com.hypixel.hytale.component.system.tick.EntityTickingSystem;
import com.hypixel.hytale.protocol.ComponentUpdate;
import com.hypixel.hytale.protocol.ComponentUpdateType;
import com.hypixel.hytale.protocol.MovementStates;
import com.hypixel.hytale.protocol.SavedMovementStates;
import com.hypixel.hytale.server.core.entity.entities.Player;
import com.hypixel.hytale.server.core.entity.entities.player.data.PlayerWorldData;
import com.hypixel.hytale.server.core.modules.entity.AllLegacyLivingEntityTypesQuery;
import com.hypixel.hytale.server.core.modules.entity.tracker.EntityTrackerSystems;
import com.hypixel.hytale.server.core.universe.world.World;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import java.util.Map;
import java.util.Map.Entry;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;

public class MovementStatesSystems {
   public MovementStatesSystems() {
   }

   public static class AddSystem extends HolderSystem<EntityStore> {
      @Nonnull
      private final ComponentType<EntityStore, MovementStatesComponent> movementStatesComponentComponentType;

      public AddSystem(@Nonnull ComponentType<EntityStore, MovementStatesComponent> movementStatesComponentComponentType) {
         this.movementStatesComponentComponentType = movementStatesComponentComponentType;
      }

      @Override
      public void onEntityAdd(@Nonnull Holder<EntityStore> holder, @Nonnull AddReason reason, @Nonnull Store<EntityStore> store) {
         holder.ensureComponent(this.movementStatesComponentComponentType);
      }

      @Override
      public void onEntityRemoved(@Nonnull Holder<EntityStore> holder, @Nonnull RemoveReason reason, @Nonnull Store<EntityStore> store) {
      }

      @Nonnull
      @Override
      public Query<EntityStore> getQuery() {
         return AllLegacyLivingEntityTypesQuery.INSTANCE;
      }
   }

   public static class PlayerInitSystem extends RefSystem<EntityStore> {
      @Nonnull
      private final Query<EntityStore> query;
      @Nonnull
      private final ComponentType<EntityStore, Player> playerComponentType;
      @Nonnull
      private final ComponentType<EntityStore, MovementStatesComponent> movementStatesComponentType;

      public PlayerInitSystem(
         @Nonnull ComponentType<EntityStore, Player> playerComponentType,
         @Nonnull ComponentType<EntityStore, MovementStatesComponent> movementStatesComponentType
      ) {
         this.playerComponentType = playerComponentType;
         this.movementStatesComponentType = movementStatesComponentType;
         this.query = Query.and(playerComponentType, movementStatesComponentType);
      }

      @Override
      public void onEntityAdded(
         @Nonnull Ref<EntityStore> ref, @Nonnull AddReason reason, @Nonnull Store<EntityStore> store, @Nonnull CommandBuffer<EntityStore> commandBuffer
      ) {
         World world = commandBuffer.getExternalData().getWorld();
         Player playerComponent = store.getComponent(ref, this.playerComponentType);

         assert playerComponent != null;

         MovementStatesComponent movementStatesComponent = store.getComponent(ref, this.movementStatesComponentType);

         assert movementStatesComponent != null;

         PlayerWorldData perWorldData = playerComponent.getPlayerConfigData().getPerWorldData(world.getName());
         SavedMovementStates movementStates = perWorldData.getLastMovementStates();
         playerComponent.applyMovementStates(
            ref, movementStates != null ? movementStates : new SavedMovementStates(), movementStatesComponent.getMovementStates(), store
         );
      }

      @Override
      public void onEntityRemove(
         @Nonnull Ref<EntityStore> ref, @Nonnull RemoveReason reason, @Nonnull Store<EntityStore> store, @Nonnull CommandBuffer<EntityStore> commandBuffer
      ) {
      }

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

   public static class TickingSystem extends EntityTickingSystem<EntityStore> {
      @Nonnull
      private final ComponentType<EntityStore, EntityTrackerSystems.Visible> visibleComponentType;
      @Nonnull
      private final ComponentType<EntityStore, MovementStatesComponent> movementStatesComponentComponentType;
      @Nonnull
      private final Query<EntityStore> query;

      public TickingSystem(
         @Nonnull ComponentType<EntityStore, EntityTrackerSystems.Visible> visibleComponentType,
         @Nonnull ComponentType<EntityStore, MovementStatesComponent> movementStatesComponentComponentType
      ) {
         this.visibleComponentType = visibleComponentType;
         this.query = Query.and(movementStatesComponentComponentType, visibleComponentType);
         this.movementStatesComponentComponentType = movementStatesComponentComponentType;
      }

      @Nullable
      @Override
      public SystemGroup<EntityStore> getGroup() {
         return EntityTrackerSystems.QUEUE_UPDATE_GROUP;
      }

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

      @Override
      public boolean isParallel(int archetypeChunkSize, int taskCount) {
         return EntityTickingSystem.maybeUseParallel(archetypeChunkSize, taskCount);
      }

      @Override
      public void tick(
         float dt,
         int index,
         @Nonnull ArchetypeChunk<EntityStore> archetypeChunk,
         @Nonnull Store<EntityStore> store,
         @Nonnull CommandBuffer<EntityStore> commandBuffer
      ) {
         EntityTrackerSystems.Visible visibleComponent = archetypeChunk.getComponent(index, this.visibleComponentType);

         assert visibleComponent != null;

         MovementStatesComponent movementStatesComponent = archetypeChunk.getComponent(index, this.movementStatesComponentComponentType);

         assert movementStatesComponent != null;

         MovementStates newMovementStates = movementStatesComponent.getMovementStates();
         MovementStates sentMovementStates = movementStatesComponent.getSentMovementStates();
         if (!newMovementStates.equals(sentMovementStates)) {
            copyMovementStatesFrom(newMovementStates, sentMovementStates);
            queueUpdatesFor(archetypeChunk.getReferenceTo(index), visibleComponent.visibleTo, movementStatesComponent);
         } else if (!visibleComponent.newlyVisibleTo.isEmpty()) {
            queueUpdatesFor(archetypeChunk.getReferenceTo(index), visibleComponent.newlyVisibleTo, movementStatesComponent);
         }
      }

      private static void queueUpdatesFor(
         @Nonnull Ref<EntityStore> ref,
         @Nonnull Map<Ref<EntityStore>, EntityTrackerSystems.EntityViewer> visibleTo,
         @Nonnull MovementStatesComponent movementStatesComponent
      ) {
         ComponentUpdate update = new ComponentUpdate();
         update.type = ComponentUpdateType.MovementStates;
         update.movementStates = movementStatesComponent.getMovementStates();

         for (Entry<Ref<EntityStore>, EntityTrackerSystems.EntityViewer> entry : visibleTo.entrySet()) {
            if (!ref.equals(entry.getKey())) {
               entry.getValue().queueUpdate(ref, update);
            }
         }
      }

      public static void copyMovementStatesFrom(@Nonnull MovementStates from, @Nonnull MovementStates to) {
         to.idle = from.idle;
         to.horizontalIdle = from.horizontalIdle;
         to.jumping = from.jumping;
         to.flying = from.flying;
         to.walking = from.walking;
         to.running = from.running;
         to.sprinting = from.sprinting;
         to.crouching = from.crouching;
         to.forcedCrouching = from.forcedCrouching;
         to.falling = from.falling;
         to.climbing = from.climbing;
         to.inFluid = from.inFluid;
         to.swimming = from.swimming;
         to.swimJumping = from.swimJumping;
         to.onGround = from.onGround;
         to.mantling = from.mantling;
         to.sliding = from.sliding;
         to.mounting = from.mounting;
         to.rolling = from.rolling;
         to.sitting = from.sitting;
         to.gliding = from.gliding;
         to.sleeping = from.sleeping;
      }
   }
}