cn.nukkit.entity.ai.sensor.EntityAttackedByOwnerSensor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of powernukkitx Show documentation

Make Nukkit great again!

There is a newer version: 1.20.40-r1

package cn.nukkit.entity.ai.sensor;

import cn.nukkit.entity.EntityIntelligent;
import cn.nukkit.entity.ai.memory.CoreMemoryTypes;
import cn.nukkit.entity.passive.EntityTamable;

public class EntityAttackedByOwnerSensor implements ISensor {
    protected int period;
    protected boolean changeTarget;

    public EntityAttackedByOwnerSensor(int period, boolean changeTarget) {
        this.period = period;
        this.changeTarget = changeTarget;
    }

    @Override
    public void sense(EntityIntelligent entity) {
        if (entity instanceof EntityTamable entityTamable) {
            var player = entityTamable.getOwner();
            if (player != null) {
                var current = entity.getMemoryStorage().get(CoreMemoryTypes.ENTITY_ATTACKED_BY_OWNER);
                if (!changeTarget) {
                    if (current != null && current.isAlive()) return;
                    else entity.getMemoryStorage().clear(CoreMemoryTypes.ENTITY_ATTACKED_BY_OWNER);
                }
                if (player.getLastBeAttackEntity() != null) {
                    entity.getMemoryStorage().put(CoreMemoryTypes.ENTITY_ATTACKING_OWNER, player.getLastBeAttackEntity());
                } else if (player.getLastAttackEntity() != null) {
                    entity.getMemoryStorage().put(CoreMemoryTypes.ENTITY_ATTACKED_BY_OWNER, player.getLastAttackEntity());
                } else entity.getMemoryStorage().clear(CoreMemoryTypes.ENTITY_ATTACKED_BY_OWNER);
            }
        }
    }

    @Override
    public int getPeriod() {
        return period;
    }
}