1 files changed, 59 insertions, 325 deletions

diff --git a/keyboards/ergodox_infinity/keymaps/dudeofawesome/visualizer.c b/keyboards/ergodox_infinity/keymaps/dudeofawesome/visualizer.c
index d339ecc970..54fc2363c7 100644
--- a/keyboards/ergodox_infinity/keymaps/dudeofawesome/visualizer.c
+++ b/keyboards/ergodox_infinity/keymaps/dudeofawesome/visualizer.c
@@ -1,5 +1,5 @@
 /*
-Copyright 2016 Fred Sundvik <fsundvik@gmail.com>
+Copyright 2017 Fred Sundvik
 
 This program is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
@@ -15,331 +15,65 @@ You should have received a copy of the GNU General Public License
 along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
-/**
- * Currently we are assuming that both the backlight and LCD are enabled
- * But it's entirely possible to write a custom visualizer that use only
- * one of them
- */
-#ifndef LCD_BACKLIGHT_ENABLE
-#error This visualizer needs that LCD backlight is enabled
-#endif
-
-#ifndef LCD_ENABLE
-#error This visualizer needs that LCD is enabled
-#endif
-
-#include "./visualizer.h"
-#include "visualizer_keyframes.h"
-#include "lcd_keyframes.h"
-#include "lcd_backlight_keyframes.h"
-#include "system/serial_link.h"
-#include "default_animations.h"
-
-static const uint32_t logo_background_color = LCD_COLOR(0, 0, 255);
-static const uint32_t initial_color = LCD_COLOR(84, 255, 255);
-
-static const uint32_t led_emulation_colors[4] = {
-  LCD_COLOR(0, 0, 255),
-  LCD_COLOR(141, 255, 255),
-  LCD_COLOR(18, 255, 255),
-  LCD_COLOR(194, 255, 255),
-};
-
-static uint32_t next_led_target_color = 0;
-
-typedef enum {
-  LCD_STATE_INITIAL,
-  LCD_STATE_LAYER_BITMAP,
-  LCD_STATE_BITMAP_AND_LEDS,
-} lcd_state_t;
-
-static lcd_state_t lcd_state = LCD_STATE_INITIAL;
-
-typedef struct {
-  uint8_t led_on;
-  uint8_t led1;
-  uint8_t led2;
-  uint8_t led3;
-} visualizer_user_data_t;
-
-/**
- * Don't access from visualization function, use the visualizer state instead
- */
-static visualizer_user_data_t user_data_keyboard = {
-  .led_on = 0,
-  .led1 = LED_BRIGHTNESS_HI,
-  .led2 = LED_BRIGHTNESS_HI,
-  .led3 = LED_BRIGHTNESS_HI,
-};
-
-_Static_assert(sizeof(visualizer_user_data_t) <= VISUALIZER_USER_DATA_SIZE,
-  "Please increase the VISUALIZER_USER_DATA_SIZE");
-
-// Feel free to modify the animations below, or even add new ones if needed
-
-// The color animation animates the LCD color when you change layers
-static keyframe_animation_t one_led_color = {
-  .num_frames = 1,
-  .loop = false,
-  .frame_lengths = {gfxMillisecondsToTicks(0)},
-  .frame_functions = {lcd_backlight_keyframe_set_color},
-};
-
-bool swap_led_target_color(keyframe_animation_t* animation, visualizer_state_t* state) {
-  uint32_t temp = next_led_target_color;
-  next_led_target_color = state->target_lcd_color;
-  state->target_lcd_color = temp;
-  return false;
-}
-
-// The color animation animates the LCD color when you change layers
-static keyframe_animation_t two_led_colors = {
-  .num_frames = 2,
-  .loop = true,
-  .frame_lengths = {gfxMillisecondsToTicks(1000), gfxMillisecondsToTicks(0)},
-  .frame_functions = {lcd_backlight_keyframe_set_color, swap_led_target_color},
-};
-
-/**
- * The LCD animation alternates between the layer name display and a
- * bitmap that displays all active layers
- */
-static keyframe_animation_t lcd_bitmap_animation = {
-  .num_frames = 1,
-  .loop = false,
-  .frame_lengths = {gfxMillisecondsToTicks(0)},
-  .frame_functions = {lcd_keyframe_display_layer_bitmap},
-};
-
-static keyframe_animation_t lcd_bitmap_leds_animation = {
-  .num_frames = 2,
-  .loop = true,
-  .frame_lengths = {gfxMillisecondsToTicks(2000), gfxMillisecondsToTicks(2000)},
-  .frame_functions = {lcd_keyframe_display_layer_bitmap, lcd_keyframe_display_led_states},
-};
-
-void initialize_user_visualizer(visualizer_state_t* state) {
-  /**
-   * The brightness will be dynamically adjustable in the future
-   * But for now, change it here.
-   */
-  lcd_backlight_brightness(180);
-  state->current_lcd_color = initial_color;
-  state->target_lcd_color = logo_background_color;
-  lcd_state = LCD_STATE_INITIAL;
-  start_keyframe_animation(&default_startup_animation);
-}
-
-static inline bool is_led_on(visualizer_user_data_t* user_data, uint8_t num) {
-  return user_data->led_on & (1u << num);
-}
-
-static uint8_t get_led_index_master(visualizer_user_data_t* user_data) {
-  for (int i=0; i < 4; i++) {
-    if (is_led_on(user_data, i)) {
-      return i + 1;
-    }
-  }
-  return 0;
-}
-
-static uint8_t get_led_index_slave(visualizer_user_data_t* user_data) {
-  uint8_t master_index = get_led_index_master(user_data);
-  if (master_index!=0) {
-    for (int i=master_index; i < 4; i++) {
-      if (is_led_on(user_data, i)) {
-        return i + 1;
-      }
-    }
-  }
-
-  return 0;
-}
-
-static uint8_t get_secondary_led_index(visualizer_user_data_t* user_data) {
-  if (
-    is_led_on(user_data, 0) &&
-    is_led_on(user_data, 1) &&
-    is_led_on(user_data, 2)
-  ) {
-    return 3;
-  }
-  return 0;
-}
-
-static uint8_t get_brightness(visualizer_user_data_t* user_data, uint8_t index) {
-  switch (index) {
-    case 1:
-      return user_data->led1;
-    case 2:
-      return user_data->led2;
-    case 3:
-      return user_data->led3;
-  }
-  return 0;
-}
-
-static void update_emulated_leds(visualizer_state_t* state, visualizer_keyboard_status_t* prev_status) {
-  visualizer_user_data_t* user_data_new = (visualizer_user_data_t*)state->status.user_data;
-  visualizer_user_data_t* user_data_old = (visualizer_user_data_t*)prev_status->user_data;
-
-  uint8_t new_index;
-  uint8_t old_index;
-
-  if (true || is_serial_link_master()) {
-    new_index = get_led_index_master(user_data_new);
-    old_index = get_led_index_master(user_data_old);
-  } else {
-    new_index = get_led_index_slave(user_data_new);
-    old_index = get_led_index_slave(user_data_old);
-  }
-
-  uint8_t new_secondary_index = get_secondary_led_index(user_data_new);
-  uint8_t old_secondary_index = get_secondary_led_index(user_data_old);
-
-  uint8_t new_brightness = get_brightness(user_data_new, new_index);
-  uint8_t old_brightness = get_brightness(user_data_old, old_index);
-
-  uint8_t new_secondary_brightness = get_brightness(user_data_new, new_secondary_index);
-  uint8_t old_secondary_brightness = get_brightness(user_data_old, old_secondary_index);
-
-  if (
-    lcd_state == LCD_STATE_INITIAL ||
-    new_index != old_index ||
-    new_secondary_index != old_secondary_index ||
-    new_brightness != old_brightness ||
-    new_secondary_brightness != old_secondary_brightness
-  ) {
-    if (new_secondary_index != 0) {
-      state->target_lcd_color = change_lcd_color_intensity(
-        led_emulation_colors[new_index], new_brightness);
-      next_led_target_color = change_lcd_color_intensity(
-        led_emulation_colors[new_secondary_index], new_secondary_brightness);
-
-      stop_keyframe_animation(&one_led_color);
-      start_keyframe_animation(&two_led_colors);
-    } else {
-      state->target_lcd_color = change_lcd_color_intensity(
-        led_emulation_colors[new_index], new_brightness);
-
-      stop_keyframe_animation(&two_led_colors);
-      start_keyframe_animation(&one_led_color);
-    }
+#include "./simple_visualizer.h"
+#include "util.h"
+#include "layers.h"
+
+// This function should be implemented by the keymap visualizer
+// Don't change anything else than state->target_lcd_color and state->layer_text as that's the only thing
+// that the simple_visualizer assumes that you are updating
+// Also make sure that the buffer passed to state->layer_text remains valid until the previous animation is
+// stopped. This can be done by either double buffering it or by using constant strings
+static void get_visualizer_layer_and_color(visualizer_state_t* state) {
+  switch(biton32(default_layer_state)) {
+    case _QWERTY:
+      state->layer_text = "QWERTY";
+      state->target_lcd_color = LCD_COLOR(0, 255, 128);
+      break;
+    case _WORKMAN:
+      state->layer_text = "Workman";
+      state->target_lcd_color = LCD_COLOR(80, 255, 128);
+      break;
+    case _DVORAK:
+      state->layer_text = "Dvorak";
+      state->target_lcd_color = LCD_COLOR(194, 255, 128);
+      break;
+    case _COLEMAK:
+      state->layer_text = "Colemak";
+      state->target_lcd_color = LCD_COLOR(18, 255, 128);
+      break;
   }
-}
-
-static void update_lcd_text(visualizer_state_t* state, visualizer_keyboard_status_t* prev_status) {
-  if (state->status.leds) {
-    if (
-      lcd_state != LCD_STATE_BITMAP_AND_LEDS ||
-      state->status.leds != prev_status->leds ||
-      state->status.layer != prev_status->layer ||
-      state->status.default_layer != prev_status->default_layer
-    ) {
-      // NOTE: that it doesn't matter if the animation isn't playing, stop will do nothing in that case
-      stop_keyframe_animation(&lcd_bitmap_animation);
 
-      lcd_state = LCD_STATE_BITMAP_AND_LEDS;
-      /**
-       * For information:
-       * The logic in this function makes sure that this doesn't happen, but if you call start on an
-       * animation that is already playing it will be restarted.
-       */
-      start_keyframe_animation(&lcd_bitmap_leds_animation);
-    }
-  } else {
-    if (
-      lcd_state != LCD_STATE_LAYER_BITMAP ||
-      state->status.layer != prev_status->layer ||
-      state->status.default_layer != prev_status->default_layer
-    ) {
-      stop_keyframe_animation(&lcd_bitmap_leds_animation);
-
-      lcd_state = LCD_STATE_LAYER_BITMAP;
-      start_keyframe_animation(&lcd_bitmap_animation);
-    }
+  switch(biton32(state->status.layer)) {
+    case _LOWER:
+      state->layer_text = "Lower";
+      state->target_lcd_color = LCD_COLOR(141, 255, 255);
+      break;
+    case _RAISE:
+      state->layer_text = "Raise";
+      state->target_lcd_color = LCD_COLOR(18, 255, 255);
+      break;
+    case _ADJUST:
+      state->layer_text = "Adjust";
+      state->target_lcd_color = LCD_COLOR(194, 255, 255);
+      break;
+    case _NUM:
+      state->layer_text = "Numpad";
+      state->target_lcd_color = LCD_COLOR(80, 255, 255);
+      break;
+    case _MOUSE:
+      state->layer_text = "Mouse";
+      state->target_lcd_color = LCD_COLOR(300, 255, 255);
+      break;
+    case _GAME:
+      state->layer_text = "Game";
+      state->target_lcd_color = LCD_COLOR(300, 255, 255);
+      break;
+    case _QWERTY: case _WORKMAN: case _DVORAK: case _COLEMAK:
+      break;
+    default:
+      state->layer_text = "NONE";
+      state->target_lcd_color = LCD_COLOR(0, 255, 255);
+      break;
   }
 }
-
-void update_user_visualizer_state(visualizer_state_t* state, visualizer_keyboard_status_t* prev_status) {
-  /**
-   * Check the status here to start and stop animations
-   * You might have to save some state, like the current animation here so that you can start the right
-   * This function is called every time the status changes
-   *
-   * NOTE that this is called from the visualizer thread, so don't access anything else outside the status
-   * This is also important because the slave won't have access to the active layer for example outside the
-   * status.
-   */
-
-  update_emulated_leds(state, prev_status);
-  update_lcd_text(state, prev_status);
-}
-
-void user_visualizer_suspend(visualizer_state_t* state) {
-  state->layer_text = "Suspending...";
-  uint8_t hue = LCD_HUE(state->current_lcd_color);
-  uint8_t sat = LCD_SAT(state->current_lcd_color);
-  state->target_lcd_color = LCD_COLOR(hue, sat, 0);
-  start_keyframe_animation(&default_suspend_animation);
-}
-
-void user_visualizer_resume(visualizer_state_t* state) {
-  state->current_lcd_color = initial_color;
-  state->target_lcd_color = logo_background_color;
-  lcd_state = LCD_STATE_INITIAL;
-  start_keyframe_animation(&default_startup_animation);
-}
-
-void ergodox_board_led_on(void){
-  // No board led support
-}
-
-void ergodox_led_lower_on(void){
-  user_data_keyboard.led_on |= (1u << 0);
-  visualizer_set_user_data(&user_data_keyboard);
-}
-
-void ergodox_led_raise_on(void){
-  user_data_keyboard.led_on |= (1u << 1);
-  visualizer_set_user_data(&user_data_keyboard);
-}
-
-void ergodox_led_adjust_on(void){
-  user_data_keyboard.led_on |= (1u << 2);
-  visualizer_set_user_data(&user_data_keyboard);
-}
-
-void ergodox_board_led_off(void){
-  // No board led support
-}
-
-void ergodox_led_lower_off(void){
-  user_data_keyboard.led_on &= ~(1u << 0);
-  visualizer_set_user_data(&user_data_keyboard);
-}
-
-void ergodox_led_raise_off(void){
-  user_data_keyboard.led_on &= ~(1u << 1);
-  visualizer_set_user_data(&user_data_keyboard);
-}
-
-void ergodox_led_adjust_off(void){
-  user_data_keyboard.led_on &= ~(1u << 2);
-  visualizer_set_user_data(&user_data_keyboard);
-}
-
-void ergodox_led_lower_set(uint8_t n) {
-  user_data_keyboard.led1 = n;
-  visualizer_set_user_data(&user_data_keyboard);
-}
-
-void ergodox_led_raise_set(uint8_t n) {
-  user_data_keyboard.led2 = n;
-  visualizer_set_user_data(&user_data_keyboard);
-}
-
-void ergodox_led_adjust_set(uint8_t n) {
-  user_data_keyboard.led3 = n;
-  visualizer_set_user_data(&user_data_keyboard);
-}