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

package org.gvsig.tools.swing.impl;

import javax.swing.event.EventListenerList;

import javax.swing.event.TreeModelEvent;

import javax.swing.event.TreeModelListener;

import javax.swing.tree.TreeModel;

import javax.swing.tree.TreePath;

import org.apache.commons.lang3.StringUtils;

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

/**

 *

 * @author osc

 */

public final class FilteredTreeModelImpl implements FilteredTreeModel {

    private final TreeModel treeModel;

    private FilterFunction filterfn;

    protected EventListenerList listenerList = new EventListenerList();

    private String filter;

    public FilteredTreeModelImpl(final TreeModel treeModel) {

        this.treeModel = treeModel;

        this.filterfn = (Object model, Object value, String filter1) -> true;

    }

    @Override

    public TreeModel getTreeModel() {

        return treeModel;

    }

    @Override

    public Object getRoot() {

        return treeModel.getRoot();

    }

    @Override

    public Object getChild(final Object parent, final int index) {

        int count = 0;

        int childCount = treeModel.getChildCount(parent);

        for (int i = 0; i < childCount; i++) {

            Object child = treeModel.getChild(parent, i);

            if( filterfn.test(treeModel, child, null) ) {

                if (count == index) {

                    return child;

                }

                count++;

            }

        }

        return null;

    }

    @Override

    public int getChildCount(final Object parent) {

        int count = 0;

        int childCount = treeModel.getChildCount(parent);

        for (int i = 0; i < childCount; i++) {

            Object child = treeModel.getChild(parent, i);

            if( filterfn.test(treeModel, child, null) ) {

                count++;

            }

        }

        return count;

    }

    @Override

    public boolean isLeaf(final Object node) {

        return treeModel.isLeaf(node);

    }

    @Override

    public void valueForPathChanged(final TreePath path, final Object newValue) {

        treeModel.valueForPathChanged(path, newValue);

    }

    @Override

    public int getIndexOfChild(final Object parent, final Object childToFind) {

        int childCount = treeModel.getChildCount(parent);

        for (int i = 0; i < childCount; i++) {

            Object child = treeModel.getChild(parent, i);

            if( filterfn.test(treeModel, child, null) ) {

                if (childToFind.equals(child)) {

                    return i;

                }

            }

        }

        return -1;

    }

    @Override

    public void addTreeModelListener(final TreeModelListener l) {

        listenerList.add(TreeModelListener.class, l);

        treeModel.addTreeModelListener(l);

    }

    @Override

    public void removeTreeModelListener(final TreeModelListener l) {

        listenerList.remove(TreeModelListener.class, l);

        treeModel.removeTreeModelListener(l);

    }

    private void fireTreeStructureChanged(Object source, TreePath path) {

        // Guaranteed to return a non-null array

        Object[] listeners = listenerList.getListenerList();

        TreeModelEvent e = null;

        // Process the listeners last to first, notifying

        // those that are interested in this event

        for (int i = listeners.length - 2; i >= 0; i -= 2) {

            if (listeners[i] == TreeModelListener.class) {

                // Lazily create the event:

                if (e == null) {

                    e = new TreeModelEvent(source, path);

                }

                ((TreeModelListener) listeners[i + 1]).treeStructureChanged(e);

            }

        }

    }

    @Override

    public String getFilter() {

        return this.filter;

    }

    @Override

    public void setFilter(String filter) {

        if (StringUtils.isBlank(filter)) {

            this.filter = null;

        } else {

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

                this.filter = filter;

            } else {

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

            }

        }

        TreePath tp = new TreePath(new Object[]{this.getRoot()});

        fireTreeStructureChanged(this, tp);

    }

    @Override

    public void setFilterFunction(FilterFunction filter) {

        if( filter == null ) {

            this.filterfn = (Object model, Object value, String filter1) -> true;

        } else {

            this.filterfn = filter;

        }

        TreePath tp = new TreePath(new Object[]{this.getRoot()});

        fireTreeStructureChanged(this, tp);

    }

}