Application: gvSIG desktop

package org.gvsig.fmap.dal.swing.impl.expressionevaluator;

import java.util.ArrayList;

import java.util.Iterator;

import java.util.List;

import org.apache.commons.lang.mutable.MutableInt;

import org.gvsig.expressionevaluator.Function;

import org.gvsig.expressionevaluator.swing.Element;

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

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

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

import org.gvsig.expressionevaluator.swing.ExpressionBuilderConfig;

import org.gvsig.expressionevaluator.swing.spi.AbstractElement;

import org.gvsig.expressionevaluator.swing.ExpressionEvaluatorSwingLocator;

import org.gvsig.expressionevaluator.swing.ExpressionEvaluatorSwingManager;

import org.gvsig.fmap.dal.feature.Feature;

import org.gvsig.fmap.dal.feature.FeatureAttributeDescriptor;

import org.gvsig.fmap.dal.feature.FeatureSet;

import org.gvsig.fmap.dal.feature.FeatureStore;

import org.gvsig.tools.exception.BaseException;

import org.gvsig.tools.visitor.VisitCanceledException;

import org.gvsig.tools.visitor.Visitor;

/**

 *

 * @author jjdelcerro

 */

public class FeatureAttributeElement        

        extends AbstractElement

        implements 

            GroupElement,

            SimpleElement,

            VariableElement

{

    final FeatureStore store;

    final FeatureAttributeDescriptor attrdesc;

    List<Element> elements;

    final ExpressionBuilderConfig configPanel;

    boolean hasMoreElements; 

    FeatureAttributeElement(ExpressionBuilderConfig configPanel, FeatureStore store, FeatureAttributeDescriptor attrdesc) {

        super(attrdesc.getName(), attrdesc.getDescription(), "expressionbuilder-element-table-field");

        this.configPanel = configPanel;

        this.store = store;

        this.attrdesc = attrdesc;

        this.elements = null;

        this.hasMoreElements = false;

    }

    @Override

    public String getTypeName() {

        return this.attrdesc.getDataTypeName();

    }

    @Override

    public Element get(int index) {

        return this.getElements().get(index);

    }

    @Override

    public int size() {

        return this.getElements().size();

    }

    @Override

    public Iterator<Element> iterator() {

        return this.getElements().iterator();

    }

    @Override

    public void addElement(Element element) {

    }

    @Override

    public void addElement(Function function) {

    }

    @Override

    public List<Element> getElements() {

        if( this.elements==null ) {

            hasMoreElements = false;

            final int limit = this.configPanel==null? 1000 : this.configPanel.getSimpleElementsLimit();

            ExpressionEvaluatorSwingManager manager = ExpressionEvaluatorSwingLocator.getManager();

            final List<Object> values = new ArrayList<>();

            try {

                FeatureSet set = this.store.getFeatureSet();

                final MutableInt count = new MutableInt(0);

                set.accept(new Visitor() {

                    @Override

                    public void visit(Object o) throws VisitCanceledException, BaseException {

                        Object value = ((Feature)o).get(attrdesc.getName());

                        if( !values.contains(value ) ) {

                            values.add(value);

                        }

                        count.increment();

                        if( count.intValue()>= limit ) {

                            hasMoreElements = true;

                            throw new VisitCanceledException();

                        }

                    }

                });

            } catch(VisitCanceledException ex) {

            } catch (Exception ex) {

                // FIXME

            }

            this.elements = new ArrayList<>();

            for (Object value : values) {

                this.elements.add(manager.createConstantElement(value));

            }

        }

        return this.elements;

    }

    @Override

    public boolean hasSubgroups() {

        return false;

    }

    @Override

    public String getSource() {

        return this.store.getName();

    }

    @Override

    public String getLabel() {

        return "Field";

    }

    @Override

    public boolean hasMoreElements() {

        return this.hasMoreElements;

    }

}