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

package org.gvsig.tools.swing.impl;

import java.awt.BorderLayout;

import java.awt.Color;

import java.awt.Component;

import java.awt.Dimension;

import java.awt.Point;

import java.awt.Rectangle;

import java.awt.event.MouseAdapter;

import java.awt.event.MouseEvent;

import java.beans.Transient;

import java.util.HashSet;

import java.util.Iterator;

import java.util.List;

import java.util.Set;

import java.util.Vector;

import javax.accessibility.AccessibleContext;

import javax.swing.DefaultListSelectionModel;

import javax.swing.JCheckBox;

import javax.swing.JList;

import javax.swing.JPanel;

import javax.swing.ListCellRenderer;

import javax.swing.ListModel;

import javax.swing.ListSelectionModel;

import javax.swing.event.ListSelectionEvent;

import javax.swing.event.ListSelectionListener;

import javax.swing.plaf.ListUI;

import javax.swing.text.Position;

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class DefaultJListWithCheckbox extends JListWithCheckbox {

    private static final Logger LOGGER = LoggerFactory.getLogger(DefaultJListWithCheckbox.class);

    private class CheckListCellRenderer extends JPanel implements ListCellRenderer {

        private final ListCellRenderer delegate;

        private ListSelectionModel selectionModel;

        private final JCheckBox checkBox = new JCheckBox();

        public CheckListCellRenderer(ListCellRenderer renderer, ListSelectionModel selectionModel) {

            this.delegate = renderer;

            this.selectionModel = selectionModel;

            this.setLayout(new BorderLayout());

            this.setOpaque(false);

            this.checkBox.setOpaque(false);

            checkBoxWidth = (int) this.checkBox.getPreferredSize().getWidth();

        }

        @Override

        public Component getListCellRendererComponent(JList list, Object value, int index, boolean isSelected, boolean cellHasFocus) {

            Component renderer = delegate.getListCellRendererComponent(list, value, index, isSelected, cellHasFocus);

            checkBox.setSelected(selectionModel.isSelectedIndex(index));

            removeAll();

            add(checkBox, BorderLayout.WEST);

            add(renderer, BorderLayout.CENTER);

            return this;

        }

        public void setCheckedModel(ListSelectionModel selectionModel) {

            this.selectionModel = selectionModel;

        }

    }

    private int checkBoxWidth = 0;

    private final JList wrappedList;

    private ListSelectionModel checkedsModel = new DefaultListSelectionModel();

    private Set<ListSelectionListener> checkedsListeners;

    private final ListSelectionListener checksModelListener;

    private final CheckListCellRenderer checkListCellRenderer;

    public DefaultJListWithCheckbox(final JList wrappedList) {

        this.wrappedList = wrappedList;

        this.checkedsModel.setSelectionMode(ListSelectionModel.MULTIPLE_INTERVAL_SELECTION);

        this.checkedsListeners = new HashSet<>();

        this.checkListCellRenderer = new CheckListCellRenderer(

                this.wrappedList.getCellRenderer(),

                this.checkedsModel

        );

        this.wrappedList.setCellRenderer(this.checkListCellRenderer);

        this.wrappedList.addMouseListener(new MouseAdapter() {

            @Override

            public void mouseClicked(MouseEvent me) {

                int index = wrappedList.locationToIndex(me.getPoint());

                if (index < 0) {

                    return;

                }

                if (me.getX() > wrappedList.getCellBounds(index, index).getX() + checkBoxWidth) {

                    return;

                }

                toggleCheck(index);

            }

        });

        this.checksModelListener = new ListSelectionListener() {

            @Override

            public void valueChanged(ListSelectionEvent lse) {

                wrappedList.repaint();

                //fireCheckedsListeners(lse);

            }

        };

        this.setCheckedModel(checkedsModel);

    }

    @Override

    public final void setCheckedModel(ListSelectionModel checkedsModel) {

        this.checkedsModel.removeListSelectionListener(this.checksModelListener);

        this.checkedsModel = checkedsModel;

        this.checkedsModel.addListSelectionListener(this.checksModelListener);

        this.checkListCellRenderer.setCheckedModel(checkedsModel);

    }

    @Override

    public void toggleCheck(int index) {

        if (index < 0) {

            return;

        }

        if (this.checkedsModel.isSelectedIndex(index)) {

            this.checkedsModel.removeSelectionInterval(index, index);

        } else {

            this.checkedsModel.addSelectionInterval(index, index);

        }

        // FIXME: isAdjusting is set to false always

        fireCheckedsListeners(new ListSelectionEvent(this, index, index, false));

    }

    @Override

    public ListUI getUI() {

        return this.wrappedList.getUI();

    }

    @Override

    public void setUI(ListUI ui) {

        this.wrappedList.setUI(ui);

    }

    @Override

    public void updateUI() {

        if( this.wrappedList!= null ) {

            this.wrappedList.updateUI();

        }

    }

    @Override

    public String getUIClassID() {

        return this.wrappedList.getUIClassID();

    }

    @Override

    public Object getPrototypeCellValue() {

        return this.wrappedList.getPrototypeCellValue();

    }

    @Override

    public void setPrototypeCellValue(Object prototypeCellValue) {

        this.wrappedList.setPrototypeCellValue(prototypeCellValue);

    }

    @Override

    public int getFixedCellWidth() {

        return this.wrappedList.getFixedCellWidth();

    }

    @Override

    public void setFixedCellWidth(int width) {

        this.wrappedList.setFixedCellWidth(width);

    }

    @Override

    public int getFixedCellHeight() {

        return this.wrappedList.getFixedCellHeight();

    }

    @Override

    public void setFixedCellHeight(int height) {

        this.wrappedList.setFixedCellHeight(height);

    }

    @Transient

    @Override

    public ListCellRenderer getCellRenderer() {

        return this.wrappedList.getCellRenderer();

    }

    @Override

    public void setCellRenderer(ListCellRenderer cellRenderer) {

        this.wrappedList.setCellRenderer(cellRenderer);

    }

    @Override

    public Color getSelectionForeground() {

        return this.wrappedList.getSelectionForeground();

    }

    @Override

    public void setSelectionForeground(Color selectionForeground) {

        this.wrappedList.setSelectionForeground(selectionForeground);

    }

    @Override

    public Color getSelectionBackground() {

        return this.wrappedList.getSelectionBackground();

    }

    @Override

    public void setSelectionBackground(Color selectionBackground) {

        this.wrappedList.setSelectionBackground(selectionBackground);

    }

    @Override

    public int getVisibleRowCount() {

        return this.wrappedList.getVisibleRowCount();

    }

    @Override

    public void setVisibleRowCount(int visibleRowCount) {

        this.wrappedList.setVisibleRowCount(visibleRowCount);

    }

    @Override

    public int getLayoutOrientation() {

        return this.wrappedList.getLayoutOrientation();

    }

    @Override

    public void setLayoutOrientation(int layoutOrientation) {

        this.wrappedList.setLayoutOrientation(layoutOrientation);

    }

    @Override

    public int getFirstVisibleIndex() {

        return this.wrappedList.getFirstVisibleIndex();

    }

    @Override

    public int getLastVisibleIndex() {

        return this.wrappedList.getLastVisibleIndex();

    }

    @Override

    public void ensureIndexIsVisible(int index) {

        this.wrappedList.ensureIndexIsVisible(index);

    }

    @Override

    public void setDragEnabled(boolean b) {

        this.wrappedList.setDragEnabled(b);

    }

    @Override

    public boolean getDragEnabled() {

        return this.wrappedList.getDragEnabled();

    }

//

//    public final void setDropMode(DropMode dropMode) {

//        this.wrappedList.setDropMode(dropMode);

//    }

//

//    public final DropMode getDropMode() {

//        

//    }

//

//    public final DropLocation getDropLocation() {

//        

//    }

    @Override

    public int getNextMatch(String prefix, int startIndex, Position.Bias bias) {

        return this.wrappedList.getNextMatch(prefix, startIndex, bias);

    }

    @Override

    public String getToolTipText(MouseEvent event) {

        return this.wrappedList.getToolTipText(event);

    }

    @Override

    public int locationToIndex(Point location) {

        return this.wrappedList.locationToIndex(location);

    }

    @Override

    public Point indexToLocation(int index) {

        return this.wrappedList.indexToLocation(index);

    }

    @Override

    public Rectangle getCellBounds(int index0, int index1) {

        return this.wrappedList.getCellBounds(index0, index1);

    }

    @Override

    public ListModel getModel() {

        return this.wrappedList.getModel();

    }

    @Override

    public void setModel(ListModel model) {

        this.wrappedList.setModel(model);

    }

    @Override

    public void setListData(Object[] listData) {

        this.wrappedList.setListData(listData);

    }

    @Override

    public void setListData(Vector listData) {

        this.wrappedList.setListData(listData);

    }

    @Override

    public ListSelectionModel getSelectionModel() {

        return this.wrappedList.getSelectionModel();

    }

    @Override

    public ListSelectionModel getCheckedModel() {

        return this.checkedsModel;

    }

    @Override

    public void addListSelectionListener(ListSelectionListener listener) {

        this.wrappedList.addListSelectionListener(listener);

    }

    @Override

    public void addChecksListener(ListSelectionListener listener) {

       this.checkedsListeners.add(listener);

    }

    @Override

    public void removeListSelectionListener(ListSelectionListener listener) {

        this.wrappedList.removeListSelectionListener(listener);

    }    

    @Override

    public void removeChecksListener(ListSelectionListener listener) {

        this.checkedsListeners.remove(listener);

    }

    @Override

    public ListSelectionListener[] getListSelectionListeners() {

        return this.wrappedList.getListSelectionListeners();

    }

    @Override

    public ListSelectionListener[] getChecksListeners() {

        return this.checkedsListeners.toArray(new ListSelectionListener[1]);

    }

    protected void fireCheckedsListeners(ListSelectionEvent event) {

        Iterator<ListSelectionListener> it = this.checkedsListeners.iterator();

        while(it.hasNext()) {

            ListSelectionListener listener = it.next();

            try {

                listener.valueChanged(event);

            } catch(Throwable th) {

                LOGGER.warn("Problems calling check listener ("+listener+").",th);

            }

        }

    }

    @Override

    public void setSelectionModel(ListSelectionModel selectionModel) {

        this.wrappedList.setSelectionModel(selectionModel);

    }

    @Override

    public void setSelectionMode(int selectionMode) {

        this.wrappedList.setSelectionMode(selectionMode);

    }

    @Override

    public int getSelectionMode() {

        return this.wrappedList.getSelectionMode();

    }

    @Override

    public int getAnchorSelectionIndex() {

        return this.wrappedList.getAnchorSelectionIndex();

    }

    @Override

    public int getLeadSelectionIndex() {

        return this.wrappedList.getLeadSelectionIndex();

    }

    @Override

    public int getMinSelectionIndex() {

        return this.wrappedList.getMinSelectionIndex();

    }

    @Override

    public int getMaxSelectionIndex() {

        return this.wrappedList.getMaxSelectionIndex();

    }

    @Override

    public boolean isSelectedIndex(int index) {

        return this.wrappedList.isSelectedIndex(index);

    }

    @Override

    public boolean isSelectionEmpty() {

        return this.wrappedList.isSelectionEmpty();

    }

    @Override

    public void clearSelection() {

        this.wrappedList.clearSelection();

    }

    @Override

    public void setSelectionInterval(int anchor, int lead) {

        this.wrappedList.setSelectionInterval(anchor, lead);

    }

    @Override

    public void addSelectionInterval(int anchor, int lead) {

        this.wrappedList.addSelectionInterval(anchor, lead);

    }

    @Override

    public void removeSelectionInterval(int index0, int index1) {

        this.wrappedList.removeSelectionInterval(index0, index1);

    }

    @Override

    public void setValueIsAdjusting(boolean b) {

        this.wrappedList.setValueIsAdjusting(b);

    }

    @Override

    public boolean getValueIsAdjusting() {

        return this.wrappedList.getValueIsAdjusting();

    }

    @Transient

    @Override

    public int[] getSelectedIndices() {

        return this.wrappedList.getSelectedIndices();

    }

    @Override

    public void setSelectedIndex(int index) {

        this.wrappedList.setSelectedIndex(index);

    }

    @Override

    public void setSelectedIndices(int[] indices) {

        this.wrappedList.setSelectedIndices(indices);

    }

    @Deprecated

    @Override

    public Object[] getSelectedValues() {

        return this.wrappedList.getSelectedValues();

    }

    @Override

    public List getSelectedValuesList() {

        return this.wrappedList.getSelectedValuesList();

    }

    @Override

    public int getSelectedIndex() {

        return this.wrappedList.getSelectedIndex();

    }

    @Override

    public Object getSelectedValue() {

        return this.wrappedList.getSelectedValue();

    }

    @Override

    public void setSelectedValue(Object anObject, boolean shouldScroll) {

        this.wrappedList.setSelectedValue(anObject, shouldScroll);

    }

    @Override

    public Dimension getPreferredScrollableViewportSize() {

        return this.wrappedList.getPreferredScrollableViewportSize();

    }

    @Override

    public int getScrollableUnitIncrement(Rectangle visibleRect, int orientation, int direction) {

        return this.wrappedList.getScrollableUnitIncrement(visibleRect, orientation, direction);

    }

    @Override

    public int getScrollableBlockIncrement(Rectangle visibleRect, int orientation, int direction) {

        return this.wrappedList.getScrollableBlockIncrement(visibleRect, orientation, direction);

    }

    @Override

    public boolean getScrollableTracksViewportWidth() {

        return this.wrappedList.getScrollableTracksViewportWidth();

    }

    @Override

    public boolean getScrollableTracksViewportHeight() {

        return this.wrappedList.getScrollableTracksViewportHeight();

    }

    @Override

    public AccessibleContext getAccessibleContext() {

        return this.wrappedList.getAccessibleContext();

    }

}