MagicBees/thaumicbees_src/thaumicbees/bees/AlleleEffectAuraNodeFlux.java

package thaumicbees.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 effectTimeout;
	private int nodeRange;
	
	AlleleEffectAuraNodeFlux(String id, boolean dominant, int timeout, int range)
	{
		super(id, dominant);
		this.effectTimeout = timeout;
		this.nodeRange = range;
	}

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

	@Override
	public IEffectData doEffect(IBeeGenome genome, IEffectData storedData, IBeeHousing housing)
	{
		int timeout = storedData.getInteger(0);
		if (timeout >= this.effectTimeout)
		{
			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);
			}
		}
		else
		{
			storedData.setInteger(0, timeout + 1);
		}
		return storedData;
	}

}