Input: cyapa - add gen5 trackpad force re-calibrate function support

Add force re-calibrate function support for gen5 trackpad device, it can be
used through sysfs calibrate interface.

Signed-off-by: Dudley Du <dudl@cypress.com>
Tested-by: Jeremiah Mahler <jmmahler@gmail.com>
Signed-off-by: Dmitry Torokhov <dmitry.torokhov@gmail.com>

1 files changed, 65 insertions, 0 deletions

diff --git a/drivers/input/mouse/cyapa_gen5.c b/drivers/input/mouse/cyapa_gen5.c
index 7c6e3393fef1..ced2a2c641cc 100644
--- a/drivers/input/mouse/cyapa_gen5.c
+++ b/drivers/input/mouse/cyapa_gen5.c
@@ -1715,6 +1715,70 @@ static int cyapa_gen5_suspend_scanning(struct cyapa *cyapa)
         return 0;
 }
 
+static int cyapa_gen5_calibrate_pwcs(struct cyapa *cyapa,
+                u8 calibrate_sensing_mode_type)
+{
+        struct gen5_app_cmd_head *app_cmd_head;
+        u8 cmd[8];
+        u8 resp_data[6];
+        int resp_len;
+        int error;
+
+        /* Try to dump all buffered data before doing command. */
+        cyapa_empty_pip_output_data(cyapa, NULL, NULL, NULL);
+
+        memset(cmd, 0, sizeof(cmd));
+        app_cmd_head = (struct gen5_app_cmd_head *)cmd;
+        put_unaligned_le16(GEN5_OUTPUT_REPORT_ADDR, &app_cmd_head->addr);
+        put_unaligned_le16(sizeof(cmd) - 2, &app_cmd_head->length);
+        app_cmd_head->report_id = GEN5_APP_CMD_REPORT_ID;
+        app_cmd_head->cmd_code = GEN5_CMD_CALIBRATE;
+        app_cmd_head->parameter_data[0] = calibrate_sensing_mode_type;
+        resp_len = sizeof(resp_data);
+        error = cyapa_i2c_pip_cmd_irq_sync(cyapa,
+                        cmd, sizeof(cmd),
+                        resp_data, &resp_len,
+                        5000, cyapa_gen5_sort_tsg_pip_app_resp_data, true);
+        if (error || !VALID_CMD_RESP_HEADER(resp_data, GEN5_CMD_CALIBRATE) ||
+                        !GEN5_CMD_COMPLETE_SUCCESS(resp_data[5]))
+                return error < 0 ? error : -EAGAIN;
+
+        return 0;
+}
+
+static ssize_t cyapa_gen5_do_calibrate(struct device *dev,
+                                     struct device_attribute *attr,
+                                     const char *buf, size_t count)
+{
+        struct cyapa *cyapa = dev_get_drvdata(dev);
+        int error, calibrate_error;
+
+        /* 1. Suspend Scanning*/
+        error = cyapa_gen5_suspend_scanning(cyapa);
+        if (error)
+                return error;
+
+        /* 2. Do mutual capacitance fine calibrate. */
+        calibrate_error = cyapa_gen5_calibrate_pwcs(cyapa,
+                                CYAPA_SENSING_MODE_MUTUAL_CAP_FINE);
+        if (calibrate_error)
+                goto resume_scanning;
+
+        /* 3. Do self capacitance calibrate. */
+        calibrate_error = cyapa_gen5_calibrate_pwcs(cyapa,
+                                CYAPA_SENSING_MODE_SELF_CAP);
+        if (calibrate_error)
+                goto resume_scanning;
+
+resume_scanning:
+        /* 4. Resume Scanning*/
+        error = cyapa_gen5_resume_scanning(cyapa);
+        if (error || calibrate_error)
+                return error ? error : calibrate_error;
+
+        return count;
+}
+
 static s32 twos_complement_to_s32(s32 value, int num_bits)
 {
         if (value >> (num_bits - 1))
@@ -2696,6 +2760,7 @@ const struct cyapa_dev_ops cyapa_gen5_ops = {
         .bl_deactivate = cyapa_gen5_bl_deactivate,
 
         .show_baseline = cyapa_gen5_show_baseline,
+        .calibrate_store = cyapa_gen5_do_calibrate,
 
         .initialize = cyapa_gen5_initialize,