import

import
import
import
import
import
import

java.awt.BorderLayout;
java.awt.Toolkit;
java.awt.event.ActionEvent;
java.awt.event.InputEvent;
java.awt.event.KeyEvent;
java.awt.event.WindowAdapter;
java.awt.event.WindowEvent;

import
import
import
import
import
import
import
import
import
import
import
import
import
import
import
import
import
import
import

javax.swing.AbstractAction;
javax.swing.Action;
javax.swing.JButton;
javax.swing.JComponent;
javax.swing.JFrame;
javax.swing.JPanel;
javax.swing.JScrollPane;
javax.swing.JTree;
javax.swing.KeyStroke;
javax.swing.UIManager;
javax.swing.event.UndoableEditListener;
javax.swing.tree.DefaultMutableTreeNode;
javax.swing.tree.TreeNode;
javax.swing.tree.TreePath;
javax.swing.undo.AbstractUndoableEdit;
javax.swing.undo.CannotRedoException;
javax.swing.undo.CannotUndoException;
javax.swing.undo.UndoManager;
javax.swing.undo.UndoableEditSupport;

public class undo extends JFrame {

public undo() {
super("Undo/Redo Example 3");
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"));
UndoableTree tree = new UndoableTree(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.WindowsLookAnd
Feel");
} catch (Exception evt) {}

JFrame f = new undo();

f.addWindowListener(new WindowAdapter() {
public void windowClosing(WindowEvent evt) {
System.exit(0);
}
});
f.pack();
f.setVisible(true);
}
}
class UndoableTree extends JTree {
// Only one constructor for brevity
public UndoableTree(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);
super.collapsePath(path);
boolean isExpanded = isExpanded(path);
if (isExpanded != wasExpanded) {
support.postEdit(new CollapseEdit(path));
}
}
public void expandPath(TreePath path) {
boolean wasExpanded = isExpanded(path);
super.expandPath(path);
boolean isExpanded = isExpanded(path);
if (isExpanded != wasExpanded) {
support.postEdit(new ExpandEdit(path));
}
}
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();
UndoableTree.this.undoCollapse(path);
}
public void redo() throws CannotRedoException {
super.redo();
UndoableTree.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();
UndoableTree.this.undoExpansion(path);
}
public void redo() throws CannotRedoException {
super.redo();
UndoableTree.this.undoCollapse(path);
}
public String getPresentationName() {
return "node expansion";
}
private TreePath path;
}
private UndoableEditSupport support = new UndoableEditSupport(this);
}