hc
2024-03-22 a0752693d998599af469473b8dc239ef973a012f
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
/*
 *
 * Copyright (C) Fuzhou Rockchip Electronics Co.Ltd
 *
 * Authors: Bin Yang <yangbin@rock-chips.com>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2  of
 * the License as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 */
 
#include <linux/interrupt.h>
#include <linux/i2c.h>
#include <linux/slab.h>
#include <linux/irq.h>
#include <linux/miscdevice.h>
#include <linux/gpio.h>
#include <linux/uaccess.h>
#include <linux/atomic.h>
#include <linux/delay.h>
#include <linux/input.h>
#include <linux/workqueue.h>
#include <linux/freezer.h>
#include <linux/of_gpio.h>
#ifdef CONFIG_HAS_EARLYSUSPEND
#include <linux/earlysuspend.h>
#endif
#include <linux/sensor-dev.h>
 
#define    STK_STATE    0x00
#define    PS_CTRL        0x01
#define    ALS_CTRL    0x02
#define    LED_CTRL    0x03
#define    INT_CTRL    0x04
#define    STK_WAIT    0x05
#define    THDH1_PS    0x06
#define    THDH2_PS    0x07
#define    THDL1_PS    0x08
#define    THDL2_PS    0x09
#define    THDH1_ALS    0x0A
#define    THDH2_ALS    0x0B
#define    THDL1_ALS    0x0C
#define    THDL2_ALS    0x0D
#define    STK_FLAG    0x10
#define    DATA1_PS    0x11
#define    DATA2_PS    0x12
#define    DATA1_ALS    0x13
#define    DATA2_ALS    0x14
#define    DATA1_OFFSET    0x15
#define    DATA2_OFFSET    0x16
#define    DATA1_IR    0x17
#define    DATA2_IR    0x18
#define    DATA1_GS0    0x24
#define    DATA2_GS0    0x25
#define    DATA1_GS1    0x26
#define    DATA2_GS1    0x27
#define    STKPDT_ID    0x3E
#define    SOFT_RESET    0x80
 
/* STK_STATE    0x00 */
#define    PS_DISABLE    (0 << 0)
#define    PS_ENABLE    (1 << 0)
#define    ALS_DISABLE    (0 << 1)
#define    ALS_ENABLE    (1 << 1)
#define    WAIT_DISABLE    (0 << 2)
#define    WAIT_ENABLE    (1 << 2)
#define    IRO_DISABLE    (0 << 4)
#define    IRO_ENABLE    (1 << 4)
#define    ASO_DISABLE    (0 << 5)
#define    ASO_ENABLE    (1 << 5)
#define    AK_DISABLE    (0 << 6)
#define    AK_ENABLE    (1 << 6)
#define    IRS_DISABLE    (0 << 7)
#define    IRS_ENABLE    (1 << 7)
 
/* PS/GS_CTRL 0x01 */
#define    PS_REFT_MS    (1 << 0)    /* [3:0] 0.185 ms ,  default value is 0.37ms */
#define    PS_GAIN_1G    (0 << 4)
#define    PS_GAIN_4G    (1 << 4)
#define    PS_GAIN_16G    (2 << 4)
#define    PS_GAIN_64G    (3 << 4)
#define    PS_PRS_1T    (0 << 6)
#define    PS_PRS_4T    (1 << 6)
 
/* ALS_CTRL 0x02 */
#define    ALS_REFT_MS    (9 << 0)/* [3:0] 0.185 ms,  default value is 94.85ms */
#define    ALS_GAIN_1G    (0 << 4)
#define    ALS_GAIN_4G    (1 << 4)
#define    ALS_GAIN_16G    (2 << 4)
#define    ALS_GAIN_64G    (3 << 4)
#define    ALS_PRS_1T    (0 << 6)
#define    ALS_PRS_4T    (1 << 6)
 
/* LED_CTRL 0x03 */
#define    LED_REFT_US    0x3F    /* [5:0] 2.89us ,  default value is 0.185ms */
#define    LED_CUR_12MA    (0 << 6)
#define    LED_CUR_25MA    (1 << 6)
#define    LED_CUR_50MA    (2 << 6)
#define    LED_CUR_100MA    (3 << 6)
 
