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