Mekanism-tilera-Edition/common/mekanism/api/GasTransmission.java

package mekanism.api;

import java.util.HashSet;
import java.util.Set;

import net.minecraft.tileentity.TileEntity;
import net.minecraftforge.common.ForgeDirection;
import net.minecraftforge.common.MinecraftForge;

/**
 * A handy class containing several utilities for efficient gas transfer.
 * @author AidanBrady
 *
 */
public final class GasTransmission 
{
	/** Do not mess with this, used to only register NetworkLoader once. */
	private static boolean networkLoader_loaded = false;
	
	/**
	 * Call this method in your mod's init phase in order for the GasNetwork to function properly.
	 */
	public static void register()
	{
		if(!networkLoader_loaded)
		{
			MinecraftForge.EVENT_BUS.register(new GasNetwork.NetworkLoader());
			networkLoader_loaded = true;
			
			System.out.println("[Mekanism] Successfully loaded gas transmission NetworkLoader.");
		}
	}
	
    /**
     * Gets all the tubes around a tile entity.
     * @param tileEntity - center tile entity
     * @return array of TileEntities
     */
    public static TileEntity[] getConnectedTubes(TileEntity tileEntity)
    {
    	TileEntity[] tubes = new TileEntity[] {null, null, null, null, null, null};
    	
    	if(!(tileEntity instanceof IPressurizedTube) || ((IPressurizedTube)tileEntity).canTransferGas())
    	{
	    	for(ForgeDirection orientation : ForgeDirection.VALID_DIRECTIONS)
	    	{
				TileEntity tube = Object3D.get(tileEntity).getFromSide(orientation).getTileEntity(tileEntity.worldObj);
				
				if(tileEntity instanceof IPressurizedTube)
				{
				    if(((IPressurizedTube)tileEntity).canTransferGasToTube(tube))
				    {
	                    tubes[orientation.ordinal()] = tube;
				    }
				}
				else if(tube instanceof IPressurizedTube && ((IPressurizedTube)tube).canTransferGas())
				{
                    tubes[orientation.ordinal()] = tube;
				}
	    	}
    	}
    	
    	return tubes;
    }
    
    /**
     * Gets all the acceptors around a tile entity.
     * @param tileEntity - center tile entity
     * @return array of IGasAcceptors
     */
    public static IGasAcceptor[] getConnectedAcceptors(TileEntity tileEntity)
    {
    	IGasAcceptor[] acceptors = new IGasAcceptor[] {null, null, null, null, null, null};
    	
    	if(!(tileEntity instanceof IPressurizedTube) || ((IPressurizedTube)tileEntity).canTransferGas())
    	{
	    	for(ForgeDirection orientation : ForgeDirection.VALID_DIRECTIONS)
	    	{
				TileEntity acceptor = Object3D.get(tileEntity).getFromSide(orientation).getTileEntity(tileEntity.worldObj);
				
				if(acceptor instanceof IGasAcceptor)
				{
					acceptors[orientation.ordinal()] = (IGasAcceptor)acceptor;
				}
	    	}
    	}
    	
    	return acceptors;
    }
    
    /**
     * Gets all the tube connections around a tile entity.
     * @param tileEntity - center tile entity
     * @return array of ITubeConnections
     */
    public static ITubeConnection[] getConnections(TileEntity tileEntity)
    {
    	ITubeConnection[] connections = new ITubeConnection[] {null, null, null, null, null, null};
    	
    	if(!(tileEntity instanceof IPressurizedTube) || ((IPressurizedTube)tileEntity).canTransferGas())
    	{
    		for(ForgeDirection orientation : ForgeDirection.VALID_DIRECTIONS)
	    	{
				TileEntity connection = Object3D.get(tileEntity).getFromSide(orientation).getTileEntity(tileEntity.worldObj);
				
				if(connection instanceof ITubeConnection)
				{
					connections[orientation.ordinal()] = (ITubeConnection)connection;
				}
	    	}
    	}
    	
    	return connections;
    }
    
    /**
     * Emits a defined gas to the network.
     * @param type - gas type to send
     * @param amount - amount of gas to send
     * @param sender - the sender of the gas
     * @param facing - side the sender is outputting from
     * @return rejected gas
     */
    public static int emitGasToNetwork(EnumGas type, int amount, TileEntity sender, ForgeDirection facing)
    {
    	TileEntity pointer = Object3D.get(sender).getFromSide(facing).getTileEntity(sender.worldObj);
    	
    	if(pointer instanceof IPressurizedTube)
    	{
	    	return ((IPressurizedTube)pointer).getNetwork().emit(amount, type, sender);
    	}
    	
    	return amount;
    }
    
    /**
     * Emits gas from all sides of a TileEntity.
     * @param type - gas type to send
     * @param amount - amount of gas to send
     * @param pointer - sending TileEntity
     * @return rejected gas
     */
    public static int emitGasFromAllSides(EnumGas type, int amount, TileEntity pointer)
    {
    	if(pointer != null)
    	{
       		Set<GasNetwork> networks = new HashSet<GasNetwork>();
    		int totalRemaining = 0;
    		
    		for(ForgeDirection side : ForgeDirection.VALID_DIRECTIONS)
    		{
    			TileEntity sideTile = Object3D.get(pointer).getFromSide(side).getTileEntity(pointer.worldObj);
    			
    			if(sideTile instanceof IPressurizedTube)
    			{
    				networks.add(((IPressurizedTube)sideTile).getNetwork());
    			}
    		}
    		
    		int remaining = amount%networks.size();
    		int splitGas = (amount-remaining)/networks.size();
    		
    		for(GasNetwork network : networks)
    		{
    			totalRemaining += network.emit(splitGas+remaining, type, pointer);
    			remaining = 0;
    		}
    		
    		return totalRemaining;
    	}
    	
    	return amount;
    }
}