SelfStatAbsoluteCondition.java

package com.hypixel.hytale.server.npc.decisionmaker.core.conditions;

import com.hypixel.hytale.codec.Codec;
import com.hypixel.hytale.codec.KeyedCodec;
import com.hypixel.hytale.codec.builder.BuilderCodec;
import com.hypixel.hytale.codec.validation.Validators;
import com.hypixel.hytale.component.ArchetypeChunk;
import com.hypixel.hytale.component.CommandBuffer;
import com.hypixel.hytale.component.Ref;
import com.hypixel.hytale.server.core.modules.entitystats.EntityStatMap;
import com.hypixel.hytale.server.core.modules.entitystats.EntityStatValue;
import com.hypixel.hytale.server.core.modules.entitystats.EntityStatsModule;
import com.hypixel.hytale.server.core.modules.entitystats.asset.EntityStatType;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import com.hypixel.hytale.server.npc.decisionmaker.core.EvaluationContext;
import com.hypixel.hytale.server.npc.decisionmaker.core.conditions.base.ScaledCurveCondition;
import java.util.Objects;
import javax.annotation.Nonnull;

public class SelfStatAbsoluteCondition extends ScaledCurveCondition {
   public static final BuilderCodec<SelfStatAbsoluteCondition> CODEC;
   protected String stat;
   protected int statIndex;

   protected SelfStatAbsoluteCondition() {
   }

   protected double getInput(int selfIndex, @Nonnull ArchetypeChunk<EntityStore> archetypeChunk, Ref<EntityStore> target, CommandBuffer<EntityStore> commandBuffer, EvaluationContext context) {
      EntityStatMap entityStatMapComponent = (EntityStatMap)archetypeChunk.getComponent(selfIndex, EntityStatsModule.get().getEntityStatMapComponentType());

      assert entityStatMapComponent != null;

      return (double)((EntityStatValue)Objects.requireNonNull(entityStatMapComponent.get(this.statIndex))).get();
   }

   @Nonnull
   public String toString() {
      String var10000 = this.stat;
      return "SelfStatAbsoluteCondition{stat='" + var10000 + "', statIndex=" + this.statIndex + "}" + super.toString();
   }

   static {
      CODEC = ((BuilderCodec.Builder)((BuilderCodec.Builder)((BuilderCodec.Builder)BuilderCodec.builder(SelfStatAbsoluteCondition.class, SelfStatAbsoluteCondition::new, ScaledCurveCondition.ABSTRACT_CODEC).documentation("A scaled curve condition that returns a utility value based on the absolute value of one of this NPC's stats.")).appendInherited(new KeyedCodec("Stat", Codec.STRING), (condition, s) -> condition.stat = s, (condition) -> condition.stat, (condition, parent) -> condition.stat = parent.stat).addValidator(Validators.nonNull()).addValidator(EntityStatType.VALIDATOR_CACHE.getValidator()).documentation("The stat to check.").add()).afterDecode((condition) -> condition.statIndex = EntityStatType.getAssetMap().getIndex(condition.stat))).build();
   }
}