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

package com.hypixel.hytale.builtin.npccombatactionevaluator.corecomponents;

import it.unimi.dsi.fastutil.ints.Int2FloatOpenHashMap;
import com.hypixel.hytale.component.Store;
import javax.annotation.Nullable;
import com.hypixel.hytale.server.npc.sensorinfo.InfoProvider;
import com.hypixel.hytale.server.npc.role.Role;
import com.hypixel.hytale.component.Ref;
import com.hypixel.hytale.server.npc.corecomponents.builders.BuilderActionBase;
import javax.annotation.Nonnull;
import com.hypixel.hytale.builtin.npccombatactionevaluator.corecomponents.builders.BuilderActionAddToTargetMemory;
import com.hypixel.hytale.builtin.npccombatactionevaluator.memory.TargetMemory;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import com.hypixel.hytale.component.ComponentType;
import com.hypixel.hytale.server.npc.corecomponents.ActionBase;

public class ActionAddToTargetMemory extends ActionBase
{
    private static final ComponentType<EntityStore, TargetMemory> TARGET_MEMORY;
    
    public ActionAddToTargetMemory(@Nonnull final BuilderActionAddToTargetMemory builder) {
        super(builder);
    }
    
    @Override
    public boolean canExecute(@Nonnull final Ref<EntityStore> ref, @Nonnull final Role role, @Nullable final InfoProvider sensorInfo, final double dt, @Nonnull final Store<EntityStore> store) {
        return super.canExecute(ref, role, sensorInfo, dt, store) && sensorInfo != null && sensorInfo.hasPosition();
    }
    
    @Override
    public boolean execute(@Nonnull final Ref<EntityStore> ref, @Nonnull final Role role, @Nonnull final InfoProvider sensorInfo, final double dt, @Nonnull final Store<EntityStore> store) {
        super.execute(ref, role, sensorInfo, dt, store);
        final TargetMemory targetMemory = ref.getStore().getComponent(ref, ActionAddToTargetMemory.TARGET_MEMORY);
        if (targetMemory == null) {
            return true;
        }
        final Ref<EntityStore> target = sensorInfo.getPositionProvider().getTarget();
        final Int2FloatOpenHashMap hostiles = targetMemory.getKnownHostiles();
        if (hostiles.put(target.getIndex(), targetMemory.getRememberFor()) <= 0.0f) {
            targetMemory.getKnownHostilesList().add(target);
        }
        return true;
    }
    
    static {
        TARGET_MEMORY = TargetMemory.getComponentType();
    }
}