quantum/process_keycode/process_tap_dance.c

   1 #include "quantum.h"
   2 #include "action_tapping.h"
   3
   4 static uint16_t last_td;
   5 static int8_t highest_td = -1;
   6
   7 void qk_tap_dance_pair_finished (qk_tap_dance_state_t *state, void *user_data) {
   8   qk_tap_dance_pair_t *pair = (qk_tap_dance_pair_t *)user_data;
   9
  10   if (state->count == 1) {
  11     register_code16 (pair->kc1);
  12   } else if (state->count == 2) {
  13     register_code16 (pair->kc2);
  14   }
  15 }
  16
  17 void qk_tap_dance_pair_reset (qk_tap_dance_state_t *state, void *user_data) {
  18   qk_tap_dance_pair_t *pair = (qk_tap_dance_pair_t *)user_data;
  19
  20   if (state->count == 1) {
  21     unregister_code16 (pair->kc1);
  22   } else if (state->count == 2) {
  23     unregister_code16 (pair->kc2);
  24   }
  25 }
  26
  27 static inline void _process_tap_dance_action_fn (qk_tap_dance_state_t *state,
  28                                                  void *user_data,
  29                                                  qk_tap_dance_user_fn_t fn)
  30 {
  31   if (fn) {
  32     fn(state, user_data);
  33   }
  34 }
  35
  36 static inline void process_tap_dance_action_on_each_tap (qk_tap_dance_action_t *action)
  37 {
  38   _process_tap_dance_action_fn (&action->state, action->user_data, action->fn.on_each_tap);
  39 }
  40
  41 static inline void process_tap_dance_action_on_dance_finished (qk_tap_dance_action_t *action)
  42 {
  43   if (action->state.finished)
  44     return;
  45   action->state.finished = true;
  46   _process_tap_dance_action_fn (&action->state, action->user_data, action->fn.on_dance_finished);
  47 }
  48
  49 static inline void process_tap_dance_action_on_reset (qk_tap_dance_action_t *action)
  50 {
  51   _process_tap_dance_action_fn (&action->state, action->user_data, action->fn.on_reset);
  52 }
  53
  54 bool process_tap_dance(uint16_t keycode, keyrecord_t *record) {
  55   uint16_t idx = keycode - QK_TAP_DANCE;
  56   qk_tap_dance_action_t *action;
  57
  58   if (last_td && last_td != keycode) {
  59     (&tap_dance_actions[last_td - QK_TAP_DANCE])->state.interrupted = true;
  60   }
  61
  62   switch(keycode) {
  63   case QK_TAP_DANCE ... QK_TAP_DANCE_MAX:
  64     if ((int16_t)idx > highest_td)
  65       highest_td = idx;
  66     action = &tap_dance_actions[idx];
  67
  68     action->state.pressed = record->event.pressed;
  69     if (record->event.pressed) {
  70       action->state.keycode = keycode;
  71       action->state.count++;
  72       action->state.timer = timer_read();
  73
  74       if (last_td && last_td != keycode) {
  75         qk_tap_dance_action_t *paction = &tap_dance_actions[last_td - QK_TAP_DANCE];
  76         paction->state.interrupted = true;
  77         process_tap_dance_action_on_dance_finished (paction);
  78         reset_tap_dance (&paction->state);
  79       }
  80
  81       last_td = keycode;
  82     }
  83
  84     break;
  85
  86   default:
  87     if (!record->event.pressed)
  88       return true;
  89
  90     if (highest_td == -1)
  91       return true;
  92
  93     for (int i = 0; i <= highest_td; i++) {
  94       action = &tap_dance_actions[i];
  95       if (action->state.count == 0)
  96         continue;
  97       action->state.interrupted = true;
  98       process_tap_dance_action_on_dance_finished (action);
  99       reset_tap_dance (&action->state);
 100     }
 101     break;
 102   }
 103
 104   return true;
 105 }
 106
 107 void matrix_scan_tap_dance () {
 108   if (highest_td == -1)
 109     return;
 110
 111   for (int i = 0; i <= highest_td; i++) {
 112     qk_tap_dance_action_t *action = &tap_dance_actions[i];
 113
 114     if (action->state.count && timer_elapsed (action->state.timer) > TAPPING_TERM) {
 115       process_tap_dance_action_on_dance_finished (action);
 116       reset_tap_dance (&action->state);
 117     }
 118   }
 119 }
 120
 121 void reset_tap_dance (qk_tap_dance_state_t *state) {
 122   qk_tap_dance_action_t *action;
 123
 124   if (state->pressed)
 125     return;
 126
 127   action = &tap_dance_actions[state->keycode - QK_TAP_DANCE];
 128
 129   process_tap_dance_action_on_reset (action);
 130
 131   state->count = 0;
 132   state->interrupted = false;
 133   state->finished = false;
 134   last_td = 0;
 135 }