tmk_core/protocol/ps2_mouse.c

   1 /*
   2 Copyright 2011,2013 Jun Wako <wakojun@gmail.com>
   3
   4 This program is free software: you can redistribute it and/or modify
   5 it under the terms of the GNU General Public License as published by
   6 the Free Software Foundation, either version 2 of the License, or
   7 (at your option) any later version.
   8
   9 This program is distributed in the hope that it will be useful,
  10 but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 GNU General Public License for more details.
  13
  14 You should have received a copy of the GNU General Public License
  15 along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16 */
  17
  18 #include <stdbool.h>
  19 #include<avr/io.h>
  20 #include<util/delay.h>
  21 #include "ps2.h"
  22 #include "ps2_mouse.h"
  23 #include "report.h"
  24 #include "host.h"
  25 #include "timer.h"
  26 #include "print.h"
  27 #include "debug.h"
  28
  29
  30 static report_mouse_t mouse_report = {};
  31
  32
  33 static void print_usb_data(void);
  34
  35
  36 /* supports only 3 button mouse at this time */
  37 uint8_t ps2_mouse_init(void) {
  38     uint8_t rcv;
  39
  40     ps2_host_init();
  41
  42     _delay_ms(1000);    // wait for powering up
  43
  44     // send Reset
  45     rcv = ps2_host_send(0xFF);
  46     print("ps2_mouse_init: send Reset: ");
  47     phex(rcv); phex(ps2_error); print("\n");
  48
  49     // read completion code of BAT
  50     rcv = ps2_host_recv_response();
  51     print("ps2_mouse_init: read BAT: ");
  52     phex(rcv); phex(ps2_error); print("\n");
  53
  54     // read Device ID
  55     rcv = ps2_host_recv_response();
  56     print("ps2_mouse_init: read DevID: ");
  57     phex(rcv); phex(ps2_error); print("\n");
  58
  59     // send Set Remote mode
  60     rcv = ps2_host_send(0xF0);
  61     print("ps2_mouse_init: send 0xF0: ");
  62     phex(rcv); phex(ps2_error); print("\n");
  63
  64     return 0;
  65 }
  66
  67 #define X_IS_NEG  (mouse_report.buttons & (1<<PS2_MOUSE_X_SIGN))
  68 #define Y_IS_NEG  (mouse_report.buttons & (1<<PS2_MOUSE_Y_SIGN))
  69 #define X_IS_OVF  (mouse_report.buttons & (1<<PS2_MOUSE_X_OVFLW))
  70 #define Y_IS_OVF  (mouse_report.buttons & (1<<PS2_MOUSE_Y_OVFLW))
  71 void ps2_mouse_task(void)
  72 {
  73     enum { SCROLL_NONE, SCROLL_BTN, SCROLL_SENT };
  74     static uint8_t scroll_state = SCROLL_NONE;
  75     static uint8_t buttons_prev = 0;
  76
  77     /* receives packet from mouse */
  78     uint8_t rcv;
  79     rcv = ps2_host_send(PS2_MOUSE_READ_DATA);
  80     if (rcv == PS2_ACK) {
  81         mouse_report.buttons = ps2_host_recv_response();
  82         mouse_report.x = ps2_host_recv_response();
  83         mouse_report.y = ps2_host_recv_response();
  84     } else {
  85         if (debug_mouse) print("ps2_mouse: fail to get mouse packet\n");
  86         return;
  87     }
  88
  89     /* if mouse moves or buttons state changes */
  90     if (mouse_report.x || mouse_report.y ||
  91             ((mouse_report.buttons ^ buttons_prev) & PS2_MOUSE_BTN_MASK)) {
  92
  93 #ifdef PS2_MOUSE_DEBUG
  94         xprintf("%ud ", timer_read());
  95         print("ps2_mouse raw: [");
  96         phex(mouse_report.buttons); print("|");
  97         print_hex8((uint8_t)mouse_report.x); print(" ");
  98         print_hex8((uint8_t)mouse_report.y); print("]\n");
  99 #endif
 100
 101         buttons_prev = mouse_report.buttons;
 102
 103         // PS/2 mouse data is '9-bit integer'(-256 to 255) which is comprised of sign-bit and 8-bit value.
 104         // bit: 8    7 ... 0
 105         //      sign \8-bit/
 106         //
 107         // Meanwhile USB HID mouse indicates 8bit data(-127 to 127), note that -128 is not used.
 108         //
 109         // This converts PS/2 data into HID value. Use only -127-127 out of PS/2 9-bit.
 110         mouse_report.x = X_IS_NEG ?
 111                           ((!X_IS_OVF && -127 <= mouse_report.x && mouse_report.x <= -1) ?  mouse_report.x : -127) :
 112                           ((!X_IS_OVF && 0 <= mouse_report.x && mouse_report.x <= 127) ? mouse_report.x : 127);
 113         mouse_report.y = Y_IS_NEG ?
 114                           ((!Y_IS_OVF && -127 <= mouse_report.y && mouse_report.y <= -1) ?  mouse_report.y : -127) :
 115                           ((!Y_IS_OVF && 0 <= mouse_report.y && mouse_report.y <= 127) ? mouse_report.y : 127);
 116
 117         // remove sign and overflow flags
 118         mouse_report.buttons &= PS2_MOUSE_BTN_MASK;
 119
 120         // invert coordinate of y to conform to USB HID mouse
 121         mouse_report.y = -mouse_report.y;
 122
 123
 124 #if PS2_MOUSE_SCROLL_BTN_MASK
 125         static uint16_t scroll_button_time = 0;
 126         if ((mouse_report.buttons & (PS2_MOUSE_SCROLL_BTN_MASK)) == (PS2_MOUSE_SCROLL_BTN_MASK)) {
 127             if (scroll_state == SCROLL_NONE) {
 128                 scroll_button_time = timer_read();
 129                 scroll_state = SCROLL_BTN;
 130             }
 131
 132             // doesn't send Scroll Button
 133             //mouse_report.buttons &= ~(PS2_MOUSE_SCROLL_BTN_MASK);
 134
 135             if (mouse_report.x || mouse_report.y) {
 136                 scroll_state = SCROLL_SENT;
 137
 138                 mouse_report.v = -mouse_report.y/(PS2_MOUSE_SCROLL_DIVISOR_V);
 139                 mouse_report.h =  mouse_report.x/(PS2_MOUSE_SCROLL_DIVISOR_H);
 140                 mouse_report.x = 0;
 141                 mouse_report.y = 0;
 142                 //host_mouse_send(&mouse_report);
 143             }
 144         }
 145         else if ((mouse_report.buttons & (PS2_MOUSE_SCROLL_BTN_MASK)) == 0) {
 146 #if PS2_MOUSE_SCROLL_BTN_SEND
 147             if (scroll_state == SCROLL_BTN &&
 148                     TIMER_DIFF_16(timer_read(), scroll_button_time) < PS2_MOUSE_SCROLL_BTN_SEND) {
 149                 // send Scroll Button(down and up at once) when not scrolled
 150                 mouse_report.buttons |= (PS2_MOUSE_SCROLL_BTN_MASK);
 151                 host_mouse_send(&mouse_report);
 152                 _delay_ms(100);
 153                 mouse_report.buttons &= ~(PS2_MOUSE_SCROLL_BTN_MASK);
 154             }
 155 #endif
 156             scroll_state = SCROLL_NONE;
 157         }
 158         // doesn't send Scroll Button
 159         mouse_report.buttons &= ~(PS2_MOUSE_SCROLL_BTN_MASK);
 160 #endif
 161
 162
 163         host_mouse_send(&mouse_report);
 164         print_usb_data();
 165     }
 166     // clear report
 167     mouse_report.x = 0;
 168     mouse_report.y = 0;
 169     mouse_report.v = 0;
 170     mouse_report.h = 0;
 171     mouse_report.buttons = 0;
 172 }
 173
 174 static void print_usb_data(void)
 175 {
 176     if (!debug_mouse) return;
 177     print("ps2_mouse usb: [");
 178     phex(mouse_report.buttons); print("|");
 179     print_hex8((uint8_t)mouse_report.x); print(" ");
 180     print_hex8((uint8_t)mouse_report.y); print(" ");
 181     print_hex8((uint8_t)mouse_report.v); print(" ");
 182     print_hex8((uint8_t)mouse_report.h); print("]\n");
 183 }
 184
 185
 186 /* PS/2 Mouse Synopsis
 187  * http://www.computer-engineering.org/ps2mouse/
 188  *
 189  * Command:
 190  * 0xFF: Reset
 191  * 0xF6: Set Defaults Sampling; rate=100, resolution=4cnt/mm, scaling=1:1, reporting=disabled
 192  * 0xF5: Disable Data Reporting
 193  * 0xF4: Enable Data Reporting
 194  * 0xF3: Set Sample Rate
 195  * 0xF2: Get Device ID
 196  * 0xF0: Set Remote Mode
 197  * 0xEB: Read Data
 198  * 0xEA: Set Stream Mode
 199  * 0xE9: Status Request
 200  * 0xE8: Set Resolution
 201  * 0xE7: Set Scaling 2:1
 202  * 0xE6: Set Scaling 1:1
 203  *
 204  * Mode:
 205  * Stream Mode: devices sends the data when it changs its state
 206  * Remote Mode: host polls the data periodically
 207  *
 208  * This code uses Remote Mode and polls the data with Read Data(0xEB).
 209  *
 210  * Data format:
 211  * byte|7       6       5       4       3       2       1       0
 212  * ----+--------------------------------------------------------------
 213  *    0|Yovflw  Xovflw  Ysign   Xsign   1       Middle  Right   Left
 214  *    1|                    X movement
 215  *    2|                    Y movement
 216  */