a7033dbba5eb02136065d87f9aef8f7e5c2c11d6
[fw/altos] / src / stm / ao_led.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; 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 #include "ao.h"
20
21 #if LED_PER_LED
22 static const struct {
23         struct stm_gpio *port;
24         uint16_t        pin;
25 } ao_leds[] = {
26 #ifdef LED_0_PORT
27     [0] { LED_0_PORT, LED_0_PIN },
28 #endif
29 #ifdef LED_1_PORT
30     [1] { LED_1_PORT, LED_1_PIN },
31 #endif
32 #ifdef LED_2_PORT
33     [2] { LED_2_PORT, LED_2_PIN },
34 #endif
35 #ifdef LED_3_PORT
36     [3] { LED_3_PORT, LED_3_PIN },
37 #endif
38 #ifdef LED_4_PORT
39     [4] { LED_4_PORT, LED_4_PIN },
40 #endif
41 #ifdef LED_5_PORT
42     [5] { LED_5_PORT, LED_5_PIN },
43 #endif
44 #ifdef LED_6_PORT
45     [6] { LED_6_PORT, LED_6_PIN },
46 #endif
47 #ifdef LED_7_PORT
48     [7] { LED_7_PORT, LED_7_PIN },
49 #endif
50 #ifdef LED_8_PORT
51     [8] { LED_8_PORT, LED_8_PIN },
52 #endif
53 #ifdef LED_9_PORT
54     [9] { LED_9_PORT, LED_9_PIN },
55 #endif
56 #ifdef LED_10_PORT
57     [10] { LED_10_PORT, LED_10_PIN },
58 #endif
59 #ifdef LED_11_PORT
60     [11] { LED_11_PORT, LED_11_PIN },
61 #endif
62 #ifdef LED_12_PORT
63     [12] { LED_12_PORT, LED_12_PIN },
64 #endif
65 #ifdef LED_13_PORT
66     [13] { LED_13_PORT, LED_13_PIN },
67 #endif
68 #ifdef LED_14_PORT
69     [14] { LED_14_PORT, LED_14_PIN },
70 #endif
71 #ifdef LED_15_PORT
72     [15] { LED_15_PORT, LED_15_PIN },
73 #endif
74 };
75 #define N_LED   (sizeof (ao_leds)/sizeof(ao_leds[0]))
76 #endif
77 static AO_LED_TYPE ao_led_enable;
78
79 void
80 ao_led_on(AO_LED_TYPE colors)
81 {
82 #ifdef LED_PER_LED
83         AO_LED_TYPE i;
84         for (i = 0; i < N_LED; i++)
85                 if (colors & (1 << i))
86                         ao_gpio_set(ao_leds[i].port, ao_leds[i].pin, foo, 1);
87 #else
88 #ifdef LED_PORT
89         LED_PORT->bsrr = (colors & ao_led_enable);
90 #else
91 #ifdef LED_PORT_0
92         LED_PORT_0->bsrr = ((colors & ao_led_enable) & LED_PORT_0_MASK) << LED_PORT_0_SHIFT;
93 #endif
94 #ifdef LED_PORT_1
95         LED_PORT_1->bsrr = ((colors & ao_led_enable) & LED_PORT_1_MASK) << LED_PORT_1_SHIFT;
96 #endif
97 #endif
98 #endif
99 }
100
101 void
102 ao_led_off(AO_LED_TYPE colors)
103 {
104 #ifdef LED_PER_LED
105         AO_LED_TYPE i;
106         for (i = 0; i < N_LED; i++)
107                 if (colors & (1 << i))
108                         ao_gpio_set(ao_leds[i].port, ao_leds[i].pin, foo, 0);
109 #else
110 #ifdef LED_PORT
111         LED_PORT->bsrr = (uint32_t) (colors & ao_led_enable) << 16;
112 #else
113 #ifdef LED_PORT_0
114         LED_PORT_0->bsrr = ((uint32_t) (colors & ao_led_enable) & LED_PORT_0_MASK) << (LED_PORT_0_SHIFT + 16);
115 #endif
116 #ifdef LED_PORT_1
117         LED_PORT_1->bsrr = ((uint32_t) (colors & ao_led_enable) & LED_PORT_1_MASK) << (LED_PORT_1_SHIFT + 16);
118 #endif
119 #endif
120 #endif
121 }
122
123 void
124 ao_led_set(AO_LED_TYPE colors)
125 {
126         AO_LED_TYPE     on = colors & ao_led_enable;
127         AO_LED_TYPE     off = ~colors & ao_led_enable;
128
129         ao_led_off(off);
130         ao_led_on(on);
131 }
132
133 void
134 ao_led_toggle(AO_LED_TYPE colors)
135 {
136 #ifdef LED_PER_LED
137         AO_LED_TYPE i;
138         for (i = 0; i < N_LED; i++)
139                 if (colors & (1 << i))
140                         ao_gpio_set(ao_leds[i].port, ao_leds[i].pin, foo, ~ao_gpio_get(ao_leds[i].port, ao_leds[i].pin, foo));
141 #else
142 #ifdef LED_PORT
143         LED_PORT->odr ^= (colors & ao_led_enable);
144 #else
145 #ifdef LED_PORT_0
146         LED_PORT_0->odr ^= ((colors & ao_led_enable) & LED_PORT_0_MASK) << LED_PORT_0_SHIFT;
147 #endif
148 #ifdef LED_PORT_1
149         LED_PORT_1->odr ^= ((colors & ao_led_enable) & LED_PORT_1_MASK) << LED_PORT_1_SHIFT;
150 #endif
151 #endif
152 #endif
153 }
154
155 void
156 ao_led_for(AO_LED_TYPE colors, AO_LED_TYPE ticks) __reentrant
157 {
158         ao_led_on(colors);
159         ao_delay(ticks);
160         ao_led_off(colors);
161 }
162
163 #define init_led_pin(port, bit) do { \
164                 stm_moder_set(port, bit, STM_MODER_OUTPUT);             \
165                 stm_otyper_set(port, bit, STM_OTYPER_PUSH_PULL);        \
166         } while (0)
167
168 void
169 ao_led_init(AO_LED_TYPE enable)
170 {
171         AO_LED_TYPE     bit;
172
173         ao_led_enable = enable;
174 #if LED_PER_LED
175         for (bit = 0; bit < N_LED; bit++)
176                 ao_enable_output(ao_leds[bit].port, ao_leds[bit].pin, foo, 0);
177 #else
178 #ifdef LED_PORT
179         stm_rcc.ahbenr |= (1 << LED_PORT_ENABLE);
180         LED_PORT->odr &= ~enable;
181 #else
182 #ifdef LED_PORT_0
183         stm_rcc.ahbenr |= (1 << LED_PORT_0_ENABLE);
184         LED_PORT_0->odr &= ~((enable & ao_led_enable) & LED_PORT_0_MASK) << LED_PORT_0_SHIFT;
185 #endif
186 #ifdef LED_PORT_1
187         stm_rcc.ahbenr |= (1 << LED_PORT_1_ENABLE);
188         LED_PORT_1->odr &= ~((enable & ao_led_enable) & LED_PORT_1_MASK) << LED_PORT_1_SHIFT;
189 #endif
190 #ifdef LED_PORT_2
191         stm_rcc.ahbenr |= (1 << LED_PORT_1_ENABLE);
192         LED_PORT_1->odr &= ~((enable & ao_led_enable) & LED_PORT_1_MASK) << LED_PORT_1_SHIFT;
193 #endif
194 #endif
195         for (bit = 0; bit < 16; bit++) {
196                 if (enable & (1 << bit)) {
197 #ifdef LED_PORT
198                         init_led_pin(LED_PORT, bit);
199 #else
200 #ifdef LED_PORT_0
201                         if (LED_PORT_0_MASK & (1 << bit))
202                                 init_led_pin(LED_PORT_0, bit + LED_PORT_0_SHIFT);
203 #endif
204 #ifdef LED_PORT_1
205                         if (LED_PORT_1_MASK & (1 << bit))
206                                 init_led_pin(LED_PORT_1, bit + LED_PORT_1_SHIFT);
207 #endif
208 #ifdef LED_PORT_2
209                         if (LED_PORT_2_MASK & (1 << bit))
210                                 init_led_pin(LED_PORT_2, bit + LED_PORT_2_SHIFT);
211 #endif
212 #endif
213                 }
214         }
215 #endif
216 }