revert delete clear

Author: kaijiec

src/ui/helper/classsearch/ClassSearchFrame.java

@@ -140,6 +140,7 @@ public void run() {
 	public void actionPerformed(ActionEvent e) {
 
 		// User cancel the class search
+		
 		if (e.getSource() == btnCancel) {
 			setVisible(false);
 			dispose();

src/ui/helper/historyui/HistoryUI.java

@@ -3,7 +3,10 @@
 import java.awt.BorderLayout;
 import java.awt.event.ActionEvent;
 import java.awt.event.ActionListener;
-
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.Stack;
+import java.util.stream.IntStream;
 
 import javax.swing.JFrame;
 import javax.swing.JPanel;
@@ -16,6 +19,10 @@
 import javax.swing.DefaultListSelectionModel;
 import javax.swing.JButton;
 
+
+
+
+
 public class HistoryUI extends JFrame
 		{
 	private static final long serialVersionUID = 8141494344180865577L;
@@ -27,6 +34,8 @@ public class HistoryUI extends JFrame
 	private JPanel panel;
 	private JScrollPane scrollPane;
 
+	@SuppressWarnings("rawtypes")
+	private Stack<ArrayList> delete_history;
 
 	/*
 	 * Setup class finders
@@ -53,36 +62,110 @@ public boolean isCellEditable(int row, int column) {
 		resultsTable = new JTable();
 		resultsTable.setModel(defaultTableModel);
 		resultsTable.setSelectionModel(new ForcedListSelectionModel());
-
+		
 		// scroll option
 		scrollPane = new JScrollPane(resultsTable);
 		panel.add(scrollPane, BorderLayout.CENTER);
 
 		// add small JPanel for Cancel and Confirm buttons
-		JPanel buttom_panel = new JPanel();
-		getContentPane().add(buttom_panel, BorderLayout.SOUTH);
+		JPanel button_panel = new JPanel();
+		getContentPane().add(button_panel, BorderLayout.SOUTH);
 
-		//create buttons from array list
+		//initialize buttons
+		createButtons(titles, button_panel);
+		
+		delete_history = new Stack<>();
+	}
+
+	/*
+	 * create buttons
+	 * param: a list of buttons' name and button panel
+	 */
+	public void createButtons(String[] titles, JPanel button_panel){
 		for (String title : titles){
 			JButton button = new JButton(title);
-			buttom_panel.add(button);
+			button_panel.add(button);
 			button.addActionListener(new ActionListener() {
-				
+			
 				@Override
 				public void actionPerformed(ActionEvent e) {
+					//if delete button is pressed, delete selected rows
+					if(title == "delete"){
+						int[] indices = resultsTable.getSelectedRows();
+						if(indices.length != 0)
+							removeSeveralRows(indices);
+					}
+					//if exit button pressed, close window
+					else if(title == "exit"){
+						System.exit(EXIT_ON_CLOSE);
+					}
+					
+					//if revert button pressed
+					else if(title =="revert"){
+						revert();
+					}
+					
+					//if clear button pressed
+					else if(title =="clear"){
+						int num_rows = resultsTable.getRowCount();
+						int[] indices = IntStream.range(0, num_rows).toArray();
+						removeSeveralRows(indices);
+					}
 					delegate.didPressButton(title, resultsTable.getSelectedRow());
 
 				}
 			});
 		}
 	}
 
-	//add a row to the end of table
+
+	/*
+	 * revert deleted rows
+	 */
+	@SuppressWarnings("rawtypes")
+	public void revert(){
+		if(!delete_history.isEmpty()){
+			ArrayList arr = delete_history.pop();
+			int[] indices = (int[]) arr.get(0);
+			String[] objects = (String[]) arr.get(1);
+			for(int i = 0; i<indices.length; i++){
+				defaultTableModel.insertRow(indices[i], new Object[] {objects[i]} );
+			}
+		}
+	}
+	
+	
+	
+	/*
+	 * add a row to the end of table
+	 * receive a HistoryDataObject
+	 */
 	public void insert(HistoryDataObject e){
 		defaultTableModel.addRow(new Object[] {e.toString()});
 	}
 
-	
+
+	/*
+	 * remove several selected rows from table
+	 * param: an array of rows number to be removed
+	 * return an ArrayList contains a list of indices and corresponding data
+	 */
+	@SuppressWarnings({ "rawtypes", "unchecked" })
+	public void removeSeveralRows(int[] indices){
+		//create a arr to hold the deleted data and rows
+		ArrayList arr = new ArrayList<>();
+		String[] objects = new String[indices.length];
+		Arrays.sort(indices);
+		
+		for(int i=indices.length-1;i>=0;i--){
+			objects[i] = (String) resultsTable.getModel().getValueAt(indices[i], 0);
+			defaultTableModel.removeRow(indices[i]);
+		}
+		
+		arr.add(indices);
+		arr.add(objects);
+		delete_history.push(arr);
+	}
 
 	public void setDelegate(HistoryUIInterface delegate) {
 		this.delegate = delegate;
@@ -97,7 +180,7 @@ private static class ForcedListSelectionModel
 		private static final long serialVersionUID = 1L;
 
 		public ForcedListSelectionModel() {
-			this.setSelectionMode(ListSelectionModel.SINGLE_SELECTION);
+			this.setSelectionMode(ListSelectionModel.MULTIPLE_INTERVAL_SELECTION);
 		};
 	}
 

src/ui/helper/historyui/HistoryUITest.java

@@ -16,10 +16,15 @@ public class HistoryUITest {
 
 	@Before
 	public void setUp(){
-		String[] arr = {"cacel","revert","confirm"};
+		String[] arr = {"delete","exit","clear","revert","confirm"};
 		historyUI = new HistoryUI(arr);
 		historyUI.insert(new HistoryDataObject("AAAA"));
 		historyUI.insert(new HistoryDataObject("BBB"));
+		historyUI.insert(new HistoryDataObject("Test 1"));
+		historyUI.insert(new HistoryDataObject("This is a item"));
+		historyUI.insert(new HistoryDataObject("turn right"));
+		historyUI.insert(new HistoryDataObject("112345"));
+		
 	}
 
 	@Test
@@ -41,7 +46,7 @@ public void didPressButton(String buttonName, int selectedRow) {
 
 
 				historyUI.setVisible(true);
-				historyUI.setSize(new Dimension(300, 200));
+				historyUI.setSize(new Dimension(500, 400));
 
 
 			}