light_scheduler.c

Go to the documentation of this file.

/*
 * Copyright (C) 2019 Xavier Paris
 *
 * This file is part of paparazzi.
 *
 * paparazzi is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2, or (at your option)
 * any later version.
 *
 * paparazzi is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with paparazzi; see the file COPYING.  If not, see
 * <http://www.gnu.org/licenses/>.
 *
 */
 
#include "modules/light/light_scheduler.h"
#include "modules/light/light_ws2812_arch.h"
#include "generated/airframe.h"
 
#ifdef LIGHT_SCHEDULER_ALERT_HEADER
#include LIGHT_SCHEDULER_ALERT_HEADER
#endif
 
#ifndef LIGHT_SCHEDULER_MODE
#define LIGHT_SCHEDULER_MODE LIGHT_SCHEDULER_MODE_RAINBOW
#endif
 
#ifndef LIGHT_SCHEDULER_RED
#define LIGHT_SCHEDULER_RED 255
#endif
 
#ifndef LIGHT_SCHEDULER_GREEN
#define LIGHT_SCHEDULER_GREEN 255
#endif
 
#ifndef LIGHT_SCHEDULER_BLUE
#define LIGHT_SCHEDULER_BLUE 255
#endif
 
#ifndef LIGHT_SCHEDULER_BRIGHTNESS
#define LIGHT_SCHEDULER_BRIGHTNESS 255
#endif
 
#ifndef LIGHT_SCHEDULER_COLOR
#define LIGHT_SCHEDULER_COLOR LIGHT_SCHEDULER_COLOR_CUSTOM
#endif
 
#ifndef LIGHT_SCHEDULER_ALERT_COLOR
#define LIGHT_SCHEDULER_ALERT_COLOR LIGHT_SCHEDULER_COLOR_RED
#endif
 
#ifndef LIGHT_SCHEDULER_ALERT_SPARKLE
#define LIGHT_SCHEDULER_ALERT_SPARKLE FALSE
#endif
 
uint8_t light_scheduler_mode = LIGHT_SCHEDULER_MODE;
uint8_t light_scheduler_red = LIGHT_SCHEDULER_RED;
uint8_t light_scheduler_green = LIGHT_SCHEDULER_GREEN;
uint8_t light_scheduler_blue = LIGHT_SCHEDULER_BLUE;
uint8_t light_scheduler_brightness = LIGHT_SCHEDULER_BRIGHTNESS;
uint8_t light_scheduler_color = LIGHT_SCHEDULER_COLOR;
 
static const uint8_t light_scheduler_presets[LIGHT_SCHEDULER_COLOR_FLAG_FIRST][3] = {
  [LIGHT_SCHEDULER_COLOR_CUSTOM]  = {255, 255, 255},
  [LIGHT_SCHEDULER_COLOR_RED]     = {255,   0,   0},
  [LIGHT_SCHEDULER_COLOR_GREEN]   = {  0, 255,   0},
  [LIGHT_SCHEDULER_COLOR_BLUE]    = {  0,   0, 255},
  [LIGHT_SCHEDULER_COLOR_WHITE]   = {255, 255, 255},
  [LIGHT_SCHEDULER_COLOR_YELLOW]  = {255, 180,   0},
  [LIGHT_SCHEDULER_COLOR_ORANGE]  = {255,  60,   0},
  [LIGHT_SCHEDULER_COLOR_CYAN]    = {  0, 255, 255},
  [LIGHT_SCHEDULER_COLOR_MAGENTA] = {255,   0, 255},
  [LIGHT_SCHEDULER_COLOR_PURPLE]  = {128,   0, 255},
};
 
static const uint8_t light_scheduler_flags[LIGHT_SCHEDULER_COLOR_NB - LIGHT_SCHEDULER_COLOR_FLAG_FIRST][3] = {
  [LIGHT_SCHEDULER_COLOR_FLAG_ITALY - LIGHT_SCHEDULER_COLOR_FLAG_FIRST] =
    {LIGHT_SCHEDULER_COLOR_RED, LIGHT_SCHEDULER_COLOR_WHITE, LIGHT_SCHEDULER_COLOR_GREEN},
  [LIGHT_SCHEDULER_COLOR_FLAG_NETHERLANDS - LIGHT_SCHEDULER_COLOR_FLAG_FIRST] =
    {LIGHT_SCHEDULER_COLOR_RED, LIGHT_SCHEDULER_COLOR_WHITE, LIGHT_SCHEDULER_COLOR_BLUE},
  [LIGHT_SCHEDULER_COLOR_FLAG_FRANCE - LIGHT_SCHEDULER_COLOR_FLAG_FIRST] =
    {LIGHT_SCHEDULER_COLOR_BLUE, LIGHT_SCHEDULER_COLOR_WHITE, LIGHT_SCHEDULER_COLOR_RED},
};
 
static uint32_t s = 0;
static uint32_t sparkle_seed = 0x2545F491;
 
static void light_scheduler_set(uint32_t n, uint8_t r, uint8_t g, uint8_t b)
{
  uint16_t k = light_scheduler_brightness;
  light_ws2812_arch_set(n, (r * k) / 255, (g * k) / 255, (b * k) / 255);
}
 
