ps2_mouse.c

   1 #include <stdbool.h>
   2 #include<avr/io.h>
   3 #include<util/delay.h>
   4 #include "ps2.h"
   5 #include "ps2_mouse.h"
   6 #include "usb_mouse.h"
   7
   8 #define PS2_MOUSE_DEBUG
   9 #ifdef PS2_MOUSE_DEBUG
  10 #   include "print.h"
  11 #   include "debug.h"
  12 #else
  13 #   define print(s)
  14 #   define phex(h)
  15 #   define phex16(h)
  16 #endif
  17
  18 // disable when errors occur 255 times.
  19 #define ERROR_RETURN() do { \
  20     if (ps2_error) { \
  21         if (ps2_mouse_error_count < 255) { \
  22             ps2_mouse_error_count++; \
  23         } else { \
  24             ps2_mouse_error_count = 0; \
  25             ps2_mouse_enable = false; \
  26         } \
  27         return ps2_error; \
  28     } \
  29 } while (0)
  30
  31
  32 /*
  33 TODO
  34 ----
  35 - Stream mode
  36 - Tracpoint command support: needed
  37 - Middle button + move = Wheel traslation
  38 */
  39 bool ps2_mouse_enable = true;
  40 uint8_t ps2_mouse_x = 0;
  41 uint8_t ps2_mouse_y = 0;
  42 uint8_t ps2_mouse_btn = 0;
  43 uint8_t ps2_mouse_error_count = 0;
  44
  45 static uint8_t ps2_mouse_btn_prev = 0;
  46
  47
  48 uint8_t ps2_mouse_init(void) {
  49     uint8_t rcv;
  50
  51     if (!ps2_mouse_enable) return 1;
  52
  53     ps2_host_init();
  54
  55     // Reset
  56     rcv = ps2_host_send(0xFF);
  57     print("ps2_mouse_init: send 0xFF: ");
  58     phex(ps2_error); print("\n");
  59     ERROR_RETURN();
  60
  61     // ACK
  62     rcv = ps2_host_recv();
  63     print("ps2_mouse_init: read ACK: ");
  64     phex(rcv); phex(ps2_error); print("\n");
  65     ERROR_RETURN();
  66
  67     // BAT takes some time
  68     _delay_ms(100);
  69     rcv = ps2_host_recv();
  70     print("ps2_mouse_init: read BAT: ");
  71     phex(rcv); phex(ps2_error); print("\n");
  72     ERROR_RETURN();
  73
  74     // Device ID
  75     rcv = ps2_host_recv();
  76     print("ps2_mouse_init: read DevID: ");
  77     phex(rcv); phex(ps2_error); print("\n");
  78     ERROR_RETURN();
  79
  80     // Enable data reporting
  81     ps2_host_send(0xF4);
  82     print("ps2_mouse_init: send 0xF4: ");
  83     phex(ps2_error); print("\n");
  84     ERROR_RETURN();
  85
  86     // ACK
  87     rcv = ps2_host_recv();
  88     print("ps2_mouse_init: read ACK: ");
  89     phex(rcv); phex(ps2_error); print("\n");
  90     ERROR_RETURN();
  91
  92     // Set Remote mode
  93     ps2_host_send(0xF0);
  94     print("ps2_mouse_init: send 0xF0: ");
  95     phex(ps2_error); print("\n");
  96     ERROR_RETURN();
  97
  98     // ACK
  99     rcv = ps2_host_recv();
 100     print("ps2_mouse_init: read ACK: ");
 101     phex(rcv); phex(ps2_error); print("\n");
 102     ERROR_RETURN();
 103
 104     return 0;
 105 }
 106
 107 /*
 108 Data format:
 109     bit: 7       6       5       4       3       2       1       0
 110 -----------------------------------------------------------------------
 111 0   btn: Yovflw  Xovflw  Ysign   Xsign   1       Middle  Right   Left
 112 1   x:   X movement(0-255)
 113 2   y:   Y movement(0-255)
 114 */
 115 uint8_t ps2_mouse_read(void)
 116 {
 117     uint8_t rcv;
 118
 119     if (!ps2_mouse_enable) return 1;
 120
 121     ps2_host_send(0xEB);
 122     ERROR_RETURN();
 123
 124     rcv=ps2_host_recv();
 125     ERROR_RETURN();
 126
 127     if(rcv==0xFA) {
 128         ps2_mouse_btn = ps2_host_recv();
 129         ERROR_RETURN();
 130         ps2_mouse_x = ps2_host_recv();
 131         ERROR_RETURN();
 132         ps2_mouse_y = ps2_host_recv();
 133         ERROR_RETURN();
 134     }
 135     return 0;
 136 }
 137
 138 bool ps2_mouse_changed(void)
 139 {
 140     return (ps2_mouse_x || ps2_mouse_y || (ps2_mouse_btn & PS2_MOUSE_BTN_MASK) != ps2_mouse_btn_prev);
 141 }
 142
 143 #define PS2_MOUSE_SCROLL_BUTTON 0x04
 144 void ps2_mouse_usb_send(void)
 145 {
 146     static bool scrolled = false;
 147
 148     if (!ps2_mouse_enable) return;
 149
 150     if (ps2_mouse_changed()) {
 151         int8_t x, y, v, h;
 152         x = y = v = h = 0;
 153
 154         // convert scale of X, Y: PS/2(-256/255) -> USB(-127/127)
 155         if (ps2_mouse_btn & (1<<PS2_MOUSE_X_SIGN))
 156             x = ps2_mouse_x > 128 ? (int8_t)ps2_mouse_x : -127;
 157         else
 158             x = ps2_mouse_x < 128 ? (int8_t)ps2_mouse_x : 127;
 159
 160         if (ps2_mouse_btn & (1<<PS2_MOUSE_Y_SIGN))
 161             y = ps2_mouse_y > 128 ? (int8_t)ps2_mouse_y : -127;
 162         else
 163             y = ps2_mouse_y < 128 ? (int8_t)ps2_mouse_y : 127;
 164
 165         // Y is needed to reverse
 166         y = -y;
 167
 168         if (ps2_mouse_btn & PS2_MOUSE_SCROLL_BUTTON) {
 169             // scroll
 170             if (x > 0 || x < 0) h = (x > 64 ? 64 : (x < -64 ? -64 :x));
 171             if (y > 0 || y < 0) v = (y > 64 ? 64 : (y < -64 ? -64 :y));
 172             if (h || v) {
 173                 scrolled = true;
 174                 usb_mouse_send(0,0, -v/16, h/16, 0);
 175                 _delay_ms(100);
 176             }
 177         } else if (!scrolled && (ps2_mouse_btn_prev & PS2_MOUSE_SCROLL_BUTTON)) {
 178             usb_mouse_send(0,0,0,0, PS2_MOUSE_SCROLL_BUTTON);
 179             _delay_ms(100);
 180             usb_mouse_send(0,0,0,0, 0);
 181         } else {
 182             scrolled = false;
 183             usb_mouse_send(x, y, 0, 0, ps2_mouse_btn & PS2_MOUSE_BTN_MASK);
 184         }
 185
 186         ps2_mouse_btn_prev = (ps2_mouse_btn & PS2_MOUSE_BTN_MASK);
 187         ps2_mouse_print();
 188     }
 189     ps2_mouse_x = 0;
 190     ps2_mouse_y = 0;
 191     ps2_mouse_btn = 0;
 192 }
 193
 194 void ps2_mouse_print(void)
 195 {
 196     if (!debug_mouse) return;
 197     print("ps2_mouse[btn|x y]: ");
 198     phex(ps2_mouse_btn); print("|");
 199     phex(ps2_mouse_x); print(" ");
 200     phex(ps2_mouse_y); print("\n");
 201 }