Выбор элемента интерфейса

package ru.skipy.tests.swing;
 
import javax.swing.*;
import java.awt.*;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.MouseAdapter;
import java.awt.event.MouseEvent;
import java.lang.reflect.Array;
import java.util.*;
import java.util.List;
 
/**
 * MultipleSelectionComponentTest
 *
 * @author Eugene Matyushkin aka Skipy
 * @since 26.12.12
 */
public class MultipleSelectionComponentTest extends JFrame {
 
    MultipleSelectionComponentTest() {
        super("Multiple selection component test");
        JPanel cp = new JPanel(new BorderLayout());
        cp.setBorder(BorderFactory.createEmptyBorder(5, 5, 5, 5));
        setContentPane(cp);
        cp.add(new MultipleSelectionComponent<>(new ElementListSourceImpl(), ElementImpl.class), BorderLayout.NORTH);
        setSize(600, 300);
        JButton close = new JButton("Close");
        close.addActionListener(new ActionListener() {
            @Override
            public void actionPerformed(ActionEvent e) {
                System.exit(0);
            }
        });
        setDefaultCloseOperation(WindowConstants.EXIT_ON_CLOSE);
        JPanel bp = new JPanel(new FlowLayout(FlowLayout.CENTER));
        bp.add(close);
        cp.add(bp, BorderLayout.SOUTH);
        setLocationRelativeTo(null);
    }
 
 
    public static void main(String[] args) {
        new MultipleSelectionComponentTest().setVisible(true);
    }
 
}
 
class ElementListSourceImpl implements ElementListSource<ElementImpl> {
 
    private java.util.List<ElementImpl> elements;
 
    public ElementListSourceImpl() {
        elements = new ArrayList<>();
        elements.add(new ElementImpl("ya_paul"));
        elements.add(new ElementImpl("Skipy"));
        elements.add(new ElementImpl("kamit"));
        elements.add(new ElementImpl("mutagen"));
        elements.add(new ElementImpl("turbanoff"));
        elements.add(new ElementImpl("Drygba"));
        elements.add(new ElementImpl("exiga"));
        elements.add(new ElementImpl("Vortal"));
        elements.add(new ElementImpl("MILAN"));
        elements.add(new ElementImpl("lemegeton"));
    }
 
    @Override
    public List<ElementImpl> getElements() {
        return elements;
    }
}
 
class ElementImpl implements SelectableElement {
 
    private String name;
 
    public ElementImpl(String name) {
        this.name = name;
    }
 
    @Override
    public String getElementDescription() {
        return name;
    }
 
    @Override
    public Image getElementImage() {
        return null;
    }
 
    @Override
    public int hashCode() {
        return name.hashCode();
    }
 
    @Override
    public boolean equals(Object obj) {
        if (!(obj instanceof ElementImpl))
            return false;
        ElementImpl ei = (ElementImpl) obj;
        return name.equals(ei.name);
    }
}
 
class MultipleSelectionComponent<Element extends SelectableElement> extends JPanel implements ActionListener {
 
    private JPanel elementsPanel;
 
    private Set<Element> elements;
 
    private ElementListSource<Element> els;
    private Class<Element> cls;
 
    public MultipleSelectionComponent(ElementListSource<Element> els, Class<Element> cls) {
        super(new BorderLayout(3, 3));
        this.els = els;
        this.cls = cls;
        JButton addElementsButton = new JButton();
        addElementsButton.setIcon(new ImageIcon(getClass().getResource("/images/add.png")));
        add(addElementsButton, BorderLayout.LINE_END);
        elementsPanel = new JPanel(new FlowLayout(FlowLayout.LEADING));
        add(elementsPanel, BorderLayout.CENTER);
        setBorder(BorderFactory.createCompoundBorder(BorderFactory.createLineBorder(Color.darkGray, 1),
                BorderFactory.createEmptyBorder(3, 3, 3, 3)));
        addElementsButton.addActionListener(this);
        elements = new HashSet<>();
    }
 
