Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit f363e4f6 authored by David Herrmann's avatar David Herrmann Committed by Jiri Kosina
Browse files

HID: wiimote: Add IR input device 



The IR cam of the wiimote reports 4 trackable lights as absolute values. Since
we can turn the IR cam on and off, we register a separate input device so we can
react on open/close callbacks to save wiimote battery power when IR cam is not
needed.

The cam can be in four states: off, basic, extended and full
The DRM chooser automatically selects a proper DRM that includes all required IR
data so no information is lost.

Signed-off-by: default avatarDavid Herrmann <dh.herrmann@googlemail.com>
Signed-off-by: default avatarJiri Kosina <jkosina@suse.cz>
parent efcf9188

drivers/hid/hid-wiimote.c

+67 −4
Original line number Diff line number Diff line
@@ -38,6 +38,7 @@ struct wiimote_data { 
	struct input_dev *input;

	struct led_classdev *leds[4];

	struct input_dev *accel;

	struct input_dev *ir;



	spinlock_t qlock;

	__u8 head;
@@ -54,8 +55,13 @@ struct wiimote_data { 
#define WIIPROTO_FLAG_LED4		0x08

#define WIIPROTO_FLAG_RUMBLE		0x10

#define WIIPROTO_FLAG_ACCEL		0x20

#define WIIPROTO_FLAG_IR_BASIC		0x40

#define WIIPROTO_FLAG_IR_EXT		0x80

#define WIIPROTO_FLAG_IR_FULL		0xc0 /* IR_BASIC | IR_EXT */

#define WIIPROTO_FLAGS_LEDS (WIIPROTO_FLAG_LED1 | WIIPROTO_FLAG_LED2 | \

					WIIPROTO_FLAG_LED3 | WIIPROTO_FLAG_LED4)

#define WIIPROTO_FLAGS_IR (WIIPROTO_FLAG_IR_BASIC | WIIPROTO_FLAG_IR_EXT | \

							WIIPROTO_FLAG_IR_FULL)



/* return flag for led \num */

#define WIIPROTO_FLAG_LED(num) (WIIPROTO_FLAG_LED1 << (num - 1))
@@ -71,6 +77,7 @@ enum wiiproto_reqs { 
	WIIPROTO_REQ_DRM_KA = 0x31,

	WIIPROTO_REQ_DRM_KAI = 0x33,

	WIIPROTO_REQ_DRM_KAE = 0x35,

	WIIPROTO_REQ_DRM_KIE = 0x36,

	WIIPROTO_REQ_DRM_KAIE = 0x37,

	WIIPROTO_REQ_DRM_SKAI1 = 0x3e,

	WIIPROTO_REQ_DRM_SKAI2 = 0x3f,
@@ -248,11 +255,24 @@ static void wiiproto_req_leds(struct wiimote_data *wdata, int leds) 
 */

static __u8 select_drm(struct wiimote_data *wdata)

{

	__u8 ir = wdata->state.flags & WIIPROTO_FLAGS_IR;



	if (ir == WIIPROTO_FLAG_IR_BASIC) {

		if (wdata->state.flags & WIIPROTO_FLAG_ACCEL)

			return WIIPROTO_REQ_DRM_KAIE;

		else

			return WIIPROTO_REQ_DRM_KIE;

	} else if (ir == WIIPROTO_FLAG_IR_EXT) {

		return WIIPROTO_REQ_DRM_KAI;

	} else if (ir == WIIPROTO_FLAG_IR_FULL) {

		return WIIPROTO_REQ_DRM_SKAI1;

	} else {

		if (wdata->state.flags & WIIPROTO_FLAG_ACCEL)

			return WIIPROTO_REQ_DRM_KA;

		else

			return WIIPROTO_REQ_DRM_K;

	}

}



static void wiiproto_req_drm(struct wiimote_data *wdata, __u8 drm)

{
@@ -681,6 +701,36 @@ static struct wiimote_data *wiimote_create(struct hid_device *hdev) 
	input_set_abs_params(wdata->accel, ABS_RY, -500, 500, 2, 4);

	input_set_abs_params(wdata->accel, ABS_RZ, -500, 500, 2, 4);



	wdata->ir = input_allocate_device();

	if (!wdata->ir)

		goto err_ir;



	input_set_drvdata(wdata->ir, wdata);

	wdata->ir->dev.parent = &wdata->hdev->dev;

	wdata->ir->id.bustype = wdata->hdev->bus;

	wdata->ir->id.vendor = wdata->hdev->vendor;

	wdata->ir->id.product = wdata->hdev->product;

	wdata->ir->id.version = wdata->hdev->version;

	wdata->ir->name = WIIMOTE_NAME " IR";



	set_bit(EV_ABS, wdata->ir->evbit);

	set_bit(ABS_HAT0X, wdata->ir->absbit);

	set_bit(ABS_HAT0Y, wdata->ir->absbit);

	set_bit(ABS_HAT1X, wdata->ir->absbit);

	set_bit(ABS_HAT1Y, wdata->ir->absbit);

	set_bit(ABS_HAT2X, wdata->ir->absbit);

	set_bit(ABS_HAT2Y, wdata->ir->absbit);

	set_bit(ABS_HAT3X, wdata->ir->absbit);

	set_bit(ABS_HAT3Y, wdata->ir->absbit);

	input_set_abs_params(wdata->ir, ABS_HAT0X, 0, 1023, 2, 4);

	input_set_abs_params(wdata->ir, ABS_HAT0Y, 0, 767, 2, 4);

	input_set_abs_params(wdata->ir, ABS_HAT1X, 0, 1023, 2, 4);

	input_set_abs_params(wdata->ir, ABS_HAT1Y, 0, 767, 2, 4);

	input_set_abs_params(wdata->ir, ABS_HAT2X, 0, 1023, 2, 4);

	input_set_abs_params(wdata->ir, ABS_HAT2Y, 0, 767, 2, 4);

	input_set_abs_params(wdata->ir, ABS_HAT3X, 0, 1023, 2, 4);

	input_set_abs_params(wdata->ir, ABS_HAT3Y, 0, 767, 2, 4);



	spin_lock_init(&wdata->qlock);

	INIT_WORK(&wdata->worker, wiimote_worker);
@@ -688,6 +738,8 @@ static struct wiimote_data *wiimote_create(struct hid_device *hdev) 


	return wdata;



err_ir:

	input_free_device(wdata->accel);

err_input:

	input_free_device(wdata->input);

err:
@@ -700,6 +752,7 @@ static void wiimote_destroy(struct wiimote_data *wdata) 
	wiimote_leds_destroy(wdata);



	input_unregister_device(wdata->accel);

	input_unregister_device(wdata->ir);

	input_unregister_device(wdata->input);

	cancel_work_sync(&wdata->worker);

	hid_hw_stop(wdata->hdev);
@@ -737,6 +790,12 @@ static int wiimote_hid_probe(struct hid_device *hdev, 
		goto err_stop;

	}



	ret = input_register_device(wdata->ir);

	if (ret) {

		hid_err(hdev, "Cannot register input device\n");

		goto err_ir;

	}



	ret = input_register_device(wdata->input);

	if (ret) {

		hid_err(hdev, "Cannot register input device\n");
@@ -761,11 +820,15 @@ err_free: 
	return ret;



err_input:

	input_unregister_device(wdata->ir);

	wdata->ir = NULL;

err_ir:

	input_unregister_device(wdata->accel);

	wdata->accel = NULL;

err_stop:

	hid_hw_stop(hdev);

err:

	input_free_device(wdata->ir);

	input_free_device(wdata->accel);

	input_free_device(wdata->input);

	kfree(wdata);