source: josm/trunk/src/org/openstreetmap/josm/gui/dialogs/CommandStackDialog.java@ 14281

Last change on this file since 14281 was 14281, checked in by Don-vip, 6 years ago

see #12726 - accessors for undo/redo commands
  • Property svn:eol-style set to native
File size: 20.2 KB
Line 
1// License: GPL. For details, see LICENSE file.
2package org.openstreetmap.josm.gui.dialogs;
3
4import static org.openstreetmap.josm.tools.I18n.tr;
5
6import java.awt.Component;
7import java.awt.Dimension;
8import java.awt.GridBagLayout;
9import java.awt.event.ActionEvent;
10import java.awt.event.KeyEvent;
11import java.awt.event.MouseEvent;
12import java.util.ArrayList;
13import java.util.Arrays;
14import java.util.Collection;
15import java.util.LinkedHashSet;
16import java.util.List;
17import java.util.Set;
18
19import javax.swing.AbstractAction;
20import javax.swing.Box;
21import javax.swing.JComponent;
22import javax.swing.JLabel;
23import javax.swing.JPanel;
24import javax.swing.JPopupMenu;
25import javax.swing.JScrollPane;
26import javax.swing.JSeparator;
27import javax.swing.JTree;
28import javax.swing.event.TreeModelEvent;
29import javax.swing.event.TreeModelListener;
30import javax.swing.event.TreeSelectionEvent;
31import javax.swing.event.TreeSelectionListener;
32import javax.swing.tree.DefaultMutableTreeNode;
33import javax.swing.tree.DefaultTreeCellRenderer;
34import javax.swing.tree.DefaultTreeModel;
35import javax.swing.tree.MutableTreeNode;
36import javax.swing.tree.TreePath;
37import javax.swing.tree.TreeSelectionModel;
38
39import org.openstreetmap.josm.actions.AutoScaleAction;
40import org.openstreetmap.josm.actions.AutoScaleAction.AutoScaleMode;
41import org.openstreetmap.josm.command.Command;
42import org.openstreetmap.josm.command.PseudoCommand;
43import org.openstreetmap.josm.data.UndoRedoHandler;
44import org.openstreetmap.josm.data.UndoRedoHandler.CommandAddedEvent;
45import org.openstreetmap.josm.data.UndoRedoHandler.CommandQueueCleanedEvent;
46import org.openstreetmap.josm.data.UndoRedoHandler.CommandQueuePreciseListener;
47import org.openstreetmap.josm.data.UndoRedoHandler.CommandRedoneEvent;
48import org.openstreetmap.josm.data.UndoRedoHandler.CommandUndoneEvent;
49import org.openstreetmap.josm.data.osm.DataSet;
50import org.openstreetmap.josm.data.osm.OsmPrimitive;
51import org.openstreetmap.josm.gui.MainApplication;
52import org.openstreetmap.josm.gui.SideButton;
53import org.openstreetmap.josm.gui.layer.OsmDataLayer;
54import org.openstreetmap.josm.gui.widgets.PopupMenuLauncher;
55import org.openstreetmap.josm.tools.GBC;
56import org.openstreetmap.josm.tools.ImageProvider;
57import org.openstreetmap.josm.tools.InputMapUtils;
58import org.openstreetmap.josm.tools.Shortcut;
59import org.openstreetmap.josm.tools.SubclassFilteredCollection;
60
61/**
62 * Dialog displaying list of all executed commands (undo/redo buffer).
63 * @since 94
64 */
65public class CommandStackDialog extends ToggleDialog implements CommandQueuePreciseListener {
66
67 private final DefaultTreeModel undoTreeModel = new DefaultTreeModel(new DefaultMutableTreeNode());
68 private final DefaultTreeModel redoTreeModel = new DefaultTreeModel(new DefaultMutableTreeNode());
69
70 private final JTree undoTree = new JTree(undoTreeModel);
71 private final JTree redoTree = new JTree(redoTreeModel);
72
73 private DefaultMutableTreeNode undoRoot;
74 private DefaultMutableTreeNode redoRoot;
75
76 private final transient UndoRedoSelectionListener undoSelectionListener;
77 private final transient UndoRedoSelectionListener redoSelectionListener;
78
79 private final JScrollPane scrollPane;
80 private final JSeparator separator = new JSeparator();
81 // only visible, if separator is the top most component
82 private final Component spacer = Box.createRigidArea(new Dimension(0, 3));
83
84 // last operation is remembered to select the next undo/redo entry in the list
85 // after undo/redo command
86 private UndoRedoType lastOperation = UndoRedoType.UNDO;
87
88 // Actions for context menu and Enter key
89 private final SelectAction selectAction = new SelectAction();
90 private final SelectAndZoomAction selectAndZoomAction = new SelectAndZoomAction();
91
92 /**
93 * Constructs a new {@code CommandStackDialog}.
94 */
95 public CommandStackDialog() {
96 super(tr("Command Stack"), "commandstack", tr("Open a list of all commands (undo buffer)."),
97 Shortcut.registerShortcut("subwindow:commandstack", tr("Toggle: {0}",
98 tr("Command Stack")), KeyEvent.VK_O, Shortcut.ALT_SHIFT), 100);
99 undoTree.addMouseListener(new MouseEventHandler());
100 undoTree.setRootVisible(false);
101 undoTree.getSelectionModel().setSelectionMode(TreeSelectionModel.SINGLE_TREE_SELECTION);
102 undoTree.setShowsRootHandles(true);
103 undoTree.expandRow(0);
104 undoTree.setCellRenderer(new CommandCellRenderer());
105 undoSelectionListener = new UndoRedoSelectionListener(undoTree);
106 undoTree.getSelectionModel().addTreeSelectionListener(undoSelectionListener);
107 InputMapUtils.unassignCtrlShiftUpDown(undoTree, JComponent.WHEN_FOCUSED);
108
109 redoTree.addMouseListener(new MouseEventHandler());
110 redoTree.setRootVisible(false);
111 redoTree.getSelectionModel().setSelectionMode(TreeSelectionModel.SINGLE_TREE_SELECTION);
112 redoTree.setShowsRootHandles(true);
113 redoTree.expandRow(0);
114 redoTree.setCellRenderer(new CommandCellRenderer());
115 redoSelectionListener = new UndoRedoSelectionListener(redoTree);
116 redoTree.getSelectionModel().addTreeSelectionListener(redoSelectionListener);
117 InputMapUtils.unassignCtrlShiftUpDown(redoTree, JComponent.WHEN_FOCUSED);
118
119 JPanel treesPanel = new JPanel(new GridBagLayout());
120
121 treesPanel.add(spacer, GBC.eol());
122 spacer.setVisible(false);
123 treesPanel.add(undoTree, GBC.eol().fill(GBC.HORIZONTAL));
124 separator.setVisible(false);
125 treesPanel.add(separator, GBC.eol().fill(GBC.HORIZONTAL));
126 treesPanel.add(redoTree, GBC.eol().fill(GBC.HORIZONTAL));
127 treesPanel.add(Box.createRigidArea(new Dimension(0, 0)), GBC.std().weight(0, 1));
128 treesPanel.setBackground(redoTree.getBackground());
129
130 wireUpdateEnabledStateUpdater(selectAction, undoTree);
131 wireUpdateEnabledStateUpdater(selectAction, redoTree);
132
133 UndoRedoAction undoAction = new UndoRedoAction(UndoRedoType.UNDO);
134 wireUpdateEnabledStateUpdater(undoAction, undoTree);
135
136 UndoRedoAction redoAction = new UndoRedoAction(UndoRedoType.REDO);
137 wireUpdateEnabledStateUpdater(redoAction, redoTree);
138
139 scrollPane = (JScrollPane) createLayout(treesPanel, true, Arrays.asList(
140 new SideButton(selectAction),
141 new SideButton(undoAction),
142 new SideButton(redoAction)
143 ));
144
145 InputMapUtils.addEnterAction(undoTree, selectAndZoomAction);
146 InputMapUtils.addEnterAction(redoTree, selectAndZoomAction);
147 }
148
149 private static class CommandCellRenderer extends DefaultTreeCellRenderer {
150 @Override
151 public Component getTreeCellRendererComponent(JTree tree, Object value, boolean sel, boolean expanded, boolean leaf, int row,
152 boolean hasFocus) {
153 super.getTreeCellRendererComponent(tree, value, sel, expanded, leaf, row, hasFocus);
154 DefaultMutableTreeNode v = (DefaultMutableTreeNode) value;
155 if (v.getUserObject() instanceof JLabel) {
156 JLabel l = (JLabel) v.getUserObject();
157 setIcon(l.getIcon());
158 setText(l.getText());
159 }
160 return this;
161 }
162 }
163
164 private void updateTitle() {
165 int undo = undoTreeModel.getChildCount(undoTreeModel.getRoot());
166 int redo = redoTreeModel.getChildCount(redoTreeModel.getRoot());
167 if (undo > 0 || redo > 0) {
168 setTitle(tr("Command Stack: Undo: {0} / Redo: {1}", undo, redo));
169 } else {
170 setTitle(tr("Command Stack"));
171 }
172 }
173
174 /**
175 * Selection listener for undo and redo area.
176 * If one is clicked, takes away the selection from the other, so
177 * it behaves as if it was one component.
178 */
179 private class UndoRedoSelectionListener implements TreeSelectionListener {
180 private final JTree source;
181
182 UndoRedoSelectionListener(JTree source) {
183 this.source = source;
184 }
185
186 @Override
187 public void valueChanged(TreeSelectionEvent e) {
188 if (source == undoTree) {
189 redoTree.getSelectionModel().removeTreeSelectionListener(redoSelectionListener);
190 redoTree.clearSelection();
191 redoTree.getSelectionModel().addTreeSelectionListener(redoSelectionListener);
192 }
193 if (source == redoTree) {
194 undoTree.getSelectionModel().removeTreeSelectionListener(undoSelectionListener);
195 undoTree.clearSelection();
196 undoTree.getSelectionModel().addTreeSelectionListener(undoSelectionListener);
197 }
198 }
199 }
200
201 /**
202 * Wires updater for enabled state to the events. Also updates dialog title if needed.
203 * @param updater updater
204 * @param tree tree on which wire updater
205 */
206 protected void wireUpdateEnabledStateUpdater(final IEnabledStateUpdating updater, JTree tree) {
207 addShowNotifyListener(updater);
208
209 tree.addTreeSelectionListener(e -> updater.updateEnabledState());
210
211 tree.getModel().addTreeModelListener(new TreeModelListener() {
212 @Override
213 public void treeNodesChanged(TreeModelEvent e) {
214 updater.updateEnabledState();
215 updateTitle();
216 }
217
218 @Override
219 public void treeNodesInserted(TreeModelEvent e) {
220 treeNodesChanged(e);
221 }
222
223 @Override
224 public void treeNodesRemoved(TreeModelEvent e) {
225 treeNodesChanged(e);
226 }
227
228 @Override
229 public void treeStructureChanged(TreeModelEvent e) {
230 treeNodesChanged(e);
231 }
232 });
233 }
234
235 @Override
236 public void showNotify() {
237 buildTrees();
238 for (IEnabledStateUpdating listener : showNotifyListener) {
239 listener.updateEnabledState();
240 }
241 UndoRedoHandler.getInstance().addCommandQueuePreciseListener(this);
242 }
243
244 /**
245 * Simple listener setup to update the button enabled state when the side dialog shows.
246 */
247 private final transient Set<IEnabledStateUpdating> showNotifyListener = new LinkedHashSet<>();
248
249 private void addShowNotifyListener(IEnabledStateUpdating listener) {
250 showNotifyListener.add(listener);
251 }
252
253 @Override
254 public void hideNotify() {
255 undoRoot = new DefaultMutableTreeNode();
256 redoRoot = new DefaultMutableTreeNode();
257 undoTreeModel.setRoot(undoRoot);
258 redoTreeModel.setRoot(redoRoot);
259 UndoRedoHandler.getInstance().removeCommandQueuePreciseListener(this);
260 }
261
262 /**
263 * Build the trees of undo and redo commands (initially or when
264 * they have changed).
265 */
266 private void buildTrees() {
267 setTitle(tr("Command Stack"));
268 buildUndoTree();
269 buildRedoTree();
270 ensureTreesConsistency();
271 }
272
273 private void buildUndoTree() {
274 List<Command> undoCommands = UndoRedoHandler.getInstance().getUndoCommands();
275 undoRoot = new DefaultMutableTreeNode();
276 for (int i = 0; i < undoCommands.size(); ++i) {
277 undoRoot.add(getNodeForCommand(undoCommands.get(i)));
278 }
279 undoTreeModel.setRoot(undoRoot);
280 }
281
282 private void buildRedoTree() {
283 List<Command> redoCommands = UndoRedoHandler.getInstance().getRedoCommands();
284 redoRoot = new DefaultMutableTreeNode();
285 for (int i = 0; i < redoCommands.size(); ++i) {
286 redoRoot.add(getNodeForCommand(redoCommands.get(i)));
287 }
288 redoTreeModel.setRoot(redoRoot);
289 }
290
291 private void ensureTreesConsistency() {
292 List<Command> undoCommands = UndoRedoHandler.getInstance().getUndoCommands();
293 List<Command> redoCommands = UndoRedoHandler.getInstance().getRedoCommands();
294 if (redoTreeModel.getChildCount(redoRoot) > 0) {
295 redoTree.scrollRowToVisible(0);
296 scrollPane.getHorizontalScrollBar().setValue(0);
297 }
298
299 separator.setVisible(!undoCommands.isEmpty() || !redoCommands.isEmpty());
300 spacer.setVisible(undoCommands.isEmpty() && !redoCommands.isEmpty());
301
302 // if one tree is empty, move selection to the other
303 switch (lastOperation) {
304 case UNDO:
305 if (undoCommands.isEmpty()) {
306 lastOperation = UndoRedoType.REDO;
307 }
308 break;
309 case REDO:
310 if (redoCommands.isEmpty()) {
311 lastOperation = UndoRedoType.UNDO;
312 }
313 break;
314 }
315
316 // select the next command to undo/redo
317 switch (lastOperation) {
318 case UNDO:
319 undoTree.setSelectionRow(undoTree.getRowCount()-1);
320 break;
321 case REDO:
322 redoTree.setSelectionRow(0);
323 break;
324 }
325
326 undoTree.scrollRowToVisible(undoTreeModel.getChildCount(undoRoot)-1);
327 scrollPane.getHorizontalScrollBar().setValue(0);
328 }
329
330 /**
331 * Wraps a command in a CommandListMutableTreeNode.
332 * Recursively adds child commands.
333 * @param c the command
334 * @return the resulting node
335 */
336 protected CommandListMutableTreeNode getNodeForCommand(PseudoCommand c) {
337 CommandListMutableTreeNode node = new CommandListMutableTreeNode(c);
338 if (c.getChildren() != null) {
339 List<PseudoCommand> children = new ArrayList<>(c.getChildren());
340 for (int i = 0; i < children.size(); ++i) {
341 node.add(getNodeForCommand(children.get(i)));
342 }
343 }
344 return node;
345 }
346
347 /**
348 * Return primitives that are affected by some command
349 * @param path GUI elements
350 * @return collection of affected primitives, onluy usable ones
351 */
352 protected static Collection<? extends OsmPrimitive> getAffectedPrimitives(TreePath path) {
353 PseudoCommand c = ((CommandListMutableTreeNode) path.getLastPathComponent()).getCommand();
354 final OsmDataLayer currentLayer = MainApplication.getLayerManager().getEditLayer();
355 return new SubclassFilteredCollection<>(
356 c.getParticipatingPrimitives(),
357 o -> {
358 OsmPrimitive p = currentLayer.data.getPrimitiveById(o);
359 return p != null && p.isUsable();
360 }
361 );
362 }
363
364 @Override
365 public void cleaned(CommandQueueCleanedEvent e) {
366 if (isVisible()) {
367 buildTrees();
368 }
369 }
370
371 @Override
372 public void commandAdded(CommandAddedEvent e) {
373 if (isVisible()) {
374 undoRoot.add(getNodeForCommand(e.getCommand()));
375 undoTreeModel.nodeStructureChanged(undoRoot);
376 ensureTreesConsistency();
377 }
378 }
379
380 @Override
381 public void commandUndone(CommandUndoneEvent e) {
382 if (isVisible()) {
383 swapNode(undoTreeModel, undoRoot, undoRoot.getChildCount() - 1, redoTreeModel, redoRoot, 0);
384 }
385 }
386
387 @Override
388 public void commandRedone(CommandRedoneEvent e) {
389 if (isVisible()) {
390 swapNode(redoTreeModel, redoRoot, 0, undoTreeModel, undoRoot, undoRoot.getChildCount());
391 }
392 }
393
394 private void swapNode(DefaultTreeModel srcModel, DefaultMutableTreeNode srcRoot, int srcIndex,
395 DefaultTreeModel dstModel, DefaultMutableTreeNode dstRoot, int dstIndex) {
396 MutableTreeNode node = (MutableTreeNode) srcRoot.getChildAt(srcIndex);
397 srcRoot.remove(node);
398 srcModel.nodeStructureChanged(srcRoot);
399 dstRoot.insert(node, dstIndex);
400 dstModel.nodeStructureChanged(dstRoot);
401 ensureTreesConsistency();
402 }
403
404 /**
405 * Action that selects the objects that take part in a command.
406 */
407 public class SelectAction extends AbstractAction implements IEnabledStateUpdating {
408
409 /**
410 * Constructs a new {@code SelectAction}.
411 */
412 public SelectAction() {
413 putValue(NAME, tr("Select"));
414 putValue(SHORT_DESCRIPTION, tr("Selects the objects that take part in this command (unless currently deleted)"));
415 new ImageProvider("dialogs", "select").getResource().attachImageIcon(this, true);
416 }
417
418 @Override
419 public void actionPerformed(ActionEvent e) {
420 TreePath path;
421 if (!undoTree.isSelectionEmpty()) {
422 path = undoTree.getSelectionPath();
423 } else if (!redoTree.isSelectionEmpty()) {
424 path = redoTree.getSelectionPath();
425 } else
426 throw new IllegalStateException();
427
428 DataSet dataSet = MainApplication.getLayerManager().getEditDataSet();
429 if (dataSet == null) return;
430 dataSet.setSelected(getAffectedPrimitives(path));
431 }
432
433 @Override
434 public void updateEnabledState() {
435 setEnabled(!undoTree.isSelectionEmpty() || !redoTree.isSelectionEmpty());
436 }
437 }
438
439 /**
440 * Action that selects the objects that take part in a command, then zoom to them.
441 */
442 public class SelectAndZoomAction extends SelectAction {
443 /**
444 * Constructs a new {@code SelectAndZoomAction}.
445 */
446 public SelectAndZoomAction() {
447 putValue(NAME, tr("Select and zoom"));
448 putValue(SHORT_DESCRIPTION,
449 tr("Selects the objects that take part in this command (unless currently deleted), then and zooms to it"));
450 new ImageProvider("dialogs/autoscale", "selection").getResource().attachImageIcon(this, true);
451 }
452
453 @Override
454 public void actionPerformed(ActionEvent e) {
455 super.actionPerformed(e);
456 AutoScaleAction.autoScale(AutoScaleMode.SELECTION);
457 }
458 }
459
460 /**
461 * undo / redo switch to reduce duplicate code
462 */
463 protected enum UndoRedoType {
464 UNDO,
465 REDO
466 }
467
468 /**
469 * Action to undo or redo all commands up to (and including) the seleced item.
470 */
471 protected class UndoRedoAction extends AbstractAction implements IEnabledStateUpdating {
472 private final UndoRedoType type;
473 private final JTree tree;
474
475 /**
476 * constructor
477 * @param type decide whether it is an undo action or a redo action
478 */
479 public UndoRedoAction(UndoRedoType type) {
480 this.type = type;
481 if (UndoRedoType.UNDO == type) {
482 tree = undoTree;
483 putValue(NAME, tr("Undo"));
484 putValue(SHORT_DESCRIPTION, tr("Undo the selected and all later commands"));
485 new ImageProvider("undo").getResource().attachImageIcon(this, true);
486 } else {
487 tree = redoTree;
488 putValue(NAME, tr("Redo"));
489 putValue(SHORT_DESCRIPTION, tr("Redo the selected and all earlier commands"));
490 new ImageProvider("redo").getResource().attachImageIcon(this, true);
491 }
492 }
493
494 @Override
495 public void actionPerformed(ActionEvent e) {
496 lastOperation = type;
497 TreePath path = tree.getSelectionPath();
498
499 // we can only undo top level commands
500 if (path.getPathCount() != 2)
501 throw new IllegalStateException();
502
503 int idx = ((CommandListMutableTreeNode) path.getLastPathComponent()).getIndex();
504
505 // calculate the number of commands to undo/redo; then do it
506 switch (type) {
507 case UNDO:
508 int numUndo = ((DefaultMutableTreeNode) undoTreeModel.getRoot()).getChildCount() - idx;
509 UndoRedoHandler.getInstance().undo(numUndo);
510 break;
511 case REDO:
512 int numRedo = idx+1;
513 UndoRedoHandler.getInstance().redo(numRedo);
514 break;
515 }
516 MainApplication.getMap().repaint();
517 }
518
519 @Override
520 public void updateEnabledState() {
521 // do not allow execution if nothing is selected or a sub command was selected
522 setEnabled(!tree.isSelectionEmpty() && tree.getSelectionPath().getPathCount() == 2);
523 }
524 }
525
526 class MouseEventHandler extends PopupMenuLauncher {
527
528 MouseEventHandler() {
529 super(new CommandStackPopup());
530 }
531
532 @Override
533 public void mouseClicked(MouseEvent evt) {
534 if (isDoubleClick(evt)) {
535 selectAndZoomAction.actionPerformed(null);
536 }
537 }
538 }
539
540 private class CommandStackPopup extends JPopupMenu {
541 CommandStackPopup() {
542 add(selectAction);
543 add(selectAndZoomAction);
544 }
545 }
546}
Note: See TracBrowser for help on using the repository browser.