altoslib: Create data file open helper in AltosLib

[fw/altos] / altoslib / AltosEepromRecordSet.java

diff --git a/altoslib/AltosEepromRecordSet.java b/altoslib/AltosEepromRecordSet.java
index 000d9c02a848d03b3a793ba5c9426e3341e44e56..0c60c1a5ab5b8db2196a626b3a22a822b0b55748 100644 (file)
--- a/altoslib/AltosEepromRecordSet.java
+++ b/altoslib/AltosEepromRecordSet.java
@@ -17,41 +17,39 @@ package org.altusmetrum.altoslib_11;
 import java.io.*;
 import java.util.*;
 
-public class AltosEepromRecordSet implements Iterable<AltosState> {
+public class AltosEepromRecordSet implements AltosRecordSet {
+       AltosEepromNew                  eeprom;
        TreeSet<AltosEepromRecord>      ordered;
-       AltosState                      start_state;
+       AltosCalData                    cal_data;
 
-       class RecordIterator implements Iterator<AltosState> {
-               Iterator<AltosEepromRecord> riterator;
-               AltosState state;
-               boolean started;
-
-               public boolean hasNext() {
-                       return state == null || riterator.hasNext();
-               }
+       public AltosConfigData config_data() {
+               return eeprom.config_data();
+       }
 
-               public AltosState next() {
-                       if (state == null)
-                               state = start_state.clone();
-                       else {
-                               state = state.clone();
-                               AltosEepromRecord       r = riterator.next();
-                               r.update_state(state);
+       public AltosCalData cal_data() {
+               if (cal_data == null) {
+                       cal_data = new AltosCalData(config_data());
+                       for (AltosEepromRecord record : ordered) {
+                               if (record.cmd() == AltosLib.AO_LOG_FLIGHT) {
+                                       cal_data.set_tick(record.tick());
+                                       cal_data.set_boost_tick();
+                               }
                        }
-                       return state;
-               }
-
-               public RecordIterator() {
-                       riterator = ordered.iterator();
-                       state = null;
                }
+               return cal_data;
        }
 
-       public Iterator<AltosState> iterator() {
-               return new RecordIterator();
+       public void capture_series(AltosDataListener listener) {
+               AltosCalData    cal_data = cal_data();
+               for (AltosEepromRecord record : ordered) {
+                       record.provide_data(listener, cal_data);
+               }
+               listener.finish();
        }
 
        public AltosEepromRecordSet(AltosEepromNew eeprom) {
+               this.eeprom = eeprom;
+
                AltosConfigData         config_data = eeprom.config_data();
 
                AltosEepromRecord       record = null;
@@ -93,9 +91,6 @@ public class AltosEepromRecordSet implements Iterable<AltosState> {
                int     tick = 0;
                boolean first = true;
 
-               start_state = new AltosState();
-               start_state.set_config_data(record.eeprom.config_data());
-
                for (;;) {
                        int     t = record.tick();
 
@@ -115,7 +110,7 @@ public class AltosEepromRecordSet implements Iterable<AltosState> {
                }
        }
 
-       public AltosEepromRecordSet(Reader input) throws IOException {
+       public AltosEepromRecordSet(InputStream input) throws IOException {
                this(new AltosEepromNew(input));
        }
 }