git.gag.com Git - fw/altos/blob - altosuilib/AltosGraph.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.altosuilib_9;
  19
  20 import java.io.*;
  21 import java.util.ArrayList;
  22
  23 import java.awt.*;
  24 import javax.swing.*;
  25 import org.altusmetrum.altoslib_9.*;
  26
  27 import org.jfree.ui.*;
  28 import org.jfree.chart.*;
  29 import org.jfree.chart.plot.*;
  30 import org.jfree.chart.axis.*;
  31 import org.jfree.chart.renderer.*;
  32 import org.jfree.chart.renderer.xy.*;
  33 import org.jfree.chart.labels.*;
  34 import org.jfree.data.xy.*;
  35 import org.jfree.data.*;
  36
  37 class AltosVoltage extends AltosUnits {
  38
  39         public double value(double v, boolean imperial_units) {
  40                 return v;
  41         }
  42
  43         public double inverse(double v, boolean imperial_units) {
  44                 return v;
  45         }
  46
  47         public String show_units(boolean imperial_units) {
  48                 return "V";
  49         }
  50
  51         public String say_units(boolean imperial_units) {
  52                 return "volts";
  53         }
  54
  55         public int show_fraction(int width, boolean imperial_units) {
  56                 return width / 2;
  57         }
  58 }
  59
  60 class AltosNsat extends AltosUnits {
  61
  62         public double value(double v, boolean imperial_units) {
  63                 return v;
  64         }
  65
  66         public double inverse(double v, boolean imperial_units) {
  67                 return v;
  68         }
  69
  70         public String show_units(boolean imperial_units) {
  71                 return "Sats";
  72         }
  73
  74         public String say_units(boolean imperial_units) {
  75                 return "Satellites";
  76         }
  77
  78         public int show_fraction(int width, boolean imperial_units) {
  79                 return 0;
  80         }
  81 }
  82
  83 class AltosPressure extends AltosUnits {
  84
  85         public double value(double p, boolean imperial_units) {
  86                 return p;
  87         }
  88
  89         public double inverse(double p, boolean imperial_units) {
  90                 return p;
  91         }
  92
  93         public String show_units(boolean imperial_units) {
  94                 return "Pa";
  95         }
  96
  97         public String say_units(boolean imperial_units) {
  98                 return "pascals";
  99         }
 100
 101         public int show_fraction(int width, boolean imperial_units) {
 102                 return 0;
 103         }
 104 }
 105
 106 class AltosDbm extends AltosUnits {
 107
 108         public double value(double d, boolean imperial_units) {
 109                 return d;
 110         }
 111
 112         public double inverse(double d, boolean imperial_units) {
 113                 return d;
 114         }
 115
 116         public String show_units(boolean imperial_units) {
 117                 return "dBm";
 118         }
 119
 120         public String say_units(boolean imperial_units) {
 121                 return "D B M";
 122         }
 123
 124         public int show_fraction(int width, boolean imperial_units) {
 125                 return 0;
 126         }
 127 }
 128
 129 class AltosGyroUnits extends AltosUnits {
 130
 131         public double value(double p, boolean imperial_units) {
 132                 return p;
 133         }
 134
 135         public double inverse(double p, boolean imperial_units) {
 136                 return p;
 137         }
 138
 139         public String show_units(boolean imperial_units) {
 140                 return "°/sec";
 141         }
 142
 143         public String say_units(boolean imperial_units) {
 144                 return "degrees per second";
 145         }
 146
 147         public int show_fraction(int width, boolean imperial_units) {
 148                 return 1;
 149         }
 150 }
 151
 152 class AltosMagUnits extends AltosUnits {
 153
 154         public double value(double p, boolean imperial_units) {
 155                 return p;
 156         }
 157
 158         public double inverse(double p, boolean imperial_units) {
 159                 return p;
 160         }
 161
 162         public String show_units(boolean imperial_units) {
 163                 return "Ga";
 164         }
 165
 166         public String say_units(boolean imperial_units) {
 167                 return "gauss";
 168         }
 169
 170         public int show_fraction(int width, boolean imperial_units) {
 171                 return 2;
 172         }
 173 }
 174
 175 class AltosDopUnits extends AltosUnits {
 176
 177         public double value(double p, boolean imperial_units) {
 178                 return p;
 179         }
 180
 181         public double inverse(double p, boolean imperial_units) {
 182                 return p;
 183         }
 184
 185         public String show_units(boolean imperial_units) {
 186                 return null;
 187         }
 188
 189         public String say_units(boolean imperial_units) {
 190                 return null;
 191         }
 192
 193         public int show_fraction(int width, boolean imperial_units) {
 194                 return 1;
 195         }
 196 }
 197
 198 public class AltosGraph extends AltosUIGraph {
 199
 200         static final private Color height_color = new Color(194,31,31);
 201         static final private Color gps_height_color = new Color(150,31,31);
 202         static final private Color pressure_color = new Color (225,31,31);
 203         static final private Color range_color = new Color(100, 31, 31);
 204         static final private Color distance_color = new Color(100, 31, 194);
 205         static final private Color speed_color = new Color(31,194,31);
 206         static final private Color accel_color = new Color(31,31,194);
 207         static final private Color voltage_color = new Color(194, 194, 31);
 208         static final private Color battery_voltage_color = new Color(194, 194, 31);
 209         static final private Color drogue_voltage_color = new Color(150, 150, 31);
 210         static final private Color main_voltage_color = new Color(100, 100, 31);
 211         static final private Color gps_nsat_color = new Color (194, 31, 194);
 212         static final private Color gps_nsat_solution_color = new Color (194, 31, 194);
 213         static final private Color gps_nsat_view_color = new Color (150, 31, 150);
 214         static final private Color gps_course_color = new Color (100, 31, 112);
 215         static final private Color gps_ground_speed_color = new Color (31, 112, 100);
 216         static final private Color gps_climb_rate_color = new Color (31, 31, 112);
 217         static final private Color gps_pdop_color = new Color(50, 194, 0);
 218         static final private Color gps_hdop_color = new Color(50, 0, 194);
 219         static final private Color gps_vdop_color = new Color(194, 0, 50);
 220         static final private Color temperature_color = new Color (31, 194, 194);
 221         static final private Color dbm_color = new Color(31, 100, 100);
 222         static final private Color state_color = new Color(0,0,0);
 223         static final private Color accel_x_color = new Color(255, 0, 0);
 224         static final private Color accel_y_color = new Color(0, 255, 0);
 225         static final private Color accel_z_color = new Color(0, 0, 255);
 226         static final private Color gyro_x_color = new Color(192, 0, 0);
 227         static final private Color gyro_y_color = new Color(0, 192, 0);
 228         static final private Color gyro_z_color = new Color(0, 0, 192);
 229         static final private Color mag_x_color = new Color(128, 0, 0);
 230         static final private Color mag_y_color = new Color(0, 128, 0);
 231         static final private Color mag_z_color = new Color(0, 0, 128);
 232         static final private Color orient_color = new Color(31, 31, 31);
 233
 234         static AltosVoltage voltage_units = new AltosVoltage();
 235         static AltosPressure pressure_units = new AltosPressure();
 236         static AltosNsat nsat_units = new AltosNsat();
 237         static AltosDbm dbm_units = new AltosDbm();
 238         static AltosGyroUnits gyro_units = new AltosGyroUnits();
 239         static AltosOrient orient_units = new AltosOrient();
 240         static AltosMagUnits mag_units = new AltosMagUnits();
 241         static AltosDopUnits dop_units = new AltosDopUnits();
 242
 243         AltosUIAxis     height_axis, speed_axis, accel_axis, voltage_axis, temperature_axis, nsat_axis, dbm_axis;
 244         AltosUIAxis     distance_axis, pressure_axis;
 245         AltosUIAxis     gyro_axis, orient_axis, mag_axis;
 246         AltosUIAxis     course_axis, dop_axis;
 247
 248         public AltosGraph(AltosUIEnable enable, AltosFlightStats stats, AltosGraphDataSet dataSet) {
 249                 super(enable);
 250
 251                 height_axis = newAxis("Height", AltosConvert.height, height_color);
 252                 pressure_axis = newAxis("Pressure", pressure_units, pressure_color, 0);
 253                 speed_axis = newAxis("Speed", AltosConvert.speed, speed_color);
 254                 accel_axis = newAxis("Acceleration", AltosConvert.accel, accel_color);
 255                 voltage_axis = newAxis("Voltage", voltage_units, voltage_color);
 256                 temperature_axis = newAxis("Temperature", AltosConvert.temperature, temperature_color, 0);
 257                 nsat_axis = newAxis("Satellites", nsat_units, gps_nsat_color,
 258                                     AltosUIAxis.axis_include_zero | AltosUIAxis.axis_integer);
 259                 dbm_axis = newAxis("Signal Strength", dbm_units, dbm_color, 0);
 260                 distance_axis = newAxis("Distance", AltosConvert.distance, range_color);
 261
 262                 gyro_axis = newAxis("Rotation Rate", gyro_units, gyro_z_color, 0);
 263                 orient_axis = newAxis("Tilt Angle", orient_units, orient_color, 0);
 264                 mag_axis = newAxis("Magnetic Field", mag_units, mag_x_color, 0);
 265                 course_axis = newAxis("Course", orient_units, gps_course_color, 0);
 266                 dop_axis = newAxis("Dilution of Precision", dop_units, gps_pdop_color, 0);
 267
 268                 addMarker("State", AltosGraphDataPoint.data_state, state_color);
 269
 270                 if (stats.has_flight_data) {
 271                         addSeries("Height",
 272                                   AltosGraphDataPoint.data_height,
 273                                   AltosConvert.height,
 274                                   height_color,
 275                                   true,
 276                                   height_axis);
 277                         addSeries("Pressure",
 278                                   AltosGraphDataPoint.data_pressure,
 279                                   pressure_units,
 280                                   pressure_color,
 281                                   false,
 282                                   pressure_axis);
 283                         addSeries("Speed",
 284                                   AltosGraphDataPoint.data_speed,
 285                                   AltosConvert.speed,
 286                                   speed_color,
 287                                   true,
 288                                   speed_axis);
 289                         addSeries("Acceleration",
 290                                   AltosGraphDataPoint.data_accel,
 291                                   AltosConvert.accel,
 292                                   accel_color,
 293                                   true,
 294                                   accel_axis);
 295                 }
 296                 if (stats.has_gps) {
 297                         boolean enable_gps = false;
 298
 299                         if (!stats.has_flight_data)
 300                                 enable_gps = true;
 301
 302                         addSeries("Range",
 303                                   AltosGraphDataPoint.data_range,
 304                                   AltosConvert.distance,
 305                                   range_color,
 306                                   false,
 307                                   distance_axis);
 308                         addSeries("Distance",
 309                                   AltosGraphDataPoint.data_distance,
 310                                   AltosConvert.distance,
 311                                   distance_color,
 312                                   enable_gps,
 313                                   distance_axis);
 314                         addSeries("GPS Height",
 315                                   AltosGraphDataPoint.data_gps_height,
 316                                   AltosConvert.height,
 317                                   gps_height_color,
 318                                   enable_gps,
 319                                   height_axis);
 320                         addSeries("GPS Altitude",
 321                                   AltosGraphDataPoint.data_gps_altitude,
 322                                   AltosConvert.height,
 323                                   gps_height_color,
 324                                   false,
 325                                   height_axis);
 326                         addSeries("GPS Satellites in Solution",
 327                                   AltosGraphDataPoint.data_gps_nsat_solution,
 328                                   nsat_units,
 329                                   gps_nsat_solution_color,
 330                                   false,
 331                                   nsat_axis);
 332                         if (stats.has_gps_sats) {
 333                                 addSeries("GPS Satellites in View",
 334                                           AltosGraphDataPoint.data_gps_nsat_view,
 335                                           nsat_units,
 336                                           gps_nsat_view_color,
 337                                           false,
 338                                           nsat_axis);
 339                         }
 340                         if (stats.has_gps_detail) {
 341                                 addSeries("GPS Course",
 342                                           AltosGraphDataPoint.data_gps_course,
 343                                           orient_units,
 344                                           gps_course_color,
 345                                           false,
 346                                           course_axis);
 347                                 addSeries("GPS Ground Speed",
 348                                           AltosGraphDataPoint.data_gps_ground_speed,
 349                                           AltosConvert.speed,
 350                                           gps_ground_speed_color,
 351                                           enable_gps,
 352                                           speed_axis);
 353                                 addSeries("GPS Climb Rate",
 354                                           AltosGraphDataPoint.data_gps_climb_rate,
 355                                           AltosConvert.speed,
 356                                           gps_climb_rate_color,
 357                                           enable_gps,
 358                                           speed_axis);
 359                         }
 360                         addSeries("GPS Position DOP",
 361                                   AltosGraphDataPoint.data_gps_pdop,
 362                                   dop_units,
 363                                   gps_pdop_color,
 364                                   false,
 365                                   dop_axis);
 366                         if (stats.has_gps_detail) {
 367                                 addSeries("GPS Horizontal DOP",
 368                                           AltosGraphDataPoint.data_gps_hdop,
 369                                           dop_units,
 370                                           gps_hdop_color,
 371                                           false,
 372                                           dop_axis);
 373                                 addSeries("GPS Vertical DOP",
 374                                           AltosGraphDataPoint.data_gps_vdop,
 375                                           dop_units,
 376                                           gps_vdop_color,
 377                                           false,
 378                                           dop_axis);
 379                         }
 380                 }
 381                 if (stats.has_rssi)
 382                         addSeries("Received Signal Strength",
 383                                   AltosGraphDataPoint.data_rssi,
 384                                   dbm_units,
 385                                   dbm_color,
 386                                   false,
 387                                   dbm_axis);
 388
 389                 if (stats.has_battery)
 390                         addSeries("Battery Voltage",
 391                                   AltosGraphDataPoint.data_battery_voltage,
 392                                   voltage_units,
 393                                   battery_voltage_color,
 394                                   false,
 395                                   voltage_axis);
 396
 397                 if (stats.has_flight_adc) {
 398                         addSeries("Temperature",
 399                                   AltosGraphDataPoint.data_temperature,
 400                                   AltosConvert.temperature,
 401                                   temperature_color,
 402                                   false,
 403                                   temperature_axis);
 404                         addSeries("Drogue Voltage",
 405                                   AltosGraphDataPoint.data_drogue_voltage,
 406                                   voltage_units,
 407                                   drogue_voltage_color,
 408                                   false,
 409                                   voltage_axis);
 410                         addSeries("Main Voltage",
 411                                   AltosGraphDataPoint.data_main_voltage,
 412                                   voltage_units,
 413                                   main_voltage_color,
 414                                   false,
 415                                   voltage_axis);
 416                 }
 417
 418                 if (stats.has_imu) {
 419                         addSeries("Acceleration Along",
 420                                   AltosGraphDataPoint.data_accel_along,
 421                                   AltosConvert.accel,
 422                                   accel_x_color,
 423                                   false,
 424                                   accel_axis);
 425                         addSeries("Acceleration Across",
 426                                   AltosGraphDataPoint.data_accel_across,
 427                                   AltosConvert.accel,
 428                                   accel_y_color,
 429                                   false,
 430                                   accel_axis);
 431                         addSeries("Acceleration Through",
 432                                   AltosGraphDataPoint.data_accel_through,
 433                                   AltosConvert.accel,
 434                                   accel_z_color,
 435                                   false,
 436                                   accel_axis);
 437                         addSeries("Roll Rate",
 438                                   AltosGraphDataPoint.data_gyro_roll,
 439                                   gyro_units,
 440                                   gyro_x_color,
 441                                   false,
 442                                   gyro_axis);
 443                         addSeries("Pitch Rate",
 444                                   AltosGraphDataPoint.data_gyro_pitch,
 445                                   gyro_units,
 446                                   gyro_y_color,
 447                                   false,
 448                                   gyro_axis);
 449                         addSeries("Yaw Rate",
 450                                   AltosGraphDataPoint.data_gyro_yaw,
 451                                   gyro_units,
 452                                   gyro_z_color,
 453                                   false,
 454                                   gyro_axis);
 455                 }
 456                 if (stats.has_mag) {
 457                         addSeries("Magnetometer Along",
 458                                   AltosGraphDataPoint.data_mag_along,
 459                                   mag_units,
 460                                   mag_x_color,
 461                                   false,
 462                                   mag_axis);
 463                         addSeries("Magnetometer Across",
 464                                   AltosGraphDataPoint.data_mag_across,
 465                                   mag_units,
 466                                   mag_y_color,
 467                                   false,
 468                                   mag_axis);
 469                         addSeries("Magnetometer Through",
 470                                   AltosGraphDataPoint.data_mag_through,
 471                                   mag_units,
 472                                   mag_z_color,
 473                                   false,
 474                                   mag_axis);
 475                 }
 476                 if (stats.has_orient)
 477                         addSeries("Tilt Angle",
 478                                   AltosGraphDataPoint.data_orient,
 479                                   orient_units,
 480                                   orient_color,
 481                                   false,
 482                                   orient_axis);
 483                 if (stats.num_ignitor > 0) {
 484                         for (int i = 0; i < stats.num_ignitor; i++)
 485                                 addSeries(AltosLib.ignitor_name(i),
 486                                           AltosGraphDataPoint.data_ignitor_0 + i,
 487                                           voltage_units,
 488                                           main_voltage_color,
 489                                           false,
 490                                           voltage_axis);
 491                         for (int i = 0; i < stats.num_ignitor; i++)
 492                                 addMarker(AltosLib.ignitor_name(i), AltosGraphDataPoint.data_ignitor_fired_0 + i, state_color);
 493                 }
 494
 495                 setDataSet(dataSet);
 496         }
 497 }