altoslib: Correct sign of telemega v4 pitch
[fw/altos] / altoslib / AltosEepromRecordMega.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_13;
16
17 public class AltosEepromRecordMega extends AltosEepromRecord {
18         public static final int record_length = 32;
19
20         public static final int max_sat = 12;
21
22         private int log_format;
23
24         /* AO_LOG_FLIGHT elements */
25         private int flight() { return data16(0); }
26         private int ground_accel() { return data16(2); }
27         private int ground_pres() { return data32(4); }
28         private int ground_accel_along() { return data16(8); }
29         private int ground_accel_across() { return data16(10); }
30         private int ground_accel_through() { return data16(12); }
31         private int ground_roll() {
32                 switch (log_format) {
33                 case AltosLib.AO_LOG_FORMAT_TELEMEGA:
34                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_3:
35                 case AltosLib.AO_LOG_FORMAT_EASYMEGA_2:
36                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_4:
37                         return data32(16);
38                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_OLD:
39                         return data16(14);
40                 default:
41                         return AltosLib.MISSING;
42                 }
43         }
44         private int ground_pitch() {
45                 switch (log_format) {
46                 case AltosLib.AO_LOG_FORMAT_TELEMEGA:
47                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_3:
48                 case AltosLib.AO_LOG_FORMAT_EASYMEGA_2:
49                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_4:
50                         return data32(20);
51                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_OLD:
52                         return data16(16);
53                 default:
54                         return AltosLib.MISSING;
55                 }
56         }
57         private int ground_yaw() {
58                 switch (log_format) {
59                 case AltosLib.AO_LOG_FORMAT_TELEMEGA:
60                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_3:
61                 case AltosLib.AO_LOG_FORMAT_EASYMEGA_2:
62                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_4:
63                         return data32(24);
64                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_OLD:
65                         return data16(18);
66                 default:
67                         return AltosLib.MISSING;
68                 }
69         }
70
71         /* AO_LOG_STATE elements */
72         private int state() { return data16(0); }
73         private int reason() { return data16(2); }
74
75         /* AO_LOG_SENSOR elements */
76         private int pres() { return data32(0); }
77         private int temp() { return data32(4); }
78         private int accel_x() { return data16(8); }
79         private int accel_y() { return data16(10); }
80         private int accel_z() { return data16(12); }
81         private int gyro_x() { return data16(14); }
82         private int gyro_y() { return data16(16); }
83         private int gyro_z() { return data16(18); }
84         private int mag_x() { return data16(20); }
85         private int mag_z() { return data16(22); }
86         private int mag_y() { return data16(24); }
87
88         private int imu_type() {
89                 switch (log_format) {
90                 case AltosLib.AO_LOG_FORMAT_TELEMEGA:
91                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_OLD:
92                         return AltosIMU.imu_type_telemega_v1_v2;
93                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_3:
94                         return AltosIMU.imu_type_telemega_v3;
95                 case AltosLib.AO_LOG_FORMAT_EASYMEGA_2:
96                         return AltosIMU.imu_type_easymega_v2;
97                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_4:
98                         return AltosIMU.imu_type_telemega_v4;
99                 default:
100                         return AltosLib.MISSING;
101                 }
102         }
103
104         private int accel_across() {
105                 switch (log_format) {
106                 case AltosLib.AO_LOG_FORMAT_TELEMEGA:
107                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_3:
108                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_OLD:
109                         return accel_x();
110                 case AltosLib.AO_LOG_FORMAT_EASYMEGA_2:
111                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_4:
112                         return -accel_y();
113                 default:
114                         return AltosLib.MISSING;
115                 }
116         }
117
118         private int accel_along(){
119                 switch (log_format) {
120                 case AltosLib.AO_LOG_FORMAT_TELEMEGA:
121                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_3:
122                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_OLD:
123                         return accel_y();
124                 case AltosLib.AO_LOG_FORMAT_EASYMEGA_2:
125                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_4:
126                         return accel_x();
127                 default:
128                         return AltosLib.MISSING;
129                 }
130         }
131
132         private int accel_through() {
133                 return accel_z();
134         }
135
136         private int gyro_pitch() {
137                 switch (log_format) {
138                 case AltosLib.AO_LOG_FORMAT_TELEMEGA:
139                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_3:
140                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_OLD:
141                         return gyro_x();
142                 case AltosLib.AO_LOG_FORMAT_EASYMEGA_2:
143                         return -gyro_y();
144                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_4:
145                         return -gyro_y();
146                 default:
147                         return AltosLib.MISSING;
148                 }
149         }
150
151         private int gyro_roll() {
152                 switch (log_format) {
153                 case AltosLib.AO_LOG_FORMAT_TELEMEGA:
154                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_3:
155                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_OLD:
156                         return gyro_y();
157                 case AltosLib.AO_LOG_FORMAT_EASYMEGA_2:
158                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_4:
159                         return gyro_x();
160                 default:
161                         return AltosLib.MISSING;
162                 }
163         }
164
165         private int gyro_yaw() {
166                 return gyro_z();
167         }
168
169         private int mag_across() {
170                 switch (log_format) {
171                 case AltosLib.AO_LOG_FORMAT_TELEMEGA:
172                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_3:
173                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_OLD:
174                         return mag_x();
175                 case AltosLib.AO_LOG_FORMAT_EASYMEGA_2:
176                         return -mag_y();
177                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_4:
178                         return mag_y();
179                 default:
180                         return AltosLib.MISSING;
181                 }
182         }
183
184         private int mag_along() {
185                 switch (log_format) {
186                 case AltosLib.AO_LOG_FORMAT_TELEMEGA:
187                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_3:
188                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_OLD:
189                         return mag_y();
190                 case AltosLib.AO_LOG_FORMAT_EASYMEGA_2:
191                 case AltosLib.AO_LOG_FORMAT_TELEMEGA_4:
192                         return mag_x();
193                 default:
194                         return AltosLib.MISSING;
195                 }
196         }
197
198         private int mag_through() {
199                 return mag_z();
200         }
201
202
203         private int accel() { return data16(26); }
204
205         /* AO_LOG_TEMP_VOLT elements */
206         private int v_batt() { return data16(0); }
207         private int v_pbatt() { return data16(2); }
208         private int nsense() { return data16(4); }
209         private int sense(int i) { return data16(6 + i * 2); }
210         private int pyro() { return data16(26); }
211
212         /* AO_LOG_GPS_TIME elements */
213         private int latitude() { return data32(0); }
214         private int longitude() { return data32(4); }
215         private int altitude_low() { return data16(8); }
216         private int hour() { return data8(10); }
217         private int minute() { return data8(11); }
218         private int second() { return data8(12); }
219         private int flags() { return data8(13); }
220         private int year() { return data8(14); }
221         private int month() { return data8(15); }
222         private int day() { return data8(16); }
223         private int course() { return data8(17); }
224         private int ground_speed() { return data16(18); }
225         private int climb_rate() { return data16(20); }
226         private int pdop() { return data8(22); }
227         private int hdop() { return data8(23); }
228         private int vdop() { return data8(24); }
229         private int mode() { return data8(25); }
230         private int altitude_high() { return data16(26); }
231
232         /* AO_LOG_GPS_SAT elements */
233         private int nsat() { return data16(0); }
234         private int svid(int n) { return data8(2 + n * 2); }
235         private int c_n(int n) { return data8(2 + n * 2 + 1); }
236
237         public void provide_data(AltosDataListener listener, AltosCalData cal_data) {
238                 super.provide_data(listener, cal_data);
239
240                 AltosGPS        gps;
241
242                 cal_data.set_imu_type(imu_type());
243
244                 switch (cmd()) {
245                 case AltosLib.AO_LOG_FLIGHT:
246                         cal_data.set_flight(flight());
247                         cal_data.set_ground_accel(ground_accel());
248                         cal_data.set_ground_pressure(ground_pres());
249                         listener.set_accel_ground(cal_data.accel_along(ground_accel_along()),
250                                                   cal_data.accel_across(ground_accel_across()),
251                                                   cal_data.accel_through(ground_accel_through()));
252                         cal_data.set_gyro_zero(ground_roll() / 512.0,
253                                                ground_pitch() / 512.0,
254                                                ground_yaw() / 512.0);
255                         break;
256                 case AltosLib.AO_LOG_STATE:
257                         listener.set_state(state());
258                         break;
259                 case AltosLib.AO_LOG_SENSOR:
260                         AltosConfigData config_data = eeprom.config_data();
261                         AltosPresTemp pt = config_data.ms5607().pres_temp(pres(), temp());;
262                         listener.set_pressure(pt.pres);
263                         listener.set_temperature(pt.temp);
264
265                         int     accel_along = accel_along();
266                         int     accel_across = accel_across();
267                         int     accel_through = accel_through();
268                         int     gyro_roll = gyro_roll();
269                         int     gyro_pitch = gyro_pitch();
270                         int     gyro_yaw = gyro_yaw();
271
272                         int     mag_along = mag_along();
273                         int     mag_across = mag_across();
274                         int     mag_through = mag_through();
275
276                         if (log_format == AltosLib.AO_LOG_FORMAT_TELEMEGA_OLD)
277                                 cal_data.check_imu_wrap(gyro_roll, gyro_pitch, gyro_yaw);
278
279                         listener.set_accel(cal_data.accel_along(accel_along),
280                                            cal_data.accel_across(accel_across),
281                                            cal_data.accel_through(accel_through));
282                         listener.set_gyro(cal_data.gyro_roll(gyro_roll),
283                                           cal_data.gyro_pitch(gyro_pitch),
284                                           cal_data.gyro_yaw(gyro_yaw));
285
286                         listener.set_mag(cal_data.mag_along(mag_along),
287                                          cal_data.mag_across(mag_across),
288                                          cal_data.mag_through(mag_through));
289
290
291                         final double lsb_per_g = 1920.0/105.5;
292
293                         double acceleration = AltosConvert.acceleration_from_sensor(
294                                 accel(),
295                                 cal_data.ground_accel,
296                                 cal_data.ground_accel + 2 * lsb_per_g,
297                                 cal_data.ground_accel);
298
299                         listener.set_acceleration(acceleration);
300                         break;
301                 case AltosLib.AO_LOG_TEMP_VOLT:
302                         listener.set_battery_voltage(AltosConvert.mega_battery_voltage(v_batt()));
303                         listener.set_pyro_voltage(AltosConvert.mega_pyro_voltage(v_pbatt()));
304
305                         int nsense = nsense();
306
307                         listener.set_apogee_voltage(AltosConvert.mega_pyro_voltage(sense(nsense-2)));
308                         listener.set_main_voltage(AltosConvert.mega_pyro_voltage(sense(nsense-1)));
309
310                         double voltages[] = new double[nsense-2];
311                         for (int i = 0; i < nsense-2; i++)
312                                 voltages[i] = AltosConvert.mega_pyro_voltage(sense(i));
313
314                         listener.set_igniter_voltage(voltages);
315                         listener.set_pyro_fired(pyro());
316                         break;
317                 case AltosLib.AO_LOG_GPS_TIME:
318                         gps = listener.make_temp_gps(false);
319                         gps.lat = latitude() / 1e7;
320                         gps.lon = longitude() / 1e7;
321
322                         if (config_data().altitude_32())
323                                 gps.alt = (altitude_low() & 0xffff) | (altitude_high() << 16);
324                         else
325                                 gps.alt = altitude_low();
326
327                         gps.hour = hour();
328                         gps.minute = minute();
329                         gps.second = second();
330
331                         int flags = flags();
332
333                         gps.connected = (flags & AltosLib.AO_GPS_RUNNING) != 0;
334                         gps.locked = (flags & AltosLib.AO_GPS_VALID) != 0;
335                         gps.nsat = (flags & AltosLib.AO_GPS_NUM_SAT_MASK) >>
336                                 AltosLib.AO_GPS_NUM_SAT_SHIFT;
337
338                         gps.year = 2000 + year();
339                         gps.month = month();
340                         gps.day = day();
341                         gps.ground_speed = ground_speed() * 1.0e-2;
342                         gps.course = course() * 2;
343                         gps.climb_rate = climb_rate() * 1.0e-2;
344                         if (config_data().compare_version("1.4.9") >= 0) {
345                                 gps.pdop = pdop() / 10.0;
346                                 gps.hdop = hdop() / 10.0;
347                                 gps.vdop = vdop() / 10.0;
348                         } else {
349                                 gps.pdop = pdop() / 100.0;
350                                 if (gps.pdop < 0.8)
351                                         gps.pdop += 2.56;
352                                 gps.hdop = hdop() / 100.0;
353                                 if (gps.hdop < 0.8)
354                                         gps.hdop += 2.56;
355                                 gps.vdop = vdop() / 100.0;
356                                 if (gps.vdop < 0.8)
357                                         gps.vdop += 2.56;
358                         }
359                         break;
360                 case AltosLib.AO_LOG_GPS_SAT:
361                         gps = listener.make_temp_gps(true);
362
363                         int n = nsat();
364                         if (n > max_sat)
365                                 n = max_sat;
366                         for (int i = 0; i < n; i++)
367                                 gps.add_sat(svid(i), c_n(i));
368                         break;
369                 }
370         }
371
372         public AltosEepromRecord next() {
373                 int     s = next_start();
374                 if (s < 0)
375                         return null;
376                 return new AltosEepromRecordMega(eeprom, s);
377         }
378
379         public AltosEepromRecordMega(AltosEeprom eeprom, int start) {
380                 super(eeprom, start, record_length);
381                 log_format = eeprom.config_data().log_format;
382         }
383
384         public AltosEepromRecordMega(AltosEeprom eeprom) {
385                 this(eeprom, 0);
386         }
387 }