MagicBees/src/main/java/magicbees/world/feature/FeatureOreVein.java

package magicbees.world.feature;

import net.minecraft.block.Block;
import net.minecraft.world.World;

import java.util.Random;

public class FeatureOreVein
{
	private Block veinBlock;
	private int veinBlockMeta;
	private Block replacesBlock;
	
	public FeatureOreVein(Block block, Block replacesBlock)
	{
		this(block, 0, replacesBlock);
	}
	
	public FeatureOreVein(Block block, int meta, Block replacesBlock)
	{
		this.veinBlock = block;
		this.veinBlockMeta = meta;
		this.replacesBlock = replacesBlock;
	}
	
	public void generateVein(World world, Random random, int startX, int startY, int startZ, int maxSpawnCount)
	{
		int currentX = startX;
		int currentY = startY;
		int currentZ = startZ;
		int spawnAttempts = 0;
		int spawnCount = 0;
		
		while (spawnCount < maxSpawnCount && spawnAttempts < maxSpawnCount * 2)
		{
			++spawnAttempts;
			
			if (!world.isAirBlock(currentX, currentY, currentZ) && canBlockReplaceAt(world, currentX, currentY, currentZ, replacesBlock))
			{
				world.setBlock(currentX, currentY, currentZ, this.veinBlock, this.veinBlockMeta, 2);
				++spawnCount;
			}

			switch (random.nextInt(6))
			{
			case 0:
				++currentY;
				break;
			case 1:
				--currentY;
				break;
			case 2:
				++currentZ;
				break;
			case 3:
				--currentZ;
				break;
			case 4:
				++currentX;
				break;
			case 5:
				--currentX;
				break;
			}
		}
	}
	
	private static boolean canBlockReplaceAt(World world, int x, int y, int z, Block block)
	{
		return world.getBlock(x, y, z) != null &&
				world.getBlock(x, y, z).isReplaceableOreGen(world, x, y, z, block);
	}
}