1*9e4c596bSVincent Huang // SPDX-License-Identifier: GPL-2.0-only
2*9e4c596bSVincent Huang /*
3*9e4c596bSVincent Huang * Copyright (c) 2012-2020 Synaptics Incorporated
4*9e4c596bSVincent Huang */
5*9e4c596bSVincent Huang
6*9e4c596bSVincent Huang #include <linux/kernel.h>
7*9e4c596bSVincent Huang #include <linux/rmi.h>
8*9e4c596bSVincent Huang #include <linux/input.h>
9*9e4c596bSVincent Huang #include <linux/slab.h>
10*9e4c596bSVincent Huang #include "rmi_driver.h"
11*9e4c596bSVincent Huang
12*9e4c596bSVincent Huang #define RMI_F3A_MAX_GPIO_COUNT 128
13*9e4c596bSVincent Huang #define RMI_F3A_MAX_REG_SIZE DIV_ROUND_UP(RMI_F3A_MAX_GPIO_COUNT, 8)
14*9e4c596bSVincent Huang
15*9e4c596bSVincent Huang /* Defs for Query 0 */
16*9e4c596bSVincent Huang #define RMI_F3A_GPIO_COUNT 0x7F
17*9e4c596bSVincent Huang
18*9e4c596bSVincent Huang #define RMI_F3A_DATA_REGS_MAX_SIZE RMI_F3A_MAX_REG_SIZE
19*9e4c596bSVincent Huang
20*9e4c596bSVincent Huang #define TRACKSTICK_RANGE_START 3
21*9e4c596bSVincent Huang #define TRACKSTICK_RANGE_END 6
22*9e4c596bSVincent Huang
23*9e4c596bSVincent Huang struct f3a_data {
24*9e4c596bSVincent Huang /* Query Data */
25*9e4c596bSVincent Huang u8 gpio_count;
26*9e4c596bSVincent Huang
27*9e4c596bSVincent Huang u8 register_count;
28*9e4c596bSVincent Huang
29*9e4c596bSVincent Huang u8 data_regs[RMI_F3A_DATA_REGS_MAX_SIZE];
30*9e4c596bSVincent Huang u16 *gpio_key_map;
31*9e4c596bSVincent Huang
32*9e4c596bSVincent Huang struct input_dev *input;
33*9e4c596bSVincent Huang
34*9e4c596bSVincent Huang struct rmi_function *f03;
35*9e4c596bSVincent Huang bool trackstick_buttons;
36*9e4c596bSVincent Huang };
37*9e4c596bSVincent Huang
rmi_f3a_report_button(struct rmi_function * fn,struct f3a_data * f3a,unsigned int button)38*9e4c596bSVincent Huang static void rmi_f3a_report_button(struct rmi_function *fn,
39*9e4c596bSVincent Huang struct f3a_data *f3a, unsigned int button)
40*9e4c596bSVincent Huang {
41*9e4c596bSVincent Huang u16 key_code = f3a->gpio_key_map[button];
42*9e4c596bSVincent Huang bool key_down = !(f3a->data_regs[0] & BIT(button));
43*9e4c596bSVincent Huang
44*9e4c596bSVincent Huang if (f3a->trackstick_buttons &&
45*9e4c596bSVincent Huang button >= TRACKSTICK_RANGE_START &&
46*9e4c596bSVincent Huang button <= TRACKSTICK_RANGE_END) {
47*9e4c596bSVincent Huang rmi_f03_overwrite_button(f3a->f03, key_code, key_down);
48*9e4c596bSVincent Huang } else {
49*9e4c596bSVincent Huang rmi_dbg(RMI_DEBUG_FN, &fn->dev,
50*9e4c596bSVincent Huang "%s: call input report key (0x%04x) value (0x%02x)",
51*9e4c596bSVincent Huang __func__, key_code, key_down);
52*9e4c596bSVincent Huang input_report_key(f3a->input, key_code, key_down);
53*9e4c596bSVincent Huang }
54*9e4c596bSVincent Huang }
55*9e4c596bSVincent Huang
rmi_f3a_attention(int irq,void * ctx)56*9e4c596bSVincent Huang static irqreturn_t rmi_f3a_attention(int irq, void *ctx)
57*9e4c596bSVincent Huang {
58*9e4c596bSVincent Huang struct rmi_function *fn = ctx;
59*9e4c596bSVincent Huang struct f3a_data *f3a = dev_get_drvdata(&fn->dev);
60*9e4c596bSVincent Huang struct rmi_driver_data *drvdata = dev_get_drvdata(&fn->rmi_dev->dev);
61*9e4c596bSVincent Huang int error;
62*9e4c596bSVincent Huang int i;
63*9e4c596bSVincent Huang
64*9e4c596bSVincent Huang if (drvdata->attn_data.data) {
65*9e4c596bSVincent Huang if (drvdata->attn_data.size < f3a->register_count) {
66*9e4c596bSVincent Huang dev_warn(&fn->dev,
67*9e4c596bSVincent Huang "F3A interrupted, but data is missing\n");
68*9e4c596bSVincent Huang return IRQ_HANDLED;
69*9e4c596bSVincent Huang }
70*9e4c596bSVincent Huang memcpy(f3a->data_regs, drvdata->attn_data.data,
71*9e4c596bSVincent Huang f3a->register_count);
72*9e4c596bSVincent Huang drvdata->attn_data.data += f3a->register_count;
73*9e4c596bSVincent Huang drvdata->attn_data.size -= f3a->register_count;
74*9e4c596bSVincent Huang } else {
75*9e4c596bSVincent Huang error = rmi_read_block(fn->rmi_dev, fn->fd.data_base_addr,
76*9e4c596bSVincent Huang f3a->data_regs, f3a->register_count);
77*9e4c596bSVincent Huang if (error) {
78*9e4c596bSVincent Huang dev_err(&fn->dev,
79*9e4c596bSVincent Huang "%s: Failed to read F3a data registers: %d\n",
80*9e4c596bSVincent Huang __func__, error);
81*9e4c596bSVincent Huang return IRQ_RETVAL(error);
82*9e4c596bSVincent Huang }
83*9e4c596bSVincent Huang }
84*9e4c596bSVincent Huang
85*9e4c596bSVincent Huang for (i = 0; i < f3a->gpio_count; i++)
86*9e4c596bSVincent Huang if (f3a->gpio_key_map[i] != KEY_RESERVED)
87*9e4c596bSVincent Huang rmi_f3a_report_button(fn, f3a, i);
88*9e4c596bSVincent Huang if (f3a->trackstick_buttons)
89*9e4c596bSVincent Huang rmi_f03_commit_buttons(f3a->f03);
90*9e4c596bSVincent Huang
91*9e4c596bSVincent Huang return IRQ_HANDLED;
92*9e4c596bSVincent Huang }
93*9e4c596bSVincent Huang
rmi_f3a_config(struct rmi_function * fn)94*9e4c596bSVincent Huang static int rmi_f3a_config(struct rmi_function *fn)
95*9e4c596bSVincent Huang {
96*9e4c596bSVincent Huang struct f3a_data *f3a = dev_get_drvdata(&fn->dev);
97*9e4c596bSVincent Huang struct rmi_driver *drv = fn->rmi_dev->driver;
98*9e4c596bSVincent Huang const struct rmi_device_platform_data *pdata =
99*9e4c596bSVincent Huang rmi_get_platform_data(fn->rmi_dev);
100*9e4c596bSVincent Huang
101*9e4c596bSVincent Huang if (!f3a)
102*9e4c596bSVincent Huang return 0;
103*9e4c596bSVincent Huang
104*9e4c596bSVincent Huang if (pdata->gpio_data.trackstick_buttons) {
105*9e4c596bSVincent Huang /* Try [re-]establish link to F03. */
106*9e4c596bSVincent Huang f3a->f03 = rmi_find_function(fn->rmi_dev, 0x03);
107*9e4c596bSVincent Huang f3a->trackstick_buttons = f3a->f03 != NULL;
108*9e4c596bSVincent Huang }
109*9e4c596bSVincent Huang
110*9e4c596bSVincent Huang drv->set_irq_bits(fn->rmi_dev, fn->irq_mask);
111*9e4c596bSVincent Huang
112*9e4c596bSVincent Huang return 0;
113*9e4c596bSVincent Huang }
114*9e4c596bSVincent Huang
rmi_f3a_is_valid_button(int button,struct f3a_data * f3a,u8 * query1_regs,u8 * ctrl1_regs)115*9e4c596bSVincent Huang static bool rmi_f3a_is_valid_button(int button, struct f3a_data *f3a,
116*9e4c596bSVincent Huang u8 *query1_regs, u8 *ctrl1_regs)
117*9e4c596bSVincent Huang {
118*9e4c596bSVincent Huang /* gpio exist && direction input */
119*9e4c596bSVincent Huang return (query1_regs[0] & BIT(button)) && !(ctrl1_regs[0] & BIT(button));
120*9e4c596bSVincent Huang }
121*9e4c596bSVincent Huang
rmi_f3a_map_gpios(struct rmi_function * fn,struct f3a_data * f3a,u8 * query1_regs,u8 * ctrl1_regs)122*9e4c596bSVincent Huang static int rmi_f3a_map_gpios(struct rmi_function *fn, struct f3a_data *f3a,
123*9e4c596bSVincent Huang u8 *query1_regs, u8 *ctrl1_regs)
124*9e4c596bSVincent Huang {
125*9e4c596bSVincent Huang const struct rmi_device_platform_data *pdata =
126*9e4c596bSVincent Huang rmi_get_platform_data(fn->rmi_dev);
127*9e4c596bSVincent Huang struct input_dev *input = f3a->input;
128*9e4c596bSVincent Huang unsigned int button = BTN_LEFT;
129*9e4c596bSVincent Huang unsigned int trackstick_button = BTN_LEFT;
130*9e4c596bSVincent Huang bool button_mapped = false;
131*9e4c596bSVincent Huang int i;
132*9e4c596bSVincent Huang int button_count = min_t(u8, f3a->gpio_count, TRACKSTICK_RANGE_END);
133*9e4c596bSVincent Huang
134*9e4c596bSVincent Huang f3a->gpio_key_map = devm_kcalloc(&fn->dev,
135*9e4c596bSVincent Huang button_count,
136*9e4c596bSVincent Huang sizeof(f3a->gpio_key_map[0]),
137*9e4c596bSVincent Huang GFP_KERNEL);
138*9e4c596bSVincent Huang if (!f3a->gpio_key_map) {
139*9e4c596bSVincent Huang dev_err(&fn->dev, "Failed to allocate gpio map memory.\n");
140*9e4c596bSVincent Huang return -ENOMEM;
141*9e4c596bSVincent Huang }
142*9e4c596bSVincent Huang
143*9e4c596bSVincent Huang for (i = 0; i < button_count; i++) {
144*9e4c596bSVincent Huang if (!rmi_f3a_is_valid_button(i, f3a, query1_regs, ctrl1_regs))
145*9e4c596bSVincent Huang continue;
146*9e4c596bSVincent Huang
147*9e4c596bSVincent Huang if (pdata->gpio_data.trackstick_buttons &&
148*9e4c596bSVincent Huang i >= TRACKSTICK_RANGE_START &&
149*9e4c596bSVincent Huang i < TRACKSTICK_RANGE_END) {
150*9e4c596bSVincent Huang f3a->gpio_key_map[i] = trackstick_button++;
151*9e4c596bSVincent Huang } else if (!pdata->gpio_data.buttonpad || !button_mapped) {
152*9e4c596bSVincent Huang f3a->gpio_key_map[i] = button;
153*9e4c596bSVincent Huang input_set_capability(input, EV_KEY, button++);
154*9e4c596bSVincent Huang button_mapped = true;
155*9e4c596bSVincent Huang }
156*9e4c596bSVincent Huang }
157*9e4c596bSVincent Huang input->keycode = f3a->gpio_key_map;
158*9e4c596bSVincent Huang input->keycodesize = sizeof(f3a->gpio_key_map[0]);
159*9e4c596bSVincent Huang input->keycodemax = f3a->gpio_count;
160*9e4c596bSVincent Huang
161*9e4c596bSVincent Huang if (pdata->gpio_data.buttonpad || (button - BTN_LEFT == 1))
162*9e4c596bSVincent Huang __set_bit(INPUT_PROP_BUTTONPAD, input->propbit);
163*9e4c596bSVincent Huang
164*9e4c596bSVincent Huang return 0;
165*9e4c596bSVincent Huang }
166*9e4c596bSVincent Huang
rmi_f3a_initialize(struct rmi_function * fn,struct f3a_data * f3a)167*9e4c596bSVincent Huang static int rmi_f3a_initialize(struct rmi_function *fn, struct f3a_data *f3a)
168*9e4c596bSVincent Huang {
169*9e4c596bSVincent Huang u8 query1[RMI_F3A_MAX_REG_SIZE];
170*9e4c596bSVincent Huang u8 ctrl1[RMI_F3A_MAX_REG_SIZE];
171*9e4c596bSVincent Huang u8 buf;
172*9e4c596bSVincent Huang int error;
173*9e4c596bSVincent Huang
174*9e4c596bSVincent Huang error = rmi_read(fn->rmi_dev, fn->fd.query_base_addr, &buf);
175*9e4c596bSVincent Huang if (error < 0) {
176*9e4c596bSVincent Huang dev_err(&fn->dev, "Failed to read general info register: %d\n",
177*9e4c596bSVincent Huang error);
178*9e4c596bSVincent Huang return -ENODEV;
179*9e4c596bSVincent Huang }
180*9e4c596bSVincent Huang
181*9e4c596bSVincent Huang f3a->gpio_count = buf & RMI_F3A_GPIO_COUNT;
182*9e4c596bSVincent Huang f3a->register_count = DIV_ROUND_UP(f3a->gpio_count, 8);
183*9e4c596bSVincent Huang
184*9e4c596bSVincent Huang /* Query1 -> gpio exist */
185*9e4c596bSVincent Huang error = rmi_read_block(fn->rmi_dev, fn->fd.query_base_addr + 1,
186*9e4c596bSVincent Huang query1, f3a->register_count);
187*9e4c596bSVincent Huang if (error) {
188*9e4c596bSVincent Huang dev_err(&fn->dev, "Failed to read query1 register\n");
189*9e4c596bSVincent Huang return error;
190*9e4c596bSVincent Huang }
191*9e4c596bSVincent Huang
192*9e4c596bSVincent Huang /* Ctrl1 -> gpio direction */
193*9e4c596bSVincent Huang error = rmi_read_block(fn->rmi_dev, fn->fd.control_base_addr + 1,
194*9e4c596bSVincent Huang ctrl1, f3a->register_count);
195*9e4c596bSVincent Huang if (error) {
196*9e4c596bSVincent Huang dev_err(&fn->dev, "Failed to read control1 register\n");
197*9e4c596bSVincent Huang return error;
198*9e4c596bSVincent Huang }
199*9e4c596bSVincent Huang
200*9e4c596bSVincent Huang error = rmi_f3a_map_gpios(fn, f3a, query1, ctrl1);
201*9e4c596bSVincent Huang if (error)
202*9e4c596bSVincent Huang return error;
203*9e4c596bSVincent Huang
204*9e4c596bSVincent Huang return 0;
205*9e4c596bSVincent Huang }
206*9e4c596bSVincent Huang
rmi_f3a_probe(struct rmi_function * fn)207*9e4c596bSVincent Huang static int rmi_f3a_probe(struct rmi_function *fn)
208*9e4c596bSVincent Huang {
209*9e4c596bSVincent Huang struct rmi_device *rmi_dev = fn->rmi_dev;
210*9e4c596bSVincent Huang struct rmi_driver_data *drv_data = dev_get_drvdata(&rmi_dev->dev);
211*9e4c596bSVincent Huang struct f3a_data *f3a;
212*9e4c596bSVincent Huang int error;
213*9e4c596bSVincent Huang
214*9e4c596bSVincent Huang if (!drv_data->input) {
215*9e4c596bSVincent Huang dev_info(&fn->dev, "F3A: no input device found, ignoring\n");
216*9e4c596bSVincent Huang return -ENXIO;
217*9e4c596bSVincent Huang }
218*9e4c596bSVincent Huang
219*9e4c596bSVincent Huang f3a = devm_kzalloc(&fn->dev, sizeof(*f3a), GFP_KERNEL);
220*9e4c596bSVincent Huang if (!f3a)
221*9e4c596bSVincent Huang return -ENOMEM;
222*9e4c596bSVincent Huang
223*9e4c596bSVincent Huang f3a->input = drv_data->input;
224*9e4c596bSVincent Huang
225*9e4c596bSVincent Huang error = rmi_f3a_initialize(fn, f3a);
226*9e4c596bSVincent Huang if (error)
227*9e4c596bSVincent Huang return error;
228*9e4c596bSVincent Huang
229*9e4c596bSVincent Huang dev_set_drvdata(&fn->dev, f3a);
230*9e4c596bSVincent Huang return 0;
231*9e4c596bSVincent Huang }
232*9e4c596bSVincent Huang
233*9e4c596bSVincent Huang struct rmi_function_handler rmi_f3a_handler = {
234*9e4c596bSVincent Huang .driver = {
235*9e4c596bSVincent Huang .name = "rmi4_f3a",
236*9e4c596bSVincent Huang },
237*9e4c596bSVincent Huang .func = 0x3a,
238*9e4c596bSVincent Huang .probe = rmi_f3a_probe,
239*9e4c596bSVincent Huang .config = rmi_f3a_config,
240*9e4c596bSVincent Huang .attention = rmi_f3a_attention,
241*9e4c596bSVincent Huang };
242