Change AltosLib to altoslib
[fw/altos] / altoslib / AltosLog.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 package org.altusmetrum.altoslib;
19
20 import java.io.*;
21 import java.text.ParseException;
22 import java.util.concurrent.LinkedBlockingQueue;
23
24 /*
25  * This creates a thread to capture telemetry data and write it to
26  * a log file
27  */
28 public class AltosLog implements Runnable {
29
30         LinkedBlockingQueue<AltosLine>  input_queue;
31         LinkedBlockingQueue<String>     pending_queue;
32         int                             serial;
33         int                             flight;
34         FileWriter                      log_file;
35         Thread                          log_thread;
36         AltosFile                       file;
37
38         private void close_log_file() {
39                 if (log_file != null) {
40                         try {
41                                 log_file.close();
42                         } catch (IOException io) {
43                         }
44                         log_file = null;
45                 }
46         }
47
48         public void close() {
49                 close_log_file();
50                 if (log_thread != null) {
51                         log_thread.interrupt();
52                         log_thread = null;
53                 }
54         }
55
56         public File file() {
57                 return file;
58         }
59
60         boolean open (AltosRecord telem) throws IOException {
61                 AltosFile       a = new AltosFile(telem);
62
63                 log_file = new FileWriter(a, true);
64                 if (log_file != null) {
65                         while (!pending_queue.isEmpty()) {
66                                 try {
67                                         String s = pending_queue.take();
68                                         log_file.write(s);
69                                         log_file.write('\n');
70                                 } catch (InterruptedException ie) {
71                                 }
72                         }
73                         log_file.flush();
74                         file = a;
75                 }
76                 return log_file != null;
77         }
78
79         public void run () {
80                 try {
81                         AltosRecord     previous = null;
82                         for (;;) {
83                                 AltosLine       line = input_queue.take();
84                                 if (line.line == null)
85                                         continue;
86                                 try {
87                                         AltosRecord     telem = AltosTelemetry.parse(line.line, previous);
88                                         if ((telem.seen & AltosRecord.seen_flight) != 0 &&
89                                             (telem.serial != serial || telem.flight != flight || log_file == null))
90                                         {
91                                                 close_log_file();
92                                                 serial = telem.serial;
93                                                 flight = telem.flight;
94                                                 open(telem);
95                                         }
96                                         previous = telem;
97                                 } catch (ParseException pe) {
98                                 } catch (AltosCRCException ce) {
99                                 }
100                                 if (log_file != null) {
101                                         log_file.write(line.line);
102                                         log_file.write('\n');
103                                         log_file.flush();
104                                 } else
105                                         pending_queue.put(line.line);
106                         }
107                 } catch (InterruptedException ie) {
108                 } catch (IOException ie) {
109                 }
110                 close();
111         }
112
113         public AltosLog (AltosLink link) {
114                 pending_queue = new LinkedBlockingQueue<String> ();
115                 input_queue = new LinkedBlockingQueue<AltosLine> ();
116                 link.add_monitor(input_queue);
117                 serial = -1;
118                 flight = -1;
119                 log_file = null;
120                 log_thread = new Thread(this);
121                 log_thread.start();
122         }
123 }