+ led_set(host_keyboard_leds());
+
+ device_scan();
+
+ // LED off
+ DDRD |= (1<<6); PORTD &= ~(1<<6);
+ return;
+}
+
+#ifdef ADB_MOUSE_ENABLE
+static void mouse_init(uint8_t orig_addr)
+{
+ uint16_t reg3;
+ uint8_t mouse_handler;
+ uint8_t addr;
+
+again:
+ // Move to tmp address 15 to setup mouse function
+ mouse_handler = (reg3 = adb_host_talk(orig_addr, ADB_REG_3)) & 0xFF;
+ if (!reg3) return;
+ dmprintf("addr%d reg3: %04X\n", orig_addr, reg3);
+
+ // Move device to tmp address
+ adb_host_flush(orig_addr);
+ adb_host_listen(orig_addr, ADB_REG_3, ((reg3 >> 8) & 0xF0) | ADB_ADDR_TMP, 0xFE);
+ adb_host_flush(ADB_ADDR_TMP);
+
+ mouse_handler = (reg3 = adb_host_talk(ADB_ADDR_TMP, ADB_REG_3)) & 0xFF;
+ if (!reg3) {
+ dmprintf("move fail\n");
+ goto again;
+ }
+ addr = ADB_ADDR_TMP;
+
+
+detect_again:
+ // Try to escalate into extended/classic2 protocol
+ if (mouse_handler == ADB_HANDLER_CLASSIC1_MOUSE || mouse_handler == ADB_HANDLER_CLASSIC2_MOUSE) {
+ adb_host_flush(addr);
+ adb_host_listen(addr, ADB_REG_3, (reg3 >> 8), ADB_HANDLER_EXTENDED_MOUSE);
+
+ mouse_handler = (reg3 = adb_host_talk(addr, ADB_REG_3)) & 0xFF;
+
+
+ if (mouse_handler == ADB_HANDLER_CLASSIC1_MOUSE) {
+ adb_host_flush(addr);
+ adb_host_listen(addr, ADB_REG_3, (reg3 >> 8), ADB_HANDLER_CLASSIC2_MOUSE);
+
+ mouse_handler = (reg3 = adb_host_talk(addr, ADB_REG_3)) & 0xFF;
+ }
+ dmprintf("addr%d reg3: %04X\n", addr, reg3);
+
+ }
+
+ // Classic Protocol 100cpi
+ if (mouse_handler == ADB_HANDLER_CLASSIC1_MOUSE) {
+ xprintf("Classic 100cpi\n");
+ mouse_cpi = 100;
+ }
+
+ // Classic Protocol 200cpi
+ if (mouse_handler == ADB_HANDLER_CLASSIC2_MOUSE) {
+ xprintf("Classic 200cpi\n");
+ mouse_cpi = 200;
+ }
+
+ // Extended Mouse Protocol
+ if (mouse_handler == ADB_HANDLER_EXTENDED_MOUSE) {
+ // Device info format(reg1 8-byte data)
+ // 0-3: device id
+ // 4-5: resolution in units/inch (0xC8=200upi)
+ // 6 : device class (0: Tablet, 1: Mouse, 2: Trackball)
+ // 7 : num of buttons
+ uint8_t len;
+ uint8_t buf[8];
+ len = adb_host_talk_buf(addr, ADB_REG_1, buf, sizeof(buf));
+
+ if (len > 5) {
+ mouse_cpi = (buf[4]<<8) | buf[5];
+ } else {
+ mouse_cpi = 100;
+ }
+
+ if (len) {
+ xprintf("Ext: [", len);
+ for (int8_t i = 0; i < len; i++) xprintf("%02X ", buf[i]);
+ xprintf("] cpi=%d\n", mouse_cpi);
+ }
+
+
+ // Kensington Turbo Mouse 5: default device
+ if (buf[0] == 0x4B && buf[1] == 0x4D && buf[2] == 0x4C && buf[3] == 0x31) {
+ xprintf("TM5: default\n");
+ // Move it to addr0 to remove this device and get new device with handle id 50 on addr 3
+ // and the new device on address 3 should be handled with command sequence later.
+ //
+ // Turbo Mouse 5 has one default device on addr3 as normal mouse at first, and another device
+ // with hander id 50 appears after the default device is moved from addr3.
+ // The mouse has the two devices at same time transiently in the result. The default device is
+ // removed automatically after the another device receives command sequence.
+ // NOTE: The mouse hangs if you try moving the two deivces to same address.
+ adb_host_flush(addr);
+ adb_host_listen(addr, ADB_REG_3, ((reg3 >> 8) & 0xF0) | ADB_ADDR_0, 0xFE);
+ } else {
+ xprintf("Unknown\n");
+ }
+ }
+
+ // Kensington Turbo Mouse 5: setup
+ if (mouse_handler == ADB_HANDLER_TURBO_MOUSE) {
+ xprintf("TM5: ext\n");
+
+ // Kensington Turbo Mouse 5 command sequence to enable four buttons
+ // https://elixir.bootlin.com/linux/v4.4/source/drivers/macintosh/adbhid.c#L1176
+ // https://github.com/NetBSD/src/blob/64b8a48e1288eb3902ed73113d157af50b2ec596/sys/arch/macppc/dev/ams.c#L261
+ static uint8_t cmd1[] = { 0xE7, 0x8C, 0x00, 0x00, 0x00, 0xFF, 0xFF, 0x94 };
+ static uint8_t cmd2[] = { 0xA5, 0x14, 0x00, 0x00, 0x69, 0xFF, 0xFF, 0x27 };
+
+ adb_host_flush(addr);
+ adb_host_listen_buf(addr, ADB_REG_2, cmd1, sizeof(cmd1));
+ adb_host_flush(addr);
+ adb_host_listen_buf(addr, ADB_REG_2, cmd2, sizeof(cmd2));
+ }
+
+
+ // Move to address 10 for mouse polling
+ adb_host_flush(addr);
+ adb_host_listen(addr, ADB_REG_3, ((reg3 >> 8) & 0xF0) | ADB_ADDR_MOUSE_POLL, 0xFE);
+ adb_host_flush(ADB_ADDR_MOUSE_POLL);
+
+ mouse_handler = (reg3 = adb_host_talk(addr, ADB_REG_3)) & 0xFF;
+ if (reg3) {
+ dmprintf("detect again\n");
+ goto detect_again;
+ }
+
+ goto again;
+}
+
+#ifdef MAX
+#undef MAX
+#endif
+#define MAX(X, Y) ((X) > (Y) ? (X) : (Y))
+
+static report_mouse_t mouse_report = {};
+
+void adb_mouse_task(void)
+{
+ uint8_t len;
+ uint8_t buf[5];
+ int16_t x, y;
+ static int8_t mouseacc;
+
+ /* tick of last polling */
+ static uint16_t tick_ms;
+
+ // polling with 12ms interval
+ if (timer_elapsed(tick_ms) < 12) return;
+ tick_ms = timer_read();
+
+ static uint16_t detect_ms;
+ if (timer_elapsed(detect_ms) > 1000) {
+ detect_ms = timer_read();
+ // check new device on addr3
+ mouse_init(ADB_ADDR_MOUSE);
+ }
+
+ // Extended Mouse Protocol data can be 2-5 bytes
+ // https://developer.apple.com/library/archive/technotes/hw/hw_01.html#Extended
+ //
+ // Byte 0: b00 y06 y05 y04 y03 y02 y01 y00
+ // Byte 1: b01 x06 x05 x04 x03 x02 x01 x00
+ // Byte 2: b02 y09 y08 y07 b03 x09 x08 x07
+ // Byte 3: b04 y12 y11 y10 b05 x12 x11 x10
+ // Byte 4: b06 y15 y14 y13 b07 x15 x14 x13
+ //
+ // b--: Button state.(0: on, 1: off)
+ // x--: X axis movement.
+ // y--: Y axis movement.
+ len = adb_host_talk_buf(ADB_ADDR_MOUSE_POLL, ADB_REG_0, buf, sizeof(buf));
+
+ // If nothing received reset mouse acceleration, and quit.
+ if (len < 2) {
+ mouseacc = 1;
+ return;
+ };
+ dmprintf("[%02X %02X %02X %02X %02X]\n", buf[0], buf[1], buf[2], buf[3], buf[4]);
+
+ // Store off-buttons and 0-movements in unused bytes
+ bool xneg = false;
+ bool yneg = false;
+ if (len == 2) {
+ if (buf[0] & 0x40) yneg = true;
+ if (buf[1] & 0x40) xneg = true;
+ } else {
+ if (buf[len - 1] & 0x40) yneg = true;
+ if (buf[len - 1] & 0x04) xneg = true;
+ }
+
+ for (int8_t i = len; i < sizeof(buf); i++) {
+ buf[i] = 0x88;
+ if (yneg) buf[i] |= 0x70;
+ if (xneg) buf[i] |= 0x07;
+ }
+
+ // 8 buttons at max
+ // TODO: Fix HID report descriptor for mouse to support button6-8
+ uint8_t buttons = 0;
+ if (!(buf[4] & 0x08)) buttons |= MOUSE_BTN8;
+ if (!(buf[4] & 0x80)) buttons |= MOUSE_BTN7;
+ if (!(buf[3] & 0x08)) buttons |= MOUSE_BTN6;
+ if (!(buf[3] & 0x80)) buttons |= MOUSE_BTN5;
+ if (!(buf[2] & 0x08)) buttons |= MOUSE_BTN4;
+ if (!(buf[2] & 0x80)) buttons |= MOUSE_BTN3;
+ if (!(buf[1] & 0x80)) buttons |= MOUSE_BTN2;
+ if (!(buf[0] & 0x80)) buttons |= MOUSE_BTN1;
+ mouse_report.buttons = buttons;
+
+ int16_t xx, yy;
+ yy = (buf[0] & 0x7F) | (buf[2] & 0x70) << 3 | (buf[3] & 0x70) << 6 | (buf[4] & 0x70) << 9;
+ xx = (buf[1] & 0x7F) | (buf[2] & 0x07) << 7 | (buf[3] & 0x07) << 10 | (buf[4] & 0x07) << 13;
+
+ // Accelerate mouse. (They weren't meant to be used on screens larger than 320x200).
+ x = xx * mouseacc;
+ y = yy * mouseacc;
+
+ // TODO: Fix HID report descriptor for mouse to support finer resolution
+ // Cap our two bytes per axis to one byte.
+ // Easier with a MIN-function, but since -MAX(-a,-b) = MIN(a,b)...
+ // I.E. MIN(MAX(x,-127),127) = -MAX(-MAX(x, -127), -127) = MIN(-MIN(-x,127),127)
+ mouse_report.x = -MAX(-MAX(x, -127), -127);
+ mouse_report.y = -MAX(-MAX(y, -127), -127);
+
+ dmprintf("[B:%02X X:%d(%d) Y:%d(%d) A:%d]\n", mouse_report.buttons, mouse_report.x, xx, mouse_report.y, yy, mouseacc);
+
+ // Send result by usb.
+ host_mouse_send(&mouse_report);
+
+ // TODO: acceleration curve is needed for precise operation?
+ // increase acceleration of mouse
+ mouseacc += ( mouseacc < (mouse_cpi < 200 ? ADB_MOUSE_MAXACC : ADB_MOUSE_MAXACC/2) ? 1 : 0 );
+