    001package net.minecraft.entity.ai;
    002
    003import net.minecraft.entity.EntityLiving;
    004
    005public class EntityAISwimming extends EntityAIBase
    006{
    007    private EntityLiving theEntity;
    008
    009    public EntityAISwimming(EntityLiving par1EntityLiving)
    010    {
    011        this.theEntity = par1EntityLiving;
    012        this.setMutexBits(4);
    013        par1EntityLiving.getNavigator().setCanSwim(true);
    014    }
    015
    016    /**
    017     * Returns whether the EntityAIBase should begin execution.
    018     */
    019    public boolean shouldExecute()
    020    {
    021        return this.theEntity.isInWater() || this.theEntity.handleLavaMovement();
    022    }
    023
    024    /**
    025     * Updates the task
    026     */
    027    public void updateTask()
    028    {
    029        if (this.theEntity.getRNG().nextFloat() < 0.8F)
    030        {
    031            this.theEntity.getJumpHelper().setJumping();
    032        }
    033    }
    034}