altosuilib: Remove 'cal_data' param from AltosGraphNew constructor
[fw/altos] / 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                 height_axis = newAxis("Height", AltosConvert.height, height_color);
94                 pressure_axis = newAxis("Pressure", AltosConvert.pressure, pressure_color, 0);
95                 speed_axis = newAxis("Speed", AltosConvert.speed, speed_color);
96                 thrust_axis = newAxis("Thrust", AltosConvert.force, thrust_color);
97                 accel_axis = newAxis("Acceleration", AltosConvert.accel, accel_color);
98                 voltage_axis = newAxis("Voltage", AltosConvert.voltage, voltage_color);
99                 temperature_axis = newAxis("Temperature", AltosConvert.temperature, temperature_color, 0);
100                 nsat_axis = newAxis("Satellites", null, gps_nsat_color,
101                                     AltosUIAxis.axis_include_zero | AltosUIAxis.axis_integer);
102                 dbm_axis = newAxis("Signal Strength", null, dbm_color, 0);
103                 distance_axis = newAxis("Distance", AltosConvert.distance, range_color);
104
105                 gyro_axis = newAxis("Rotation Rate", AltosConvert.rotation_rate, gyro_roll_color, 0);
106                 orient_axis = newAxis("Tilt Angle", AltosConvert.orient, orient_color, 0);
107                 mag_axis = newAxis("Magnetic Field", AltosConvert.magnetic_field, mag_along_color, 0);
108                 course_axis = newAxis("Course", AltosConvert.orient, gps_course_color, 0);
109                 dop_axis = newAxis("Dilution of Precision", dop_units, gps_pdop_color, 0);
110
111                 flight_series.register_axis("default",
112                                             speed_color,
113                                             false,
114                                             speed_axis);
115
116                 flight_series.register_marker(AltosUIFlightSeries.state_name,
117                                               state_color,
118                                               true,
119                                               plot,
120                                               true);
121
122                 flight_series.register_marker(AltosUIFlightSeries.pyro_fired_name,
123                                               igniter_marker_color,
124                                               true,
125                                               plot,
126                                               false);
127
128                 flight_series.register_axis(AltosUIFlightSeries.accel_name,
129                                             accel_color,
130                                             true,
131                                             accel_axis);
132
133                 flight_series.register_axis(AltosUIFlightSeries.kalman_accel_name,
134                                             kalman_accel_color,
135                                             false,
136                                             accel_axis);
137
138                 flight_series.register_axis(AltosUIFlightSeries.rssi_name,
139                                             dbm_color,
140                                             false,
141                                             dbm_axis);
142
143                 flight_series.register_axis(AltosUIFlightSeries.speed_name,
144                                             speed_color,
145                                             true,
146                                             speed_axis);
147
148                 flight_series.register_axis(AltosUIFlightSeries.kalman_speed_name,
149                                             kalman_speed_color,
150                                             true,
151                                             speed_axis);
152
153                 flight_series.register_axis(AltosUIFlightSeries.pressure_name,
154                                             pressure_color,
155                                             false,
156                                             pressure_axis);
157
158                 flight_series.register_axis(AltosUIFlightSeries.height_name,
159                                             height_color,
160                                             true,
161                                             height_axis);
162
163                 flight_series.register_axis(AltosUIFlightSeries.altitude_name,
164                                             height_color,
165                                             false,
166                                             height_axis);
167
168                 flight_series.register_axis(AltosUIFlightSeries.kalman_height_name,
169                                             kalman_height_color,
170                                             false,
171                                             height_axis);
172
173
174                 flight_series.register_axis(AltosUIFlightSeries.temperature_name,
175                                             temperature_color,
176                                             false,
177                                             temperature_axis);
178
179                 flight_series.register_axis(AltosUIFlightSeries.battery_voltage_name,
180                                             battery_voltage_color,
181                                             false,
182                                             voltage_axis);
183
184                 flight_series.register_axis(AltosUIFlightSeries.apogee_voltage_name,
185                                             drogue_voltage_color,
186                                             false,
187                                             voltage_axis);
188
189                 flight_series.register_axis(AltosUIFlightSeries.main_voltage_name,
190                                             main_voltage_color,
191                                             false,
192                                             voltage_axis);
193
194                 flight_series.register_axis(AltosUIFlightSeries.sats_in_view_name,
195                                             gps_nsat_view_color,
196                                             false,
197                                             nsat_axis);
198
199                 flight_series.register_axis(AltosUIFlightSeries.sats_in_soln_name,
200                                             gps_nsat_solution_color,
201                                             false,
202                                             nsat_axis);
203
204                 flight_series.register_axis(AltosUIFlightSeries.gps_pdop_name,
205                                             gps_pdop_color,
206                                             false,
207                                             dop_axis);
208
209                 flight_series.register_axis(AltosUIFlightSeries.gps_hdop_name,
210                                             gps_hdop_color,
211                                             false,
212                                             dop_axis);
213
214                 flight_series.register_axis(AltosUIFlightSeries.gps_vdop_name,
215                                             gps_vdop_color,
216                                             false,
217                                             dop_axis);
218
219                 flight_series.register_axis(AltosUIFlightSeries.gps_altitude_name,
220                                             gps_height_color,
221                                             false,
222                                             height_axis);
223
224                 flight_series.register_axis(AltosUIFlightSeries.gps_height_name,
225                                             gps_height_color,
226                                             false,
227                                             height_axis);
228
229                 flight_series.register_axis(AltosUIFlightSeries.gps_ground_speed_name,
230                                             gps_ground_speed_color,
231                                             false,
232                                             speed_axis);
233
234                 flight_series.register_axis(AltosUIFlightSeries.gps_ascent_rate_name,
235                                             gps_climb_rate_color,
236                                             false,
237                                             speed_axis);
238
239                 flight_series.register_axis(AltosUIFlightSeries.gps_course_name,
240                                             gps_course_color,
241                                             false,
242                                             course_axis);
243
244                 flight_series.register_axis(AltosUIFlightSeries.gps_speed_name,
245                                             gps_speed_color,
246                                             false,
247                                             speed_axis);
248
249                 flight_series.register_axis(AltosUIFlightSeries.accel_along_name,
250                                             accel_along_color,
251                                             false,
252                                             accel_axis);
253
254                 flight_series.register_axis(AltosUIFlightSeries.accel_across_name,
255                                             accel_across_color,
256                                             false,
257                                             accel_axis);
258
259                 flight_series.register_axis(AltosUIFlightSeries.accel_through_name,
260                                             accel_through_color,
261                                             false,
262                                             accel_axis);
263
264                 flight_series.register_axis(AltosUIFlightSeries.gyro_roll_name,
265                                             gyro_roll_color,
266                                             false,
267                                             gyro_axis);
268
269                 flight_series.register_axis(AltosUIFlightSeries.gyro_pitch_name,
270                                             gyro_pitch_color,
271                                             false,
272                                             gyro_axis);
273
274                 flight_series.register_axis(AltosUIFlightSeries.gyro_yaw_name,
275                                             gyro_yaw_color,
276                                             false,
277                                             gyro_axis);
278
279                 flight_series.register_axis(AltosUIFlightSeries.mag_along_name,
280                                             mag_along_color,
281                                             false,
282                                             mag_axis);
283
284                 flight_series.register_axis(AltosUIFlightSeries.mag_across_name,
285                                             mag_across_color,
286                                             false,
287                                             mag_axis);
288
289                 flight_series.register_axis(AltosUIFlightSeries.mag_through_name,
290                                             mag_through_color,
291                                             false,
292                                             mag_axis);
293
294                 flight_series.register_axis(AltosUIFlightSeries.orient_name,
295                                             orient_color,
296                                             false,
297                                             orient_axis);
298
299                 for (int channel = 0; channel < 26; channel++) {
300                         flight_series.register_axis(flight_series.igniter_voltage_name(channel),
301                                                     igniter_voltage_color,
302                                                     false,
303                                                     voltage_axis);
304                 }
305
306                 flight_series.register_axis(AltosUIFlightSeries.thrust_name,
307                                             thrust_color,
308                                             true,
309                                             thrust_axis);
310
311                 return flight_series.series(cal_data);
312         }
313
314         public void set_data(AltosFlightStats stats, AltosUIFlightSeries flight_series) {
315                 set_series(setup(stats, flight_series));
316         }
317
318         public AltosGraphNew(AltosUIEnable enable) {
319                 super(enable, "Flight");
320         }
321
322         public AltosGraphNew(AltosUIEnable enable, AltosFlightStats stats, AltosUIFlightSeries flight_series) {
323                 this(enable);
324                 set_series(setup(stats, flight_series));
325         }
326 }