Preventing pipe feed back loop

This commit is contained in:
Calclavia 2014-01-25 20:37:25 +08:00
parent 676af2e409
commit dcc1a0caf5
3 changed files with 134 additions and 105 deletions

View file

@ -16,4 +16,6 @@ public interface IFluidConnector extends IConnector<IFluidNetwork>, IFluidHandle
public FluidTank getInternalTank(); public FluidTank getInternalTank();
public void onFluidChanged(); public void onFluidChanged();
public boolean canFlow();
} }

View file

@ -14,12 +14,15 @@ import resonantinduction.api.fluid.IFluidPipe;
import universalelectricity.api.vector.Vector3; import universalelectricity.api.vector.Vector3;
import calclavia.lib.utility.FluidUtility; import calclavia.lib.utility.FluidUtility;
/** The network for pipe fluid transfer. getNodes() is NOT used. /**
* The network for pipe fluid transfer. getNodes() is NOT used.
* *
* @author DarkGuardsman */ * @author DarkGuardsman
*/
public class PipeNetwork extends FluidNetwork public class PipeNetwork extends FluidNetwork
{ {
public HashMap<IFluidHandler, EnumSet<ForgeDirection>> connectionMap = new HashMap<IFluidHandler, EnumSet<ForgeDirection>>(); public HashMap<IFluidHandler, EnumSet<ForgeDirection>> sideMap = new HashMap<IFluidHandler, EnumSet<ForgeDirection>>();
public HashMap<IFluidHandler, IFluidConnector> connectionMap = new HashMap<IFluidHandler, IFluidConnector>();
public int maxFlowRate = 0; public int maxFlowRate = 0;
public int maxPressure = 0; public int maxPressure = 0;
@ -31,17 +34,26 @@ public class PipeNetwork extends FluidNetwork
* tank * tank
*/ */
FluidStack stack = this.getTank().getFluid().copy(); FluidStack stack = this.getTank().getFluid().copy();
int count = this.connectionMap.size(); int count = this.sideMap.size();
for (Entry<IFluidHandler, EnumSet<ForgeDirection>> entry : this.connectionMap.entrySet()) for (Entry<IFluidHandler, EnumSet<ForgeDirection>> entry : this.sideMap.entrySet())
{ {
int sideCount = entry.getValue().size(); int sideCount = entry.getValue().size();
for (ForgeDirection dir : entry.getValue()) for (ForgeDirection dir : entry.getValue())
{ {
int volPer = (stack.amount / count); int volPer = (stack.amount / count);
int volPerSide = (volPer / sideCount); int volPerSide = (volPer / sideCount);
IFluidHandler handler = entry.getKey();
stack.amount -= entry.getKey().fill(dir, FluidUtility.getStack(stack, Math.min(volPerSide, this.maxFlowRate)), true); /*
* Don't input to tanks from the sides where the pipe is extraction mode. This
* prevents feed-back loops.
*/
if (connectionMap.get(handler).canFlow())
{
stack.amount -= handler.fill(dir, FluidUtility.getStack(stack, Math.min(volPerSide, this.maxFlowRate)), true);
}
if (sideCount > 1) if (sideCount > 1)
--sideCount; --sideCount;
@ -67,7 +79,7 @@ public class PipeNetwork extends FluidNetwork
@Override @Override
public boolean canUpdate() public boolean canUpdate()
{ {
return getTank().getFluidAmount() > 0 && connectionMap.size() > 0 && getConnectors().size() > 0; return getTank().getFluidAmount() > 0 && sideMap.size() > 0 && getConnectors().size() > 0;
} }
@Override @Override
@ -79,35 +91,38 @@ public class PipeNetwork extends FluidNetwork
@Override @Override
public void reconstruct() public void reconstruct()
{ {
this.connectionMap.clear(); this.sideMap.clear();
this.maxFlowRate = Integer.MAX_VALUE; this.maxFlowRate = Integer.MAX_VALUE;
this.maxPressure = Integer.MAX_VALUE; this.maxPressure = Integer.MAX_VALUE;
super.reconstruct(); super.reconstruct();
} }
@Override @Override
public void reconstructConnector(IFluidConnector part) public void reconstructConnector(IFluidConnector connector)
{ {
super.reconstructConnector(part); super.reconstructConnector(connector);
if (part instanceof IFluidPipe)
{
if (((IFluidPipe) part).getMaxFlowRate() < this.maxFlowRate)
this.maxFlowRate = ((IFluidPipe) part).getMaxFlowRate();
if (((IFluidPipe) part).getMaxPressure() < this.maxPressure) if (connector instanceof IFluidPipe)
this.maxPressure = ((IFluidPipe) part).getMaxPressure(); {
if (((IFluidPipe) connector).getMaxFlowRate() < this.maxFlowRate)
this.maxFlowRate = ((IFluidPipe) connector).getMaxFlowRate();
if (((IFluidPipe) connector).getMaxPressure() < this.maxPressure)
this.maxPressure = ((IFluidPipe) connector).getMaxPressure();
} }
for (int i = 0; i < 6; i++) for (int i = 0; i < 6; i++)
{ {
if (part.getConnections()[i] instanceof IFluidHandler && !(part.getConnections()[i] instanceof IFluidPipe)) if (connector.getConnections()[i] instanceof IFluidHandler && !(connector.getConnections()[i] instanceof IFluidPipe))
{ {
EnumSet<ForgeDirection> set = this.connectionMap.get(part.getConnections()[i]); EnumSet<ForgeDirection> set = this.sideMap.get(connector.getConnections()[i]);
if (set == null) if (set == null)
{ {
set = EnumSet.noneOf(ForgeDirection.class); set = EnumSet.noneOf(ForgeDirection.class);
} }
set.add(ForgeDirection.getOrientation(i).getOpposite()); set.add(ForgeDirection.getOrientation(i).getOpposite());
this.connectionMap.put((IFluidHandler) part.getConnections()[i], set); sideMap.put((IFluidHandler) connector.getConnections()[i], set);
connectionMap.put((IFluidHandler) connector.getConnections()[i], connector);
} }
} }
} }

View file

@ -1,5 +1,6 @@
package resonantinduction.mechanical.fluid.pipe; package resonantinduction.mechanical.fluid.pipe;
import calclavia.lib.prefab.block.BlockAdvanced;
import net.minecraft.entity.player.EntityPlayer; import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.item.ItemStack; import net.minecraft.item.ItemStack;
import net.minecraft.nbt.NBTTagCompound; import net.minecraft.nbt.NBTTagCompound;
@ -70,10 +71,15 @@ public class PartPipe extends PartFramedConnection<EnumPipeMaterial, IFluidPipe,
@Override @Override
public boolean activate(EntityPlayer player, MovingObjectPosition part, ItemStack item) public boolean activate(EntityPlayer player, MovingObjectPosition part, ItemStack item)
{ {
if (!world().isRemote && player.isSneaking()) if (!world().isRemote)
{
if (BlockAdvanced.isUsableWrench(player, player.getCurrentEquippedItem(), x(), y(), z()))
{ {
isExtracting = !isExtracting; isExtracting = !isExtracting;
player.addChatMessage("Pipe extraction mode: " + isExtracting); player.addChatMessage("Pipe extraction mode: " + isExtracting);
BlockAdvanced.damageWrench(player, player.getCurrentEquippedItem(), x(), y(), z());
return true;
}
} }
return super.activate(player, part, item); return super.activate(player, part, item);
@ -213,4 +219,10 @@ public class PartPipe extends PartFramedConnection<EnumPipeMaterial, IFluidPipe,
isExtracting = nbt.getBoolean("isExtracting"); isExtracting = nbt.getBoolean("isExtracting");
} }
@Override
public boolean canFlow()
{
return !isExtracting;
}
} }