Application: gvSIG desktop

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

import org.gvsig.fmap.dal.swing.expressionevaluator.FeatureStoreElement;

import java.util.ArrayList;

import java.util.Collections;

import java.util.Comparator;

import java.util.Iterator;

import java.util.List;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.expressionevaluator.Function;

import org.gvsig.expressionevaluator.swing.Element;

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

import org.gvsig.expressionevaluator.swing.ExpressionBuilderConfig;

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

import org.gvsig.fmap.dal.exception.DataException;

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

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

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

/**

 *

 * @author jjdelcerro

 */

public class DefaultFeatureStoreElement 

        extends AbstractElement

        implements FeatureStoreElement , SimpleElement

{

    private FeatureStore store;

    private FeatureType type;

    private List<Element> elements;

    private final ExpressionBuilderConfig config;

    private String myName;

    @SuppressWarnings("OverridableMethodCallInConstructor")

    public DefaultFeatureStoreElement(ExpressionBuilderConfig config, FeatureStore store) {

        super("store", "expressionbuilder-element-table");

        this.config = config;

        if( store!=null ) {

            this.setFeatureStore(store);

        }

    }

    @Override

    public Element get(int index) {

        return this.elements.get(index);

    }

    @Override

    public void setName(String name) {

        this.myName = name;

    }

    @Override

    public String getName() {

        if( !StringUtils.isBlank(this.myName) ) {

            return this.myName;

        }

        if (this.store == null) {

            return super.getName();

        }

        return this.store.getName();

    }

    @Override

    public int size() {

        return this.type.size();

    }

    @Override

    public Iterator<Element> iterator() {

        return this.elements.iterator();

    }

    @Override

    public void addElement(Element element) {

    }

    @Override

    public void addElement(Function function) {

    }

    @Override

    public List<Element> getElements() {

        if( this.elements == null ) {

            return Collections.EMPTY_LIST;

        }

        return Collections.unmodifiableList(this.elements);

    }

    @Override

    public boolean hasSubgroups() {

        return true;

    }

    @Override

    public boolean hasMoreElements() {

        return false;

    }

    @Override

    public void setFeatureStore(FeatureStore store) {

          if (store==null) {

              this.elements = Collections.EMPTY_LIST;

              return;

          }

          try {

            this.store = store;

            this.type = this.store.getDefaultFeatureType();

            this.elements = new ArrayList<>();

            for (FeatureAttributeDescriptor attrdesc : type) {

                this.elements.add(new FeatureAttributeElement(

                                this.config,

                                this.store,

                                attrdesc

                        )

                );

            }

            Collections.sort(elements, new Comparator<Element>() {

                @Override

                public int compare(Element o1, Element o2) {

                    return o1.getName().compareTo(o2.getName());

                }

            });

        } catch (DataException ex) {

            throw new RuntimeException(ex);

        }

    }

    @Override

    public FeatureStore getFeatureStore() {

        return this.store;

    }

}