Application: gvSIG desktop

package org.gvsig.expressionevaluator.swing.impl;

import java.util.ArrayList;

import java.util.Collection;

import java.util.Collections;

import java.util.Comparator;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import javax.swing.JButton;

import javax.swing.text.JTextComponent;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.expressionevaluator.Function;

import org.gvsig.expressionevaluator.SymbolTable;

import org.gvsig.expressionevaluator.swing.CalculatorController;

import org.gvsig.expressionevaluator.swing.ExpressionEvaluatorSwingManager;

import org.gvsig.expressionevaluator.swing.JExpressionBuilder;

import org.gvsig.expressionevaluator.swing.Element;

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

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

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

import org.gvsig.expressionevaluator.swing.ElementFactory;

import org.gvsig.expressionevaluator.swing.ExpressionBuilderConfig;

import org.gvsig.expressionevaluator.swing.ExpressionPickerController;

import org.gvsig.expressionevaluator.swing.FunctionAdditionalPanelFactory;

import org.gvsig.expressionevaluator.swing.JExpressionPicker;

import org.gvsig.expressionevaluator.swing.impl.elements.DefaultAggregateElements;

import org.gvsig.expressionevaluator.swing.impl.elements.DefaultConstantValueElement;

import org.gvsig.expressionevaluator.swing.impl.elements.DefaultFunctionElement;

import org.gvsig.expressionevaluator.swing.impl.elements.DefaultGroupElement;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.i18n.I18nManager;

public class DefaultExpressionEvaluatorSwingManager

        implements ExpressionEvaluatorSwingManager {

    private final Map<String, ElementFactory> elementsFactories;

    private final Map<String, FunctionAdditionalPanelFactory> functionAdditionalPanelFactories;

    public DefaultExpressionEvaluatorSwingManager() {

        this.elementsFactories = new HashMap<>();

        this.functionAdditionalPanelFactories = new HashMap<>();

    }

    @Override

    public List<Element> createElements(Collection<SymbolTable> symbolTables) {

        I18nManager i18n = ToolsLocator.getI18nManager();

        Map<String, Function> functions = new HashMap<>();

        for (SymbolTable symbolTable : symbolTables) {

            for (Function function : symbolTable.functions()) {

                functions.put(function.name().toUpperCase(), function);

            }

        }

        Map<String, GroupElement> functionGroups = new HashMap<>();

        Map<String, GroupElement> operatorGroups = new HashMap<>();

        for (Function function : functions.values()) {

            if (function.isOperator()) {

                String groupName = function.group();

                GroupElement group = operatorGroups.get(groupName);

                if (group == null) {

                    group = new DefaultGroupElement(groupName);

                    operatorGroups.put(groupName, group);

                }

                group.addElement(new DefaultFunctionElement(function));

            } else {

                String groupName = function.group();

                GroupElement group = functionGroups.get(groupName);

                if (group == null) {

                    group = new DefaultGroupElement(groupName);

                    functionGroups.put(groupName, group);

                }

                group.addElement(new DefaultFunctionElement(function));

            }

        }

        GroupElement builtinFunctions = new DefaultAggregateElements(this, i18n.getTranslation("_Functions"));

        for (GroupElement group : functionGroups.values()) {

            builtinFunctions.addElement(group);

        }

        GroupElement operatorFunctions = new DefaultAggregateElements(this, i18n.getTranslation("_Operators"));

        for (GroupElement group : operatorGroups.values()) {

            operatorFunctions.addElement(group);

        }

        List<Element> elements = new ArrayList<>();

        elements.add(operatorFunctions);

        elements.add(builtinFunctions);

        return elements;

    }

    @Override

    public void registerElement(ElementFactory factory) {

        this.elementsFactories.put(factory.getName().toLowerCase(), factory);

    }

    @Override

    public Collection<ElementFactory> getElementFactories() {

        List<ElementFactory> factories = new ArrayList<>();

        factories.addAll(this.elementsFactories.values());

        Collections.sort(factories, new Comparator<ElementFactory>() {

            @Override

            public int compare(ElementFactory o1, ElementFactory o2) {

                return StringUtils.compare(o1.getName(), o2.getName());

            }

        });

        return factories;

    }

    @Override

    public JExpressionBuilder createJExpressionBuilder() {

        DefaultJExpressionBuilder x = new DefaultJExpressionBuilder(

                this,

                new DefaultExpressionBuilderConfig(this)

        );

        return x;

    }

    @Override

    public Element createElement(String name, Object... parameters) {

        ElementFactory factory = this.elementsFactories.get(name.toLowerCase());

        if( factory==null ) {

            throw new RuntimeException("Can't locate element factory '"+name+"'.");

        }

        return factory.createElement(parameters);

    }

    @Override

    public GroupElement createGroupElement(String name) {

        GroupElement e = new DefaultGroupElement(name);

        return e;

    }

    @Override

    public GroupElement createGroupElement(String name, List<Element> elements) {

        GroupElement e = new DefaultGroupElement(this, name, elements);

        return e;

    }

    @Override

    public FunctionElement createFunctionElement(Function function) {

        FunctionElement e = new DefaultFunctionElement(function);

        return e;

    }

    @Override

    public ConstantElement createConstantElement(Object value, String label) {

        ConstantElement e = new DefaultConstantValueElement(

                value,

                label,

                null

        );

        return e;

    }

    @Override

    public ConstantElement createConstantElement(Object value, String label, String description) {

        ConstantElement e = new DefaultConstantValueElement(

                value,

                StringUtils.defaultIfEmpty(label, ""),

                description

        );

        return e;

    }

//

//    @Override

//    public VariableElement createVariableElement(ExpressionBuilderConfig config, String label, String source, String name, String typeName, String description) {

//        VariableElement e = new DefaultVariableElement(config, label, source, name, typeName, description);

//        return e;

//    }

    @Override

    public ExpressionPickerController createExpressionPickerController(JTextComponent text, JButton button) {

        ExpressionPickerController c = new DefaultExpressionPickerController(this, text, button);

        return c;

    }

    @Override

    public JExpressionPicker createJExpressionPicker() {

        JExpressionPicker x = new DefaultJExpressionPicker(this);

        return x;

    }

    @Override

    public CalculatorController createCalculatorController(JTextComponent text, int dataType) {

        DefaultCalculatorController controller = new DefaultCalculatorController(this,text,dataType);

        return controller;

    }

    @Override

    public void registerAdditionalPanel(FunctionAdditionalPanelFactory factory) {

        if( factory == null ) {

            throw new IllegalArgumentException("factory can't be null");

        }

        this.functionAdditionalPanelFactories.put(factory.getName().toUpperCase(),factory);

    }

    @Override

    public Collection<FunctionAdditionalPanelFactory> getFunctionAdditionalPanelFactories() {

        return this.functionAdditionalPanelFactories.values();

    }

    @Override

    public FunctionAdditionalPanelFactory getFunctionAdditionalPanelFactory(String functionName) {

        FunctionAdditionalPanelFactory factory = this.functionAdditionalPanelFactories.get(functionName.toUpperCase());

        return factory;

    }

}