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

class="kw">import com.hypixel.hytale.builtin.hytalegenerator.graph.GraphSpace;
class="kw">import com.hypixel.hytale.builtin.hytalegenerator.graph.nodeselectors.NodeSelector;
class="kw">import javax.annotation.Nonnull;

class="kw">public class NodesEdgeSelector class="kw">extends EdgeSelector {
   @Nonnull
   class="kw">private class="kw">final NodeSelector nodeSelector;
   @Nonnull
   class="kw">private class="kw">final NodesEdgeSelector.Operator operator;

   class="kw">public NodesEdgeSelector(@Nonnull NodeSelector var1, @Nonnull NodesEdgeSelector.Operator var2) {
      this.nodeSelector = var1;
      this.operator = var2;
   }

   @Override
   class="kw">public boolean isSelected(@Nonnull GraphSpace var1, @Nonnull GraphSpace.Edge var2) {
      class="kw">return this.operator == NodesEdgeSelector.Operator.OR
         ? this.nodeSelector.isSelected(var1, var2.nodeA()) || this.nodeSelector.isSelected(var1, var2.nodeB())
         : this.nodeSelector.isSelected(var1, var2.nodeA()) && this.nodeSelector.isSelected(var1, var2.nodeB());
   }

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

   class="kw">public enum Operator {
      AND,
      OR;

      Operator() {
      }
   }
}