gdx-studio

package gdxstudio.panel;
import gdxstudio.Style;

import java.awt.Dimension;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.util.Arrays;
import java.util.Collection;
import java.util.Iterator;
import java.util.SortedSet;
import java.util.TreeSet;

import javax.swing.AbstractListModel;
import javax.swing.BorderFactory;
import javax.swing.DefaultListModel;
import javax.swing.JList;
import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.ListSelectionModel;
import javax.swing.event.ListSelectionListener;

import web.laf.lite.layout.VerticalFlowLayout;
import web.laf.lite.utils.UIUtils;

public abstract class BaseList extends JPanel implements ListSelectionListener, ActionListener {
	private static final long serialVersionUID = 1L;
	public JList<String> list;
	protected DefaultListModel<String> listModel;
	protected SortedListModel listModel2;
	protected JScrollPane scrollPane;
	protected final Style.TitleButton header;
	
	BaseList(String title){
		super(new VerticalFlowLayout());
		UIUtils.setShadeWidth(this, 0);
		UIUtils.setRound(this, 0);
		listModel = new DefaultListModel<String>();
		list = new JList<String>(listModel);
		list.setSelectionMode(ListSelectionModel.SINGLE_SELECTION);
		list.addListSelectionListener(this);
		UIUtils.setDecorateSelection(list, false);
		header = new Style.TitleButton(title, this);
		add(header);
		scrollPane = new JScrollPane(list);
		scrollPane.setPreferredSize(new Dimension(200, 175));
		UIUtils.setDrawBorder(scrollPane, false);
		add(scrollPane);
		setBorder(BorderFactory.createMatteBorder(0, 0, 1, 0, Style.border));
	}
	
	BaseList(String title, boolean dontAddScroll){
		super(new VerticalFlowLayout());
		UIUtils.setShadeWidth(this, 0);
		UIUtils.setRound(this, 0);
		listModel = new DefaultListModel<String>();
		list = new JList<String>(listModel);
		list.setSelectionMode(ListSelectionModel.SINGLE_SELECTION);
		list.addListSelectionListener(this);
		header = new Style.TitleButton(title, this);
		add(header);
		scrollPane = new JScrollPane(list);
		scrollPane.setPreferredSize(new Dimension(200, 175));
		UIUtils.setDrawBorder(scrollPane, false);
		setBorder(BorderFactory.createMatteBorder(0, 0, 1, 0, Style.border));
	}
	
	BaseList(String title, String... items){
		super(new VerticalFlowLayout());
		UIUtils.setShadeWidth(this, 0);
		UIUtils.setRound(this, 0);
		//listModel = new DefaultListModel<String>();
		list = new JList<String>(items);
		list.setSelectionMode(ListSelectionModel.SINGLE_SELECTION);
		list.addListSelectionListener(this);
		header = new Style.TitleButton(title, this);
		add(header);
		scrollPane = new JScrollPane(list);
		scrollPane.setPreferredSize(new Dimension(200, 175));
		UIUtils.setDrawBorder(scrollPane, false);
		add(scrollPane);
		setBorder(BorderFactory.createMatteBorder(0, 0, 1, 0, Style.border));
	}
	
	@SuppressWarnings("unchecked")
	BaseList(String title, String useSort){
		super(new VerticalFlowLayout());
		UIUtils.setShadeWidth(this, 0);
		UIUtils.setRound(this, 0);
		listModel2 = new  SortedListModel();
		list = new JList<String>(listModel2);
		list.setSelectionMode(ListSelectionModel.SINGLE_SELECTION);
		list.addListSelectionListener(this);
		header = new Style.TitleButton(title, this);
		add(header);
		scrollPane = new JScrollPane(list);
		scrollPane.setPreferredSize(new Dimension(200, 175));
		UIUtils.setDrawBorder(scrollPane, false);
		setBorder(BorderFactory.createMatteBorder(0, 0, 1, 0, Style.border));
	}
	
	public void enable(){
		list.setEnabled(true);
	}
	
	public void disable(){
		list.setEnabled(false);
	}
	
	public void lock(){
		list.removeListSelectionListener(this);
	}
	
	public void unlock(){
		list.addListSelectionListener(this);
	}
	
	public void clear(){
		lock();
		listModel.clear();
		unlock();
	}
	
	@Override
	public void actionPerformed(ActionEvent arg0) {
		setVisible(false);
		validate();
		repaint();
	}
}

@SuppressWarnings("rawtypes")
class SortedListModel extends AbstractListModel {
	private static final long serialVersionUID = 1L;
	SortedSet<Object> model;

	public SortedListModel() {
		model = new TreeSet<Object>();
	}

	public int getSize() {
		return model.size();
	}

	public Object getElementAt(int index) {
		return model.toArray()[index];
	}
	
	public int indexOf(Object element){
		int index = 0;
		for(Object e: model.toArray()){
			if(e.equals(element))
				break;
			index++;
		}
		return index;
	}
	

	public void addElement(Object element) {
		if (model.add(element)) {
			fireContentsChanged(this, 0, getSize());
		}
	}
	public void addAll(Object elements[]) {
		Collection<Object> c = Arrays.asList(elements);
		model.addAll(c);
		fireContentsChanged(this, 0, getSize());
	}

	public void clear() {
		model.clear();
		fireContentsChanged(this, 0, getSize());
	}

	public boolean contains(Object element) {
		return model.contains(element);
	}

	public Object firstElement() {
		return model.first();
	}

	public Iterator iterator() {
		return model.iterator();
	}

	public Object lastElement() {
		return model.last();
	}

	public boolean removeElement(Object element) {
		boolean removed = model.remove(element);
		if (removed) {
			fireContentsChanged(this, 0, getSize());
		}
		return removed;
	}
}