]> www.fi.muni.cz Git - bike-lights.git/blob - firmware/pattern.c
78f988034fad04bfe0432aeac18d0a566f23c31c
[bike-lights.git] / firmware / pattern.c
1 #include <avr/io.h>
2
3 #include "lights.h"
4
5 typedef struct {
6         unsigned char mode: 3;
7         unsigned char duration: 5;
8 } pattern_t;
9
10 static unsigned char led_counters[N_LEDS];
11 static pattern_t *led_patterns[N_LEDS];
12
13 #define PATTERN_END { 0, 0 }
14 pattern_t off_pattern[] = {
15         { 0, 0x1F },
16         PATTERN_END
17 };
18
19 pattern_t blink_pattern[] = {
20         { 1, 0x4 },
21         { 0, 0x8 },
22         PATTERN_END
23 };
24
25 pattern_t mode1_pattern[] = {
26         { 4, 0x1 },
27         { 0, 0x1 },
28         { 4, 0x1 },
29         { 0, 0x8 },
30         { 1, 0x1 },
31         { 0, 0x1 },
32         { 1, 0x1 },
33         { 0, 0x8 },
34         PATTERN_END
35 };
36
37 pattern_t boot_pattern[] = {
38         { 1, 0x6 },
39         { 0, 0x6 },
40         { 1, 0x3 },
41         { 0, 0x3 },
42         { 1, 0x2 },
43         { 0, 0x2 },
44         { 1, 0x1 },
45         { 0, 0x1 },
46         { 1, 0x1 },
47         { 0, 0x1 },
48         { 1, 0x1 },
49         { 0, 0x1 },
50         { 1, 0x1 },
51         { 0, 0x1 },
52         { 1, 0x10 },
53         { 0, 0x10 },
54         PATTERN_END
55 };
56
57 pattern_t pattern_num[] = {
58         { 0, 0x5 },
59         { 1, 0x1 }, /* 10 */
60         { 0, 0x5 },
61         { 1, 0x1 }, /*  9 */
62         { 0, 0x5 },
63         { 1, 0x1 }, /*  8 */
64         { 0, 0x5 },
65         { 1, 0x1 }, /*  7 */
66         { 0, 0x5 },
67         { 1, 0x1 }, /*  6 */
68         { 0, 0x5 },
69         { 1, 0x1 }, /*  5 */
70         { 0, 0x5 },
71         { 1, 0x1 }, /*  4 */
72         { 0, 0x5 },
73         { 1, 0x1 }, /*  3 */
74         { 0, 0x5 },
75         { 1, 0x1 }, /*  2 */
76         { 0, 0x5 },
77         { 1, 0x1 }, /*  1 */
78         { 0, 0x1F },
79         PATTERN_END
80 };
81
82 static unsigned char test_running;
83
84 static void led_set_mode(unsigned char n, unsigned char mode)
85 {
86         if (n < N_PWMLEDS) {
87                 pwmled_set_mode(n, mode);
88         } else if (n < N_LEDS) {
89                 gpio_set(n - N_PWMLEDS, mode);
90         }
91 }
92
93 void led_set_pattern(unsigned char n, pattern_t *pattern)
94 {
95         if (!pattern)
96                 pattern = off_pattern;
97
98         led_patterns[n] = pattern;
99         led_counters[n] = pattern->duration;
100         led_set_mode(n, pattern->mode);
101 }
102
103 void pattern_init()
104 {
105         unsigned char i;
106
107         for (i = 0; i < N_LEDS; i++) {
108                 led_counters[i] = 0;
109                 led_patterns[i] = off_pattern;
110         }
111         led_set_pattern(N_PWMLEDS+1, boot_pattern);
112         test_running = 0;
113 }
114
115 static pattern_t *number_pattern(unsigned char num)
116 {
117         if (num >= 9)
118                 num = 9;
119
120         return pattern_num + sizeof(pattern_num)/sizeof(pattern_t)
121                 - 2 - 2*num;
122 }
123
124 static inline pattern_t *pattern_select(unsigned char n)
125 {
126         if (n < N_PWMLEDS && !pwmled_enabled(n))
127                 return off_pattern; // Don't mess with non-enabled LEDs
128         else if (n == 2)
129                 return mode1_pattern;
130         else if (n == 3)
131                 return number_pattern(1+ambient_zone);
132         return off_pattern;
133 }
134
135 void patterns_next_tick()
136 {
137         unsigned char i;
138
139         for (i = 0; i < N_LEDS; i++) {
140                 if (led_counters[i] == 0) {
141                         pattern_t *p = led_patterns[i];
142                         p++;
143                         if (p->duration == 0) { // END
144                                 p = pattern_select(i);
145                         }
146                         led_set_pattern(i, p);
147                 }
148
149                 led_counters[i]--;
150         }
151 }
152
153 void led_set_status(unsigned char status)
154 {
155         led_set_pattern(N_PWMLEDS+1, number_pattern(status));
156 }