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

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

import com.hypixel.hytale.math.vector.Vector3d;
import com.hypixel.hytale.server.npc.decisionmaker.core.EvaluationContext;
import com.hypixel.hytale.component.CommandBuffer;
import javax.annotation.Nullable;
import com.hypixel.hytale.component.Ref;
import com.hypixel.hytale.component.ArchetypeChunk;
import javax.annotation.Nonnull;
import com.hypixel.hytale.server.core.modules.entity.component.TransformComponent;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import com.hypixel.hytale.component.ComponentType;
import com.hypixel.hytale.codec.builder.BuilderCodec;
import com.hypixel.hytale.server.npc.decisionmaker.core.conditions.base.ScaledCurveCondition;

public class TargetDistanceCondition extends ScaledCurveCondition
{
    public static final BuilderCodec<TargetDistanceCondition> CODEC;
    @Nonnull
    private static final ComponentType<EntityStore, TransformComponent> TRANSFORM_COMPONENT_TYPE;
    
    @Override
    protected double getInput(final int selfIndex, @Nonnull final ArchetypeChunk<EntityStore> archetypeChunk, @Nullable final Ref<EntityStore> target, @Nonnull final CommandBuffer<EntityStore> commandBuffer, final EvaluationContext context) {
        if (target == null || !target.isValid()) {
            return Double.MAX_VALUE;
        }
        final TransformComponent selfTransformComponent = archetypeChunk.getComponent(selfIndex, TargetDistanceCondition.TRANSFORM_COMPONENT_TYPE);
        assert selfTransformComponent != null;
        final Vector3d selfPos = selfTransformComponent.getPosition();
        final TransformComponent targetTransformComponent = commandBuffer.getComponent(target, TargetDistanceCondition.TRANSFORM_COMPONENT_TYPE);
        assert targetTransformComponent != null;
        final Vector3d targetPos = targetTransformComponent.getPosition();
        return selfPos.distanceTo(targetPos);
    }
    
    static {
        CODEC = BuilderCodec.builder(TargetDistanceCondition.class, TargetDistanceCondition::new, ScaledCurveCondition.ABSTRACT_CODEC).documentation("A scaled curve condition that returns a utility value based on the distance between the NPC and the target.").build();
        TRANSFORM_COMPONENT_TYPE = TransformComponent.getComponentType();
    }
}