git.gag.com Git - fw/altos/blob - altosuilib/AltosGraphNew.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; 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.altosuilib_11;
  20
  21 import java.io.*;
  22 import java.util.ArrayList;
  23
  24 import java.awt.*;
  25 import javax.swing.*;
  26 import org.altusmetrum.altoslib_11.*;
  27
  28 import org.jfree.ui.*;
  29 import org.jfree.chart.*;
  30 import org.jfree.chart.plot.*;
  31 import org.jfree.chart.axis.*;
  32 import org.jfree.chart.renderer.*;
  33 import org.jfree.chart.renderer.xy.*;
  34 import org.jfree.chart.labels.*;
  35 import org.jfree.data.xy.*;
  36 import org.jfree.data.*;
  37
  38 public class AltosGraphNew extends AltosUIGraphNew {
  39
  40         static final private Color height_color = new Color(194,31,31);
  41         static final private Color kalman_height_color = new Color(255,0,0);
  42         static final private Color gps_height_color = new Color(150,31,31);
  43         static final private Color pressure_color = new Color (225,31,31);
  44         static final private Color range_color = new Color(100, 31, 31);
  45         static final private Color distance_color = new Color(100, 31, 194);
  46         static final private Color speed_color = new Color(31,194,31);
  47         static final private Color kalman_speed_color = new Color(0,255,0);
  48         static final private Color thrust_color = new Color(31,194,31);
  49         static final private Color accel_color = new Color(31,31,194);
  50         static final private Color kalman_accel_color = new Color(0,0,255);
  51         static final private Color voltage_color = new Color(194, 194, 31);
  52         static final private Color battery_voltage_color = new Color(194, 194, 31);
  53         static final private Color drogue_voltage_color = new Color(150, 150, 31);
  54         static final private Color main_voltage_color = new Color(100, 100, 31);
  55         static final private Color igniter_voltage_color = new Color(80, 80, 31);
  56         static final private Color igniter_marker_color = new Color(255, 0, 0);
  57         static final private Color gps_nsat_color = new Color (194, 31, 194);
  58         static final private Color gps_nsat_solution_color = new Color (194, 31, 194);
  59         static final private Color gps_nsat_view_color = new Color (150, 31, 150);
  60         static final private Color gps_course_color = new Color (100, 31, 112);
  61         static final private Color gps_ground_speed_color = new Color (31, 112, 100);
  62         static final private Color gps_speed_color = new Color (31, 112, 100);
  63         static final private Color gps_climb_rate_color = new Color (31, 31, 112);
  64         static final private Color gps_pdop_color = new Color(50, 194, 0);
  65         static final private Color gps_hdop_color = new Color(50, 0, 194);
  66         static final private Color gps_vdop_color = new Color(194, 0, 50);
  67         static final private Color temperature_color = new Color (31, 194, 194);
  68         static final private Color dbm_color = new Color(31, 100, 100);
  69         static final private Color state_color = new Color(0,0,0);
  70         static final private Color accel_along_color = new Color(255, 0, 0);
  71         static final private Color accel_across_color = new Color(0, 255, 0);
  72         static final private Color accel_through_color = new Color(0, 0, 255);
  73         static final private Color gyro_roll_color = new Color(192, 0, 0);
  74         static final private Color gyro_pitch_color = new Color(0, 192, 0);
  75         static final private Color gyro_yaw_color = new Color(0, 0, 192);
  76         static final private Color mag_along_color = new Color(128, 0, 0);
  77         static final private Color mag_across_color = new Color(0, 128, 0);
  78         static final private Color mag_through_color = new Color(0, 0, 128);
  79         static final private Color orient_color = new Color(31, 31, 31);
  80
  81         static AltosUnits dop_units = null;
  82
  83         AltosUIFlightSeries flight_series;
  84
  85         AltosUITimeSeries[] setup(AltosFlightStats stats, AltosUIFlightSeries flight_series) {
  86                 AltosCalData    cal_data = flight_series.cal_data;
  87
  88                 AltosUIAxis     height_axis, speed_axis, accel_axis, voltage_axis, temperature_axis, nsat_axis, dbm_axis;
  89                 AltosUIAxis     distance_axis, pressure_axis, thrust_axis;
  90                 AltosUIAxis     gyro_axis, orient_axis, mag_axis;
  91                 AltosUIAxis     course_axis, dop_axis;
  92
  93                 if (stats.serial != AltosLib.MISSING && stats.product != null && stats.flight != AltosLib.MISSING)
  94                         setName(String.format("%s %d flight %d\n", stats.product, stats.serial, stats.flight));
  95
  96                 height_axis = newAxis("Height", AltosConvert.height, height_color);
  97                 pressure_axis = newAxis("Pressure", AltosConvert.pressure, pressure_color, 0);
  98                 speed_axis = newAxis("Speed", AltosConvert.speed, speed_color);
  99                 thrust_axis = newAxis("Thrust", AltosConvert.force, thrust_color);
 100                 accel_axis = newAxis("Acceleration", AltosConvert.accel, accel_color);
 101                 voltage_axis = newAxis("Voltage", AltosConvert.voltage, voltage_color);
 102                 temperature_axis = newAxis("Temperature", AltosConvert.temperature, temperature_color, 0);
 103                 nsat_axis = newAxis("Satellites", null, gps_nsat_color,
 104                                     AltosUIAxis.axis_include_zero | AltosUIAxis.axis_integer);
 105                 dbm_axis = newAxis("Signal Strength", null, dbm_color, 0);
 106                 distance_axis = newAxis("Distance", AltosConvert.distance, range_color);
 107
 108                 gyro_axis = newAxis("Rotation Rate", AltosConvert.rotation_rate, gyro_roll_color, 0);
 109                 orient_axis = newAxis("Tilt Angle", AltosConvert.orient, orient_color, 0);
 110                 mag_axis = newAxis("Magnetic Field", AltosConvert.magnetic_field, mag_along_color, 0);
 111                 course_axis = newAxis("Course", AltosConvert.orient, gps_course_color, 0);
 112                 dop_axis = newAxis("Dilution of Precision", dop_units, gps_pdop_color, 0);
 113
 114                 flight_series.register_axis("default",
 115                                             speed_color,
 116                                             false,
 117                                             speed_axis);
 118
 119                 flight_series.register_marker(AltosUIFlightSeries.state_name,
 120                                               state_color,
 121                                               true,
 122                                               plot,
 123                                               true);
 124
 125                 flight_series.register_marker(AltosUIFlightSeries.pyro_fired_name,
 126                                               igniter_marker_color,
 127                                               true,
 128                                               plot,
 129                                               false);
 130
 131                 flight_series.register_axis(AltosUIFlightSeries.accel_name,
 132                                             accel_color,
 133                                             true,
 134                                             accel_axis);
 135
 136                 flight_series.register_axis(AltosUIFlightSeries.kalman_accel_name,
 137                                             kalman_accel_color,
 138                                             false,
 139                                             accel_axis);
 140
 141                 flight_series.register_axis(AltosUIFlightSeries.rssi_name,
 142                                             dbm_color,
 143                                             false,
 144                                             dbm_axis);
 145
 146                 flight_series.register_axis(AltosUIFlightSeries.speed_name,
 147                                             speed_color,
 148                                             true,
 149                                             speed_axis);
 150
 151                 flight_series.register_axis(AltosUIFlightSeries.kalman_speed_name,
 152                                             kalman_speed_color,
 153                                             true,
 154                                             speed_axis);
 155
 156                 flight_series.register_axis(AltosUIFlightSeries.pressure_name,
 157                                             pressure_color,
 158                                             false,
 159                                             pressure_axis);
 160
 161                 flight_series.register_axis(AltosUIFlightSeries.height_name,
 162                                             height_color,
 163                                             true,
 164                                             height_axis);
 165
 166                 flight_series.register_axis(AltosUIFlightSeries.altitude_name,
 167                                             height_color,
 168                                             false,
 169                                             height_axis);
 170
 171                 flight_series.register_axis(AltosUIFlightSeries.kalman_height_name,
 172                                             kalman_height_color,
 173                                             false,
 174                                             height_axis);
 175
 176
 177                 flight_series.register_axis(AltosUIFlightSeries.temperature_name,
 178                                             temperature_color,
 179                                             false,
 180                                             temperature_axis);
 181
 182                 flight_series.register_axis(AltosUIFlightSeries.battery_voltage_name,
 183                                             battery_voltage_color,
 184                                             false,
 185                                             voltage_axis);
 186
 187                 flight_series.register_axis(AltosUIFlightSeries.apogee_voltage_name,
 188                                             drogue_voltage_color,
 189                                             false,
 190                                             voltage_axis);
 191
 192                 flight_series.register_axis(AltosUIFlightSeries.main_voltage_name,
 193                                             main_voltage_color,
 194                                             false,
 195                                             voltage_axis);
 196
 197                 flight_series.register_axis(AltosUIFlightSeries.sats_in_view_name,
 198                                             gps_nsat_view_color,
 199                                             false,
 200                                             nsat_axis);
 201
 202                 flight_series.register_axis(AltosUIFlightSeries.sats_in_soln_name,
 203                                             gps_nsat_solution_color,
 204                                             false,
 205                                             nsat_axis);
 206
 207                 flight_series.register_axis(AltosUIFlightSeries.gps_pdop_name,
 208                                             gps_pdop_color,
 209                                             false,
 210                                             dop_axis);
 211
 212                 flight_series.register_axis(AltosUIFlightSeries.gps_hdop_name,
 213                                             gps_hdop_color,
 214                                             false,
 215                                             dop_axis);
 216
 217                 flight_series.register_axis(AltosUIFlightSeries.gps_vdop_name,
 218                                             gps_vdop_color,
 219                                             false,
 220                                             dop_axis);
 221
 222                 flight_series.register_axis(AltosUIFlightSeries.gps_altitude_name,
 223                                             gps_height_color,
 224                                             false,
 225                                             height_axis);
 226
 227                 flight_series.register_axis(AltosUIFlightSeries.gps_height_name,
 228                                             gps_height_color,
 229                                             false,
 230                                             height_axis);
 231
 232                 flight_series.register_axis(AltosUIFlightSeries.gps_ground_speed_name,
 233                                             gps_ground_speed_color,
 234                                             false,
 235                                             speed_axis);
 236
 237                 flight_series.register_axis(AltosUIFlightSeries.gps_ascent_rate_name,
 238                                             gps_climb_rate_color,
 239                                             false,
 240                                             speed_axis);
 241
 242                 flight_series.register_axis(AltosUIFlightSeries.gps_course_name,
 243                                             gps_course_color,
 244                                             false,
 245                                             course_axis);
 246
 247                 flight_series.register_axis(AltosUIFlightSeries.gps_speed_name,
 248                                             gps_speed_color,
 249                                             false,
 250                                             speed_axis);
 251
 252                 flight_series.register_axis(AltosUIFlightSeries.accel_along_name,
 253                                             accel_along_color,
 254                                             false,
 255                                             accel_axis);
 256
 257                 flight_series.register_axis(AltosUIFlightSeries.accel_across_name,
 258                                             accel_across_color,
 259                                             false,
 260                                             accel_axis);
 261
 262                 flight_series.register_axis(AltosUIFlightSeries.accel_through_name,
 263                                             accel_through_color,
 264                                             false,
 265                                             accel_axis);
 266
 267                 flight_series.register_axis(AltosUIFlightSeries.gyro_roll_name,
 268                                             gyro_roll_color,
 269                                             false,
 270                                             gyro_axis);
 271
 272                 flight_series.register_axis(AltosUIFlightSeries.gyro_pitch_name,
 273                                             gyro_pitch_color,
 274                                             false,
 275                                             gyro_axis);
 276
 277                 flight_series.register_axis(AltosUIFlightSeries.gyro_yaw_name,
 278                                             gyro_yaw_color,
 279                                             false,
 280                                             gyro_axis);
 281
 282                 flight_series.register_axis(AltosUIFlightSeries.mag_along_name,
 283                                             mag_along_color,
 284                                             false,
 285                                             mag_axis);
 286
 287                 flight_series.register_axis(AltosUIFlightSeries.mag_across_name,
 288                                             mag_across_color,
 289                                             false,
 290                                             mag_axis);
 291
 292                 flight_series.register_axis(AltosUIFlightSeries.mag_through_name,
 293                                             mag_through_color,
 294                                             false,
 295                                             mag_axis);
 296
 297                 flight_series.register_axis(AltosUIFlightSeries.orient_name,
 298                                             orient_color,
 299                                             false,
 300                                             orient_axis);
 301
 302                 for (int channel = 0; channel < 26; channel++) {
 303                         flight_series.register_axis(flight_series.igniter_voltage_name(channel),
 304                                                     igniter_voltage_color,
 305                                                     false,
 306                                                     voltage_axis);
 307                 }
 308
 309                 flight_series.register_axis(AltosUIFlightSeries.thrust_name,
 310                                             thrust_color,
 311                                             true,
 312                                             thrust_axis);
 313
 314                 return flight_series.series(cal_data);
 315         }
 316
 317         public void set_data(AltosFlightStats stats, AltosUIFlightSeries flight_series) {
 318                 set_series(setup(stats, flight_series));
 319         }
 320
 321         public AltosGraphNew(AltosUIEnable enable) {
 322                 super(enable, "Flight");
 323         }
 324
 325         public AltosGraphNew(AltosUIEnable enable, AltosFlightStats stats, AltosUIFlightSeries flight_series) {
 326                 this(enable);
 327                 set_series(setup(stats, flight_series));
 328         }
 329 }