/* INT 0x04 */
#define    PS_INT_DISABLE        0xF8
#define    PS_INT_ENABLE        (1 << 0)
#define    PS_INT_ENABLE_FLGNFH    (2 << 0)
#define    PS_INT_ENABLE_FLGNFL    (3 << 0)
#define    PS_INT_MODE_ENABLE    (4 << 0)
#define    PS_INT_ENABLE_THL    (5 << 0)
#define    PS_INT_ENABLE_THH    (6 << 0)
#define    PS_INT_ENABLE_THHL    (7 << 0)
#define    ALS_INT_DISABLE        (0 << 3)
#define    ALS_INT_ENABLE        (1 << 3)
#define    INT_CTRL_PS_OR_LS    (0 << 7)
#define    INT_CTRL_PS_AND_LS    (1 << 7)
 
/* FLAG 0x10 */
#define    STK_FLAG_NF    (1 << 0)
#define    STK_FLAG_IR_RDY    (1 << 1)
#define    STK_FLAG_OUI    (1 << 2)
#define    STK_FLAG_PSINT    (1 << 4)
#define    STK_FLAG_ALSINT    (1 << 5)
#define    STK_FLAG_PSDR    (1 << 6)
#define    STK_FLAG_ALSDR    (1 << 7)
 
static int ps_threshold_low;
static int ps_threshold_high;
 
static int sensor_active(struct i2c_client *client, int enable, int rate)
{
   struct sensor_private_data *sensor =
       (struct sensor_private_data *)i2c_get_clientdata(client);
   int result = 0;
   int status = 0;
 
   sensor->ops->ctrl_data = sensor_read_reg(client, sensor->ops->ctrl_reg);
   if (!enable) {
       status = ~PS_ENABLE;
       sensor->ops->ctrl_data &= status;
   } else {
       status = PS_ENABLE;
       sensor->ops->ctrl_data |= status;
   }
 
   DBG
   ("%s:reg=0x%x, reg_ctrl=0x%x, enable=%d\n",
   __func__,
   sensor->ops->ctrl_reg,
   sensor->ops->ctrl_data,
   enable);
 
   result = sensor_write_reg
           (client, sensor->ops->ctrl_reg,
           sensor->ops->ctrl_data);
   if (result)
       dev_err(&client->dev, "%s:fail to active sensor\n", __func__);
 
   return result;
}
 
static int sensor_init(struct i2c_client *client)
{
   struct sensor_private_data *sensor =
       (struct sensor_private_data *)i2c_get_clientdata(client);
   struct device_node *np = client->dev.of_node;
   int ps_val = 0;
   int ret = 0;
   int result = 0;
   int val = 0;
 
   result = sensor->ops->active(client, 0, 0);
   if (result) {
       dev_err(&client->dev, "%s:sensor active fail\n", __func__);
       return result;
   }
   sensor->status_cur = SENSOR_OFF;
 
   ret = of_property_read_u32(np, "ps_threshold_low", &ps_val);
   if (ret)
       dev_err(&client->dev, "%s:Unable to read ps_threshold_low\n",
           __func__);
 
   ps_threshold_low = ps_val;
   result = sensor_write_reg(client, THDL1_PS, (unsigned char)ps_val);
   if (result) {
       dev_err(&client->dev, "%s:write THDL1_PS fail\n", __func__);
       return result;
   }
   result = sensor_write_reg
           (client, THDL2_PS,
           (unsigned char)(ps_val >> 8));
   if (result) {
       dev_err(&client->dev, "%s:write THDL1_PS fail\n", __func__);
       return result;
   }
 
   ret = of_property_read_u32(np, "ps_threshold_high", &ps_val);
   if (ret)
       dev_err(&client->dev, "%s:Unable to read ps_threshold_high\n",
           __func__);
 
   ps_threshold_high = ps_val;
   result = sensor_write_reg(client, THDH1_PS, (unsigned char)ps_val);
   if (result) {
       dev_err(&client->dev, "%s:write THDH1_PS fail\n",
           __func__);
       return result;
   }
   result = sensor_write_reg
       (client, THDH2_PS, (unsigned char)(ps_val >> 8));
   if (result) {
       dev_err(&client->dev, "%s:write THDH1_PS fail\n",
           __func__);
       return result;
   }
 
   ret = of_property_read_u32(np, "ps_ctrl_gain", &ps_val);
   if (ret)
       dev_err(&client->dev, "%s:Unable to read ps_ctrl_gain\n",
           __func__);
 
   result = sensor_write_reg
       (client, PS_CTRL,
       (unsigned char)((ps_val << 4) | PS_REFT_MS));
   if (result) {
       dev_err(&client->dev, "%s:write PS_CTRL fail\n", __func__);
       return result;
   }
 
   ret = of_property_read_u32(np, "ps_led_current", &ps_val);
   if (ret)
       dev_err(&client->dev, "%s:Unable to read ps_led_current\n",
           __func__);
 
   result = sensor_write_reg
       (client, LED_CTRL,
       (unsigned char)((ps_val << 6) | LED_REFT_US));
   if (result) {
       dev_err(&client->dev, "%s:write LED_CTRL fail\n", __func__);
       return result;
   }
 
   val = sensor_read_reg(client, INT_CTRL);
   val &= ~INT_CTRL_PS_AND_LS;
   if (sensor->pdata->irq_enable)
       val |= PS_INT_ENABLE_FLGNFL;
   else
       val &= PS_INT_DISABLE;
   result = sensor_write_reg(client, INT_CTRL, val);
   if (result) {
       dev_err(&client->dev, "%s:write INT_CTRL fail\n", __func__);
       return result;
   }
 
   return result;
}
 
