BptBuilderTemplate.java

/**
 * Copyright (c) 2011-2014, SpaceToad and the BuildCraft Team
 * http://www.mod-buildcraft.com
 *
 * BuildCraft is distributed under the terms of the Minecraft Mod Public
 * License 1.0, or MMPL. Please check the contents of the license located in
 * http://www.mod-buildcraft.com/MMPL-1.0.txt
 */
package buildcraft.core.blueprints;

import java.util.LinkedList;

import net.minecraft.item.ItemStack;
import net.minecraft.world.World;
import net.minecraftforge.common.util.ForgeDirection;
import buildcraft.api.blueprints.SchematicBlockBase;
import buildcraft.api.core.BuildCraftAPI;
import buildcraft.builders.TileAbstractBuilder;
import buildcraft.core.BlockIndex;
import buildcraft.core.blueprints.BuildingSlotBlock.Mode;
import buildcraft.core.inventory.InventoryIterator;
import buildcraft.core.inventory.InventoryIterator.IInvSlot;
import buildcraft.core.utils.BlockUtil;

public class BptBuilderTemplate extends BptBuilderBase {

	private LinkedList<BuildingSlotBlock> buildList = new LinkedList<BuildingSlotBlock>();
	private BuildingSlotIterator iterator;

	public BptBuilderTemplate(BlueprintBase bluePrint, World world, int x, int y, int z) {
		super(bluePrint, world, x, y, z);
	}

	@Override
	protected void initialize () {
		if (blueprint.excavate) {
			for (int j = blueprint.sizeY - 1; j >= 0; --j) {
				for (int i = 0; i < blueprint.sizeX; ++i) {
					for (int k = 0; k < blueprint.sizeZ; ++k) {
						int xCoord = i + x - blueprint.anchorX;
						int yCoord = j + y - blueprint.anchorY;
						int zCoord = k + z - blueprint.anchorZ;

						SchematicBlockBase slot = blueprint.contents[i][j][k];

						if (slot == null
								&& !clearedLocations.contains(new BlockIndex(
										xCoord, yCoord, zCoord))) {
							BuildingSlotBlock b = new BuildingSlotBlock();

							b.schematic = null;
							b.x = xCoord;
							b.y = yCoord;
							b.z = zCoord;
							b.mode = Mode.ClearIfInvalid;
							b.buildStage = 0;

							buildList.add(b);
						}
					}
				}
			}
		}

		for (int j = 0; j < blueprint.sizeY; ++j) {
			for (int i = 0; i < blueprint.sizeX; ++i) {
				for (int k = 0; k < blueprint.sizeZ; ++k) {
					int xCoord = i + x - blueprint.anchorX;
					int yCoord = j + y - blueprint.anchorY;
					int zCoord = k + z - blueprint.anchorZ;

					SchematicBlockBase slot = blueprint.contents[i][j][k];

					if (slot != null && !builtLocations.contains(new BlockIndex(xCoord, yCoord, zCoord))) {
						BuildingSlotBlock b = new BuildingSlotBlock();

						b.schematic = slot;
						b.x = xCoord;
						b.y = yCoord;
						b.z = zCoord;

						b.mode = Mode.Build;
						b.buildStage = 1;

						buildList.add(b);
					}
				}
			}
		}

		iterator = new BuildingSlotIterator(buildList);
	}

	private void checkDone() {
		if (buildList.size() == 0) {
			done = true;
		} else {
			done = false;
		}
	}

	@Override
	public BuildingSlot getNextBlock(World world, TileAbstractBuilder inv) {
		if (buildList.size() != 0) {
			BuildingSlotBlock slot = internalGetNextBlock(world, inv, buildList);
			checkDone();

			if (slot != null) {
				return slot;
			}
		}

		checkDone();

		return null;
	}

	public BuildingSlotBlock internalGetNextBlock(World world, TileAbstractBuilder builder, LinkedList<BuildingSlotBlock> list) {
		BuildingSlotBlock result = null;

		IInvSlot firstSlotToConsume = null;

		for (IInvSlot invSlot : InventoryIterator.getIterable(builder, ForgeDirection.UNKNOWN)) {
			if (!builder.isBuildingMaterialSlot(invSlot.getIndex())) {
				continue;
			}

			ItemStack stack = invSlot.getStackInSlot();

			if (stack != null && stack.stackSize > 0) {
				firstSlotToConsume = invSlot;
				break;
			}
		}

		iterator.startIteration();

		while (iterator.hasNext()) {
			BuildingSlotBlock slot = iterator.next();

			if (slot.buildStage > buildList.getFirst().buildStage) {
				iterator.reset ();
				return null;
			}

			if (slot.mode == Mode.ClearIfInvalid) {
				if (BuildCraftAPI.isSoftBlock(world, slot.x, slot.y, slot.z)
						|| BlockUtil.isUnbreakableBlock(world, slot.x, slot.y, slot.z)) {
					iterator.remove();
					clearedLocations.add(new BlockIndex(slot.x, slot.y, slot.z));
				} else {
					if (setupForDestroy(builder, context, slot)) {
						result = slot;
						iterator.remove();
						clearedLocations.add(new BlockIndex(slot.x, slot.y, slot.z));

						break;
					}
				}
			} else if (slot.mode == Mode.Build) {
				if (!BuildCraftAPI.isSoftBlock(world, slot.x, slot.y, slot.z)) {
					iterator.remove();
					builtLocations.add(new BlockIndex(slot.x, slot.y, slot.z));
				} else {
					if (builder.energyAvailable() > TileAbstractBuilder.BUILD_ENERGY && firstSlotToConsume != null) {
						builder.consumeEnergy(TileAbstractBuilder.BUILD_ENERGY);

						slot.addStackConsumed(firstSlotToConsume
								.decreaseStackInSlot());
						result = slot;
						iterator.remove();
						builtLocations.add(new BlockIndex(slot.x, slot.y, slot.z));

						break;
					}
				}
			}
		}

		return result;
	}
}