Wakeup LPME-IO on master key press event

Author: sekigon-gonnoc

tmk_core/protocol/nrf/lpme.c

@@ -150,16 +150,23 @@ static matrix_row_t lpme_read_row(lpme_config_t const * const config, uint8_t ro
 }
 
 extern const uint8_t MAINTASK_INTERVAL;
-uint32_t lpme_scan(lpme_config_t const * const config, matrix_row_t *rows)
-{
+uint32_t lpme_scan(lpme_config_t const *const config, matrix_row_t *rows,
+                   bool wakeup) {
     uint32_t        change                 = 0;
     static uint32_t run_mode               = 0;
     static uint32_t sleep_counter          = 0;
     static uint32_t sleep_enter_counter    = 0;
     const uint32_t  sleep_enter_threthold1 = 10000 / MAINTASK_INTERVAL;
     const uint32_t  sleep1_interval        = 60 / MAINTASK_INTERVAL;
     const uint32_t  sleep_enter_threthold2 = 30000 / MAINTASK_INTERVAL;
-    const uint32_t  sleep2_interval        = 120 / MAINTASK_INTERVAL;
+    const uint32_t  sleep2_interval        = 90 / MAINTASK_INTERVAL;
+
+    if (wakeup) {
+        run_mode            = 0;
+        sleep_enter_counter = 0;
+        sleep_counter       = 0;
+        dprint("LPME wake up\n");
+    }
 
     if (run_mode == 0) {
     } else if (run_mode == 1) {
@@ -228,4 +235,3 @@ uint32_t lpme_scan(lpme_config_t const * const config, matrix_row_t *rows)
 
     return change;
 }
-

tmk_core/protocol/nrf/lpme.h

@@ -18,4 +18,5 @@ typedef struct {
 } lpme_config_t;
 
 void     lpme_init(lpme_config_t const *const config);
-uint32_t lpme_scan(lpme_config_t const *const config, matrix_row_t *rows);
+uint32_t lpme_scan(lpme_config_t const *const config, matrix_row_t *rows,
+                   bool wakeup);

tmk_core/protocol/nrf/matrix_lpme.c

@@ -112,7 +112,7 @@ static uint32_t scan_col2row(matrix_row_t *matrix_raw)
     uint32_t change = matrix_func_col2row.scan(matrix_raw);
 
     uint8_t matrix_offset_slave = config->matrix.is_left_hand ?                                 config->matrix.device_rows : 0;
-    change |= lpme_scan(&lpme_config, &matrix_raw[matrix_offset_slave]);
+    change |= lpme_scan(&lpme_config, &matrix_raw[matrix_offset_slave], change);
 
     return change;
 }
@@ -152,7 +152,7 @@ static uint32_t scan_row2col(matrix_row_t *matrix_raw)
     uint32_t change = matrix_func_row2col.scan(matrix_raw);
 
     uint8_t matrix_offset_slave = config->matrix.is_left_hand ?                                 config->matrix.device_rows : 0;
-    change |= lpme_scan(&lpme_config, &matrix_raw[matrix_offset_slave]);
+    change |= lpme_scan(&lpme_config, &matrix_raw[matrix_offset_slave], change);
 
     return change;
 }