path: root/dozentenmodul/src/main/java/org/openslx/dozmod/gui/control/RoomSelector.java

package org.openslx.dozmod.gui.control;

import java.awt.Component;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.MouseAdapter;
import java.awt.event.MouseEvent;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Enumeration;
import java.util.List;

import javax.swing.DefaultListCellRenderer;
import javax.swing.DefaultListModel;
import javax.swing.JButton;
import javax.swing.JList;
import javax.swing.JPanel;
import javax.swing.JScrollPane;

import org.apache.log4j.Logger;
import org.openslx.bwlp.thrift.iface.Location;
import org.openslx.dozmod.gui.Gui;
import org.openslx.dozmod.gui.helper.GridManager;
import org.openslx.dozmod.thrift.cache.MetaDataCache;
import org.openslx.util.QuickTimer;
import org.openslx.util.QuickTimer.Task;

public class RoomSelector extends QLabel {
	private final static Logger LOGGER = Logger.getLogger(RoomSelector.class);

	private DefaultListModel<Location> roomModel = new DefaultListModel<Location>();
	private DefaultListModel<Location> selectedRoomModel = new DefaultListModel<Location>();

	private JButton addAllButton = new JButton(">>");
	private JButton addButton = new JButton(">");
	private JButton removeButton = new JButton("<");
	private JButton removeAllButton = new JButton("<<");

	private boolean initDone = false;
	private List<Integer> preselection;

	public RoomSelector() {
		GridManager grid = new GridManager(this, 3);
		LocationRenderer locationRenderer = new LocationRenderer();

		// the available list of rooms
		final JList<Location> roomList = new JList<Location>();
		roomList.setCellRenderer(locationRenderer);
		roomList.setModel(roomModel);

		// the selected list of rooms
		final JList<Location> selectedRoomList = new JList<Location>();
		selectedRoomList.setCellRenderer(locationRenderer);
		selectedRoomList.setModel(selectedRoomModel);

		// the listeners
		selectedRoomList.addMouseListener(new MouseAdapter() {
			public void mouseClicked(MouseEvent e) {
				if (e.getClickCount() == 2) {
					Location selection = selectedRoomList.getSelectedValue();
					if (selection != null) {
						// LOGGER.debug("Removing: " + selection);
						if (selectedRoomModel.contains(selection)) {
							selectedRoomModel.removeElement(selection);
							if (!roomModel.contains(selection)) {
								roomModel.addElement(selection);
							}
							sortLists();
						}
					}
				}
			}
		});

		roomList.addMouseListener(new MouseAdapter() {
			public void mouseClicked(MouseEvent e) {
				if (e.getClickCount() == 2) {
					Location selection = roomList.getSelectedValue();
					if (selection != null) {
						// LOGGER.debug("Adding: " + selection);
						if (!selectedRoomModel.contains(selection)) {
							selectedRoomModel.addElement(selection);
							roomModel.removeElement(selection);
						}
						sortLists();
					}
				}
			}
		});

		addAllButton.addActionListener(new ActionListener() {
			@Override
			public void actionPerformed(ActionEvent e) {
				moveAll(roomModel, selectedRoomModel);
			}
		});
		addButton.addActionListener(new ActionListener() {
			@Override
			public void actionPerformed(ActionEvent e) {
				addToSelection(roomList.getSelectedValuesList());

			}
		});
		removeButton.addActionListener(new ActionListener() {
			@Override
			public void actionPerformed(ActionEvent e) {
				removeFromSelection(selectedRoomList.getSelectedValuesList());

			}
		});
		removeAllButton.addActionListener(new ActionListener() {
			@Override
			public void actionPerformed(ActionEvent e) {
				moveAll(selectedRoomModel, roomModel);
			}
		});

		grid.add(new QLabel("Verfügbare Räume"));
		grid.skip();
		grid.add(new QLabel("Ausgewählte Räume"));
		grid.nextRow();
		grid.add(new JScrollPane(roomList)).fill(true, true).expand(true, true);
		// Vertical BoxLayout for buttons
		JPanel buttonPanel = new JPanel();
		GridManager buttonGrid = new GridManager(buttonPanel, 1);
		buttonGrid.add(addAllButton).fill(true, false);
		buttonGrid.nextRow();
		buttonGrid.add(addButton).fill(true, false);
		buttonGrid.nextRow();
		buttonGrid.add(removeButton).fill(true, false);
		buttonGrid.nextRow();
		buttonGrid.add(removeAllButton).fill(true, false);
		buttonGrid.nextRow();
		buttonGrid.finish(true);
		grid.add(buttonPanel).fill(false, false).expand(false, false);
		grid.add(new JScrollPane(selectedRoomList)).fill(true, true)
				.expand(true, true);
		grid.nextRow();
		grid.finish(true);

		// initialise the data
		init();
	}

