WorldPregenerateSystem.java

package com.hypixel.hytale.server.core.universe.world.system;

import com.hypixel.hytale.common.util.FormatUtil;
import com.hypixel.hytale.component.Ref;
import com.hypixel.hytale.component.Store;
import com.hypixel.hytale.component.dependency.Dependency;
import com.hypixel.hytale.component.dependency.Order;
import com.hypixel.hytale.component.dependency.SystemGroupDependency;
import com.hypixel.hytale.component.system.StoreSystem;
import com.hypixel.hytale.logger.HytaleLogger;
import com.hypixel.hytale.math.shape.Box2D;
import com.hypixel.hytale.math.util.ChunkUtil;
import com.hypixel.hytale.math.util.MathUtil;
import com.hypixel.hytale.server.core.universe.world.World;
import com.hypixel.hytale.server.core.universe.world.storage.ChunkStore;
import it.unimi.dsi.fastutil.objects.ObjectArrayList;
import java.util.List;
import java.util.Set;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.logging.Level;
import javax.annotation.Nonnull;

public class WorldPregenerateSystem extends StoreSystem<ChunkStore> {
   private static final Set<Dependency<ChunkStore>> DEPENDENCIES;

   public WorldPregenerateSystem() {
   }

   @Nonnull
   public Set<Dependency<ChunkStore>> getDependencies() {
      return DEPENDENCIES;
   }

   public void onSystemAddedToStore(@Nonnull Store<ChunkStore> store) {
      World world = ((ChunkStore)store.getExternalData()).getWorld();
      Box2D region = world.getWorldConfig().getChunkConfig().getPregenerateRegion();
      if (region != null) {
         world.getLogger().at(Level.INFO).log("Ensuring region is generated: %s", region);
         long start = System.nanoTime();
         int lowX = MathUtil.floor(region.min.x);
         int lowZ = MathUtil.floor(region.min.y);
         int highX = MathUtil.floor(region.max.x);
         int highZ = MathUtil.floor(region.max.y);
         List<CompletableFuture<Ref<ChunkStore>>> futures = new ObjectArrayList<CompletableFuture<Ref<ChunkStore>>>();

         for(int x = lowX; x <= highX; x += 32) {
            for(int z = lowZ; z <= highZ; z += 32) {
               futures.add(world.getChunkStore().getChunkReferenceAsync(ChunkUtil.indexChunkFromBlock(x, z)));
            }
         }

         int allFutures = futures.size();
         AtomicInteger done = new AtomicInteger();
         futures.forEach((f) -> f.whenComplete((worldChunk, throwable) -> {
               if (throwable != null) {
                  ((HytaleLogger.Api)world.getLogger().at(Level.SEVERE).withCause(throwable)).log("Failed to load/generate chunk:");
               }

               if (done.incrementAndGet() == allFutures) {
                  long end = System.nanoTime();
                  world.getLogger().at(Level.INFO).log("Finished loading %d chunks. Finished in %s", allFutures, FormatUtil.nanosToString(end - start));
               }

            }));
      }

   }

   public void onSystemRemovedFromStore(@Nonnull Store<ChunkStore> store) {
   }

   static {
      DEPENDENCIES = Set.of(new SystemGroupDependency(Order.AFTER, ChunkStore.INIT_GROUP));
   }
}