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);
        this.getNavigator().setCanSwim(true);
        this.tasks.addTask(0, new EntityAISwimming(this));
	}

	// FLY CHANGE  [WIP]

	@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()
    {
        if(!canFly()){
            super.onLivingUpdate();
            return;
        }

        double speed = this.getEntityAttribute(SharedMonsterAttributes.movementSpeed).getAttributeValue();
        double speedMult = 0.6000000238418579D/0.5D;

        this.motionY *= speed*speedMult;

        if(isEntityAlive()) {
            if (!this.worldObj.isRemote) {
                Entity entity = this.getAttackTarget();
                if (entity != null && entity.isEntityAlive()) {
                    if (this.posY < entity.posY) {
                        if (this.motionY < 0.0D) {
                            this.motionY = 0.0D;
                        }

                        this.motionY += (speed - this.motionY) * speed * speedMult;
                    }

                    double d0 = entity.posX - this.posX;
                    double d1 = entity.posY - this.posY;
                    double d2 = entity.posZ - this.posZ;
                    double d3 = d0 * d0 + d2 * d2;

                    if (d3 > 9.0D) {
                        double d5 = (double) MathHelper.sqrt_double(d3);
                        this.motionX += (d0 / d5 * speed - this.motionX) * speed * speedMult;
                        this.motionZ += (d2 / d5 * speed - this.motionZ) * speed * speedMult;
                    }

                    this.rotationYaw = (float) ((Math.atan2(-d0, -d2) + Math.PI) * -(180F / Math.PI));
                }
            }

            if (this.motionX * this.motionX + this.motionZ * this.motionZ > 0.05000000074505806D && isEntityAlive()) {
                this.rotationYaw = (float) Math.atan2(this.motionZ, this.motionX) * (180F / (float) Math.PI) - 90.0F;
            }
        }

        super.onLivingUpdate();
    }

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




}