tmk_core/common/action_layer.c

   1 #include <stdint.h>
   2 #include "keyboard.h"
   3 #include "action.h"
   4 #include "util.h"
   5 #include "action_layer.h"
   6
   7 #ifdef DEBUG_ACTION
   8 #include "debug.h"
   9 #else
  10 #include "nodebug.h"
  11 #endif
  12
  13
  14 /*
  15  * Default Layer State
  16  */
  17 uint32_t default_layer_state = 0;
  18
  19 __attribute__((weak))
  20 uint32_t default_layer_state_set_kb(uint32_t state) {
  21     return state;
  22 }
  23
  24 static void default_layer_state_set(uint32_t state)
  25 {
  26     state = default_layer_state_set_kb(state);
  27     debug("default_layer_state: ");
  28     default_layer_debug(); debug(" to ");
  29     default_layer_state = state;
  30     default_layer_debug(); debug("\n");
  31     clear_keyboard_but_mods(); // To avoid stuck keys
  32 }
  33
  34 void default_layer_debug(void)
  35 {
  36     dprintf("%08lX(%u)", default_layer_state, biton32(default_layer_state));
  37 }
  38
  39 void default_layer_set(uint32_t state)
  40 {
  41     default_layer_state_set(state);
  42 }
  43
  44 #ifndef NO_ACTION_LAYER
  45 void default_layer_or(uint32_t state)
  46 {
  47     default_layer_state_set(default_layer_state | state);
  48 }
  49 void default_layer_and(uint32_t state)
  50 {
  51     default_layer_state_set(default_layer_state & state);
  52 }
  53 void default_layer_xor(uint32_t state)
  54 {
  55     default_layer_state_set(default_layer_state ^ state);
  56 }
  57 #endif
  58
  59
  60 #ifndef NO_ACTION_LAYER
  61 /*
  62  * Keymap Layer State
  63  */
  64 uint32_t layer_state = 0;
  65
  66 __attribute__((weak))
  67 uint32_t layer_state_set_kb(uint32_t state) {
  68     return state;
  69 }
  70
  71 static void layer_state_set(uint32_t state)
  72 {
  73     state = layer_state_set_kb(state);
  74     dprint("layer_state: ");
  75     layer_debug(); dprint(" to ");
  76     layer_state = state;
  77     layer_debug(); dprintln();
  78     clear_keyboard_but_mods(); // To avoid stuck keys
  79 }
  80
  81 void layer_clear(void)
  82 {
  83     layer_state_set(0);
  84 }
  85
  86 void layer_move(uint8_t layer)
  87 {
  88     layer_state_set(1UL<<layer);
  89 }
  90
  91 void layer_on(uint8_t layer)
  92 {
  93     layer_state_set(layer_state | (1UL<<layer));
  94 }
  95
  96 void layer_off(uint8_t layer)
  97 {
  98     layer_state_set(layer_state & ~(1UL<<layer));
  99 }
 100
 101 void layer_invert(uint8_t layer)
 102 {
 103     layer_state_set(layer_state ^ (1UL<<layer));
 104 }
 105
 106 void layer_or(uint32_t state)
 107 {
 108     layer_state_set(layer_state | state);
 109 }
 110 void layer_and(uint32_t state)
 111 {
 112     layer_state_set(layer_state & state);
 113 }
 114 void layer_xor(uint32_t state)
 115 {
 116     layer_state_set(layer_state ^ state);
 117 }
 118
 119 void layer_debug(void)
 120 {
 121     dprintf("%08lX(%u)", layer_state, biton32(layer_state));
 122 }
 123 #endif
 124
 125 #if !defined(NO_ACTION_LAYER) && defined(PREVENT_STUCK_MODIFIERS)
 126 uint8_t source_layers_cache[(MATRIX_ROWS * MATRIX_COLS + 7) / 8][MAX_LAYER_BITS] = {{0}};
 127
 128 void update_source_layers_cache(keypos_t key, uint8_t layer)
 129 {
 130     const uint8_t key_number = key.col + (key.row * MATRIX_COLS);
 131     const uint8_t storage_row = key_number / 8;
 132     const uint8_t storage_bit = key_number % 8;
 133
 134     for (uint8_t bit_number = 0; bit_number < MAX_LAYER_BITS; bit_number++) {
 135         source_layers_cache[storage_row][bit_number] ^=
 136             (-((layer & (1U << bit_number)) != 0)
 137              ^ source_layers_cache[storage_row][bit_number])
 138             & (1U << storage_bit);
 139     }
 140 }
 141
 142 uint8_t read_source_layers_cache(keypos_t key)
 143 {
 144     const uint8_t key_number = key.col + (key.row * MATRIX_COLS);
 145     const uint8_t storage_row = key_number / 8;
 146     const uint8_t storage_bit = key_number % 8;
 147     uint8_t layer = 0;
 148
 149     for (uint8_t bit_number = 0; bit_number < MAX_LAYER_BITS; bit_number++) {
 150         layer |=
 151             ((source_layers_cache[storage_row][bit_number]
 152               & (1U << storage_bit)) != 0)
 153             << bit_number;
 154     }
 155
 156     return layer;
 157 }
 158 #endif
 159
 160 /*
 161  * Make sure the action triggered when the key is released is the same
 162  * one as the one triggered on press. It's important for the mod keys
 163  * when the layer is switched after the down event but before the up
 164  * event as they may get stuck otherwise.
 165  */
 166 action_t store_or_get_action(bool pressed, keypos_t key)
 167 {
 168 #if !defined(NO_ACTION_LAYER) && defined(PREVENT_STUCK_MODIFIERS)
 169     if (disable_action_cache) {
 170         return layer_switch_get_action(key);
 171     }
 172
 173     uint8_t layer;
 174
 175     if (pressed) {
 176         layer = layer_switch_get_layer(key);
 177         update_source_layers_cache(key, layer);
 178     }
 179     else {
 180         layer = read_source_layers_cache(key);
 181     }
 182     return action_for_key(layer, key);
 183 #else
 184     return layer_switch_get_action(key);
 185 #endif
 186 }
 187
 188
 189 int8_t layer_switch_get_layer(keypos_t key)
 190 {
 191     action_t action;
 192     action.code = ACTION_TRANSPARENT;
 193
 194 #ifndef NO_ACTION_LAYER
 195     uint32_t layers = layer_state | default_layer_state;
 196     /* check top layer first */
 197     for (int8_t i = 31; i >= 0; i--) {
 198         if (layers & (1UL<<i)) {
 199             action = action_for_key(i, key);
 200             if (action.code != ACTION_TRANSPARENT) {
 201                 return i;
 202             }
 203         }
 204     }
 205     /* fall back to layer 0 */
 206     return 0;
 207 #else
 208     return biton32(default_layer_state);
 209 #endif
 210 }
 211
 212 action_t layer_switch_get_action(keypos_t key)
 213 {
 214     return action_for_key(layer_switch_get_layer(key), key);
 215 }