MagicBees/magicbees_src/magicbees/bees/HiveDrop.java

package magicbees.bees;

import java.util.ArrayList;
import java.util.Collection;

import net.minecraft.item.ItemStack;
import net.minecraft.world.World;
import forestry.api.apiculture.EnumBeeType;
import forestry.api.apiculture.IBee;
import forestry.api.apiculture.IHiveDrop;
import forestry.api.genetics.IAllele;

public class HiveDrop implements IHiveDrop
{
	private IAllele[] template;
	private int chance;
	private ItemStack[] bonus;
	private float ignoblePercent = 0f;
	
	public HiveDrop(IAllele[] beeTemplate, ItemStack[] bonusDrops, int dropChance)
	{
		this.template = beeTemplate;
		this.bonus = bonusDrops;
		this.chance = dropChance;
	}
	
	public HiveDrop setIgnoblePercentage(float f)
	{
		this.ignoblePercent = f;
		
		return this;
	}
	
	private IBee getBee(World w)
	{
		IBee bee = BeeManager.beeRoot.getBee(w, BeeManager.beeRoot.templateAsGenome(this.template));
		if (w.rand.nextFloat() < this.ignoblePercent)
		{
			bee.setIsNatural(false);
		}
		
		return bee;
	}

	@Override
	public ItemStack getPrincess(World world, int x, int y, int z, int fortune)
	{
		return BeeManager.beeRoot.getMemberStack(getBee(world), EnumBeeType.PRINCESS.ordinal());
	}

	@Override
	public Collection<ItemStack> getDrones(World world, int x, int y, int z, int fortune)
	{
		ArrayList<ItemStack> value = new ArrayList<ItemStack>(1);
		value.add(BeeManager.beeRoot.getMemberStack(getBee(world), EnumBeeType.DRONE.ordinal()));
		return value;
	}

	@Override
	public Collection<ItemStack> getAdditional(World world, int x, int y, int z, int fortune)
	{
		ArrayList<ItemStack> value = new ArrayList<ItemStack>(bonus.length);
		
		for (int i = 0; i < bonus.length; ++i)
		{
			value.add(bonus[i].copy());
		}
		
		return value;
	}

	@Override
	public int getChance(World world, int x, int y, int z)
	{
		return this.chance;
	}
}