MagicBees/magicbees_src/magicbees/bees/AlleleEffectAuraNodeFlux.java

package magicbees.bees;

import net.minecraft.world.World;
import thaumcraft.api.EnumTag;
import thaumcraft.api.ObjectTags;
import thaumcraft.api.ThaumcraftApi;
import forestry.api.apiculture.IBeeGenome;
import forestry.api.apiculture.IBeeHousing;
import forestry.api.genetics.IEffectData;

public class AlleleEffectAuraNodeFlux extends AlleleEffect
{
	private int nodeRange;
	
	AlleleEffectAuraNodeFlux(String id, boolean dominant, int timeout, int range)
	{
		super(id, dominant, timeout);
		this.nodeRange = range;
	}

	@Override
	public IEffectData validateStorage(IEffectData storedData)
	{
		if (storedData == null)
		{
			storedData = new EffectData(1, 0, 0);
		}
		return storedData;
	}

	@Override
	public IEffectData doEffectThrottled(IBeeGenome genome, IEffectData storedData, IBeeHousing housing)
	{
		World world = housing.getWorld();
		int x = housing.getXCoord();
		int y = housing.getYCoord();
		int z = housing.getZCoord();
		
		EnumTag tag;
		do
		{
			tag = EnumTag.values()[world.rand.nextInt(EnumTag.values().length)];
		}
		while (tag == EnumTag.UNKNOWN || tag == EnumTag.WEATHER);
		
		int nodeId = ThaumcraftApi.getClosestAuraWithinRange(world, x, y, z, this.nodeRange);
		if (nodeId != -1)
		{
			int quantity = world.rand.nextInt(2) + 1;
			ThaumcraftApi.queueNodeChanges(nodeId, 0, 0, false, new ObjectTags().add(tag, quantity), 0f, 0f, 0f);
		}
		storedData.setInteger(0, 0);

		return storedData;
	}

}