Application: gvSIG desktop » Library: org.gvsig.tools

import javax.swing.table.TableModel;

import org.gvsig.tools.swing.api.FilteredTableModel;

    @Override

    public FilteredTableModel createFilteredTableModel(TableModel tableModel, int filterColumn) {

        FilteredTableModelImpl model = new FilteredTableModelImpl(tableModel, filterColumn);

        return model;

    }

package org.gvsig.tools.swing.impl;

import org.gvsig.tools.swing.api.FilteredTableModel;

import java.util.ArrayList;

import java.util.Comparator;

import java.util.List;

import java.util.Objects;

import javax.swing.SwingUtilities;

import javax.swing.table.AbstractTableModel;

import javax.swing.table.TableModel;

import org.apache.commons.io.FilenameUtils;

import org.apache.commons.lang3.ObjectUtils;

import org.apache.commons.lang3.StringUtils;

/**

 *

 * @author jjdelcerro

 */

class FilteredTableModelImpl extends AbstractTableModel implements FilteredTableModel {

    private final TableModel baseModel ;

    private final int filterColumn;

    private String filter;

    private List<Integer> filteredElements;

    public FilteredTableModelImpl(TableModel tableModel, int filterColumn) {

        this.baseModel = tableModel;

        this.filterColumn = filterColumn;

        this.filteredElements = null;

        this.setFilter(null);

    }

    @Override

    public String getColumnName(int column) {

        return this.baseModel.getColumnName(column);

    }

    @Override

    public final void setFilter(String filter) {

        if (StringUtils.isBlank(filter)) {

            this.filter = null;

            this.filteredElements = new ArrayList<>(this.baseModel.getRowCount());

            for (int i = 0; i < this.baseModel.getRowCount(); i++) {

                filteredElements.add(i);

            }

        } else {

            this.filter = filter;

            this.filteredElements = new ArrayList<>();

            for (int i = 0; i < this.baseModel.getRowCount(); i++) {

                Object value = this.baseModel.getValueAt(i, this.filterColumn);

                if (FilenameUtils.wildcardMatch(Objects.toString(value, "").toLowerCase(), this.filter.toLowerCase())) {

                    this.filteredElements.add(i);

                }

            }

        }

        fireContentsChanged();

    }

    private void fireContentsChanged() {

        if( !SwingUtilities.isEventDispatchThread() ) {

            SwingUtilities.invokeLater(new Runnable() {

                @Override

                public void run() {

                    fireContentsChanged();

                }

            });

            return;

        }

        this.fireTableDataChanged();

     }

    @Override

    public String getFilter() {

        return this.filter;

    }

    @Override

    public void sort(final boolean ascending) {

        this.sortBackground(ascending);

    }

    private void sortBackground(final boolean ascending) {

        Thread th = new Thread(new Runnable() {

            @Override

            public void run() {

                sortForeground(ascending);

            }

        }, "FilteredTableModelSorter");

        th.start();

    }

    private void sortForeground(final boolean ascending) {

        filteredElements.sort(new Comparator<Integer>() {

            @Override

            public int compare(Integer row1, Integer row2) {

                try {

                    Comparable o1 = (Comparable) baseModel.getValueAt(row1, filterColumn);

                    Comparable o2 = (Comparable) baseModel.getValueAt(row2, filterColumn);

                    int r = ObjectUtils.compare(o1, o2);

                    if (ascending) {

                        return r;

                    }

                    return r * -1;

                } catch(Exception ex) {

                    return 0;

                }

            }

        });

        fireContentsChanged();

    }

    @Override

    public int getRowCount() {

        return this.filteredElements.size();

    }

    @Override

    public int getColumnCount() {

        return this.baseModel.getColumnCount();

    }

    @Override

    public Object getValueAt(int rowIndex, int columnIndex) {

        int row = this.filteredElements.get(rowIndex);

        return this.baseModel.getValueAt(row, columnIndex);

    }

}

package org.gvsig.tools.swing.impl;

import java.util.ArrayList;

import java.util.Comparator;

import java.util.List;

import java.util.Objects;

import javax.swing.AbstractListModel;

import javax.swing.SwingUtilities;

import org.apache.commons.io.FilenameUtils;

import org.apache.commons.lang3.ObjectUtils;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.tools.util.LabeledValue;

import org.gvsig.tools.util.LabeledValueImpl;

/**

 *

 * @author jjdelcerro

 */

class FilteredListModel extends AbstractListModel {

    private List<LabeledValue> elements;

    private List<LabeledValue> filteredElements;

    private String filter;

    public FilteredListModel() {

        this.elements = new ArrayList<>();

        this.filteredElements = this.elements;

        this.filter = null;

    }

    @Override

    public int getSize() {

        return this.filteredElements.size();

    }

    @Override

    public LabeledValue getElementAt(int index) {

        return this.filteredElements.get(index);

    }

    void addElement(LabeledValue element) {

        this.elements.add(element);

        if (this.elements != this.filteredElements) {

            this.filteredElements.add(element);

        }

    }

    void addElement(Object element) {

        this.addElement(new LabeledValueImpl<>(Objects.toString(element, ""), element));

    }

    public void setFilter(String filter) {

        if (StringUtils.isBlank(filter)) {

            this.filter = null;

            this.filteredElements = this.elements;

        } else {

            this.filter = filter;

            this.filteredElements = new ArrayList<>();

            for (LabeledValue element : elements) {

                if (FilenameUtils.wildcardMatch(element.getValue().toString().toLowerCase(), this.filter.toLowerCase())) {

                    this.filteredElements.add(element);

                }

            }

        }

        fireContentsChanged();

    }

    private void fireContentsChanged() {

        if( !SwingUtilities.isEventDispatchThread() ) {

            SwingUtilities.invokeLater(new Runnable() {

                @Override

                public void run() {

                    fireContentsChanged();

                }

            });

            return;

        }

        this.fireContentsChanged(this, 0, this.filteredElements.size());

     }

    public String getFilter() {

        return this.filter;

    }

    public void sort(final boolean ascending) {

        this.sortBackground(ascending);

    }

    private void sortBackground(final boolean ascending) {

        Thread th = new Thread(new Runnable() {

            @Override

            public void run() {

                sortForeground(ascending);

            }

        }, "FilteredListModelSorter");

        th.start();

    }

    private void sortForeground(final boolean ascending) {

        final List items = new ArrayList<>(elements);

        items.sort(new Comparator<Comparable>() {

            @Override

            public int compare(Comparable o1, Comparable o2) {

                int r = ObjectUtils.compare(o1, o2);

                if (ascending) {

                    return r;

                }

                return r * -1;

            }

        });

        elements = items;

        filteredElements = elements;

        if (filter != null) {

            setFilter(filter);

        } else {

            fireContentsChanged();

        }

    }

}

package org.gvsig.tools.swing.api;

import javax.swing.table.TableModel;

/**

 *

 * @author jjdelcerro

 */

public interface FilteredTableModel extends TableModel {

    String getFilter();

    void setFilter(String filter);

    void sort(final boolean ascending);

}

import javax.swing.table.TableModel;

    public FilteredTableModel createFilteredTableModel(TableModel model, int filterColumn);