/*
   *  Bluetooth Wacom Tablet support
   *
   *  Copyright (c) 1999 Andreas Gal
   *  Copyright (c) 2000-2005 Vojtech Pavlik <vojtech@suse.cz>
   *  Copyright (c) 2005 Michael Haboustak <mike-@cinci.rr.com> for Concept2, Inc
   *  Copyright (c) 2006-2007 Jiri Kosina
   *  Copyright (c) 2008 Jiri Slaby <jirislaby@gmail.com>
   *  Copyright (c) 2006 Andrew Zabolotny <zap@homelink.ru>
   *  Copyright (c) 2009 Bastien Nocera <hadess@hadess.net>
   *  Copyright (c) 2011 Przemysław Firszt <przemo@firszt.eu>
   */
  
  /*
   * This program is free software; you can redistribute it and/or modify it
   * under the terms of the GNU General Public License as published by the Free
   * Software Foundation; either version 2 of the License, or (at your option)
   * any later version.
   */
  
  #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
  
  #include <linux/device.h>
  #include <linux/hid.h>
  #include <linux/module.h>
  #include <linux/leds.h>
  #include <linux/slab.h>
  #include <linux/power_supply.h>
  
  #include "hid-ids.h"
  
  #define PAD_DEVICE_ID	0x0F
  
  #define WAC_CMD_LED_CONTROL     0x20
  #define WAC_CMD_ICON_START_STOP     0x21
  #define WAC_CMD_ICON_TRANSFER       0x26
  
  struct wacom_data {
  	__u16 tool;
  	__u16 butstate;
  	__u8 whlstate;
  	__u8 features;
  	__u32 id;
  	__u32 serial;
  	unsigned char high_speed;
  	__u8 battery_capacity;
  	__u8 power_raw;
  	__u8 ps_connected;
  	__u8 bat_charging;
  	struct power_supply battery;
  	struct power_supply ac;
  	__u8 led_selector;
  	struct led_classdev *leds[4];
  };
  
  /*percent of battery capacity for Graphire
    8th value means AC online and show 100% capacity */
  static unsigned short batcap_gr[8] = { 1, 15, 25, 35, 50, 70, 100, 100 };
  /*percent of battery capacity for Intuos4 WL, AC has a separate bit*/
  static unsigned short batcap_i4[8] = { 1, 15, 30, 45, 60, 70, 85, 100 };
  
  static enum power_supply_property wacom_battery_props[] = {
  	POWER_SUPPLY_PROP_PRESENT,
  	POWER_SUPPLY_PROP_CAPACITY,
  	POWER_SUPPLY_PROP_SCOPE,
  	POWER_SUPPLY_PROP_STATUS,
  };
  
  static enum power_supply_property wacom_ac_props[] = {
  	POWER_SUPPLY_PROP_PRESENT,
  	POWER_SUPPLY_PROP_ONLINE,
  	POWER_SUPPLY_PROP_SCOPE,
  };
  
  static void wacom_scramble(__u8 *image)
  {
  	__u16 mask;
  	__u16 s1;
  	__u16 s2;
  	__u16 r1 ;
  	__u16 r2 ;
  	__u16 r;
  	__u8 buf[256];
  	int i, w, x, y, z;
  
  	for (x = 0; x < 32; x++) {
  		for (y = 0; y < 8; y++)
  			buf[(8 * x) + (7 - y)] = image[(8 * x) + y];
  	}
  
  	/* Change 76543210 into GECA6420 as required by Intuos4 WL
  	 *        HGFEDCBA      HFDB7531
  	 */
  	for (x = 0; x < 4; x++) {
  		for (y = 0; y < 4; y++) {
  			for (z = 0; z < 8; z++) {
  				mask = 0x0001;
  				r1 = 0;
  				r2 = 0;
  				i = (x << 6) + (y << 4) + z;
  				s1 = buf[i];
  				s2 = buf[i+8];
  				for (w = 0; w < 8; w++) {
  					r1 |= (s1 & mask);
  					r2 |= (s2 & mask);
  					s1 <<= 1;
  					s2 <<= 1;
  					mask <<= 2;
  				}
  				r = r1 | (r2 << 1);
  				i = (x << 6) + (y << 4) + (z << 1);
  				image[i] = 0xFF & r;
  				image[i+1] = (0xFF00 & r) >> 8;
  			}
  		}
  	}
  }
  
  static void wacom_set_image(struct hid_device *hdev, const char *image,
  						__u8 icon_no)
  {
  	__u8 rep_data[68];
  	__u8 p[256];
  	int ret, i, j;
  
  	for (i = 0; i < 256; i++)
  		p[i] = image[i];
  
  	rep_data[0] = WAC_CMD_ICON_START_STOP;
  	rep_data[1] = 0;
  	ret = hdev->hid_output_raw_report(hdev, rep_data, 2,
  				HID_FEATURE_REPORT);
  	if (ret < 0)
  		goto err;
  
  	rep_data[0] = WAC_CMD_ICON_TRANSFER;
  	rep_data[1] = icon_no & 0x07;
  
  	wacom_scramble(p);
  
  	for (i = 0; i < 4; i++) {
  		for (j = 0; j < 64; j++)
  			rep_data[j + 3] = p[(i << 6) + j];
  
  		rep_data[2] = i;
  		ret = hdev->hid_output_raw_report(hdev, rep_data, 67,
  					HID_FEATURE_REPORT);
  	}
  
  	rep_data[0] = WAC_CMD_ICON_START_STOP;
  	rep_data[1] = 0;
  
  	ret = hdev->hid_output_raw_report(hdev, rep_data, 2,
  				HID_FEATURE_REPORT);
  
  err:
  	return;
  }
  
  static void wacom_leds_set_brightness(struct led_classdev *led_dev,
  						enum led_brightness value)
  {
  	struct device *dev = led_dev->dev->parent;
  	struct hid_device *hdev;
  	struct wacom_data *wdata;
  	unsigned char *buf;
  	__u8 led = 0;
  	int i;
  
  	hdev = container_of(dev, struct hid_device, dev);
  	wdata = hid_get_drvdata(hdev);
  	for (i = 0; i < 4; ++i) {
  		if (wdata->leds[i] == led_dev)
  			wdata->led_selector = i;
  	}
  
  	led = wdata->led_selector | 0x04;
  	buf = kzalloc(9, GFP_KERNEL);
  	if (buf) {
  		buf[0] = WAC_CMD_LED_CONTROL;
  		buf[1] = led;
  		buf[2] = value >> 2;
  		buf[3] = value;
  		/* use fixed brightness for OLEDs */
  		buf[4] = 0x08;
  		hdev->hid_output_raw_report(hdev, buf, 9, HID_FEATURE_REPORT);
  		kfree(buf);
  	}
  
  	return;
  }
  
  static enum led_brightness wacom_leds_get_brightness(struct led_classdev *led_dev)
  {
  	struct wacom_data *wdata;
  	struct device *dev = led_dev->dev->parent;
  	int value = 0;
  	int i;
  
  	wdata = hid_get_drvdata(container_of(dev, struct hid_device, dev));
  
  	for (i = 0; i < 4; ++i) {
  		if (wdata->leds[i] == led_dev) {
  			value = wdata->leds[i]->brightness;
  			break;
  		}
  	}
  
  	return value;
  }
  
  
  static int wacom_initialize_leds(struct hid_device *hdev)
  {
  	struct wacom_data *wdata = hid_get_drvdata(hdev);
  	struct led_classdev *led;
  	struct device *dev = &hdev->dev;
  	size_t namesz = strlen(dev_name(dev)) + 12;
  	char *name;
  	int i, ret;
  
  	wdata->led_selector = 0;
  
  	for (i = 0; i < 4; i++) {
  		led = kzalloc(sizeof(struct led_classdev) + namesz, GFP_KERNEL);
  		if (!led) {
  			hid_warn(hdev,
  				 "can't allocate memory for LED selector
  ");
  			ret = -ENOMEM;
  			goto err;
  		}
  
  		name = (void *)&led[1];
  		snprintf(name, namesz, "%s:selector:%d", dev_name(dev), i);
  		led->name = name;
  		led->brightness = 0;
  		led->max_brightness = 127;
  		led->brightness_get = wacom_leds_get_brightness;
  		led->brightness_set = wacom_leds_set_brightness;
  
  		wdata->leds[i] = led;
  
  		ret = led_classdev_register(dev, wdata->leds[i]);
  
  		if (ret) {
  			wdata->leds[i] = NULL;
  			kfree(led);
  			hid_warn(hdev, "can't register LED
  ");
  			goto err;
  		}
  	}
  
  err:
  	return ret;
  }
  
  static void wacom_destroy_leds(struct hid_device *hdev)
  {
  	struct wacom_data *wdata = hid_get_drvdata(hdev);
  	struct led_classdev *led;
  	int i;
  
  	for (i = 0; i < 4; ++i) {
  		if (wdata->leds[i]) {
  			led = wdata->leds[i];
  			wdata->leds[i] = NULL;
  			led_classdev_unregister(led);
  			kfree(led);
  		}
  	}
  
  }
  
  static int wacom_battery_get_property(struct power_supply *psy,
  				enum power_supply_property psp,
  				union power_supply_propval *val)
  {
  	struct wacom_data *wdata = container_of(psy,
  					struct wacom_data, battery);
  	int ret = 0;
  
  	switch (psp) {
  	case POWER_SUPPLY_PROP_PRESENT:
  		val->intval = 1;
  		break;
  	case POWER_SUPPLY_PROP_SCOPE:
  		val->intval = POWER_SUPPLY_SCOPE_DEVICE;
  		break;
  	case POWER_SUPPLY_PROP_CAPACITY:
  		val->intval = wdata->battery_capacity;
  		break;
  	case POWER_SUPPLY_PROP_STATUS:
  		if (wdata->bat_charging)
  			val->intval = POWER_SUPPLY_STATUS_CHARGING;
  		else
  			if (wdata->battery_capacity == 100 && wdata->ps_connected)
  				val->intval = POWER_SUPPLY_STATUS_FULL;
  			else
  				val->intval = POWER_SUPPLY_STATUS_DISCHARGING;
  		break;
  	default:
  		ret = -EINVAL;
  		break;
  	}
  	return ret;
  }
  
  static int wacom_ac_get_property(struct power_supply *psy,
  				enum power_supply_property psp,
  				union power_supply_propval *val)
  {
  	struct wacom_data *wdata = container_of(psy, struct wacom_data, ac);
  	int ret = 0;
  
  	switch (psp) {
  	case POWER_SUPPLY_PROP_PRESENT:
  		/* fall through */
  	case POWER_SUPPLY_PROP_ONLINE:
  		val->intval = wdata->ps_connected;
  		break;
  	case POWER_SUPPLY_PROP_SCOPE:
  		val->intval = POWER_SUPPLY_SCOPE_DEVICE;
  		break;
  	default:
  		ret = -EINVAL;
  		break;
  	}
  	return ret;
  }
  
  static void wacom_set_features(struct hid_device *hdev, u8 speed)
  {
  	struct wacom_data *wdata = hid_get_drvdata(hdev);
  	int limit, ret;
  	__u8 rep_data[2];
  
  	switch (hdev->product) {
  	case USB_DEVICE_ID_WACOM_GRAPHIRE_BLUETOOTH:
  		rep_data[0] = 0x03 ; rep_data[1] = 0x00;
  		limit = 3;
  		do {
  			ret = hdev->hid_output_raw_report(hdev, rep_data, 2,
  					HID_FEATURE_REPORT);
  		} while (ret < 0 && limit-- > 0);
  
  		if (ret >= 0) {
  			if (speed == 0)
  				rep_data[0] = 0x05;
  			else
  				rep_data[0] = 0x06;
  
  			rep_data[1] = 0x00;
  			limit = 3;
  			do {
  				ret = hdev->hid_output_raw_report(hdev,
  					rep_data, 2, HID_FEATURE_REPORT);
  			} while (ret < 0 && limit-- > 0);
  
  			if (ret >= 0) {
  				wdata->high_speed = speed;
  				return;
  			}
  		}
  
  		/*
  		 * Note that if the raw queries fail, it's not a hard failure
  		 * and it is safe to continue
  		 */
  		hid_warn(hdev, "failed to poke device, command %d, err %d
  ",
  			 rep_data[0], ret);
  		break;
  	case USB_DEVICE_ID_WACOM_INTUOS4_BLUETOOTH:
  		if (speed == 1)
  			wdata->features &= ~0x20;
  		else
  			wdata->features |= 0x20;
  
  		rep_data[0] = 0x03;
  		rep_data[1] = wdata->features;
  
  		ret = hdev->hid_output_raw_report(hdev, rep_data, 2,
  					HID_FEATURE_REPORT);
  		if (ret >= 0)
  			wdata->high_speed = speed;
  		break;
  	}
  
  	return;
  }
  
  static ssize_t wacom_show_speed(struct device *dev,
  				struct device_attribute
  				*attr, char *buf)
  {
  	struct wacom_data *wdata = dev_get_drvdata(dev);
  
  	return snprintf(buf, PAGE_SIZE, "%i
  ", wdata->high_speed);
  }
  
  static ssize_t wacom_store_speed(struct device *dev,
  				struct device_attribute *attr,
  				const char *buf, size_t count)
  {
  	struct hid_device *hdev = container_of(dev, struct hid_device, dev);
  	int new_speed;
  
  	if (sscanf(buf, "%1d", &new_speed ) != 1)
  		return -EINVAL;
  
  	if (new_speed == 0 || new_speed == 1) {
  		wacom_set_features(hdev, new_speed);
  		return strnlen(buf, PAGE_SIZE);
  	} else
  		return -EINVAL;
  }
  
  static DEVICE_ATTR(speed, S_IRUGO | S_IWUSR | S_IWGRP,
  		wacom_show_speed, wacom_store_speed);
  
  #define WACOM_STORE(OLED_ID)						\
  static ssize_t wacom_oled##OLED_ID##_store(struct device *dev,		\
  				struct device_attribute *attr,		\
  				const char *buf, size_t count)		\
  {									\
  	struct hid_device *hdev = container_of(dev, struct hid_device,	\
  				dev);					\
  									\
  	if (count != 256)						\
  		return -EINVAL;						\
  									\
  	wacom_set_image(hdev, buf, OLED_ID);				\
  									\
  	return count;							\
  }									\
  									\
  static DEVICE_ATTR(oled##OLED_ID##_img, S_IWUSR | S_IWGRP, NULL,	\
  				wacom_oled##OLED_ID##_store)
  
  WACOM_STORE(0);
  WACOM_STORE(1);
  WACOM_STORE(2);
  WACOM_STORE(3);
  WACOM_STORE(4);
  WACOM_STORE(5);
  WACOM_STORE(6);
  WACOM_STORE(7);
  
  static int wacom_gr_parse_report(struct hid_device *hdev,
  			struct wacom_data *wdata,
  			struct input_dev *input, unsigned char *data)
  {
  	int tool, x, y, rw;
  
  	tool = 0;
  	/* Get X & Y positions */
  	x = le16_to_cpu(*(__le16 *) &data[2]);
  	y = le16_to_cpu(*(__le16 *) &data[4]);
  
  	/* Get current tool identifier */
  	if (data[1] & 0x90) { /* If pen is in the in/active area */
  		switch ((data[1] >> 5) & 3) {
  		case 0:	/* Pen */
  			tool = BTN_TOOL_PEN;
  			break;
  
  		case 1: /* Rubber */
  			tool = BTN_TOOL_RUBBER;
  			break;
  
  		case 2: /* Mouse with wheel */
  		case 3: /* Mouse without wheel */
  			tool = BTN_TOOL_MOUSE;
  			break;
  		}
  
  		/* Reset tool if out of active tablet area */
  		if (!(data[1] & 0x10))
  			tool = 0;
  	}
  
  	/* If tool changed, notify input subsystem */
  	if (wdata->tool != tool) {
  		if (wdata->tool) {
  			/* Completely reset old tool state */
  			if (wdata->tool == BTN_TOOL_MOUSE) {
  				input_report_key(input, BTN_LEFT, 0);
  				input_report_key(input, BTN_RIGHT, 0);
  				input_report_key(input, BTN_MIDDLE, 0);
  				input_report_abs(input, ABS_DISTANCE,
  					input_abs_get_max(input, ABS_DISTANCE));
  			} else {
  				input_report_key(input, BTN_TOUCH, 0);
  				input_report_key(input, BTN_STYLUS, 0);
  				input_report_key(input, BTN_STYLUS2, 0);
  				input_report_abs(input, ABS_PRESSURE, 0);
  			}
  			input_report_key(input, wdata->tool, 0);
  			input_sync(input);
  		}
  		wdata->tool = tool;
  		if (tool)
  			input_report_key(input, tool, 1);
  	}
  
  	if (tool) {
  		input_report_abs(input, ABS_X, x);
  		input_report_abs(input, ABS_Y, y);
  
  		switch ((data[1] >> 5) & 3) {
  		case 2: /* Mouse with wheel */
  			input_report_key(input, BTN_MIDDLE, data[1] & 0x04);
  			rw = (data[6] & 0x01) ? -1 :
  				(data[6] & 0x02) ? 1 : 0;
  			input_report_rel(input, REL_WHEEL, rw);
  			/* fall through */
  
  		case 3: /* Mouse without wheel */
  			input_report_key(input, BTN_LEFT, data[1] & 0x01);
  			input_report_key(input, BTN_RIGHT, data[1] & 0x02);
  			/* Compute distance between mouse and tablet */
  			rw = 44 - (data[6] >> 2);
  			if (rw < 0)
  				rw = 0;
  			else if (rw > 31)
  				rw = 31;
  			input_report_abs(input, ABS_DISTANCE, rw);
  			break;
  
  		default:
  			input_report_abs(input, ABS_PRESSURE,
  					data[6] | (((__u16) (data[1] & 0x08)) << 5));
  			input_report_key(input, BTN_TOUCH, data[1] & 0x01);
  			input_report_key(input, BTN_STYLUS, data[1] & 0x02);
  			input_report_key(input, BTN_STYLUS2, (tool == BTN_TOOL_PEN) && data[1] & 0x04);
  			break;
  		}
  
  		input_sync(input);
  	}
  
  	/* Report the state of the two buttons at the top of the tablet
  	 * as two extra fingerpad keys (buttons 4 & 5). */
  	rw = data[7] & 0x03;
  	if (rw != wdata->butstate) {
  		wdata->butstate = rw;
  		input_report_key(input, BTN_0, rw & 0x02);
  		input_report_key(input, BTN_1, rw & 0x01);
  		input_report_key(input, BTN_TOOL_FINGER, 0xf0);
  		input_event(input, EV_MSC, MSC_SERIAL, 0xf0);
  		input_sync(input);
  	}
  
  	/* Store current battery capacity and power supply state*/
  	rw = (data[7] >> 2 & 0x07);
  	if (rw != wdata->power_raw) {
  		wdata->power_raw = rw;
  		wdata->battery_capacity = batcap_gr[rw];
  		if (rw == 7)
  			wdata->ps_connected = 1;
  		else
  			wdata->ps_connected = 0;
  	}
  	return 1;
  }
  
  static void wacom_i4_parse_button_report(struct wacom_data *wdata,
  			struct input_dev *input, unsigned char *data)
  {
  	__u16 new_butstate;
  	__u8 new_whlstate;
  	__u8 sync = 0;
  
  	new_whlstate = data[1];
  	if (new_whlstate != wdata->whlstate) {
  		wdata->whlstate = new_whlstate;
  		if (new_whlstate & 0x80) {
  			input_report_key(input, BTN_TOUCH, 1);
  			input_report_abs(input, ABS_WHEEL, (new_whlstate & 0x7f));
  			input_report_key(input, BTN_TOOL_FINGER, 1);
  		} else {
  			input_report_key(input, BTN_TOUCH, 0);
  			input_report_abs(input, ABS_WHEEL, 0);
  			input_report_key(input, BTN_TOOL_FINGER, 0);
  		}
  		sync = 1;
  	}
  
  	new_butstate = (data[3] << 1) | (data[2] & 0x01);
  	if (new_butstate != wdata->butstate) {
  		wdata->butstate = new_butstate;
  		input_report_key(input, BTN_0, new_butstate & 0x001);
  		input_report_key(input, BTN_1, new_butstate & 0x002);
  		input_report_key(input, BTN_2, new_butstate & 0x004);
  		input_report_key(input, BTN_3, new_butstate & 0x008);
  		input_report_key(input, BTN_4, new_butstate & 0x010);
  		input_report_key(input, BTN_5, new_butstate & 0x020);
  		input_report_key(input, BTN_6, new_butstate & 0x040);
  		input_report_key(input, BTN_7, new_butstate & 0x080);
  		input_report_key(input, BTN_8, new_butstate & 0x100);
  		input_report_key(input, BTN_TOOL_FINGER, 1);
  		sync = 1;
  	}
  
  	if (sync) {
  		input_report_abs(input, ABS_MISC, PAD_DEVICE_ID);
  		input_event(input, EV_MSC, MSC_SERIAL, 0xffffffff);
  		input_sync(input);
  	}
  }
  
  static void wacom_i4_parse_pen_report(struct wacom_data *wdata,
  			struct input_dev *input, unsigned char *data)
  {
  	__u16 x, y, pressure;
  	__u8 distance;
  	__u8 tilt_x, tilt_y;
  
  	switch (data[1]) {
  	case 0x80: /* Out of proximity report */
  		input_report_key(input, BTN_TOUCH, 0);
  		input_report_abs(input, ABS_PRESSURE, 0);
  		input_report_key(input, BTN_STYLUS, 0);
  		input_report_key(input, BTN_STYLUS2, 0);
  		input_report_key(input, wdata->tool, 0);
  		input_report_abs(input, ABS_MISC, 0);
  		input_event(input, EV_MSC, MSC_SERIAL, wdata->serial);
  		wdata->tool = 0;
  		input_sync(input);
  		break;
  	case 0xC2: /* Tool report */
  		wdata->id = ((data[2] << 4) | (data[3] >> 4) |
  			((data[7] & 0x0f) << 20) |
  			((data[8] & 0xf0) << 12));
  		wdata->serial = ((data[3] & 0x0f) << 28) +
  				(data[4] << 20) + (data[5] << 12) +
  				(data[6] << 4) + (data[7] >> 4);
  
  		switch (wdata->id) {
  		case 0x100802:
  			wdata->tool = BTN_TOOL_PEN;
  			break;
  		case 0x10080A:
  			wdata->tool = BTN_TOOL_RUBBER;
  			break;
  		}
  		break;
  	default: /* Position/pressure report */
  		x = data[2] << 9 | data[3] << 1 | ((data[9] & 0x02) >> 1);
  		y = data[4] << 9 | data[5] << 1 | (data[9] & 0x01);
  		pressure = (data[6] << 3) | ((data[7] & 0xC0) >> 5)
  			| (data[1] & 0x01);
  		distance = (data[9] >> 2) & 0x3f;
  		tilt_x = ((data[7] << 1) & 0x7e) | (data[8] >> 7);
  		tilt_y = data[8] & 0x7f;
  
  		input_report_key(input, BTN_TOUCH, pressure > 1);
  
  		input_report_key(input, BTN_STYLUS, data[1] & 0x02);
  		input_report_key(input, BTN_STYLUS2, data[1] & 0x04);
  		input_report_key(input, wdata->tool, 1);
  		input_report_abs(input, ABS_X, x);
  		input_report_abs(input, ABS_Y, y);
  		input_report_abs(input, ABS_PRESSURE, pressure);
  		input_report_abs(input, ABS_DISTANCE, distance);
  		input_report_abs(input, ABS_TILT_X, tilt_x);
  		input_report_abs(input, ABS_TILT_Y, tilt_y);
  		input_report_abs(input, ABS_MISC, wdata->id);
  		input_event(input, EV_MSC, MSC_SERIAL, wdata->serial);
  		input_report_key(input, wdata->tool, 1);
  		input_sync(input);
  		break;
  	}
  
  	return;
  }
  
  static void wacom_i4_parse_report(struct hid_device *hdev,
  			struct wacom_data *wdata,
  			struct input_dev *input, unsigned char *data)
  {
  	switch (data[0]) {
  	case 0x00: /* Empty report */
  		break;
  	case 0x02: /* Pen report */
  		wacom_i4_parse_pen_report(wdata, input, data);
  		break;
  	case 0x03: /* Features Report */
  		wdata->features = data[2];
  		break;
  	case 0x0C: /* Button report */
  		wacom_i4_parse_button_report(wdata, input, data);
  		break;
  	default:
  		hid_err(hdev, "Unknown report: %d,%d
  ", data[0], data[1]);
  		break;
  	}
  }
  
  static int wacom_raw_event(struct hid_device *hdev, struct hid_report *report,
  		u8 *raw_data, int size)
  {
  	struct wacom_data *wdata = hid_get_drvdata(hdev);
  	struct hid_input *hidinput;
  	struct input_dev *input;
  	unsigned char *data = (unsigned char *) raw_data;
  	int i;
  	__u8 power_raw;
  
  	if (!(hdev->claimed & HID_CLAIMED_INPUT))
  		return 0;
  
  	hidinput = list_entry(hdev->inputs.next, struct hid_input, list);
  	input = hidinput->input;
  
  	switch (hdev->product) {
  	case USB_DEVICE_ID_WACOM_GRAPHIRE_BLUETOOTH:
  		if (data[0] == 0x03) {
  			return wacom_gr_parse_report(hdev, wdata, input, data);
  		} else {
  			hid_err(hdev, "Unknown report: %d,%d size:%d
  ",
  					data[0], data[1], size);
  			return 0;
  		}
  		break;
  	case USB_DEVICE_ID_WACOM_INTUOS4_BLUETOOTH:
  		i = 1;
  
  		switch (data[0]) {
  		case 0x04:
  			wacom_i4_parse_report(hdev, wdata, input, data + i);
  			i += 10;
  			/* fall through */
  		case 0x03:
  			wacom_i4_parse_report(hdev, wdata, input, data + i);
  			i += 10;
  			wacom_i4_parse_report(hdev, wdata, input, data + i);
  			power_raw = data[i+10];
  			if (power_raw != wdata->power_raw) {
  				wdata->power_raw = power_raw;
  				wdata->battery_capacity = batcap_i4[power_raw & 0x07];
  				wdata->bat_charging = (power_raw & 0x08) ? 1 : 0;
  				wdata->ps_connected = (power_raw & 0x10) ? 1 : 0;
  			}
  
  			break;
  		default:
  			hid_err(hdev, "Unknown report: %d,%d size:%d
  ",
  					data[0], data[1], size);
  			return 0;
  		}
  	}
  	return 1;
  }
  
  static int wacom_input_mapped(struct hid_device *hdev, struct hid_input *hi,
  	struct hid_field *field, struct hid_usage *usage, unsigned long **bit,
  								int *max)
  {
  	struct input_dev *input = hi->input;
  
  	__set_bit(INPUT_PROP_POINTER, input->propbit);
  
  	/* Basics */
  	input->evbit[0] |= BIT(EV_KEY) | BIT(EV_ABS) | BIT(EV_REL);
  
  	__set_bit(REL_WHEEL, input->relbit);
  
  	__set_bit(BTN_TOOL_PEN, input->keybit);
  	__set_bit(BTN_TOUCH, input->keybit);
  	__set_bit(BTN_STYLUS, input->keybit);
  	__set_bit(BTN_STYLUS2, input->keybit);
  	__set_bit(BTN_LEFT, input->keybit);
  	__set_bit(BTN_RIGHT, input->keybit);
  	__set_bit(BTN_MIDDLE, input->keybit);
  
  	/* Pad */
  	input_set_capability(input, EV_MSC, MSC_SERIAL);
  
  	__set_bit(BTN_0, input->keybit);
  	__set_bit(BTN_1, input->keybit);
  	__set_bit(BTN_TOOL_FINGER, input->keybit);
  
  	/* Distance, rubber and mouse */
  	__set_bit(BTN_TOOL_RUBBER, input->keybit);
  	__set_bit(BTN_TOOL_MOUSE, input->keybit);
  
  	switch (hdev->product) {
  	case USB_DEVICE_ID_WACOM_GRAPHIRE_BLUETOOTH:
  		input_set_abs_params(input, ABS_X, 0, 16704, 4, 0);
  		input_set_abs_params(input, ABS_Y, 0, 12064, 4, 0);
  		input_set_abs_params(input, ABS_PRESSURE, 0, 511, 0, 0);
  		input_set_abs_params(input, ABS_DISTANCE, 0, 32, 0, 0);
  		break;
  	case USB_DEVICE_ID_WACOM_INTUOS4_BLUETOOTH:
  		__set_bit(ABS_WHEEL, input->absbit);
  		__set_bit(ABS_MISC, input->absbit);
  		__set_bit(BTN_2, input->keybit);
  		__set_bit(BTN_3, input->keybit);
  		__set_bit(BTN_4, input->keybit);
  		__set_bit(BTN_5, input->keybit);
  		__set_bit(BTN_6, input->keybit);
  		__set_bit(BTN_7, input->keybit);
  		__set_bit(BTN_8, input->keybit);
  		input_set_abs_params(input, ABS_WHEEL, 0, 71, 0, 0);
  		input_set_abs_params(input, ABS_X, 0, 40640, 4, 0);
  		input_set_abs_params(input, ABS_Y, 0, 25400, 4, 0);
  		input_set_abs_params(input, ABS_PRESSURE, 0, 2047, 0, 0);
  		input_set_abs_params(input, ABS_DISTANCE, 0, 63, 0, 0);
  		input_set_abs_params(input, ABS_TILT_X, 0, 127, 0, 0);
  		input_set_abs_params(input, ABS_TILT_Y, 0, 127, 0, 0);
  		break;
  	}
  
  	return 0;
  }
  
  static int wacom_probe(struct hid_device *hdev,
  		const struct hid_device_id *id)
  {
  	struct wacom_data *wdata;
  	int ret;
  
  	wdata = kzalloc(sizeof(*wdata), GFP_KERNEL);
  	if (wdata == NULL) {
  		hid_err(hdev, "can't alloc wacom descriptor
  ");
  		return -ENOMEM;
  	}
  
  	hid_set_drvdata(hdev, wdata);
  
  	/* Parse the HID report now */
  	ret = hid_parse(hdev);
  	if (ret) {
  		hid_err(hdev, "parse failed
  ");
  		goto err_free;
  	}
  
  	ret = hid_hw_start(hdev, HID_CONNECT_DEFAULT);
  	if (ret) {
  		hid_err(hdev, "hw start failed
  ");
  		goto err_free;
  	}
  
  	ret = device_create_file(&hdev->dev, &dev_attr_speed);
  	if (ret)
  		hid_warn(hdev,
  			 "can't create sysfs speed attribute err: %d
  ", ret);
  
  #define OLED_INIT(OLED_ID)						\
  	do {								\
  		ret = device_create_file(&hdev->dev,			\
  				&dev_attr_oled##OLED_ID##_img);		\
  		if (ret)						\
  			hid_warn(hdev,					\
  			 "can't create sysfs oled attribute, err: %d
  ", ret);\
  	} while (0)
  
  OLED_INIT(0);
  OLED_INIT(1);
  OLED_INIT(2);
  OLED_INIT(3);
  OLED_INIT(4);
  OLED_INIT(5);
  OLED_INIT(6);
  OLED_INIT(7);
  
  	wdata->features = 0;
  	wacom_set_features(hdev, 1);
  
  	if (hdev->product == USB_DEVICE_ID_WACOM_INTUOS4_BLUETOOTH) {
  		sprintf(hdev->name, "%s", "Wacom Intuos4 WL");
  		ret = wacom_initialize_leds(hdev);
  		if (ret)
  			hid_warn(hdev,
  				 "can't create led attribute, err: %d
  ", ret);
  	}
  
  	wdata->battery.properties = wacom_battery_props;
  	wdata->battery.num_properties = ARRAY_SIZE(wacom_battery_props);
  	wdata->battery.get_property = wacom_battery_get_property;
  	wdata->battery.name = "wacom_battery";
  	wdata->battery.type = POWER_SUPPLY_TYPE_BATTERY;
  	wdata->battery.use_for_apm = 0;
  
  
  	ret = power_supply_register(&hdev->dev, &wdata->battery);
  	if (ret) {
  		hid_err(hdev, "can't create sysfs battery attribute, err: %d
  ",
  			ret);
  		goto err_battery;
  	}
  
  	power_supply_powers(&wdata->battery, &hdev->dev);
  
  	wdata->ac.properties = wacom_ac_props;
  	wdata->ac.num_properties = ARRAY_SIZE(wacom_ac_props);
  	wdata->ac.get_property = wacom_ac_get_property;
  	wdata->ac.name = "wacom_ac";
  	wdata->ac.type = POWER_SUPPLY_TYPE_MAINS;
  	wdata->ac.use_for_apm = 0;
  
  	ret = power_supply_register(&hdev->dev, &wdata->ac);
  	if (ret) {
  		hid_err(hdev,
  			"can't create ac battery attribute, err: %d
  ", ret);
  		goto err_ac;
  	}
  
  	power_supply_powers(&wdata->ac, &hdev->dev);
  	return 0;
  
  err_ac:
  	power_supply_unregister(&wdata->battery);
  err_battery:
  	wacom_destroy_leds(hdev);
  	device_remove_file(&hdev->dev, &dev_attr_oled0_img);
  	device_remove_file(&hdev->dev, &dev_attr_oled1_img);
  	device_remove_file(&hdev->dev, &dev_attr_oled2_img);
  	device_remove_file(&hdev->dev, &dev_attr_oled3_img);
  	device_remove_file(&hdev->dev, &dev_attr_oled4_img);
  	device_remove_file(&hdev->dev, &dev_attr_oled5_img);
  	device_remove_file(&hdev->dev, &dev_attr_oled6_img);
  	device_remove_file(&hdev->dev, &dev_attr_oled7_img);
  	device_remove_file(&hdev->dev, &dev_attr_speed);
  	hid_hw_stop(hdev);
  err_free:
  	kfree(wdata);
  	return ret;
  }
  
  static void wacom_remove(struct hid_device *hdev)
  {
  	struct wacom_data *wdata = hid_get_drvdata(hdev);
  
  	wacom_destroy_leds(hdev);
  	device_remove_file(&hdev->dev, &dev_attr_oled0_img);
  	device_remove_file(&hdev->dev, &dev_attr_oled1_img);
  	device_remove_file(&hdev->dev, &dev_attr_oled2_img);
  	device_remove_file(&hdev->dev, &dev_attr_oled3_img);
  	device_remove_file(&hdev->dev, &dev_attr_oled4_img);
  	device_remove_file(&hdev->dev, &dev_attr_oled5_img);
  	device_remove_file(&hdev->dev, &dev_attr_oled6_img);
  	device_remove_file(&hdev->dev, &dev_attr_oled7_img);
  	device_remove_file(&hdev->dev, &dev_attr_speed);
  	hid_hw_stop(hdev);
  
  	power_supply_unregister(&wdata->battery);
  	power_supply_unregister(&wdata->ac);
  	kfree(hid_get_drvdata(hdev));
  }
  
  static const struct hid_device_id wacom_devices[] = {
  	{ HID_BLUETOOTH_DEVICE(USB_VENDOR_ID_WACOM, USB_DEVICE_ID_WACOM_GRAPHIRE_BLUETOOTH) },
  	{ HID_BLUETOOTH_DEVICE(USB_VENDOR_ID_WACOM, USB_DEVICE_ID_WACOM_INTUOS4_BLUETOOTH) },
  
  	{ }
  };
  MODULE_DEVICE_TABLE(hid, wacom_devices);
  
  static struct hid_driver wacom_driver = {
  	.name = "wacom",
  	.id_table = wacom_devices,
  	.probe = wacom_probe,
  	.remove = wacom_remove,
  	.raw_event = wacom_raw_event,
  	.input_mapped = wacom_input_mapped,
  };
  module_hid_driver(wacom_driver);
  
  MODULE_DESCRIPTION("Driver for Wacom Graphire Bluetooth and Wacom Intuos4 WL");
  MODULE_LICENSE("GPL");