class="kw">package com.hypixel.hytale.server.flock.decisionmaker.conditions;

class="kw">import com.hypixel.hytale.codec.builder.BuilderCodec;
class="kw">import com.hypixel.hytale.component.ArchetypeChunk;
class="kw">import com.hypixel.hytale.component.CommandBuffer;
class="kw">import com.hypixel.hytale.component.Ref;
class="kw">import com.hypixel.hytale.server.core.entity.group.EntityGroup;
class="kw">import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
class="kw">import com.hypixel.hytale.server.flock.FlockMembership;
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 FlockSizeCondition class="kw">extends ScaledCurveCondition {
   class="kw">public class="kw">static class="kw">final BuilderCodec<FlockSizeCondition> CODEC = BuilderCodec.builder(
         FlockSizeCondition.class, FlockSizeCondition::new, ScaledCurveCondition.ABSTRACT_CODEC
      )
      .build();

   class="kw">protected FlockSizeCondition() {
   }

   @Override
   class="kw">protected double getInput(
      int var1, @Nonnull ArchetypeChunk<EntityStore> var2, Ref<EntityStore> var3, @Nonnull CommandBuffer<EntityStore> var4, EvaluationContext var5
   ) {
      FlockMembership var6 = var2.getComponent(var1, FlockMembership.getComponentType());
      if (var6 == null) {
         class="kw">return 1.0;
      }

      Ref var7 = var6.getFlockRef();
      class="kw">return var7 != null && var7.isValid() ? var4.getComponent(var7, EntityGroup.getComponentType()).size() : 1.0;
   }

   @Nonnull
   @Override
   class="kw">public String toString() {
      class="kw">return "FlockSizeCondition{} " + super.toString();
   }
}