firmware/pattern.c

   1 #include <avr/io.h>
   2 #include <stdlib.h> // for NULL
   3
   4 #include "lights.h"
   5
   6 static unsigned char led_counters[N_LEDS];
   7 static pattern_t *led_patterns[N_LEDS];
   8
   9 static pattern_t boot_pattern[] = {
  10         { 1, 0x6 },
  11         { 0, 0x6 },
  12         { 1, 0x3 },
  13         { 0, 0x3 },
  14         { 1, 0x2 },
  15         { 0, 0x2 },
  16         { 1, 0x1 },
  17         { 0, 0x1 },
  18         { 1, 0x1 },
  19         { 0, 0x1 },
  20         { 1, 0x1 },
  21         { 0, 0x1 },
  22         { 1, 0x1 },
  23         { 0, 0x1 },
  24         { 1, 0x10 },
  25         { 0, 0x10 },
  26         PATTERN_END
  27 };
  28
  29 static pattern_t pattern_num[] = {
  30         { 0, 0x5 },
  31         { 1, 0x1 }, /* 10 */
  32         { 0, 0x5 },
  33         { 1, 0x1 }, /*  9 */
  34         { 0, 0x5 },
  35         { 1, 0x1 }, /*  8 */
  36         { 0, 0x5 },
  37         { 1, 0x1 }, /*  7 */
  38         { 0, 0x5 },
  39         { 1, 0x1 }, /*  6 */
  40         { 0, 0x5 },
  41         { 1, 0x1 }, /*  5 */
  42         { 0, 0x5 },
  43         { 1, 0x1 }, /*  4 */
  44         { 0, 0x5 },
  45         { 1, 0x1 }, /*  3 */
  46         { 0, 0x5 },
  47         { 1, 0x1 }, /*  2 */
  48         { 0, 0x5 },
  49         { 1, 0x1 }, /*  1 */
  50         { 0, 0xF },
  51         PATTERN_END
  52 };
  53
  54 static pattern_t pattern_invnum[] = {
  55         { 1, 0x5 },
  56         { 0, 0x1 }, /* 10 */
  57         { 1, 0x5 },
  58         { 0, 0x1 }, /*  9 */
  59         { 1, 0x5 },
  60         { 0, 0x1 }, /*  8 */
  61         { 1, 0x5 },
  62         { 0, 0x1 }, /*  7 */
  63         { 1, 0x5 },
  64         { 0, 0x1 }, /*  6 */
  65         { 1, 0x5 },
  66         { 0, 0x1 }, /*  5 */
  67         { 1, 0x5 },
  68         { 0, 0x1 }, /*  4 */
  69         { 1, 0x5 },
  70         { 0, 0x1 }, /*  3 */
  71         { 1, 0x5 },
  72         { 0, 0x1 }, /*  2 */
  73         { 1, 0x5 },
  74         { 0, 0x1 }, /*  1 */
  75         { 1, 0xF },
  76         PATTERN_END
  77 };
  78
  79 pattern_t off_pattern[] = {
  80         { 0, 0x1 },
  81         PATTERN_END
  82 };
  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
 100         led_counters[n] = pattern->duration;
 101         led_set_mode(n, pattern->mode);
 102 }
 103
 104 void init_pattern()
 105 {
 106         unsigned char i;
 107
 108         for (i = 0; i < N_LEDS; i++)
 109                 led_set_pattern(i, NULL);
 110
 111         led_set_pattern(N_PWMLEDS+1, boot_pattern);
 112 }
 113
 114 pattern_t *number_pattern(unsigned char num, unsigned char inv)
 115 {
 116         if (num >= 10)
 117                 num = 10;
 118
 119         if (inv) {
 120                 return pattern_invnum
 121                         + sizeof(pattern_invnum)/sizeof(pattern_t)
 122                         - 2 - 2*num;
 123         } else {
 124                 return pattern_num
 125                         + sizeof(pattern_num)/sizeof(pattern_t)
 126                         - 2 - 2*num;
 127         }
 128 }
 129
 130 static pattern_t *pattern_select(unsigned char n)
 131 {
 132         switch(n) {
 133         case 0: return pwmled0_pattern_select();
 134         case 1: return pwmled1_pattern_select();
 135         case 2: return pwmled2_pattern_select();
 136         case 3: return status_led_pattern_select();
 137         case 4: return illumination_led_pattern_select();
 138         case 6: return laser_pattern_select();
 139         default: return NULL;
 140         }
 141 }
 142
 143 void pattern_reload()
 144 {
 145         unsigned char i;
 146
 147         for (i = 0; i < N_LEDS; i++)
 148                 led_set_pattern(i, pattern_select(i));
 149 }
 150
 151 static void inline pattern_finished(unsigned char n)
 152 {
 153         unsigned char i;
 154
 155         led_patterns[n] = NULL;
 156
 157         if (n < N_PWMLEDS) {
 158                 for (i = 0; i < N_PWMLEDS; i++)
 159                         if (led_patterns[i])
 160                                 return;
 161
 162                 /* all pwmleds finished; restart them */
 163                 for (i = 0; i < N_PWMLEDS; i++)
 164                         led_set_pattern(i, pattern_select(i));
 165         } else if (n == 3) {
 166                 if (!led_patterns[4])
 167                         led_set_pattern(4, pattern_select(4));
 168         } else if (n == 4) {
 169                 if (!led_patterns[3])
 170                         led_set_pattern(3, pattern_select(3));
 171         } else {
 172                 led_set_pattern(n, pattern_select(n));
 173         }
 174 }
 175
 176 void patterns_next_tick()
 177 {
 178         unsigned char i;
 179
 180         for (i = 0; i < N_LEDS; i++) {
 181                 if (!led_patterns[i]) {
 182                         pattern_finished(i);
 183                         continue;
 184                 }
 185
 186                 if (--led_counters[i] == 0) {
 187                         pattern_t *p = led_patterns[i];
 188                         p++;
 189                         if (p->duration == 0) { // END
 190                                 /* Keep the last state, wait for others */
 191                                 pattern_finished(i);
 192                                 continue;
 193                         }
 194                         led_set_pattern(i, p);
 195                 }
 196
 197         }
 198 }
 199