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

package com.hypixel.hytale.server.core.modules.entity.repulsion;

import java.util.List;
import com.hypixel.hytale.server.core.modules.splitvelocity.VelocityConfig;
import com.hypixel.hytale.protocol.ChangeVelocityType;
import com.hypixel.hytale.math.vector.Vector3d;
import com.hypixel.hytale.server.core.modules.physics.component.Velocity;
import it.unimi.dsi.fastutil.objects.ObjectArrayList;
import com.hypixel.hytale.math.vector.Vector2d;
import com.hypixel.hytale.component.dependency.SystemDependency;
import com.hypixel.hytale.component.dependency.OrderPriority;
import com.hypixel.hytale.server.core.modules.entity.system.PlayerSpatialSystem;
import com.hypixel.hytale.component.dependency.Order;
import com.hypixel.hytale.component.spatial.SpatialResource;
import com.hypixel.hytale.component.ResourceType;
import com.hypixel.hytale.component.dependency.Dependency;
import java.util.Set;
import com.hypixel.hytale.server.core.modules.entity.component.TransformComponent;
import com.hypixel.hytale.server.core.modules.physics.systems.IVelocityModifyingSystem;
import com.hypixel.hytale.component.Component;
import com.hypixel.hytale.component.system.RefChangeSystem;
import java.util.Iterator;
import com.hypixel.hytale.protocol.ComponentUpdateType;
import com.hypixel.hytale.protocol.ComponentUpdate;
import java.util.Map;
import com.hypixel.hytale.component.Ref;
import com.hypixel.hytale.component.CommandBuffer;
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.component.RemoveReason;
import com.hypixel.hytale.server.core.universe.world.World;
import com.hypixel.hytale.component.Store;
import com.hypixel.hytale.component.AddReason;
import com.hypixel.hytale.component.Holder;
import com.hypixel.hytale.server.core.entity.entities.Player;
import javax.annotation.Nonnull;
import com.hypixel.hytale.component.query.Query;
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 RepulsionSystems
{
    public static class PlayerSetup extends HolderSystem<EntityStore>
    {
        private final ComponentType<EntityStore, Repulsion> componentType;
        @Nonnull
        private final Query<EntityStore> query;
        
        public PlayerSetup(final ComponentType<EntityStore, Repulsion> componentType, final ComponentType<EntityStore, Player> playerComponentType) {
            this.componentType = componentType;
            this.query = (Query<EntityStore>)Query.and(playerComponentType, Query.not((Query<Object>)componentType));
        }
        
        @Nonnull
        @Override
        public Query<EntityStore> getQuery() {
            return this.query;
        }
        
        @Override
        public void onEntityAdd(@Nonnull final Holder<EntityStore> holder, @Nonnull final AddReason reason, @Nonnull final Store<EntityStore> store) {
            final World world = store.getExternalData().getWorld();
            final int repulsionConfigIndex = world.getGameplayConfig().getPlayerConfig().getRepulsionConfigIndex();
            if (repulsionConfigIndex == -1) {
                if (holder.getComponent(this.componentType) != null) {
                    holder.removeComponent(this.componentType);
                }
            }
            else {
                final RepulsionConfig repulsion = RepulsionConfig.getAssetMap().getAsset(repulsionConfigIndex);
                if (holder.getComponent(this.componentType) != null) {
                    holder.removeComponent(this.componentType);
                }
                holder.addComponent(this.componentType, new Repulsion(repulsion));
            }
        }
        
        @Override
        public void onEntityRemoved(@Nonnull final Holder<EntityStore> holder, @Nonnull final RemoveReason reason, @Nonnull final Store<EntityStore> store) {
        }
    }
    
    public static class EntityTrackerUpdate extends EntityTickingSystem<EntityStore>
    {
        private final ComponentType<EntityStore, EntityTrackerSystems.Visible> visibleComponentType;
        private final ComponentType<EntityStore, Repulsion> componentType;
        @Nonnull
        private final Query<EntityStore> query;
        
        public EntityTrackerUpdate(final ComponentType<EntityStore, EntityTrackerSystems.Visible> visibleComponentType, final ComponentType<EntityStore, Repulsion> componentType) {
            this.visibleComponentType = visibleComponentType;
            this.componentType = componentType;
            this.query = (Query<EntityStore>)Query.and(visibleComponentType, componentType);
        }
        
        @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 visible = archetypeChunk.getComponent(index, this.visibleComponentType);
            final Repulsion repulsion = archetypeChunk.getComponent(index, this.componentType);
            if (repulsion.consumeNetworkOutdated()) {
                queueUpdatesFor(archetypeChunk.getReferenceTo(index), repulsion, visible.visibleTo);
            }
            else if (!visible.newlyVisibleTo.isEmpty()) {
                queueUpdatesFor(archetypeChunk.getReferenceTo(index), repulsion, visible.newlyVisibleTo);
            }
        }
        
        private static void queueUpdatesFor(final Ref<EntityStore> ref, @Nonnull final Repulsion repulsion, @Nonnull final Map<Ref<EntityStore>, EntityTrackerSystems.EntityViewer> visibleTo) {
            final ComponentUpdate update = new ComponentUpdate();
            update.type = ComponentUpdateType.Repulsion;
            update.repulsionConfigIndex = repulsion.getRepulsionConfigIndex();
            for (final EntityTrackerSystems.EntityViewer viewer : visibleTo.values()) {
                viewer.queueUpdate(ref, update);
            }
        }
    }
    
    public static class EntityTrackerRemove extends RefChangeSystem<EntityStore, Repulsion>
    {
        private final ComponentType<EntityStore, Repulsion> componentType;
        private final ComponentType<EntityStore, EntityTrackerSystems.Visible> visibleComponentType;
        @Nonnull
        private final Query<EntityStore> query;
        
        public EntityTrackerRemove(final ComponentType<EntityStore, EntityTrackerSystems.Visible> visibleComponentType, final ComponentType<EntityStore, Repulsion> componentType) {
            this.visibleComponentType = visibleComponentType;
            this.componentType = componentType;
            this.query = (Query<EntityStore>)Query.and(visibleComponentType, componentType);
        }
        
        @Nonnull
        @Override
        public Query<EntityStore> getQuery() {
            return this.query;
        }
        
        @Nonnull
        @Override
        public ComponentType<EntityStore, Repulsion> componentType() {
            return this.componentType;
        }
        
        @Override
        public void onComponentAdded(@Nonnull final Ref<EntityStore> ref, @Nonnull final Repulsion component, @Nonnull final Store<EntityStore> store, @Nonnull final CommandBuffer<EntityStore> commandBuffer) {
        }
        
        @Override
        public void onComponentSet(@Nonnull final Ref<EntityStore> ref, final Repulsion oldComponent, @Nonnull final Repulsion newComponent, @Nonnull final Store<EntityStore> store, @Nonnull final CommandBuffer<EntityStore> commandBuffer) {
        }
        
        @Override
        public void onComponentRemoved(@Nonnull final Ref<EntityStore> ref, @Nonnull final Repulsion component, @Nonnull final Store<EntityStore> store, @Nonnull final CommandBuffer<EntityStore> commandBuffer) {
            for (final EntityTrackerSystems.EntityViewer viewer : store.getComponent(ref, this.visibleComponentType).visibleTo.values()) {
                viewer.queueRemove(ref, ComponentUpdateType.Repulsion);
            }
        }
    }
    
    public static class RepulsionTicker extends EntityTickingSystem<EntityStore> implements IVelocityModifyingSystem
    {
        private final ComponentType<EntityStore, Repulsion> repulsionComponentType;
        private final ComponentType<EntityStore, TransformComponent> transformComponentComponentType;
        @Nonnull
        private final Query<EntityStore> query;
        @Nonnull
        private final Set<Dependency<EntityStore>> dependencies;
        private final ResourceType<EntityStore, SpatialResource<Ref<EntityStore>, EntityStore>> spatialComponent;
        
        public RepulsionTicker(final ComponentType<EntityStore, Repulsion> repulsionComponentType, final ComponentType<EntityStore, TransformComponent> transformComponentComponentType, final ResourceType<EntityStore, SpatialResource<Ref<EntityStore>, EntityStore>> spatialComponent) {
            this.repulsionComponentType = repulsionComponentType;
            this.transformComponentComponentType = transformComponentComponentType;
            this.query = (Query<EntityStore>)Query.and(repulsionComponentType, transformComponentComponentType);
            this.spatialComponent = spatialComponent;
            this.dependencies = (Set<Dependency<EntityStore>>)Set.of(new SystemDependency(Order.AFTER, PlayerSpatialSystem.class, OrderPriority.CLOSEST));
        }
        
        @Nonnull
        @Override
        public Set<Dependency<EntityStore>> getDependencies() {
            return this.dependencies;
        }
        
        @Nullable
        @Override
        public SystemGroup<EntityStore> getGroup() {
            return EntityTrackerSystems.QUEUE_UPDATE_GROUP;
        }
        
        @Nonnull
        @Override
        public Query<EntityStore> getQuery() {
            return this.query;
        }
        
        @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 Repulsion repulsionComponent = archetypeChunk.getComponent(index, this.repulsionComponentType);
            assert repulsionComponent != null;
            final int repulsionConfigIndex = repulsionComponent.getRepulsionConfigIndex();
            if (repulsionConfigIndex == -1) {
                return;
            }
            final RepulsionConfig repulsion = RepulsionConfig.getAssetMap().getAsset(repulsionConfigIndex);
            final float radius = repulsion.radius;
            final TransformComponent transformComponent = archetypeChunk.getComponent(index, this.transformComponentComponentType);
            assert transformComponent != null;
            final Vector2d position = new Vector2d(transformComponent.getPosition().x, transformComponent.getPosition().z);
            final SpatialResource<Ref<EntityStore>, EntityStore> spatialResource = store.getResource(this.spatialComponent);
            final List<Ref<EntityStore>> results = new ObjectArrayList<Ref<EntityStore>>();
            spatialResource.getSpatialStructure().ordered(transformComponent.getPosition(), radius, results);
            for (final Ref<EntityStore> entityRef : results) {
                final TransformComponent entityTransformComponent = commandBuffer.getComponent(entityRef, this.transformComponentComponentType);
                if (entityTransformComponent == null) {
                    continue;
                }
                final Vector2d entityPosition = new Vector2d(entityTransformComponent.getPosition().x, entityTransformComponent.getPosition().z);
                if (entityPosition.equals(position)) {
                    continue;
                }
                final double distance = position.distanceTo(entityPosition);
                if (distance < 0.1) {
                    continue;
                }
                final double fraction = (radius - distance) / radius;
                float maxForce = repulsion.maxForce;
                int flip = 1;
                if (maxForce < 0.0f) {
                    flip = -1;
                    maxForce *= flip;
                }
                double force = Math.max(repulsion.minForce, maxForce * fraction);
                force *= flip;
                final Vector2d push = entityPosition.subtract(position);
                push.normalize();
                push.scale(force);
                final Velocity entityVelocityComponent = commandBuffer.getComponent(entityRef, Velocity.getComponentType());
                if (entityVelocityComponent == null) {
                    continue;
                }
                final Vector3d addedVelocity = new Vector3d((float)push.x, 0.0, (float)push.y);
                entityVelocityComponent.addInstruction(addedVelocity, null, ChangeVelocityType.Add);
            }
        }
    }
}