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

package org.gvsig.tools.swing.impl;

import java.awt.event.ActionEvent;

import java.awt.event.KeyAdapter;

import java.awt.event.KeyEvent;

import javax.swing.JButton;

import javax.swing.JTree;

import javax.swing.text.JTextComponent;

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

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

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

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

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

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

/**

 *

 * @author jjdelcerro

 */

public class DefaultFilteredTreeController implements FilteredTreeController {

    private final JTree tree;

    private final JTextComponent text;

    private final JButton button;

    public DefaultFilteredTreeController(JTree tree, JTextComponent text, JButton button) {

        this.tree = tree;

        this.text = text;

        this.button = button;

        this.initComponents();

    }

    private void initComponents() {

        ToolsSwingManager toolsSwingManager = ToolsSwingLocator.getToolsSwingManager();

        FilteredTreeModel model = this.createModel(tree);

        this.tree.setModel(model);

        toolsSwingManager.addClearButton(this.text, (ActionEvent e) -> {

            text.setText("");

            clearFilter();

        });

        toolsSwingManager.setDefaultPopupMenu(this.text);

        this.text.addKeyListener(new KeyAdapter() {

            @Override

            public void keyTyped(KeyEvent e) {

                if (e.getKeyChar() == '\n') {

                    setFilter(text.getText());

                }

            }

        });

        this.button.addActionListener((ActionEvent e) -> {

            setFilter(text.getText());

        });

    }

    @Override

    public FilteredTreeModel createModel(JTree tree) {

        ToolsSwingManager toolsSwingManager = ToolsSwingLocator.getToolsSwingManager();

        FilteredTreeModel model = toolsSwingManager.createFilteredTreeModel(tree);

        return model;

    }

    @Override

    public FilteredTreeModel getModel() {

        return (FilteredTreeModel) this.tree.getModel();

    }

    @Override

    public void clearFilter() {

      this.getModel().setFilter(null);

    }

    @Override

    public void setFilter(String filter) {

      this.getModel().setFilter(filter);

    }

    @Override

    public void setFilter(FilterFunction filter) {

        if( filter == null ) {

            filter = new TreeModelRecursiveWildcarMatchFilter(this.getModel());

        }

      this.getModel().setFilterFunction(filter);

    }

}