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

package org.gvsig.tools.swing.impl;

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

import java.util.ArrayList;

import java.util.List;

import java.util.Objects;

import javax.swing.SwingUtilities;

import javax.swing.event.TableModelEvent;

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;

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

/**

 *

 * @author jjdelcerro

 */

class FilteredTableModelImpl extends AbstractTableModel implements FilteredTableModel {

    private final TableModel baseModel ;

    private final int filterColumn;

    private String filter;

    private List<Integer> filteredElements;

    private FilteredModel.FilterFunction filterfn;

    private boolean caseSensitive;

    public FilteredTableModelImpl(TableModel tableModel, int theFilterColumn) {

        this.baseModel = tableModel;

        this.filterColumn = theFilterColumn;

        this.filteredElements = null;

        this.setFilter(null);

        this.baseModel.addTableModelListener((TableModelEvent e) -> {

            fireContentsChanged();

        });

        this.filterfn = (Object model, Object row, String filter1) -> {

            int  i = (Integer) row;

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

            return FilenameUtils.wildcardMatch(

                    Objects.toString(value, "").toLowerCase(),

                    filter1.toLowerCase()

            );

        };

    }

    @Override

    public void setFilterFunction(FilterFunction filter) {

        this.filterfn = filter;

    }

    @Override

    public String getColumnName(int column) {

        return this.baseModel.getColumnName(column);

    }

    @Override

    public Class<?> getColumnClass(int columnIndex) {

        return this.baseModel.getColumnClass(columnIndex);

    }

    @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 {

            if( StringUtils.containsAny(filter, "*?") ) {

                this.filter = filter;

            } else {

                this.filter = filter + "*";

            }

            this.filteredElements = new ArrayList<>();

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

                if( this.filterfn.test(this.baseModel, i, this.filter) ) {

                    this.filteredElements.add(i);

                }

            }

        }

        fireContentsChanged();

    }

    private void fireContentsChanged() {

        if( !SwingUtilities.isEventDispatchThread() ) {

            SwingUtilities.invokeLater(this::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(() -> {

            sortForeground(ascending);

        }, "FilteredTableModelSorter");

        th.start();

    }

    private void sortForeground(final boolean ascending) {

        filteredElements.sort((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);

    }

    @Override

    public void setValueAt(Object aValue, int rowIndex, int columnIndex) {

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

        this.baseModel.setValueAt(aValue, row, columnIndex);

    }

    @Override

    public boolean isCellEditable(int rowIndex, int columnIndex) {

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

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

    }

    @Override

    public void setCaseSensitive(boolean sensitive) {

        this.caseSensitive = sensitive;

    }

    @Override

    public boolean isCaseSensitive() {

        return this.caseSensitive;

    }

}