path: root/ao-tools/altosui/AltosUI.java

/*
 * Copyright © 2010 Keith Packard <keithp@keithp.com>
 *
 * 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; version 2 of the License.
 *
 * 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, write to the Free Software Foundation, Inc.,
 * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.
 */

package altosui;

import java.awt.*;
import java.awt.event.*;
import javax.swing.*;
import javax.swing.filechooser.FileNameExtensionFilter;
import javax.swing.table.AbstractTableModel;
import java.io.*;
import java.util.*;
import java.text.*;
import gnu.io.CommPortIdentifier;

import altosui.AltosSerial;
import altosui.AltosSerialMonitor;
import altosui.AltosTelemetry;
import altosui.AltosState;

class AltosUIMonitor implements AltosSerialMonitor {
	public void data(String data) {
		System.out.println(data);
	}
}

class AltosFlightStatusTableModel extends AbstractTableModel {
	private String[] columnNames = {"Height (m)", "State", "RSSI (dBm)", "Speed (m/s)" };
	private Object[] data = { 0, "idle", 0, 0 };

	public int getColumnCount() { return columnNames.length; }
	public int getRowCount() { return 1; }
	public String getColumnName(int col) { return columnNames[col]; }
	public Object getValueAt(int row, int col) { return data[col]; }

	public void setValueAt(Object value, int col) {
		data[col] = value;
		fireTableCellUpdated(0, col);
	}

	public void setValueAt(Object value, int row, int col) {
		setValueAt(value, col);
	}

	public void set(AltosState state) {
		setValueAt(state.height, 0);
		setValueAt(state.data.state, 1);
		setValueAt(state.data.rssi, 2);
		setValueAt(state.speed, 3);
	}
}

public class AltosUI extends JFrame {
	private int channel = -1;

	private JTable flightStatus;
	private JTable flightInfo;
	private AltosSerial serialLine;

	public AltosUI() {

		String[] statusNames = { "Height (m)", "State", "RSSI (dBm)", "Speed (m/s)" };
		Object[][] statusData = { { "0", "pad", "-50", "0" } };

		flightStatus = new JTable(statusData, statusNames);

		flightStatus.setShowGrid(false);

		this.add(flightStatus);

		setTitle("AltOS");

		createMenu();

		serialLine = new AltosSerial();
		serialLine.monitor(new AltosUIMonitor());
		int dpi = Toolkit.getDefaultToolkit().getScreenResolution();
		this.setSize(new Dimension (dpi * 5, dpi * 4));
		this.validate();
		setDefaultCloseOperation(JFrame.DO_NOTHING_ON_CLOSE);
		addWindowListener(new WindowAdapter() {
			@Override
			public void windowClosing(WindowEvent e) {
				System.exit(0);
			}
		});
	}

	final JFileChooser deviceChooser = new JFileChooser();
	final JFileChooser logdirChooser = new JFileChooser();
	final String logdirName = "TeleMetrum";
	File logdir = null;

	private void setLogdir() {
		if (logdir == null)
			logdir = new File(logdirChooser.getCurrentDirectory(), logdirName);
		logdirChooser.setCurrentDirectory(logdir);
	}

	private void makeLogdir() {
		setLogdir();
		if (!logdir.exists()) {
			if (!logdir.mkdirs())
				JOptionPane.showMessageDialog(AltosUI.this,
							      logdir.getName(),
							      "Cannot create directory",
							      JOptionPane.ERROR_MESSAGE);
		} else if (!logdir.isDirectory()) {
			JOptionPane.showMessageDialog(AltosUI.this,
						      logdir.getName(),
						      "Is not a directory",
						      JOptionPane.ERROR_MESSAGE);
		}
	}

	private void PickSerialDevice() {
		java.util.Enumeration<CommPortIdentifier> port_list = CommPortIdentifier.getPortIdentifiers();
		while (port_list.hasMoreElements()) {
			CommPortIdentifier identifier = port_list.nextElement();
			System.out.println("Serial port " + identifier.getName());
		}
	}

	private void ConnectToDevice() {
		PickSerialDevice();
		int returnVal = deviceChooser.showOpenDialog(AltosUI.this);

		if (returnVal == JFileChooser.APPROVE_OPTION) {
			File file = deviceChooser.getSelectedFile();
			try {
				serialLine.open(file);
			} catch (FileNotFoundException ee) {
				JOptionPane.showMessageDialog(AltosUI.this,
							      file.getName(),
							      "Cannot open serial port",
							      JOptionPane.ERROR_MESSAGE);
			}
		}
	}

	String readline(FileInputStream s) throws IOException {
		int c;
		String	line = "";

		while ((c = s.read()) != -1) {
			if (c == '\r')
				continue;
			if (c == '\n')
				return line;
			line = line + (char) c;
		}
		return null;
	}

	/*
	 * Open an existing telemetry file and replay it in realtime
	 */

