BiomesOPlenty/src/minecraft/biomesoplenty/helpers/EntitiesHelper.java

package biomesoplenty.helpers;

import net.minecraft.entity.Entity;
import net.minecraft.entity.EntityCreature;
import net.minecraft.entity.EntityLiving;
import net.minecraft.entity.passive.EntityAnimal;
import net.minecraft.entity.passive.EntityChicken;
import net.minecraft.entity.passive.EntityCow;
import net.minecraft.entity.passive.EntityPig;
import net.minecraft.entity.passive.EntitySheep;
import net.minecraft.util.AxisAlignedBB;
import net.minecraft.util.MathHelper;
import net.minecraftforge.event.Event.Result;
import net.minecraftforge.event.ForgeSubscribe;
import net.minecraftforge.event.entity.EntityJoinWorldEvent;
import net.minecraftforge.event.entity.living.LivingSpawnEvent.CheckSpawn;
import biomesoplenty.api.Blocks;
import biomesoplenty.api.Items;
import biomesoplenty.mobs.ai.EntityAITemptArmour;

public class EntitiesHelper
{
    @ForgeSubscribe
    public void onEntitySpawn(EntityJoinWorldEvent event)
    {
        Entity entity = event.entity;
        
        if (!(entity instanceof EntityLiving))
            return;
        
        if (entity instanceof EntityChicken)
            ((EntityLiving)entity).tasks.addTask(3, new EntityAITemptArmour((EntityCreature) entity, 0.25F, Items.flowerBand.get().itemID, 0, false));
        
        if (entity instanceof EntitySheep)
            ((EntityLiving)entity).tasks.addTask(3, new EntityAITemptArmour((EntityCreature) entity, 0.25F, Items.flowerBand.get().itemID, 1, false));
        
        if (entity instanceof EntityPig)
            ((EntityLiving)entity).tasks.addTask(4, new EntityAITemptArmour((EntityCreature) entity, 0.25F, Items.flowerBand.get().itemID, 2, false));
        
        if (entity instanceof EntityCow)
            ((EntityLiving)entity).tasks.addTask(3, new EntityAITemptArmour((EntityCreature) entity, 0.25F, Items.flowerBand.get().itemID, 3, false));
    }
    
    @ForgeSubscribe
    public void canEntitySpawn(CheckSpawn event)
    {
    	if (event.entityLiving instanceof EntityAnimal)
    	{
            int i = MathHelper.floor_double(event.entityLiving.posX);
            int j = MathHelper.floor_double(event.entityLiving.boundingBox.minY);
            int k = MathHelper.floor_double(event.entityLiving.posZ);
            
            if (event.entityLiving.getMaxSpawnedInChunk() <= event.world.getEntitiesWithinAABB(event.entityLiving.getClass(), AxisAlignedBB.getAABBPool().getAABB((double)(i - 16), (double)(j - 4), (double)(k - 16), (double)(i + 17), (double)(j + 5), (double)(k + 17))).size())
                return;
            
            if (event.entityLiving.worldObj.getBlockId(i, j - 1, k) == Blocks.holyGrass.get().blockID && event.entityLiving.worldObj.getFullBlockLightValue(i, j, k) > 8
                    && event.entityLiving.worldObj.checkNoEntityCollision(event.entity.boundingBox) && !event.entityLiving.worldObj.isAnyLiquid(event.entityLiving.boundingBox)
                    && event.entityLiving.worldObj.getCollidingBoundingBoxes(event.entityLiving, event.entityLiving.boundingBox).isEmpty())
            	event.setResult(Result.ALLOW);
    	}
    }
}