// 
// Decompiled by Procyon v0.6.0
// 

package com.hypixel.hytale.server.flock.decisionmaker.conditions;

import com.hypixel.hytale.server.core.entity.group.EntityGroup;
import com.hypixel.hytale.server.flock.FlockMembership;
import com.hypixel.hytale.server.npc.entities.NPCEntity;
import com.hypixel.hytale.server.npc.decisionmaker.core.EvaluationContext;
import com.hypixel.hytale.component.CommandBuffer;
import com.hypixel.hytale.component.Ref;
import javax.annotation.Nonnull;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import com.hypixel.hytale.component.ArchetypeChunk;
import com.hypixel.hytale.codec.builder.BuilderCodec;
import com.hypixel.hytale.server.npc.decisionmaker.core.conditions.base.ScaledCurveCondition;

public class FlockSizeCondition extends ScaledCurveCondition
{
    public static final BuilderCodec<FlockSizeCondition> CODEC;
    
    protected FlockSizeCondition() {
    }
    
    @Override
    protected double getInput(final int selfIndex, @Nonnull final ArchetypeChunk<EntityStore> archetypeChunk, final Ref<EntityStore> target, @Nonnull final CommandBuffer<EntityStore> commandBuffer, final EvaluationContext context) {
        final NPCEntity self = archetypeChunk.getComponent(selfIndex, NPCEntity.getComponentType());
        final FlockMembership membership = archetypeChunk.getComponent(selfIndex, FlockMembership.getComponentType());
        if (membership == null) {
            return 1.0;
        }
        final Ref<EntityStore> flockReference = membership.getFlockRef();
        if (flockReference == null || !flockReference.isValid()) {
            return 1.0;
        }
        return commandBuffer.getComponent(flockReference, EntityGroup.getComponentType()).size();
    }
    
    @Nonnull
    @Override
    public String toString() {
        return "FlockSizeCondition{} " + super.toString();
    }
    
    static {
        CODEC = BuilderCodec.builder(FlockSizeCondition.class, FlockSizeCondition::new, ScaledCurveCondition.ABSTRACT_CODEC).build();
    }
}