问JTable与RowSorter的变化--过滤器变化时的选择

EN

import java.awt.BorderLayout;
import java.util.ArrayList;
import java.util.List;
import java.util.TimeZone;

import javax.swing.Box;
import javax.swing.DefaultListSelectionModel;
import javax.swing.JButton;
import javax.swing.JFrame;
import javax.swing.JLabel;
import javax.swing.JScrollPane;
import javax.swing.JTable;
import javax.swing.ListSelectionModel;
import javax.swing.RowFilter;
import javax.swing.SwingUtilities;
import javax.swing.table.AbstractTableModel;
import javax.swing.table.TableModel;
import javax.swing.table.TableRowSorter;

public class TableProblem
    extends JFrame
{
    public static final class ApplicationModel
    {
        String[] data = TimeZone.getAvailableIDs();
        public String[] getData() { return data; }

        private String modelSelection;
        public String getModelSelection() { return modelSelection; }
        public void setModelSelection(String value) { modelSelection = value; fireModelSelectionChange(); }

        private void fireModelSelectionChange()
        { selectionListeners.forEach(l -> l.modelSelectionChanged(modelSelection, findModelIndex(modelSelection))); }

        private int findModelIndex(String value)
        {
            if (value != null)
                for (int i = 0; i < data.length; i++)
                    if (value.equals(data[i]))
                        return i;
            return -1;
        }
        private List<ApplicationModelSelectionListener> selectionListeners = new ArrayList<>();
        public void addSelectionListener(ApplicationModelSelectionListener l) { selectionListeners.add(l); }
    }

    public interface ApplicationModelSelectionListener
    {
        void modelSelectionChanged(String selection, int selectedModelIndex);
    }

    /** This class acts as the selection ViewModel.  The actual model is the
    * passed-in ApplicationModel.
    */
    public final class TimeZoneListSelectionModel
        extends DefaultListSelectionModel
        implements ApplicationModelSelectionListener
    {
        private final ApplicationModel appMdl;
        private static final long serialVersionUID = 1L;

        private TimeZoneListSelectionModel(ApplicationModel appMdl)
        {
            setSelectionMode(ListSelectionModel.SINGLE_SELECTION);
            this.appMdl = appMdl;
            appMdl.addSelectionListener(this);
        }

        // Requests to ListSelectionModel to modify the selection are passed
        // to the appMdl
        @Override
        public void clearSelection()
        {
            appMdl.setModelSelection(null);
        }

        @Override
        public void setSelectionInterval(int index0, int index1)
        {
            int modelIdx = tbl.convertRowIndexToModel(index0);
            String value = appMdl.getData()[modelIdx];
            appMdl.setModelSelection(value);
        }

        @Override
        public void addSelectionInterval(int index0, int index1)
        {
            int modelIdx = tbl.convertRowIndexToModel(index0);
            String value = appMdl.getData()[modelIdx];
            appMdl.setModelSelection(value);
        }

        // Notification from the app model about selection change gets
        // percolated back to the user interface
        @Override
        public void modelSelectionChanged(String selection, int selectedModelIndex)
        {
            if (selectedModelIndex == -1)
            {
                super.clearSelection();
                return;
            }
            int viewIndex = tbl.convertRowIndexToView(selectedModelIndex);
            if (viewIndex == -1)
                super.clearSelection();
            else
                super.setSelectionInterval(viewIndex, viewIndex);
        }
    }

    public static final class TimeZoneTableModel
        extends AbstractTableModel
    {
        private static final long serialVersionUID = 1L;
        private final String[] data;

        public TimeZoneTableModel(String[] data)
        {
            this.data = data;
        }

        @Override public int getRowCount() { return data.length; }

        @Override public int getColumnCount() { return 1; }

        @Override
        public Object getValueAt(int rowIndex, int columnIndex)
        {
            if (columnIndex == 0)
                return data[rowIndex];
            throw new IllegalArgumentException("columnIndex="+columnIndex+" should be < 1");
        }

        @Override public String getColumnName(int column)
        { return "Time Zone"; }
    }

    private static final class StringRowFilter
        extends RowFilter<TableModel, Integer>
    {
        private String prefix;
        public void setPrefix(String value) { prefix = value; rowSorter.sort(); }

        private final TableRowSorter<TableModel> rowSorter;

        public StringRowFilter(TableRowSorter<TableModel> rowSorter)
        {
            this.rowSorter = rowSorter;
        }

        @Override
        public boolean include(
            Entry<? extends TableModel, ? extends Integer> entry)
        {
            if (prefix == null)
                return true;
            String lowerCase = entry.getStringValue(0).toLowerCase();
            return lowerCase.startsWith(prefix);
        }
    }

    private static final long serialVersionUID = 1L;

    public static void main(String[] args)
    {
        ApplicationModel appMdl = new ApplicationModel();
        SwingUtilities.invokeLater(() -> new TableProblem(appMdl).setVisible(true));
    }

    private final JTable tbl;

    public TableProblem(ApplicationModel appMdl)
    {
        super("View-ModelView-Model Test");
        setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);

        TimeZoneTableModel mdl = new TimeZoneTableModel(appMdl.getData());

        tbl = new JTable(mdl);
        tbl.setAutoCreateRowSorter(true);
        TimeZoneListSelectionModel tzListSelectionModel = new TimeZoneListSelectionModel(appMdl);
        tbl.setSelectionModel(tzListSelectionModel);

        @SuppressWarnings("unchecked")
        TableRowSorter<TableModel> rowSorter = (TableRowSorter<TableModel>)tbl.getRowSorter();
        StringRowFilter filter = new StringRowFilter(rowSorter);
        rowSorter.setRowFilter(filter);

        Box filterButtons = createFilterButtons(filter);

        Box vbox = Box.createVerticalBox();
        vbox.add(filterButtons);
        vbox.add(new JScrollPane(tbl));

        JLabel mdlSelect = new JLabel("App Model selection: ");
        appMdl.addSelectionListener((selection, selectedModelIndex) ->
            mdlSelect.setText("App Model selection: " + selection + " (" +
                selectedModelIndex + ")"));
        vbox.add(mdlSelect);

        add(vbox, BorderLayout.CENTER);

        pack();
    }

    private static Box createFilterButtons(StringRowFilter filter)
    {
        Box filterButtons = Box.createHorizontalBox();
        filterButtons.add(new JLabel("Filter: "));
        for (String filterStr : "All,Africa,America,Antarctica,Asia,Australia,Canada,Europe,Pacific,Us".split(","))
            addFilterButton(filter, filterButtons, filterStr);
        return filterButtons;
    }

    private static void addFilterButton(StringRowFilter filter,
        Box filterButtons, String buttonName)
    {
        String filterPrefix = "All".equals(buttonName) ? null : buttonName.toLowerCase();
        JButton asiaButton = new JButton(buttonName);
        asiaButton.addActionListener(ae -> filter.setPrefix(filterPrefix));
        filterButtons.add(asiaButton);
    }
}