Application: gvSIG desktop » Library: org.gvsig.tools

package org.gvsig.tools.swing.impl;

import java.awt.Color;

import java.awt.Cursor;

import java.awt.Point;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.awt.event.ItemEvent;

import java.awt.event.ItemListener;

import java.awt.event.MouseAdapter;

import java.awt.event.MouseEvent;

import java.util.HashSet;

import java.util.List;

import java.util.Objects;

import java.util.Set;

import javax.swing.AbstractButton;

import javax.swing.ComboBoxModel;

import javax.swing.ImageIcon;

import javax.swing.JComponent;

import javax.swing.JMenuItem;

import javax.swing.JPopupMenu;

import javax.swing.UIManager;

import org.apache.commons.lang3.StringUtils;

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

/**

 *

 * @author jjdelcerro

 */

@SuppressWarnings("UseSpecificCatch")

public class DropDownButton implements DropDown {

    private final AbstractButton component;

    private final Set<ItemListener> itemListeners;

    private ComboBoxModel model;

    private int selectedIndex;

    private JPopupMenu popup;

    private List<ImageIcon>icons;

    private boolean readonly = false;

    private Color disabledForeground = null;

    private Color enabledForeground = null;

    private boolean visibleDropdownArrow = true;

    public DropDownButton(final AbstractButton button) {

        this.component = button;

        this.itemListeners = new HashSet<>();

        this.component.setCursor(Cursor.getPredefinedCursor(Cursor.HAND_CURSOR));

        this.component.addMouseListener(new MouseAdapter() {

            @Override

            public void mouseClicked(MouseEvent e) {

                doShowPopup();

            }

        });

        this.icons=null;

        String s = this.component.getText();

        if( !s.endsWith(" \u25BE")) {

            this.component.setText(s+" \u25BE");

        }

        this.component.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                doShowPopup();

            }

        });

    }

    @Override

    public JComponent asJComponent() {

        return this.component;

    }

    @Override

    public boolean isReadOnly() {

        return this.readonly;

    }

    @Override

    public void setReadOnly(boolean readOnly) {

        if( disabledForeground == null ) {

            disabledForeground = UIManager.getColor("Button.disabledText");        

            enabledForeground = UIManager.getColor("Button.foreground");        

        }

        this.readonly = readOnly;

        if( readOnly ) {

            component.setForeground(disabledForeground);        

        } else {

            component.setForeground(enabledForeground);        

        }

    }

    @Override

    public void setVisibleDropdownArrow(boolean visible) {

        this.visibleDropdownArrow = visible;

        String s = this.component.getText();

        if( this.visibleDropdownArrow ) {

            if( !s.endsWith(" \u25BE")) {

                this.component.setText(s+" \u25BE");

            }

        } else {

            if( s.endsWith(" \u25BE")) {

                this.component.setText(s.substring(0, s.length()-1));

            }

        }

    }

    @Override

    public boolean isVisibleDropdownArrow() {

        return this.visibleDropdownArrow;

    }

    @Override

    public void setEnabled(boolean enabled) {

        this.component.setEnabled(enabled);

    }

    private void doShowPopup() {

        if (model == null) {

            return;

        }

        this.popup = new JPopupMenu();

        for (int i = 0; i < model.getSize(); i++) {

            this.popup.add(this.createItem(i));

        }

        Point p = component.getLocationOnScreen();

        popup.show(component, 0, component.getHeight());

    }

    @Override

    public void setIcons(List<ImageIcon> icons) {

        this.icons= icons;

    }

    @Override

    public List<ImageIcon> getIcons() {

        return this.icons;

    }

    private ImageIcon getIcon(int index) {

        if( this.icons == null || index<0 || index>this.icons.size() ) {

            return null;

        }

        return this.icons.get(index);

    }

    private JMenuItem createItem(final int index) {

        JMenuItem item = new JMenuItem(Objects.toString(model.getElementAt(index), ""));

        item.setEnabled(!readonly);

        item.setIcon(this.getIcon(index));

        item.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                if( readonly ) {

                    return;

                }

                setSelectedIndex(index);

                ItemEvent ie = new ItemEvent(DropDownButton.this, 1, index, ItemEvent.SELECTED);

                for (ItemListener itemListener : itemListeners) {

                    if (itemListener != null) {

                        itemListener.itemStateChanged(ie);

                    }

                }

            }

        });

        return item;

    }

    @Override

    public void setModel(ComboBoxModel model) {

        this.model = model;

    }

    @Override

    public ComboBoxModel getModel() {

        return this.model;

    }

    @Override

    public void setSelectedIndex(int i) {

        if( model == null ) {

            return;

        }

        String value=null;

        if( i<0 ) {

            this.component.setToolTipText(null);

        } else {

            value = Objects.toString(this.model.getElementAt(i), "");

            this.component.setToolTipText(value);

        }

        if( StringUtils.isBlank(value) ) {

            value = "     ";

        }

        this.setText(value);

        this.selectedIndex = i;

    }

    private void setText(String s) {

        if( this.visibleDropdownArrow ) {

            if( !s.endsWith(" \u25BE")) {

                this.component.setText(s+" \u25BE");

            }

        } else {

            if( s.endsWith(" \u25BE")) {

                this.component.setText(s.substring(0, s.length()-1));

            }

        }

    }

   @Override

    public Object getSelectedItem() {

        if( model == null ) {

            return null;

        }

        return this.model.getElementAt(selectedIndex);

    }

    @Override

    public int getSelectedIndex() {

        return this.selectedIndex;

    }

    @Override

    public Object[] getSelectedObjects() {

        if( model == null ) {

            return null;

        }

        return new Object[]{this.getSelectedItem()};

    }

    @Override

    public void addItemListener(ItemListener itemListener) {

        this.itemListeners.add(itemListener);

    }

    @Override

    public void removeItemListener(ItemListener l) {

        this.itemListeners.remove(l);

    }

}