Application: gvSIG desktop

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

import org.gvsig.fmap.dal.swing.featuretype.AttributeDescriptorPickerController;

import java.awt.event.ItemEvent;

import java.awt.event.ItemListener;

import java.util.ArrayList;

import java.util.Comparator;

import java.util.List;

import java.util.function.Predicate;

import javax.swing.DefaultComboBoxModel;

import javax.swing.JComboBox;

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

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

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

import org.gvsig.tools.swing.api.pickercontroller.AbstractPickerController;

/**

 *

 * @author jjdelcerro

 */

public class DefaultAttributeDescriptorPickerController         

        extends AbstractPickerController<FeatureAttributeDescriptor> 

        implements AttributeDescriptorPickerController

    {

    private final JComboBox combo;

    private FeatureType featureType;

    private final Predicate<FeatureAttributeDescriptor> filter;

    public DefaultAttributeDescriptorPickerController(JComboBox combo, Predicate<FeatureAttributeDescriptor> filter) {

        this.combo = combo; 

        this.combo.setEditable(false);

        this.combo.addItemListener(new ItemListener() {

            @Override

            public void itemStateChanged(ItemEvent e) {

                fireChangeEvent();

            }

        });

        this.filter = filter;

    }

    public void setFeatureType(FeatureType featureType) {

        this.featureType = featureType;

        List<FeatureAttributeDescriptor> attributes = new ArrayList<>();

        for (FeatureAttributeDescriptor descriptor : featureType) {

            if( filter==null || filter.test(descriptor) ) {

                attributes.add(descriptor);

            }

        }

        attributes.sort(new Comparator<FeatureAttributeDescriptor>() {

            @Override

            public int compare(FeatureAttributeDescriptor o1, FeatureAttributeDescriptor o2) {

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

            }

        });

        DefaultComboBoxModel model = new DefaultComboBoxModel();

        for (FeatureAttributeDescriptor descriptor : attributes) {

            String nameDescription = descriptor.getName() + " [" + descriptor.getDataTypeName() + "]";

            model.addElement(new ListElement<>(nameDescription, descriptor));

        }

        this.combo.setModel(model);

        if( this.combo.getModel().getSize()>0 ) {

            this.combo.setSelectedIndex(0);

        } else {

            this.combo.setSelectedIndex(-1);

        }

    }

    @Override

    public FeatureAttributeDescriptor get() {

        FeatureAttributeDescriptor descriptor = (FeatureAttributeDescriptor) ListElement.getSelected(this.combo);

        return descriptor;

    }

    @Override

    public void set(FeatureAttributeDescriptor descriptor) {

        ListElement.setSelected(this.combo, descriptor);

    }

    @Override

    public void set(String name) {

        FeatureAttributeDescriptor descriptor = this.featureType.getAttributeDescriptor(name);

        if( descriptor == null ) {

            // throw new IllegalArgumentException("The field '"+name+" don't exists.");

            this.combo.setSelectedIndex(-1);

            return;

        }

        this.set(descriptor);

    }

    @Override

    public void coerceAndSet(Object descriptor) {

        this.set((FeatureAttributeDescriptor) descriptor);

    }

    @Override

    public void setEnabled(boolean enabled) {

        this.combo.setEnabled(enabled);

    }

    @Override

    public boolean isEnabled() {

        return this.combo.isEnabled();

    }

    @Override

    public String getName() {

        FeatureAttributeDescriptor descriptor = this.get();

        if( descriptor==null ) {

            return null;

        }

        return descriptor.getName();

    }

    @Override

    public void clean() {

        this.combo.setSelectedIndex(-1);

    }

}