altoslib/altosui: Further AltosState transition work
[fw/altos] / altoslib / AltosEepromFile.java
1 /*
2  * Copyright © 2013 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_1;
19
20 import java.io.*;
21 import java.util.*;
22 import java.text.*;
23
24 class AltosEepromIterator implements Iterator<AltosState> {
25         AltosState              state;
26         Iterator<AltosEeprom>   body;
27         AltosEeprom             next;
28         boolean                 seen;
29
30         public boolean hasNext() {
31                 return !seen || body.hasNext();
32         }
33
34         public AltosState next() {
35                 if (seen) {
36                         AltosState      n = state.clone();
37                         AltosEeprom     e = body.next();
38
39                         e.update_state(n);
40                         state = n;
41                 }
42                 seen = true;
43                 return state;
44         }
45
46         public void remove () {
47         }
48
49         public AltosEepromIterator(AltosState start, Iterator<AltosEeprom> body) {
50                 this.state = start;
51                 this.body = body;
52                 this.seen = false;
53         }
54 }
55
56 public class AltosEepromFile extends AltosStateIterable {
57
58         AltosEepromIterable     headers;
59         AltosEepromIterable     body;
60         AltosState              start;
61
62         public void write_comments(PrintStream out) {
63                 headers.write(out);
64         }
65
66         public void write(PrintStream out) {
67                 headers.write(out);
68                 body.write(out);
69         }
70
71         public AltosEepromFile(FileInputStream input) {
72                 headers = new AltosEepromIterable(AltosEepromHeader.read(input));
73
74                 start = headers.state();
75                 start.set_state(AltosLib.ao_flight_pad);
76
77                 switch (start.log_format) {
78                 case AltosLib.AO_LOG_FORMAT_FULL:
79                         body = new AltosEepromIterable(AltosEepromTM.read(input));
80                         break;
81                 case AltosLib.AO_LOG_FORMAT_TINY:
82                 case AltosLib.AO_LOG_FORMAT_TELEMETRY:
83                 case AltosLib.AO_LOG_FORMAT_TELESCIENCE:
84                 case AltosLib.AO_LOG_FORMAT_TELEMEGA:
85                         body = new AltosEepromIterable(AltosEepromMega.read(input));
86                         break;
87                 case AltosLib.AO_LOG_FORMAT_TELEMETRUM:
88                         body = new AltosEepromIterable(AltosEepromMetrum2.read(input));
89                         break;
90                 case AltosLib.AO_LOG_FORMAT_TELEMINI:
91                 case AltosLib.AO_LOG_FORMAT_EASYMINI:
92                         body = new AltosEepromIterable(AltosEepromMini.read(input));
93                         break;
94                 }
95
96                 /* Find boost tick */
97                 AltosState      state = start.clone();
98                 for (AltosEeprom eeprom : body) {
99                         eeprom.update_state(state);
100                         if (state.state >= AltosLib.ao_flight_boost) {
101                                 start.set_boost_tick(state.tick);
102                                 break;
103                         }
104                 }
105         }
106
107         public Iterator<AltosState> iterator() {
108                 AltosState              state = start.clone();
109                 Iterator<AltosEeprom>   i = body.iterator();
110
111                 while (i.hasNext() && !state.valid())
112                         i.next().update_state(state);
113                 return new AltosEepromIterator(state, i);
114         }
115 }