mousekey.c 13 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373
  1. /*
  2. Copyright 2011 Jun Wako <wakojun@gmail.com>
  3. This program is free software: you can redistribute it and/or modify
  4. it under the terms of the GNU General Public License as published by
  5. the Free Software Foundation, either version 2 of the License, or
  6. (at your option) any later version.
  7. This program is distributed in the hope that it will be useful,
  8. but WITHOUT ANY WARRANTY; without even the implied warranty of
  9. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  10. GNU General Public License for more details.
  11. You should have received a copy of the GNU General Public License
  12. along with this program. If not, see <http://www.gnu.org/licenses/>.
  13. */
  14. #include <stdint.h>
  15. #include "keycode.h"
  16. #include "host.h"
  17. #include "timer.h"
  18. #include "print.h"
  19. #include "debug.h"
  20. #include "mousekey.h"
  21. inline int8_t times_inv_sqrt2(int8_t x) {
  22. // 181/256 is pretty close to 1/sqrt(2)
  23. // 0.70703125 0.707106781
  24. // 1 too small for x=99 and x=198
  25. // This ends up being a mult and discard lower 8 bits
  26. return (x * 181) >> 8;
  27. }
  28. static report_mouse_t mouse_report = {0};
  29. static void mousekey_debug(void);
  30. static uint8_t mousekey_accel = 0;
  31. static uint8_t mousekey_repeat = 0;
  32. static uint16_t last_timer = 0;
  33. #ifndef MK_3_SPEED
  34. /*
  35. * Mouse keys acceleration algorithm
  36. * http://en.wikipedia.org/wiki/Mouse_keys
  37. *
  38. * speed = delta * max_speed * (repeat / time_to_max)**((1000+curve)/1000)
  39. */
  40. /* milliseconds between the initial key press and first repeated motion event (0-2550) */
  41. uint8_t mk_delay = MOUSEKEY_DELAY / 10;
  42. /* milliseconds between repeated motion events (0-255) */
  43. uint8_t mk_interval = MOUSEKEY_INTERVAL;
  44. /* steady speed (in action_delta units) applied each event (0-255) */
  45. uint8_t mk_max_speed = MOUSEKEY_MAX_SPEED;
  46. /* number of events (count) accelerating to steady speed (0-255) */
  47. uint8_t mk_time_to_max = MOUSEKEY_TIME_TO_MAX;
  48. /* ramp used to reach maximum pointer speed (NOT SUPPORTED) */
  49. // int8_t mk_curve = 0;
  50. /* wheel params */
  51. uint8_t mk_wheel_max_speed = MOUSEKEY_WHEEL_MAX_SPEED;
  52. uint8_t mk_wheel_time_to_max = MOUSEKEY_WHEEL_TIME_TO_MAX;
  53. static uint8_t move_unit(void) {
  54. uint16_t unit;
  55. if (mousekey_accel & (1 << 0)) {
  56. unit = (MOUSEKEY_MOVE_DELTA * mk_max_speed) / 4;
  57. } else if (mousekey_accel & (1 << 1)) {
  58. unit = (MOUSEKEY_MOVE_DELTA * mk_max_speed) / 2;
  59. } else if (mousekey_accel & (1 << 2)) {
  60. unit = (MOUSEKEY_MOVE_DELTA * mk_max_speed);
  61. } else if (mousekey_repeat == 0) {
  62. unit = MOUSEKEY_MOVE_DELTA;
  63. } else if (mousekey_repeat >= mk_time_to_max) {
  64. unit = MOUSEKEY_MOVE_DELTA * mk_max_speed;
  65. } else {
  66. unit = (MOUSEKEY_MOVE_DELTA * mk_max_speed * mousekey_repeat) / mk_time_to_max;
  67. }
  68. return (unit > MOUSEKEY_MOVE_MAX ? MOUSEKEY_MOVE_MAX : (unit == 0 ? 1 : unit));
  69. }
  70. static uint8_t wheel_unit(void) {
  71. uint16_t unit;
  72. if (mousekey_accel & (1 << 0)) {
  73. unit = (MOUSEKEY_WHEEL_DELTA * mk_wheel_max_speed) / 4;
  74. } else if (mousekey_accel & (1 << 1)) {
  75. unit = (MOUSEKEY_WHEEL_DELTA * mk_wheel_max_speed) / 2;
  76. } else if (mousekey_accel & (1 << 2)) {
  77. unit = (MOUSEKEY_WHEEL_DELTA * mk_wheel_max_speed);
  78. } else if (mousekey_repeat == 0) {
  79. unit = MOUSEKEY_WHEEL_DELTA;
  80. } else if (mousekey_repeat >= mk_wheel_time_to_max) {
  81. unit = MOUSEKEY_WHEEL_DELTA * mk_wheel_max_speed;
  82. } else {
  83. unit = (MOUSEKEY_WHEEL_DELTA * mk_wheel_max_speed * mousekey_repeat) / mk_wheel_time_to_max;
  84. }
  85. return (unit > MOUSEKEY_WHEEL_MAX ? MOUSEKEY_WHEEL_MAX : (unit == 0 ? 1 : unit));
  86. }
  87. void mousekey_task(void) {
  88. if (timer_elapsed(last_timer) < (mousekey_repeat ? mk_interval : mk_delay * 10)) {
  89. return;
  90. }
  91. if (mouse_report.x == 0 && mouse_report.y == 0 && mouse_report.v == 0 && mouse_report.h == 0) {
  92. return;
  93. }
  94. if (mousekey_repeat != UINT8_MAX) mousekey_repeat++;
  95. if (mouse_report.x > 0) mouse_report.x = move_unit();
  96. if (mouse_report.x < 0) mouse_report.x = move_unit() * -1;
  97. if (mouse_report.y > 0) mouse_report.y = move_unit();
  98. if (mouse_report.y < 0) mouse_report.y = move_unit() * -1;
  99. /* diagonal move [1/sqrt(2)] */
  100. if (mouse_report.x && mouse_report.y) {
  101. mouse_report.x = times_inv_sqrt2(mouse_report.x);
  102. if (mouse_report.x == 0) {
  103. mouse_report.x = 1;
  104. }
  105. mouse_report.y = times_inv_sqrt2(mouse_report.y);
  106. if (mouse_report.y == 0) {
  107. mouse_report.y = 1;
  108. }
  109. }
  110. if (mouse_report.v > 0) mouse_report.v = wheel_unit();
  111. if (mouse_report.v < 0) mouse_report.v = wheel_unit() * -1;
  112. if (mouse_report.h > 0) mouse_report.h = wheel_unit();
  113. if (mouse_report.h < 0) mouse_report.h = wheel_unit() * -1;
  114. mousekey_send();
  115. }
  116. void mousekey_on(uint8_t code) {
  117. if (code == KC_MS_UP)
  118. mouse_report.y = move_unit() * -1;
  119. else if (code == KC_MS_DOWN)
  120. mouse_report.y = move_unit();
  121. else if (code == KC_MS_LEFT)
  122. mouse_report.x = move_unit() * -1;
  123. else if (code == KC_MS_RIGHT)
  124. mouse_report.x = move_unit();
  125. else if (code == KC_MS_WH_UP)
  126. mouse_report.v = wheel_unit();
  127. else if (code == KC_MS_WH_DOWN)
  128. mouse_report.v = wheel_unit() * -1;
  129. else if (code == KC_MS_WH_LEFT)
  130. mouse_report.h = wheel_unit() * -1;
  131. else if (code == KC_MS_WH_RIGHT)
  132. mouse_report.h = wheel_unit();
  133. else if (code == KC_MS_BTN1)
  134. mouse_report.buttons |= MOUSE_BTN1;
  135. else if (code == KC_MS_BTN2)
  136. mouse_report.buttons |= MOUSE_BTN2;
  137. else if (code == KC_MS_BTN3)
  138. mouse_report.buttons |= MOUSE_BTN3;
  139. else if (code == KC_MS_BTN4)
  140. mouse_report.buttons |= MOUSE_BTN4;
  141. else if (code == KC_MS_BTN5)
  142. mouse_report.buttons |= MOUSE_BTN5;
  143. else if (code == KC_MS_ACCEL0)
  144. mousekey_accel |= (1 << 0);
  145. else if (code == KC_MS_ACCEL1)
  146. mousekey_accel |= (1 << 1);
  147. else if (code == KC_MS_ACCEL2)
  148. mousekey_accel |= (1 << 2);
  149. }
  150. void mousekey_off(uint8_t code) {
  151. if (code == KC_MS_UP && mouse_report.y < 0)
  152. mouse_report.y = 0;
  153. else if (code == KC_MS_DOWN && mouse_report.y > 0)
  154. mouse_report.y = 0;
  155. else if (code == KC_MS_LEFT && mouse_report.x < 0)
  156. mouse_report.x = 0;
  157. else if (code == KC_MS_RIGHT && mouse_report.x > 0)
  158. mouse_report.x = 0;
  159. else if (code == KC_MS_WH_UP && mouse_report.v > 0)
  160. mouse_report.v = 0;
  161. else if (code == KC_MS_WH_DOWN && mouse_report.v < 0)
  162. mouse_report.v = 0;
  163. else if (code == KC_MS_WH_LEFT && mouse_report.h < 0)
  164. mouse_report.h = 0;
  165. else if (code == KC_MS_WH_RIGHT && mouse_report.h > 0)
  166. mouse_report.h = 0;
  167. else if (code == KC_MS_BTN1)
  168. mouse_report.buttons &= ~MOUSE_BTN1;
  169. else if (code == KC_MS_BTN2)
  170. mouse_report.buttons &= ~MOUSE_BTN2;
  171. else if (code == KC_MS_BTN3)
  172. mouse_report.buttons &= ~MOUSE_BTN3;
  173. else if (code == KC_MS_BTN4)
  174. mouse_report.buttons &= ~MOUSE_BTN4;
  175. else if (code == KC_MS_BTN5)
  176. mouse_report.buttons &= ~MOUSE_BTN5;
  177. else if (code == KC_MS_ACCEL0)
  178. mousekey_accel &= ~(1 << 0);
  179. else if (code == KC_MS_ACCEL1)
  180. mousekey_accel &= ~(1 << 1);
  181. else if (code == KC_MS_ACCEL2)
  182. mousekey_accel &= ~(1 << 2);
  183. if (mouse_report.x == 0 && mouse_report.y == 0 && mouse_report.v == 0 && mouse_report.h == 0) mousekey_repeat = 0;
  184. }
  185. #else /* #ifndef MK_3_SPEED */
  186. enum { mkspd_unmod, mkspd_0, mkspd_1, mkspd_2, mkspd_COUNT };
  187. # ifndef MK_MOMENTARY_ACCEL
  188. static uint8_t mk_speed = mkspd_1;
  189. # else
  190. static uint8_t mk_speed = mkspd_unmod;
  191. static uint8_t mkspd_DEFAULT = mkspd_unmod;
  192. # endif
  193. static uint16_t last_timer_c = 0;
  194. static uint16_t last_timer_w = 0;
  195. uint16_t c_offsets[mkspd_COUNT] = {MK_C_OFFSET_UNMOD, MK_C_OFFSET_0, MK_C_OFFSET_1, MK_C_OFFSET_2};
  196. uint16_t c_intervals[mkspd_COUNT] = {MK_C_INTERVAL_UNMOD, MK_C_INTERVAL_0, MK_C_INTERVAL_1, MK_C_INTERVAL_2};
  197. uint16_t w_offsets[mkspd_COUNT] = {MK_W_OFFSET_UNMOD, MK_W_OFFSET_0, MK_W_OFFSET_1, MK_W_OFFSET_2};
  198. uint16_t w_intervals[mkspd_COUNT] = {MK_W_INTERVAL_UNMOD, MK_W_INTERVAL_0, MK_W_INTERVAL_1, MK_W_INTERVAL_2};
  199. void mousekey_task(void) {
  200. // report cursor and scroll movement independently
  201. report_mouse_t const tmpmr = mouse_report;
  202. if ((mouse_report.x || mouse_report.y) && timer_elapsed(last_timer_c) > c_intervals[mk_speed]) {
  203. mouse_report.h = 0;
  204. mouse_report.v = 0;
  205. mousekey_send();
  206. last_timer_c = last_timer;
  207. mouse_report = tmpmr;
  208. }
  209. if ((mouse_report.h || mouse_report.v) && timer_elapsed(last_timer_w) > w_intervals[mk_speed]) {
  210. mouse_report.x = 0;
  211. mouse_report.y = 0;
  212. mousekey_send();
  213. last_timer_w = last_timer;
  214. mouse_report = tmpmr;
  215. }
  216. }
  217. void adjust_speed(void) {
  218. uint16_t const c_offset = c_offsets[mk_speed];
  219. uint16_t const w_offset = w_offsets[mk_speed];
  220. if (mouse_report.x > 0) mouse_report.x = c_offset;
  221. if (mouse_report.x < 0) mouse_report.x = c_offset * -1;
  222. if (mouse_report.y > 0) mouse_report.y = c_offset;
  223. if (mouse_report.y < 0) mouse_report.y = c_offset * -1;
  224. if (mouse_report.h > 0) mouse_report.h = w_offset;
  225. if (mouse_report.h < 0) mouse_report.h = w_offset * -1;
  226. if (mouse_report.v > 0) mouse_report.v = w_offset;
  227. if (mouse_report.v < 0) mouse_report.v = w_offset * -1;
  228. // adjust for diagonals
  229. if (mouse_report.x && mouse_report.y) {
  230. mouse_report.x = times_inv_sqrt2(mouse_report.x);
  231. if (mouse_report.x == 0) {
  232. mouse_report.x = 1;
  233. }
  234. mouse_report.y = times_inv_sqrt2(mouse_report.y);
  235. if (mouse_report.y == 0) {
  236. mouse_report.y = 1;
  237. }
  238. }
  239. if (mouse_report.h && mouse_report.v) {
  240. mouse_report.h = times_inv_sqrt2(mouse_report.h);
  241. mouse_report.v = times_inv_sqrt2(mouse_report.v);
  242. }
  243. }
  244. void mousekey_on(uint8_t code) {
  245. uint16_t const c_offset = c_offsets[mk_speed];
  246. uint16_t const w_offset = w_offsets[mk_speed];
  247. uint8_t const old_speed = mk_speed;
  248. if (code == KC_MS_UP)
  249. mouse_report.y = c_offset * -1;
  250. else if (code == KC_MS_DOWN)
  251. mouse_report.y = c_offset;
  252. else if (code == KC_MS_LEFT)
  253. mouse_report.x = c_offset * -1;
  254. else if (code == KC_MS_RIGHT)
  255. mouse_report.x = c_offset;
  256. else if (code == KC_MS_WH_UP)
  257. mouse_report.v = w_offset;
  258. else if (code == KC_MS_WH_DOWN)
  259. mouse_report.v = w_offset * -1;
  260. else if (code == KC_MS_WH_LEFT)
  261. mouse_report.h = w_offset * -1;
  262. else if (code == KC_MS_WH_RIGHT)
  263. mouse_report.h = w_offset;
  264. else if (code == KC_MS_BTN1)
  265. mouse_report.buttons |= MOUSE_BTN1;
  266. else if (code == KC_MS_BTN2)
  267. mouse_report.buttons |= MOUSE_BTN2;
  268. else if (code == KC_MS_BTN3)
  269. mouse_report.buttons |= MOUSE_BTN3;
  270. else if (code == KC_MS_BTN4)
  271. mouse_report.buttons |= MOUSE_BTN4;
  272. else if (code == KC_MS_BTN5)
  273. mouse_report.buttons |= MOUSE_BTN5;
  274. else if (code == KC_MS_ACCEL0)
  275. mk_speed = mkspd_0;
  276. else if (code == KC_MS_ACCEL1)
  277. mk_speed = mkspd_1;
  278. else if (code == KC_MS_ACCEL2)
  279. mk_speed = mkspd_2;
  280. if (mk_speed != old_speed) adjust_speed();
  281. }
  282. void mousekey_off(uint8_t code) {
  283. # ifdef MK_MOMENTARY_ACCEL
  284. uint8_t const old_speed = mk_speed;
  285. # endif
  286. if (code == KC_MS_UP && mouse_report.y < 0)
  287. mouse_report.y = 0;
  288. else if (code == KC_MS_DOWN && mouse_report.y > 0)
  289. mouse_report.y = 0;
  290. else if (code == KC_MS_LEFT && mouse_report.x < 0)
  291. mouse_report.x = 0;
  292. else if (code == KC_MS_RIGHT && mouse_report.x > 0)
  293. mouse_report.x = 0;
  294. else if (code == KC_MS_WH_UP && mouse_report.v > 0)
  295. mouse_report.v = 0;
  296. else if (code == KC_MS_WH_DOWN && mouse_report.v < 0)
  297. mouse_report.v = 0;
  298. else if (code == KC_MS_WH_LEFT && mouse_report.h < 0)
  299. mouse_report.h = 0;
  300. else if (code == KC_MS_WH_RIGHT && mouse_report.h > 0)
  301. mouse_report.h = 0;
  302. else if (code == KC_MS_BTN1)
  303. mouse_report.buttons &= ~MOUSE_BTN1;
  304. else if (code == KC_MS_BTN2)
  305. mouse_report.buttons &= ~MOUSE_BTN2;
  306. else if (code == KC_MS_BTN3)
  307. mouse_report.buttons &= ~MOUSE_BTN3;
  308. else if (code == KC_MS_BTN4)
  309. mouse_report.buttons &= ~MOUSE_BTN4;
  310. else if (code == KC_MS_BTN5)
  311. mouse_report.buttons &= ~MOUSE_BTN5;
  312. # ifdef MK_MOMENTARY_ACCEL
  313. else if (code == KC_MS_ACCEL0)
  314. mk_speed = mkspd_DEFAULT;
  315. else if (code == KC_MS_ACCEL1)
  316. mk_speed = mkspd_DEFAULT;
  317. else if (code == KC_MS_ACCEL2)
  318. mk_speed = mkspd_DEFAULT;
  319. if (mk_speed != old_speed) adjust_speed();
  320. # endif
  321. }
  322. #endif /* #ifndef MK_3_SPEED */
  323. void mousekey_send(void) {
  324. mousekey_debug();
  325. host_mouse_send(&mouse_report);
  326. last_timer = timer_read();
  327. }
  328. void mousekey_clear(void) {
  329. mouse_report = (report_mouse_t){};
  330. mousekey_repeat = 0;
  331. mousekey_accel = 0;
  332. }
  333. static void mousekey_debug(void) {
  334. if (!debug_mouse) return;
  335. print("mousekey [btn|x y v h](rep/acl): [");
  336. phex(mouse_report.buttons);
  337. print("|");
  338. print_decs(mouse_report.x);
  339. print(" ");
  340. print_decs(mouse_report.y);
  341. print(" ");
  342. print_decs(mouse_report.v);
  343. print(" ");
  344. print_decs(mouse_report.h);
  345. print("](");
  346. print_dec(mousekey_repeat);
  347. print("/");
  348. print_dec(mousekey_accel);
  349. print(")\n");
  350. }