class="kw">package com.hypixel.hytale.builtin.hytalegenerator.engine.stages;

class="kw">import com.hypixel.hytale.builtin.hytalegenerator.bounds.Bounds3i;
class="kw">import com.hypixel.hytale.builtin.hytalegenerator.engine.bufferbundle.BufferBundle;
class="kw">import com.hypixel.hytale.builtin.hytalegenerator.engine.bufferbundle.buffers.CountedPixelBuffer;
class="kw">import com.hypixel.hytale.builtin.hytalegenerator.engine.bufferbundle.buffers.type.BufferType;
class="kw">import com.hypixel.hytale.builtin.hytalegenerator.engine.bufferbundle.buffers.type.ParametrizedBufferType;
class="kw">import com.hypixel.hytale.builtin.hytalegenerator.engine.views.PixelBufferView;
class="kw">import com.hypixel.hytale.builtin.hytalegenerator.workerindexer.WorkerIndexer;
class="kw">import com.hypixel.hytale.builtin.hytalegenerator.worldstructure.BiCarta;
class="kw">import com.hypixel.hytale.builtin.hytalegenerator.worldstructure.WorldStructure;
class="kw">import java.util.List;
class="kw">import java.util.Map;
class="kw">import javax.annotation.Nonnull;

class="kw">public class BiomeStage class="kw">implements Stage {
   @Nonnull
   class="kw">public class="kw">static class="kw">final Class<CountedPixelBuffer> bufferClass = CountedPixelBuffer.class;
   @Nonnull
   class="kw">public class="kw">static class="kw">final Class<Integer> biomeClass = Integer.class;
   @Nonnull
   class="kw">private class="kw">final ParametrizedBufferType biomeOutputBufferType;
   @Nonnull
   class="kw">private class="kw">final String stageName;
   @Nonnull
   class="kw">private class="kw">final WorkerIndexer.Data<WorldStructure> worldStructure_workerData;

   class="kw">public BiomeStage(@Nonnull String var1, @Nonnull ParametrizedBufferType var2, @Nonnull WorkerIndexer.Data<WorldStructure> var3) {
      this.stageName = var1;
      this.biomeOutputBufferType = var2;
      this.worldStructure_workerData = var3;
   }

   @Override
   class="kw">public void run(@Nonnull Stage.Context var1) {
      BufferBundle.Access.View var2 = var1.bufferAccess.get(this.biomeOutputBufferType);
      PixelBufferView var3 = new PixelBufferView<>(var2, biomeClass);
      BiCarta var4 = this.worldStructure_workerData.get(var1.workerId).getBiomeMap();
      Bounds3i var5 = var3.getBounds();

      for (int var6 = var5.min.x; var6 < var5.max.x; var6++) {
         for (int var7 = var5.min.z; var7 < var5.max.z; var7++) {
            Integer var8 = var4.apply(var6, var7, var1.workerId);
            var3.set(var8, var6, 0, var7);
         }
      }
   }

   @Nonnull
   @Override
   class="kw">public Map<BufferType, Bounds3i> getInputTypesAndBounds_bufferGrid() {
      class="kw">return Map.of();
   }

   @Nonnull
   @Override
   class="kw">public List<BufferType> getOutputTypes() {
      class="kw">return List.of(this.biomeOutputBufferType);
   }

   @Nonnull
   @Override
   class="kw">public String getName() {
      class="kw">return this.stageName;
   }
}