X-Git-Url: https://mograsim.net/gitweb/?a=blobdiff_plain;f=era.mi%2Fsrc%2Fera%2Fmi%2Flogic%2Fcomponents%2FTriStateBuffer.java;h=2b72bad42a3d26807dd4f30d6abb478edaeeef4d;hb=c1d0ddc342c482051fa6c455bb286617135bd3c3;hp=4000ae66e1778fbc506a61def85d0ce19ce681c7;hpb=49f569b513c36e8ad421fd5a547bf34bd830652a;p=Mograsim.git

diff --git a/era.mi/src/era/mi/logic/components/TriStateBuffer.java b/era.mi/src/era/mi/logic/components/TriStateBuffer.java
index 4000ae66..2b72bad4 100644
--- a/era.mi/src/era/mi/logic/components/TriStateBuffer.java
+++ b/era.mi/src/era/mi/logic/components/TriStateBuffer.java
@@ -1,33 +1,52 @@
 package era.mi.logic.components;
 
+import java.util.Arrays;
+import java.util.Collections;
+import java.util.List;
+
 import era.mi.logic.Bit;
-import era.mi.logic.wires.WireArray;
-import era.mi.logic.wires.WireArray.WireArrayInput;
-
-public class TriStateBuffer extends BasicComponent{
-	WireArray in, enable;
-	WireArrayInput outI;
-	
-	public TriStateBuffer(int processTime, WireArray in, WireArray out, WireArray enable) {
+import era.mi.logic.wires.Wire;
+import era.mi.logic.wires.Wire.WireEnd;
+
+public class TriStateBuffer extends BasicComponent
+{
+	WireEnd in, enable;
+	WireEnd out;
+
+	public TriStateBuffer(int processTime, WireEnd in, WireEnd out, WireEnd enable)
+	{
 		super(processTime);
-		if(in.length != out.length)
-			throw new IllegalArgumentException("Tri-state output must have the same amount of bits as the input. Input: " + in.length + " Output: " + out.length);
-		if(enable.length != 1)
-			throw new IllegalArgumentException("Tri-state enable must have exactly one bit, not " + enable.length + ".");
+		if (in.length() != out.length())
+			throw new IllegalArgumentException(
+					"Tri-state output must have the same amount of bits as the input. Input: " + in.length() + " Output: " + out.length());
+		if (enable.length() != 1)
+			throw new IllegalArgumentException("Tri-state enable must have exactly one bit, not " + enable.length() + ".");
 		this.in = in;
 		in.addObserver(this);
 		this.enable = enable;
 		enable.addObserver(this);
-		outI = out.createInput();
+		this.out = out;
 	}
-	
+
 	@Override
 	protected void compute()
 	{
-		if(enable.getValue() == Bit.ONE)
-			outI.feedSignals(in.getValues());
+		if (enable.getValue() == Bit.ONE)
+			out.feedSignals(in.getValues());
 		else
-			outI.clearSignals();
+			out.clearSignals();
+	}
+
+	@Override
+	public List<WireEnd> getAllInputs()
+	{
+		return Collections.unmodifiableList(Arrays.asList(in, enable));
+	}
+
+	@Override
+	public List<WireEnd> getAllOutputs()
+	{
+		return Collections.unmodifiableList(Arrays.asList(out));
 	}
 
 }