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

package com.hypixel.hytale.server.npc.corecomponents.entity;

import com.hypixel.hytale.math.vector.Vector3d;
import com.hypixel.hytale.server.npc.sensorinfo.InfoProvider;
import com.hypixel.hytale.component.ComponentAccessor;
import com.hypixel.hytale.component.Store;
import com.hypixel.hytale.server.npc.role.Role;
import com.hypixel.hytale.component.Ref;
import com.hypixel.hytale.server.npc.corecomponents.builders.BuilderSensorBase;
import com.hypixel.hytale.server.npc.corecomponents.ISensorEntityPrioritiser;
import com.hypixel.hytale.server.npc.asset.builder.ComponentContext;
import com.hypixel.hytale.server.npc.asset.builder.BuilderSupport;
import javax.annotation.Nonnull;
import com.hypixel.hytale.server.npc.corecomponents.entity.builders.BuilderSensorTarget;
import com.hypixel.hytale.server.npc.sensorinfo.EntityPositionProvider;
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.server.npc.corecomponents.SensorWithEntityFilters;

public class SensorTarget extends SensorWithEntityFilters
{
    protected static final ComponentType<EntityStore, TransformComponent> TRANSFORM_COMPONENT_TYPE;
    protected final int targetSlot;
    protected final boolean autoUnlockTarget;
    protected final double range;
    protected final EntityPositionProvider positionProvider;
    
    public SensorTarget(@Nonnull final BuilderSensorTarget builderSensorTarget, @Nonnull final BuilderSupport support) {
        super(builderSensorTarget, builderSensorTarget.getFilters(support, null, ComponentContext.SensorTarget));
        this.positionProvider = new EntityPositionProvider();
        this.targetSlot = builderSensorTarget.getTargetSlot(support);
        this.range = builderSensorTarget.getRange(support);
        this.autoUnlockTarget = builderSensorTarget.getAutoUnlockTarget(support);
    }
    
    @Override
    public boolean matches(@Nonnull final Ref<EntityStore> ref, @Nonnull final Role role, final double dt, @Nonnull final Store<EntityStore> store) {
        if (!super.matches(ref, role, dt, store)) {
            this.positionProvider.clear();
            return false;
        }
        final Ref<EntityStore> target = role.getMarkedEntitySupport().getMarkedEntityRef(this.targetSlot);
        if (target == null) {
            return false;
        }
        if (!this.fulfilsRequirements(ref, role, target, store)) {
            if (this.autoUnlockTarget) {
                this.positionProvider.clear();
                role.getMarkedEntitySupport().clearMarkedEntity(this.targetSlot);
            }
            return false;
        }
        return this.positionProvider.setTarget(target, store) != null;
    }
    
    @Override
    public InfoProvider getSensorInfo() {
        return this.positionProvider;
    }
    
    protected boolean fulfilsRequirements(@Nonnull final Ref<EntityStore> ref, @Nonnull final Role role, @Nonnull final Ref<EntityStore> target, @Nonnull final Store<EntityStore> store) {
        final TransformComponent transformComponent = store.getComponent(ref, SensorTarget.TRANSFORM_COMPONENT_TYPE);
        if (transformComponent == null) {
            return false;
        }
        final Vector3d position = transformComponent.getPosition();
        if (this.range != Double.MAX_VALUE) {
            final TransformComponent targetTransformComponent = store.getComponent(target, SensorTarget.TRANSFORM_COMPONENT_TYPE);
            if (targetTransformComponent == null) {
                return false;
            }
            final double squaredDistance = position.distanceSquaredTo(targetTransformComponent.getPosition());
            if (squaredDistance > this.range * this.range) {
                return false;
            }
        }
        return this.matchesFilters(ref, target, role, store);
    }
    
    static {
        TRANSFORM_COMPONENT_TYPE = TransformComponent.getComponentType();
    }
}