projects / linux-2.6-block.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
HID: wiimote: add Balance Board support
[linux-2.6-block.git] / drivers / hid / hid-wiimote-modules.c
CommitLineData
27f06942
DH		 1/*
2 * Device Modules for Nintendo Wii / Wii U HID Driver
3 * Copyright (c) 2011-2013 David Herrmann <dh.herrmann@gmail.com>
4 */
5
6/*
7 * This program is free software; you can redistribute it and/or modify it
8 * under the terms of the GNU General Public License as published by the Free
9 * Software Foundation; either version 2 of the License, or (at your option)
10 * any later version.
11 */
12
13/*
14 * Wiimote Modules
15 * Nintendo devices provide different peripherals and many new devices lack
16 * initial features like the IR camera. Therefore, each peripheral device is
17 * implemented as an independent module and we probe on each device only the
18 * modules for the hardware that really is available.
19 *
20 * Module registration is sequential. Unregistration is done in reverse order.
21 * After device detection, the needed modules are loaded. Users can trigger
22 * re-detection which causes all modules to be unloaded and then reload the
23 * modules for the new detected device.
24 *
25 * wdata->input is a shared input device. It is always initialized prior to
26 * module registration. If at least one registered module is marked as
27 * WIIMOD_FLAG_INPUT, then the input device will get registered after all
28 * modules were registered.
29 * Please note that it is unregistered _before_ the "remove" callbacks are
30 * called. This guarantees that no input interaction is done, anymore. However,
31 * the wiimote core keeps a reference to the input device so it is freed only
32 * after all modules were removed. It is safe to send events to unregistered
33 * input devices.
34 */
35
36#include <linux/device.h>
37#include <linux/hid.h>
38#include <linux/input.h>
39#include <linux/spinlock.h>
40#include "hid-wiimote.h"
41
20cef813
DH		 42/*
43 * Keys
44 * The initial Wii Remote provided a bunch of buttons that are reported as
45 * part of the core protocol. Many later devices dropped these and report
46 * invalid data in the core button reports. Load this only on devices which
47 * correctly send button reports.
48 * It uses the shared input device.
49 */
50
51static const __u16 wiimod_keys_map[] = {
52 KEY_LEFT, /* WIIPROTO_KEY_LEFT */
53 KEY_RIGHT, /* WIIPROTO_KEY_RIGHT */
54 KEY_UP, /* WIIPROTO_KEY_UP */
55 KEY_DOWN, /* WIIPROTO_KEY_DOWN */
56 KEY_NEXT, /* WIIPROTO_KEY_PLUS */
57 KEY_PREVIOUS, /* WIIPROTO_KEY_MINUS */
58 BTN_1, /* WIIPROTO_KEY_ONE */
59 BTN_2, /* WIIPROTO_KEY_TWO */
60 BTN_A, /* WIIPROTO_KEY_A */
61 BTN_B, /* WIIPROTO_KEY_B */
62 BTN_MODE, /* WIIPROTO_KEY_HOME */
63};
64
65static void wiimod_keys_in_keys(struct wiimote_data *wdata, const __u8 *keys)
66{
67 input_report_key(wdata->input, wiimod_keys_map[WIIPROTO_KEY_LEFT],
68 !!(keys[0] & 0x01));
69 input_report_key(wdata->input, wiimod_keys_map[WIIPROTO_KEY_RIGHT],
70 !!(keys[0] & 0x02));
71 input_report_key(wdata->input, wiimod_keys_map[WIIPROTO_KEY_DOWN],
72 !!(keys[0] & 0x04));
73 input_report_key(wdata->input, wiimod_keys_map[WIIPROTO_KEY_UP],
74 !!(keys[0] & 0x08));
75 input_report_key(wdata->input, wiimod_keys_map[WIIPROTO_KEY_PLUS],
76 !!(keys[0] & 0x10));
77 input_report_key(wdata->input, wiimod_keys_map[WIIPROTO_KEY_TWO],
78 !!(keys[1] & 0x01));
79 input_report_key(wdata->input, wiimod_keys_map[WIIPROTO_KEY_ONE],
80 !!(keys[1] & 0x02));
81 input_report_key(wdata->input, wiimod_keys_map[WIIPROTO_KEY_B],
82 !!(keys[1] & 0x04));
83 input_report_key(wdata->input, wiimod_keys_map[WIIPROTO_KEY_A],
84 !!(keys[1] & 0x08));
85 input_report_key(wdata->input, wiimod_keys_map[WIIPROTO_KEY_MINUS],
86 !!(keys[1] & 0x10));
87 input_report_key(wdata->input, wiimod_keys_map[WIIPROTO_KEY_HOME],
88 !!(keys[1] & 0x80));
89 input_sync(wdata->input);
90}
91
92static int wiimod_keys_probe(const struct wiimod_ops *ops,
93 struct wiimote_data *wdata)
94{
95 unsigned int i;
96
97 set_bit(EV_KEY, wdata->input->evbit);
98 for (i = 0; i < WIIPROTO_KEY_COUNT; ++i)
99 set_bit(wiimod_keys_map[i], wdata->input->keybit);
100
101 return 0;
102}
103
104static const struct wiimod_ops wiimod_keys = {
105 .flags = WIIMOD_FLAG_INPUT,
106 .arg = 0,
107 .probe = wiimod_keys_probe,
108 .remove = NULL,
109 .in_keys = wiimod_keys_in_keys,
110};
111
112/*
113 * Rumble
114 * Nearly all devices provide a rumble feature. A small motor for
115 * force-feedback effects. We provide an FF_RUMBLE memless ff device on the
116 * shared input device if this module is loaded.
117 * The rumble motor is controlled via a flag on almost every output report so
118 * the wiimote core handles the rumble flag. But if a device doesn't provide
119 * the rumble motor, this flag shouldn't be set.
120 */
121
122static int wiimod_rumble_play(struct input_dev *dev, void *data,
123 struct ff_effect *eff)
124{
125 struct wiimote_data *wdata = input_get_drvdata(dev);
126 __u8 value;
127 unsigned long flags;
128
129 /*
130 * The wiimote supports only a single rumble motor so if any magnitude
131 * is set to non-zero then we start the rumble motor. If both are set to
132 * zero, we stop the rumble motor.
133 */
134
135 if (eff->u.rumble.strong_magnitude || eff->u.rumble.weak_magnitude)
136 value = 1;
137 else
138 value = 0;
139
140 spin_lock_irqsave(&wdata->state.lock, flags);
141 wiiproto_req_rumble(wdata, value);
142 spin_unlock_irqrestore(&wdata->state.lock, flags);
143
144 return 0;
145}
146
147static int wiimod_rumble_probe(const struct wiimod_ops *ops,
148 struct wiimote_data *wdata)
149{
150 set_bit(FF_RUMBLE, wdata->input->ffbit);
151 if (input_ff_create_memless(wdata->input, NULL, wiimod_rumble_play))
152 return -ENOMEM;
153
154 return 0;
155}
156
157static void wiimod_rumble_remove(const struct wiimod_ops *ops,
158 struct wiimote_data *wdata)
159{
160 unsigned long flags;
161
162 spin_lock_irqsave(&wdata->state.lock, flags);
163 wiiproto_req_rumble(wdata, 0);
164 spin_unlock_irqrestore(&wdata->state.lock, flags);
165}
166
167static const struct wiimod_ops wiimod_rumble = {
168 .flags = WIIMOD_FLAG_INPUT,
169 .arg = 0,
170 .probe = wiimod_rumble_probe,
171 .remove = wiimod_rumble_remove,
172};
173
dcf39231
DH		 174/*
175 * Battery
176 * 1 byte of battery capacity information is sent along every protocol status
177 * report. The wiimote core caches it but we try to update it on every
178 * user-space request.
179 * This is supported by nearly every device so it's almost always enabled.
180 */
181
182static enum power_supply_property wiimod_battery_props[] = {
183 POWER_SUPPLY_PROP_CAPACITY,
184 POWER_SUPPLY_PROP_SCOPE,
185};
186
187static int wiimod_battery_get_property(struct power_supply *psy,
188 enum power_supply_property psp,
189 union power_supply_propval *val)
190{
191 struct wiimote_data *wdata = container_of(psy, struct wiimote_data,
192 battery);
193 int ret = 0, state;
194 unsigned long flags;
195
196 if (psp == POWER_SUPPLY_PROP_SCOPE) {
197 val->intval = POWER_SUPPLY_SCOPE_DEVICE;
198 return 0;
199 } else if (psp != POWER_SUPPLY_PROP_CAPACITY) {
200 return -EINVAL;
201 }
202
203 ret = wiimote_cmd_acquire(wdata);
204 if (ret)
205 return ret;
206
207 spin_lock_irqsave(&wdata->state.lock, flags);
208 wiimote_cmd_set(wdata, WIIPROTO_REQ_SREQ, 0);
209 wiiproto_req_status(wdata);
210 spin_unlock_irqrestore(&wdata->state.lock, flags);
211
212 wiimote_cmd_wait(wdata);
213 wiimote_cmd_release(wdata);
214
215 spin_lock_irqsave(&wdata->state.lock, flags);
216 state = wdata->state.cmd_battery;
217 spin_unlock_irqrestore(&wdata->state.lock, flags);
218
219 val->intval = state * 100 / 255;
220 return ret;
221}
222
223static int wiimod_battery_probe(const struct wiimod_ops *ops,
224 struct wiimote_data *wdata)
225{
226 int ret;
227
228 wdata->battery.properties = wiimod_battery_props;
229 wdata->battery.num_properties = ARRAY_SIZE(wiimod_battery_props);
230 wdata->battery.get_property = wiimod_battery_get_property;
231 wdata->battery.type = POWER_SUPPLY_TYPE_BATTERY;
232 wdata->battery.use_for_apm = 0;
233 wdata->battery.name = kasprintf(GFP_KERNEL, "wiimote_battery_%s",
234 wdata->hdev->uniq);
235 if (!wdata->battery.name)
236 return -ENOMEM;
237
238 ret = power_supply_register(&wdata->hdev->dev, &wdata->battery);
239 if (ret) {
240 hid_err(wdata->hdev, "cannot register battery device\n");
241 goto err_free;
242 }
243
244 power_supply_powers(&wdata->battery, &wdata->hdev->dev);
245 return 0;
246
247err_free:
248 kfree(wdata->battery.name);
249 wdata->battery.name = NULL;
250 return ret;
251}
252
253static void wiimod_battery_remove(const struct wiimod_ops *ops,
254 struct wiimote_data *wdata)
255{
256 if (!wdata->battery.name)
257 return;
258
259 power_supply_unregister(&wdata->battery);
260 kfree(wdata->battery.name);
261 wdata->battery.name = NULL;
262}
263
264static const struct wiimod_ops wiimod_battery = {
265 .flags = 0,
266 .arg = 0,
267 .probe = wiimod_battery_probe,
268 .remove = wiimod_battery_remove,
269};
270
6c5ae018
DH		 271/*
272 * LED
273 * 0 to 4 player LEDs are supported by devices. The "arg" field of the
274 * wiimod_ops structure specifies which LED this module controls. This allows
275 * to register a limited number of LEDs.
276 * State is managed by wiimote core.
277 */
278
279static enum led_brightness wiimod_led_get(struct led_classdev *led_dev)
280{
281 struct wiimote_data *wdata;
282 struct device *dev = led_dev->dev->parent;
283 int i;
284 unsigned long flags;
285 bool value = false;
286
287 wdata = hid_get_drvdata(container_of(dev, struct hid_device, dev));
288
289 for (i = 0; i < 4; ++i) {
290 if (wdata->leds[i] == led_dev) {
291 spin_lock_irqsave(&wdata->state.lock, flags);
292 value = wdata->state.flags & WIIPROTO_FLAG_LED(i + 1);
293 spin_unlock_irqrestore(&wdata->state.lock, flags);
294 break;
295 }
296 }
297
298 return value ? LED_FULL : LED_OFF;
299}
300
301static void wiimod_led_set(struct led_classdev *led_dev,
302 enum led_brightness value)
303{
304 struct wiimote_data *wdata;
305 struct device *dev = led_dev->dev->parent;
306 int i;
307 unsigned long flags;
308 __u8 state, flag;
309
310 wdata = hid_get_drvdata(container_of(dev, struct hid_device, dev));
311
312 for (i = 0; i < 4; ++i) {
313 if (wdata->leds[i] == led_dev) {
314 flag = WIIPROTO_FLAG_LED(i + 1);
315 spin_lock_irqsave(&wdata->state.lock, flags);
316 state = wdata->state.flags;
317 if (value == LED_OFF)
318 wiiproto_req_leds(wdata, state & ~flag);
319 else
320 wiiproto_req_leds(wdata, state | flag);
321 spin_unlock_irqrestore(&wdata->state.lock, flags);
322 break;
323 }
324 }
325}
326
327static int wiimod_led_probe(const struct wiimod_ops *ops,
328 struct wiimote_data *wdata)
329{
330 struct device *dev = &wdata->hdev->dev;
331 size_t namesz = strlen(dev_name(dev)) + 9;
332 struct led_classdev *led;
333 unsigned long flags;
334 char *name;
335 int ret;
336
337 led = kzalloc(sizeof(struct led_classdev) + namesz, GFP_KERNEL);
338 if (!led)
339 return -ENOMEM;
340
341 name = (void*)&led[1];
342 snprintf(name, namesz, "%s:blue:p%lu", dev_name(dev), ops->arg);
343 led->name = name;
344 led->brightness = 0;
345 led->max_brightness = 1;
346 led->brightness_get = wiimod_led_get;
347 led->brightness_set = wiimod_led_set;
348
349 wdata->leds[ops->arg] = led;
350 ret = led_classdev_register(dev, led);
351 if (ret)
352 goto err_free;
353
354 /* enable LED1 to stop initial LED-blinking */
355 if (ops->arg == 0) {
356 spin_lock_irqsave(&wdata->state.lock, flags);
357 wiiproto_req_leds(wdata, WIIPROTO_FLAG_LED1);
358 spin_unlock_irqrestore(&wdata->state.lock, flags);
359 }
360
361 return 0;
362
363err_free:
364 wdata->leds[ops->arg] = NULL;
365 kfree(led);
366 return ret;
367}
368
369static void wiimod_led_remove(const struct wiimod_ops *ops,
370 struct wiimote_data *wdata)
371{
372 if (!wdata->leds[ops->arg])
373 return;
374
375 led_classdev_unregister(wdata->leds[ops->arg]);
376 kfree(wdata->leds[ops->arg]);
377 wdata->leds[ops->arg] = NULL;
378}
379
380static const struct wiimod_ops wiimod_leds[4] = {
381 {
382 .flags = 0,
383 .arg = 0,
384 .probe = wiimod_led_probe,
385 .remove = wiimod_led_remove,
386 },
387 {
388 .flags = 0,
389 .arg = 1,
390 .probe = wiimod_led_probe,
391 .remove = wiimod_led_remove,
392 },
393 {
394 .flags = 0,
395 .arg = 2,
396 .probe = wiimod_led_probe,
397 .remove = wiimod_led_remove,
398 },
399 {
400 .flags = 0,
401 .arg = 3,
402 .probe = wiimod_led_probe,
403 .remove = wiimod_led_remove,
404 },
405};
406
0ea16757
DH		 407/*
408 * Accelerometer
409 * 3 axis accelerometer data is part of nearly all DRMs. If not supported by a
410 * device, it's mostly cleared to 0. This module parses this data and provides
411 * it via a separate input device.
412 */
413
414static void wiimod_accel_in_accel(struct wiimote_data *wdata,
415 const __u8 *accel)
416{
417 __u16 x, y, z;
418
419 if (!(wdata->state.flags & WIIPROTO_FLAG_ACCEL))
420 return;
421
422 /*
423 * payload is: BB BB XX YY ZZ
424 * Accelerometer data is encoded into 3 10bit values. XX, YY and ZZ
425 * contain the upper 8 bits of each value. The lower 2 bits are
426 * contained in the buttons data BB BB.
427 * Bits 6 and 7 of the first buttons byte BB is the lower 2 bits of the
428 * X accel value. Bit 5 of the second buttons byte is the 2nd bit of Y
429 * accel value and bit 6 is the second bit of the Z value.
430 * The first bit of Y and Z values is not available and always set to 0.
431 * 0x200 is returned on no movement.
432 */
433
434 x = accel[2] << 2;
435 y = accel[3] << 2;
436 z = accel[4] << 2;
437
438 x |= (accel[0] >> 5) & 0x3;
439 y |= (accel[1] >> 4) & 0x2;
440 z |= (accel[1] >> 5) & 0x2;
441
442 input_report_abs(wdata->accel, ABS_RX, x - 0x200);
443 input_report_abs(wdata->accel, ABS_RY, y - 0x200);
444 input_report_abs(wdata->accel, ABS_RZ, z - 0x200);
445 input_sync(wdata->accel);
446}
447
448static int wiimod_accel_open(struct input_dev *dev)
449{
450 struct wiimote_data *wdata = input_get_drvdata(dev);
451 unsigned long flags;
452
453 spin_lock_irqsave(&wdata->state.lock, flags);
454 wiiproto_req_accel(wdata, true);
455 spin_unlock_irqrestore(&wdata->state.lock, flags);
456
457 return 0;
458}
459
460static void wiimod_accel_close(struct input_dev *dev)
461{
462 struct wiimote_data *wdata = input_get_drvdata(dev);
463 unsigned long flags;
464
465 spin_lock_irqsave(&wdata->state.lock, flags);
466 wiiproto_req_accel(wdata, false);
467 spin_unlock_irqrestore(&wdata->state.lock, flags);
468}
469
470static int wiimod_accel_probe(const struct wiimod_ops *ops,
471 struct wiimote_data *wdata)
472{
473 int ret;
474
475 wdata->accel = input_allocate_device();
476 if (!wdata->accel)
477 return -ENOMEM;
478
479 input_set_drvdata(wdata->accel, wdata);
480 wdata->accel->open = wiimod_accel_open;
481 wdata->accel->close = wiimod_accel_close;
482 wdata->accel->dev.parent = &wdata->hdev->dev;
483 wdata->accel->id.bustype = wdata->hdev->bus;
484 wdata->accel->id.vendor = wdata->hdev->vendor;
485 wdata->accel->id.product = wdata->hdev->product;
486 wdata->accel->id.version = wdata->hdev->version;
487 wdata->accel->name = WIIMOTE_NAME " Accelerometer";
488
489 set_bit(EV_ABS, wdata->accel->evbit);
490 set_bit(ABS_RX, wdata->accel->absbit);
491 set_bit(ABS_RY, wdata->accel->absbit);
492 set_bit(ABS_RZ, wdata->accel->absbit);
493 input_set_abs_params(wdata->accel, ABS_RX, -500, 500, 2, 4);
494 input_set_abs_params(wdata->accel, ABS_RY, -500, 500, 2, 4);
495 input_set_abs_params(wdata->accel, ABS_RZ, -500, 500, 2, 4);
496
497 ret = input_register_device(wdata->accel);
498 if (ret) {
499 hid_err(wdata->hdev, "cannot register input device\n");
500 goto err_free;
501 }
502
503 return 0;
504
505err_free:
506 input_free_device(wdata->accel);
507 wdata->accel = NULL;
508 return ret;
509}
510
511static void wiimod_accel_remove(const struct wiimod_ops *ops,
512 struct wiimote_data *wdata)
513{
514 if (!wdata->accel)
515 return;
516
517 input_unregister_device(wdata->accel);
518 wdata->accel = NULL;
519}
520
521static const struct wiimod_ops wiimod_accel = {
522 .flags = 0,
523 .arg = 0,
524 .probe = wiimod_accel_probe,
525 .remove = wiimod_accel_remove,
526 .in_accel = wiimod_accel_in_accel,
527};
528
3b5f03c4
DH		 529/*
530 * IR Cam
531 * Up to 4 IR sources can be tracked by a normal Wii Remote. The IR cam needs
532 * to be initialized with a fairly complex procedure and consumes a lot of
533 * power. Therefore, as long as no application uses the IR input device, it is
534 * kept offline.
535 * Nearly no other device than the normal Wii Remotes supports the IR cam so
536 * you can disable this module for these devices.
537 */
538
539static void wiimod_ir_in_ir(struct wiimote_data *wdata, const __u8 *ir,
540 bool packed, unsigned int id)
541{
542 __u16 x, y;
543 __u8 xid, yid;
544 bool sync = false;
545
546 if (!(wdata->state.flags & WIIPROTO_FLAGS_IR))
547 return;
548
549 switch (id) {
550 case 0:
551 xid = ABS_HAT0X;
552 yid = ABS_HAT0Y;
553 break;
554 case 1:
555 xid = ABS_HAT1X;
556 yid = ABS_HAT1Y;
557 break;
558 case 2:
559 xid = ABS_HAT2X;
560 yid = ABS_HAT2Y;
561 break;
562 case 3:
563 xid = ABS_HAT3X;
564 yid = ABS_HAT3Y;
565 sync = true;
566 break;
567 default:
568 return;
569 };
570
571 /*
572 * Basic IR data is encoded into 3 bytes. The first two bytes are the
573 * lower 8 bit of the X/Y data, the 3rd byte contains the upper 2 bits
574 * of both.
575 * If data is packed, then the 3rd byte is put first and slightly
576 * reordered. This allows to interleave packed and non-packed data to
577 * have two IR sets in 5 bytes instead of 6.
578 * The resulting 10bit X/Y values are passed to the ABS_HAT? input dev.
579 */
580
581 if (packed) {
582 x = ir[1] | ((ir[0] & 0x03) << 8);
583 y = ir[2] | ((ir[0] & 0x0c) << 6);
584 } else {
585 x = ir[0] | ((ir[2] & 0x30) << 4);
586 y = ir[1] | ((ir[2] & 0xc0) << 2);
587 }
588
589 input_report_abs(wdata->ir, xid, x);
590 input_report_abs(wdata->ir, yid, y);
591
592 if (sync)
593 input_sync(wdata->ir);
594}
595
596static int wiimod_ir_change(struct wiimote_data *wdata, __u16 mode)
597{
598 int ret;
599 unsigned long flags;
600 __u8 format = 0;
601 static const __u8 data_enable[] = { 0x01 };
602 static const __u8 data_sens1[] = { 0x02, 0x00, 0x00, 0x71, 0x01,
603 0x00, 0xaa, 0x00, 0x64 };
604 static const __u8 data_sens2[] = { 0x63, 0x03 };
605 static const __u8 data_fin[] = { 0x08 };
606
607 spin_lock_irqsave(&wdata->state.lock, flags);
608
609 if (mode == (wdata->state.flags & WIIPROTO_FLAGS_IR)) {
610 spin_unlock_irqrestore(&wdata->state.lock, flags);
611 return 0;
612 }
613
614 if (mode == 0) {
615 wdata->state.flags &= ~WIIPROTO_FLAGS_IR;
616 wiiproto_req_ir1(wdata, 0);
617 wiiproto_req_ir2(wdata, 0);
618 wiiproto_req_drm(wdata, WIIPROTO_REQ_NULL);
619 spin_unlock_irqrestore(&wdata->state.lock, flags);
620 return 0;
621 }
622
623 spin_unlock_irqrestore(&wdata->state.lock, flags);
624
625 ret = wiimote_cmd_acquire(wdata);
626 if (ret)
627 return ret;
628
629 /* send PIXEL CLOCK ENABLE cmd first */
630 spin_lock_irqsave(&wdata->state.lock, flags);
631 wiimote_cmd_set(wdata, WIIPROTO_REQ_IR1, 0);
632 wiiproto_req_ir1(wdata, 0x06);
633 spin_unlock_irqrestore(&wdata->state.lock, flags);
634
635 ret = wiimote_cmd_wait(wdata);
636 if (ret)
637 goto unlock;
638 if (wdata->state.cmd_err) {
639 ret = -EIO;
640 goto unlock;
641 }
642
643 /* enable IR LOGIC */
644 spin_lock_irqsave(&wdata->state.lock, flags);
645 wiimote_cmd_set(wdata, WIIPROTO_REQ_IR2, 0);
646 wiiproto_req_ir2(wdata, 0x06);
647 spin_unlock_irqrestore(&wdata->state.lock, flags);
648
649 ret = wiimote_cmd_wait(wdata);
650 if (ret)
651 goto unlock;
652 if (wdata->state.cmd_err) {
653 ret = -EIO;
654 goto unlock;
655 }
656
657 /* enable IR cam but do not make it send data, yet */
658 ret = wiimote_cmd_write(wdata, 0xb00030, data_enable,
659 sizeof(data_enable));
660 if (ret)
661 goto unlock;
662
663 /* write first sensitivity block */
664 ret = wiimote_cmd_write(wdata, 0xb00000, data_sens1,
665 sizeof(data_sens1));
666 if (ret)
667 goto unlock;
668
669 /* write second sensitivity block */
670 ret = wiimote_cmd_write(wdata, 0xb0001a, data_sens2,
671 sizeof(data_sens2));
672 if (ret)
673 goto unlock;
674
675 /* put IR cam into desired state */
676 switch (mode) {
677 case WIIPROTO_FLAG_IR_FULL:
678 format = 5;
679 break;
680 case WIIPROTO_FLAG_IR_EXT:
681 format = 3;
682 break;
683 case WIIPROTO_FLAG_IR_BASIC:
684 format = 1;
685 break;
686 }
687 ret = wiimote_cmd_write(wdata, 0xb00033, &format, sizeof(format));
688 if (ret)
689 goto unlock;
690
691 /* make IR cam send data */
692 ret = wiimote_cmd_write(wdata, 0xb00030, data_fin, sizeof(data_fin));
693 if (ret)
694 goto unlock;
695
696 /* request new DRM mode compatible to IR mode */
697 spin_lock_irqsave(&wdata->state.lock, flags);
698 wdata->state.flags &= ~WIIPROTO_FLAGS_IR;
699 wdata->state.flags |= mode & WIIPROTO_FLAGS_IR;
700 wiiproto_req_drm(wdata, WIIPROTO_REQ_NULL);
701 spin_unlock_irqrestore(&wdata->state.lock, flags);
702
703unlock:
704 wiimote_cmd_release(wdata);
705 return ret;
706}
707
708static int wiimod_ir_open(struct input_dev *dev)
709{
710 struct wiimote_data *wdata = input_get_drvdata(dev);
711
712 return wiimod_ir_change(wdata, WIIPROTO_FLAG_IR_BASIC);
713}
714
715static void wiimod_ir_close(struct input_dev *dev)
716{
717 struct wiimote_data *wdata = input_get_drvdata(dev);
718
719 wiimod_ir_change(wdata, 0);
720}
721
722static int wiimod_ir_probe(const struct wiimod_ops *ops,
723 struct wiimote_data *wdata)
724{
725 int ret;
726
727 wdata->ir = input_allocate_device();
728 if (!wdata->ir)
729 return -ENOMEM;
730
731 input_set_drvdata(wdata->ir, wdata);
732 wdata->ir->open = wiimod_ir_open;
733 wdata->ir->close = wiimod_ir_close;
734 wdata->ir->dev.parent = &wdata->hdev->dev;
735 wdata->ir->id.bustype = wdata->hdev->bus;
736 wdata->ir->id.vendor = wdata->hdev->vendor;
737 wdata->ir->id.product = wdata->hdev->product;
738 wdata->ir->id.version = wdata->hdev->version;
739 wdata->ir->name = WIIMOTE_NAME " IR";
740
741 set_bit(EV_ABS, wdata->ir->evbit);
742 set_bit(ABS_HAT0X, wdata->ir->absbit);
743 set_bit(ABS_HAT0Y, wdata->ir->absbit);
744 set_bit(ABS_HAT1X, wdata->ir->absbit);
745 set_bit(ABS_HAT1Y, wdata->ir->absbit);
746 set_bit(ABS_HAT2X, wdata->ir->absbit);
747 set_bit(ABS_HAT2Y, wdata->ir->absbit);
748 set_bit(ABS_HAT3X, wdata->ir->absbit);
749 set_bit(ABS_HAT3Y, wdata->ir->absbit);
750 input_set_abs_params(wdata->ir, ABS_HAT0X, 0, 1023, 2, 4);
751 input_set_abs_params(wdata->ir, ABS_HAT0Y, 0, 767, 2, 4);
752 input_set_abs_params(wdata->ir, ABS_HAT1X, 0, 1023, 2, 4);
753 input_set_abs_params(wdata->ir, ABS_HAT1Y, 0, 767, 2, 4);
754 input_set_abs_params(wdata->ir, ABS_HAT2X, 0, 1023, 2, 4);
755 input_set_abs_params(wdata->ir, ABS_HAT2Y, 0, 767, 2, 4);
756 input_set_abs_params(wdata->ir, ABS_HAT3X, 0, 1023, 2, 4);
757 input_set_abs_params(wdata->ir, ABS_HAT3Y, 0, 767, 2, 4);
758
759 ret = input_register_device(wdata->ir);
760 if (ret) {
761 hid_err(wdata->hdev, "cannot register input device\n");
762 goto err_free;
763 }
764
765 return 0;
766
767err_free:
768 input_free_device(wdata->ir);
769 wdata->ir = NULL;
770 return ret;
771}
772
773static void wiimod_ir_remove(const struct wiimod_ops *ops,
774 struct wiimote_data *wdata)
775{
776 if (!wdata->ir)
777 return;
778
779 input_unregister_device(wdata->ir);
780 wdata->ir = NULL;
781}
782
783static const struct wiimod_ops wiimod_ir = {
784 .flags = 0,
785 .arg = 0,
786 .probe = wiimod_ir_probe,
787 .remove = wiimod_ir_remove,
788 .in_ir = wiimod_ir_in_ir,
789};
790
f1d4bed4
DH		 791/*
792 * Balance Board Extension
793 * The Nintendo Wii Balance Board provides four hardware weight sensor plus a
794 * single push button. No other peripherals are available. However, the
795 * balance-board data is sent via a standard Wii Remote extension. All other
796 * data for non-present hardware is zeroed out.
797 * Some 3rd party devices react allergic if we try to access normal Wii Remote
798 * hardware, so this extension module should be the only module that is loaded
799 * on balance boards.
800 * The balance board needs 8 bytes extension data instead of basic 6 bytes so
801 * it needs the WIIMOD_FLAG_EXT8 flag.
802 */
803
804static void wiimod_bboard_in_keys(struct wiimote_data *wdata, const __u8 *keys)
805{
806 input_report_key(wdata->extension.input, BTN_A,
807 !!(keys[1] & 0x08));
808 input_sync(wdata->extension.input);
809}
810
811static void wiimod_bboard_in_ext(struct wiimote_data *wdata,
812 const __u8 *ext)
813{
814 __s32 val[4], tmp, div;
815 unsigned int i;
816 struct wiimote_state *s = &wdata->state;
817
818 /*
819 * Balance board data layout:
820 *
821 * Byte | 8 7 6 5 4 3 2 1 |
822 * -----+--------------------------+
823 * 1 | Top Right <15:8> |
824 * 2 | Top Right <7:0> |
825 * -----+--------------------------+
826 * 3 | Bottom Right <15:8> |
827 * 4 | Bottom Right <7:0> |
828 * -----+--------------------------+
829 * 5 | Top Left <15:8> |
830 * 6 | Top Left <7:0> |
831 * -----+--------------------------+
832 * 7 | Bottom Left <15:8> |
833 * 8 | Bottom Left <7:0> |
834 * -----+--------------------------+
835 *
836 * These values represent the weight-measurements of the Wii-balance
837 * board with 16bit precision.
838 *
839 * The balance-board is never reported interleaved with motionp.
840 */
841
842 val[0] = ext[0];
843 val[0] <<= 8;
844 val[0] |= ext[1];
845
846 val[1] = ext[2];
847 val[1] <<= 8;
848 val[1] |= ext[3];
849
850 val[2] = ext[4];
851 val[2] <<= 8;
852 val[2] |= ext[5];
853
854 val[3] = ext[6];
855 val[3] <<= 8;
856 val[3] |= ext[7];
857
858 /* apply calibration data */
859 for (i = 0; i < 4; i++) {
860 if (val[i] <= s->calib_bboard[i][0]) {
861 tmp = 0;
862 } else if (val[i] < s->calib_bboard[i][1]) {
863 tmp = val[i] - s->calib_bboard[i][0];
864 tmp *= 1700;
865 div = s->calib_bboard[i][1] - s->calib_bboard[i][0];
866 tmp /= div ? div : 1;
867 } else {
868 tmp = val[i] - s->calib_bboard[i][1];
869 tmp *= 1700;
870 div = s->calib_bboard[i][2] - s->calib_bboard[i][1];
871 tmp /= div ? div : 1;
872 tmp += 1700;
873 }
874 val[i] = tmp;
875 }
876
877 input_report_abs(wdata->extension.input, ABS_HAT0X, val[0]);
878 input_report_abs(wdata->extension.input, ABS_HAT0Y, val[1]);
879 input_report_abs(wdata->extension.input, ABS_HAT1X, val[2]);
880 input_report_abs(wdata->extension.input, ABS_HAT1Y, val[3]);
881 input_sync(wdata->extension.input);
882}
883
884static int wiimod_bboard_open(struct input_dev *dev)
885{
886 struct wiimote_data *wdata = input_get_drvdata(dev);
887 unsigned long flags;
888
889 spin_lock_irqsave(&wdata->state.lock, flags);
890 wdata->state.flags |= WIIPROTO_FLAG_EXT_USED;
891 wiiproto_req_drm(wdata, WIIPROTO_REQ_NULL);
892 spin_unlock_irqrestore(&wdata->state.lock, flags);
893
894 return 0;
895}
896
897static void wiimod_bboard_close(struct input_dev *dev)
898{
899 struct wiimote_data *wdata = input_get_drvdata(dev);
900 unsigned long flags;
901
902 spin_lock_irqsave(&wdata->state.lock, flags);
903 wdata->state.flags &= ~WIIPROTO_FLAG_EXT_USED;
904 wiiproto_req_drm(wdata, WIIPROTO_REQ_NULL);
905 spin_unlock_irqrestore(&wdata->state.lock, flags);
906}
907
908static int wiimod_bboard_probe(const struct wiimod_ops *ops,
909 struct wiimote_data *wdata)
910{
911 int ret, i, j;
912 __u8 buf[24], offs;
913
914 wiimote_cmd_acquire_noint(wdata);
915
916 ret = wiimote_cmd_read(wdata, 0xa40024, buf, 12);
917 if (ret != 12) {
918 wiimote_cmd_release(wdata);
919 return ret < 0 ? ret : -EIO;
920 }
921 ret = wiimote_cmd_read(wdata, 0xa40024 + 12, buf + 12, 12);
922 if (ret != 12) {
923 wiimote_cmd_release(wdata);
924 return ret < 0 ? ret : -EIO;
925 }
926
927 wiimote_cmd_release(wdata);
928
929 offs = 0;
930 for (i = 0; i < 3; ++i) {
931 for (j = 0; j < 4; ++j) {
932 wdata->state.calib_bboard[j][i] = buf[offs];
933 wdata->state.calib_bboard[j][i] <<= 8;
934 wdata->state.calib_bboard[j][i] |= buf[offs + 1];
935 offs += 2;
936 }
937 }
938
939 wdata->extension.input = input_allocate_device();
940 if (!wdata->extension.input)
941 return -ENOMEM;
942
943 input_set_drvdata(wdata->extension.input, wdata);
944 wdata->extension.input->open = wiimod_bboard_open;
945 wdata->extension.input->close = wiimod_bboard_close;
946 wdata->extension.input->dev.parent = &wdata->hdev->dev;
947 wdata->extension.input->id.bustype = wdata->hdev->bus;
948 wdata->extension.input->id.vendor = wdata->hdev->vendor;
949 wdata->extension.input->id.product = wdata->hdev->product;
950 wdata->extension.input->id.version = wdata->hdev->version;
951 wdata->extension.input->name = WIIMOTE_NAME " Balance Board";
952
953 set_bit(EV_KEY, wdata->extension.input->evbit);
954 set_bit(BTN_A, wdata->extension.input->keybit);
955
956 set_bit(EV_ABS, wdata->extension.input->evbit);
957 set_bit(ABS_HAT0X, wdata->extension.input->absbit);
958 set_bit(ABS_HAT0Y, wdata->extension.input->absbit);
959 set_bit(ABS_HAT1X, wdata->extension.input->absbit);
960 set_bit(ABS_HAT1Y, wdata->extension.input->absbit);
961 input_set_abs_params(wdata->extension.input,
962 ABS_HAT0X, 0, 65535, 2, 4);
963 input_set_abs_params(wdata->extension.input,
964 ABS_HAT0Y, 0, 65535, 2, 4);
965 input_set_abs_params(wdata->extension.input,
966 ABS_HAT1X, 0, 65535, 2, 4);
967 input_set_abs_params(wdata->extension.input,
968 ABS_HAT1Y, 0, 65535, 2, 4);
969
970 ret = input_register_device(wdata->extension.input);
971 if (ret)
972 goto err_free;
973
974 return 0;
975
976err_free:
977 input_free_device(wdata->extension.input);
978 wdata->extension.input = NULL;
979 return ret;
980}
981
982static void wiimod_bboard_remove(const struct wiimod_ops *ops,
983 struct wiimote_data *wdata)
984{
985 if (!wdata->extension.input)
986 return;
987
988 input_unregister_device(wdata->extension.input);
989 wdata->extension.input = NULL;
990}
991
992static const struct wiimod_ops wiimod_bboard = {
993 .flags = WIIMOD_FLAG_EXT8,
994 .arg = 0,
995 .probe = wiimod_bboard_probe,
996 .remove = wiimod_bboard_remove,
997 .in_keys = wiimod_bboard_in_keys,
998 .in_ext = wiimod_bboard_in_ext,
999};
1000
4148b6bf
DH		 1001/*
1002 * Motion Plus
1003 */
1004
1005const struct wiimod_ops wiimod_mp = {
1006 .flags = 0,
1007 .arg = 0,
1008};
1009
27f06942
DH		 1010/* module table */
1011
4148b6bf
DH		 1012static const struct wiimod_ops wiimod_dummy;
1013
27f06942 1014const struct wiimod_ops *wiimod_table[WIIMOD_NUM] = {
20cef813
DH		 1015 [WIIMOD_KEYS] = &wiimod_keys,
1016 [WIIMOD_RUMBLE] = &wiimod_rumble,
dcf39231 1017 [WIIMOD_BATTERY] = &wiimod_battery,
6c5ae018
DH		 1018 [WIIMOD_LED1] = &wiimod_leds[0],
1019 [WIIMOD_LED2] = &wiimod_leds[1],
1020 [WIIMOD_LED3] = &wiimod_leds[2],
1021 [WIIMOD_LED4] = &wiimod_leds[3],
0ea16757 1022 [WIIMOD_ACCEL] = &wiimod_accel,
3b5f03c4 1023 [WIIMOD_IR] = &wiimod_ir,
27f06942 1024};
4148b6bf
DH		 1025
1026const struct wiimod_ops *wiimod_ext_table[WIIMOTE_EXT_NUM] = {
1027 [WIIMOTE_EXT_NONE] = &wiimod_dummy,
1028 [WIIMOTE_EXT_UNKNOWN] = &wiimod_dummy,
f1d4bed4 1029 [WIIMOTE_EXT_BALANCE_BOARD] = &wiimod_bboard,
4148b6bf 1030};
Linux 6.x block layer and io_uring tree(s)