From b536a284fb8ad7a6a3f0aeff9f6b753e4351e519 Mon Sep 17 00:00:00 2001
From: tgame14 <tgameBusiness@gmail.com>
Date: Wed, 16 Apr 2014 16:44:06 +0300
Subject: [PATCH] Adding utility methods and properly implementing several
 interfaces, Railings now are able to search around them

---
 .../electrical/itemrailing/GridRailing.java   | 26 +++++--
 .../electrical/itemrailing/NodeRailing.java   | 68 ++++++++++++++++++-
 .../electrical/itemrailing/PartRailing.java   |  2 +-
 .../itemrailing/interfaces/IItemRailing.java  |  5 +-
 4 files changed, 88 insertions(+), 13 deletions(-)

diff --git a/electrical/src/main/scala/resonantinduction/electrical/itemrailing/GridRailing.java b/electrical/src/main/scala/resonantinduction/electrical/itemrailing/GridRailing.java
index f3b4eea3..674845b7 100644
--- a/electrical/src/main/scala/resonantinduction/electrical/itemrailing/GridRailing.java
+++ b/electrical/src/main/scala/resonantinduction/electrical/itemrailing/GridRailing.java
@@ -1,10 +1,11 @@
 package resonantinduction.electrical.itemrailing;
 
 import calclavia.lib.grid.NodeGrid;
+import com.google.common.collect.Sets;
 import resonantinduction.electrical.itemrailing.interfaces.IItemRailing;
 import resonantinduction.electrical.itemrailing.interfaces.IItemRailingTransfer;
 
-import java.util.Comparator;
+import java.util.*;
 
 /**
  * @since 18/03/14
@@ -17,18 +18,29 @@ public class GridRailing extends NodeGrid<NodeRailing>
         super(type);
     }
 
-	public IItemRailing findTargetForIItemTransfer(IItemRailingTransfer itemRailingTransfer)
+	public IItemRailing findTargetForIItemTransfer(IItemRailingTransfer itemwrapper)
 	{
-
+		if (itemwrapper.getColor() == null)
+		{
+			return findNearestInventory(itemwrapper);
+		}
+		return findNearestColoredTarget(itemwrapper);
 	}
 
-	public static class ComparatorRailing implements Comparator<IItemRailing>
+	public IItemRailing findNearestInventory(IItemRailingTransfer itemwrapper)
 	{
+		IItemRailing endGoal = null;
+		IItemRailing[] arr = (IItemRailing[]) this.getNodes().toArray();
+		Arrays.sort();
 
-		@Override
-		public int compare(IItemRailing o1, IItemRailing o2)
+		for (NodeRailing node : this.getNodes())
 		{
-			return 0;
+
 		}
 	}
+
+	public IItemRailing findNearestColoredTarget(IItemRailingTransfer itemwrapper)
+	{
+
+	}
 }
diff --git a/electrical/src/main/scala/resonantinduction/electrical/itemrailing/NodeRailing.java b/electrical/src/main/scala/resonantinduction/electrical/itemrailing/NodeRailing.java
index eb8922d2..e99e3a4e 100644
--- a/electrical/src/main/scala/resonantinduction/electrical/itemrailing/NodeRailing.java
+++ b/electrical/src/main/scala/resonantinduction/electrical/itemrailing/NodeRailing.java
@@ -1,14 +1,21 @@
 package resonantinduction.electrical.itemrailing;
 
-import calclavia.lib.grid.Grid;
 import calclavia.lib.grid.Node;
+import net.minecraft.inventory.IInventory;
+import net.minecraft.tileentity.TileEntity;
+import net.minecraft.world.World;
+import net.minecraftforge.common.ForgeDirection;
 import resonantinduction.electrical.itemrailing.interfaces.IItemRailing;
+import universalelectricity.api.vector.IVectorWorld;
+
+import java.util.ArrayList;
+import java.util.List;
 
 /**
  * @since 18/03/14
  * @author tgame14
  */
