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

class="kw">import com.hypixel.hytale.builtin.hytalegenerator.graph.GraphSpace;
class="kw">import it.unimi.dsi.fastutil.ints.IntOpenHashSet;
class="kw">import it.unimi.dsi.fastutil.ints.IntSet;
class="kw">import java.util.List;
class="kw">import javax.annotation.Nonnull;

class="kw">public class ConnectionCountNodeSelector class="kw">extends NodeSelector {
   @Nonnull
   class="kw">private class="kw">final IntSet countsSet;

   class="kw">public ConnectionCountNodeSelector(@Nonnull List<Integer> var1) {
      assert !var1.contains(null);
      this.countsSet = new IntOpenHashSet(var1);
   }

   @Override
   class="kw">public boolean isSelected(@Nonnull GraphSpace var1, @Nonnull GraphSpace.Node var2) {
      class="kw">return this.countsSet.contains(var2.getEdgesCount());
   }

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