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

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

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

public class WorldPregenerateSystem extends StoreSystem<ChunkStore>
{
    private static final Set<Dependency<ChunkStore>> DEPENDENCIES;
    
    @Nonnull
    @Override
    public Set<Dependency<ChunkStore>> getDependencies() {
        return WorldPregenerateSystem.DEPENDENCIES;
    }
    
    @Override
    public void onSystemAddedToStore(@Nonnull final Store<ChunkStore> store) {
        final World world = store.getExternalData().getWorld();
        final Box2D region = world.getWorldConfig().getChunkConfig().getPregenerateRegion();
        if (region != null) {
            world.getLogger().at(Level.INFO).log("Ensuring region is generated: %s", region);
            final long start = java.lang.System.nanoTime();
            final int lowX = MathUtil.floor(region.min.x);
            final int lowZ = MathUtil.floor(region.min.y);
            final int highX = MathUtil.floor(region.max.x);
            final int highZ = MathUtil.floor(region.max.y);
            final 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)));
                }
            }
            final int allFutures = futures.size();
            final AtomicInteger done = new AtomicInteger();
            futures.forEach(f -> f.whenComplete((worldChunk, throwable) -> {
                if (throwable != null) {
                    world.getLogger().at(Level.SEVERE).withCause(throwable).log("Failed to load/generate chunk:");
                }
                if (done.incrementAndGet() == allFutures) {
                    final long end = java.lang.System.nanoTime();
                    world.getLogger().at(Level.INFO).log("Finished loading %d chunks. Finished in %s", allFutures, FormatUtil.nanosToString(end - start));
                }
            }));
        }
    }
    
    @Override
    public void onSystemRemovedFromStore(@Nonnull final Store<ChunkStore> store) {
    }
    
    static {
        DEPENDENCIES = Set.of(new SystemGroupDependency(Order.AFTER, ChunkStore.INIT_GROUP));
    }
}