class="kw">package com.hypixel.hytale.builtin.npccombatactionevaluator.conditions;

class="kw">import com.hypixel.hytale.builtin.npccombatactionevaluator.memory.TargetMemory;
class="kw">import com.hypixel.hytale.codec.KeyedCodec;
class="kw">import com.hypixel.hytale.codec.builder.BuilderCodec;
class="kw">import com.hypixel.hytale.codec.codecs.EnumCodec;
class="kw">import com.hypixel.hytale.component.ArchetypeChunk;
class="kw">import com.hypixel.hytale.component.CommandBuffer;
class="kw">import com.hypixel.hytale.component.ComponentType;
class="kw">import com.hypixel.hytale.component.Ref;
class="kw">import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
class="kw">import com.hypixel.hytale.server.npc.decisionmaker.core.EvaluationContext;
class="kw">import com.hypixel.hytale.server.npc.decisionmaker.core.conditions.base.ScaledCurveCondition;
class="kw">import javax.annotation.Nonnull;

class="kw">public class TargetMemoryCountCondition class="kw">extends ScaledCurveCondition {
   @Nonnull
   class="kw">public class="kw">static class="kw">final EnumCodec<TargetMemoryCountCondition.TargetType> TARGET_TYPE_CODEC = new EnumCodec<>(TargetMemoryCountCondition.TargetType.class)
      .documentKey(TargetMemoryCountCondition.TargetType.All, "All known targets.")
      .documentKey(TargetMemoryCountCondition.TargetType.Friendly, "Known friendly targets.")
      .documentKey(TargetMemoryCountCondition.TargetType.Hostile, "Known hostile targets.");
   @Nonnull
   class="kw">public class="kw">static class="kw">final BuilderCodec<TargetMemoryCountCondition> CODEC = BuilderCodec.builder(
         TargetMemoryCountCondition.class, TargetMemoryCountCondition::new, ScaledCurveCondition.ABSTRACT_CODEC
      )
      .documentation("A scaled curve condition that returns a utility value based on the number of known targets in the memory.")
      .<TargetMemoryCountCondition.TargetType>appendInherited(
         new KeyedCodec<>("TargetType", TARGET_TYPE_CODEC),
         (var0, var1) -> var0.targetType = var1,
         var0 -> var0.targetType,
         (var0, var1) -> var0.targetType = var1.targetType
      )
      .documentation("The type of targets to count.")
      .add()
      .build();
   @Nonnull
   class="kw">protected class="kw">static class="kw">final ComponentType<EntityStore, TargetMemory> TARGET_MEMORY_COMPONENT_TYPE = TargetMemory.getComponentType();
   class="kw">protected TargetMemoryCountCondition.TargetType targetType = TargetMemoryCountCondition.TargetType.Hostile;

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

   @Override
   class="kw">protected double getInput(
      int var1, @Nonnull ArchetypeChunk<EntityStore> var2, Ref<EntityStore> var3, CommandBuffer<EntityStore> var4, EvaluationContext var5
   ) {
      TargetMemory var6 = var2.getComponent(var1, TARGET_MEMORY_COMPONENT_TYPE);
      assert var6 != null;

      class="kw">return class="kw">switch (this.targetType) {
         case Hostile -> var6.getKnownHostiles().size();
         case Friendly -> var6.getKnownFriendlies().size();
         case All -> var6.getKnownFriendlies().size() + var6.getKnownHostiles().size();
      };
   }

   class="kw">private enum TargetType {
      Hostile,
      Friendly,
      All;

      TargetType() {
      }
   }
}