@@ -0,0 +1,294 @@
+// SPDX-License-Identifier: GPL-2.0-or-later
+/*
+ * Pixart Touchpad Controller 1336U PS2 driver
+ *
+ * Author: Jon Xie <jon_xie@pixart.com>
+ * Jay Lee <jay_lee@pixart.com>
+ * Further cleanup and restructuring by:
+ * Binbin Zhou <zhoubinbin@loongson.cn>
+ *
+ * Copyright (C) 2021-2024 Pixart Imaging
+ * Copyright (C) 2024 Loongson Technology Corporation Limited.
+ *
+ */
+
+#include <linux/delay.h>
+#include <linux/device.h>
+#include <linux/libps2.h>
+#include <linux/input/mt.h>
+#include <linux/serio.h>
+#include <linux/slab.h>
+#include "pixart_ps2.h"
+
+static int pixart_mode_detect(struct psmouse *psmouse)
+{
+ u8 param[1];
+
+ if (ps2_command(&psmouse->ps2dev, param, PIXART_CMD_REPORT_FORMAT))
+ return -EIO;
+
+ return (param[0] == 1) ? PIXART_MODE_ABS : PIXART_MODE_REL;
+}
+
+static int pixart_type_detect(struct psmouse *psmouse)
+{
+ struct ps2dev *ps2dev = &psmouse->ps2dev;
+ u8 param[3];
+
+ param[0] = 0x0a;
+ ps2_command(ps2dev, param, PSMOUSE_CMD_SETRATE);
+ param[0] = 0x0;
+ ps2_command(ps2dev, param, PSMOUSE_CMD_SETRES);
+ param[0] = 0x0;
+ ps2_command(ps2dev, param, PSMOUSE_CMD_SETRES);
+ param[0] = 0x0;
+ ps2_command(ps2dev, param, PSMOUSE_CMD_SETRES);
+ param[0] = 0x03;
+ ps2_command(ps2dev, param, PSMOUSE_CMD_SETRES);
+ ps2_command(ps2dev, param, PSMOUSE_CMD_GETINFO);
+
+ return (param[0] == 0x0e) ? PIXART_TYPE_TOUCHPAD : PIXART_TYPE_CLICKPAD;
+}
+
+static int pixart_intellimouse_detect(struct psmouse *psmouse)
+{
+ struct ps2dev *ps2dev = &psmouse->ps2dev;
+ u8 param[2];
+
+ param[0] = 200;
+ ps2_command(ps2dev, param, PSMOUSE_CMD_SETRATE);
+ param[0] = 100;
+ ps2_command(ps2dev, param, PSMOUSE_CMD_SETRATE);
+ param[0] = 80;
+ ps2_command(ps2dev, param, PSMOUSE_CMD_SETRATE);
+ ps2_command(ps2dev, param, PSMOUSE_CMD_GETID);
+
+ return (param[0] == 3) ? 0 : -1;
+}
+
+static void pixart_reset(struct psmouse *psmouse)
+{
+ ps2_command(&psmouse->ps2dev, NULL, PSMOUSE_CMD_RESET_DIS);
+ msleep(100);
+ psmouse_reset(psmouse);
+}
+
+static psmouse_ret_t pixart_process_abs_packet(struct psmouse *psmouse)
+{
+ struct pixart_data *priv = psmouse->private;
+ struct input_dev *dev = psmouse->dev;
+ u8 *packet = psmouse->packet;
+ u8 contact_cnt = (packet[0] >> 4) & 0x07;
+ unsigned int abs_x, abs_y;
+ int i, id, fingers = 0;
+ bool tip;
+
+ if (contact_cnt > MAX_FINGERS || ((packet[0] & 0x8c) != 0x80))
+ return PSMOUSE_BAD_DATA;
+
+ if (contact_cnt == MAX_FINGERS && psmouse->pktcnt < psmouse->pktsize)
+ return PSMOUSE_GOOD_DATA;
+ else if (contact_cnt == 0 && psmouse->pktcnt < 5)
+ return PSMOUSE_GOOD_DATA;
+ else if (psmouse->pktcnt < (3 * contact_cnt + 2))
+ return PSMOUSE_GOOD_DATA;
+
+ for (i = 0; i < contact_cnt; i++) {
+ id = packet[3 * i + 3] & 0x07;
+ abs_y = ((packet[3 * i + 3] & 0x30) << 4) | packet[3 * i + 1];
+ abs_x = ((packet[3 * i + 3] & 0xc0) << 2) | packet[3 * i + 2];
+
+ if (i == MAX_FINGERS - 1)
+ tip = packet[14] & (0x01 << 1);
+ else
+ tip = packet[3 * contact_cnt + 1] & (0x01 << (2 * i + 1));
+
+ input_mt_slot(dev, id);
+ input_mt_report_slot_state(dev, MT_TOOL_FINGER, tip);
+ if (tip) {
+ fingers++;
+ input_report_abs(dev, ABS_MT_POSITION_Y, abs_y);
+ input_report_abs(dev, ABS_MT_POSITION_X, abs_x);
+ }
+ }
+
+ input_mt_drop_unused(dev);
+ input_mt_report_pointer_emulation(dev, false);
+ input_mt_report_finger_count(dev, fingers);
+
+ if (priv->type == PIXART_TYPE_CLICKPAD) {
+ input_report_key(dev, BTN_LEFT, packet[0] & 0x03);
+ } else {
+ input_report_key(dev, BTN_LEFT, packet[0] & 0x01);
+ input_report_key(dev, BTN_RIGHT, packet[0] & 0x02);
+ }
+
+ input_mt_report_pointer_emulation(dev, true);
+ input_sync(dev);
+
+ return PSMOUSE_FULL_PACKET;
+}
+
+static psmouse_ret_t pixart_process_rel_packet(struct psmouse *psmouse)
+{
+ struct input_dev *dev = psmouse->dev;
+ u8 *packet = psmouse->packet;
+ int x = packet[1] ? packet[1] - ((packet[0] << 4) & 0x100) : 0;
+ int y = packet[2] ? ((packet[0] << 3) & 0x100) - packet[2] : 0;
+ int w = -(s8)packet[3];
+
+ if (psmouse->pktcnt < psmouse->pktsize)
+ return PSMOUSE_GOOD_DATA;
+
+ psmouse_report_standard_buttons(dev, packet[0]);
+ input_report_rel(dev, REL_X, x);
+ input_report_rel(dev, REL_Y, y);
+ input_report_rel(dev, REL_WHEEL, w);
+ input_sync(dev);
+
+ return PSMOUSE_FULL_PACKET;
+}
+
+static void pixart_disconnect(struct psmouse *psmouse)
+{
+ psmouse_info(psmouse, "Device disconnect");
+ pixart_reset(psmouse);
+ kfree(psmouse->private);
+ psmouse->private = NULL;
+}
+
+static int pixart_reconnect(struct psmouse *psmouse)
+{
+ struct pixart_data *priv = psmouse->private;
+
+ psmouse_info(psmouse, "Device reconnect");
+ pixart_reset(psmouse);
+
+ priv->mode = pixart_mode_detect(psmouse);
+ if (priv->mode < 0) {
+ psmouse_err(psmouse, "Unable to detect the PixArt device");
+ return -EIO;
+ }
+
+ if (priv->mode == PIXART_MODE_ABS)
+ ps2_command(&psmouse->ps2dev, NULL, PIXART_CMD_SWITCH_PROTO);
+
+ return 0;
+}
+
+static void pixart_set_abs_input_params(struct input_dev *dev, struct pixart_data *priv)
+{
+ __clear_bit(EV_REL, dev->evbit);
+ __clear_bit(REL_X, dev->relbit);
+ __clear_bit(REL_Y, dev->relbit);
+ __clear_bit(BTN_MIDDLE, dev->keybit);
+
+ __set_bit(EV_KEY, dev->evbit);
+ __set_bit(BTN_LEFT, dev->keybit);
+ __set_bit(BTN_RIGHT, dev->keybit);
+ if (priv->type == PIXART_TYPE_CLICKPAD) {
+ __set_bit(INPUT_PROP_BUTTONPAD, dev->propbit);
+ __clear_bit(BTN_RIGHT, dev->keybit);
+ }
+ __set_bit(BTN_TOUCH, dev->keybit);
+ __set_bit(BTN_TOOL_FINGER, dev->keybit);
+ __set_bit(BTN_TOOL_DOUBLETAP, dev->keybit);
+ __set_bit(BTN_TOOL_TRIPLETAP, dev->keybit);
+ __set_bit(BTN_TOOL_QUADTAP, dev->keybit);
+ __set_bit(INPUT_PROP_POINTER, dev->propbit);
+
+ __set_bit(EV_ABS, dev->evbit);
+ input_set_abs_params(dev, ABS_X, 0, ABS_MAX_X, 0, 0);
+ input_set_abs_params(dev, ABS_Y, 0, ABS_MAX_Y, 0, 0);
+
+ input_set_abs_params(dev, ABS_MT_POSITION_X, 0, ABS_MAX_X, 0, 0);
+ input_set_abs_params(dev, ABS_MT_POSITION_Y, 0, ABS_MAX_Y, 0, 0);
+
+ input_mt_init_slots(dev, SLOTS_NUMBER, 0);
+}
+
+static void pixart_set_rel_input_params(struct input_dev *dev, struct pixart_data *priv)
+{
+ __set_bit(EV_KEY, dev->evbit);
+ __set_bit(EV_REL, dev->evbit);
+ __set_bit(BTN_LEFT, dev->keybit);
+ __set_bit(BTN_RIGHT, dev->keybit);
+ __set_bit(REL_X, dev->relbit);
+ __set_bit(REL_Y, dev->relbit);
+ __set_bit(REL_WHEEL, dev->relbit);
+ __set_bit(INPUT_PROP_POINTER, dev->propbit);
+}
+
+int pixart_detect(struct psmouse *psmouse, bool set_properties)
+{
+ int mode, type;
+
+ pixart_reset(psmouse);
+
+ type = pixart_type_detect(psmouse);
+ if (type < 0)
+ return -ENODEV;
+
+ mode = pixart_mode_detect(psmouse);
+ if (mode < 0)
+ return -ENODEV;
+
+ psmouse_info(psmouse, "Detect PixArt Device.");
+ if (set_properties) {
+ psmouse->vendor = "PixArt";
+ psmouse->name = (type == PIXART_TYPE_TOUCHPAD) ?
+ "touchpad" : "clickpad";
+ }
+
+ return 0;
+}
+
+int pixart_init(struct psmouse *psmouse)
+{
+ struct pixart_data *priv;
+
+ priv = kzalloc(sizeof(*priv), GFP_KERNEL);
+ if (!priv)
+ return -ENOMEM;
+
+ psmouse->private = priv;
+ pixart_reset(psmouse);
+
+ priv->type = pixart_type_detect(psmouse);
+ if (priv->type < 0)
+ goto fail;
+
+ priv->mode = pixart_mode_detect(psmouse);
+ if (priv->mode < 0) {
+ psmouse_err(psmouse, "Unable to initialize the PixArt device\n");
+ goto fail;
+ }
+
+ if (priv->mode == PIXART_MODE_ABS) {
+ ps2_command(&psmouse->ps2dev, NULL, PIXART_CMD_SWITCH_PROTO);
+ pixart_set_abs_input_params(psmouse->dev, priv);
+ psmouse->protocol_handler = pixart_process_abs_packet;
+ psmouse->pktsize = 15;
+ } else {
+ pixart_reset(psmouse);
+ pixart_intellimouse_detect(psmouse);
+ pixart_set_rel_input_params(psmouse->dev, priv);
+ psmouse->protocol_handler = pixart_process_rel_packet;
+ psmouse->pktsize = 4;
+ }
+
+ psmouse->disconnect = pixart_disconnect;
+ psmouse->reconnect = pixart_reconnect;
+ psmouse->cleanup = pixart_reset;
+ /* resync is not supported yet */
+ psmouse->resync_time = 0;
+
+ psmouse_info(psmouse, "Set device as Type: %x, mode: %x", priv->type, priv->mode);
+ return 0;
+
+fail:
+ pixart_reset(psmouse);
+ kfree(priv);
+ psmouse->private = NULL;
+ return -1;
+}