HID: wiimote: Support rumble device

This adds support for the wiimote's rumble device. Every output report can
enable and disable the rumble motor. Hence, every output report must look up our
new RUMBLE flag and make sure that it does not unintentionally toggle the
rumble motor.

Signed-off-by: David Herrmann <dh.herrmann@googlemail.com>
Signed-off-by: Jiri Kosina <jkosina@suse.cz>

1 files changed, 41 insertions, 4 deletions

diff --git a/drivers/hid/hid-wiimote.c b/drivers/hid/hid-wiimote.c
index 85a02e5f9fe8..680975436289 100644
--- a/drivers/hid/hid-wiimote.c
+++ b/drivers/hid/hid-wiimote.c
@@ -46,10 +46,11 @@ struct wiimote_data {
         struct wiimote_state state;
 };
 
-#define WIIPROTO_FLAG_LED1 0x01
-#define WIIPROTO_FLAG_LED2 0x02
-#define WIIPROTO_FLAG_LED3 0x04
-#define WIIPROTO_FLAG_LED4 0x08
+#define WIIPROTO_FLAG_LED1              0x01
+#define WIIPROTO_FLAG_LED2              0x02
+#define WIIPROTO_FLAG_LED3              0x04
+#define WIIPROTO_FLAG_LED4              0x08
+#define WIIPROTO_FLAG_RUMBLE            0x10
 #define WIIPROTO_FLAGS_LEDS (WIIPROTO_FLAG_LED1 | WIIPROTO_FLAG_LED2 | \
                                         WIIPROTO_FLAG_LED3 | WIIPROTO_FLAG_LED4)
 
@@ -58,6 +59,7 @@ struct wiimote_data {
 
 enum wiiproto_reqs {
         WIIPROTO_REQ_NULL = 0x0,
+        WIIPROTO_REQ_RUMBLE = 0x10,
         WIIPROTO_REQ_LED = 0x11,
         WIIPROTO_REQ_DRM = 0x12,
         WIIPROTO_REQ_STATUS = 0x20,
@@ -172,6 +174,39 @@ static void wiimote_queue(struct wiimote_data *wdata, const __u8 *buffer,
         spin_unlock_irqrestore(&wdata->qlock, flags);
 }
 
+/*
+ * This sets the rumble bit on the given output report if rumble is
+ * currently enabled.
+ * \cmd1 must point to the second byte in the output report => &cmd[1]
+ * This must be called on nearly every output report before passing it
+ * into the output queue!
+ */
+static inline void wiiproto_keep_rumble(struct wiimote_data *wdata, __u8 *cmd1)
+{
+        if (wdata->state.flags & WIIPROTO_FLAG_RUMBLE)
+                *cmd1 |= 0x01;
+}
+
+static void wiiproto_req_rumble(struct wiimote_data *wdata, __u8 rumble)
+{
+        __u8 cmd[2];
+
+        rumble = !!rumble;
+        if (rumble == !!(wdata->state.flags & WIIPROTO_FLAG_RUMBLE))
+                return;
+
+        if (rumble)
+                wdata->state.flags |= WIIPROTO_FLAG_RUMBLE;
+        else
+                wdata->state.flags &= ~WIIPROTO_FLAG_RUMBLE;
+
+        cmd[0] = WIIPROTO_REQ_RUMBLE;
+        cmd[1] = 0;
+
+        wiiproto_keep_rumble(wdata, &cmd[1]);
+        wiimote_queue(wdata, cmd, sizeof(cmd));
+}
+
 static void wiiproto_req_leds(struct wiimote_data *wdata, int leds)
 {
         __u8 cmd[2];
@@ -193,6 +228,7 @@ static void wiiproto_req_leds(struct wiimote_data *wdata, int leds)
         if (leds & WIIPROTO_FLAG_LED4)
                 cmd[1] |= 0x80;
 
+        wiiproto_keep_rumble(wdata, &cmd[1]);
         wiimote_queue(wdata, cmd, sizeof(cmd));
 }
 
@@ -217,6 +253,7 @@ static void wiiproto_req_drm(struct wiimote_data *wdata, __u8 drm)
         cmd[1] = 0;
         cmd[2] = drm;
 
+        wiiproto_keep_rumble(wdata, &cmd[1]);
         wiimote_queue(wdata, cmd, sizeof(cmd));
 }