AIRobotUnload.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.robots.boards;

import net.minecraft.inventory.IInventory;
import net.minecraft.item.ItemStack;
import net.minecraft.tileentity.TileEntity;

import net.minecraftforge.common.util.ForgeDirection;

import buildcraft.api.robots.AIRobot;
import buildcraft.api.robots.EntityRobotBase;
import buildcraft.core.inventory.ITransactor;
import buildcraft.core.inventory.Transactor;
import buildcraft.core.inventory.filters.ArrayStackFilter;
import buildcraft.core.robots.DockingStation;
import buildcraft.silicon.statements.StateStationRequestItems;
import buildcraft.transport.Pipe;

public class AIRobotUnload extends AIRobot {

	public AIRobotUnload(EntityRobotBase iRobot) {
		super(iRobot, 0, 1);
	}

	@Override
	public void start() {
		DockingStation station = (DockingStation) robot.getDockingStation();

		if (station == null) {
			return;
		}

		Pipe pipe = station.pipe.pipe;

		for (int i = 0; i < robot.getSizeInventory(); ++i) {
			boolean found = false;
			ItemStack stackToAdd = robot.getStackInSlot(i);

			if (stackToAdd == null) {
				continue;
			}

			for (Object s : pipe.getActionStates()) {
				if (s instanceof StateStationRequestItems) {
					if (((StateStationRequestItems) s).matches(new ArrayStackFilter(stackToAdd))) {
						found = true;
						break;
					}
				}
			}

			if (!found) {
				// This stack is not accepted by any of the action states
				// currently active on this pipe - look for another one
				continue;
			}

			for (ForgeDirection dir : ForgeDirection.VALID_DIRECTIONS) {
				TileEntity nearbyTile = robot.worldObj.getTileEntity(station.x() + dir.offsetX, station.y()
						+ dir.offsetY, station.z()
						+ dir.offsetZ);

				if (nearbyTile != null && nearbyTile instanceof IInventory) {
					ITransactor trans = Transactor.getTransactorFor(nearbyTile);

					if (stackToAdd != null) {
						ItemStack added = trans.add(stackToAdd, dir, true);
						robot.decrStackSize(i, added.stackSize);
					}
				}
			}
		}

		terminate();
	}
}