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

package noppes.npcs.entity;

import net.minecraft.init.Blocks;
import net.minecraft.world.World;

public abstract class EntityNPCFlying extends EntityNPCInterface {

    public boolean flyLimitAllow = false;

	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_);
    }

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

    public void moveEntityWithHeading(float p_70612_1_, float p_70612_2_)
    {
        if(!this.canFly() || this.hurtTime != 0) {
            super.moveEntityWithHeading(p_70612_1_, p_70612_2_);
            return;
        }

        double heightOffGround = this.posY - this.worldObj.getTopSolidOrLiquidBlock((int) this.posX, (int) this.posZ);
        if (heightOffGround > this.ai.flyHeightLimit && this.ai.hasFlyLimit || (heightOffGround < this.height && this.motionY == 0)) {
            this.flyLimitAllow = false;
            if (!this.getNavigator().noPath() && this.motionY > 0) {
                this.motionY = 0;
            } else {
                super.moveEntityWithHeading(p_70612_1_, p_70612_2_);
                return;
            }
        }
        this.flyLimitAllow = true;

        double d3 = this.motionY;
        super.moveEntityWithHeading(p_70612_1_, p_70612_2_);
        this.motionY = d3;

        if (this.getNavigator().noPath()) {
            this.motionY = -Math.abs(this.ai.flyGravity);
        }

        this.velocityChanged = true;
    }
}