git.gag.com Git - fw/altos/blob - 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_2;
  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                         body = new AltosEepromIterable(AltosEepromTm.read(input));
  83                         break;
  84                 case AltosLib.AO_LOG_FORMAT_TELEMETRY:
  85                 case AltosLib.AO_LOG_FORMAT_TELESCIENCE:
  86                 case AltosLib.AO_LOG_FORMAT_TELEMEGA:
  87                         body = new AltosEepromIterable(AltosEepromMega.read(input));
  88                         break;
  89                 case AltosLib.AO_LOG_FORMAT_TELEMETRUM:
  90                         body = new AltosEepromIterable(AltosEepromMetrum2.read(input));
  91                         break;
  92                 case AltosLib.AO_LOG_FORMAT_TELEMINI:
  93                 case AltosLib.AO_LOG_FORMAT_EASYMINI:
  94                         body = new AltosEepromIterable(AltosEepromMini.read(input));
  95                         break;
  96                 default:
  97                         body = new AltosEepromIterable(new LinkedList<AltosEeprom>());
  98                         break;
  99                 }
 100
 101                 /* Find boost tick */
 102                 AltosState      state = start.clone();
 103                 for (AltosEeprom eeprom : body) {
 104                         eeprom.update_state(state);
 105                         state.finish_update();
 106                         if (state.state >= AltosLib.ao_flight_boost) {
 107                                 start.set_boost_tick(state.tick);
 108                                 break;
 109                         }
 110                 }
 111         }
 112
 113         public Iterator<AltosState> iterator() {
 114                 AltosState              state = start.clone();
 115                 Iterator<AltosEeprom>   i = body.iterator();
 116
 117                 while (i.hasNext() && !state.valid()) {
 118                         i.next().update_state(state);
 119                         state.finish_update();
 120                 }
 121                 return new AltosEepromIterator(state, i);
 122         }
 123 }