613 lines
14 KiB
Java
613 lines
14 KiB
Java
package appeng.me;
|
|
|
|
import java.util.Collections;
|
|
import java.util.Comparator;
|
|
import java.util.EnumSet;
|
|
import java.util.LinkedList;
|
|
import java.util.List;
|
|
import java.util.concurrent.Callable;
|
|
|
|
import net.minecraft.nbt.NBTTagCompound;
|
|
import net.minecraft.tileentity.TileEntity;
|
|
import net.minecraft.world.World;
|
|
import net.minecraftforge.common.util.ForgeDirection;
|
|
import appeng.api.exceptions.FailedConnection;
|
|
import appeng.api.networking.GridFlags;
|
|
import appeng.api.networking.GridNotification;
|
|
import appeng.api.networking.IGrid;
|
|
import appeng.api.networking.IGridBlock;
|
|
import appeng.api.networking.IGridCache;
|
|
import appeng.api.networking.IGridConnectionVisitor;
|
|
import appeng.api.networking.IGridConnection;
|
|
import appeng.api.networking.IGridHost;
|
|
import appeng.api.networking.IGridNode;
|
|
import appeng.api.networking.IGridVisitor;
|
|
import appeng.api.networking.energy.IEnergyGrid;
|
|
import appeng.api.networking.events.MENetworkChannelsChanged;
|
|
import appeng.api.networking.pathing.IPathingGrid;
|
|
import appeng.api.util.AEColor;
|
|
import appeng.api.util.DimensionalCoord;
|
|
import appeng.api.util.IReadOnlyCollection;
|
|
import appeng.core.WorldSettings;
|
|
import appeng.hooks.TickHandler;
|
|
import appeng.me.pathfinding.IPathItem;
|
|
import appeng.util.ReadOnlyCollection;
|
|
|
|
public class GridNode implements IGridNode, IPathItem
|
|
{
|
|
|
|
final static private MENetworkChannelsChanged event = new MENetworkChannelsChanged();
|
|
final static private int channelCount[] = new int[] { 0, 8, 32 };
|
|
|
|
final List<IGridConnection> Connections = new LinkedList();
|
|
GridStorage myStorage = null;
|
|
|
|
IGridBlock gridProxy;
|
|
Grid myGrid;
|
|
|
|
Object visitorIterationNumber = null;
|
|
|
|
// connection criteria
|
|
private int compressedData = 0;
|
|
|
|
@Override
|
|
public void updateState()
|
|
{
|
|
EnumSet<GridFlags> set = gridProxy.getFlags();
|
|
|
|
compressedData = set.contains( GridFlags.CANNOT_CARRY ) ? 0 : (set.contains( GridFlags.DENSE_CAPACITY ) ? 2 : 1);
|
|
|
|
compressedData = compressedData | (gridProxy.getGridColor().ordinal() << 3);
|
|
|
|
for (ForgeDirection dir : gridProxy.getConnectableSides())
|
|
compressedData = compressedData | (1 << (dir.ordinal() + 8));
|
|
|
|
FindConnections();
|
|
getInternalGrid();
|
|
}
|
|
|
|
public int getMaxChannels()
|
|
{
|
|
return channelCount[compressedData & 0x03];
|
|
}
|
|
|
|
public AEColor getColor()
|
|
{
|
|
return AEColor.values()[(compressedData >> 3) & 0x1F];
|
|
}
|
|
|
|
private boolean isValidDirection(ForgeDirection dir)
|
|
{
|
|
return (compressedData & (1 << (8 + dir.ordinal()))) > 0;
|
|
}
|
|
|
|
// old power draw, used to diff
|
|
public double previousDraw = 0.0;
|
|
|
|
private int channelData = 0;
|
|
|
|
public long lastSecurityKey = -1;
|
|
public int playerID = -1;
|
|
|
|
@Override
|
|
public void setPlayerID(int playerID)
|
|
{
|
|
if ( playerID >= 0 )
|
|
this.playerID = playerID;
|
|
}
|
|
|
|
public int usedChannels()
|
|
{
|
|
return channelData >> 8;
|
|
}
|
|
|
|
public GridNode(IGridBlock what) {
|
|
gridProxy = what;
|
|
}
|
|
|
|
@Override
|
|
public void loadFromNBT(String name, NBTTagCompound nodeData)
|
|
{
|
|
if ( myGrid == null )
|
|
{
|
|
NBTTagCompound node = nodeData.getCompoundTag( name );
|
|
playerID = node.getInteger( "p" );
|
|
lastSecurityKey = node.getLong( "k" );
|
|
setGridStorage( WorldSettings.getInstance().getGridStorage( node.getLong( "g" ) ) );
|
|
}
|
|
else
|
|
throw new RuntimeException( "Loading data after part of a grid, this is invalid." );
|
|
}
|
|
|
|
@Override
|
|
public void saveToNBT(String name, NBTTagCompound nodeData)
|
|
{
|
|
if ( myStorage != null )
|
|
{
|
|
NBTTagCompound node = new NBTTagCompound();
|
|
|
|
node.setInteger( "p", playerID );
|
|
node.setLong( "k", lastSecurityKey );
|
|
node.setLong( "g", myStorage.getID() );
|
|
|
|
nodeData.setTag( name, node );
|
|
}
|
|
else
|
|
nodeData.removeTag( name );
|
|
}
|
|
|
|
@Override
|
|
public IGridBlock getGridBlock()
|
|
{
|
|
return gridProxy;
|
|
}
|
|
|
|
@Override
|
|
public EnumSet<ForgeDirection> getConnectedSides()
|
|
{
|
|
EnumSet<ForgeDirection> set = EnumSet.noneOf( ForgeDirection.class );
|
|
for (IGridConnection gc : Connections)
|
|
set.add( gc.getDirection( this ) );
|
|
return set;
|
|
}
|
|
|
|
public Class<? extends IGridHost> getMachineClass()
|
|
{
|
|
return getMachine().getClass();
|
|
}
|
|
|
|
@Override
|
|
public IGridHost getMachine()
|
|
{
|
|
return gridProxy.getMachine();
|
|
}
|
|
|
|
@Override
|
|
public void beginVisit(IGridVisitor g)
|
|
{
|
|
Object tracker = new Object();
|
|
|
|
LinkedList<GridNode> nextRun = new LinkedList();
|
|
nextRun.add( this );
|
|
|
|
visitorIterationNumber = tracker;
|
|
|
|
if ( g instanceof IGridConnectionVisitor )
|
|
{
|
|
LinkedList<IGridConnection> nextConn = new LinkedList();
|
|
IGridConnectionVisitor gcv = (IGridConnectionVisitor) g;
|
|
|
|
while (!nextRun.isEmpty())
|
|
{
|
|
while (!nextConn.isEmpty())
|
|
gcv.visitConnection( nextConn.poll() );
|
|
|
|
LinkedList<GridNode> thisRun = nextRun;
|
|
nextRun = new LinkedList();
|
|
|
|
for (GridNode n : thisRun)
|
|
n.visitorConnection( tracker, g, nextRun, nextConn );
|
|
}
|
|
}
|
|
else
|
|
{
|
|
while (!nextRun.isEmpty())
|
|
{
|
|
LinkedList<GridNode> thisRun = nextRun;
|
|
nextRun = new LinkedList();
|
|
|
|
for (GridNode n : thisRun)
|
|
n.visitorNode( tracker, g, nextRun );
|
|
}
|
|
}
|
|
}
|
|
|
|
private void visitorConnection(Object tracker, IGridVisitor g, LinkedList<GridNode> nextRun, LinkedList<IGridConnection> nextConnections)
|
|
{
|
|
if ( g.visitNode( this ) )
|
|
{
|
|
for (IGridConnection gc : getConnections())
|
|
{
|
|
GridNode gn = (GridNode) gc.getOtherSide( this );
|
|
GridConnection gcc = (GridConnection) gc;
|
|
|
|
if ( gcc.visitorIterationNumber != tracker )
|
|
{
|
|
gcc.visitorIterationNumber = tracker;
|
|
nextConnections.add( gc );
|
|
}
|
|
|
|
if ( tracker == gn.visitorIterationNumber )
|
|
continue;
|
|
|
|
gn.visitorIterationNumber = tracker;
|
|
|
|
nextRun.add( gn );
|
|
}
|
|
}
|
|
}
|
|
|
|
private void visitorNode(Object tracker, IGridVisitor g, LinkedList<GridNode> nextRun)
|
|
{
|
|
if ( g.visitNode( this ) )
|
|
{
|
|
for (IGridConnection gc : getConnections())
|
|
{
|
|
GridNode gn = (GridNode) gc.getOtherSide( this );
|
|
|
|
if ( tracker == gn.visitorIterationNumber )
|
|
continue;
|
|
|
|
gn.visitorIterationNumber = tracker;
|
|
|
|
nextRun.add( gn );
|
|
}
|
|
}
|
|
}
|
|
|
|
public void FindConnections()
|
|
{
|
|
if ( !gridProxy.isWorldAccessible() )
|
|
return;
|
|
|
|
EnumSet<ForgeDirection> newSecurityConnections = EnumSet.noneOf( ForgeDirection.class );
|
|
|
|
DimensionalCoord dc = gridProxy.getLocation();
|
|
for (ForgeDirection f : ForgeDirection.VALID_DIRECTIONS)
|
|
{
|
|
IGridHost te = findGridHost( dc.getWorld(), dc.x + f.offsetX, dc.y + f.offsetY, dc.z + f.offsetZ );
|
|
if ( te != null )
|
|
{
|
|
GridNode node = (GridNode) te.getGridNode( f.getOpposite() );
|
|
if ( node == null )
|
|
continue;
|
|
|
|
boolean isValidConnection = this.canConnect( node, f ) && node.canConnect( this, f.getOpposite() );
|
|
|
|
IGridConnection con = null; // find the connection for this
|
|
// direction..
|
|
for (IGridConnection c : getConnections())
|
|
{
|
|
if ( c.getDirection( this ) == f )
|
|
{
|
|
con = c;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if ( con != null )
|
|
{
|
|
IGridNode os = (IGridNode) con.getOtherSide( this );
|
|
if ( os == node )
|
|
{
|
|
// if this connection is no longer valid, destroy it.
|
|
if ( !isValidConnection )
|
|
con.destroy();
|
|
}
|
|
else
|
|
{
|
|
con.destroy();
|
|
// throw new GridException( "invalid state found, encountered connection to phantom block." );
|
|
}
|
|
}
|
|
else if ( isValidConnection )
|
|
{
|
|
if ( node.lastSecurityKey != -1 )
|
|
newSecurityConnections.add( f );
|
|
else
|
|
{
|
|
// construct a new connection between these two nodes.
|
|
try
|
|
{
|
|
new GridConnection( node, this, f.getOpposite() );
|
|
}
|
|
catch (FailedConnection e)
|
|
{
|
|
TickHandler.instance.addCallable( node.getWorld(), new Callable() {
|
|
|
|
@Override
|
|
public Object call() throws Exception
|
|
{
|
|
getMachine().securityBreak();
|
|
return null;
|
|
}
|
|
|
|
} );
|
|
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
for (ForgeDirection f : newSecurityConnections)
|
|
{
|
|
IGridHost te = findGridHost( dc.getWorld(), dc.x + f.offsetX, dc.y + f.offsetY, dc.z + f.offsetZ );
|
|
if ( te != null )
|
|
{
|
|
GridNode node = (GridNode) te.getGridNode( f.getOpposite() );
|
|
if ( node == null )
|
|
continue;
|
|
|
|
// construct a new connection between these two nodes.
|
|
try
|
|
{
|
|
new GridConnection( node, this, f.getOpposite() );
|
|
}
|
|
catch (FailedConnection e)
|
|
{
|
|
TickHandler.instance.addCallable( node.getWorld(), new Callable() {
|
|
|
|
@Override
|
|
public Object call() throws Exception
|
|
{
|
|
getMachine().securityBreak();
|
|
return null;
|
|
}
|
|
|
|
} );
|
|
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
private IGridHost findGridHost(World world, int x, int y, int z)
|
|
{
|
|
if ( world.blockExists( x, y, z ) )
|
|
{
|
|
TileEntity te = world.getTileEntity( x, y, z );
|
|
if ( te instanceof IGridHost )
|
|
return (IGridHost) te;
|
|
}
|
|
return null;
|
|
}
|
|
|
|
public void addConnection(IGridConnection gridConnection)
|
|
{
|
|
Connections.add( gridConnection );
|
|
if ( gridConnection.hasDirection() )
|
|
gridProxy.onGridNotification( GridNotification.ConnectionsChanged );
|
|
|
|
final IGridNode gn = this;
|
|
|
|
Collections.sort( Connections, new Comparator<IGridConnection>() {
|
|
|
|
@Override
|
|
public int compare(IGridConnection o1, IGridConnection o2)
|
|
{
|
|
boolean preferredA = o1.getOtherSide( gn ).hasFlag( GridFlags.PREFERRED );
|
|
boolean preferredB = o2.getOtherSide( gn ).hasFlag( GridFlags.PREFERRED );
|
|
|
|
return preferredA == preferredB ? 0 : (preferredA ? -1 : 1);
|
|
}
|
|
|
|
} );
|
|
}
|
|
|
|
public void removeConnection(IGridConnection gridConnection)
|
|
{
|
|
Connections.remove( gridConnection );
|
|
if ( gridConnection.hasDirection() )
|
|
gridProxy.onGridNotification( GridNotification.ConnectionsChanged );
|
|
}
|
|
|
|
@Override
|
|
public IReadOnlyCollection<IGridConnection> getConnections()
|
|
{
|
|
return new ReadOnlyCollection<IGridConnection>( Connections );
|
|
}
|
|
|
|
public boolean hasConnection(IGridNode otherside)
|
|
{
|
|
for (IGridConnection gc : Connections)
|
|
{
|
|
if ( gc.a() == otherside || gc.b() == otherside )
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
public boolean canConnect(GridNode from, ForgeDirection dir)
|
|
{
|
|
if ( !isValidDirection( dir ) )
|
|
return false;
|
|
|
|
if ( !from.getColor().matches( getColor() ) )
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
@Override
|
|
public IGrid getGrid()
|
|
{
|
|
return myGrid;
|
|
}
|
|
|
|
public Grid getInternalGrid()
|
|
{
|
|
if ( myGrid == null )
|
|
myGrid = new Grid( this );
|
|
|
|
return myGrid;
|
|
}
|
|
|
|
public void setGrid(Grid grid)
|
|
{
|
|
if ( myGrid == grid )
|
|
return;
|
|
|
|
if ( myGrid != null )
|
|
{
|
|
myGrid.remove( this );
|
|
|
|
if ( myGrid.isEmpty() )
|
|
{
|
|
myGrid.saveState();
|
|
|
|
for (IGridCache c : grid.caches.values())
|
|
c.onJoin( myGrid.myStorage );
|
|
}
|
|
}
|
|
|
|
myGrid = grid;
|
|
myGrid.add( this );
|
|
}
|
|
|
|
public void validateGrid()
|
|
{
|
|
GridSplitDetector gsd = new GridSplitDetector( getInternalGrid().getPivot() );
|
|
beginVisit( gsd );
|
|
if ( !gsd.pivotFound )
|
|
{
|
|
GridPropagator gp = new GridPropagator( new Grid( this ) );
|
|
beginVisit( gp );
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void destroy()
|
|
{
|
|
while (!Connections.isEmpty())
|
|
{
|
|
// not part of this network for real anymore.
|
|
if ( Connections.size() == 1 )
|
|
setGridStorage( null );
|
|
|
|
IGridConnection c = Connections.listIterator().next();
|
|
GridNode otherSide = (GridNode) c.getOtherSide( this );
|
|
otherSide.getInternalGrid().pivot = otherSide;
|
|
c.destroy();
|
|
}
|
|
|
|
if ( myGrid != null )
|
|
myGrid.remove( this );
|
|
}
|
|
|
|
@Override
|
|
public World getWorld()
|
|
{
|
|
return gridProxy.getLocation().getWorld();
|
|
}
|
|
|
|
@Override
|
|
public boolean meetsChannelRequirements()
|
|
{
|
|
return (!getGridBlock().getFlags().contains( GridFlags.REQUIRE_CHANNEL ) || getUsedChannels() > 0);
|
|
}
|
|
|
|
@Override
|
|
public boolean isActive()
|
|
{
|
|
IGrid g = getGrid();
|
|
if ( g != null )
|
|
{
|
|
IPathingGrid pg = g.getCache( IPathingGrid.class );
|
|
IEnergyGrid eg = g.getCache( IEnergyGrid.class );
|
|
return meetsChannelRequirements() && eg.isNetworkPowered() && !pg.isNetworkBooting();
|
|
}
|
|
return false;
|
|
}
|
|
|
|
@Override
|
|
public boolean canSupportMoreChannels()
|
|
{
|
|
return getUsedChannels() < getMaxChannels();
|
|
}
|
|
|
|
@Override
|
|
public IReadOnlyCollection<IPathItem> getPossibleOptions()
|
|
{
|
|
return (ReadOnlyCollection) getConnections();
|
|
}
|
|
|
|
public int getLastUsedChannels()
|
|
{
|
|
return (channelData >> 8) & 0xff;
|
|
}
|
|
|
|
public int getUsedChannels()
|
|
{
|
|
return channelData & 0xff;
|
|
}
|
|
|
|
@Override
|
|
public void incrementChannelCount(int usedChannels)
|
|
{
|
|
channelData += usedChannels;
|
|
}
|
|
|
|
public void setGridStorage(GridStorage s)
|
|
{
|
|
myStorage = s;
|
|
channelData = 0;
|
|
}
|
|
|
|
public GridStorage getGridStorage()
|
|
{
|
|
return myStorage;
|
|
}
|
|
|
|
@Override
|
|
public EnumSet<GridFlags> getFlags()
|
|
{
|
|
return getGridBlock().getFlags();
|
|
}
|
|
|
|
@Override
|
|
public void finalizeChannels()
|
|
{
|
|
if ( getFlags().contains( GridFlags.CANNOT_CARRY ) )
|
|
return;
|
|
|
|
if ( getLastUsedChannels() != getUsedChannels() )
|
|
{
|
|
channelData = (channelData & 0xff);
|
|
channelData |= channelData << 8;
|
|
|
|
if ( getInternalGrid() != null )
|
|
getInternalGrid().postEventTo( this, event );
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public IPathItem getControllerRoute()
|
|
{
|
|
if ( Connections.isEmpty() || getFlags().contains( GridFlags.CANNOT_CARRY ) )
|
|
return null;
|
|
|
|
return (IPathItem) Connections.get( 0 );
|
|
}
|
|
|
|
@Override
|
|
public void setControllerRoute(IPathItem fast, boolean zeroOut)
|
|
{
|
|
if ( zeroOut )
|
|
channelData &= ~0xff;
|
|
|
|
int idx = Connections.indexOf( fast );
|
|
if ( idx > 0 )
|
|
{
|
|
Connections.remove( fast );
|
|
Connections.add( 0, (IGridConnection) fast );
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public boolean hasFlag(GridFlags flag)
|
|
{
|
|
return getGridBlock().getFlags().contains( flag );
|
|
}
|
|
|
|
@Override
|
|
public int getPlayerID()
|
|
{
|
|
return playerID;
|
|
}
|
|
|
|
}
|