static int stk3410_get_ps_value(int ps)
{
   int index = 0;
   static int val = 1;
 
   if (ps > ps_threshold_high) {
       index = 0;
       val = 0;
   } else if (ps < ps_threshold_low) {
       index = 1;
       val = 1;
   } else {
       index = val;
   }
 
   return index;
}
 
static int sensor_report_value(struct i2c_client *client)
{
   struct sensor_private_data *sensor =
       (struct sensor_private_data *)i2c_get_clientdata(client);
   int result = 0;
   int value = 0;
   char buffer[2] = {0};
   int index = 1;
 
   if (sensor->ops->read_len < 2) {
       dev_err(&client->dev, "%s:length is error, len=%d\n", __func__,
           sensor->ops->read_len);
       return -1;
   }
 
   buffer[0] = sensor->ops->read_reg;
   result = sensor_rx_data(client, buffer, sensor->ops->read_len);
   if (result) {
       dev_err(&client->dev, "%s:sensor read data fail\n", __func__);
       return result;
   }
   value = (buffer[0] << 8) | buffer[1];
   index = stk3410_get_ps_value(value);
   input_report_abs(sensor->input_dev, ABS_DISTANCE, index);
   input_sync(sensor->input_dev);
 
   DBG("%s:%s value=0x%x\n", __func__, sensor->ops->name, value);
 
   if (sensor->pdata->irq_enable && sensor->ops->int_status_reg) {
       value = sensor_read_reg(client, sensor->ops->int_status_reg);
       if (value & STK_FLAG_NF) {
           value &= ~STK_FLAG_NF;
           result = sensor_write_reg
               (client, sensor->ops->int_status_reg,
               value);
           if (result) {
               dev_err(&client->dev, "%s:write status reg error\n",
                   __func__);
               return result;
           }
       }
   }
 
   return result;
}
 
static struct sensor_operate psensor_stk3410_ops = {
   .name            = "ps_stk3410",
   .type            = SENSOR_TYPE_PROXIMITY,
   .id_i2c            = PROXIMITY_ID_STK3410,
   .read_reg        = DATA1_PS,
   .read_len        = 2,
   .id_reg            = SENSOR_UNKNOW_DATA,
   .id_data        = SENSOR_UNKNOW_DATA,
   .precision        = 16,
   .ctrl_reg        = STK_STATE,
   .int_status_reg    = STK_FLAG,
   .range            = {100, 65535},
   .brightness        = {10, 255},
   .trig            = IRQF_TRIGGER_LOW | IRQF_ONESHOT | IRQF_SHARED,
   .active            = sensor_active,
   .init            = sensor_init,
   .report            = sensor_report_value,
};
 
static int proximity_stk3410_probe(struct i2c_client *client,
                  const struct i2c_device_id *devid)
{
   return sensor_register_device(client, NULL, devid, &psensor_stk3410_ops);
}
 
static int proximity_stk3410_remove(struct i2c_client *client)
{
   return sensor_unregister_device(client, NULL, &psensor_stk3410_ops);
}
 
static const struct i2c_device_id proximity_stk3410_id[] = {
   {"ps_stk3410", PROXIMITY_ID_STK3410},
   {}
};
 
static struct i2c_driver proximity_stk3410_driver = {
   .probe = proximity_stk3410_probe,
   .remove = proximity_stk3410_remove,
   .shutdown = sensor_shutdown,
   .id_table = proximity_stk3410_id,
   .driver = {
       .name = "proximity_stk3410",
   #ifdef CONFIG_PM
       .pm = &sensor_pm_ops,
   #endif
   },
};
 
module_i2c_driver(proximity_stk3410_driver);
 
MODULE_AUTHOR("Bin Yang <yangbin@rock-chips.com>");
MODULE_DESCRIPTION("stk3410 proximity driver");
MODULE_LICENSE("GPL");