From 8f6bac1dca7cfd2e06beda4a8e0622fbb6078f61 Mon Sep 17 00:00:00 2001
From: MURAOKA Taro <koron.kaoriya@gmail.com>
Date: Mon, 7 Feb 2022 10:57:06 +0900
Subject: [PATCH] suppress ball a while, when scroll mode changed

---
 .../keyboards/keyball/lib/keyball/keyball.c   | 50 +++++++++++++------
 .../keyboards/keyball/lib/keyball/keyball.h   |  9 +++-
 2 files changed, 43 insertions(+), 16 deletions(-)

diff --git a/qmk_firmware/keyboards/keyball/lib/keyball/keyball.c b/qmk_firmware/keyboards/keyball/lib/keyball/keyball.c
index 72a68185f..e6bd852ec 100644
--- a/qmk_firmware/keyboards/keyball/lib/keyball/keyball.c
+++ b/qmk_firmware/keyboards/keyball/lib/keyball/keyball.c
@@ -167,11 +167,33 @@ static void motion_to_mouse(keyball_motion_t *m, report_mouse_t *r, bool is_left
     }
 }
 
+static inline bool should_add_motion(uint32_t now) {
+#if defined(KEYBALL_SCROLLBALL_INHIVITOR) && KEYBALL_SCROLLBALL_INHIVITOR > 0
+    if (TIMER_DIFF_32(now, keyball.scroll_changed) < KEYBALL_SCROLLBALL_INHIVITOR) {
+        return false;
+    }
+#endif
+    return true;
+}
+
+static inline bool should_report(uint32_t now) {
+#if defined(KEYBALL_REPORTMOUSE_INTERVAL) && KEYBALL_REPORTMOUSE_INTERVAL > 0
+    // throttling mouse report rate.
+    static uint32_t last = 0;
+    if (TIMER_DIFF_32(now, last) < KEYBALL_REPORTMOUSE_INTERVAL) {
+        return false;
+    }
+    last = now;
+#endif
+    return true;
+}
+
 report_mouse_t pointing_device_driver_get_report(report_mouse_t rep) {
+    uint32_t now = timer_read32();
     // fetch from optical sensor.
     if (keyball.this_have_ball) {
         pmw3360_motion_t d = {0};
-        if (pmw3360_motion_burst(&d)) {
+        if (pmw3360_motion_burst(&d) && should_add_motion(now)) {
             ATOMIC_BLOCK_FORCEON {
                 keyball.this_motion.x = add16(keyball.this_motion.x, d.x);
                 keyball.this_motion.y = add16(keyball.this_motion.y, d.y);
@@ -179,16 +201,7 @@ report_mouse_t pointing_device_driver_get_report(report_mouse_t rep) {
         }
     }
     // report mouse event, if keyboard is primary.
-    if (is_keyboard_master()) {
-#if defined(KEYBALL_REPORTMOUSE_INTERVAL) && KEYBALL_REPORTMOUSE_INTERVAL > 0
-        // throttling mouse report rate.
-        static uint32_t last_report = 0;
-        uint32_t        now         = timer_read32();
-        if (TIMER_DIFF_32(now, last_report) < KEYBALL_REPORTMOUSE_INTERVAL) {
-            return rep;
-        }
-        last_report = now;
-#endif
+    if (is_keyboard_master() && should_report(now)) {
         // modify mouse report by PMW3360 motion.
         if (keyball.this_have_ball) {
             motion_to_mouse(&keyball.this_motion, &rep, is_keyboard_left(), keyball.scroll_mode);
@@ -388,7 +401,16 @@ void keyball_oled_render_keyinfo(void) {
 
 bool keyball_get_scroll_mode(void) { return keyball.scroll_mode; }
 
-void keyball_set_scroll_mode(bool mode) { keyball.scroll_mode = mode; }
+void keyball_set_scroll_mode(bool mode) {
+    if (mode != keyball.scroll_mode) {
+        keyball.scroll_changed = timer_read32();
+        keyball.this_motion.x  = 0;
+        keyball.this_motion.y  = 0;
+        keyball.that_motion.x  = 0;
+        keyball.that_motion.y  = 0;
+    }
+    keyball.scroll_mode = mode;
+}
 
 uint8_t keyball_get_scroll_div(void) { return keyball.scroll_div == 0 ? KEYBALL_SCROLL_DIV_DEFAULT : keyball.scroll_div; }
 
@@ -499,11 +521,11 @@ bool process_record_kb(uint16_t keycode, keyrecord_t *record) {
 
         case SCRL_TO:
             if (record->event.pressed) {
-                keyball.scroll_mode = !keyball.scroll_mode;
+                keyball_set_scroll_mode(!keyball.scroll_mode);
             }
             break;
         case SCRL_MO:
-            keyball.scroll_mode = record->event.pressed;
+            keyball_set_scroll_mode(record->event.pressed);
             break;
         case SCRL_DVI:
             if (record->event.pressed) {
diff --git a/qmk_firmware/keyboards/keyball/lib/keyball/keyball.h b/qmk_firmware/keyboards/keyball/lib/keyball/keyball.h
index 6fc776a9e..43e5eecad 100644
--- a/qmk_firmware/keyboards/keyball/lib/keyball/keyball.h
+++ b/qmk_firmware/keyboards/keyball/lib/keyball/keyball.h
@@ -32,6 +32,10 @@ along with this program.  If not, see <http://www.gnu.org/licenses/>.
 #    define KEYBALL_REPORTMOUSE_INTERVAL 8  // mouse report rate: 125Hz
 #endif
 
+#ifndef KEYBALL_SCROLLBALL_INHIVITOR
+#    define KEYBALL_SCROLLBALL_INHIVITOR 50
+#endif
+
 //////////////////////////////////////////////////////////////////////////////
 // Constants
 
@@ -103,8 +107,9 @@ typedef struct {
     uint8_t cpi_value;
     bool    cpi_changed;
 
-    bool    scroll_mode;
-    uint8_t scroll_div;
+    bool     scroll_mode;
+    uint8_t  scroll_div;
+    uint32_t scroll_changed;
 
     uint16_t       last_kc;
     keypos_t       last_pos;