1
2
3
4
5
6
7
8
9
10#include <linux/counter.h>
11#include <linux/mfd/stm32-timers.h>
12#include <linux/mod_devicetable.h>
13#include <linux/module.h>
14#include <linux/pinctrl/consumer.h>
15#include <linux/platform_device.h>
16
17#define TIM_CCMR_CCXS (BIT(8) | BIT(0))
18#define TIM_CCMR_MASK (TIM_CCMR_CC1S | TIM_CCMR_CC2S | \
19 TIM_CCMR_IC1F | TIM_CCMR_IC2F)
20#define TIM_CCER_MASK (TIM_CCER_CC1P | TIM_CCER_CC1NP | \
21 TIM_CCER_CC2P | TIM_CCER_CC2NP)
22
23struct stm32_timer_regs {
24 u32 cr1;
25 u32 cnt;
26 u32 smcr;
27 u32 arr;
28};
29
30struct stm32_timer_cnt {
31 struct counter_device counter;
32 struct regmap *regmap;
33 struct clk *clk;
34 u32 ceiling;
35 bool enabled;
36 struct stm32_timer_regs bak;
37};
38
39
40
41
42
43
44
45
46enum stm32_count_function {
47 STM32_COUNT_SLAVE_MODE_DISABLED = -1,
48 STM32_COUNT_ENCODER_MODE_1,
49 STM32_COUNT_ENCODER_MODE_2,
50 STM32_COUNT_ENCODER_MODE_3,
51};
52
53static enum counter_count_function stm32_count_functions[] = {
54 [STM32_COUNT_ENCODER_MODE_1] = COUNTER_COUNT_FUNCTION_QUADRATURE_X2_A,
55 [STM32_COUNT_ENCODER_MODE_2] = COUNTER_COUNT_FUNCTION_QUADRATURE_X2_B,
56 [STM32_COUNT_ENCODER_MODE_3] = COUNTER_COUNT_FUNCTION_QUADRATURE_X4,
57};
58
59static int stm32_count_read(struct counter_device *counter,
60 struct counter_count *count, unsigned long *val)
61{
62 struct stm32_timer_cnt *const priv = counter->priv;
63 u32 cnt;
64
65 regmap_read(priv->regmap, TIM_CNT, &cnt);
66 *val = cnt;
67
68 return 0;
69}
70
71static int stm32_count_write(struct counter_device *counter,
72 struct counter_count *count,
73 const unsigned long val)
74{
75 struct stm32_timer_cnt *const priv = counter->priv;
76
77 if (val > priv->ceiling)
78 return -EINVAL;
79
80 return regmap_write(priv->regmap, TIM_CNT, val);
81}
82
83static int stm32_count_function_get(struct counter_device *counter,
84 struct counter_count *count,
85 size_t *function)
86{
87 struct stm32_timer_cnt *const priv = counter->priv;
88 u32 smcr;
89
90 regmap_read(priv->regmap, TIM_SMCR, &smcr);
91
92 switch (smcr & TIM_SMCR_SMS) {
93 case 1:
94 *function = STM32_COUNT_ENCODER_MODE_1;
95 return 0;
96 case 2:
97 *function = STM32_COUNT_ENCODER_MODE_2;
98 return 0;
99 case 3:
100 *function = STM32_COUNT_ENCODER_MODE_3;
101 return 0;
102 }
103
104 return -EINVAL;
105}
106
107static int stm32_count_function_set(struct counter_device *counter,
108 struct counter_count *count,
109 size_t function)
110{
111 struct stm32_timer_cnt *const priv = counter->priv;
112 u32 cr1, sms;
113
114 switch (function) {
115 case STM32_COUNT_ENCODER_MODE_1:
116 sms = 1;
117 break;
118 case STM32_COUNT_ENCODER_MODE_2:
119 sms = 2;
120 break;
121 case STM32_COUNT_ENCODER_MODE_3:
122 sms = 3;
123 break;
124 default:
125 sms = 0;
126 break;
127 }
128
129
130 regmap_read(priv->regmap, TIM_CR1, &cr1);
131
132 regmap_update_bits(priv->regmap, TIM_CR1, TIM_CR1_CEN, 0);
133
134
135 regmap_update_bits(priv->regmap, TIM_CR1, TIM_CR1_ARPE, 0);
136 regmap_write(priv->regmap, TIM_ARR, priv->ceiling);
137
138 regmap_update_bits(priv->regmap, TIM_SMCR, TIM_SMCR_SMS, sms);
139
140
141 regmap_update_bits(priv->regmap, TIM_EGR, TIM_EGR_UG, TIM_EGR_UG);
142
143
144 regmap_update_bits(priv->regmap, TIM_CR1, TIM_CR1_CEN, cr1);
145
146 return 0;
147}
148
149static ssize_t stm32_count_direction_read(struct counter_device *counter,
150 struct counter_count *count,
151 void *private, char *buf)
152{
153 struct stm32_timer_cnt *const priv = counter->priv;
154 const char *direction;
155 u32 cr1;
156
157 regmap_read(priv->regmap, TIM_CR1, &cr1);
158 direction = (cr1 & TIM_CR1_DIR) ? "backward" : "forward";
159
160 return scnprintf(buf, PAGE_SIZE, "%s\n", direction);
161}
162
163static ssize_t stm32_count_ceiling_read(struct counter_device *counter,
164 struct counter_count *count,
165 void *private, char *buf)
166{
167 struct stm32_timer_cnt *const priv = counter->priv;
168 u32 arr;
169
170 regmap_read(priv->regmap, TIM_ARR, &arr);
171
172 return snprintf(buf, PAGE_SIZE, "%u\n", arr);
173}
174
175static ssize_t stm32_count_ceiling_write(struct counter_device *counter,
176 struct counter_count *count,
177 void *private,
178 const char *buf, size_t len)
179{
180 struct stm32_timer_cnt *const priv = counter->priv;
181 unsigned int ceiling;
182 int ret;
183
184 ret = kstrtouint(buf, 0, &ceiling);
185 if (ret)
186 return ret;
187
188
189 regmap_update_bits(priv->regmap, TIM_CR1, TIM_CR1_ARPE, 0);
190 regmap_write(priv->regmap, TIM_ARR, ceiling);
191
192 priv->ceiling = ceiling;
193 return len;
194}
195
196static ssize_t stm32_count_enable_read(struct counter_device *counter,
197 struct counter_count *count,
198 void *private, char *buf)
199{
200 struct stm32_timer_cnt *const priv = counter->priv;
201 u32 cr1;
202
203 regmap_read(priv->regmap, TIM_CR1, &cr1);
204
205 return scnprintf(buf, PAGE_SIZE, "%d\n", (bool)(cr1 & TIM_CR1_CEN));
206}
207
208static ssize_t stm32_count_enable_write(struct counter_device *counter,
209 struct counter_count *count,
210 void *private,
211 const char *buf, size_t len)
212{
213 struct stm32_timer_cnt *const priv = counter->priv;
214 int err;
215 u32 cr1;
216 bool enable;
217
218 err = kstrtobool(buf, &enable);
219 if (err)
220 return err;
221
222 if (enable) {
223 regmap_read(priv->regmap, TIM_CR1, &cr1);
224 if (!(cr1 & TIM_CR1_CEN))
225 clk_enable(priv->clk);
226
227 regmap_update_bits(priv->regmap, TIM_CR1, TIM_CR1_CEN,
228 TIM_CR1_CEN);
229 } else {
230 regmap_read(priv->regmap, TIM_CR1, &cr1);
231 regmap_update_bits(priv->regmap, TIM_CR1, TIM_CR1_CEN, 0);
232 if (cr1 & TIM_CR1_CEN)
233 clk_disable(priv->clk);
234 }
235
236
237 priv->enabled = enable;
238
239 return len;
240}
241
242static const struct counter_count_ext stm32_count_ext[] = {
243 {
244 .name = "direction",
245 .read = stm32_count_direction_read,
246 },
247 {
248 .name = "enable",
249 .read = stm32_count_enable_read,
250 .write = stm32_count_enable_write
251 },
252 {
253 .name = "ceiling",
254 .read = stm32_count_ceiling_read,
255 .write = stm32_count_ceiling_write
256 },
257};
258
259enum stm32_synapse_action {
260 STM32_SYNAPSE_ACTION_NONE,
261 STM32_SYNAPSE_ACTION_BOTH_EDGES
262};
263
264static enum counter_synapse_action stm32_synapse_actions[] = {
265 [STM32_SYNAPSE_ACTION_NONE] = COUNTER_SYNAPSE_ACTION_NONE,
266 [STM32_SYNAPSE_ACTION_BOTH_EDGES] = COUNTER_SYNAPSE_ACTION_BOTH_EDGES
267};
268
269static int stm32_action_get(struct counter_device *counter,
270 struct counter_count *count,
271 struct counter_synapse *synapse,
272 size_t *action)
273{
274 size_t function;
275 int err;
276
277
278 *action = STM32_SYNAPSE_ACTION_NONE;
279
280 err = stm32_count_function_get(counter, count, &function);
281 if (err)
282 return 0;
283
284 switch (function) {
285 case STM32_COUNT_ENCODER_MODE_1:
286
287 if (synapse->signal->id == count->synapses[0].signal->id)
288 *action = STM32_SYNAPSE_ACTION_BOTH_EDGES;
289 break;
290 case STM32_COUNT_ENCODER_MODE_2:
291
292 if (synapse->signal->id == count->synapses[1].signal->id)
293 *action = STM32_SYNAPSE_ACTION_BOTH_EDGES;
294 break;
295 case STM32_COUNT_ENCODER_MODE_3:
296
297 *action = STM32_SYNAPSE_ACTION_BOTH_EDGES;
298 break;
299 }
300
301 return 0;
302}
303
304static const struct counter_ops stm32_timer_cnt_ops = {
305 .count_read = stm32_count_read,
306 .count_write = stm32_count_write,
307 .function_get = stm32_count_function_get,
308 .function_set = stm32_count_function_set,
309 .action_get = stm32_action_get,
310};
311
312static struct counter_signal stm32_signals[] = {
313 {
314 .id = 0,
315 .name = "Channel 1 Quadrature A"
316 },
317 {
318 .id = 1,
319 .name = "Channel 1 Quadrature B"
320 }
321};
322
323static struct counter_synapse stm32_count_synapses[] = {
324 {
325 .actions_list = stm32_synapse_actions,
326 .num_actions = ARRAY_SIZE(stm32_synapse_actions),
327 .signal = &stm32_signals[0]
328 },
329 {
330 .actions_list = stm32_synapse_actions,
331 .num_actions = ARRAY_SIZE(stm32_synapse_actions),
332 .signal = &stm32_signals[1]
333 }
334};
335
336static struct counter_count stm32_counts = {
337 .id = 0,
338 .name = "Channel 1 Count",
339 .functions_list = stm32_count_functions,
340 .num_functions = ARRAY_SIZE(stm32_count_functions),
341 .synapses = stm32_count_synapses,
342 .num_synapses = ARRAY_SIZE(stm32_count_synapses),
343 .ext = stm32_count_ext,
344 .num_ext = ARRAY_SIZE(stm32_count_ext)
345};
346
347static int stm32_timer_cnt_probe(struct platform_device *pdev)
348{
349 struct stm32_timers *ddata = dev_get_drvdata(pdev->dev.parent);
350 struct device *dev = &pdev->dev;
351 struct stm32_timer_cnt *priv;
352
353 if (IS_ERR_OR_NULL(ddata))
354 return -EINVAL;
355
356 priv = devm_kzalloc(dev, sizeof(*priv), GFP_KERNEL);
357 if (!priv)
358 return -ENOMEM;
359
360 priv->regmap = ddata->regmap;
361 priv->clk = ddata->clk;
362 priv->ceiling = ddata->max_arr;
363
364 priv->counter.name = dev_name(dev);
365 priv->counter.parent = dev;
366 priv->counter.ops = &stm32_timer_cnt_ops;
367 priv->counter.counts = &stm32_counts;
368 priv->counter.num_counts = 1;
369 priv->counter.signals = stm32_signals;
370 priv->counter.num_signals = ARRAY_SIZE(stm32_signals);
371 priv->counter.priv = priv;
372
373 platform_set_drvdata(pdev, priv);
374
375
376 return devm_counter_register(dev, &priv->counter);
377}
378
379static int __maybe_unused stm32_timer_cnt_suspend(struct device *dev)
380{
381 struct stm32_timer_cnt *priv = dev_get_drvdata(dev);
382
383
384 if (priv->enabled) {
385
386 regmap_read(priv->regmap, TIM_SMCR, &priv->bak.smcr);
387 regmap_read(priv->regmap, TIM_ARR, &priv->bak.arr);
388 regmap_read(priv->regmap, TIM_CNT, &priv->bak.cnt);
389 regmap_read(priv->regmap, TIM_CR1, &priv->bak.cr1);
390
391
392 regmap_update_bits(priv->regmap, TIM_CR1, TIM_CR1_CEN, 0);
393 clk_disable(priv->clk);
394 }
395
396 return pinctrl_pm_select_sleep_state(dev);
397}
398
399static int __maybe_unused stm32_timer_cnt_resume(struct device *dev)
400{
401 struct stm32_timer_cnt *priv = dev_get_drvdata(dev);
402 int ret;
403
404 ret = pinctrl_pm_select_default_state(dev);
405 if (ret)
406 return ret;
407
408 if (priv->enabled) {
409 clk_enable(priv->clk);
410
411
412 regmap_write(priv->regmap, TIM_SMCR, priv->bak.smcr);
413 regmap_write(priv->regmap, TIM_ARR, priv->bak.arr);
414 regmap_write(priv->regmap, TIM_CNT, priv->bak.cnt);
415
416
417 regmap_write(priv->regmap, TIM_CR1, priv->bak.cr1);
418 }
419
420 return 0;
421}
422
423static SIMPLE_DEV_PM_OPS(stm32_timer_cnt_pm_ops, stm32_timer_cnt_suspend,
424 stm32_timer_cnt_resume);
425
426static const struct of_device_id stm32_timer_cnt_of_match[] = {
427 { .compatible = "st,stm32-timer-counter", },
428 {},
429};
430MODULE_DEVICE_TABLE(of, stm32_timer_cnt_of_match);
431
432static struct platform_driver stm32_timer_cnt_driver = {
433 .probe = stm32_timer_cnt_probe,
434 .driver = {
435 .name = "stm32-timer-counter",
436 .of_match_table = stm32_timer_cnt_of_match,
437 .pm = &stm32_timer_cnt_pm_ops,
438 },
439};
440module_platform_driver(stm32_timer_cnt_driver);
441
442MODULE_AUTHOR("Benjamin Gaignard <benjamin.gaignard@st.com>");
443MODULE_ALIAS("platform:stm32-timer-counter");
444MODULE_DESCRIPTION("STMicroelectronics STM32 TIMER counter driver");
445MODULE_LICENSE("GPL v2");
446