	private void Replay() {
		setLogdir();
		logdirChooser.setDialogTitle("Select Telemetry File");
		logdirChooser.setFileFilter(new FileNameExtensionFilter("Telemetry file", "telem"));
		int returnVal = logdirChooser.showOpenDialog(AltosUI.this);

		if (returnVal == JFileChooser.APPROVE_OPTION) {
			File file = logdirChooser.getSelectedFile();
			if (file == null)
				System.out.println("No file selected?");
			String	filename = file.getName();
			try {
				FileInputStream	replay = new FileInputStream(file);
				String	line;

				try {
					while ((line = readline(replay)) != null) {
						try {
							AltosTelemetry	t = new AltosTelemetry(line);
							System.out.println ("Version " + t.version + t.callsign);
						} catch (ParseException pp) {
							JOptionPane.showMessageDialog(AltosUI.this,
										      line,
										      "error parsing",
										      JOptionPane.ERROR_MESSAGE);
							break;
						}
					}
				} catch (IOException ee) {
					JOptionPane.showMessageDialog(AltosUI.this,
								      filename,
								      "error reading",
								      JOptionPane.ERROR_MESSAGE);
				} finally {
					try {
						replay.close();
					} catch (IOException e) {}
				}
			} catch (FileNotFoundException ee) {
				JOptionPane.showMessageDialog(AltosUI.this,
							      filename,
							      "Cannot open serial port",
							      JOptionPane.ERROR_MESSAGE);
			}
		}
	}

	private void SaveFlightData() {
	}

	private void createMenu() {
		JMenuBar menubar = new JMenuBar();
		JMenu menu;
		JMenuItem item;
		JRadioButtonMenuItem radioitem;

		// File menu
		{
			menu = new JMenu("File");
			menu.setMnemonic(KeyEvent.VK_F);
			menubar.add(menu);

			item = new JMenuItem("Quit",KeyEvent.VK_Q);
			item.setAccelerator(KeyStroke.getKeyStroke(KeyEvent.VK_Q,
								   ActionEvent.CTRL_MASK));
			item.addActionListener(new ActionListener() {
					public void actionPerformed(ActionEvent e) {
						System.exit(0);
					}
				});
			menu.add(item);
		}

		// Device menu
		{
			menu = new JMenu("Device");
			menu.setMnemonic(KeyEvent.VK_D);
			menubar.add(menu);

			item = new JMenuItem("Connect to Device",KeyEvent.VK_C);
			item.addActionListener(new ActionListener() {
					public void actionPerformed(ActionEvent e) {
						ConnectToDevice();
					}
				});
			menu.add(item);

			item = new JMenuItem("Disconnect from Device",KeyEvent.VK_D);
			item.addActionListener(new ActionListener() {
					public void actionPerformed(ActionEvent e) {
						serialLine.close();
					}
				});
			menu.add(item);

			menu.addSeparator();

			item = new JMenuItem("Save Flight Data",KeyEvent.VK_S);
			item.addActionListener(new ActionListener() {
					public void actionPerformed(ActionEvent e) {
						SaveFlightData();
					}
				});
			menu.add(item);

			item = new JMenuItem("Replay",KeyEvent.VK_R);
			item.addActionListener(new ActionListener() {
					public void actionPerformed(ActionEvent e) {
						Replay();
					}
				});
			menu.add(item);
		}
		// Log menu
		{
			menu = new JMenu("Log");
			menu.setMnemonic(KeyEvent.VK_L);
			menubar.add(menu);

			item = new JMenuItem("New Log",KeyEvent.VK_N);
			item.addActionListener(new ActionListener() {
					public void actionPerformed(ActionEvent e) {
					}
				});
			menu.add(item);

			item = new JMenuItem("Configure Log",KeyEvent.VK_C);
			item.addActionListener(new ActionListener() {
					public void actionPerformed(ActionEvent e) {
					}
				});
			menu.add(item);
		}
		// Voice menu
		{
			menu = new JMenu("Voice", true);
			menu.setMnemonic(KeyEvent.VK_V);
			menubar.add(menu);

			radioitem = new JRadioButtonMenuItem("Enable Voice");
			radioitem.addActionListener(new ActionListener() {
					public void actionPerformed(ActionEvent e) {
					}
				});
			menu.add(radioitem);
		}

		// Channel menu
		{
			menu = new JMenu("Channel", true);
			menu.setMnemonic(KeyEvent.VK_C);
			menubar.add(menu);
			ButtonGroup group = new ButtonGroup();

			for (int c = 0; c <= 9; c++) {
				radioitem = new JRadioButtonMenuItem(String.format("Channel %1d (%7.3fMHz)", c,
										   434.550 + c * 0.1),
								     c == 0);
				radioitem.setActionCommand(String.format("%d", c));
				radioitem.addActionListener(new ActionListener() {
						public void actionPerformed(ActionEvent e) {
							System.out.println("Command: " + e.getActionCommand() + " param: " +
									   e.paramString());
						}
					});
				menu.add(radioitem);
				group.add(radioitem);
			}
		}

		this.setJMenuBar(menubar);

	}
	public static void main(final String[] args) {
		AltosUI altosui = new AltosUI();
		altosui.setVisible(true);
	}
}