20a2a413da4f9499a7f1ce052dd52f882440b0c9
[fw/altos] / altoslib / AltosIMU.java
1 /*
2  * Copyright © 2012 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.altoslib_11;
20
21 import java.util.concurrent.*;
22 import java.io.*;
23
24 public class AltosIMU implements Cloneable {
25         public int              accel_along;
26         public int              accel_across;
27         public int              accel_through;
28
29         public int              gyro_roll;
30         public int              gyro_pitch;
31         public int              gyro_yaw;
32
33         public static final double      counts_per_g = 2048.0;
34
35         public static double convert_accel(double counts) {
36                 return counts / counts_per_g * AltosConvert.gravity;
37         }
38
39         public static final double      counts_per_degsec = 16.4;
40
41         public static double convert_gyro(double counts) {
42                 return counts / counts_per_degsec;
43         }
44
45         public boolean parse_string(String line) {
46                 if (!line.startsWith("Accel:"))
47                         return false;
48
49                 String[] items = line.split("\\s+");
50
51                 if (items.length >= 8) {
52                         accel_along = Integer.parseInt(items[1]);
53                         accel_across = Integer.parseInt(items[2]);
54                         accel_through = Integer.parseInt(items[3]);
55                         gyro_roll = Integer.parseInt(items[5]);
56                         gyro_pitch = Integer.parseInt(items[6]);
57                         gyro_yaw = Integer.parseInt(items[7]);
58                 }
59                 return true;
60         }
61
62         public AltosIMU clone() {
63                 AltosIMU        n = new AltosIMU();
64
65                 n.accel_along = accel_along;
66                 n.accel_across = accel_across;
67                 n.accel_through = accel_through;
68
69                 n.gyro_roll = gyro_roll;
70                 n.gyro_pitch = gyro_pitch;
71                 n.gyro_yaw = gyro_yaw;
72                 return n;
73         }
74
75         static public void provide_data(AltosDataListener listener, AltosLink link, AltosCalData cal_data) throws InterruptedException {
76                 try {
77                         AltosIMU        imu = new AltosIMU(link);
78
79                         if (imu != null) {
80                                 listener.set_accel(cal_data.accel_along(imu.accel_along),
81                                                    cal_data.accel_across(imu.accel_across),
82                                                    cal_data.accel_through(imu.accel_through));
83                                 listener.set_gyro(cal_data.gyro_roll(imu.gyro_roll),
84                                                   cal_data.gyro_pitch(imu.gyro_pitch),
85                                                   cal_data.gyro_yaw(imu.gyro_yaw));
86                         }
87                 } catch (TimeoutException te) {
88                 }
89         }
90
91         public AltosIMU() {
92                 accel_along = AltosLib.MISSING;
93                 accel_across = AltosLib.MISSING;
94                 accel_through = AltosLib.MISSING;
95
96                 gyro_roll = AltosLib.MISSING;
97                 gyro_pitch = AltosLib.MISSING;
98                 gyro_yaw = AltosLib.MISSING;
99         }
100
101         public AltosIMU(int accel_along, int accel_across, int accel_through,
102                         int gyro_roll, int gyro_pitch, int gyro_yaw) {
103
104                 this.accel_along = accel_along;
105                 this.accel_across = accel_across;
106                 this.accel_through = accel_through;
107
108                 this.gyro_roll = gyro_roll;
109                 this.gyro_pitch = gyro_pitch;
110                 this.gyro_yaw = gyro_yaw;
111         }
112
113         public AltosIMU(AltosLink link) throws InterruptedException, TimeoutException {
114                 this();
115                 link.printf("I\n");
116                 for (;;) {
117                         String line = link.get_reply_no_dialog(5000);
118                         if (line == null) {
119                                 throw new TimeoutException();
120                         }
121                         if (parse_string(line))
122                                 break;
123                 }
124         }
125 }