CustomNPC-Plus/src/main/java/noppes/npcs/entity/EntityNPCFlying.java

package noppes.npcs.entity;

import net.minecraft.entity.Entity;
import net.minecraft.entity.SharedMonsterAttributes;
import net.minecraft.entity.ai.EntityAISwimming;
import net.minecraft.util.MathHelper;
import net.minecraft.world.World;

public abstract class EntityNPCFlying extends EntityNPCInterface {

	public EntityNPCFlying(World world) {
		super(world);
	}

	@Override
	public boolean canFly(){
		return ai.movementType == 1;
	}

    @Override
    public void fall(float distance) {
    	if(!canFly())
    		super.fall(distance);
    }

    @Override
    protected void updateFallState(double p_180433_1_, boolean p_180433_3_) {
    	if(!canFly())
    		super.updateFallState(p_180433_1_, p_180433_3_);
    }

    public void onLivingUpdate()
    {
        super.onLivingUpdate();
        if(!canFly()){
            return;
        }

        if (!this.onGround && !this.getMoveHelper().isUpdating())
        {
            this.motionY *= 0.0D;
            this.velocityChanged = true;
        }
    }

    @Override
    public boolean isOnLadder(){
        return false;
    }




}