static uint32_t sparkle_rand(void)
{
  sparkle_seed ^= sparkle_seed << 13;
  sparkle_seed ^= sparkle_seed >> 17;
  sparkle_seed ^= sparkle_seed << 5;
  return sparkle_seed;
}
 
static void light_scheduler_sparkle(void)
{
  static const uint8_t palette[] = {
    LIGHT_SCHEDULER_COLOR_YELLOW, LIGHT_SCHEDULER_COLOR_WHITE, LIGHT_SCHEDULER_COLOR_ORANGE,
    LIGHT_SCHEDULER_COLOR_RED, LIGHT_SCHEDULER_COLOR_GREEN, LIGHT_SCHEDULER_COLOR_CYAN,
    LIGHT_SCHEDULER_COLOR_BLUE, LIGHT_SCHEDULER_COLOR_MAGENTA
  };
  for (uint32_t n = 0; n < WS2812_NB_LEDS; n++) {
    uint32_t r = sparkle_rand();
    if ((r >> 8) % 5 == 0) {
      light_ws2812_arch_set(n, 0, 0, 0);
    } else {
      const uint8_t *c = light_scheduler_presets[palette[r % sizeof(palette)]];
      light_scheduler_set(n, c[0], c[1], c[2]);
    }
  }
}
 
void light_scheduler_set_color(uint8_t color)
{
  light_scheduler_color = color;
  if (color >= LIGHT_SCHEDULER_COLOR_FLAG_FIRST && color < LIGHT_SCHEDULER_COLOR_NB) {
    light_scheduler_mode = LIGHT_SCHEDULER_MODE_COLOR;
  } else if (color != LIGHT_SCHEDULER_COLOR_CUSTOM && color < LIGHT_SCHEDULER_COLOR_FLAG_FIRST) {
    light_scheduler_red = light_scheduler_presets[color][0];
    light_scheduler_green = light_scheduler_presets[color][1];
    light_scheduler_blue = light_scheduler_presets[color][2];
    light_scheduler_mode = LIGHT_SCHEDULER_MODE_COLOR;
  }
}
 
void light_scheduler_init(void)
{
  if (light_scheduler_color != LIGHT_SCHEDULER_COLOR_CUSTOM) {
    uint8_t mode = light_scheduler_mode;
    light_scheduler_set_color(light_scheduler_color);
    light_scheduler_mode = mode;
  }
  light_ws2812_arch_init();
}
 
void light_scheduler_periodic(void)
{
  uint32_t n, s0;
#ifdef LIGHT_SCHEDULER_ALERT
  if (LIGHT_SCHEDULER_ALERT) {
    if (LIGHT_SCHEDULER_ALERT_SPARKLE) {
      light_scheduler_sparkle();
      return;
    }
    const uint8_t *c = light_scheduler_presets[LIGHT_SCHEDULER_ALERT_COLOR];
    for (n = 0; n < WS2812_NB_LEDS; n++) {
      light_scheduler_set(n, c[0], c[1], c[2]);
    }
    return;
  }
#endif
 
  if (light_scheduler_mode == LIGHT_SCHEDULER_MODE_SPARKLE) {
    light_scheduler_sparkle();
    return;
  }
 
  for (n = 0; n < WS2812_NB_LEDS; n++) {
    if (light_scheduler_mode == LIGHT_SCHEDULER_MODE_COLOR &&
        light_scheduler_color >= LIGHT_SCHEDULER_COLOR_FLAG_FIRST && light_scheduler_color < LIGHT_SCHEDULER_COLOR_NB) {
      // flag: three equal bands, led n is in the band containing its center (5 leds: 2 | 1 | 2)
      uint32_t band = ((2 * n + 1) * 3) / (2 * WS2812_NB_LEDS);
      const uint8_t *c = light_scheduler_presets[light_scheduler_flags[light_scheduler_color - LIGHT_SCHEDULER_COLOR_FLAG_FIRST][band]];
      light_scheduler_set(n, c[0], c[1], c[2]);
    } else if (light_scheduler_mode == LIGHT_SCHEDULER_MODE_COLOR) {
      light_scheduler_set(n, light_scheduler_red, light_scheduler_green, light_scheduler_blue);
    } else if (light_scheduler_mode == LIGHT_SCHEDULER_MODE_OFF) {
      light_ws2812_arch_set(n, 0, 0, 0);
    } else if (WS2812_NB_LEDS == 32) {
      if (n<=7) light_scheduler_set(n, 0,255,0);  // LEFT = GREEN
      else if (n<=15)
        light_scheduler_set(n, 0,0,255);           // BACK = BLUE
      else if (n<=23)
        light_scheduler_set(n, 255,0,0);           // RIGHT = RED
      else
        light_scheduler_set(n, 255,255,255);       // FRONT = WHITE
    } else {
      s0 = s + 10 * n;
      light_scheduler_set(n, s0 % 255, (s0 + 85) % 255, (s0 + 170) % 255);
    }
  }
  s += 5;  // rainbow speed at 16 Hz
}