Application: gvSIG desktop

/*

 * To change this license header, choose License Headers in Project Properties.

 * To change this template file, choose Tools | Templates

 * and open the template in the editor.

 */

package org.gvsig.expressionevaluator.swing.impl;

import java.util.ArrayList;

import java.util.Comparator;

import java.util.List;

import javax.swing.AbstractListModel;

import javax.swing.SwingUtilities;

import org.apache.commons.io.FilenameUtils;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.expressionevaluator.swing.Element.ConstantElement;

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

/**

 *

 * @author jjdelcerro

 */

class FilteredListModel extends AbstractListModel {

    private List<ListElement> elements;

    private List<ListElement> 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 Object getElementAt(int index) {

        return this.filteredElements.get(index);

    }

    void addElement(ListElement element) {

        this.elements.add(element);

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

            this.filteredElements.add(element);

        }

    }

    public void setFilter(String filter) {

        if (StringUtils.isEmpty(filter)) {

            this.filter = null;

            this.filteredElements = this.elements;

        } else {

            this.filter = filter;

            this.filteredElements = new ArrayList<>();

            for (ListElement 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.sortForeground(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<ListElement> items = new ArrayList<>(elements);

        final Comparator comparator = ComparatorUtils.naturalComparator();

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

            @Override

            public int compare(ListElement o1, ListElement o2) {

                int r;

                if (o1 == null && o2 == null) {

                    r = 0;

                } else if (o1 == null && o2 != null) {

                    r = -1;

                } else if (o1 != null && o2 == null) {

                    r = 1;

                } else {

                    if( o1.getValue() instanceof ConstantElement ) {

                        r = comparator.compare(

                            ((ConstantElement)o1.getValue()).getValue(),

                            ((ConstantElement)o2.getValue()).getValue()

                        );

                    } else {

                        r = comparator.compare(o1.getValue(),o2.getValue());

                    }

                }

                if (ascending) {

                    return r;

                }

                return r * -1;

            }

        });

        elements = items;

        filteredElements = elements;

        if (filter != null) {

            setFilter(filter);

        } else {

            fireContentsChanged();

        }

    }

    void addElement(Object data) {

        ListElement e = new ListElement(data.toString(), data);

        this.addElement(e);

    }

}