CustomNPC-Plus/src/main/java/noppes/npcs/ai/EntityAIWaterNav.java

package noppes.npcs.ai;

import net.minecraft.entity.ai.EntityAIBase;
import noppes.npcs.entity.EntityNPCInterface;

public class EntityAIWaterNav extends EntityAIBase
{
    private EntityNPCInterface theEntity;

    public EntityAIWaterNav(EntityNPCInterface par1EntityNPCInterface)
    {
        this.theEntity = par1EntityNPCInterface;
        par1EntityNPCInterface.getNavigator().setCanSwim(true);
    }

    /**
     * Returns whether the EntityAIBase should begin execution.
     */
    public boolean shouldExecute()
    {
        if (this.theEntity.isInWater() || this.theEntity.handleLavaMovement())
        {
        	if (this.theEntity.ai.canSwim)
        	{
        		return true;
        	}
        	else if (this.theEntity.isCollidedHorizontally)
        	{
        		return true;
        	}
        		
        }
        return false;
    }

    /**
     * Updates the task
     */
    public void updateTask()
    {
        if (this.theEntity.getRNG().nextFloat() < 0.8F)
        {
            this.theEntity.getJumpHelper().setJumping();
        }
    }
}