DisplayPanel.java

package au.gov.amsa.geo.distance;

import java.awt.Color;
import java.awt.Dimension;
import java.awt.FlowLayout;
import java.awt.Graphics;
import java.awt.Graphics2D;
import java.awt.event.ComponentEvent;
import java.awt.event.KeyEvent;
import java.awt.event.KeyListener;
import java.awt.event.MouseEvent;
import java.awt.event.MouseListener;
import java.io.File;

import javax.swing.JFrame;
import javax.swing.JPanel;
import javax.swing.SwingUtilities;

import org.apache.log4j.Logger;

import rx.Observable;
import au.gov.amsa.geo.distance.DistanceTravelledCalculator.CalculationResult;
import au.gov.amsa.geo.model.Bounds;
import au.gov.amsa.geo.model.Options;

/**
 * Swing application that allows viewing of traffic density plots. Supports
 * zooming via double-click.
 *
 */
public class DisplayPanel extends JPanel {

	private static final long serialVersionUID = 7844558863774822599L;

	private static Logger log = Logger.getLogger(DisplayPanel.class);

	private Options options;
	private int numStandardDeviations;

	private CalculationResult calculationResult;

	public DisplayPanel(int numberStandardDeviations, final CellsUpdater updater) {
		this.numStandardDeviations = numberStandardDeviations;
		setPreferredSize(java.awt.Toolkit.getDefaultToolkit().getScreenSize());
		setBackground(Color.white);
		setFocusable(true);
		addKeyListener(createKeyListener());
		addMouseListener(createMouseListener(updater));
	}

	public void setCalculationResult(CalculationResult calculationResult,
			Options options) {
		this.calculationResult = calculationResult;
		this.options = options;
		repaint();
	}

	@Override
	protected void paintComponent(Graphics gOld) {
		super.paintComponent(gOld);
		Graphics2D g = (Graphics2D) gOld;
		Renderer.paintAll(g, options, numStandardDeviations, getWidth(),
				getHeight(), calculationResult, true, true);
	}

	public static interface CellsUpdater {

		void update(DisplayPanel display, Bounds bounds);
	}

	private KeyListener createKeyListener() {
		return new KeyListener() {

			@Override
			public void keyTyped(KeyEvent e) {

			}

			@Override
			public void keyPressed(KeyEvent e) {
				int keyCode = e.getKeyCode();
				if (keyCode == 49) {
					numStandardDeviations = 1;
					repaint();
				} else if (keyCode == 50) {
					numStandardDeviations = 2;
					repaint();
				} else if (keyCode == 51) {
					numStandardDeviations = 3;
					repaint();
				}
			}

			@Override
			public void keyReleased(KeyEvent e) {

			}
		};
	}

	private MouseListener createMouseListener(final CellsUpdater updater) {
		return new MouseListener() {

			@Override
			public void mouseClicked(final MouseEvent e) {
				if (e.getClickCount() == 2) {
					log.info("double clicked");
					Thread t = new Thread(new Runnable() {

						@Override
						public void run() {
							log.info("updating cells");
							double propX = (double) e.getX() / getWidth();
							double propY = (double) e.getY() / getHeight();
							double zoomFactor = 0.5;
							Bounds b = options.getBounds();
							double topLeftLat = b.getTopLeftLat() - propY
									* (b.getHeightDegrees()) + zoomFactor
									* b.getHeightDegrees() / 2;
							double topLeftLon = b.getTopLeftLon() + propX
									* b.getWidthDegrees() - zoomFactor
									* b.getWidthDegrees() / 2;
							double bottomRightLat = topLeftLat - zoomFactor
									* b.getHeightDegrees();
							double bottomRightLon = topLeftLon + zoomFactor
									* b.getWidthDegrees();
							updater.update(DisplayPanel.this, new Bounds(
									topLeftLat, topLeftLon, bottomRightLat,
									bottomRightLon));
							repaint();
						}
					});
					t.start();
				}
			}

			@Override
			public void mousePressed(MouseEvent e) {

			}

			@Override
			public void mouseReleased(MouseEvent e) {

			}

			@Override
			public void mouseEntered(MouseEvent e) {

			}

			@Override
			public void mouseExited(MouseEvent e) {

			}
		};
	}

	public void run() {
		SwingUtilities.invokeLater(new Runnable() {

			@Override
			public void run() {
				final JFrame f = new JFrame("Vessel Traffic Density");
				// Sets the behavior for when the window is closed
				f.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
				// Add a layout manager so that the button is not placed on top
				// of the
				// label
				f.setLayout(new FlowLayout());
				// // Add a label and a button
				// f.add(new JLabel("Hello, world!"));
				// f.add(new JButton("Press me!"));
				f.add(DisplayPanel.this);
				// Arrange the components inside the window
				f.pack();
				// By default, the window is not visible. Make it visible.
				f.setVisible(true);

				f.addComponentListener(new java.awt.event.ComponentAdapter() {
					@Override
					public void componentResized(ComponentEvent e) {
						DisplayPanel.this.setPreferredSize(new Dimension(f
								.getWidth(), f.getHeight()));
					}
				});
			}
		});

	}

	public static void displayGui(final Observable<File> files,
			final Options options, final CalculationResult calculationResult) {
		final DisplayPanel display = new DisplayPanel(2, new CellsUpdater() {
			@Override
			public void update(DisplayPanel display, Bounds bounds) {
				// decrease cell size by 0.8 beyond the normal proportional
				// decrease
				double cellSizeDegrees = bounds.getWidthDegrees()
						/ options.getBounds().getWidthDegrees()
						* options.getCellSizeDegreesAsDouble() * 0.8;
				Options o2 = Options.builder().cellSizeDegrees(cellSizeDegrees)
						.bounds(bounds).build();
				display.setCalculationResult(DistanceTravelledCalculator
						.calculateTrafficDensity(o2, files), o2);
			}
		});
		display.setCalculationResult(calculationResult, options);
		display.run();
	}

}