88b38bb176f01ad00db321de36fb698d2c14019c
[fw/altos] / altosui / AltosSerial.java
1 /*
2  * Copyright © 2010 Keith Packard <keithp@keithp.com>
3  *
4  * This program is free software; you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation; version 2 of the License.
7  *
8  * This program is distributed in the hope that it will be useful, but
9  * WITHOUT ANY WARRANTY; without even the implied warranty of
10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
11  * General Public License for more details.
12  *
13  * You should have received a copy of the GNU General Public License along
14  * with this program; if not, write to the Free Software Foundation, Inc.,
15  * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.
16  */
17
18 /*
19  * Deal with TeleDongle on a serial port
20  */
21
22 package altosui;
23
24 import java.lang.*;
25 import java.io.*;
26 import java.util.concurrent.*;
27 import java.util.*;
28 import java.awt.*;
29 import java.awt.event.*;
30 import javax.swing.*;
31 import javax.swing.filechooser.FileNameExtensionFilter;
32 import javax.swing.table.*;
33
34 import libaltosJNI.*;
35
36 /*
37  * This class reads from the serial port and places each received
38  * line in a queue. Dealing with that queue is left up to other
39  * threads.
40  */
41
42 public class AltosSerial implements Runnable {
43
44         static java.util.List<String> devices_opened = Collections.synchronizedList(new LinkedList<String>());
45
46         AltosDevice device;
47         SWIGTYPE_p_altos_file altos;
48         LinkedList<LinkedBlockingQueue<AltosLine>> monitors;
49         LinkedBlockingQueue<AltosLine> reply_queue;
50         Thread input_thread;
51         String line;
52         byte[] line_bytes;
53         int line_count;
54         boolean monitor_mode;
55         int telemetry;
56         int channel;
57         static boolean debug;
58         boolean remote;
59         LinkedList<String> pending_output = new LinkedList<String>();
60         Frame frame;
61
62         static void set_debug(boolean new_debug) {
63                 debug = new_debug;
64         }
65
66         public void run () {
67                 int c;
68
69                 try {
70                         for (;;) {
71                                 c = libaltos.altos_getchar(altos, 0);
72                                 if (Thread.interrupted())
73                                         break;
74                                 if (c == libaltosConstants.LIBALTOS_ERROR) {
75                                         for (int e = 0; e < monitors.size(); e++) {
76                                                 LinkedBlockingQueue<AltosLine> q = monitors.get(e);
77                                                 q.put(new AltosLine());
78                                         }
79                                         reply_queue.put (new AltosLine());
80                                         break;
81                                 }
82                                 if (c == libaltosConstants.LIBALTOS_TIMEOUT)
83                                         continue;
84                                 if (c == '\r')
85                                         continue;
86                                 synchronized(this) {
87                                         if (c == '\n') {
88                                                 if (line_count != 0) {
89                                                         try {
90                                                                 line = new String(line_bytes, 0, line_count, "UTF-8");
91                                                         } catch (UnsupportedEncodingException ue) {
92                                                                 line = "";
93                                                                 for (int i = 0; i < line_count; i++)
94                                                                         line = line + line_bytes[i];
95                                                         }
96                                                         if (debug)
97                                                                 System.out.printf("\t\t\t\t\t%s\n", line);
98                                                         if (line.startsWith("VERSION") || line.startsWith("CRC")) {
99                                                                 for (int e = 0; e < monitors.size(); e++) {
100                                                                         LinkedBlockingQueue<AltosLine> q = monitors.get(e);
101                                                                         q.put(new AltosLine (line));
102                                                                 }
103                                                         } else {
104                                                                 reply_queue.put(new AltosLine (line));
105                                                         }
106                                                         line_count = 0;
107                                                         line = "";
108                                                 }
109                                         } else {
110                                                 if (line_bytes == null) {
111                                                         line_bytes = new byte[256];
112                                                 } else if (line_count == line_bytes.length) {
113                                                         byte[] new_line_bytes = new byte[line_count * 2];
114                                                         System.arraycopy(line_bytes, 0, new_line_bytes, 0, line_count);
115                                                         line_bytes = new_line_bytes;
116                                                 }
117                                                 line_bytes[line_count] = (byte) c;
118                                                 line_count++;
119                                         }
120                                 }
121                         }
122                 } catch (InterruptedException e) {
123                 }
124         }
125
126         public void flush_output() {
127                 if (altos != null) {
128                         for (String s : pending_output)
129                                 System.out.print(s);
130                         pending_output.clear();
131                         libaltos.altos_flush(altos);
132                 }
133         }
134
135         boolean         abort;
136         JDialog         timeout_dialog;
137         boolean timeout_started = false;
138
139         private void stop_timeout_dialog() {
140                 System.out.printf("stop_timeout_dialog\n");
141                 Runnable r = new Runnable() {
142                                 public void run() {
143                                         if (timeout_dialog != null)
144                                                 timeout_dialog.setVisible(false);
145                                 }
146                         };
147                 SwingUtilities.invokeLater(r);
148         }
149
150         private void start_timeout_dialog_internal() {
151                 System.out.printf("Creating timeout dialog\n");
152                 Object[] options = { "Cancel" };
153
154                 JOptionPane     pane = new JOptionPane();
155                 pane.setMessage(String.format("Connecting to %s", device.getPath()));
156                 pane.setOptions(options);
157                 pane.setInitialValue(null);
158
159                 timeout_dialog = pane.createDialog(frame, "Connecting...");
160
161                 timeout_dialog.setVisible(true);
162
163                 Object o = pane.getValue();
164                 if (o == null)
165                         return;
166                 if (options[0].equals(o))
167                         abort = true;
168         }
169
170         private boolean check_timeout() {
171                 if (!timeout_started && frame != null) {
172                         timeout_started = true;
173                         System.out.printf("Starting timeout dialog\n");
174                         if (SwingUtilities.isEventDispatchThread()) {
175                                 start_timeout_dialog_internal();
176                         } else {
177                                 Runnable r = new Runnable() {
178                                                 public void run() {
179                                                         start_timeout_dialog_internal();
180                                                 }
181                                         };
182                                 SwingUtilities.invokeLater(r);
183                         }
184                 }
185                 return abort;
186         }
187
188         public void flush_input() {
189                 flush_output();
190                 boolean got_some;
191
192                 int timeout = 100;
193                 if (remote)
194                         timeout = 300;
195                 do {
196                         try {
197                                 Thread.sleep(timeout);
198                         } catch (InterruptedException ie) {
199                         }
200                         got_some = !reply_queue.isEmpty();
201                         synchronized(this) {
202                                 if (!"VERSION".startsWith(line) &&
203                                     !line.startsWith("VERSION"))
204                                         line = "";
205                                 reply_queue.clear();
206                         }
207                 } while (got_some);
208         }
209
210         public String get_reply() throws InterruptedException {
211                 flush_output();
212                 AltosLine line = reply_queue.take();
213                 return line.line;
214         }
215
216         public String get_reply(int timeout) throws InterruptedException {
217                 flush_output();
218                 if (remote) {
219                         timeout = 300;
220                         System.out.printf("Doing remote timout\n");
221                 }
222                 abort = false;
223                 timeout_started = false;
224                 for (;;) {
225                         AltosLine line = reply_queue.poll(timeout, TimeUnit.MILLISECONDS);
226                         if (line != null) {
227                                 stop_timeout_dialog();
228                                 return line.line;
229                         }
230                         if (!remote || check_timeout())
231                                 return null;
232                 }
233         }
234
235         public void add_monitor(LinkedBlockingQueue<AltosLine> q) {
236                 set_monitor(true);
237                 monitors.add(q);
238         }
239
240         public void remove_monitor(LinkedBlockingQueue<AltosLine> q) {
241                 monitors.remove(q);
242                 if (monitors.isEmpty())
243                         set_monitor(false);
244         }
245
246         public void close() {
247                 if (altos != null) {
248                         libaltos.altos_close(altos);
249                 }
250                 if (input_thread != null) {
251                         try {
252                                 input_thread.interrupt();
253                                 input_thread.join();
254                         } catch (InterruptedException e) {
255                         }
256                         input_thread = null;
257                 }
258                 if (altos != null) {
259                         libaltos.altos_free(altos);
260                         altos = null;
261                 }
262                 synchronized (devices_opened) {
263                         devices_opened.remove(device.getPath());
264                 }
265                 if (debug)
266                         System.out.printf("Closing %s\n", device.getPath());
267         }
268
269         private void putc(char c) {
270                 if (altos != null)
271                         libaltos.altos_putchar(altos, c);
272         }
273
274         public void print(String data) {
275                 if (debug)
276                         pending_output.add(data);
277                 for (int i = 0; i < data.length(); i++)
278                         putc(data.charAt(i));
279         }
280
281         public void printf(String format, Object ... arguments) {
282                 print(String.format(format, arguments));
283         }
284
285         private void open() throws FileNotFoundException, AltosSerialInUseException {
286                 synchronized (devices_opened) {
287                         if (devices_opened.contains(device.getPath()))
288                                 throw new AltosSerialInUseException(device);
289                         devices_opened.add(device.getPath());
290                 }
291                 altos = libaltos.altos_open(device);
292                 if (altos == null) {
293                         close();
294                         throw new FileNotFoundException(device.toShortString());
295                 }
296                 if (debug)
297                         System.out.printf("Open %s\n", device.getPath());
298                 input_thread = new Thread(this);
299                 input_thread.start();
300                 print("~\nE 0\n");
301                 set_monitor(false);
302                 flush_output();
303         }
304
305         public void set_radio() {
306                 telemetry = AltosPreferences.telemetry(device.getSerial());
307                 channel = AltosPreferences.channel(device.getSerial());
308                 set_channel(channel);
309                 set_callsign(AltosPreferences.callsign());
310         }
311
312         public void set_channel(int in_channel) {
313                 channel = in_channel;
314                 if (altos != null) {
315                         if (monitor_mode)
316                                 printf("m 0\nc r %d\nm %d\n", channel, telemetry);
317                         else
318                                 printf("c r %d\n", channel);
319                         flush_output();
320                 }
321         }
322
323         public void set_telemetry(int in_telemetry) {
324                 telemetry = in_telemetry;
325                 if (altos != null) {
326                         if (monitor_mode)
327                                 printf("m 0\nm %d\n", telemetry);
328                         flush_output();
329                 }
330         }
331
332         void set_monitor(boolean monitor) {
333                 monitor_mode = monitor;
334                 if (altos != null) {
335                         if (monitor)
336                                 printf("m %d\n", telemetry);
337                         else
338                                 printf("m 0\n");
339                         flush_output();
340                 }
341         }
342
343         public void set_callsign(String callsign) {
344                 if (altos != null) {
345                         printf ("c c %s\n", callsign);
346                         flush_output();
347                 }
348         }
349
350         public void start_remote() {
351                 if (debug)
352                         System.out.printf("start remote\n");
353                 set_radio();
354                 printf("p\nE 0\n");
355                 flush_input();
356                 remote = true;
357         }
358
359         public void stop_remote() {
360                 if (debug)
361                         System.out.printf("stop remote\n");
362                 try {
363                         flush_input();
364                 } finally {
365                         System.out.printf("Sending tilde\n");
366                         printf ("~\n");
367                         flush_output();
368                 }
369                 remote = false;
370         }
371
372         public void set_frame(Frame in_frame) {
373                 frame = in_frame;
374         }
375
376         public AltosSerial(AltosDevice in_device) throws FileNotFoundException, AltosSerialInUseException {
377                 device = in_device;
378                 line = "";
379                 monitor_mode = false;
380                 frame = null;
381                 telemetry = Altos.ao_telemetry_full;
382                 monitors = new LinkedList<LinkedBlockingQueue<AltosLine>> ();
383                 reply_queue = new LinkedBlockingQueue<AltosLine> ();
384                 open();
385         }
386 }