-public class NodeRailing extends Node<IItemRailing, GridRailing, NodeRailing>
+public class NodeRailing extends Node<IItemRailing, GridRailing, NodeRailing> implements IVectorWorld
 {
     public NodeRailing (PartRailing parent)
     {
@@ -21,6 +28,63 @@ public class NodeRailing extends Node<IItemRailing, GridRailing, NodeRailing>
         return new GridRailing(getClass());
     }
 
+	/**
+	 *
+	 * @return possibly null, a IInventory to target
+	 */
+	public IInventory getInventoriesNearby()
+	{
+		List<IInventory> invList = new ArrayList<IInventory>();
+		for (ForgeDirection dir : ForgeDirection.VALID_DIRECTIONS)
+		{
+			TileEntity te = parent.getWorldPos().translate(dir).getTileEntity(parent.getWorldPos().world());
+			if (te != null && te instanceof IInventory)
+			{
+				invList.add((IInventory) te);
+			}
+		}
+		return invList.get(parent.getWorldPos().world().rand.nextInt(invList.size()));
+	}
+
+	public boolean isLeaf()
+	{
+		int connectionsCount = 0;
+		for (ForgeDirection dir : ForgeDirection.VALID_DIRECTIONS)
+		{
+			if (parent.getWorldPos().translate(dir).getTileEntity(parent.getWorldPos().world()) instanceof IItemRailing)
+			{
+				connectionsCount++;
+				if (connectionsCount >= 2)
+				{
+					return false;
+				}
+			}
+		}
+		return true;
+	}
 
 
+	@Override
+	public double z()
+	{
+		return parent.getWorldPos().z();
+	}
+
+	@Override
+	public World world()
+	{
+		return parent.getWorldPos().world();
+	}
+
+	@Override
+	public double x()
+	{
+		return parent.getWorldPos().x();
+	}
+
+	@Override
+	public double y()
+	{
+		return parent.getWorldPos().y()
+	}
 }
diff --git a/electrical/src/main/scala/resonantinduction/electrical/itemrailing/PartRailing.java b/electrical/src/main/scala/resonantinduction/electrical/itemrailing/PartRailing.java
index 32f711de..566a3d8d 100644
--- a/electrical/src/main/scala/resonantinduction/electrical/itemrailing/PartRailing.java
+++ b/electrical/src/main/scala/resonantinduction/electrical/itemrailing/PartRailing.java
@@ -97,7 +97,7 @@ public class PartRailing extends PartFramedConnection<PartRailing.EnumRailing, I
     }
 
 	@Override
-	public IVectorWorld getWorldPos()
+	public VectorWorld getWorldPos()
 	{
 		return new VectorWorld(getWorld(), x(), y(), z());
 	}
diff --git a/electrical/src/main/scala/resonantinduction/electrical/itemrailing/interfaces/IItemRailing.java b/electrical/src/main/scala/resonantinduction/electrical/itemrailing/interfaces/IItemRailing.java
index cac97191..32bafe83 100644
--- a/electrical/src/main/scala/resonantinduction/electrical/itemrailing/interfaces/IItemRailing.java
+++ b/electrical/src/main/scala/resonantinduction/electrical/itemrailing/interfaces/IItemRailing.java
@@ -2,9 +2,8 @@ package resonantinduction.electrical.itemrailing.interfaces;
 
 import calclavia.lib.grid.INodeProvider;
 import calclavia.lib.render.EnumColor;
-import net.minecraft.world.World;
 import net.minecraftforge.common.ForgeDirection;
-import universalelectricity.api.vector.IVectorWorld;
+import universalelectricity.api.vector.VectorWorld;
 
 /**
  * implement on Part Railings.
@@ -22,7 +21,7 @@ public interface IItemRailing extends INodeProvider
 
     public IItemRailing setRailingColor (EnumColor color);
 
-	public IVectorWorld getWorldPos();
+	public VectorWorld getWorldPos();