Switch from GPLv2 to GPLv2+
[fw/altos] / altoslib / AltosTelemetryIterable.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; either version 2 of the License, or
7  * (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful, but
10  * WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License along
15  * with this program; if not, write to the Free Software Foundation, Inc.,
16  * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.
17  */
18
19 package org.altusmetrum.altoslib_11;
20
21 import java.io.*;
22 import java.util.*;
23 import java.text.*;
24
25 class AltosTelemetryOrdered implements Comparable<AltosTelemetryOrdered> {
26         AltosTelemetry  telem;
27         int             index;
28         int             tick;
29
30         public int compareTo(AltosTelemetryOrdered o) {
31                 int     tick_diff = tick - o.tick;
32
33                 if (tick_diff != 0)
34                         return tick_diff;
35                 return index - o.index;
36         }
37
38         AltosTelemetryOrdered (AltosTelemetry telem, int index, int tick) {
39                 this.telem = telem;
40                 this.index = index;
41                 this.tick = tick;
42         }
43 }
44
45 class AltosTelemetryOrderedIterator implements Iterator<AltosTelemetry> {
46         Iterator<AltosTelemetryOrdered> iterator;
47
48         public AltosTelemetryOrderedIterator(TreeSet<AltosTelemetryOrdered> telems) {
49                 iterator = telems.iterator();
50         }
51
52         public boolean hasNext() {
53                 return iterator.hasNext();
54         }
55
56         public AltosTelemetry next() {
57                 return iterator.next().telem;
58         }
59
60         public void remove () {
61         }
62 }
63
64 public class AltosTelemetryIterable implements Iterable<AltosTelemetry> {
65         TreeSet<AltosTelemetryOrdered>  telems;
66         int tick;
67         int index;
68
69         public void add (AltosTelemetry telem) {
70                 int     t = telem.tick;
71                 if (!telems.isEmpty()) {
72                         while (t < tick - 1000)
73                                 t += 65536;
74                 }
75                 tick = t;
76                 telems.add(new AltosTelemetryOrdered(telem, index++, tick));
77         }
78
79         public Iterator<AltosTelemetry> iterator () {
80                 return new AltosTelemetryOrderedIterator(telems);
81         }
82
83         public AltosTelemetryIterable (FileInputStream input) {
84                 telems = new TreeSet<AltosTelemetryOrdered> ();
85                 tick = 0;
86                 index = 0;
87
88                 try {
89                         for (;;) {
90                                 String line = AltosLib.gets(input);
91                                 if (line == null) {
92                                         break;
93                                 }
94                                 try {
95                                         AltosTelemetry telem = AltosTelemetry.parse(line);
96                                         if (telem == null)
97                                                 break;
98                                         add(telem);
99                                 } catch (ParseException pe) {
100                                         System.out.printf("parse exception %s\n", pe.getMessage());
101                                 } catch (AltosCRCException ce) {
102                                 }
103                         }
104                 } catch (IOException io) {
105                         System.out.printf("io exception\n");
106                 }
107         }
108 }