Java tutorial
/* Core SWING Advanced Programming By Kim Topley ISBN: 0 13 083292 8 Publisher: Prentice Hall */ import java.awt.BorderLayout; import java.awt.Toolkit; import java.awt.event.ActionEvent; import java.awt.event.InputEvent; import java.awt.event.KeyEvent; import java.awt.event.WindowAdapter; import java.awt.event.WindowEvent; import javax.swing.AbstractAction; import javax.swing.Action; import javax.swing.JButton; import javax.swing.JComponent; import javax.swing.JFrame; import javax.swing.JPanel; import javax.swing.JScrollPane; import javax.swing.JTree; import javax.swing.KeyStroke; import javax.swing.UIManager; import javax.swing.event.UndoableEditListener; import javax.swing.tree.DefaultMutableTreeNode; import javax.swing.tree.TreeNode; import javax.swing.tree.TreePath; import javax.swing.undo.AbstractUndoableEdit; import javax.swing.undo.CannotRedoException; import javax.swing.undo.CannotUndoException; import javax.swing.undo.UndoManager; import javax.swing.undo.UndoableEditSupport; public class UndoExample4 extends JFrame { public UndoExample4() { super("Undo/Redo Example 4"); DefaultMutableTreeNode rootNode = new DefaultMutableTreeNode("root"); DefaultMutableTreeNode node = new DefaultMutableTreeNode("Apollo 8"); rootNode.add(node); node.add(new DefaultMutableTreeNode("Borman")); node.add(new DefaultMutableTreeNode("Lovell")); node.add(new DefaultMutableTreeNode("Anders")); node = new DefaultMutableTreeNode("Apollo 11"); rootNode.add(node); node.add(new DefaultMutableTreeNode("Armstrong")); node.add(new DefaultMutableTreeNode("Aldrin")); node.add(new DefaultMutableTreeNode("Collins")); node = new DefaultMutableTreeNode("Apollo 12"); rootNode.add(node); node.add(new DefaultMutableTreeNode("Conrad")); node.add(new DefaultMutableTreeNode("Gordon")); node.add(new DefaultMutableTreeNode("Bean")); UndoableTree2 tree = new UndoableTree2(rootNode); getContentPane().add(new JScrollPane(tree), BorderLayout.CENTER); // Create the undo manager and actions UndoManager manager = new UndoManager(); tree.addUndoableEditListener(manager); Action undoAction = new UndoAction(manager); Action redoAction = new RedoAction(manager); // Add the actions to buttons JPanel panel = new JPanel(); JButton undoButton = new JButton("Undo"); JButton redoButton = new JButton("Redo"); undoButton.addActionListener(undoAction); redoButton.addActionListener(redoAction); panel.add(undoButton); panel.add(redoButton); getContentPane().add(panel, BorderLayout.SOUTH); // Assign the actions to keys ((JComponent) getContentPane()).registerKeyboardAction(undoAction, KeyStroke.getKeyStroke(KeyEvent.VK_Z, InputEvent.CTRL_MASK), JComponent.WHEN_IN_FOCUSED_WINDOW); ((JComponent) getContentPane()).registerKeyboardAction(redoAction, KeyStroke.getKeyStroke(KeyEvent.VK_Y, InputEvent.CTRL_MASK), JComponent.WHEN_IN_FOCUSED_WINDOW); } // The Undo action public class UndoAction extends AbstractAction { public UndoAction(UndoManager manager) { this.manager = manager; } public void actionPerformed(ActionEvent evt) { try { manager.undo(); } catch (CannotUndoException e) { Toolkit.getDefaultToolkit().beep(); } } private UndoManager manager; } // The Redo action public class RedoAction extends AbstractAction { public RedoAction(UndoManager manager) { this.manager = manager; } public void actionPerformed(ActionEvent evt) { try { manager.redo(); } catch (CannotRedoException e) { Toolkit.getDefaultToolkit().beep(); } } private UndoManager manager; } public static void main(String[] args) { try { UIManager.setLookAndFeel("com.sun.java.swing.plaf.windows.WindowsLookAndFeel"); } catch (Exception evt) { } JFrame f = new UndoExample4(); f.addWindowListener(new WindowAdapter() { public void windowClosing(WindowEvent evt) { System.exit(0); } }); f.pack(); f.setVisible(true); } } class UndoableTree2 extends JTree { public UndoableTree2(TreeNode root) { super(root); } public void addUndoableEditListener(UndoableEditListener l) { support.addUndoableEditListener(l); } public void removeUndoableEditListener(UndoableEditListener l) { support.removeUndoableEditListener(l); } public void collapsePath(TreePath path) { boolean wasExpanded = isExpanded(path); TreePath[] selections = getSelectionPaths(); super.collapsePath(path); boolean isExpanded = isExpanded(path); if (isExpanded != wasExpanded) { TreePath[] newSelections = getSelectionPaths(); support.beginUpdate(); support.postEdit(new SelectionEdit(selections, newSelections)); support.postEdit(new CollapseEdit(path)); support.endUpdate(); } } public void expandPath(TreePath path) { boolean wasExpanded = isExpanded(path); TreePath[] selections = getSelectionPaths(); super.expandPath(path); boolean isExpanded = isExpanded(path); if (isExpanded != wasExpanded) { TreePath[] newSelections = getSelectionPaths(); support.beginUpdate(); support.postEdit(new SelectionEdit(selections, newSelections)); support.postEdit(new ExpandEdit(path)); support.endUpdate(); } } private void undoExpansion(TreePath path) { super.collapsePath(path); } private void undoCollapse(TreePath path) { super.expandPath(path); } private class CollapseEdit extends AbstractUndoableEdit { public CollapseEdit(TreePath path) { this.path = path; } public void undo() throws CannotUndoException { super.undo(); UndoableTree2.this.undoCollapse(path); } public void redo() throws CannotRedoException { super.redo(); UndoableTree2.this.undoExpansion(path); } public String getPresentationName() { return "node collapse"; } private TreePath path; } private class ExpandEdit extends AbstractUndoableEdit { public ExpandEdit(TreePath path) { this.path = path; } public void undo() throws CannotUndoException { super.undo(); UndoableTree2.this.undoExpansion(path); } public void redo() throws CannotRedoException { super.redo(); UndoableTree2.this.undoCollapse(path); } public String getPresentationName() { return "node expansion"; } private TreePath path; } private class SelectionEdit extends AbstractUndoableEdit { public SelectionEdit(TreePath[] oldSelections, TreePath[] newSelections) { this.oldSelections = oldSelections; this.newSelections = newSelections; } public void undo() throws CannotUndoException { super.undo(); UndoableTree2.this.setSelectionPaths(oldSelections); } public void redo() throws CannotRedoException { super.redo(); UndoableTree2.this.setSelectionPaths(newSelections); } public String getPresentationName() { return "selection change"; } private TreePath[] oldSelections; private TreePath[] newSelections; } private UndoableEditSupport support = new UndoableEditSupport(this); }