2d217bcf9d392fbd70099beaa3f21eee6ff8f871
[fw/altos] / src / drivers / ao_hmc5883.c
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; 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 #include <ao.h>
19 #include <ao_hmc5883.h>
20 #include <ao_exti.h>
21
22 #if HAS_HMC5883
23
24 static uint8_t  ao_hmc5883_configured;
25
26 static uint8_t  ao_hmc5883_addr;
27
28 static void
29 ao_hmc5883_reg_write(uint8_t addr, uint8_t data)
30 {
31         uint8_t d[2];
32
33         d[0] = addr;
34         d[1] = data;
35         ao_i2c_get(AO_HMC5883_I2C_INDEX);
36         ao_i2c_start(AO_HMC5883_I2C_INDEX, HMC5883_ADDR_WRITE);
37         ao_i2c_send(d, 2, AO_HMC5883_I2C_INDEX, TRUE);
38         ao_i2c_put(AO_HMC5883_I2C_INDEX);
39         ao_hmc5883_addr = addr + 1;
40 }
41
42 static void
43 ao_hmc5883_read(uint8_t addr, uint8_t *data, uint8_t len)
44 {
45         ao_i2c_get(AO_HMC5883_I2C_INDEX);
46         if (addr != ao_hmc5883_addr) {
47                 ao_i2c_start(AO_HMC5883_I2C_INDEX, HMC5883_ADDR_WRITE);
48                 ao_i2c_send(&addr, 1, AO_HMC5883_I2C_INDEX, FALSE);
49         }
50         ao_i2c_start(AO_HMC5883_I2C_INDEX, HMC5883_ADDR_READ);
51         ao_i2c_recv(data, len, AO_HMC5883_I2C_INDEX, TRUE);
52         ao_i2c_put(AO_HMC5883_I2C_INDEX);
53         ao_hmc5883_addr = 0xff;
54 }
55
56 static uint8_t ao_hmc5883_done;
57
58 static void
59 ao_hmc5883_isr(void)
60 {
61         ao_exti_disable(AO_HMC5883_INT_PORT, AO_HMC5883_INT_PIN);
62         ao_hmc5883_done = 1;
63         ao_wakeup(&ao_hmc5883_done);
64 }
65
66 static uint32_t ao_hmc5883_missed_irq;
67
68 void
69 ao_hmc5883_sample(struct ao_hmc5883_sample *sample)
70 {
71         uint16_t        *d = (uint16_t *) sample;
72         int             i = sizeof (*sample) / 2;
73
74         ao_hmc5883_done = 0;
75         ao_exti_enable(AO_HMC5883_INT_PORT, AO_HMC5883_INT_PIN);
76         ao_hmc5883_reg_write(HMC5883_MODE, HMC5883_MODE_SINGLE);
77
78         ao_alarm(AO_MS_TO_TICKS(10));
79         ao_arch_block_interrupts();
80         while (!ao_hmc5883_done)
81                 if (ao_sleep(&ao_hmc5883_done))
82                         ++ao_hmc5883_missed_irq;
83         ao_arch_release_interrupts();
84         ao_clear_alarm();
85
86         ao_hmc5883_read(HMC5883_X_MSB, (uint8_t *) sample, sizeof (struct ao_hmc5883_sample));
87 #if __BYTE_ORDER == __LITTLE_ENDIAN
88         /* byte swap */
89         while (i--) {
90                 uint16_t        t = *d;
91                 *d++ = (t >> 8) | (t << 8);
92         }
93 #endif
94 }
95
96 static uint8_t
97 ao_hmc5883_setup(void)
98 {
99         uint8_t d;
100         uint8_t present;
101
102         if (ao_hmc5883_configured)
103                 return 1;
104
105         ao_i2c_get(AO_HMC5883_I2C_INDEX);
106         present = ao_i2c_start(AO_HMC5883_I2C_INDEX, HMC5883_ADDR_READ);
107         ao_i2c_recv(&d, 1, AO_HMC5883_I2C_INDEX, TRUE);
108         ao_i2c_put(AO_HMC5883_I2C_INDEX);
109
110         if (!present)
111                 ao_panic(AO_PANIC_SELF_TEST_HMC5883);
112
113         ao_hmc5883_reg_write(HMC5883_CONFIG_A,
114                              (HMC5883_CONFIG_A_MA_8 << HMC5883_CONFIG_A_MA) |
115                              (HMC5883_CONFIG_A_DO_15 << HMC5883_CONFIG_A_DO) |
116                              (HMC5883_CONFIG_A_MS_NORMAL << HMC5883_CONFIG_A_MS));
117
118         ao_hmc5883_reg_write(HMC5883_CONFIG_B,
119                              (HMC5883_CONFIG_B_GN_1_3 << HMC5883_CONFIG_B_GN));
120
121         ao_hmc5883_configured = 1;
122         return 1;
123 }
124
125 struct ao_hmc5883_sample ao_hmc5883_current;
126
127 static void
128 ao_hmc5883(void)
129 {
130         ao_hmc5883_setup();
131         for (;;) {
132                 ao_hmc5883_sample(&ao_hmc5883_current);
133                 ao_arch_critical(
134                         AO_DATA_PRESENT(AO_DATA_HMC5883);
135                         AO_DATA_WAIT();
136                         );
137         }
138 }
139
140 static struct ao_task ao_hmc5883_task;
141
142 static void
143 ao_hmc5883_show(void)
144 {
145         struct ao_data  sample;
146         ao_data_get(&sample);
147         printf ("X: %d Y: %d Z: %d missed irq: %lu\n",
148                 sample.hmc5883.x, sample.hmc5883.y, sample.hmc5883.z, ao_hmc5883_missed_irq);
149 }
150
151 static const struct ao_cmds ao_hmc5883_cmds[] = {
152         { ao_hmc5883_show,      "M\0Show HMC5883 status" },
153         { 0, NULL }
154 };
155
156 void
157 ao_hmc5883_init(void)
158 {
159         ao_hmc5883_configured = 0;
160
161         ao_enable_port(AO_HMC5883_INT_PORT);
162         ao_exti_setup(AO_HMC5883_INT_PORT,
163                       AO_HMC5883_INT_PIN,
164                       AO_EXTI_MODE_FALLING | AO_EXTI_MODE_PULL_UP,
165                       ao_hmc5883_isr);
166
167         ao_add_task(&ao_hmc5883_task, ao_hmc5883, "hmc5883");
168         ao_cmd_register(&ao_hmc5883_cmds[0]);
169 }
170
171 #endif