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

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

class="kw">public class NeighborEdgesNodeSelector 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 EdgeSelector edgeSelector;
   class="kw">private class="kw">final int selectedNeighborsThreshold;

   class="kw">public NeighborEdgesNodeSelector(@Nonnull EdgeSelector var1, int var2) {
      this.edgeSelector = 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.edgeSelector.isSelected(var1, var6)) {
            var3++;
         }
      }

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

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