git.gag.com Git - fw/altos/blob - altoslib/AltosFlightSeries.java

   1 /*
   2  * Copyright © 2017 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
  15 package org.altusmetrum.altoslib_11;
  16
  17 import java.util.*;
  18
  19 public class AltosFlightSeries extends AltosDataListener {
  20
  21         public ArrayList<AltosTimeSeries> series = new ArrayList<AltosTimeSeries>();
  22
  23         public double   speed_filter_width = 4.0;
  24         public double   accel_filter_width = 4.0;
  25
  26         public int[] indices() {
  27                 int[] indices = new int[series.size()];
  28                 for (int i = 0; i < indices.length; i++)
  29                         indices[i] = -1;
  30                 step_indices(indices);
  31                 return indices;
  32         }
  33
  34         private double time(int id, int index) {
  35                 AltosTimeSeries         s = series.get(id);
  36
  37                 if (index < 0)
  38                         return Double.NEGATIVE_INFINITY;
  39
  40                 if (index < s.values.size())
  41                         return s.values.get(index).time;
  42                 return Double.POSITIVE_INFINITY;
  43         }
  44
  45         public boolean step_indices(int[] indices) {
  46                 double  min_next = time(0, indices[0]+1);
  47
  48                 for (int i = 1; i < indices.length; i++) {
  49                         double next = time(i, indices[i]+1);
  50                         if (next < min_next)
  51                                 min_next = next;
  52                 }
  53
  54                 if (min_next == Double.POSITIVE_INFINITY)
  55                         return false;
  56
  57                 for (int i = 0; i < indices.length; i++) {
  58                         double  t = time(i, indices[i] + 1);
  59
  60                         if (t <= min_next)
  61                                 indices[i]++;
  62                 }
  63                 return true;
  64         }
  65
  66         public double time(int[] indices) {
  67                 double max = time(0, indices[0]);
  68
  69                 for (int i = 1; i < indices.length; i++) {
  70                         double t = time(i, indices[i]);
  71                         if (t >= max)
  72                                 max = t;
  73                 }
  74                 return max;
  75         }
  76
  77         public double value(String name, int[] indices) {
  78                 for (int i = 0; i < indices.length; i++) {
  79                         AltosTimeSeries s = series.get(i);
  80                         if (s.label.equals(name)) {
  81                                 int index = indices[i];
  82                                 if (index < 0)
  83                                         index = 0;
  84                                 if (index >= s.values.size())
  85                                         index = s.values.size() - 1;
  86                                 return s.values.get(index).value;
  87                         }
  88                 }
  89                 return AltosLib.MISSING;
  90         }
  91
  92         public double value(String name, double time) {
  93                 for (AltosTimeSeries s : series) {
  94                         if (s.label.equals(name))
  95                                 return s.value(time);
  96                 }
  97                 return AltosLib.MISSING;
  98         }
  99
 100         public double value_before(String name, double time) {
 101                 for (AltosTimeSeries s : series) {
 102                         if (s.label.equals(name))
 103                                 return s.value_before(time);
 104                 }
 105                 return AltosLib.MISSING;
 106         }
 107
 108         public double value_after(String name, double time) {
 109                 for (AltosTimeSeries s : series) {
 110                         if (s.label.equals(name))
 111                                 return s.value_after(time);
 112                 }
 113                 return AltosLib.MISSING;
 114         }
 115
 116         public AltosTimeSeries make_series(String label, AltosUnits units) {
 117                 return new AltosTimeSeries(label, units);
 118         }
 119
 120         public void add_series(AltosTimeSeries s) {
 121                 series.add(s);
 122         }
 123
 124         public AltosTimeSeries add_series(String label, AltosUnits units) {
 125                 AltosTimeSeries s = make_series(label, units);
 126                 add_series(s);
 127                 return s;
 128         }
 129
 130         public void remove_series(AltosTimeSeries s) {
 131                 series.remove(s);
 132         }
 133
 134         public boolean has_series(String label) {
 135                 for (AltosTimeSeries s : series)
 136                         if (s.label.equals(label))
 137                                 return true;
 138                 return false;
 139         }
 140
 141         public AltosTimeSeries state_series;
 142
 143         public static final String state_name = "State";
 144
 145         public void set_state(int state) {
 146
 147                 if (state == AltosLib.ao_flight_pad)
 148                         return;
 149
 150                 if (state_series == null)
 151                         state_series = add_series(state_name, AltosConvert.state_name);
 152                 else if (this.state == state)
 153                         return;
 154                 this.state = state;
 155                 state_series.add(time(), state);
 156         }
 157
 158         public AltosTimeSeries  accel_series;
 159
 160         public static final String accel_name = "Accel";
 161
 162         public void set_acceleration(double acceleration) {
 163                 if (acceleration == AltosLib.MISSING)
 164                         return;
 165                 if (accel_series == null)
 166                         accel_series = add_series(accel_name, AltosConvert.accel);
 167
 168                 accel_series.add(time(), acceleration);
 169         }
 170
 171         private void compute_accel() {
 172                 if (accel_series != null)
 173                         return;
 174
 175                 if (speed_series != null) {
 176                         AltosTimeSeries temp_series = make_series(speed_name, AltosConvert.speed);
 177                         speed_series.filter(temp_series, accel_filter_width);
 178                         accel_series = add_series(accel_name, AltosConvert.accel);
 179                         temp_series.differentiate(accel_series);
 180                 }
 181         }
 182
 183         public void set_received_time(long received_time) {
 184         }
 185
 186         public AltosTimeSeries rssi_series;
 187
 188         public static final String rssi_name = "RSSI";
 189
 190         public AltosTimeSeries status_series;
 191
 192         public static final String status_name = "Radio Status";
 193
 194         public void set_rssi(int rssi, int status) {
 195                 if (rssi_series == null) {
 196                         rssi_series = add_series(rssi_name, null);
 197                         status_series = add_series(status_name, null);
 198                 }
 199                 rssi_series.add(time(), rssi);
 200                 status_series.add(time(), status);
 201         }
 202
 203         public AltosTimeSeries pressure_series;
 204
 205         public static final String pressure_name = "Pressure";
 206
 207         public AltosTimeSeries altitude_series;
 208
 209         public static final String altitude_name = "Altitude";
 210
 211         public AltosTimeSeries height_series;
 212
 213         public static final String height_name = "Height";
 214
 215         public  void set_pressure(double pa) {
 216                 if (pa == AltosLib.MISSING)
 217                         return;
 218
 219                 if (pressure_series == null)
 220                         pressure_series = add_series(pressure_name, AltosConvert.pressure);
 221                 pressure_series.add(time(), pa);
 222                 if (altitude_series == null)
 223                         altitude_series = add_series(altitude_name, AltosConvert.height);
 224
 225                 if (cal_data.ground_pressure == AltosLib.MISSING)
 226                         cal_data.set_ground_pressure(pa);
 227
 228                 double altitude = AltosConvert.pressure_to_altitude(pa);
 229                 altitude_series.add(time(), altitude);
 230         }
 231
 232         private void compute_height() {
 233                 double ground_altitude = cal_data.ground_altitude;
 234                 if (height_series == null && ground_altitude != AltosLib.MISSING && altitude_series != null) {
 235                         height_series = add_series(height_name, AltosConvert.height);
 236                         for (AltosTimeValue alt : altitude_series)
 237                                 height_series.add(alt.time, alt.value - ground_altitude);
 238                 }
 239
 240                 if (gps_height == null && cal_data.gps_pad != null && cal_data.gps_pad.alt != AltosLib.MISSING && gps_altitude != null) {
 241                         double gps_ground_altitude = cal_data.gps_pad.alt;
 242                         gps_height = add_series(gps_height_name, AltosConvert.height);
 243                         for (AltosTimeValue gps_alt : gps_altitude)
 244                                 gps_height.add(gps_alt.time, gps_alt.value - gps_ground_altitude);
 245                 }
 246         }
 247
 248         public AltosTimeSeries speed_series;
 249
 250         public static final String speed_name = "Speed";
 251
 252         private void compute_speed() {
 253                 if (speed_series != null)
 254                         return;
 255
 256                 AltosTimeSeries alt_speed_series = null;
 257                 AltosTimeSeries accel_speed_series = null;
 258
 259                 if (altitude_series != null) {
 260                         AltosTimeSeries temp_series = make_series(altitude_name, AltosConvert.height);
 261                         altitude_series.filter(temp_series, speed_filter_width);
 262
 263                         alt_speed_series = make_series(speed_name, AltosConvert.speed);
 264                         temp_series.differentiate(alt_speed_series);
 265                 }
 266                 if (accel_series != null) {
 267                         AltosTimeSeries temp_series = make_series(speed_name, AltosConvert.speed);
 268                         accel_series.integrate(temp_series);
 269
 270                         accel_speed_series = make_series(speed_name, AltosConvert.speed);
 271                         temp_series.filter(accel_speed_series, 0.1);
 272                 }
 273
 274                 if (alt_speed_series != null && accel_speed_series != null) {
 275                         double  apogee_time = AltosLib.MISSING;
 276                         if (state_series != null) {
 277                                 for (AltosTimeValue d : state_series) {
 278                                         if (d.value >= AltosLib.ao_flight_drogue){
 279                                                 apogee_time = d.time;
 280                                                 break;
 281                                         }
 282                                 }
 283                         }
 284                         if (apogee_time == AltosLib.MISSING) {
 285                                 speed_series = alt_speed_series;
 286                         } else {
 287                                 speed_series = make_series(speed_name, AltosConvert.speed);
 288                                 for (AltosTimeValue d : accel_speed_series) {
 289                                         if (d.time <= apogee_time)
 290                                                 speed_series.add(d);
 291                                 }
 292                                 for (AltosTimeValue d : alt_speed_series) {
 293                                         if (d.time > apogee_time)
 294                                                 speed_series.add(d);
 295                                 }
 296
 297                         }
 298                 } else if (alt_speed_series != null) {
 299                         speed_series = alt_speed_series;
 300                 } else if (accel_speed_series != null) {
 301                         speed_series = accel_speed_series;
 302                 }
 303                 if (speed_series != null)
 304                         add_series(speed_series);
 305         }
 306
 307         public AltosTimeSeries  kalman_height_series, kalman_speed_series, kalman_accel_series;
 308
 309         public static final String kalman_height_name = "Kalman Height";
 310         public static final String kalman_speed_name = "Kalman Speed";
 311         public static final String kalman_accel_name = "Kalman Accel";
 312
 313         public void set_kalman(double height, double speed, double acceleration) {
 314                 if (kalman_height_series == null) {
 315                         kalman_height_series = add_series(kalman_height_name, AltosConvert.height);
 316                         kalman_speed_series = add_series(kalman_speed_name, AltosConvert.speed);
 317                         kalman_accel_series = add_series(kalman_accel_name, AltosConvert.accel);
 318                 }
 319                 kalman_height_series.add(time(), height);
 320                 kalman_speed_series.add(time(), speed);
 321                 kalman_accel_series.add(time(), acceleration);
 322         }
 323
 324         public AltosTimeSeries thrust_series;
 325
 326         public static final String thrust_name = "Thrust";
 327
 328         public  void set_thrust(double N) {
 329                 if (thrust_series == null)
 330                         thrust_series = add_series(thrust_name, AltosConvert.force);
 331                 thrust_series.add(time(), N);
 332         }
 333
 334         public AltosTimeSeries temperature_series;
 335
 336         public static final String temperature_name = "Temperature";
 337
 338         public  void set_temperature(double deg_c) {
 339                 if (temperature_series == null)
 340                         temperature_series = add_series(temperature_name, AltosConvert.temperature);
 341                 temperature_series.add(time(), deg_c);
 342         }
 343
 344         public AltosTimeSeries battery_voltage_series;
 345
 346         public static final String battery_voltage_name = "Battery Voltage";
 347
 348         public void set_battery_voltage(double volts) {
 349                 if (volts == AltosLib.MISSING)
 350                         return;
 351                 if (battery_voltage_series == null)
 352                         battery_voltage_series = add_series(battery_voltage_name, AltosConvert.voltage);
 353                 battery_voltage_series.add(time(), volts);
 354         }
 355
 356         public AltosTimeSeries apogee_voltage_series;
 357
 358         public static final String apogee_voltage_name = "Apogee Voltage";
 359
 360         public void set_apogee_voltage(double volts) {
 361                 if (volts == AltosLib.MISSING)
 362                         return;
 363                 if (apogee_voltage_series == null)
 364                         apogee_voltage_series = add_series(apogee_voltage_name, AltosConvert.voltage);
 365                 apogee_voltage_series.add(time(), volts);
 366         }
 367
 368         public AltosTimeSeries main_voltage_series;
 369
 370         public static final String main_voltage_name = "Main Voltage";
 371
 372         public void set_main_voltage(double volts) {
 373                 if (volts == AltosLib.MISSING)
 374                         return;
 375                 if (main_voltage_series == null)
 376                         main_voltage_series = add_series(main_voltage_name, AltosConvert.voltage);
 377                 main_voltage_series.add(time(), volts);
 378         }
 379
 380         public ArrayList<AltosGPSTimeValue> gps_series;
 381
 382         public AltosGPS gps_before(double time) {
 383                 AltosGPS gps = null;
 384                 for (AltosGPSTimeValue gtv : gps_series)
 385                         if (gtv.time <= time)
 386                                 gps = gtv.gps;
 387                         else
 388                                 break;
 389                 return gps;
 390         }
 391
 392         public AltosTimeSeries  sats_in_view;
 393         public AltosTimeSeries sats_in_soln;
 394         public AltosTimeSeries gps_altitude;
 395         public AltosTimeSeries gps_height;
 396         public AltosTimeSeries gps_ground_speed;
 397         public AltosTimeSeries gps_ascent_rate;
 398         public AltosTimeSeries gps_course;
 399         public AltosTimeSeries gps_speed;
 400         public AltosTimeSeries gps_pdop, gps_vdop, gps_hdop;
 401
 402         public static final String sats_in_view_name = "Satellites in view";
 403         public static final String sats_in_soln_name = "Satellites in solution";
 404         public static final String gps_altitude_name = "GPS Altitude";
 405         public static final String gps_height_name = "GPS Height";
 406         public static final String gps_ground_speed_name = "GPS Ground Speed";
 407         public static final String gps_ascent_rate_name = "GPS Ascent Rate";
 408         public static final String gps_course_name = "GPS Course";
 409         public static final String gps_speed_name = "GPS Speed";
 410         public static final String gps_pdop_name = "GPS Dilution of Precision";
 411         public static final String gps_vdop_name = "GPS Vertical Dilution of Precision";
 412         public static final String gps_hdop_name = "GPS Horizontal Dilution of Precision";
 413
 414         public void set_gps(AltosGPS gps) {
 415                 if (gps_series == null)
 416                         gps_series = new ArrayList<AltosGPSTimeValue>();
 417                 gps_series.add(new AltosGPSTimeValue(time(), gps));
 418
 419                 if (sats_in_soln == null) {
 420                         sats_in_soln = add_series(sats_in_soln_name, null);
 421                 }
 422                 sats_in_soln.add(time(), gps.nsat);
 423                 if (gps.pdop != AltosLib.MISSING) {
 424                         if (gps_pdop == null)
 425                                 gps_pdop = add_series(gps_pdop_name, null);
 426                         gps_pdop.add(time(), gps.pdop);
 427                 }
 428                 if (gps.hdop != AltosLib.MISSING) {
 429                         if (gps_hdop == null)
 430                                 gps_hdop = add_series(gps_hdop_name, null);
 431                         gps_hdop.add(time(), gps.hdop);
 432                 }
 433                 if (gps.vdop != AltosLib.MISSING) {
 434                         if (gps_vdop == null)
 435                                 gps_vdop = add_series(gps_vdop_name, null);
 436                         gps_vdop.add(time(), gps.vdop);
 437                 }
 438                 if (gps.locked) {
 439                         if (gps.alt != AltosLib.MISSING) {
 440                                 if (gps_altitude == null)
 441                                         gps_altitude = add_series(gps_altitude_name, AltosConvert.height);
 442                                 gps_altitude.add(time(), gps.alt);
 443                         }
 444                         if (gps.ground_speed != AltosLib.MISSING) {
 445                                 if (gps_ground_speed == null)
 446                                         gps_ground_speed = add_series(gps_ground_speed_name, AltosConvert.speed);
 447                                 gps_ground_speed.add(time(), gps.ground_speed);
 448                         }
 449                         if (gps.climb_rate != AltosLib.MISSING) {
 450                                 if (gps_ascent_rate == null)
 451                                         gps_ascent_rate = add_series(gps_ascent_rate_name, AltosConvert.speed);
 452                                 gps_ascent_rate.add(time(), gps.climb_rate);
 453                         }
 454                         if (gps.course != AltosLib.MISSING) {
 455                                 if (gps_course == null)
 456                                         gps_course = add_series(gps_course_name, null);
 457                                 gps_course.add(time(), gps.course);
 458                         }
 459                         if (gps.ground_speed != AltosLib.MISSING && gps.climb_rate != AltosLib.MISSING) {
 460                                 if (gps_speed == null)
 461                                         gps_speed = add_series(gps_speed_name, null);
 462                                 gps_speed.add(time(), Math.sqrt(gps.ground_speed * gps.ground_speed +
 463                                                                 gps.climb_rate * gps.climb_rate));
 464                         }
 465                 }
 466                 if (gps.cc_gps_sat != null) {
 467                         if (sats_in_view == null)
 468                                 sats_in_view = add_series(sats_in_view_name, null);
 469                         sats_in_view.add(time(), gps.cc_gps_sat.length);
 470                 }
 471         }
 472
 473         public static final String accel_along_name = "Accel Along";
 474         public static final String accel_across_name = "Accel Across";
 475         public static final String accel_through_name = "Accel Through";
 476
 477         public AltosTimeSeries accel_along, accel_across, accel_through;
 478
 479         public static final String gyro_roll_name = "Roll Rate";
 480         public static final String gyro_pitch_name = "Pitch Rate";
 481         public static final String gyro_yaw_name = "Yaw Rate";
 482
 483         public AltosTimeSeries gyro_roll, gyro_pitch, gyro_yaw;
 484
 485         public static final String mag_along_name = "Magnetic Field Along";
 486         public static final String mag_across_name = "Magnetic Field Across";
 487         public static final String mag_through_name = "Magnetic Field Through";
 488
 489         public AltosTimeSeries mag_along, mag_across, mag_through;
 490
 491         public  void set_accel(double along, double across, double through) {
 492                 if (accel_along == null) {
 493                         accel_along = add_series(accel_along_name, AltosConvert.accel);
 494                         accel_across = add_series(accel_across_name, AltosConvert.accel);
 495                         accel_through = add_series(accel_through_name, AltosConvert.accel);
 496                 }
 497                 accel_along.add(time(), along);
 498                 accel_across.add(time(), across);
 499                 accel_through.add(time(), through);
 500         }
 501
 502         public  void set_accel_ground(double along, double across, double through) {
 503         }
 504
 505         public  void set_gyro(double roll, double pitch, double yaw) {
 506                 if (gyro_roll == null) {
 507                         gyro_roll = add_series(gyro_roll_name, AltosConvert.rotation_rate);
 508                         gyro_pitch = add_series(gyro_pitch_name, AltosConvert.rotation_rate);
 509                         gyro_yaw = add_series(gyro_yaw_name, AltosConvert.rotation_rate);
 510                 }
 511                 gyro_roll.add(time(), roll);
 512                 gyro_pitch.add(time(), pitch);
 513                 gyro_yaw.add(time(), yaw);
 514         }
 515
 516         public  void set_mag(double along, double across, double through) {
 517                 if (mag_along == null) {
 518                         mag_along = add_series(mag_along_name, AltosConvert.magnetic_field);
 519                         mag_across = add_series(mag_across_name, AltosConvert.magnetic_field);
 520                         mag_through = add_series(mag_through_name, AltosConvert.magnetic_field);
 521                 }
 522                 mag_along.add(time(), along);
 523                 mag_across.add(time(), across);
 524                 mag_through.add(time(), through);
 525         }
 526
 527         public static final String orient_name = "Tilt Angle";
 528
 529         public AltosTimeSeries orient_series;
 530
 531         public void set_orient(double orient) {
 532                 if (orient_series == null)
 533                         orient_series = add_series(orient_name, AltosConvert.orient);
 534                 orient_series.add(time(), orient);
 535         }
 536
 537         public static final String pyro_voltage_name = "Pyro Voltage";
 538
 539         public AltosTimeSeries pyro_voltage;
 540
 541         public  void set_pyro_voltage(double volts) {
 542                 if (pyro_voltage == null)
 543                         pyro_voltage = add_series(pyro_voltage_name, AltosConvert.voltage);
 544                 pyro_voltage.add(time(), volts);
 545         }
 546
 547         private static String[] igniter_voltage_names;
 548
 549         public String igniter_voltage_name(int channel) {
 550                 if (igniter_voltage_names == null || igniter_voltage_names.length <= channel) {
 551                         String[] new_igniter_voltage_names = new String[channel + 1];
 552                         int     i = 0;
 553
 554                         if (igniter_voltage_names != null) {
 555                                 for (; i < igniter_voltage_names.length; i++)
 556                                         new_igniter_voltage_names[i] = igniter_voltage_names[i];
 557                         }
 558                         for (; i < channel+1; i++)
 559                                 new_igniter_voltage_names[i] = AltosLib.igniter_name(i);
 560                         igniter_voltage_names = new_igniter_voltage_names;
 561                 }
 562                 return igniter_voltage_names[channel];
 563         }
 564
 565         public AltosTimeSeries[] igniter_voltage;
 566
 567         public  void set_igniter_voltage(double[] voltage) {
 568                 int channels = voltage.length;
 569                 if (igniter_voltage == null || igniter_voltage.length <= channels) {
 570                         AltosTimeSeries[]       new_igniter_voltage = new AltosTimeSeries[channels + 1];
 571                         int                     i = 0;
 572
 573                         if (igniter_voltage != null) {
 574                                 for (; i < igniter_voltage.length; i++)
 575                                         new_igniter_voltage[i] = igniter_voltage[i];
 576                         }
 577                         for (; i < channels; i++)
 578                                 new_igniter_voltage[i] = add_series(igniter_voltage_name(i), AltosConvert.voltage);
 579                         igniter_voltage = new_igniter_voltage;
 580                 }
 581                 for (int channel = 0; channel < voltage.length; channel++)
 582                         igniter_voltage[channel].add(time(), voltage[channel]);
 583         }
 584
 585         public static final String pyro_fired_name = "Pyro Channel State";
 586
 587         public AltosTimeSeries pyro_fired_series;
 588
 589         int     last_pyro_mask;
 590
 591         public  void set_pyro_fired(int pyro_mask) {
 592                 if (pyro_fired_series == null)
 593                         pyro_fired_series = add_series(pyro_fired_name, AltosConvert.pyro_name);
 594                 for (int channel = 0; channel < 32; channel++) {
 595                         if ((last_pyro_mask & (1 << channel)) == 0 &&
 596                             (pyro_mask & (1 << channel)) != 0) {
 597                                 pyro_fired_series.add(time(), channel);
 598                         }
 599                 }
 600                 last_pyro_mask = pyro_mask;
 601         }
 602
 603         public void set_companion(AltosCompanion companion) {
 604         }
 605
 606         public void finish() {
 607                 compute_speed();
 608                 compute_accel();
 609                 compute_height();
 610         }
 611
 612         public AltosTimeSeries[] series() {
 613                 finish();
 614                 return series.toArray(new AltosTimeSeries[0]);
 615         }
 616
 617         public AltosFlightSeries(AltosCalData cal_data) {
 618                 super(cal_data);
 619         }
 620 }