Applied-Energistics-2-tilera-Edition/src/main/java/appeng/me/GridConnection.java

/*
 * This file is part of Applied Energistics 2.
 * Copyright (c) 2013 - 2014, AlgorithmX2, All rights reserved.
 *
 * Applied Energistics 2 is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Applied Energistics 2 is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with Applied Energistics 2.  If not, see <http://www.gnu.org/licenses/lgpl>.
 */

package appeng.me;


import java.util.Arrays;
import java.util.EnumSet;

import net.minecraftforge.common.util.ForgeDirection;

import appeng.api.exceptions.FailedConnection;
import appeng.api.networking.GridFlags;
import appeng.api.networking.IGridConnection;
import appeng.api.networking.IGridNode;
import appeng.api.networking.events.MENetworkChannelsChanged;
import appeng.api.networking.pathing.IPathingGrid;
import appeng.api.util.IReadOnlyCollection;
import appeng.core.AEConfig;
import appeng.core.AELog;
import appeng.core.features.AEFeature;
import appeng.me.pathfinding.IPathItem;
import appeng.util.Platform;
import appeng.util.ReadOnlyCollection;

public class GridConnection implements IGridConnection, IPathItem
{

	final static private MENetworkChannelsChanged event = new MENetworkChannelsChanged();

	private GridNode sideA;
	private ForgeDirection fromAtoB;
	private GridNode sideB;

	Object visitorIterationNumber = null;

	public int channelData = 0;

	public GridConnection(IGridNode aNode, IGridNode bNode, ForgeDirection fromAtoB) throws FailedConnection {

		GridNode a = (GridNode) aNode;
		GridNode b = (GridNode) bNode;

		if ( Platform.securityCheck( a, b ) )
		{
			if ( AEConfig.instance.isFeatureEnabled( AEFeature.LogSecurityAudits ) )
			{
				AELog.info( "Audit Failed 1: " + a.getGridBlock().getLocation() );
				AELog.info( "Audit Failed 2: " + b.getGridBlock().getLocation() );
			}

			throw new FailedConnection();
		}

		if ( a == null || b == null )
			throw new GridException( "Connection Forged Between null entities." );

		if ( a.hasConnection( b ) || b.hasConnection( a ) )
			throw new GridException( "Connection already exists." );

		this.sideA = a;
		this.fromAtoB = fromAtoB;
		this.sideB = b;

		if ( b.getMyGrid() == null )
		{
			b.setGrid( a.getInternalGrid() );
		}
		else
		{
			if ( a.getMyGrid() == null )
			{
				GridPropagator gp = new GridPropagator( b.getInternalGrid() );
				a.beginVisit( gp );
			}
			else if ( b.getMyGrid() == null )
			{
				GridPropagator gp = new GridPropagator( a.getInternalGrid() );
				b.beginVisit( gp );
			}
			else if ( this.isNetworkABetter( a, b ) )
			{
				GridPropagator gp = new GridPropagator( a.getInternalGrid() );
				b.beginVisit( gp );
			}
			else
			{
				GridPropagator gp = new GridPropagator( b.getInternalGrid() );
				a.beginVisit( gp );
			}
		}

		// a connection was destroyed RE-PATH!!
		IPathingGrid p = this.sideA.getInternalGrid().getCache( IPathingGrid.class );
		p.repath();

		this.sideA.addConnection( this );
		this.sideB.addConnection( this );
	}

	private boolean isNetworkABetter(GridNode a, GridNode b)
	{
		return a.getMyGrid().getPriority() > b.getMyGrid().getPriority() || a.getMyGrid().size() > b.getMyGrid().size();
	}

	@Override
	public void destroy()
	{
		// a connection was destroyed RE-PATH!!
		IPathingGrid p = this.sideA.getInternalGrid().getCache( IPathingGrid.class );
		p.repath();

		this.sideA.removeConnection( this );
		this.sideB.removeConnection( this );

		this.sideA.validateGrid();
		this.sideB.validateGrid();
	}

	@Override
	public IGridNode a()
	{
		return this.sideA;
	}

	@Override
	public ForgeDirection getDirection(IGridNode side)
	{
		if ( this.fromAtoB == ForgeDirection.UNKNOWN )
			return this.fromAtoB;

		if ( this.sideA == side )
			return this.fromAtoB;
		else
			return this.fromAtoB.getOpposite();
	}

	@Override
	public IGridNode b()
	{
		return this.sideB;
	}

	@Override
	public IGridNode getOtherSide(IGridNode gridNode)
	{
		if ( gridNode == this.sideA )
			return this.sideB;
		if ( gridNode == this.sideB )
			return this.sideA;

		throw new GridException( "Invalid Side of Connection" );
	}

	@Override
	public boolean hasDirection()
	{
		return this.fromAtoB != ForgeDirection.UNKNOWN;
	}

	@Override
	public IReadOnlyCollection<IPathItem> getPossibleOptions()
	{
		return new ReadOnlyCollection<IPathItem>( Arrays.asList( (IPathItem) this.a(), (IPathItem) this.b() ) );
	}

	@Override
	public void incrementChannelCount(int usedChannels)
	{
		this.channelData += usedChannels;
	}

	@Override
	public boolean canSupportMoreChannels()
	{
		return this.getLastUsedChannels() < 32; // max, PERIOD.
	}

	@Override
	public int getUsedChannels()
	{
		return (this.channelData >> 8) & 0xff;
	}

	public int getLastUsedChannels()
	{
		return this.channelData & 0xff;
	}

	@Override
	public IPathItem getControllerRoute()
	{
		if ( this.sideA.getFlags().contains( GridFlags.CANNOT_CARRY ) )
			return null;
		return this.sideA;
	}

	@Override
	public void setControllerRoute(IPathItem fast, boolean zeroOut)
	{
		if ( zeroOut )
			this.channelData &= ~0xff;

		if ( this.sideB == fast )
		{
			GridNode tmp = this.sideA;
			this.sideA = this.sideB;
			this.sideB = tmp;
			this.fromAtoB = this.fromAtoB.getOpposite();
		}
	}

	@Override
	public void finalizeChannels()
	{
		if ( this.getUsedChannels() != this.getLastUsedChannels() )
		{
			this.channelData = (this.channelData & 0xff);
			this.channelData |= this.channelData << 8;

			if ( this.sideA.getInternalGrid() != null )
				this.sideA.getInternalGrid().postEventTo( this.sideA, event );

			if ( this.sideB.getInternalGrid() != null )
				this.sideB.getInternalGrid().postEventTo( this.sideB, event );
		}
	}

	@Override
	public EnumSet<GridFlags> getFlags()
	{
		return EnumSet.noneOf( GridFlags.class );
	}

}