class="kw">package com.hypixel.hytale.server.core.modules.entity.system;

class="kw">import com.hypixel.hytale.component.ArchetypeChunk;
class="kw">import com.hypixel.hytale.component.CommandBuffer;
class="kw">import com.hypixel.hytale.component.Component;
class="kw">import com.hypixel.hytale.component.ComponentType;
class="kw">import com.hypixel.hytale.component.Ref;
class="kw">import com.hypixel.hytale.component.Resource;
class="kw">import com.hypixel.hytale.component.ResourceType;
class="kw">import com.hypixel.hytale.component.Store;
class="kw">import com.hypixel.hytale.component.SystemGroup;
class="kw">import com.hypixel.hytale.component.query.Query;
class="kw">import com.hypixel.hytale.component.system.RefChangeSystem;
class="kw">import com.hypixel.hytale.component.system.tick.EntityTickingSystem;
class="kw">import com.hypixel.hytale.protocol.ComponentUpdate;
class="kw">import com.hypixel.hytale.protocol.ComponentUpdateType;
class="kw">import com.hypixel.hytale.server.core.modules.entity.EntityModule;
class="kw">import com.hypixel.hytale.server.core.modules.entity.tracker.EntityTrackerSystems;
class="kw">import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
class="kw">import java.util.Map;
class="kw">import java.util.Set;
class="kw">import java.util.concurrent.ConcurrentHashMap;
class="kw">import java.util.function.Supplier;
class="kw">import javax.annotation.Nonnull;
class="kw">import javax.annotation.Nullable;

class="kw">public class MarkerReplicationSystems {
   class="kw">public MarkerReplicationSystems() {
   }

   class="kw">public class="kw">abstract class="kw">static class EntityTrackerAddAndRemove<T class="kw">extends Component<EntityStore>> class="kw">extends RefChangeSystem<EntityStore, T> {
      class="kw">private class="kw">final ComponentType<EntityStore, T> markerComponentType;
      class="kw">private class="kw">final ComponentType<EntityStore, EntityTrackerSystems.Visible> visibleComponentType;
      class="kw">private class="kw">final ComponentUpdateType componentUpdateType;
      @Nonnull
      class="kw">private class="kw">final Query<EntityStore> query;

      class="kw">protected EntityTrackerAddAndRemove(
         ComponentType<EntityStore, EntityTrackerSystems.Visible> var1, ComponentType<EntityStore, T> var2, ComponentUpdateType var3
      ) {
         this.markerComponentType = var2;
         this.visibleComponentType = var1;
         this.componentUpdateType = var3;
         this.query = Query.and(var1, var2);
      }

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

      @Nonnull
      @Override
      class="kw">public ComponentType<EntityStore, T> componentType() {
         class="kw">return this.markerComponentType;
      }

      @Override
      class="kw">public void onComponentAdded(@Nonnull Ref<EntityStore> var1, @Nonnull T var2, @Nonnull Store<EntityStore> var3, @Nonnull CommandBuffer<EntityStore> var4) {
         var4.getResource(MarkerReplicationSystems.QueueResource.getResourceType()).queue(this.markerComponentType).add(var1);
      }

      @Override
      class="kw">public void onComponentSet(
         @Nonnull Ref<EntityStore> var1, T var2, @Nonnull T var3, @Nonnull Store<EntityStore> var4, @Nonnull CommandBuffer<EntityStore> var5
      ) {
      }

      @Override
      class="kw">public void onComponentRemoved(
         @Nonnull Ref<EntityStore> var1, @Nonnull T var2, @Nonnull Store<EntityStore> var3, @Nonnull CommandBuffer<EntityStore> var4
      ) {
         for (EntityTrackerSystems.EntityViewer var6 : var3.getComponent(var1, this.visibleComponentType).visibleTo.values()) {
            var6.queueRemove(var1, this.componentUpdateType);
         }
      }
   }

   class="kw">public class="kw">abstract class="kw">static class EntityTrackerUpdate class="kw">extends EntityTickingSystem<EntityStore> {
      class="kw">private class="kw">final ComponentType<EntityStore, EntityTrackerSystems.Visible> visibleComponentType;
      class="kw">private class="kw">final ComponentType<EntityStore, ?> markerComponentType;
      class="kw">private class="kw">final Supplier<ComponentUpdate> updateFactory;
      @Nonnull
      class="kw">private class="kw">final Query<EntityStore> query;

      class="kw">protected EntityTrackerUpdate(
         ComponentType<EntityStore, EntityTrackerSystems.Visible> var1, ComponentType<EntityStore, ?> var2, Supplier<ComponentUpdate> var3
      ) {
         this.visibleComponentType = var1;
         this.markerComponentType = var2;
         this.updateFactory = var3;
         this.query = Query.and(var1, var2);
      }

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

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

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

      @Override
      class="kw">public void tick(float var1, int var2, @Nonnull Store<EntityStore> var3) {
         super.tick(var1, var2, var3);
         var3.getResource(MarkerReplicationSystems.QueueResource.getResourceType()).queue(this.markerComponentType).clear();
      }

      @Override
      class="kw">public void tick(
         float var1, int var2, @Nonnull ArchetypeChunk<EntityStore> var3, @Nonnull Store<EntityStore> var4, @Nonnull CommandBuffer<EntityStore> var5
      ) {
         EntityTrackerSystems.Visible var6 = var3.getComponent(var2, this.visibleComponentType);
         Ref var7 = var3.getReferenceTo(var2);
         if (var5.getResource(MarkerReplicationSystems.QueueResource.getResourceType()).queue(this.markerComponentType).remove(var7)) {
            this.queueUpdatesFor(var7, var6.visibleTo);
         } else if (!var6.newlyVisibleTo.isEmpty()) {
            this.queueUpdatesFor(var7, var6.newlyVisibleTo);
         }
      }

      class="kw">private void queueUpdatesFor(Ref<EntityStore> var1, @Nonnull Map<Ref<EntityStore>, EntityTrackerSystems.EntityViewer> var2) {
         ComponentUpdate var3 = this.updateFactory.get();

         for (EntityTrackerSystems.EntityViewer var5 : var2.values()) {
            var5.queueUpdate(var1, var3);
         }
      }
   }

   class="kw">public class="kw">static class QueueResource class="kw">implements Resource<EntityStore> {
      class="kw">private class="kw">final Map<ComponentType<EntityStore, ?>, Set<Ref<EntityStore>>> queues = new ConcurrentHashMap<>();

      class="kw">public QueueResource() {
      }

      class="kw">public class="kw">static ResourceType<EntityStore, MarkerReplicationSystems.QueueResource> getResourceType() {
         class="kw">return EntityModule.get().getMarkerReplicationQueueResourceType();
      }

      @Nonnull
      Set<Ref<EntityStore>> queue(@Nonnull ComponentType<EntityStore, ?> var1) {
         class="kw">return this.queues.computeIfAbsent(var1, var0 -> ConcurrentHashMap.newKeySet());
      }

      @Nonnull
      @Override
      class="kw">public Resource<EntityStore> clone() {
         class="kw">return new MarkerReplicationSystems.QueueResource();
      }
   }
}