Application: gvSIG desktop

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

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.awt.event.ItemEvent;

import java.util.ArrayList;

import java.util.List;

import java.util.function.Predicate;

import javax.swing.DefaultComboBoxModel;

import javax.swing.DefaultListModel;

import javax.swing.JComboBox;

import javax.swing.JComponent;

import javax.swing.JList;

import javax.swing.event.ListSelectionEvent;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.fmap.dal.DataManager;

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

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

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

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

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

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

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

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

/**

 *

 * @author jjdelcerro

 */

public class DefaultAttributeDescriptorPickerController         

        extends AbstractPickerController<FeatureAttributeDescriptor> 

        implements AttributeDescriptorPickerController

    {

    private static class ComboOrList<T> implements Component {

        private JComboBox combo;

        private JList list;

        public ComboOrList(JComboBox combo) {

            this.combo = combo;

            this.combo.setEditable(false);

        }

        public ComboOrList(JList list) {

            this.list = list;

        }

        @Override

        public JComponent asJComponent() {

            if( this.combo == null ) {

                return this.list;

            }

            return this.combo;

        }

        private void addSelectionListener(final ActionListener listener) {

            if( this.combo == null ) {

                this.list.addListSelectionListener((ListSelectionEvent e) -> {

                    if( e.getValueIsAdjusting() ) {

                        return;

                    }

                    listener.actionPerformed(new ActionEvent(list, 0, "selected"));

                });

            } else {

                this.combo.addItemListener((ItemEvent e) -> {

                    if( e.getStateChange()==ItemEvent.SELECTED ) {

                        listener.actionPerformed(new ActionEvent(combo, 0, "selected"));

                    }

                });

            }

        }

        public void clear() {

            if( this.combo == null ) {

                DefaultListModel model = new DefaultListModel();

                this.list.setModel(model);

            } else {

                DefaultComboBoxModel model = new DefaultComboBoxModel();

                this.combo.setModel(model);

            }

        }

        public void add(String label, T value) {

            ListElement<T> element = new ListElement<>(label,value);

            if( this.combo == null ) {

                DefaultListModel model = (DefaultListModel) this.list.getModel();

                model.addElement(element);

            } else {

                DefaultComboBoxModel model = (DefaultComboBoxModel) this.combo.getModel();

                model.addElement(element);

            }

        }

        public int getSize() {

            if( this.combo == null ) {

                return this.list.getModel().getSize();

            } else {

                return this.combo.getModel().getSize();

            }

        }

        public void setSelectedIndex(int i) {

            if( this.combo == null ) {

                this.list.setSelectedIndex(i);

            } else {

                this.combo.setSelectedIndex(i);

            }

        }

        public void setEnabled(boolean enabled) {

            if( this.combo == null ) {

                this.list.setEnabled(enabled);

            } else {

                this.combo.setEnabled(enabled);

            }

        }

        public boolean isEnabled() {

            if( this.combo == null ) {

                return this.list.isEnabled();

            } else {

                return this.combo.isEnabled();

            }

        }

        public void setEditable(boolean editable){

            if( this.combo != null ) {

                ToolsSwingManager toolsSwingManager = ToolsSwingLocator.getToolsSwingManager();

                if(editable){

                    this.combo.setEditable(editable);

                    toolsSwingManager.addClearButton(this.combo, (ActionEvent e) -> {

                        this.combo.setSelectedIndex(-1);

                    });

                    toolsSwingManager.setDefaultPopupMenu(this.combo);

                } else {

                    toolsSwingManager.removeClearButton(this.combo);

                    this.combo.setEditable(editable);

                }

            }

        }

        public void setSelectedValue(T value) {

            if( this.combo == null ) {

                ListElement.setSelected(this.list, value);

            } else {

                ListElement.setSelected(this.combo, value);

            }

        }

        public T getSelectedValue() {

            T value;

            if( this.combo == null ) {

                value = (T) ListElement.getSelected(this.list);

            } else {

                value = (T) ListElement.getSelected(this.combo);

            }

            return value;

        }

    }

    private final ComboOrList<FeatureAttributeDescriptor> combo;

    private FeatureType featureType;

    private final Predicate<FeatureAttributeDescriptor> filter;

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

        this(new ComboOrList(combo), filter);

        combo.setRenderer(new FeatureAttributeListCellRendererImpl());

    }

    public DefaultAttributeDescriptorPickerController(JList list, Predicate<FeatureAttributeDescriptor> filter) {

        this(new ComboOrList(list), filter);

        list.setCellRenderer(new FeatureAttributeListCellRendererImpl());

    }

    private DefaultAttributeDescriptorPickerController(ComboOrList combo, Predicate<FeatureAttributeDescriptor> filter) {

        this.combo = combo; 

        this.combo.addSelectionListener((ActionEvent e) -> {

            fireChangeEvent();

        });

        this.filter = filter;

    }

    @Override

    public void setFeatureType(FeatureType featureType) {

        this.featureType = featureType;

        this.combo.clear();

        if( this.featureType==null ) {

            return;

        }

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

        for (FeatureAttributeDescriptor descriptor : featureType) {

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

                attributes.add(descriptor);

            }

        }

        int useLabels = DataManager.USE_LABELS_BOTH; //FIXME

        attributes.sort((FeatureAttributeDescriptor o1, FeatureAttributeDescriptor o2) -> {

            String theLabel1;

            String theLabel2;

            switch(useLabels) {

                case DataManager.USE_LABELS_NO:

                    theLabel1 = o1.getName();

                    theLabel2 = o2.getName();

                    break;

                case DataManager.USE_LABELS_YES:

                    theLabel1 = (o1.hasLabel())? o1.getLocalizedLabel() : o1.getName();

                    theLabel2 = (o2.hasLabel())? o2.getLocalizedLabel() : o2.getName();

                    break;

                case DataManager.USE_LABELS_BOTH:

                default:

                    theLabel1 = (o1.hasLabel())?String.format("%s [%s]", o1.getLocalizedLabel(), o1.getName()): o1.getName();

                    theLabel2 = (o2.hasLabel())?String.format("%s [%s]", o2.getLocalizedLabel(), o2.getName()): o2.getName();

                    break;

            }

            return theLabel1.compareToIgnoreCase(theLabel2);

        });

        for (FeatureAttributeDescriptor descriptor : attributes) {

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

            this.combo.add(nameDescription, descriptor);

        }

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

            this.combo.setSelectedIndex(0);

        } else {

            this.combo.setSelectedIndex(-1);

        }

    }

    @Override

    public FeatureAttributeDescriptor get() {

        FeatureAttributeDescriptor descriptor = this.combo.getSelectedValue();

        return descriptor;

    }

    @Override

    public void set(FeatureAttributeDescriptor descriptor) {

        this.combo.setSelectedValue(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() {

        Object selected = this.combo.getSelectedValue();

        if( selected == null ) {

            return null;

        }

        if(selected instanceof String){

            String name = (String)selected;

            if(StringUtils.isBlank(name)){

                return null;

            }

            return name;

        }

        FeatureAttributeDescriptor descriptor = (FeatureAttributeDescriptor) selected;

        return descriptor.getName();

    }

    @Override

    public void clean() {

        this.combo.setSelectedIndex(-1);

    }

    @Override

    public void setEditable(boolean editable) {

        super.setEditable(editable);

        this.combo.setEditable(editable);

    }

}