keyboards/ergodox/infinity/visualizer.c

   1 /*
   2 Copyright 2016 Fred Sundvik <fsundvik@gmail.com>
   3
   4 This program is free software: you can redistribute it and/or modify
   5 it under the terms of the GNU General Public License as published by
   6 the Free Software Foundation, either version 2 of the License, or
   7 (at your option) any later version.
   8
   9 This program is distributed in the hope that it will be useful,
  10 but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 GNU General Public License for more details.
  13
  14 You should have received a copy of the GNU General Public License
  15 along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16 */
  17
  18 // Currently we are assuming that both the backlight and LCD are enabled
  19 // But it's entirely possible to write a custom visualizer that use only
  20 // one of them
  21 #ifndef LCD_BACKLIGHT_ENABLE
  22 #error This visualizer needs that LCD backlight is enabled
  23 #endif
  24
  25 #ifndef LCD_ENABLE
  26 #error This visualizer needs that LCD is enabled
  27 #endif
  28
  29 #include "visualizer.h"
  30 #include "lcd_keyframes.h"
  31 #include "lcd_backlight_keyframes.h"
  32 #include "system/serial_link.h"
  33
  34 // To generate an image array like this
  35 // Ensure the image is 128 x 32 or smaller
  36 // Convert the bitmap to a C array using a program like http://www.riuson.com/lcd-image-converter/
  37 // Ensure the the conversion process produces a monochrome format array - 1 bit/pixel, left to right, top to bottom
  38 // Update array in the source code with the C array produced by the conversion program
  39
  40 // The image below is generated from lcd_logo.png
  41 static const uint8_t image_data_lcd_logo[512] = {
  42     0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  43     0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  44     0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  45     0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  46     0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  47     0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  48     0x00, 0xf8, 0xfe, 0x3e, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  49     0x00, 0x38, 0x38, 0x38, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  50     0x00, 0x38, 0x38, 0x38, 0x06, 0x29, 0x41, 0x24, 0x52, 0x24, 0x46, 0x00, 0x00, 0x00, 0x00, 0x00,
  51     0x00, 0x38, 0x38, 0x38, 0x09, 0x55, 0x42, 0xaa, 0xaa, 0xaa, 0xa8, 0x00, 0x00, 0x00, 0x00, 0x00,
  52     0x00, 0x38, 0x38, 0x38, 0x09, 0x55, 0x82, 0x28, 0xaa, 0xae, 0x8c, 0x00, 0x00, 0x00, 0x00, 0x00,
  53     0x00, 0x38, 0x38, 0x38, 0x09, 0x55, 0x43, 0x28, 0xaa, 0xaa, 0x88, 0x00, 0x00, 0x00, 0x00, 0x00,
  54     0x00, 0x38, 0x38, 0x38, 0x0a, 0x55, 0x42, 0x28, 0xaa, 0xaa, 0x88, 0x00, 0x00, 0x00, 0x00, 0x00,
  55     0x00, 0x38, 0x38, 0x38, 0x05, 0x45, 0x42, 0x28, 0x89, 0x4a, 0x86, 0x00, 0x00, 0x00, 0x00, 0x00,
  56     0x00, 0x18, 0x38, 0x30, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  57     0x00, 0x1c, 0x38, 0x70, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  58     0x00, 0x0e, 0x38, 0xe0, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  59     0x00, 0x03, 0xff, 0x80, 0x04, 0x45, 0x14, 0xa4, 0x92, 0x83, 0x52, 0x22, 0x22, 0x36, 0x00, 0x00,
  60     0x00, 0x00, 0x38, 0x00, 0x0a, 0xaa, 0xaa, 0xaa, 0xba, 0x84, 0x55, 0x55, 0x57, 0x45, 0x00, 0x00,
  61     0x00, 0x00, 0x38, 0x00, 0x08, 0xaa, 0xaa, 0xaa, 0x92, 0xb2, 0x55, 0x55, 0x42, 0x65, 0x00, 0x00,
  62     0x00, 0x00, 0x38, 0x00, 0x08, 0xaa, 0xaa, 0xaa, 0x92, 0x81, 0x56, 0x65, 0x42, 0x45, 0x00, 0x00,
  63     0x00, 0x00, 0x38, 0x00, 0x0a, 0xaa, 0xaa, 0xaa, 0x92, 0x81, 0x54, 0x45, 0x42, 0x45, 0x00, 0x00,
  64     0x00, 0x00, 0x38, 0x00, 0x04, 0x48, 0xa2, 0x4a, 0x89, 0x06, 0x24, 0x42, 0x41, 0x36, 0x00, 0x00,
  65     0x00, 0x00, 0x38, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  66     0x00, 0x00, 0xfe, 0x00, 0x00, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  67     0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  68     0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  69     0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  70     0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  71     0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  72     0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  73     0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
  74 };
  75
  76 static const uint32_t logo_background_color = LCD_COLOR(0x00, 0x00, 0xFF);
  77 static const uint32_t initial_color = LCD_COLOR(0, 0, 0);
  78
  79 static const uint32_t led_emulation_colors[4] = {
  80     LCD_COLOR(0, 0, 0),
  81     LCD_COLOR(255, 255, 255),
  82     LCD_COLOR(84, 255, 255),
  83     LCD_COLOR(168, 255, 255),
  84 };
  85
  86 static uint32_t next_led_target_color = 0;
  87
  88 typedef enum {
  89     LCD_STATE_INITIAL,
  90     LCD_STATE_LAYER_BITMAP,
  91     LCD_STATE_BITMAP_AND_LEDS,
  92 } lcd_state_t;
  93
  94 static lcd_state_t lcd_state = LCD_STATE_INITIAL;
  95
  96 typedef struct {
  97     uint8_t led_on;
  98     uint8_t led1;
  99     uint8_t led2;
 100     uint8_t led3;
 101 } visualizer_user_data_t;
 102
 103 // Don't access from visualization function, use the visualizer state instead
 104 static visualizer_user_data_t user_data_keyboard = {
 105     .led_on = 0,
 106     .led1 = LED_BRIGHTNESS_HI,
 107     .led2 = LED_BRIGHTNESS_HI,
 108     .led3 = LED_BRIGHTNESS_HI,
 109 };
 110
 111 _Static_assert(sizeof(visualizer_user_data_t) <= VISUALIZER_USER_DATA_SIZE,
 112     "Please increase the VISUALIZER_USER_DATA_SIZE");
 113
 114 bool display_logo(keyframe_animation_t* animation, visualizer_state_t* state) {
 115     (void)state;
 116     (void)animation;
 117     (void)state;
 118     // Read the uGFX documentation for information how to use the displays
 119     // http://wiki.ugfx.org/index.php/Main_Page
 120     gdispClear(White);
 121
 122     // You can use static variables for things that can't be found in the animation
 123     // or state structs, here we use the image
 124
 125     //gdispGBlitArea is a tricky function to use since it supports blitting part of the image
 126     // if you have full screen image, then just use 128 and 32 for both source and target dimensions
 127     gdispGBlitArea(GDISP, 0, 0, 128, 32, 0, 0, 128, (pixel_t*)image_data_lcd_logo);
 128
 129     // Always remember to flush the display
 130     gdispFlush();
 131     return false;
 132 }
 133
 134 // Feel free to modify the animations below, or even add new ones if needed
 135
 136 // Don't worry, if the startup animation is long, you can use the keyboard like normal
 137 // during that time
 138 static keyframe_animation_t startup_animation = {
 139     .num_frames = 3,
 140     .loop = false,
 141     .frame_lengths = {0, gfxMillisecondsToTicks(10000), 0},
 142     .frame_functions = {
 143             display_logo,
 144             backlight_keyframe_animate_color,
 145             enable_visualization
 146     },
 147 };
 148
 149 // The color animation animates the LCD color when you change layers
 150 static keyframe_animation_t one_led_color = {
 151     .num_frames = 1,
 152     .loop = false,
 153     .frame_lengths = {gfxMillisecondsToTicks(0)},
 154     .frame_functions = {backlight_keyframe_set_color},
 155 };
 156
 157 bool swap_led_target_color(keyframe_animation_t* animation, visualizer_state_t* state) {
 158     uint32_t temp = next_led_target_color;
 159     next_led_target_color = state->target_lcd_color;
 160     state->target_lcd_color = temp;
 161     return false;
 162 }
 163
 164 // The color animation animates the LCD color when you change layers
 165 static keyframe_animation_t two_led_colors = {
 166     .num_frames = 2,
 167     .loop = true,
 168     .frame_lengths = {gfxMillisecondsToTicks(1000), gfxMillisecondsToTicks(0)},
 169     .frame_functions = {backlight_keyframe_set_color, swap_led_target_color},
 170 };
 171
 172 // The LCD animation alternates between the layer name display and a
 173 // bitmap that displays all active layers
 174 static keyframe_animation_t lcd_bitmap_animation = {
 175     .num_frames = 1,
 176     .loop = false,
 177     .frame_lengths = {gfxMillisecondsToTicks(0)},
 178     .frame_functions = {lcd_keyframe_display_layer_bitmap},
 179 };
 180
 181 static keyframe_animation_t lcd_bitmap_leds_animation = {
 182     .num_frames = 2,
 183     .loop = true,
 184     .frame_lengths = {gfxMillisecondsToTicks(2000), gfxMillisecondsToTicks(2000)},
 185     .frame_functions = {lcd_keyframe_display_layer_bitmap, lcd_keyframe_display_led_states},
 186 };
 187
 188 static keyframe_animation_t suspend_animation = {
 189     .num_frames = 4,
 190     .loop = false,
 191     .frame_lengths = {0, gfxMillisecondsToTicks(1000), 0, 0},
 192     .frame_functions = {
 193             lcd_keyframe_display_layer_text,
 194             backlight_keyframe_animate_color,
 195             lcd_keyframe_disable,
 196             lcd_keyframe_disable,
 197     },
 198 };
 199
 200 static keyframe_animation_t resume_animation = {
 201     .num_frames = 5,
 202     .loop = false,
 203     .frame_lengths = {0, 0, 0, gfxMillisecondsToTicks(10000), 0},
 204     .frame_functions = {
 205             lcd_keyframe_enable,
 206             backlight_keyframe_enable,
 207             display_logo,
 208             backlight_keyframe_animate_color,
 209             enable_visualization,
 210     },
 211 };
 212
 213 void initialize_user_visualizer(visualizer_state_t* state) {
 214     // The brightness will be dynamically adjustable in the future
 215     // But for now, change it here.
 216     lcd_backlight_brightness(130);
 217     state->current_lcd_color = initial_color;
 218     state->target_lcd_color = logo_background_color;
 219     lcd_state = LCD_STATE_INITIAL;
 220     start_keyframe_animation(&startup_animation);
 221 }
 222
 223 static const uint32_t red;
 224 static const uint32_t green;
 225 static const uint32_t blue;
 226
 227 inline bool is_led_on(visualizer_user_data_t* user_data, uint8_t num) {
 228     return user_data->led_on & (1u << num);
 229 }
 230
 231 static uint8_t get_led_index_master(visualizer_user_data_t* user_data) {
 232     for (int i=0; i < 3; i++) {
 233         if (is_led_on(user_data, i)) {
 234             return i + 1;
 235         }
 236     }
 237     return 0;
 238 }
 239
 240 static uint8_t get_led_index_slave(visualizer_user_data_t* user_data) {
 241     uint8_t master_index = get_led_index_master(user_data);
 242     if (master_index!=0) {
 243         for (int i=master_index; i < 3; i++) {
 244             if (is_led_on(user_data, i)) {
 245                 return i + 1;
 246             }
 247         }
 248     }
 249
 250     return 0;
 251 }
 252
 253 static uint8_t get_secondary_led_index(visualizer_user_data_t* user_data) {
 254     if (is_led_on(user_data, 0) &&
 255             is_led_on(user_data, 1) &&
 256             is_led_on(user_data, 2)) {
 257         return 3;
 258     }
 259     return 0;
 260 }
 261
 262 static uint8_t get_brightness(visualizer_user_data_t* user_data, uint8_t index) {
 263     switch (index) {
 264     case 1:
 265         return user_data->led1;
 266     case 2:
 267         return user_data->led2;
 268     case 3:
 269         return user_data->led3;
 270     }
 271     return 0;
 272 }
 273
 274 static void update_emulated_leds(visualizer_state_t* state, visualizer_keyboard_status_t* prev_status) {
 275     visualizer_user_data_t* user_data_new = (visualizer_user_data_t*)state->status.user_data;
 276     visualizer_user_data_t* user_data_old = (visualizer_user_data_t*)prev_status->user_data;
 277
 278     uint8_t new_index;
 279     uint8_t old_index;
 280
 281     if (is_serial_link_master()) {
 282         new_index = get_led_index_master(user_data_new);
 283         old_index = get_led_index_master(user_data_old);
 284     }
 285     else {
 286         new_index = get_led_index_slave(user_data_new);
 287         old_index = get_led_index_slave(user_data_old);
 288     }
 289     uint8_t new_secondary_index = get_secondary_led_index(user_data_new);
 290     uint8_t old_secondary_index = get_secondary_led_index(user_data_old);
 291
 292     uint8_t old_brightness = get_brightness(user_data_old, old_index);
 293     uint8_t new_brightness = get_brightness(user_data_new, new_index);
 294
 295     uint8_t old_secondary_brightness = get_brightness(user_data_old, old_secondary_index);
 296     uint8_t new_secondary_brightness = get_brightness(user_data_new, new_secondary_index);
 297
 298     if (lcd_state == LCD_STATE_INITIAL ||
 299             new_index != old_index ||
 300             new_secondary_index != old_secondary_index ||
 301             new_brightness != old_brightness ||
 302             new_secondary_brightness != old_secondary_brightness) {
 303
 304         if (new_secondary_index != 0) {
 305             state->target_lcd_color = change_lcd_color_intensity(
 306                 led_emulation_colors[new_index], new_brightness);
 307             next_led_target_color = change_lcd_color_intensity(
 308                 led_emulation_colors[new_secondary_index], new_secondary_brightness);
 309
 310             stop_keyframe_animation(&one_led_color);
 311             start_keyframe_animation(&two_led_colors);
 312         } else {
 313             state->target_lcd_color = change_lcd_color_intensity(
 314                 led_emulation_colors[new_index], new_brightness);
 315             stop_keyframe_animation(&two_led_colors);
 316             start_keyframe_animation(&one_led_color);
 317         }
 318     }
 319 }
 320
 321 static void update_lcd_text(visualizer_state_t* state, visualizer_keyboard_status_t* prev_status) {
 322     if (state->status.leds) {
 323         if (lcd_state != LCD_STATE_BITMAP_AND_LEDS ||
 324                 state->status.leds != prev_status->leds ||
 325                 state->status.layer != prev_status->layer ||
 326                 state->status.default_layer != prev_status->default_layer) {
 327
 328             // NOTE: that it doesn't matter if the animation isn't playing, stop will do nothing in that case
 329             stop_keyframe_animation(&lcd_bitmap_animation);
 330
 331             lcd_state = LCD_STATE_BITMAP_AND_LEDS;
 332             // For information:
 333             // The logic in this function makes sure that this doesn't happen, but if you call start on an
 334             // animation that is already playing it will be restarted.
 335             start_keyframe_animation(&lcd_bitmap_leds_animation);
 336         }
 337     } else {
 338         if (lcd_state != LCD_STATE_LAYER_BITMAP ||
 339                 state->status.layer != prev_status->layer ||
 340                 state->status.default_layer != prev_status->default_layer) {
 341
 342             stop_keyframe_animation(&lcd_bitmap_leds_animation);
 343
 344             lcd_state = LCD_STATE_LAYER_BITMAP;
 345             start_keyframe_animation(&lcd_bitmap_animation);
 346         }
 347     }
 348 }
 349
 350 void update_user_visualizer_state(visualizer_state_t* state, visualizer_keyboard_status_t* prev_status) {
 351     // Check the status here to start and stop animations
 352     // You might have to save some state, like the current animation here so that you can start the right
 353     // This function is called every time the status changes
 354
 355     // NOTE that this is called from the visualizer thread, so don't access anything else outside the status
 356     // This is also important because the slave won't have access to the active layer for example outside the
 357     // status.
 358
 359     update_emulated_leds(state, prev_status);
 360     update_lcd_text(state, prev_status);
 361
 362 }
 363
 364 void user_visualizer_suspend(visualizer_state_t* state) {
 365     state->layer_text = "Suspending...";
 366     uint8_t hue = LCD_HUE(state->current_lcd_color);
 367     uint8_t sat = LCD_SAT(state->current_lcd_color);
 368     state->target_lcd_color = LCD_COLOR(hue, sat, 0);
 369     start_keyframe_animation(&suspend_animation);
 370 }
 371
 372 void user_visualizer_resume(visualizer_state_t* state) {
 373     state->current_lcd_color = initial_color;
 374     state->target_lcd_color = logo_background_color;
 375     lcd_state = LCD_STATE_INITIAL;
 376     start_keyframe_animation(&resume_animation);
 377 }
 378
 379 void ergodox_board_led_on(void){
 380     // No board led support
 381 }
 382
 383 void ergodox_right_led_1_on(void){
 384     user_data_keyboard.led_on |= (1u << 0);
 385     visualizer_set_user_data(&user_data_keyboard);
 386 }
 387
 388 void ergodox_right_led_2_on(void){
 389     user_data_keyboard.led_on |= (1u << 1);
 390     visualizer_set_user_data(&user_data_keyboard);
 391 }
 392
 393 void ergodox_right_led_3_on(void){
 394     user_data_keyboard.led_on |= (1u << 2);
 395     visualizer_set_user_data(&user_data_keyboard);
 396 }
 397
 398 void ergodox_board_led_off(void){
 399     // No board led support
 400 }
 401
 402 void ergodox_right_led_1_off(void){
 403     user_data_keyboard.led_on &= ~(1u << 0);
 404     visualizer_set_user_data(&user_data_keyboard);
 405 }
 406
 407 void ergodox_right_led_2_off(void){
 408     user_data_keyboard.led_on &= ~(1u << 1);
 409     visualizer_set_user_data(&user_data_keyboard);
 410 }
 411
 412 void ergodox_right_led_3_off(void){
 413     user_data_keyboard.led_on &= ~(1u << 2);
 414     visualizer_set_user_data(&user_data_keyboard);
 415 }
 416
 417 void ergodox_right_led_1_set(uint8_t n) {
 418     user_data_keyboard.led1 = n;
 419     visualizer_set_user_data(&user_data_keyboard);
 420 }
 421
 422 void ergodox_right_led_2_set(uint8_t n) {
 423     user_data_keyboard.led2 = n;
 424     visualizer_set_user_data(&user_data_keyboard);
 425 }
 426
 427 void ergodox_right_led_3_set(uint8_t n) {
 428     user_data_keyboard.led3 = n;
 429     visualizer_set_user_data(&user_data_keyboard);
 430 }