Hbm-s-Nuclear-Tech-GIT/src/main/java/api/hbm/fluid/IFluidUser.java

package api.hbm.fluid;

import com.hbm.inventory.fluid.FluidType;
import com.hbm.inventory.fluid.tank.FluidTank;
import com.hbm.packet.AuxParticlePacketNT;
import com.hbm.packet.PacketDispatcher;

import cpw.mods.fml.common.network.NetworkRegistry.TargetPoint;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.tileentity.TileEntity;
import net.minecraft.world.World;
import net.minecraftforge.common.util.ForgeDirection;

public interface IFluidUser extends IFluidConnector {
	
	public default void sendFluid(FluidTank tank, World world, int x, int y, int z, ForgeDirection dir) {
		sendFluid(tank.getTankType(), tank.getPressure(), world, x, y, z, dir);
	}
	
	public default void sendFluid(FluidType type, int pressure, World world, int x, int y, int z, ForgeDirection dir) {
		
		TileEntity te = world.getTileEntity(x, y, z);
		boolean wasSubscribed = false;
		boolean red = false;
		
		if(te instanceof IFluidConductor) {
			IFluidConductor con = (IFluidConductor) te;
			
			if(con.getPipeNet(type) != null && con.getPipeNet(type).isSubscribed(this)) {
				con.getPipeNet(type).unsubscribe(this);
				wasSubscribed = true;
			}
		}
		
		if(te instanceof IFluidConnector) {
			IFluidConnector con = (IFluidConnector) te;
			
			if(con.canConnect(type, dir.getOpposite())) {
				long toSend = this.getTotalFluidForSend(type, pressure);
				
				if(toSend > 0) {
					long transfer = toSend - con.transferFluid(type, pressure, toSend);
					this.removeFluidForTransfer(type, pressure, transfer);
				}
				red = true;
			}
		}
		
		if(wasSubscribed && te instanceof IFluidConductor) {
			IFluidConductor con = (IFluidConductor) te;
			
			if(con.getPipeNet(type) != null && !con.getPipeNet(type).isSubscribed(this)) {
				con.getPipeNet(type).subscribe(this);
			}
		}
		
		if(particleDebug) {
			NBTTagCompound data = new NBTTagCompound();
			data.setString("type", "network");
			data.setString("mode", "fluid");
			data.setInteger("color", type.getColor());
			double posX = x + 0.5 - dir.offsetX * 0.5 + world.rand.nextDouble() * 0.5 - 0.25;
			double posY = y + 0.5 - dir.offsetY * 0.5 + world.rand.nextDouble() * 0.5 - 0.25;
			double posZ = z + 0.5 - dir.offsetZ * 0.5 + world.rand.nextDouble() * 0.5 - 0.25;
			data.setDouble("mX", dir.offsetX * (red ? 0.025 : 0.1));
			data.setDouble("mY", dir.offsetY * (red ? 0.025 : 0.1));
			data.setDouble("mZ", dir.offsetZ * (red ? 0.025 : 0.1));
			PacketDispatcher.wrapper.sendToAllAround(new AuxParticlePacketNT(data, posX, posY, posZ), new TargetPoint(world.provider.dimensionId, posX, posY, posZ, 25));
		}
	}
	
	public static IPipeNet getPipeNet(World world, int x, int y, int z, FluidType type) {

		TileEntity te = world.getTileEntity(x, y, z);
		
		if(te instanceof IFluidConductor) {
			IFluidConductor con = (IFluidConductor) te;
			
			if(con.getPipeNet(type) != null) {
				return con.getPipeNet(type);
			}
		}
		
		return null;
	}
	
	/** Use more common conPos method instead */
	@Deprecated public default void sendFluidToAll(FluidTank tank, TileEntity te) {
		sendFluidToAll(tank.getTankType(), tank.getPressure(), te);
	}

	/** Use more common conPos method instead */
	@Deprecated public default void sendFluidToAll(FluidType type, int pressure, TileEntity te) {
		
		for(ForgeDirection dir : ForgeDirection.VALID_DIRECTIONS) {
			sendFluid(type, pressure, te.getWorldObj(), te.xCoord + dir.offsetX, te.yCoord + dir.offsetY, te.zCoord + dir.offsetZ, dir);
		}
	}

	public default long getTotalFluidForSend(FluidType type, int pressure) { return 0; }
	public default void removeFluidForTransfer(FluidType type, int pressure, long amount) { }
	
	public default void subscribeToAllAround(FluidType type, TileEntity te) {
		subscribeToAllAround(type, te.getWorldObj(), te.xCoord, te.yCoord, te.zCoord);
	}
	
	public default void subscribeToAllAround(FluidType type, World world, int x, int y, int z) {
		
		for(ForgeDirection dir : ForgeDirection.VALID_DIRECTIONS)
			this.trySubscribe(type, world, x + dir.offsetX, y + dir.offsetY, z + dir.offsetZ, dir);
	}
	
	public default void unsubscribeToAllAround(FluidType type, TileEntity te) {
		unsubscribeToAllAround(type, te.getWorldObj(), te.xCoord, te.yCoord, te.zCoord);
	}
	
	public default void unsubscribeToAllAround(FluidType type, World world, int x, int y, int z) {
		
		for(ForgeDirection dir : ForgeDirection.VALID_DIRECTIONS)
			this.tryUnsubscribe(type, world, x + dir.offsetX, y + dir.offsetY, z + dir.offsetZ);
	}
	
	/**
	 * Returns all internal tanks of this tile. Not used by the fluid network, it should only be used for display purposes or edge cases that can't be solved otherwise.
	 * The array is either composed of the original tank or outright the original tank array, so changes done to this array will extend to the IFluidUser.
	 * @return
	 */
	public FluidTank[] getAllTanks();
}