    @Override
    public void actionPerformed(ActionEvent e) {
        ElementSelector<Element> es = new ElementSelector<>(els, cls);
        int result = JOptionPane.showInternalConfirmDialog(this, es, "Select elements",
                JOptionPane.OK_CANCEL_OPTION, JOptionPane.INFORMATION_MESSAGE);
        if (result == JOptionPane.OK_OPTION) {
            java.util.List<Element> elts = es.getSelectedElements();
            if (elts != null) {
                for (Element elt : elts) {
                    if (!elements.contains(elt)) {
                        elements.add(elt);
                        elementsPanel.add(new ElementPanel(elt));
                    }
                }
            }
        }
    }
 
    class ElementPanel extends JPanel {
 
        Element element;
 
        ElementPanel(Element element) {
            super(new BorderLayout(2, 2));
            this.element = element;
            setBorder(BorderFactory.createCompoundBorder(BorderFactory.createRaisedBevelBorder(),
                    BorderFactory.createEmptyBorder(2, 2, 2, 2)));
            JLabel delete = new JLabel(new ImageIcon(getClass().getResource("/images/delete.png")));
            add(delete, BorderLayout.LINE_END);
            delete.addMouseListener(new MouseAdapter() {
                @Override
                public void mouseClicked(MouseEvent e) {
                    elements.remove(ElementPanel.this.element);
                    elementsPanel.remove(ElementPanel.this);
                    elementsPanel.repaint();
                }
            });
            JLabel lbl = new JLabel(element.getElementDescription());
            if (element.getElementImage() != null) {
                lbl.setIcon(new ImageIcon(element.getElementImage()));
            }
            add(lbl, BorderLayout.LINE_START);
        }
    }
 
}
 
class ElementPanelRenderer<Element extends SelectableElement> extends JLabel implements ListCellRenderer<Element> {
 
    public ElementPanelRenderer() {
        setOpaque(true);
    }
 
    @Override
    public Component getListCellRendererComponent(JList<? extends Element> jList, Element value, int index, boolean isSelected, boolean cellHasFocus) {
        setText(value.getElementDescription());
        if (value.getElementImage() != null) {
            setIcon(new ImageIcon(value.getElementImage()));
        }
        setBackground(isSelected ? SystemColor.textHighlight : SystemColor.window);
        setBorder(cellHasFocus ? BorderFactory.createStrokeBorder(
                new BasicStroke(1, BasicStroke.CAP_SQUARE, BasicStroke.JOIN_ROUND, 1, new float[]{1, 1}, 1)) : null);
        return this;
    }
}
 
class ElementSelector<Element extends SelectableElement> extends JPanel {
 
    private JList<Element> elementsList;
 
    public ElementSelector(ElementListSource<Element> els, Class<Element> cls) {
        super(new BorderLayout());
        java.util.List<Element> elts = els.getElements();
        elementsList = new JList<>(elts.toArray((Element[]) Array.newInstance(cls, elts.size())));
        elementsList.setSelectionMode(ListSelectionModel.MULTIPLE_INTERVAL_SELECTION);
        elementsList.setCellRenderer(new ElementPanelRenderer<Element>());
        add(new JScrollPane(elementsList), BorderLayout.CENTER);
    }
 
    public java.util.List<Element> getSelectedElements() {
        return elementsList.getSelectedValuesList();
    }
}
 
interface ElementListSource<Element extends SelectableElement> {
 
    public java.util.List<Element> getElements();
 
}
 
interface SelectableElement {
 
    public String getElementDescription();
 
    public Image getElementImage();
 
}

    @Override
    public void actionPerformed(ActionEvent e) {
        ElementSelector<Element> es = new ElementSelector<>(els, cls);
        int result = JOptionPane.showConfirmDialog(this, es, "Select elements",
                JOptionPane.OK_CANCEL_OPTION, JOptionPane.INFORMATION_MESSAGE);
        if (result == JOptionPane.OK_OPTION) {
            java.util.List<Element> elts = es.getSelectedElements();
            if (elts != null) {
                for (Element elt : elts) {
                    if (!elements.contains(elt)) {
                        elements.add(elt);
                        elementsPanel.add(new ElementPanel(elt));
                    }
                }
            }
            doLayout();
            validate();
        }
    }

                public void mouseClicked(MouseEvent e) {
                    elements.remove(ElementPanel.this.element);
                    elementsPanel.remove(ElementPanel.this);
                    elementsPanel.doLayout();
                    elementsPanel.repaint();
                }