/*
* Freeplane - mind map editor
* Copyright (C) 2008 Dimitry Polivaev
*
* This file author is Dimitry Polivaev
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package org.freeplane.core.util.collection;
import java.util.Iterator;
import java.util.SortedMap;
import java.util.TreeMap;
import javax.swing.AbstractListModel;
import javax.swing.ComboBoxModel;
/**
* @author Dimitry Polivaev
*/
public class SortedComboBoxModel extends AbstractListModel implements ComboBoxModel, IListModel, Iterable<Object> {
static private class Comparator implements Comparable<Object>{
final private Object obj;
private Comparator(Object obj) {
this.obj = obj;
}
public int compareTo(Object o) {
return compareTo((Comparator)o);
}
private int compareTo(Comparator o) {
final int stringCompare = obj.toString().compareTo(o.obj.toString());
if(stringCompare != 0)
return stringCompare;
final int typeCompare = obj.getClass().getName().compareTo(o.obj.getClass().getName());
return typeCompare;
}
@Override
public int hashCode() {
return obj.hashCode();
}
@Override
public boolean equals(Object o) {
return obj.getClass().equals(o.getClass()) && obj.equals(((Comparator)o).obj);
}
@Override
public String toString() {
return obj.toString();
}
}
private static final long serialVersionUID = 1L;
private Object selectedItem;
private final SortedMap<Comparator, Object> model;
public SortedComboBoxModel() {
model = new TreeMap<Comparator, Object>();
}
public void add(final Object element) {
if(addImpl(element))
fireContentsChanged(this, 0, getSize());
}
private boolean addImpl(final Object element) {
final Comparator key = key(element);
if(model.containsKey(key))
return false;
model.put(key, element);
return true;
}
private Comparator key(Object o){
return new Comparator(o);
}
public void addAll(final Object elements[]) {
for(Object e : elements)
addImpl(e);
fireContentsChanged(this, 0, getSize());
}
public void clear() {
final int oldSize = getSize();
if (oldSize > 0) {
model.clear();
fireIntervalRemoved(this, 0, oldSize - 1);
}
}
public boolean contains(final Object element) {
return model.containsKey(key(element));
}
public Object firstElement() {
return model.get(model.firstKey());
}
public Object getElementAt(final int index) {
return model.values().toArray()[index];
}
/**
*/
public int getIndexOf(final Object o) {
int count = -1;
for (final Object element : this) {
count++;
if (element.equals(o)) {
return count;
}
}
return -1;
}
public int getSize() {
return model.size();
}
public Iterator<Object> iterator() {
return model.values().iterator();
}
public Object lastElement() {
return model.get(model.lastKey());
}
/*
* (non-Javadoc)
* @see
* freeplane.controller.filter.util.SortedListModel#delete(java.lang.Object)
*/
public void remove(final Object element) {
if (null != model.remove(key(element))) {
fireContentsChanged(this, 0, getSize());
}
}
/*
* (non-Javadoc)
* @see
* freeplane.controller.filter.util.SortedListModel#replace(java.lang.Object,
* java.lang.Object)
*/
public void replace(final Object oldO, final Object newO) {
if (oldO.equals(newO)) {
return;
}
final boolean removed = null != model.remove(key(oldO));
final boolean added = addImpl(newO);
if (removed || added) {
fireContentsChanged(this, 0, getSize());
}
}
/*
* (non-Javadoc)
* @see javax.swing.ComboBoxModel#getSelectedItem()
*/
public Object getSelectedItem() {
return selectedItem;
}
/*
* (non-Javadoc)
* @see javax.swing.ComboBoxModel#setSelectedItem(java.lang.Object)
*/
public void setSelectedItem(final Object o) {
selectedItem = o;
fireContentsChanged(this, -1, -1);
}
}