sensehat-joystick.c 3.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135
  1. // SPDX-License-Identifier: GPL-2.0-or-later
  2. /*
  3. * Raspberry Pi Sense HAT joystick driver
  4. * http://raspberrypi.org
  5. *
  6. * Copyright (C) 2015 Raspberry Pi
  7. * Copyright (C) 2021 Charles Mirabile, Mwesigwa Guma, Joel Savitz
  8. *
  9. * Original Author: Serge Schneider
  10. * Revised for upstream Linux by: Charles Mirabile, Mwesigwa Guma, Joel Savitz
  11. */
  12. #include <linux/module.h>
  13. #include <linux/input.h>
  14. #include <linux/i2c.h>
  15. #include <linux/interrupt.h>
  16. #include <linux/platform_device.h>
  17. #include <linux/regmap.h>
  18. #include <linux/property.h>
  19. #define JOYSTICK_SMB_REG 0xf2
  20. struct sensehat_joystick {
  21. struct platform_device *pdev;
  22. struct input_dev *keys_dev;
  23. unsigned long prev_states;
  24. struct regmap *regmap;
  25. };
  26. static const unsigned int keymap[] = {
  27. BTN_DPAD_DOWN, BTN_DPAD_RIGHT, BTN_DPAD_UP, BTN_SELECT, BTN_DPAD_LEFT,
  28. };
  29. static irqreturn_t sensehat_joystick_report(int irq, void *cookie)
  30. {
  31. struct sensehat_joystick *sensehat_joystick = cookie;
  32. unsigned long curr_states, changes;
  33. unsigned int keys;
  34. int error;
  35. int i;
  36. error = regmap_read(sensehat_joystick->regmap, JOYSTICK_SMB_REG, &keys);
  37. if (error < 0) {
  38. dev_err(&sensehat_joystick->pdev->dev,
  39. "Failed to read joystick state: %d", error);
  40. return IRQ_NONE;
  41. }
  42. curr_states = keys;
  43. bitmap_xor(&changes, &curr_states, &sensehat_joystick->prev_states,
  44. ARRAY_SIZE(keymap));
  45. for_each_set_bit(i, &changes, ARRAY_SIZE(keymap))
  46. input_report_key(sensehat_joystick->keys_dev, keymap[i],
  47. curr_states & BIT(i));
  48. input_sync(sensehat_joystick->keys_dev);
  49. sensehat_joystick->prev_states = keys;
  50. return IRQ_HANDLED;
  51. }
  52. static int sensehat_joystick_probe(struct platform_device *pdev)
  53. {
  54. struct sensehat_joystick *sensehat_joystick;
  55. int error, i, irq;
  56. sensehat_joystick = devm_kzalloc(&pdev->dev, sizeof(*sensehat_joystick),
  57. GFP_KERNEL);
  58. if (!sensehat_joystick)
  59. return -ENOMEM;
  60. sensehat_joystick->pdev = pdev;
  61. sensehat_joystick->regmap = dev_get_regmap(pdev->dev.parent, NULL);
  62. if (!sensehat_joystick->regmap) {
  63. dev_err(&pdev->dev, "unable to get sensehat regmap");
  64. return -ENODEV;
  65. }
  66. sensehat_joystick->keys_dev = devm_input_allocate_device(&pdev->dev);
  67. if (!sensehat_joystick->keys_dev) {
  68. dev_err(&pdev->dev, "Could not allocate input device");
  69. return -ENOMEM;
  70. }
  71. sensehat_joystick->keys_dev->name = "Raspberry Pi Sense HAT Joystick";
  72. sensehat_joystick->keys_dev->phys = "sensehat-joystick/input0";
  73. sensehat_joystick->keys_dev->id.bustype = BUS_I2C;
  74. __set_bit(EV_KEY, sensehat_joystick->keys_dev->evbit);
  75. __set_bit(EV_REP, sensehat_joystick->keys_dev->evbit);
  76. for (i = 0; i < ARRAY_SIZE(keymap); i++)
  77. __set_bit(keymap[i], sensehat_joystick->keys_dev->keybit);
  78. error = input_register_device(sensehat_joystick->keys_dev);
  79. if (error) {
  80. dev_err(&pdev->dev, "Could not register input device");
  81. return error;
  82. }
  83. irq = platform_get_irq(pdev, 0);
  84. if (irq < 0)
  85. return irq;
  86. error = devm_request_threaded_irq(&pdev->dev, irq,
  87. NULL, sensehat_joystick_report,
  88. IRQF_ONESHOT, "keys",
  89. sensehat_joystick);
  90. if (error) {
  91. dev_err(&pdev->dev, "IRQ request failed");
  92. return error;
  93. }
  94. return 0;
  95. }
  96. static const struct of_device_id sensehat_joystick_device_id[] = {
  97. { .compatible = "raspberrypi,sensehat-joystick" },
  98. {},
  99. };
  100. MODULE_DEVICE_TABLE(of, sensehat_joystick_device_id);
  101. static struct platform_driver sensehat_joystick_driver = {
  102. .probe = sensehat_joystick_probe,
  103. .driver = {
  104. .name = "sensehat-joystick",
  105. .of_match_table = sensehat_joystick_device_id,
  106. },
  107. };
  108. module_platform_driver(sensehat_joystick_driver);
  109. MODULE_DESCRIPTION("Raspberry Pi Sense HAT joystick driver");
  110. MODULE_AUTHOR("Charles Mirabile <[email protected]>");
  111. MODULE_AUTHOR("Serge Schneider <[email protected]>");
  112. MODULE_LICENSE("GPL");