class="kw">package com.hypixel.hytale.server.npc.corecomponents.world;

class="kw">import com.hypixel.hytale.component.Ref;
class="kw">import com.hypixel.hytale.component.Store;
class="kw">import com.hypixel.hytale.server.core.modules.entity.component.TransformComponent;
class="kw">import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
class="kw">import com.hypixel.hytale.server.npc.asset.builder.BuilderSupport;
class="kw">import com.hypixel.hytale.server.npc.corecomponents.SensorBase;
class="kw">import com.hypixel.hytale.server.npc.corecomponents.world.builders.BuilderSensorLeash;
class="kw">import com.hypixel.hytale.server.npc.entities.NPCEntity;
class="kw">import com.hypixel.hytale.server.npc.instructions.ExecutionSupport;
class="kw">import com.hypixel.hytale.server.npc.sensorinfo.InfoProvider;
class="kw">import com.hypixel.hytale.server.npc.sensorinfo.PositionProvider;
class="kw">import javax.annotation.Nonnull;
class="kw">import org.joml.Vector3d;

class="kw">public class SensorLeash class="kw">extends SensorBase {
   class="kw">protected class="kw">final double range;
   class="kw">protected class="kw">final double rangeSq;
   class="kw">protected class="kw">final PositionProvider positionProvider = new PositionProvider();

   class="kw">public SensorLeash(@Nonnull BuilderSensorLeash var1, @Nonnull BuilderSupport var2) {
      super(var1);
      this.range = var1.getRange(var2);
      this.rangeSq = this.range * this.range;
   }

   @Override
   class="kw">public boolean matches(@Nonnull Ref<EntityStore> var1, @Nonnull ExecutionSupport var2, double var3, @Nonnull Store<EntityStore> var5) {
      if (!super.matches(var1, var2, var3, var5)) {
         class="kw">return false;
      } else {
         TransformComponent var6 = var5.getComponent(var1, TransformComponent.getComponentType());
         assert var6 != null;
         NPCEntity var7 = var5.getComponent(var1, NPCEntity.getComponentType());
         Vector3d var8 = var7.getLeashPoint();
         if (var6.getPosition().distanceSquared(var8) > this.rangeSq) {
            this.positionProvider.setTarget(var8);
            class="kw">return true;
         } else {
            this.positionProvider.clear();
            class="kw">return false;
         }
      }
   }

   @Override
   class="kw">public InfoProvider getSensorInfo() {
      class="kw">return this.positionProvider;
   }
}