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

class="kw">import com.google.gson.JsonElement;
class="kw">import com.hypixel.hytale.server.npc.asset.builder.Builder;
class="kw">import com.hypixel.hytale.server.npc.asset.builder.BuilderDescriptorState;
class="kw">import com.hypixel.hytale.server.npc.asset.builder.BuilderSupport;
class="kw">import com.hypixel.hytale.server.npc.asset.builder.InstructionType;
class="kw">import com.hypixel.hytale.server.npc.asset.builder.holder.EnumSetHolder;
class="kw">import com.hypixel.hytale.server.npc.corecomponents.builders.BuilderSensorBase;
class="kw">import com.hypixel.hytale.server.npc.corecomponents.combat.BodyMotionCharge;
class="kw">import com.hypixel.hytale.server.npc.corecomponents.combat.SensorChargeState;
class="kw">import com.hypixel.hytale.server.npc.instructions.Sensor;
class="kw">import java.util.EnumSet;
class="kw">import javax.annotation.Nonnull;

class="kw">public class BuilderSensorChargeState class="kw">extends BuilderSensorBase {
   class="kw">protected class="kw">final EnumSetHolder<BodyMotionCharge.ChargeState> states = new EnumSetHolder<>();

   class="kw">public BuilderSensorChargeState() {
   }

   @Nonnull
   class="kw">public Sensor build(@Nonnull BuilderSupport var1) {
      class="kw">return new SensorChargeState(this, var1);
   }

   @Nonnull
   @Override
   class="kw">public String getShortDescription() {
      class="kw">return "Test if a BodyMotion Charge is in one of the given states.";
   }

   @Nonnull
   @Override
   class="kw">public String getLongDescription() {
      class="kw">return this.getShortDescription();
   }

   @Nonnull
   @Override
   class="kw">public BuilderDescriptorState getBuilderDescriptorState() {
      class="kw">return BuilderDescriptorState.Stable;
   }

   @Nonnull
   @Override
   class="kw">public Builder<Sensor> readConfig(@Nonnull JsonElement var1) {
      this.requireEnumSet(
         var1, "States", this.states, BodyMotionCharge.ChargeState.class, BuilderDescriptorState.Stable, "The charge states that trigger this sensor", null
      );
      this.requirePreceding(
         BodyMotionCharge.class, BuilderDescriptorState.Stable, "Requires a preceding BodyMotionCharge in the same instruction list scope", null
      );
      this.requireInstructionType(InstructionType.NPCOnlyInstructions);
      class="kw">return this;
   }

   @Nonnull
   class="kw">public EnumSet<BodyMotionCharge.ChargeState> getStates(@Nonnull BuilderSupport var1) {
      class="kw">return this.states.get(var1.getExecutionContext());
   }
}