ListSnapshot (used on ListModel)

[fanfix.git] / ui / ListModel.java

diff --git a/ui/ListModel.java b/ui/ListModel.java
index 06e7914c1a80a5dbcc6287d8d636f96c8de000e2..52fa816a55a577aa1c8b3d27e7c9132443083029 100644 (file)
--- a/ui/ListModel.java
+++ b/ui/ListModel.java
@@ -81,6 +81,7 @@ public class ListModel<T> extends DefaultListModel6<T> {
        private int hoveredIndex;
        private List<T> items = new ArrayList<T>();
        private JList6<T> list;
+       private boolean keepSelection = true;
 
        /**
         * Create a new {@link ListModel}.
@@ -185,6 +186,31 @@ public class ListModel<T> extends DefaultListModel6<T> {
                });
        }
 
+       /**
+        * (Try and) keep the elements that were selected when filtering.
+        * <p>
+        * This will use toString on the elements to identify them, and can be a bit
+        * resource intensive.
+        * 
+        * @return TRUE if we do
+        */
+       public boolean isKeepSelection() {
+               return keepSelection;
+       }
+
+       /**
+        * (Try and) keep the elements that were selected when filtering.
+        * <p>
+        * This will use toString on the elements to identify them, and can be a bit
+        * resource intensive.
+        * 
+        * @param keepSelection
+        *            TRUE to try and keep them selected
+        */
+       public void setKeepSelection(boolean keepSelection) {
+               this.keepSelection = keepSelection;
+       }
+
        /**
         * Check if this element is currently under the mouse.
         * 
@@ -289,6 +315,11 @@ public class ListModel<T> extends DefaultListModel6<T> {
         */
        @SuppressWarnings("unchecked") // ListModel<T> and JList<T> are not java 1.6
        public void filter(Predicate<T> filter) {
+               ListSnapshot snapshot = null;
+
+               if (keepSelection)
+                       snapshot = new ListSnapshot(list);
+
                clear();
                for (T item : items) {
                        if (filter == null || filter.test(item)) {
@@ -296,6 +327,9 @@ public class ListModel<T> extends DefaultListModel6<T> {
                        }
                }
 
+               if (keepSelection)
+                       snapshot.apply();
+
                list.repaint();
        }