	public void init() {
		QuickTimer.scheduleOnce(new Task() {
			List<Location> locsList = null;

			@Override
			public void fire() {
				locsList = MetaDataCache.getLocations();
				Gui.asyncExec(new Runnable() {
					@Override
					public void run() {
						initDone = fillRoomList(locsList);
						// check if preselection was set before we were done initialising
						if (initDone && preselection != null) {
							setSelectionInternal(preselection);
						}
					}
				});
			}
		});
	}

	public boolean fillRoomList(final List<Location> locations) {
		if (locations == null) {
			LOGGER.error("No locations returned from the metadata cache.");
			return false;
		}
		for (Location loc : locations) {
			if (loc != null) {
				LOGGER.debug("Adding: " + loc.getLocationName());
				roomModel.addElement(loc);
			} else {
				LOGGER.error("One location returned from the server was null");
			}
		}
		return true;
	}

	private void moveAll(DefaultListModel<Location> from,
			DefaultListModel<Location> to) {
		for (Enumeration<Location> rooms = from.elements(); rooms
				.hasMoreElements();) {
			Location current = rooms.nextElement();
			to.addElement(current);
		}
		from.clear();
		sortLists();
	}

	// convenience
	private void addToSelection(Location location) {
		if (location == null) {
			return;
		}
		List<Location> newLocations = new ArrayList<Location>();
		newLocations.add(location);
		addToSelection(newLocations);
	}

	private void addToSelection(List<Location> locations) {
		if (locations == null)
			return;
		// LOGGER.debug("AddToSelection: " + locations);
		for (Location location : locations) {
			selectedRoomModel.addElement(location);
			roomModel.removeElement(location);
		}
		sortLists();
	}

	private void removeFromSelection(List<Location> locations) {
		if (locations == null)
			return;
		// LOGGER.debug("RemoveFromSelection: " + locations);
		for (Location location : locations) {
			selectedRoomModel.removeElement(location);
			roomModel.addElement(location);
		}
		sortLists();
	}

	private void sortLists() {
		// sort the available room list
		List<Location> list = Collections.list(roomModel.elements());
		Collections.sort(list);
		roomModel.clear();
		for (Location loc : list) {
			roomModel.addElement(loc);
		}
		// sort the selected room list
		list = Collections.list(selectedRoomModel.elements());
		Collections.sort(list);
		selectedRoomModel.clear();
		for (Location loc : list) {
			selectedRoomModel.addElement(loc);
		}
	}

	public DefaultListModel<Location> getRoomModel() {
		return roomModel;
	}

	public DefaultListModel<Location> getSelectedRoomModel() {
		return selectedRoomModel;
	}

	public void setSelectedRoomsAsIds(List<Integer> list) {
		if (list == null) {
			LOGGER.error("No list given!");
			return;
		}
		preselection = list;
		// if we are initialised, we need to explicitly set the selection
		if (initDone)
			setSelectionInternal(preselection);
	}
	private void setSelectionInternal(List<Integer> list){
		if (list == null || list.isEmpty()) {
			LOGGER.error("No list given for preselection!");
			return;
		}
		LOGGER.debug("Setting to: " + preselection);
		moveAll(selectedRoomModel, roomModel);

		for (Integer id : preselection) {
			LOGGER.debug("Searching: " + id);
			Location loc = MetaDataCache
					.getLocationById(id);
			LOGGER.debug("Found: " + loc);
			if (loc != null) {
				addToSelection(loc);
				LOGGER.debug("Added: " + loc);
			}
		}
	}
}

@SuppressWarnings("serial")
class LocationRenderer extends DefaultListCellRenderer {
	@Override
	public Component getListCellRendererComponent(JList<? extends Object> list,
			Object value, int index, boolean isSelected, boolean cellHasFocus) {
		if (value instanceof Location) {
			return super.getListCellRendererComponent(list,
					((Location) value).getLocationName(), index, isSelected,
					cellHasFocus);
		}
		return super.getListCellRendererComponent(list, value, index,
				isSelected, cellHasFocus);
	}
}