Added Performance label

Added a JLabel between the buttonPanel and arrPanel that shows performance statistics and is updated every time that the frame is repainted.

Author: dlarocque

src/com/example/algorithmvisualizer/AlgVisualizer.java

@@ -22,11 +22,13 @@
 
 public class AlgVisualizer implements ActionListener {
 
-	private int n;
 	private final int CONTENT_WIDTH = 800;
 	private final int CONTENT_HEIGHT = 860;
 	private final int ARR_DISPLAY_HEIGHT = 800;
 	private final String[] SIZE_OPTIONS = { "10", "50", "100", "200", "400", "800" }; // array size options
+	private int n;
+	private int numSwaps;
+	private int delay;
 	private Integer indexComparisons;
 	private long startTime; // start time of a sort
 	private long endTime; // end time of a sort
@@ -47,8 +49,8 @@ public class AlgVisualizer implements ActionListener {
 	private JButton selectionButton;
 	private JButton mergeButton;
 	private JButton quickButton;
-	private JButton performanceButton;
 	private JComboBox<String> sizeChanger;
+	private JLabel performanceLabel;
 	private SwingWorker<Void, Integer[]> arrSort;
 
 	/*
@@ -83,6 +85,9 @@ public void initializeVars() {
 		arr = shuffleArr(arr);
 
 		indexComparisons = 0;
+		startTime = 0;
+		endTime = 0;
+		setDelay(2);
 
 		// Initialize objects that will display and sort the array
 
@@ -131,10 +136,10 @@ public void initializeVars() {
 		sizeChanger.addActionListener(this);
 		sizeChanger.setBackground(Color.WHITE);
 
-		performanceButton = new JButton("Performance");
-		performanceButton.addActionListener(this);
-		performanceButton.setBackground(Color.WHITE);
-		performanceButton.setEnabled(false); // This button is not available until a sort is complete
+		// Initialize the performance label and center it
+
+		performanceLabel = new JLabel();
+		performanceLabel.setHorizontalAlignment(SwingConstants.CENTER);
 	}
 
 	/*
@@ -152,14 +157,14 @@ public void setFrame() {
 		buttonPanel.add(mergeButton);
 		buttonPanel.add(quickButton);
 		buttonPanel.add(sizeChanger);
-		buttonPanel.add(performanceButton);
 
 		// Initialize and make the frame visible
 		frame = new JFrame("Algorithm Visualizer");
 		frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
 		frame.setResizable(false); // Cannot be resizable, causes visual issues
 		frame.add(buttonPanel, BorderLayout.PAGE_START); // Button panel added to the top of the frame
 		frame.add(arrPanel, BorderLayout.PAGE_END); // Array display is added to the bottom of the frame
+		frame.add(performanceLabel);
 		frame.pack();
 		frame.setLocationRelativeTo(null); // center of the screen
 		frame.setVisible(true);
@@ -215,14 +220,6 @@ public void actionPerformed(ActionEvent event) {
 			// reset and paint the new array
 			reset();
 			arrSort.execute();
-		} else if (event.getSource() == performanceButton) {
-			int numSwaps = arrDisplay.getSwappedIndexes().size();
-			long visualizationTime = endTime - startTime; // net time
-			long sortingTime = visualizationTime - (60 * numSwaps + 1); // - NEED TO FIX
-			String statsMessage = String.format(
-					"Index Comparisons : %d  Index Swaps : %d  Visualization Time : %dms  Sorting Time : %dms",
-					indexComparisons, numSwaps, visualizationTime, sortingTime);
-			JOptionPane.showMessageDialog(frame, statsMessage, "Performance", JOptionPane.PLAIN_MESSAGE);
 		}
 	}
 
@@ -239,7 +236,6 @@ public void reset() {
 		setStopSort(true);
 		arr = shuffleArr(arr);
 		arrDisplay.clearSwappedIndexes();
-		arrDisplay.setNumChunk(0);
 		arrDisplay.setComplete(false);
 		arrDisplay.setArr(arr);
 		indexComparisons = 0;
@@ -274,6 +270,36 @@ public Integer[] fillArr(Integer[] arr) {
 		return arr;
 	}
 
+	/*
+	 * updatePerformance will be called every time the array is repainted. This
+	 * makes it slower / not real time when there is a high delay.
+	 * 
+	 * We get the values for each performance statistic we want to track (number of
+	 * swaps, number of comparisons, visualization time, sorting time), format them
+	 * into a string that will then be assigned to the JLabel's text.
+	 * 
+	 * frame.pack() makes sure that our new text will fit in the frame, and will
+	 * adjust if it does not.
+	 */
+	public void updatePerformance() {
+		numSwaps = arrDisplay.getSwappedIndexes().size();
+
+		long visualizationTime = 0;
+		long sortingTime = 0;
+		if (!getSort().equals("Not Sorting")) {
+			visualizationTime = System.currentTimeMillis() - startTime;
+			sortingTime = visualizationTime - (delay * (arrDisplay.getNumChunks() - 1));
+		}
+
+		String performance = String.format(
+				"Index Comparisons : %d  Index Swaps : %d  Visualization Time : %dms  Sorting Time : %dms",
+				indexComparisons, numSwaps, visualizationTime, sortingTime);
+
+		performanceLabel.setText(performance);
+
+		frame.pack();
+	}
+
 	public Integer[] getArr() {
 		return arr;
 	}
@@ -366,10 +392,6 @@ public void setN(int n) {
 		this.n = n;
 	}
 
-	public JButton getPerformanceButton() {
-		return performanceButton;
-	}
-
 	public Integer getIndexComparisons() {
 		return indexComparisons;
 	}
@@ -393,4 +415,28 @@ public long getEndTime() {
 	public void setEndTime(long endTime) {
 		this.endTime = endTime;
 	}
+
+	public JLabel getPerformanceLabel() {
+		return performanceLabel;
+	}
+
+	public void setPerformanceLabel(JLabel performanceLabel) {
+		this.performanceLabel = performanceLabel;
+	}
+
+	public int getNumSwaps() {
+		return numSwaps;
+	}
+
+	public void setNumSwaps(int numSwaps) {
+		this.numSwaps = numSwaps;
+	}
+
+	public int getDelay() {
+		return delay;
+	}
+
+	public void setDelay(int delay) {
+		this.delay = delay;
+	}
 }

src/com/example/algorithmvisualizer/ArrDisplay.java

@@ -22,7 +22,7 @@ public class ArrDisplay extends JComponent {
 	private static final long serialVersionUID = 1L;
 	private int swappedIndex1;
 	private int swappedIndex2;
-	private int numChunk; // amount of frames painted since the array was shuffled
+	private int numChunks; // amount of frames painted since the array was shuffled
 	private boolean isComplete; // if the array is sorted
 	private ArrayList<Integer[]> swappedIndexes;
 	private Integer[] arr;
@@ -61,8 +61,9 @@ public void paintComponent(Graphics g) {
 			swappedIndex1 = -1;
 			swappedIndex2 = -1;
 		} else if (!algVisualizer.stopSort()) {
-			swappedIndex1 = swappedIndexes.get(numChunk - 1)[0];
-			swappedIndex2 = swappedIndexes.get(numChunk - 1)[1];
+			// exclude the first chunk as its used to draw the first array, not related to the sorting
+			swappedIndex1 = swappedIndexes.get(numChunks - 1)[0];	// index out of bounds exception?
+			swappedIndex2 = swappedIndexes.get(numChunks - 1)[1];
 		}
 
 		// Iterate through the array and drawn every index
@@ -89,6 +90,7 @@ public void paintComponent(Graphics g) {
 			// Draw the current indexes bar
 			graphics2d.fillRect(x, y, width, height);
 		}
+		algVisualizer.updatePerformance();
 	}
 
 	public ArrayList<Integer[]> getSwappedIndexes() {
@@ -104,23 +106,19 @@ public void clearSwappedIndexes() {
 		swappedIndexes = new ArrayList<Integer[]>();
 	}
 
-	public void setNumChunk(int numChunk) {
-		this.numChunk = numChunk;
+	public void setNumChunk(int numChunks) {
+		this.numChunks = numChunks;
 	}
 
-	public int getNumChunk() {
-		return numChunk;
+	public int getNumChunks() {
+		return numChunks;
 	}
 
 	public boolean isComplete() {
 		return isComplete;
 	}
 
 	public void setComplete(boolean complete) {
-		// Performance Button's availability is based on whether or not the array is
-		// sorted.
-		algVisualizer.getPerformanceButton().setEnabled(complete);
-
 		this.isComplete = complete;
 	}
 

src/com/example/algorithmvisualizer/ArrSorting.java

@@ -18,12 +18,11 @@
 
 public class ArrSorting extends SwingWorker<Void, Integer[]> {
 
-	private Integer delay = 10;
 	private int n;
 	private Integer[] arr;
 	private AlgVisualizer algVisualizer;
 	private ArrDisplay arrDisplay;
-	private int numChunk;
+	private int numChunks;
 
 	public ArrSorting(AlgVisualizer algVisualizer, Integer[] arr, ArrDisplay arrDisplay) {
 		this.algVisualizer = algVisualizer;
@@ -87,8 +86,8 @@ protected Void doInBackground() throws Exception {
 	protected void process(List<Integer[]> chunks) {
 		while (chunks.size() > 0) {
 			// Paint each cloned array and update number of chunks
-			numChunk++;
-			arrDisplay.setNumChunk(this.numChunk);
+			numChunks++;
+			arrDisplay.setNumChunk(this.numChunks);
 			arrDisplay.setArr(chunks.get(0));
 			arrDisplay.repaint();
 			chunks.remove(0);
@@ -101,7 +100,7 @@ protected void process(List<Integer[]> chunks) {
 	 */
 	private void sleep() {
 		try {
-			Thread.sleep(delay);
+			Thread.sleep(algVisualizer.getDelay());
 		} catch (InterruptedException e) {
 			e.printStackTrace();
 		}