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

package org.gvsig.tools.swing.api;

import javax.swing.tree.TreeModel;

import org.apache.commons.io.FilenameUtils;

import org.apache.commons.lang3.StringUtils;

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

/**

 *

 * @author jjdelcerro

 */

public class TreeModelRecursiveWildcarMatchFilter implements TreeModelFilter {

  private final String filter;

  public TreeModelRecursiveWildcarMatchFilter(String filter) {

    if (StringUtils.isBlank(filter)) {

        this.filter = null;

    } else {

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

            this.filter = filter.toUpperCase();

        } else {

            this.filter = "*" + filter.toUpperCase() + "*";

        }

    }    

  }

  @Override

  public boolean test(TreeModel treeModel, Object node) {

        if ( this.filter==null ) {

            return true;

        }

        if ( node==null ) {

            return true;

        }

        String nodeString = node.toString().toUpperCase();

        boolean matches = FilenameUtils.wildcardMatch(nodeString, filter);

        int childCount = treeModel.getChildCount(node);

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

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

            matches |= test(treeModel, child);

        }

        return matches;

  }

}