git.gag.com Git - fw/altos/blob - ao-tools/altosui/AltosGraphUI.java

   1
   2 // Copyright (c) 2010 Anthony Towns
   3 // GPL v2 or later
   4
   5 package altosui;
   6
   7 import java.io.*;
   8 import java.util.ArrayList;
   9
  10 import javax.swing.JFrame;
  11 import java.awt.Color;
  12
  13 import org.jfree.chart.ChartPanel;
  14 import org.jfree.chart.ChartUtilities;
  15 import org.jfree.chart.JFreeChart;
  16 import org.jfree.chart.axis.AxisLocation;
  17 import org.jfree.ui.ApplicationFrame;
  18 import org.jfree.ui.RefineryUtilities;
  19
  20 public class AltosGraphUI extends JFrame
  21 {
  22     static final private Color red = new Color(194,31,31);
  23     static final private Color green = new Color(31,194,31);
  24     static final private Color blue = new Color(31,31,194);
  25     static final private Color black = new Color(31,31,31);
  26
  27     static private class OverallGraphs {
  28         AltosGraphTime.Element height =
  29             new AltosGraphTime.TimeSeries("Height (m)", "Height (AGL)", red) {
  30                 public void gotTimeData(double time, AltosDataPoint d) {
  31                     series.add(time, d.height());
  32                 }
  33             };
  34
  35         AltosGraphTime.Element speed =
  36             new AltosGraphTime.TimeSeries("Speed (m/s)", "Vertical Speed", green) {
  37                 public void gotTimeData(double time, AltosDataPoint d) {
  38                     if (d.state() < Altos.ao_flight_drogue) {
  39                         series.add(time, d.accel_speed());
  40                     } else {
  41                         series.add(time, d.baro_speed());
  42                     }
  43                 }
  44             };
  45
  46         AltosGraphTime.Element acceleration =
  47             new AltosGraphTime.TimeSeries("Acceleration (m/s\u00B2)",
  48                     "Axial Acceleration", blue)
  49             {
  50                 public void gotTimeData(double time, AltosDataPoint d) {
  51                     series.add(time, d.acceleration());
  52                 }
  53             };
  54
  55         AltosGraphTime.Element temperature =
  56             new AltosGraphTime.TimeSeries("Temperature (\u00B0C)",
  57                     "Board temperature", red)
  58             {
  59                 public void gotTimeData(double time, AltosDataPoint d) {
  60                     series.add(time, d.temperature());
  61                 }
  62             };
  63
  64         AltosGraphTime.Element drogue_voltage =
  65             new AltosGraphTime.TimeSeries("Voltage (V)", "Drogue Continuity", blue)
  66             {
  67                 public void gotTimeData(double time, AltosDataPoint d) {
  68                     series.add(time, d.drogue_voltage());
  69                 }
  70             };
  71
  72         AltosGraphTime.Element main_voltage =
  73             new AltosGraphTime.TimeSeries("Voltage (V)", "Main Continuity", green)
  74             {
  75                 public void gotTimeData(double time, AltosDataPoint d) {
  76                     series.add(time, d.main_voltage());
  77                 }
  78             };
  79
  80         AltosGraphTime.Element e_pad    = new AltosGraphTime.StateMarker(Altos.ao_flight_pad, "Pad");
  81         AltosGraphTime.Element e_boost  = new AltosGraphTime.StateMarker(Altos.ao_flight_boost, "Boost");
  82         AltosGraphTime.Element e_fast   = new AltosGraphTime.StateMarker(Altos.ao_flight_fast, "Fast");
  83         AltosGraphTime.Element e_coast  = new AltosGraphTime.StateMarker(Altos.ao_flight_coast, "Coast");
  84         AltosGraphTime.Element e_drogue = new AltosGraphTime.StateMarker(Altos.ao_flight_drogue, "Drogue");
  85         AltosGraphTime.Element e_main   = new AltosGraphTime.StateMarker(Altos.ao_flight_main, "Main");
  86         AltosGraphTime.Element e_landed = new AltosGraphTime.StateMarker(Altos.ao_flight_landed, "Landed");
  87
  88         protected AltosGraphTime myAltosGraphTime(String suffix) {
  89             return (new AltosGraphTime("Overall " + suffix))
  90                 .addElement(e_boost)
  91                 .addElement(e_drogue)
  92                 .addElement(e_main)
  93                 .addElement(e_landed);
  94         }
  95
  96         public ArrayList<AltosGraph> graphs() {
  97             ArrayList<AltosGraph> graphs = new ArrayList<AltosGraph>();
  98
  99             graphs.add( myAltosGraphTime("Summary")
 100                     .addElement(height)
 101                     .addElement(speed)
 102                     .addElement(acceleration) );
 103
 104             graphs.add( myAltosGraphTime("Altitude")
 105                     .addElement(height) );
 106
 107             graphs.add( myAltosGraphTime("Speed")
 108                     .addElement(speed) );
 109
 110             graphs.add( myAltosGraphTime("Acceleration")
 111                     .addElement(acceleration) );
 112
 113             graphs.add( myAltosGraphTime("Temperature")
 114                     .addElement(temperature) );
 115
 116             graphs.add( myAltosGraphTime("Continuity")
 117                     .addElement(drogue_voltage)
 118                     .addElement(main_voltage) );
 119
 120             return graphs;
 121         }
 122     }
 123
 124     static private class AscentGraphs extends OverallGraphs {
 125         protected AltosGraphTime myAltosGraphTime(String suffix) {
 126             return (new AltosGraphTime("Ascent " + suffix) {
 127                 public void addData(AltosDataPoint d) {
 128                     int state = d.state();
 129                     if (Altos.ao_flight_boost <= state && state <= Altos.ao_flight_coast) {
 130                         super.addData(d);
 131                     }
 132                 }
 133             }).addElement(e_boost)
 134               .addElement(e_fast)
 135               .addElement(e_coast);
 136         }
 137     }
 138
 139     static private class DescentGraphs extends OverallGraphs {
 140         protected AltosGraphTime myAltosGraphTime(String suffix) {
 141             return (new AltosGraphTime("Descent " + suffix) {
 142                 public void addData(AltosDataPoint d) {
 143                     int state = d.state();
 144                     if (Altos.ao_flight_drogue <= state && state <= Altos.ao_flight_main) {
 145                         super.addData(d);
 146                     }
 147                 }
 148             }).addElement(e_drogue)
 149               .addElement(e_main);
 150             // ((XYGraph)graph[8]).ymin = new Double(-50);
 151         }
 152     }
 153
 154     public AltosGraphUI(JFrame frame)
 155     {
 156         super("Altos Graph");
 157
 158         AltosGraphDataChooser chooser;
 159         chooser = new AltosGraphDataChooser(frame);
 160         Iterable<AltosDataPoint> reader = chooser.runDialog();
 161         if (reader == null)
 162             return;
 163
 164         init(reader, 0);
 165     }
 166
 167     public AltosGraphUI(Iterable<AltosDataPoint> data, int which)
 168     {
 169         super("Altos Graph");
 170         init(data, which);
 171     }
 172
 173     private void init(Iterable<AltosDataPoint> data, int which) {
 174         AltosGraph graph = createGraph(data, which);
 175
 176         JFreeChart chart = graph.createChart();
 177         ChartPanel chartPanel = new ChartPanel(chart);
 178         chartPanel.setMouseWheelEnabled(true);
 179         chartPanel.setPreferredSize(new java.awt.Dimension(800, 500));
 180         setContentPane(chartPanel);
 181
 182         pack();
 183
 184         RefineryUtilities.centerFrameOnScreen(this);
 185
 186         setDefaultCloseOperation(DISPOSE_ON_CLOSE);
 187         setVisible(true);
 188     }
 189
 190     private static AltosGraph createGraph(Iterable<AltosDataPoint> data,
 191             int which)
 192     {
 193         return createGraphsWhich(data, which).get(0);
 194     }
 195
 196     private static ArrayList<AltosGraph> createGraphs(
 197             Iterable<AltosDataPoint> data)
 198     {
 199         return createGraphsWhich(data, -1);
 200     }
 201
 202     private static ArrayList<AltosGraph> createGraphsWhich(
 203             Iterable<AltosDataPoint> data, int which)
 204     {
 205         ArrayList<AltosGraph> graph = new ArrayList<AltosGraph>();
 206         graph.addAll((new OverallGraphs()).graphs());
 207         graph.addAll((new AscentGraphs()).graphs());
 208         graph.addAll((new DescentGraphs()).graphs());
 209
 210         if (which > 0) {
 211             if (which >= graph.size()) {
 212                 which = 0;
 213             }
 214             AltosGraph g = graph.get(which);
 215             graph = new ArrayList<AltosGraph>();
 216             graph.add(g);
 217         }
 218
 219         for (AltosDataPoint dp : data) {
 220             for (AltosGraph g : graph) {
 221                 g.addData(dp);
 222             }
 223         }
 224
 225         return graph;
 226     }
 227 }
 228
 229 /* gnuplot bits...
 230  *
 231 300x400
 232
 233 --------------------------------------------------------
 234 TOO HARD! :)
 235
 236 "ascent-gps-accuracy.png" "Vertical error margin to apogee - GPS v Baro (m)"
 237     5:($7 < 6 ? $24-$11 : 1/0)
 238 "descent-gps-accuracy.png" "Vertical error margin during descent - GPS v Baro (m)"
 239     5:($7 < 6 ? 1/0 : $24-$11)
 240
 241 set output "overall-gps-accuracy.png"
 242 set ylabel "distance above sea level (m)"
 243 plot "telemetry.csv" using 5:11 with lines ti "baro altitude" axis x1y1, \
 244     "telemetry.csv" using 5:24 with lines ti "gps altitude" axis x1y1
 245
 246 set term png tiny size 700,700 enhanced
 247 set xlabel "m"
 248 set ylabel "m"
 249 set polar
 250 set grid polar
 251 set rrange[*:*]
 252 set angles degrees
 253
 254 set output "overall-gps-path.png"
 255 #:30 with yerrorlines
 256 plot "telemetry.csv" using (90-$33):($7 == 2 ? $31 : 1/0) with lines ti "pad", \
 257     "telemetry.csv" using (90-$33):($7 == 3 ? $31 : 1/0) with lines ti "boost", \
 258     "telemetry.csv" using (90-$33):($7 == 4 ? $31 : 1/0) with lines ti "fast", \
 259     "telemetry.csv" using (90-$33):($7 == 5 ? $31 : 1/0) with lines ti "coast", \
 260     "telemetry.csv" using (90-$33):($7 == 6 ? $31 : 1/0) with lines ti "drogue", \
 261     "telemetry.csv" using (90-$33):($7 == 7 ? $31 : 1/0) with lines ti "main", \
 262     "telemetry.csv" using (90-$33):($7 == 8 ? $31 : 1/0) with lines ti "landed"
 263
 264 set output "ascent-gps-path.png"
 265 plot "telemetry.csv" using (90-$33):($7 == 2 ? $31 : 1/0):30 with lines ti "pad", \
 266     "telemetry.csv" using (90-$33):($7 == 3 ? $31 : 1/0):20 with lines ti "boost", \
 267     "telemetry.csv" using (90-$33):($7 == 4 ? $31 : 1/0):10 with lines ti "fast", \
 268     "telemetry.csv" using (90-$33):($7 == 5 ? $31 : 1/0):5 with lines ti "coast"
 269
 270 set output "descent-gps-path.png"
 271 plot "telemetry.csv" using (90-$33):($7 == 6 ? $31 : 1/0) with lines ti "drogue", \
 272     "telemetry.csv" using (90-$33):($7 == 7 ? $31 : 1/0) with lines ti "main", \
 273     "telemetry.csv" using (90-$33):($7 == 8 ? $31 : 1/0) with lines ti "landed"
 274
 275  */
 276
 277