altos: Add TelePyro v0.1 support
[fw/altos] / src / avr / ao_adc_avr.c
1 /*
2  * Copyright © 2011 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
20 volatile __xdata struct ao_adc  ao_adc_ring[AO_ADC_RING];
21 volatile __data uint8_t         ao_adc_head;
22
23 #ifdef TELESCIENCE
24 const uint8_t   adc_channels[AO_LOG_TELESCIENCE_NUM_ADC] = {
25         0x00,
26         0x01,
27         0x04,
28         0x05,
29         0x06,
30         0x07,
31         0x20,
32         0x21,
33         0x22,
34         0x23,
35         0x24,
36         0x25,
37 };
38 #endif
39
40 #ifdef TELEPYRO
41 const uint8_t   adc_channels[AO_TELEPYRO_NUM_ADC] = {
42         0x00,   /* ADC0  v_batt */
43         0x04,   /* ADC4  sense_a */
44         0x05,   /* ADC5  sense_b */
45         0x06,   /* ADC6  sense_c */
46         0x07,   /* ADC7  sense_d */
47         0x23,   /* ADC11 sense_e */
48         0x22,   /* ADC10 sense_f */
49         0x21,   /* ADC9 sense_g */
50 };
51 #endif
52
53 #define NUM_ADC (sizeof (adc_channels) / sizeof (adc_channels[0]))
54
55 static uint8_t  ao_adc_channel;
56
57 #define ADC_CHANNEL_LOW(c)      (((c) & 0x1f) << MUX0)
58 #define ADC_CHANNEL_HIGH(c)     ((((c) & 0x20) >> 5) << MUX5)
59
60 #define ADCSRA_INIT     ((1 << ADEN) |          /* Enable ADC */                \
61                          (0 << ADATE) |         /* No auto ADC trigger */       \
62                          (1 << ADIF) |          /* Clear interrupt */           \
63                          (0 << ADIE) |          /* Enable interrupt */          \
64                          (6 << ADPS0))          /* Prescale clock by 64 */
65
66 #define ADCSRB_INIT     ((0 << ADHSM) |         /* No high-speed mode */ \
67                          (0 << ACME) |          /* Some comparitor thing */ \
68                          (2 << ADTS0))          /* Free running mode (don't care) */
69
70 static void
71 ao_adc_start(void)
72 {
73         uint8_t channel = adc_channels[ao_adc_channel];
74         ADMUX = ((0 << REFS1) |                         /* AVcc reference */
75                  (1 << REFS0) |                         /* AVcc reference */
76                  (1 << ADLAR) |                         /* Left-shift results */
77                  (ADC_CHANNEL_LOW(channel)));           /* Select channel */
78
79         ADCSRB = (ADCSRB_INIT |
80                   ADC_CHANNEL_HIGH(channel));           /* High channel bit */
81
82         ADCSRA = (ADCSRA_INIT |
83                   (1 << ADSC) |
84                   (1 << ADIE));                         /* Start conversion */
85 }
86
87 ISR(ADC_vect)
88 {
89         uint16_t        value;
90
91         /* Must read ADCL first or the value there will be lost */
92         value = ADCL;
93         value |= (ADCH << 8);
94         ao_adc_ring[ao_adc_head].adc[ao_adc_channel] = value;
95         if (++ao_adc_channel < NUM_ADC)
96                 ao_adc_start();
97         else {
98                 ADCSRA = ADCSRA_INIT;
99                 ao_adc_ring[ao_adc_head].tick = ao_time();
100                 ao_adc_head = ao_adc_ring_next(ao_adc_head);
101                 ao_wakeup((void *) &ao_adc_head);
102                 ao_cpu_sleep_disable = 0;
103         }
104 }
105
106 void
107 ao_adc_poll(void)
108 {
109         ao_cpu_sleep_disable = 1;
110         ao_adc_channel = 0;
111         ao_adc_start();
112 }
113
114 void
115 ao_adc_get(__xdata struct ao_adc *packet)
116 {
117         uint8_t i = ao_adc_ring_prev(ao_adc_head);
118         memcpy(packet, (void *) &ao_adc_ring[i], sizeof (struct ao_adc));
119 }
120
121 static void
122 ao_adc_dump(void) __reentrant
123 {
124         static __xdata struct ao_adc    packet;
125         uint8_t i;
126         ao_adc_get(&packet);
127         printf("tick: %5u",  packet.tick);
128         for (i = 0; i < NUM_ADC; i++)
129                 printf (" %2d: %5u", i, packet.adc[i]);
130         printf ("\n");
131 }
132
133 __code struct ao_cmds ao_adc_cmds[] = {
134         { ao_adc_dump,  "a\0Display current ADC values" },
135         { 0, NULL },
136 };
137
138 void
139 ao_adc_init(void)
140 {
141         DIDR0 = 0xf3;
142         DIDR2 = 0x3f;
143         ADCSRB = ADCSRB_INIT;
144         ADCSRA = ADCSRA_INIT;
145         ao_cmd_register(&ao_adc_cmds[0]);
146 }