]> www.fi.muni.cz Git - bike-lights.git/blobdiff - firmware/pattern.c
mudflap for dual rearlights
[bike-lights.git] / firmware / pattern.c
index 3aec77b947bec7b52edea8f6b1cd470504a54b66..b4edcc4309a4df77b3a5a57f79c1307b255e3712 100644 (file)
 #include <avr/io.h>
+#include <stdlib.h> // for NULL
 
 #include "lights.h"
 
 static unsigned char led_counters[N_LEDS];
 static pattern_t *led_patterns[N_LEDS];
 
-pattern_t off_pattern[] = {
-       { 0, PATTERN_FOREVER },
+static unsigned char fibonacci[8] = {
+       0, 1, 2, 3, 5, 8, 13, 21,
 };
 
-pattern_t zero_pattern[] = {
-       { 0, 0x10 },
+static pattern_t boot_pattern[] = {
+       { 1, D_5 },
+       { 0, D_5 },
+       { 1, D_3 },
+       { 0, D_3 },
+       { 1, D_2 },
+       { 0, D_2 },
+       { 1, D_1 },
+       { 0, D_1 },
+       { 1, D_1 },
+       { 0, D_1 },
+       { 1, D_1 },
+       { 0, D_1 },
+       { 1, D_1 },
+       { 0, D_1 },
+       { 1, D_8 },
+       { 0, D_8 },
        PATTERN_END
 };
 
-pattern_t on_pattern[] = {
-       { 1, PATTERN_FOREVER },
-};
-
-pattern_t blink_pattern[] = {
-       { 1, 0x4 },
-       { 0, 0x8 },
+static pattern_t pattern_num[] = {
+       { 0, D_5 },
+       { 1, D_1 }, /* 10 */
+       { 0, D_5 },
+       { 1, D_1 }, /*  9 */
+       { 0, D_5 },
+       { 1, D_1 }, /*  8 */
+       { 0, D_5 },
+       { 1, D_1 }, /*  7 */
+       { 0, D_5 },
+       { 1, D_1 }, /*  6 */
+       { 0, D_5 },
+       { 1, D_1 }, /*  5 */
+       { 0, D_5 },
+       { 1, D_1 }, /*  4 */
+       { 0, D_5 },
+       { 1, D_1 }, /*  3 */
+       { 0, D_5 },
+       { 1, D_1 }, /*  2 */
+       { 0, D_5 },
+       { 1, D_1 }, /*  1 */
+       { 0, D_13 },
        PATTERN_END
 };
 
-pattern_t mode1_pattern[] = {
-       { 2, 0x1 },
-       { 0, 0x1 },
-       { 2, 0x1 },
-       { 0, 0x8 },
-       { 1, 0x1 },
-       { 0, 0x1 },
-       { 1, 0x1 },
-       { 0, 0x8 },
+static pattern_t pattern_invnum[] = {
+       { 1, D_5 },
+       { 0, D_1 }, /* 10 */
+       { 1, D_5 },
+       { 0, D_1 }, /*  9 */
+       { 1, D_5 },
+       { 0, D_1 }, /*  8 */
+       { 1, D_5 },
+       { 0, D_1 }, /*  7 */
+       { 1, D_5 },
+       { 0, D_1 }, /*  6 */
+       { 1, D_5 },
+       { 0, D_1 }, /*  5 */
+       { 1, D_5 },
+       { 0, D_1 }, /*  4 */
+       { 1, D_5 },
+       { 0, D_1 }, /*  3 */
+       { 1, D_5 },
+       { 0, D_1 }, /*  2 */
+       { 1, D_5 },
+       { 0, D_1 }, /*  1 */
+       { 1, D_13 },
        PATTERN_END
 };
 
-pattern_t boot_pattern[] = {
-       { 1, 0x6 },
-       { 0, 0x6 },
-       { 1, 0x3 },
-       { 0, 0x3 },
-       { 1, 0x2 },
-       { 0, 0x2 },
-       { 1, 0x1 },
-       { 0, 0x1 },
-       { 1, 0x1 },
-       { 0, 0x1 },
-       { 1, 0x1 },
-       { 0, 0x1 },
-       { 1, 0x1 },
-       { 0, 0x1 },
-       { 1, 0x10 },
-       { 0, 0x10 },
+pattern_t off_pattern[] = {
+       { 0, D_1 },
        PATTERN_END
 };
 
-pattern_t pattern_num[] = {
-       { 0, 0x5 },
-       { 1, 0x1 }, /* 10 */
-       { 0, 0x5 },
-       { 1, 0x1 }, /*  9 */
-       { 0, 0x5 },
-       { 1, 0x1 }, /*  8 */
-       { 0, 0x5 },
-       { 1, 0x1 }, /*  7 */
-       { 0, 0x5 },
-       { 1, 0x1 }, /*  6 */
-       { 0, 0x5 },
-       { 1, 0x1 }, /*  5 */
-       { 0, 0x5 },
-       { 1, 0x1 }, /*  4 */
-       { 0, 0x5 },
-       { 1, 0x1 }, /*  3 */
-       { 0, 0x5 },
-       { 1, 0x1 }, /*  2 */
-       { 0, 0x5 },
-       { 1, 0x1 }, /*  1 */
-       { 0, 0x1E },
-       PATTERN_END
-};
+/*
+ * This is tricky: we use a single pattern for all three pwmleds,
+ * but on some occasions, we want to be able to modify only a single
+ * pwmled status without affecting other outputs. For example, during
+ * braking, we want to modify only the rear pwmled status. We don't
+ * use a separate "braking" pattern for every other pattern used, but instead
+ * we change the pwmled0 status regardless of the original value when
+ * braking. The rule is the following:
+ * - if during braking the pwmled2 (front) is at mode 2, we switch it to
+ *   mode 3 (which has the same target current) to avoid flicker.
+ * - if pwmled0 (rear) is off, we set it to mode 2
+ *     (if it is with mode 1, we keep it at mode 1)
+ * TODO: something similar should be done for the "entering the dark area"
+ *     condition, where we want to switch pwmled2 (front) on, to mode 2.
+ */
+void pwmleds_update_mode()
+{
+       unsigned char mode, mode0, mode1, mode2;
 
-static unsigned char test_running;
+       mode = led_patterns[0]->mode;
 
-static void led_set_mode(unsigned char n, unsigned char mode)
-{
-       if (n < N_PWMLEDS) {
-               pwmled_set_mode(n, mode);
-       } else if (n < N_LEDS) {
-               gpio_set(n - N_PWMLEDS, mode);
+       mode0 = mode & 3;
+       mode1 = (mode >> 2) & 1;
+       mode2 = (mode >> 3) & 3;
+
+       if (err_flags.braking && !err_flags.err_battery) {
+               mode0 = 2;
+               if (mode2 == 2)
+                       mode2 = 3;
        }
+
+       pwmled_set_mode(0, mode0);
+       pwmled_set_mode(1, mode1);
+       pwmled_set_mode(2, mode2);
 }
 
 void led_set_pattern(unsigned char n, pattern_t *pattern)
@@ -98,8 +127,14 @@ void led_set_pattern(unsigned char n, pattern_t *pattern)
                pattern = off_pattern;
 
        led_patterns[n] = pattern;
-       led_counters[n] = pattern->duration;
-       led_set_mode(n, pattern->mode);
+
+       led_counters[n] = fibonacci[pattern->duration_fib];
+
+       if (n == 0) {
+               pwmleds_update_mode();
+       } else if (n < N_LEDS) {
+               gpio_set(n - 1, pattern->mode);
+       }
 }
 
 void init_pattern()
@@ -107,30 +142,65 @@ void init_pattern()
        unsigned char i;
 
        for (i = 0; i < N_LEDS; i++)
-               led_set_pattern(i, zero_pattern);
+               led_set_pattern(i, NULL);
 
-       led_set_pattern(N_PWMLEDS+1, boot_pattern);
-       test_running = 0;
+       led_set_pattern(N_ILLUM_LED, boot_pattern);
 }
 
-pattern_t *number_pattern(unsigned char num)
+pattern_t *number_pattern(unsigned char num, unsigned char inv)
 {
-       if (num >= 9)
-               num = 9;
+       if (num >= 10)
+               num = 10;
+
+       if (inv) {
+               return pattern_invnum
+                       + sizeof(pattern_invnum)/sizeof(pattern_t)
+                       - 2 - 2*num;
+       } else {
+               return pattern_num
+                       + sizeof(pattern_num)/sizeof(pattern_t)
+                       - 2 - 2*num;
+       }
+}
 
-       return pattern_num + sizeof(pattern_num)/sizeof(pattern_t)
-               - 2 - 2*num;
+static pattern_t *pattern_select(unsigned char n)
+{
+       switch(n) {
+       case 0: return pwmled_pattern_select();
+       case N_STATUS_LED: return status_led_pattern_select();
+       case N_ILLUM_LED:  return illumination_led_pattern_select();
+       case N_LASER_LED:  return laser_pattern_select();
+       default: return NULL;
+       }
 }
 
-static inline pattern_t *pattern_select(unsigned char n)
+void pattern_reload()
 {
-       if (n < N_PWMLEDS)
-               return mode1_pattern;
-       else if (n == 3)
-               return status_pattern_select(0);
-       else if (n == 4)
-               return status_pattern_select(1);
-       return off_pattern;
+       unsigned char i;
+
+       for (i = 0; i < N_LEDS; i++)
+               led_set_pattern(i, pattern_select(i));
+}
+
+static void inline pattern_finished(unsigned char n)
+{
+       unsigned char i;
+
+       led_patterns[n] = NULL;
+
+       if (n == 0) {
+               led_set_pattern(0, pattern_select(0));
+       } else if (n == N_STATUS_LED) {
+               if (!led_patterns[N_ILLUM_LED])
+                       led_set_pattern(N_ILLUM_LED,
+                               pattern_select(N_ILLUM_LED));
+       } else if (n == N_ILLUM_LED) {
+               if (!led_patterns[N_STATUS_LED])
+                       led_set_pattern(N_STATUS_LED,
+                               pattern_select(N_STATUS_LED));
+       } else {
+               led_set_pattern(n, pattern_select(n));
+       }
 }
 
 void patterns_next_tick()
@@ -138,17 +208,22 @@ void patterns_next_tick()
        unsigned char i;
 
        for (i = 0; i < N_LEDS; i++) {
-               if (led_counters[i] == 0) {
+               if (!led_patterns[i]) {
+                       pattern_finished(i);
+                       continue;
+               }
+
+               if (--led_counters[i] == 0) {
                        pattern_t *p = led_patterns[i];
                        p++;
-                       if (p->duration == 0) { // END
-                               p = pattern_select(i);
+                       if (p->duration_fib == 0) { // END
+                               /* Keep the last state, wait for others */
+                               pattern_finished(i);
+                               continue;
                        }
                        led_set_pattern(i, p);
                }
 
-               if (led_counters[i] < PATTERN_FOREVER)
-                       led_counters[i]--;
        }
 }