Citizens2/v1_18_R2/src/main/java/net/citizensnpcs/nms/v1_18_R2/util/EntityMoveControl.java

package net.citizensnpcs.nms.v1_18_R2.util;

import java.util.Random;

import net.citizensnpcs.util.NMS;
import net.minecraft.core.BlockPos;
import net.minecraft.core.Direction.Axis;
import net.minecraft.tags.BlockTags;
import net.minecraft.util.Mth;
import net.minecraft.world.entity.EntityType;
import net.minecraft.world.entity.LivingEntity;
import net.minecraft.world.entity.Mob;
import net.minecraft.world.entity.ai.attributes.Attributes;
import net.minecraft.world.entity.ai.control.MoveControl;
import net.minecraft.world.entity.monster.Slime;
import net.minecraft.world.level.block.state.BlockState;
import net.minecraft.world.phys.shapes.VoxelShape;

public class EntityMoveControl extends MoveControl {
    protected LivingEntity entity;
    private int jumpTicks;
    protected boolean moving;
    protected double speedMod;
    protected double tx;
    protected double ty;
    protected double tz;

    public EntityMoveControl(LivingEntity entityinsentient) {
        super(entityinsentient instanceof Mob ? (Mob) entityinsentient
                : new Slime(EntityType.SLIME, entityinsentient.level));
        this.entity = entityinsentient;
        this.tx = entityinsentient.getX();
        this.ty = entityinsentient.getY();
        this.tz = entityinsentient.getZ();
    }

    @Override
    public double getSpeedModifier() {
        return this.speedMod;
    }

    @Override
    public double getWantedX() {
        return this.tx;
    }

    @Override
    public double getWantedY() {
        return this.ty;
    }

    @Override
    public double getWantedZ() {
        return this.tz;
    }

    @Override
    public boolean hasWanted() {
        return this.moving;
    }

    protected int jumpTicks() {
        return new Random().nextInt(20) + 10;
    }

    @Override
    protected float rotlerp(float f, float f1, float f2) {
        float f3 = Mth.wrapDegrees(f1 - f);
        if (f3 > f2) {
            f3 = f2;
        }
        if (f3 < -f2) {
            f3 = -f2;
        }
        float f4 = f + f3;
        if (f4 < 0.0F) {
            f4 += 360.0F;
        } else if (f4 > 360.0F) {
            f4 -= 360.0F;
        }
        return f4;
    }

    @Override
    public void setWantedPosition(double d0, double d1, double d2, double d3) {
        this.tx = d0;
        this.ty = d1;
        this.tz = d2;
        this.speedMod = d3;
        this.moving = true;
    }

    @Override
    public void tick() {
        this.entity.zza = 0;
        if (this.moving) {
            this.moving = false;
            double dX = this.tx - this.entity.getX();
            double dZ = this.tz - this.entity.getZ();
            double dY = this.ty - this.entity.getY();
            double dXZ = Math.sqrt(dX * dX + dZ * dZ);
            double dXYZ = Math.sqrt(dX * dX + dY * dY + dZ * dZ);
            if (dXYZ < 2.500000277905201E-7)
                // this.entity.zza = 0.0F;
                return;
            if (dXZ > 0.4) {
                float f = (float) Math.toDegrees(Mth.atan2(dZ, dX)) - 90.0F;
                entity.setYRot(rotlerp(this.entity.getYRot(), f, 90.0F));
                NMS.setHeadYaw(entity.getBukkitEntity(), entity.getYRot());
            }
            this.entity.zza = (float) (this.speedMod * entity.getAttribute(Attributes.MOVEMENT_SPEED).getBaseValue());
            if (entity instanceof Slime && jumpTicks-- <= 0) {
                this.jumpTicks = new Random().nextInt(20) + 10;
                if (((Slime) entity).isAggressive()) {
                    this.jumpTicks /= 3;
                }
                ((Slime) entity).getJumpControl().jump();
                return;
            }
            BlockPos pos = entity.blockPosition();
            BlockState bs = entity.level.getBlockState(pos);
            VoxelShape vs = bs.getCollisionShape(entity.level, pos);
            if (dY >= entity.maxUpStep && dXZ < Math.max(1.0F, entity.getBbWidth())
                    || !vs.isEmpty() && entity.getY() < vs.max(Axis.Y) + pos.getY() && !bs.is(BlockTags.DOORS)
                            && !bs.is(BlockTags.FENCES)) {
                NMS.setShouldJump(entity.getBukkitEntity());
            }
        }
    }
}