Hbm-s-Nuclear-Tech-GIT/src/main/java/com/hbm/entity/mob/ai/EntityAIStopFlying.java

package com.hbm.entity.mob.ai;

import com.hbm.entity.mob.IFlyingCreature;

import net.minecraft.entity.EntityLivingBase;
import net.minecraft.entity.ai.EntityAIBase;

public class EntityAIStopFlying extends EntityAIBase {
	
	private EntityLivingBase living;
	private IFlyingCreature flying;
	
	public EntityAIStopFlying(EntityLivingBase living, IFlyingCreature flying) {
		this.living = living;
		this.flying = flying;
	}

	@Override
	public boolean shouldExecute() {
		return this.flying.getFlyingState() == this.flying.STATE_FLYING && this.living.getRNG().nextInt(200) == 0;
	}

	@Override
	public void startExecuting() {
		this.flying.setFlyingState(this.flying.STATE_WALKING);
	}
}