class="kw">package com.hypixel.hytale.builtin.hytalegenerator.assets.density;

class="kw">import com.hypixel.hytale.builtin.hytalegenerator.density.CacheDensity;
class="kw">import com.hypixel.hytale.builtin.hytalegenerator.density.ConstantValueDensity;
class="kw">import com.hypixel.hytale.builtin.hytalegenerator.density.Density;
class="kw">import com.hypixel.hytale.builtin.hytalegenerator.density.MultiCacheDensity;
class="kw">import com.hypixel.hytale.codec.Codec;
class="kw">import com.hypixel.hytale.codec.KeyedCodec;
class="kw">import com.hypixel.hytale.codec.builder.BuilderCodec;
class="kw">import com.hypixel.hytale.codec.validation.Validators;
class="kw">import javax.annotation.Nonnull;

class="kw">public class CacheDensityAsset class="kw">extends DensityAsset {
   @Nonnull
   class="kw">public class="kw">static class="kw">final BuilderCodec<CacheDensityAsset> CODEC = BuilderCodec.builder(
         CacheDensityAsset.class, CacheDensityAsset::new, DensityAsset.ABSTRACT_CODEC
      )
      .append(new KeyedCodec<>("Capacity", Codec.INTEGER, true), (var0, var1) -> var0.capacity = var1, var0 -> var0.capacity)
      .addValidator(Validators.greaterThanOrEqual(0))
      .add()
      .build();
   class="kw">public class="kw">static int DEFAULT_CAPACITY = 3;
   class="kw">private int capacity = DEFAULT_CAPACITY;

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

   @Nonnull
   @Override
   class="kw">public Density build(@Nonnull DensityAsset.Argument var1) {
      if (this.isSkipped() || this.inputs().length == 0) {
         class="kw">return new ConstantValueDensity(0.0);
      } else if (this.capacity <= 0) {
         class="kw">return this.buildFirstInput(var1);
      } else {
         class="kw">return this.capacity == 1 ? new CacheDensity(this.buildFirstInput(var1)) : new MultiCacheDensity(this.buildFirstInput(var1), this.capacity);
      }
   }

   @Override
   class="kw">public void cleanUp() {
      this.cleanUpInputs();
   }
}