// 
// Decompiled by Procyon v0.6.0
// 

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

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

public class MovementStatesSystems
{
    public static class AddSystem extends HolderSystem<EntityStore>
    {
        @Nonnull
        private final ComponentType<EntityStore, MovementStatesComponent> movementStatesComponentComponentType;
        
        public AddSystem(@Nonnull final ComponentType<EntityStore, MovementStatesComponent> movementStatesComponentComponentType) {
            this.movementStatesComponentComponentType = movementStatesComponentComponentType;
        }
        
        @Override
        public void onEntityAdd(@Nonnull final Holder<EntityStore> holder, @Nonnull final AddReason reason, @Nonnull final Store<EntityStore> store) {
            holder.ensureComponent(this.movementStatesComponentComponentType);
        }
        
        @Override
        public void onEntityRemoved(@Nonnull final Holder<EntityStore> holder, @Nonnull final RemoveReason reason, @Nonnull final 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 final ComponentType<EntityStore, Player> playerComponentType, @Nonnull final ComponentType<EntityStore, MovementStatesComponent> movementStatesComponentType) {
            this.playerComponentType = playerComponentType;
            this.movementStatesComponentType = movementStatesComponentType;
            this.query = (Query<EntityStore>)Query.and(playerComponentType, movementStatesComponentType);
        }
        
        @Override
        public void onEntityAdded(@Nonnull final Ref<EntityStore> ref, @Nonnull final AddReason reason, @Nonnull final Store<EntityStore> store, @Nonnull final CommandBuffer<EntityStore> commandBuffer) {
            final World world = commandBuffer.getExternalData().getWorld();
            final Player playerComponent = store.getComponent(ref, this.playerComponentType);
            assert playerComponent != null;
            final MovementStatesComponent movementStatesComponent = store.getComponent(ref, this.movementStatesComponentType);
            assert movementStatesComponent != null;
            final PlayerWorldData perWorldData = playerComponent.getPlayerConfigData().getPerWorldData(world.getName());
            final SavedMovementStates movementStates = perWorldData.getLastMovementStates();
            playerComponent.applyMovementStates(ref, (movementStates != null) ? movementStates : new SavedMovementStates(), movementStatesComponent.getMovementStates(), store);
        }
        
        @Override
        public void onEntityRemove(@Nonnull final Ref<EntityStore> ref, @Nonnull final RemoveReason reason, @Nonnull final Store<EntityStore> store, @Nonnull final 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 final ComponentType<EntityStore, EntityTrackerSystems.Visible> visibleComponentType, @Nonnull final ComponentType<EntityStore, MovementStatesComponent> movementStatesComponentComponentType) {
            this.visibleComponentType = visibleComponentType;
            this.query = (Query<EntityStore>)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(final int archetypeChunkSize, final int taskCount) {
            return EntityTickingSystem.maybeUseParallel(archetypeChunkSize, taskCount);
        }
        
        @Override
        public void tick(final float dt, final int index, @Nonnull final ArchetypeChunk<EntityStore> archetypeChunk, @Nonnull final Store<EntityStore> store, @Nonnull final CommandBuffer<EntityStore> commandBuffer) {
            final EntityTrackerSystems.Visible visibleComponent = archetypeChunk.getComponent(index, this.visibleComponentType);
            assert visibleComponent != null;
            final MovementStatesComponent movementStatesComponent = archetypeChunk.getComponent(index, this.movementStatesComponentComponentType);
            assert movementStatesComponent != null;
            final MovementStates newMovementStates = movementStatesComponent.getMovementStates();
            final 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 final Ref<EntityStore> ref, @Nonnull final Map<Ref<EntityStore>, EntityTrackerSystems.EntityViewer> visibleTo, @Nonnull final MovementStatesComponent movementStatesComponent) {
            final ComponentUpdate update = new ComponentUpdate();
            update.type = ComponentUpdateType.MovementStates;
            update.movementStates = movementStatesComponent.getMovementStates();
            for (final Map.Entry<Ref<EntityStore>, EntityTrackerSystems.EntityViewer> entry : visibleTo.entrySet()) {
                if (ref.equals(entry.getKey())) {
                    continue;
                }
                entry.getValue().queueUpdate(ref, update);
            }
        }
        
        public static void copyMovementStatesFrom(@Nonnull final MovementStates from, @Nonnull final 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;
        }
    }
}