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

class="kw">import com.hypixel.hytale.builtin.hytalegenerator.assets.graph.GraphGeneratorAsset;
class="kw">import com.hypixel.hytale.builtin.hytalegenerator.graph.nodeselectors.AllNodeSelector;
class="kw">import com.hypixel.hytale.builtin.hytalegenerator.graph.nodeselectors.NeighborNodesNodeSelector;
class="kw">import com.hypixel.hytale.builtin.hytalegenerator.graph.nodeselectors.NodeSelector;
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 javax.annotation.Nonnull;

class="kw">public class NeighborNodesNodeSelectorAsset class="kw">extends NodeSelectorAsset {
   @Nonnull
   class="kw">public class="kw">static class="kw">final BuilderCodec<NeighborNodesNodeSelectorAsset> CODEC = BuilderCodec.builder(
         NeighborNodesNodeSelectorAsset.class, NeighborNodesNodeSelectorAsset::new, NodeSelectorAsset.ABSTRACT_CODEC
      )
      .append(new KeyedCodec<>("NodeSelector", NodeSelectorAsset.CODEC, true), (var0, var1) -> var0.nodeSelectorAsset = var1, var0 -> var0.nodeSelectorAsset)
      .add()
      .append(
         new KeyedCodec<>("SelectedNeighborsThreshold", Codec.INTEGER, true),
         (var0, var1) -> var0.selectedNeighborsThreshold = var1,
         var0 -> var0.selectedNeighborsThreshold
      )
      .add()
      .build();
   @Nonnull
   class="kw">private NodeSelectorAsset nodeSelectorAsset = AllNodeSelectorAsset.INSTANCE;
   class="kw">private int selectedNeighborsThreshold = -1;

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

   @Nonnull
   @Override
   class="kw">public NodeSelector build(@Nonnull GraphGeneratorAsset.Argument var1) {
      class="kw">return super.skip() ? AllNodeSelector.INSTANCE : new NeighborNodesNodeSelector(this.nodeSelectorAsset.build(var1), this.selectedNeighborsThreshold);
   }

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