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

package org.gvsig.tools.swing.impl;

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

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.task.AbstractCancellableTask;

import org.gvsig.tools.util.LabeledValue;

import org.gvsig.tools.util.LabeledValueImpl;

/**

 *

 * @author jjdelcerro

 */

class FilteredListModelImpl extends AbstractListModel implements FilteredListModel {

    private List<LabeledValue> elements;

    private List<LabeledValue> filteredElements;

    private String filter;

    private boolean sorting;

    private AbstractCancellableTask th;

    public FilteredListModelImpl() {

        this.elements = new ArrayList<>();

        this.filteredElements = this.elements;

        this.filter = null;

        this.sorting = false;

        this.th = null;

    }

    @Override

    public int getSize() {

        return this.filteredElements.size();

    }

    @Override

    public void clear() {

        this.elements = new ArrayList<>();

        this.filteredElements = this.elements;

        this.filter = null;

        fireContentsChanged();

    }

    @Override

    public LabeledValue getElementAt(int index) {

        return this.filteredElements.get(index);

    }

    @Override

    public void addElement(LabeledValue element) {

        if( element==null ) {

            return;

        }

        this.elements.add(element);

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

            this.filteredElements.add(element);

        }

        fireContentsChanged();

    }

    @Override

    public void addElement(Object element) {

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

    }

    @Override

    public void setFilter(String filter) {

        if (StringUtils.isBlank(filter)) {

            this.filter = null;

            this.filteredElements = this.elements;

        } else {

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

                this.filter = filter;

            } else {

                this.filter = "*" + filter + "*";

            }

            this.filteredElements = new ArrayList<>();

            for (LabeledValue element : elements) {

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

                    this.filteredElements.add(element);

                }

            }

        }

        fireContentsChanged();

    }

    private void fireContentsChanged() {

        if( !SwingUtilities.isEventDispatchThread() ) {

            SwingUtilities.invokeLater(() -> {

              fireContentsChanged();

            });

            return;

        }

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

     }

    @Override

    public String getFilter() {

        return this.filter;

    }

    @Override

    public void sort(final boolean ascending) {

        this.sortBackground(ascending);

    }

    private void sortBackground(final boolean ascending) {

        if (this.th != null) {

            this.th.cancelRequest();

        }

        this.th = new AbstractCancellableTask("FilteredListModelSorter") {

            @Override

            public void run() {

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

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

                    @Override

                    public int compare(LabeledValue o1, LabeledValue o2) {

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

                        if (ascending) {

                            return r;

                        }

                        return r * -1;

                    }

                });

                if (this.isCancellationRequested()) {

                    return;

                }

                elements = items;

                filteredElements = elements;

                if (filter != null) {

                    setFilter(filter);

                } else {

                    fireContentsChanged();

                }

            }

        };

        th.start();

    }

}