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

class="kw">import com.hypixel.hytale.builtin.hytalegenerator.graph.GraphSpace;
class="kw">import java.util.List;
class="kw">import javax.annotation.Nonnull;

class="kw">public class NeighborNodesNodeSelector class="kw">extends NodeSelector {
   class="kw">public class="kw">static class="kw">final int ALL_NEIGHBORS_THRESHOLD = -1;
   @Nonnull
   class="kw">private class="kw">final NodeSelector nodeSelector;
   class="kw">private class="kw">final int selectedNeighborsThreshold;

   class="kw">public NeighborNodesNodeSelector(@Nonnull NodeSelector var1, int var2) {
      this.nodeSelector = var1;
      this.selectedNeighborsThreshold = var2;
   }

   @Override
   class="kw">public boolean isSelected(@Nonnull GraphSpace var1, @Nonnull GraphSpace.Node var2) {
      int var3 = 0;
      List var4 = var2.edges();

      for (int var5 = 0; var5 < var4.size(); var5++) {
         GraphSpace.Edge var6 = var4.get(var5);
         if (this.nodeSelector.isSelected(var1, var6.otherNode(var2))) {
            var3++;
         }
      }

      class="kw">return this.selectedNeighborsThreshold != -1 ? var3 >= this.selectedNeighborsThreshold : var3 == var2.getEdgesCount() && !var4.isEmpty();
   }

   @Override
   class="kw">public double getInteractionRange(double var1) {
      class="kw">return var1 + this.nodeSelector.getInteractionRange(var1);
   }
}