SensorAdjustPosition.java

package com.hypixel.hytale.server.npc.corecomponents.utility;

import com.hypixel.hytale.component.ComponentAccessor;
import com.hypixel.hytale.component.Ref;
import com.hypixel.hytale.component.Store;
import com.hypixel.hytale.math.vector.Vector3d;
import com.hypixel.hytale.server.core.universe.world.World;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import com.hypixel.hytale.server.npc.asset.builder.BuilderSupport;
import com.hypixel.hytale.server.npc.corecomponents.SensorBase;
import com.hypixel.hytale.server.npc.corecomponents.utility.builders.BuilderSensorAdjustPosition;
import com.hypixel.hytale.server.npc.entities.NPCEntity;
import com.hypixel.hytale.server.npc.instructions.Sensor;
import com.hypixel.hytale.server.npc.movement.controllers.MotionController;
import com.hypixel.hytale.server.npc.role.Role;
import com.hypixel.hytale.server.npc.role.support.DebugSupport;
import com.hypixel.hytale.server.npc.sensorinfo.IPositionProvider;
import com.hypixel.hytale.server.npc.sensorinfo.InfoProvider;
import com.hypixel.hytale.server.npc.sensorinfo.PositionProvider;
import com.hypixel.hytale.server.npc.util.IAnnotatedComponent;
import com.hypixel.hytale.server.npc.util.IAnnotatedComponentCollection;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;

public class SensorAdjustPosition extends SensorBase implements IAnnotatedComponentCollection {
   protected final Sensor sensor;
   @Nonnull
   protected final Vector3d offset;
   protected final PositionProvider positionProvider = new PositionProvider();

   public SensorAdjustPosition(@Nonnull BuilderSensorAdjustPosition builder, @Nonnull BuilderSupport support, Sensor sensor) {
      super(builder);
      this.sensor = sensor;
      this.offset = builder.getOffset(support);
   }

   public boolean matches(@Nonnull Ref<EntityStore> ref, @Nonnull Role role, double dt, @Nonnull Store<EntityStore> store) {
      if (super.matches(ref, role, dt, store) && this.sensor.matches(ref, role, dt, store)) {
         InfoProvider sensorInfo = this.sensor.getSensorInfo();
         if (sensorInfo == null) {
            return false;
         } else {
            IPositionProvider otherPositionProvider = sensorInfo.getPositionProvider();
            double x = otherPositionProvider.getX() + this.offset.x;
            double y = otherPositionProvider.getY() + this.offset.y;
            double z = otherPositionProvider.getZ() + this.offset.z;
            this.positionProvider.setTarget(x, y, z);
            return true;
         }
      } else {
         this.positionProvider.clear();
         DebugSupport debugSupport = role.getDebugSupport();
         if (debugSupport.isTraceSensorFails()) {
            debugSupport.setLastFailingSensor(this.sensor);
         }

         return false;
      }
   }

   public InfoProvider getSensorInfo() {
      return this.positionProvider;
   }

   public void registerWithSupport(Role role) {
      this.sensor.registerWithSupport(role);
   }

   public void motionControllerChanged(@Nullable Ref<EntityStore> ref, @Nonnull NPCEntity npcComponent, MotionController motionController, @Nullable ComponentAccessor<EntityStore> componentAccessor) {
      this.sensor.motionControllerChanged(ref, npcComponent, motionController, componentAccessor);
   }

   public void loaded(Role role) {
      this.sensor.loaded(role);
   }

   public void spawned(Role role) {
      this.sensor.spawned(role);
   }

   public void unloaded(Role role) {
      this.sensor.unloaded(role);
   }

   public void removed(Role role) {
      this.sensor.removed(role);
   }

   public void teleported(Role role, World from, World to) {
      this.sensor.teleported(role, from, to);
   }

   public void done() {
      this.sensor.done();
   }

   public int componentCount() {
      return 1;
   }

   public IAnnotatedComponent getComponent(int index) {
      if (index >= this.componentCount()) {
         throw new IndexOutOfBoundsException();
      } else {
         return this.sensor;
      }
   }

   public void setContext(IAnnotatedComponent parent, int index) {
      super.setContext(parent, index);
      this.sensor.setContext(this, index);
   }
}