1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * Copyright (C) 2018 Spreadtrum Communications Inc.
4  */
5 
6 #include <linux/device.h>
7 #include <linux/input.h>
8 #include <linux/mod_devicetable.h>
9 #include <linux/module.h>
10 #include <linux/platform_device.h>
11 #include <linux/property.h>
12 #include <linux/regmap.h>
13 #include <linux/workqueue.h>
14 
15 #define CUR_DRV_CAL_SEL			GENMASK(13, 12)
16 #define SLP_LDOVIBR_PD_EN		BIT(9)
17 #define LDO_VIBR_PD			BIT(8)
18 #define SC2730_CUR_DRV_CAL_SEL		0
19 #define SC2730_SLP_LDOVIBR_PD_EN	BIT(14)
20 #define SC2730_LDO_VIBR_PD		BIT(13)
21 
22 struct sc27xx_vibra_data {
23 	u32 cur_drv_cal_sel;
24 	u32 slp_pd_en;
25 	u32 ldo_pd;
26 };
27 
28 struct vibra_info {
29 	struct input_dev	*input_dev;
30 	struct work_struct	play_work;
31 	struct regmap		*regmap;
32 	const struct sc27xx_vibra_data *data;
33 	u32			base;
34 	u32			strength;
35 	bool			enabled;
36 };
37 
38 static const struct sc27xx_vibra_data sc2731_data = {
39 	.cur_drv_cal_sel = CUR_DRV_CAL_SEL,
40 	.slp_pd_en = SLP_LDOVIBR_PD_EN,
41 	.ldo_pd = LDO_VIBR_PD,
42 };
43 
44 static const struct sc27xx_vibra_data sc2730_data = {
45 	.cur_drv_cal_sel = SC2730_CUR_DRV_CAL_SEL,
46 	.slp_pd_en = SC2730_SLP_LDOVIBR_PD_EN,
47 	.ldo_pd = SC2730_LDO_VIBR_PD,
48 };
49 
50 static const struct sc27xx_vibra_data sc2721_data = {
51 	.cur_drv_cal_sel = CUR_DRV_CAL_SEL,
52 	.slp_pd_en = SLP_LDOVIBR_PD_EN,
53 	.ldo_pd = LDO_VIBR_PD,
54 };
55 
56 static void sc27xx_vibra_set(struct vibra_info *info, bool on)
57 {
58 	const struct sc27xx_vibra_data *data = info->data;
59 	if (on) {
60 		regmap_update_bits(info->regmap, info->base, data->ldo_pd, 0);
61 		regmap_update_bits(info->regmap, info->base,
62 				   data->slp_pd_en, 0);
63 		info->enabled = true;
64 	} else {
65 		regmap_update_bits(info->regmap, info->base, data->ldo_pd,
66 				   data->ldo_pd);
67 		regmap_update_bits(info->regmap, info->base,
68 				   data->slp_pd_en, data->slp_pd_en);
69 		info->enabled = false;
70 	}
71 }
72 
73 static int sc27xx_vibra_hw_init(struct vibra_info *info)
74 {
75 	const struct sc27xx_vibra_data *data = info->data;
76 
77 	if (!data->cur_drv_cal_sel)
78 		return 0;
79 
80 	return regmap_update_bits(info->regmap, info->base,
81 				  data->cur_drv_cal_sel, 0);
82 }
83 
84 static void sc27xx_vibra_play_work(struct work_struct *work)
85 {
86 	struct vibra_info *info = container_of(work, struct vibra_info,
87 					       play_work);
88 
89 	if (info->strength && !info->enabled)
90 		sc27xx_vibra_set(info, true);
91 	else if (info->strength == 0 && info->enabled)
92 		sc27xx_vibra_set(info, false);
93 }
94 
95 static int sc27xx_vibra_play(struct input_dev *input, void *data,
96 			     struct ff_effect *effect)
97 {
98 	struct vibra_info *info = input_get_drvdata(input);
99 
100 	info->strength = effect->u.rumble.weak_magnitude;
101 	schedule_work(&info->play_work);
102 
103 	return 0;
104 }
105 
106 static void sc27xx_vibra_close(struct input_dev *input)
107 {
108 	struct vibra_info *info = input_get_drvdata(input);
109 
110 	cancel_work_sync(&info->play_work);
111 	if (info->enabled)
112 		sc27xx_vibra_set(info, false);
113 }
114 
115 static int sc27xx_vibra_probe(struct platform_device *pdev)
116 {
117 	struct vibra_info *info;
118 	const struct sc27xx_vibra_data *data;
119 	int error;
120 
121 	data = device_get_match_data(&pdev->dev);
122 	if (!data) {
123 		dev_err(&pdev->dev, "no matching driver data found\n");
124 		return -EINVAL;
125 	}
126 
127 	info = devm_kzalloc(&pdev->dev, sizeof(*info), GFP_KERNEL);
128 	if (!info)
129 		return -ENOMEM;
130 
131 	info->regmap = dev_get_regmap(pdev->dev.parent, NULL);
132 	if (!info->regmap) {
133 		dev_err(&pdev->dev, "failed to get vibrator regmap.\n");
134 		return -ENODEV;
135 	}
136 
137 	error = device_property_read_u32(&pdev->dev, "reg", &info->base);
138 	if (error) {
139 		dev_err(&pdev->dev, "failed to get vibrator base address.\n");
140 		return error;
141 	}
142 
143 	info->input_dev = devm_input_allocate_device(&pdev->dev);
144 	if (!info->input_dev) {
145 		dev_err(&pdev->dev, "failed to allocate input device.\n");
146 		return -ENOMEM;
147 	}
148 
149 	info->input_dev->name = "sc27xx:vibrator";
150 	info->input_dev->id.version = 0;
151 	info->input_dev->close = sc27xx_vibra_close;
152 	info->data = data;
153 
154 	input_set_drvdata(info->input_dev, info);
155 	input_set_capability(info->input_dev, EV_FF, FF_RUMBLE);
156 	INIT_WORK(&info->play_work, sc27xx_vibra_play_work);
157 	info->enabled = false;
158 
159 	error = sc27xx_vibra_hw_init(info);
160 	if (error) {
161 		dev_err(&pdev->dev, "failed to initialize the vibrator.\n");
162 		return error;
163 	}
164 
165 	error = input_ff_create_memless(info->input_dev, NULL,
166 					sc27xx_vibra_play);
167 	if (error) {
168 		dev_err(&pdev->dev, "failed to register vibrator to FF.\n");
169 		return error;
170 	}
171 
172 	error = input_register_device(info->input_dev);
173 	if (error) {
174 		dev_err(&pdev->dev, "failed to register input device.\n");
175 		return error;
176 	}
177 
178 	return 0;
179 }
180 
181 static const struct of_device_id sc27xx_vibra_of_match[] = {
182 	{ .compatible = "sprd,sc2721-vibrator", .data = &sc2721_data },
183 	{ .compatible = "sprd,sc2730-vibrator", .data = &sc2730_data },
184 	{ .compatible = "sprd,sc2731-vibrator", .data = &sc2731_data },
185 	{}
186 };
187 MODULE_DEVICE_TABLE(of, sc27xx_vibra_of_match);
188 
189 static struct platform_driver sc27xx_vibra_driver = {
190 	.driver = {
191 		.name = "sc27xx-vibrator",
192 		.of_match_table = sc27xx_vibra_of_match,
193 	},
194 	.probe = sc27xx_vibra_probe,
195 };
196 
197 module_platform_driver(sc27xx_vibra_driver);
198 
199 MODULE_DESCRIPTION("Spreadtrum SC27xx Vibrator Driver");
200 MODULE_LICENSE("GPL v2");
201 MODULE_AUTHOR("Xiaotong Lu <xiaotong.lu@spreadtrum.com>");
202