TotalSustainedDamageCondition.java

package com.hypixel.hytale.builtin.npccombatactionevaluator.conditions;

import com.hypixel.hytale.builtin.npccombatactionevaluator.memory.DamageMemory;
import com.hypixel.hytale.codec.builder.BuilderCodec;
import com.hypixel.hytale.component.ArchetypeChunk;
import com.hypixel.hytale.component.CommandBuffer;
import com.hypixel.hytale.component.ComponentType;
import com.hypixel.hytale.component.Holder;
import com.hypixel.hytale.component.Ref;
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 javax.annotation.Nonnull;

public class TotalSustainedDamageCondition extends ScaledCurveCondition {
   public static final BuilderCodec<TotalSustainedDamageCondition> CODEC;
   protected static final ComponentType<EntityStore, DamageMemory> DAMAGE_MEMORY_COMPONENT_TYPE;

   public TotalSustainedDamageCondition() {
   }

   protected double getInput(int selfIndex, @Nonnull ArchetypeChunk<EntityStore> archetypeChunk, Ref<EntityStore> target, CommandBuffer<EntityStore> commandBuffer, EvaluationContext context) {
      DamageMemory memory = (DamageMemory)archetypeChunk.getComponent(selfIndex, DAMAGE_MEMORY_COMPONENT_TYPE);
      return (double)memory.getTotalCombatDamage();
   }

   public void setupNPC(@Nonnull Holder<EntityStore> holder) {
      holder.ensureComponent(DAMAGE_MEMORY_COMPONENT_TYPE);
   }

   static {
      CODEC = ((BuilderCodec.Builder)BuilderCodec.builder(TotalSustainedDamageCondition.class, TotalSustainedDamageCondition::new, ScaledCurveCondition.ABSTRACT_CODEC).documentation("A scaled curve condition that returns a utility value based on total damage taken during this bout of combat.")).build();
      DAMAGE_MEMORY_COMPONENT_TYPE = DamageMemory.getComponentType();
   }
}