Inital commitHEADmaster

1 files changed, 113 insertions, 0 deletions

diff --git a/labs/lab6_toh/THComponent.java b/labs/lab6_toh/THComponent.java
new file mode 100644
index 0000000..3a61dc5
--- /dev/null
+++ b/labs/lab6_toh/THComponent.java
@@ -0,0 +1,113 @@
+import java.awt.Color;
+import java.awt.Graphics;
+import java.awt.Graphics2D;
+import java.awt.geom.Rectangle2D;
+import java.util.Random;
+
+import javax.swing.JComponent;
+
+@SuppressWarnings("serial")
+public class THComponent extends JComponent
+{
+	private int width;
+	private int height;
+	private int leftMargin = 10;
+	private int rightMargin = 10;
+	private int topMargin = 10;
+	private int bottomMargin = 10;
+	private int baseHeight = 10;
+	private double poleWidth = 10;
+	private TowerOfHanoi towers;
+	private int numberOfDiscs;
+	private double largestDiscSize;
+	private double smallestDiscSize = 20;
+	private double defaultReductionSize = 20.0;
+	private double defaultDiscHeight = 20.0;
+	private double[] discSizes;
+	
+	public THComponent(TowerOfHanoi aTowers)
+	{
+		towers = aTowers;
+		numberOfDiscs = towers.getNumberOfDiscs();
+		discSizes = new double[numberOfDiscs];
+	}
+	
+	public void paintComponent(Graphics g)
+	{
+		Graphics2D g2 = (Graphics2D) g;
+		
+		width = this.getWidth();
+		height = this.getHeight();
+
+		int drawingWidth = width - (leftMargin + rightMargin);
+		int drawingHeight = height - (topMargin + bottomMargin);
+
+		// draw the base;
+		
+		Rectangle2D.Double rect = new Rectangle2D.Double(leftMargin , height - (bottomMargin + baseHeight), drawingWidth, baseHeight);
+		g2.draw(rect);
+		
+		// draw poles;
+		
+		double halfDistance = ((double) drawingWidth / 6);
+		largestDiscSize = (halfDistance * 2) - poleWidth;
+		
+		for(int i = 1; i <= 5; i = i + 2)
+		{
+			rect.setRect((leftMargin + (halfDistance * i)) - (poleWidth / 2), topMargin, poleWidth, drawingHeight - baseHeight);
+			g2.draw(rect);
+		}
+		
+		// draw discs
+		
+		double totalReduction = (largestDiscSize - smallestDiscSize) / (numberOfDiscs - 1);
+		double actualReduction;
+		
+		if(totalReduction > defaultReductionSize)
+		{
+			actualReduction = defaultReductionSize;
+		}
+		else
+		{
+			actualReduction = totalReduction;
+		}
+		
+		double totalHeight = drawingHeight - baseHeight;
+		double tempDiscHeight = totalHeight / (numberOfDiscs + 1);
+		double actualDiscHeight;
+		
+		if(tempDiscHeight > defaultDiscHeight)
+		{
+			actualDiscHeight = defaultDiscHeight;
+		}
+		else
+		{
+			actualDiscHeight = tempDiscHeight;
+		}
+		
+		for(int i = 0; i < numberOfDiscs; i++)
+		{
+			discSizes[i] = largestDiscSize - (actualReduction * ((numberOfDiscs - 1) - i));
+		}
+		
+		int[][] t = new int[3][];
+		for(int i = 0; i < 3; i++)
+		{
+			t[i] = towers.getArrayOfDiscs(i);
+		}
+			
+		for(int i = 0; i <= 2; i++)
+		{
+			for(int j = 0; j < t[i].length; j++)
+			{
+				double x = (leftMargin + (halfDistance * (1 + (2 * i)))) - (discSizes[t[i][j]] / 2);
+				double y = ((topMargin + drawingHeight) - baseHeight) - (actualDiscHeight * (j + 1));
+				rect.setRect(x,y,discSizes[t[i][j]],actualDiscHeight);
+				g2.setColor(Color.GREEN);
+				g2.fill(rect);
+				g2.setColor(Color.BLACK);
+				g2.draw(rect);
+			}
+		}
+	}
+}