Application: gvSIG desktop

package org.gvsig.expressionevaluator.swing.impl;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.util.List;

import javax.swing.DefaultListModel;

import javax.swing.JComponent;

import javax.swing.SwingUtilities;

import javax.swing.event.ListSelectionEvent;

import javax.swing.event.ListSelectionListener;

import javax.swing.event.TreeModelListener;

import javax.swing.event.TreeSelectionEvent;

import javax.swing.event.TreeSelectionListener;

import javax.swing.tree.TreeModel;

import javax.swing.tree.TreePath;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.expressionevaluator.Function;

import org.gvsig.expressionevaluator.swing.ExpressionEvaluatorSwingManager;

import org.gvsig.expressionevaluator.swing.JExpressionBuilder;

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

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

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

/**

 *

 * @author jjdelcerro

 */

public class DefaultJExpressionBuilder

        extends DefaultJExpressionBuilderView

        implements JExpressionBuilder {

    private final ExpressionEvaluatorSwingManager manager;

    private List<Element> elements;

    private GroupElement currentGroup;

    private class ElementsTreeModel implements TreeModel {

        private final GroupElement root;

        public ElementsTreeModel(GroupElement root) {

            this.root = root;

        }

        @Override

        public Object getRoot() {

            return this.root;

        }

        private List<Element> getChilds(final GroupElement node) {

            return node.getElements();

        }

        @Override

        public Object getChild(Object parent, int index) {

            List<Element> childs = getChilds((DefaultGroupElement)parent);

            return childs.get(index);

        }

        @Override

        public int getChildCount(Object parent) {

            List<Element> childs = getChilds((DefaultGroupElement)parent);

            return childs.size();

        }

        @Override

        public boolean isLeaf(Object node) {

            boolean leaf = node instanceof SimpleElement;

            return leaf;

        }

        @Override

        public void valueForPathChanged(TreePath path, Object newValue) {

        }

        @Override

        public int getIndexOfChild(Object parent, Object child) {

            List<Element> childs = getChilds((DefaultGroupElement)parent);

            int n = childs.indexOf(child);

            return n;

        }

        @Override

        public void addTreeModelListener(TreeModelListener l) {

        }

        @Override

        public void removeTreeModelListener(TreeModelListener l) {

        }

    }

    public DefaultJExpressionBuilder(ExpressionEvaluatorSwingManager manager) {

        this.manager = manager;

        this.initComponents();

    }

    private void initComponents() {

        ToolsSwingManager toolsSwingManager = ToolsSwingLocator.getToolsSwingManager();

        toolsSwingManager.setDefaultPopupMenu(this.txtExpression);

        toolsSwingManager.setDefaultPopupMenu(this.txtGroupElement);

        toolsSwingManager.setDefaultPopupMenu(this.txtSimpleElementFilter);

        toolsSwingManager.setDefaultPopupMenu(this.txtDescription);

        toolsSwingManager.addClearButton(this.txtSimpleElementFilter);

        TreeModel model = new ElementsTreeModel(

                this.manager.createGroup("Elements", this.getElements())

        );

        this.treeElements.setModel(model);

        this.treeElements.addTreeSelectionListener(new TreeSelectionListener() {

            @Override

            public void valueChanged(TreeSelectionEvent e) {

                TreePath path = e.getPath();

                Element element = (Element) path.getLastPathComponent();

                if( element instanceof GroupElement) {

                    updateSimpleElementsList((GroupElement)element);

                }

            }

        });

        this.btnGroupElementInsert.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                doInsert(txtGroupElement.getText());

            }

        });

        this.btnSimpleElementInsert.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                SimpleElement element = (SimpleElement) lstSimpleElement.getSelectedValue();

                doInsert(element);

            }

        });

        this.btnSimpleElementFilter.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                doFilter();

            }

        });

        this.lstSimpleElement.getSelectionModel().addListSelectionListener(new ListSelectionListener() {

            @Override

            public void valueChanged(ListSelectionEvent e) {

                SimpleElement element = (SimpleElement) lstSimpleElement.getSelectedValue();

                doSelectElement(element);

            }

        });

        this.btnSimpleElementSortDown.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                FilteredListModel model = (FilteredListModel) lstSimpleElement.getModel();

                model.sort(true);

            }

        });

        this.btnSimpleElementSortUp.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                FilteredListModel model = (FilteredListModel) lstSimpleElement.getModel();

                model.sort(false);

            }

        });

    }

    private void doInsert(String s) {

        this.txtExpression.insert(s, this.txtExpression.getCaretPosition());

    }

    private void doInsert(SimpleElement element) {

        Function function = element.getFunction();

        if( function == null ) {

            doInsert(element.getName());

        } else {

            doInsert(function.template());

        }

    }

    private void doFilter() {

        FilteredListModel model = (FilteredListModel) this.lstSimpleElement.getModel();

        model.setFilter(this.txtSimpleElementFilter.getText());

    }

    private void doSelectElement(SimpleElement element) {

        StringBuilder html = new StringBuilder();

        html.append("<html>\n");

        html.append("<h1>").append(element.getName()).append("</h1>\n");

        Function function = element.getFunction();

        if( function != null ) {

                if( StringUtils.isEmpty(function.returnType()) ) {

                    html.append("<b>return</b> ").append(function.returnType()).append("<br>\n");

                }

                String[] args = function.descriptionArgs();

                if( args!=null ) {

                    html.append("<b>Arguments:</b> ").append("<ul>\n");

                    for (String arg : args) {

                        html.append("<li>").append(arg).append("</li>\n");

                    }

                    html.append("</ul>\n");

                    html.append("<br>\n");

                }

                if( StringUtils.isEmpty(function.template()) ) {

                    html.append("<b>Template:</b> ").append(function.template()).append("<br>\n");

                }

        }

        html.append(element.getDescription().replace("\n", "<br>")).append("<br>\n");

        html.append("</html>\n");

        this.txtDescription.setText(html.toString());

    }

    private void updateSimpleElementsList(GroupElement group) {

        FilteredListModel model = new FilteredListModel();

        model.addElement("Loading...");

        this.currentGroup = group;

        this.txtGroupElement.setText(group.getName());

        if( group instanceof SimpleElement ) {

            this.btnGroupElementInsert.setEnabled(true);

        } else {

            this.btnGroupElementInsert.setEnabled(false);

        }

        this.lstSimpleElement.setEnabled(false);

        this.lstSimpleElement.setModel(model);

        Thread th = new Thread(new Runnable() {

            @Override

            public void run() {

                simpleElementsUpdater();

            }

        }, "ExpressionBuilderSimpleElementsUpdater");

        th.start();

    }

    private void simpleElementsUpdater() {

        final GroupElement group = this.currentGroup;

        final DefaultListModel model = new DefaultListModel();

        for (Element child : group) {

            if( group != this.currentGroup ) {

                return;

            }

            String label = child.getName();

            if( child instanceof SimpleElement ) {

                Function function = ((SimpleElement) child).getFunction();

                if( function != null && StringUtils.isEmpty(function.returnType()) ) {

                    label = "<html><b>"  + label + "</b> return " + function.returnType() + "</html>";

                }

            }

            model.addElement(new ListElement<>(label,child));

        }

        SwingUtilities.invokeLater(new Runnable() {

            @Override

            public void run() {

                if( group != currentGroup ) {

                    return;

                }

                lstSimpleElement.setEnabled(true);

                lstSimpleElement.setModel(model);

            }

        });

    }

    @Override

    public String getExpression() {

        String expression = this.txtExpression.getText();

        if( StringUtils.isEmpty(expression) ) {

            return null;

        }

        return expression;

    }

    @Override

    public List<Element> getElements() {

        if( this.elements == null ) {

            this.elements = manager.getElements();

        }

        return elements;

    }

    @Override

    public JComponent asJComponent() {

        return this;

    }

}