From 2141d742cad38c7c1a1ef93d32316e0facf79e00 Mon Sep 17 00:00:00 2001
From: tmk <hasu@tmk-kbd.com>
Date: Mon, 4 Apr 2022 23:53:33 +0900
Subject: [PATCH] adb_usb: Add Mouse Systems A3 support

---
 converter/adb_usb/matrix.c | 22 ++++++++++++++++++++++
 tmk_core/protocol/adb.h    |  1 +
 2 files changed, 23 insertions(+)

diff --git a/converter/adb_usb/matrix.c b/converter/adb_usb/matrix.c
index 921edb3f..95657c50 100644
--- a/converter/adb_usb/matrix.c
+++ b/converter/adb_usb/matrix.c
@@ -202,6 +202,16 @@ again:
         adb_host_listen(ADB_ADDR_MOUSE_TMP, ADB_REG_3, (reg3 >> 8), ADB_HANDLER_EXTENDED_MOUSE);
         mouse_handler = (reg3 = adb_host_talk(ADB_ADDR_MOUSE_TMP, ADB_REG_3)) & 0xFF;
 
+        if (mouse_handler != ADB_HANDLER_EXTENDED_MOUSE) {
+            adb_host_flush(ADB_ADDR_MOUSE_TMP);
+            adb_host_listen(ADB_ADDR_MOUSE_TMP, ADB_REG_3, (reg3 >> 8), ADB_HANDLER_MOUSESYSTEMS_A3);
+            mouse_handler = (reg3 = adb_host_talk(ADB_ADDR_MOUSE_TMP, ADB_REG_3)) & 0xFF;
+
+            if (mouse_handler == ADB_HANDLER_MOUSESYSTEMS_A3) {
+                adb_host_listen(ADB_ADDR_MOUSE_TMP, ADB_REG_2, 0x00, 0x07);
+            }
+        }
+
         if (mouse_handler == ADB_HANDLER_CLASSIC1_MOUSE) {
             adb_host_flush(ADB_ADDR_MOUSE_TMP);
             adb_host_listen(ADB_ADDR_MOUSE_TMP, ADB_REG_3, (reg3 >> 8), ADB_HANDLER_CLASSIC2_MOUSE);
@@ -463,6 +473,18 @@ void adb_mouse_task(void)
         buf[1] = ((buf[2] & 2) << 6) | (buf[1] & 0x7F) ;
         buf[2] = ((buf[2] & 1) << 7) | (yneg ? 0x70 : 0x00) | (xneg ? 0x0F : 0x08);
         len = 3;
+    } else if (mouse_handler == ADB_HANDLER_MOUSESYSTEMS_A3) {
+        // Mouse Systems A3: 3-button mouse/trackball:
+        //   Byte0: ??? y06 y05 y04 y03 y02 y01 y00
+        //   Byte1: ??? x06 x05 x04 x03 x02 x01 x00
+        //   Byte2: ??? ??? ??? ??? ??? bR  bM  bL
+        //     b--: button state(0:pressed, 1:released)
+        if (buf[0] & 0x40) yneg = true;
+        if (buf[1] & 0x40) xneg = true;
+        buf[0] = ((buf[2] & 1) << 7) | (buf[0] & 0x7F);
+        buf[1] = ((buf[2] & 4) << 5) | (buf[1] & 0x7F) ;
+        buf[2] = ((buf[2] & 2) << 6) | (yneg ? 0x70 : 0x00) | (xneg ? 0x0F : 0x08);
+        len = 3;
     } else {
         if (buf[len - 1] & 0x40) yneg = true;
         if (buf[len - 1] & 0x04) xneg = true;
diff --git a/tmk_core/protocol/adb.h b/tmk_core/protocol/adb.h
index 1847d4a0..58ee8925 100644
--- a/tmk_core/protocol/adb.h
+++ b/tmk_core/protocol/adb.h
@@ -100,6 +100,7 @@ POSSIBILITY OF SUCH DAMAGE.
 #define ADB_HANDLER_MICROSPEED_UNKNOWN  0x5F
 #define ADB_HANDLER_CONTOUR_MOUSE       0x66
 #define ADB_HANDLER_CHPRODUCTS_PRO      0x42
+#define ADB_HANDLER_MOUSESYSTEMS_A3     0x03
 // pseudo handler
 #define ADB_HANDLER_LOGITECH            0x4C