1// SPDX-License-Identifier: GPL-2.0
2/*
3 * Copyright (C) STMicroelectronics 2016
4 *
5 * Author: Benjamin Gaignard <benjamin.gaignard@st.com>
6 *
7 */
8
9#include <linux/iio/iio.h>
10#include <linux/iio/sysfs.h>
11#include <linux/iio/timer/stm32-timer-trigger.h>
12#include <linux/iio/trigger.h>
13#include <linux/mfd/stm32-timers.h>
14#include <linux/mod_devicetable.h>
15#include <linux/module.h>
16#include <linux/platform_device.h>
17#include <linux/property.h>
18
19#define MAX_TRIGGERS 7
20#define MAX_VALIDS 5
21
22/* List the triggers created by each timer */
23static const void *triggers_table[][MAX_TRIGGERS] = {
24	{ TIM1_TRGO, TIM1_TRGO2, TIM1_CH1, TIM1_CH2, TIM1_CH3, TIM1_CH4,},
25	{ TIM2_TRGO, TIM2_CH1, TIM2_CH2, TIM2_CH3, TIM2_CH4,},
26	{ TIM3_TRGO, TIM3_CH1, TIM3_CH2, TIM3_CH3, TIM3_CH4,},
27	{ TIM4_TRGO, TIM4_CH1, TIM4_CH2, TIM4_CH3, TIM4_CH4,},
28	{ TIM5_TRGO, TIM5_CH1, TIM5_CH2, TIM5_CH3, TIM5_CH4,},
29	{ TIM6_TRGO,},
30	{ TIM7_TRGO,},
31	{ TIM8_TRGO, TIM8_TRGO2, TIM8_CH1, TIM8_CH2, TIM8_CH3, TIM8_CH4,},
32	{ TIM9_TRGO, TIM9_CH1, TIM9_CH2,},
33	{ TIM10_OC1,},
34	{ TIM11_OC1,},
35	{ TIM12_TRGO, TIM12_CH1, TIM12_CH2,},
36	{ TIM13_OC1,},
37	{ TIM14_OC1,},
38	{ TIM15_TRGO,},
39	{ TIM16_OC1,},
40	{ TIM17_OC1,},
41};
42
43/* List the triggers accepted by each timer */
44static const void *valids_table[][MAX_VALIDS] = {
45	{ TIM5_TRGO, TIM2_TRGO, TIM3_TRGO, TIM4_TRGO,},
46	{ TIM1_TRGO, TIM8_TRGO, TIM3_TRGO, TIM4_TRGO,},
47	{ TIM1_TRGO, TIM2_TRGO, TIM5_TRGO, TIM4_TRGO,},
48	{ TIM1_TRGO, TIM2_TRGO, TIM3_TRGO, TIM8_TRGO,},
49	{ TIM2_TRGO, TIM3_TRGO, TIM4_TRGO, TIM8_TRGO,},
50	{ }, /* timer 6 */
51	{ }, /* timer 7 */
52	{ TIM1_TRGO, TIM2_TRGO, TIM4_TRGO, TIM5_TRGO,},
53	{ TIM2_TRGO, TIM3_TRGO, TIM10_OC1, TIM11_OC1,},
54	{ }, /* timer 10 */
55	{ }, /* timer 11 */
56	{ TIM4_TRGO, TIM5_TRGO, TIM13_OC1, TIM14_OC1,},
57};
58
59static const void *stm32h7_valids_table[][MAX_VALIDS] = {
60	{ TIM15_TRGO, TIM2_TRGO, TIM3_TRGO, TIM4_TRGO,},
61	{ TIM1_TRGO, TIM8_TRGO, TIM3_TRGO, TIM4_TRGO,},
62	{ TIM1_TRGO, TIM2_TRGO, TIM15_TRGO, TIM4_TRGO,},
63	{ TIM1_TRGO, TIM2_TRGO, TIM3_TRGO, TIM8_TRGO,},
64	{ TIM1_TRGO, TIM8_TRGO, TIM3_TRGO, TIM4_TRGO,},
65	{ }, /* timer 6 */
66	{ }, /* timer 7 */
67	{ TIM1_TRGO, TIM2_TRGO, TIM4_TRGO, TIM5_TRGO,},
68	{ }, /* timer 9 */
69	{ }, /* timer 10 */
70	{ }, /* timer 11 */
71	{ TIM4_TRGO, TIM5_TRGO, TIM13_OC1, TIM14_OC1,},
72	{ }, /* timer 13 */
73	{ }, /* timer 14 */
74	{ TIM1_TRGO, TIM3_TRGO, TIM16_OC1, TIM17_OC1,},
75	{ }, /* timer 16 */
76	{ }, /* timer 17 */
77};
78
79struct stm32_timer_trigger_regs {
80	u32 cr1;
81	u32 cr2;
82	u32 psc;
83	u32 arr;
84	u32 cnt;
85	u32 smcr;
86};
87
88struct stm32_timer_trigger {
89	struct device *dev;
90	struct regmap *regmap;
91	struct clk *clk;
92	bool enabled;
93	u32 max_arr;
94	const void *triggers;
95	const void *valids;
96	bool has_trgo2;
97	struct mutex lock; /* concurrent sysfs configuration */
98	struct list_head tr_list;
99	struct stm32_timer_trigger_regs bak;
100};
101
102struct stm32_timer_trigger_cfg {
103	const void *(*valids_table)[MAX_VALIDS];
104	const unsigned int num_valids_table;
105};
106
107static bool stm32_timer_is_trgo2_name(const char *name)
108{
109	return !!strstr(name, "trgo2");
110}
111
112static bool stm32_timer_is_trgo_name(const char *name)
113{
114	return (!!strstr(name, "trgo") && !strstr(name, "trgo2"));
115}
116
117static int stm32_timer_start(struct stm32_timer_trigger *priv,
118			     struct iio_trigger *trig,
119			     unsigned int frequency)
120{
121	unsigned long long prd, div;
122	int prescaler = 0;
123	u32 ccer;
124
125	/* Period and prescaler values depends of clock rate */
126	div = (unsigned long long)clk_get_rate(priv->clk);
127
128	do_div(div, frequency);
129
130	prd = div;
131
132	/*
133	 * Increase prescaler value until we get a result that fit
134	 * with auto reload register maximum value.
135	 */
136	while (div > priv->max_arr) {
137		prescaler++;
138		div = prd;
139		do_div(div, (prescaler + 1));
140	}
141	prd = div;
142
143	if (prescaler > MAX_TIM_PSC) {
144		dev_err(priv->dev, "prescaler exceeds the maximum value\n");
145		return -EINVAL;
146	}
147
148	/* Check if nobody else use the timer */
149	regmap_read(priv->regmap, TIM_CCER, &ccer);
150	if (ccer & TIM_CCER_CCXE)
151		return -EBUSY;
152
153	mutex_lock(&priv->lock);
154	if (!priv->enabled) {
155		priv->enabled = true;
156		clk_enable(priv->clk);
157	}
158
159	regmap_write(priv->regmap, TIM_PSC, prescaler);
160	regmap_write(priv->regmap, TIM_ARR, prd - 1);
161	regmap_update_bits(priv->regmap, TIM_CR1, TIM_CR1_ARPE, TIM_CR1_ARPE);
162
163	/* Force master mode to update mode */
164	if (stm32_timer_is_trgo2_name(trig->name))
165		regmap_update_bits(priv->regmap, TIM_CR2, TIM_CR2_MMS2,
166				   0x2 << TIM_CR2_MMS2_SHIFT);
167	else
168		regmap_update_bits(priv->regmap, TIM_CR2, TIM_CR2_MMS,
169				   0x2 << TIM_CR2_MMS_SHIFT);
170
171	/* Make sure that registers are updated */
172	regmap_update_bits(priv->regmap, TIM_EGR, TIM_EGR_UG, TIM_EGR_UG);
173
174	/* Enable controller */
175	regmap_update_bits(priv->regmap, TIM_CR1, TIM_CR1_CEN, TIM_CR1_CEN);
176	mutex_unlock(&priv->lock);
177
178	return 0;
179}
180
181static void stm32_timer_stop(struct stm32_timer_trigger *priv,
182			     struct iio_trigger *trig)
183{
184	u32 ccer;
185
186	regmap_read(priv->regmap, TIM_CCER, &ccer);
187	if (ccer & TIM_CCER_CCXE)
188		return;
189
190	mutex_lock(&priv->lock);
191	/* Stop timer */
192	regmap_update_bits(priv->regmap, TIM_CR1, TIM_CR1_ARPE, 0);
193	regmap_update_bits(priv->regmap, TIM_CR1, TIM_CR1_CEN, 0);
194	regmap_write(priv->regmap, TIM_PSC, 0);
195	regmap_write(priv->regmap, TIM_ARR, 0);
196
197	/* Force disable master mode */
198	if (stm32_timer_is_trgo2_name(trig->name))
199		regmap_update_bits(priv->regmap, TIM_CR2, TIM_CR2_MMS2, 0);
200	else
201		regmap_update_bits(priv->regmap, TIM_CR2, TIM_CR2_MMS, 0);
202
203	/* Make sure that registers are updated */
204	regmap_update_bits(priv->regmap, TIM_EGR, TIM_EGR_UG, TIM_EGR_UG);
205
206	if (priv->enabled) {
207		priv->enabled = false;
208		clk_disable(priv->clk);
209	}
210	mutex_unlock(&priv->lock);
211}
212
213static ssize_t stm32_tt_store_frequency(struct device *dev,
214					struct device_attribute *attr,
215					const char *buf, size_t len)
216{
217	struct iio_trigger *trig = to_iio_trigger(dev);
218	struct stm32_timer_trigger *priv = iio_trigger_get_drvdata(trig);
219	unsigned int freq;
220	int ret;
221
222	ret = kstrtouint(buf, 10, &freq);
223	if (ret)
224		return ret;
225
226	if (freq == 0) {
227		stm32_timer_stop(priv, trig);
228	} else {
229		ret = stm32_timer_start(priv, trig, freq);
230		if (ret)
231			return ret;
232	}
233
234	return len;
235}
236
237static ssize_t stm32_tt_read_frequency(struct device *dev,
238				       struct device_attribute *attr, char *buf)
239{
240	struct iio_trigger *trig = to_iio_trigger(dev);
241	struct stm32_timer_trigger *priv = iio_trigger_get_drvdata(trig);
242	u32 psc, arr, cr1;
243	unsigned long long freq = 0;
244
245	regmap_read(priv->regmap, TIM_CR1, &cr1);
246	regmap_read(priv->regmap, TIM_PSC, &psc);
247	regmap_read(priv->regmap, TIM_ARR, &arr);
248
249	if (cr1 & TIM_CR1_CEN) {
250		freq = (unsigned long long)clk_get_rate(priv->clk);
251		do_div(freq, psc + 1);
252		do_div(freq, arr + 1);
253	}
254
255	return sprintf(buf, "%d\n", (unsigned int)freq);
256}
257
258static IIO_DEV_ATTR_SAMP_FREQ(0660,
259			      stm32_tt_read_frequency,
260			      stm32_tt_store_frequency);
261
262#define MASTER_MODE_MAX		7
263#define MASTER_MODE2_MAX	15
264
265static char *master_mode_table[] = {
266	"reset",
267	"enable",
268	"update",
269	"compare_pulse",
270	"OC1REF",
271	"OC2REF",
272	"OC3REF",
273	"OC4REF",
274	/* Master mode selection 2 only */
275	"OC5REF",
276	"OC6REF",
277	"compare_pulse_OC4REF",
278	"compare_pulse_OC6REF",
279	"compare_pulse_OC4REF_r_or_OC6REF_r",
280	"compare_pulse_OC4REF_r_or_OC6REF_f",
281	"compare_pulse_OC5REF_r_or_OC6REF_r",
282	"compare_pulse_OC5REF_r_or_OC6REF_f",
283};
284
285static ssize_t stm32_tt_show_master_mode(struct device *dev,
286					 struct device_attribute *attr,
287					 char *buf)
288{
289	struct stm32_timer_trigger *priv = dev_get_drvdata(dev);
290	struct iio_trigger *trig = to_iio_trigger(dev);
291	u32 cr2;
292
293	regmap_read(priv->regmap, TIM_CR2, &cr2);
294
295	if (stm32_timer_is_trgo2_name(trig->name))
296		cr2 = (cr2 & TIM_CR2_MMS2) >> TIM_CR2_MMS2_SHIFT;
297	else
298		cr2 = (cr2 & TIM_CR2_MMS) >> TIM_CR2_MMS_SHIFT;
299
300	return sysfs_emit(buf, "%s\n", master_mode_table[cr2]);
301}
302
303static ssize_t stm32_tt_store_master_mode(struct device *dev,
304					  struct device_attribute *attr,
305					  const char *buf, size_t len)
306{
307	struct stm32_timer_trigger *priv = dev_get_drvdata(dev);
308	struct iio_trigger *trig = to_iio_trigger(dev);
309	u32 mask, shift, master_mode_max;
310	int i;
311
312	if (stm32_timer_is_trgo2_name(trig->name)) {
313		mask = TIM_CR2_MMS2;
314		shift = TIM_CR2_MMS2_SHIFT;
315		master_mode_max = MASTER_MODE2_MAX;
316	} else {
317		mask = TIM_CR2_MMS;
318		shift = TIM_CR2_MMS_SHIFT;
319		master_mode_max = MASTER_MODE_MAX;
320	}
321
322	for (i = 0; i <= master_mode_max; i++) {
323		if (!strncmp(master_mode_table[i], buf,
324			     strlen(master_mode_table[i]))) {
325			mutex_lock(&priv->lock);
326			if (!priv->enabled) {
327				/* Clock should be enabled first */
328				priv->enabled = true;
329				clk_enable(priv->clk);
330			}
331			regmap_update_bits(priv->regmap, TIM_CR2, mask,
332					   i << shift);
333			mutex_unlock(&priv->lock);
334			return len;
335		}
336	}
337
338	return -EINVAL;
339}
340
341static ssize_t stm32_tt_show_master_mode_avail(struct device *dev,
342					       struct device_attribute *attr,
343					       char *buf)
344{
345	struct iio_trigger *trig = to_iio_trigger(dev);
346	unsigned int i, master_mode_max;
347	size_t len = 0;
348
349	if (stm32_timer_is_trgo2_name(trig->name))
350		master_mode_max = MASTER_MODE2_MAX;
351	else
352		master_mode_max = MASTER_MODE_MAX;
353
354	for (i = 0; i <= master_mode_max; i++)
355		len += scnprintf(buf + len, PAGE_SIZE - len,
356			"%s ", master_mode_table[i]);
357
358	/* replace trailing space by newline */
359	buf[len - 1] = '\n';
360
361	return len;
362}
363
364static IIO_DEVICE_ATTR(master_mode_available, 0444,
365		       stm32_tt_show_master_mode_avail, NULL, 0);
366
367static IIO_DEVICE_ATTR(master_mode, 0660,
368		       stm32_tt_show_master_mode,
369		       stm32_tt_store_master_mode,
370		       0);
371
372static struct attribute *stm32_trigger_attrs[] = {
373	&iio_dev_attr_sampling_frequency.dev_attr.attr,
374	&iio_dev_attr_master_mode.dev_attr.attr,
375	&iio_dev_attr_master_mode_available.dev_attr.attr,
376	NULL,
377};
378
379static const struct attribute_group stm32_trigger_attr_group = {
380	.attrs = stm32_trigger_attrs,
381};
382
383static const struct attribute_group *stm32_trigger_attr_groups[] = {
384	&stm32_trigger_attr_group,
385	NULL,
386};
387
388static const struct iio_trigger_ops timer_trigger_ops = {
389};
390
391static void stm32_unregister_iio_triggers(struct stm32_timer_trigger *priv)
392{
393	struct iio_trigger *tr;
394
395	list_for_each_entry(tr, &priv->tr_list, alloc_list)
396		iio_trigger_unregister(tr);
397}
398
399static int stm32_register_iio_triggers(struct stm32_timer_trigger *priv)
400{
401	int ret;
402	const char * const *cur = priv->triggers;
403
404	INIT_LIST_HEAD(&priv->tr_list);
405
406	while (cur && *cur) {
407		struct iio_trigger *trig;
408		bool cur_is_trgo = stm32_timer_is_trgo_name(*cur);
409		bool cur_is_trgo2 = stm32_timer_is_trgo2_name(*cur);
410
411		if (cur_is_trgo2 && !priv->has_trgo2) {
412			cur++;
413			continue;
414		}
415
416		trig = devm_iio_trigger_alloc(priv->dev, "%s", *cur);
417		if  (!trig)
418			return -ENOMEM;
419
420		trig->dev.parent = priv->dev->parent;
421		trig->ops = &timer_trigger_ops;
422
423		/*
424		 * sampling frequency and master mode attributes
425		 * should only be available on trgo/trgo2 triggers
426		 */
427		if (cur_is_trgo || cur_is_trgo2)
428			trig->dev.groups = stm32_trigger_attr_groups;
429
430		iio_trigger_set_drvdata(trig, priv);
431
432		ret = iio_trigger_register(trig);
433		if (ret) {
434			stm32_unregister_iio_triggers(priv);
435			return ret;
436		}
437
438		list_add_tail(&trig->alloc_list, &priv->tr_list);
439		cur++;
440	}
441
442	return 0;
443}
444
445static int stm32_counter_read_raw(struct iio_dev *indio_dev,
446				  struct iio_chan_spec const *chan,
447				  int *val, int *val2, long mask)
448{
449	struct stm32_timer_trigger *priv = iio_priv(indio_dev);
450	u32 dat;
451
452	switch (mask) {
453	case IIO_CHAN_INFO_RAW:
454		regmap_read(priv->regmap, TIM_CNT, &dat);
455		*val = dat;
456		return IIO_VAL_INT;
457
458	case IIO_CHAN_INFO_ENABLE:
459		regmap_read(priv->regmap, TIM_CR1, &dat);
460		*val = (dat & TIM_CR1_CEN) ? 1 : 0;
461		return IIO_VAL_INT;
462
463	case IIO_CHAN_INFO_SCALE:
464		regmap_read(priv->regmap, TIM_SMCR, &dat);
465		dat &= TIM_SMCR_SMS;
466
467		*val = 1;
468		*val2 = 0;
469
470		/* in quadrature case scale = 0.25 */
471		if (dat == 3)
472			*val2 = 2;
473
474		return IIO_VAL_FRACTIONAL_LOG2;
475	}
476
477	return -EINVAL;
478}
479
480static int stm32_counter_write_raw(struct iio_dev *indio_dev,
481				   struct iio_chan_spec const *chan,
482				   int val, int val2, long mask)
483{
484	struct stm32_timer_trigger *priv = iio_priv(indio_dev);
485
486	switch (mask) {
487	case IIO_CHAN_INFO_RAW:
488		return regmap_write(priv->regmap, TIM_CNT, val);
489
490	case IIO_CHAN_INFO_SCALE:
491		/* fixed scale */
492		return -EINVAL;
493
494	case IIO_CHAN_INFO_ENABLE:
495		mutex_lock(&priv->lock);
496		if (val) {
497			if (!priv->enabled) {
498				priv->enabled = true;
499				clk_enable(priv->clk);
500			}
501			regmap_update_bits(priv->regmap, TIM_CR1, TIM_CR1_CEN,
502					   TIM_CR1_CEN);
503		} else {
504			regmap_update_bits(priv->regmap, TIM_CR1, TIM_CR1_CEN,
505					   0);
506			if (priv->enabled) {
507				priv->enabled = false;
508				clk_disable(priv->clk);
509			}
510		}
511		mutex_unlock(&priv->lock);
512		return 0;
513	}
514
515	return -EINVAL;
516}
517
518static int stm32_counter_validate_trigger(struct iio_dev *indio_dev,
519					  struct iio_trigger *trig)
520{
521	struct stm32_timer_trigger *priv = iio_priv(indio_dev);
522	const char * const *cur = priv->valids;
523	unsigned int i = 0;
524
525	if (!is_stm32_timer_trigger(trig))
526		return -EINVAL;
527
528	while (cur && *cur) {
529		if (!strncmp(trig->name, *cur, strlen(trig->name))) {
530			regmap_update_bits(priv->regmap,
531					   TIM_SMCR, TIM_SMCR_TS,
532					   i << TIM_SMCR_TS_SHIFT);
533			return 0;
534		}
535		cur++;
536		i++;
537	}
538
539	return -EINVAL;
540}
541
542static const struct iio_info stm32_trigger_info = {
543	.validate_trigger = stm32_counter_validate_trigger,
544	.read_raw = stm32_counter_read_raw,
545	.write_raw = stm32_counter_write_raw
546};
547
548static const char *const stm32_trigger_modes[] = {
549	"trigger",
550};
551
552static int stm32_set_trigger_mode(struct iio_dev *indio_dev,
553				  const struct iio_chan_spec *chan,
554				  unsigned int mode)
555{
556	struct stm32_timer_trigger *priv = iio_priv(indio_dev);
557
558	regmap_update_bits(priv->regmap, TIM_SMCR, TIM_SMCR_SMS, TIM_SMCR_SMS);
559
560	return 0;
561}
562
563static int stm32_get_trigger_mode(struct iio_dev *indio_dev,
564				  const struct iio_chan_spec *chan)
565{
566	struct stm32_timer_trigger *priv = iio_priv(indio_dev);
567	u32 smcr;
568
569	regmap_read(priv->regmap, TIM_SMCR, &smcr);
570
571	return (smcr & TIM_SMCR_SMS) == TIM_SMCR_SMS ? 0 : -EINVAL;
572}
573
574static const struct iio_enum stm32_trigger_mode_enum = {
575	.items = stm32_trigger_modes,
576	.num_items = ARRAY_SIZE(stm32_trigger_modes),
577	.set = stm32_set_trigger_mode,
578	.get = stm32_get_trigger_mode
579};
580
581static const char *const stm32_enable_modes[] = {
582	"always",
583	"gated",
584	"triggered",
585};
586
587static int stm32_enable_mode2sms(int mode)
588{
589	switch (mode) {
590	case 0:
591		return 0;
592	case 1:
593		return 5;
594	case 2:
595		return 6;
596	}
597
598	return -EINVAL;
599}
600
601static int stm32_set_enable_mode(struct iio_dev *indio_dev,
602				 const struct iio_chan_spec *chan,
603				 unsigned int mode)
604{
605	struct stm32_timer_trigger *priv = iio_priv(indio_dev);
606	int sms = stm32_enable_mode2sms(mode);
607
608	if (sms < 0)
609		return sms;
610	/*
611	 * Triggered mode sets CEN bit automatically by hardware. So, first
612	 * enable counter clock, so it can use it. Keeps it in sync with CEN.
613	 */
614	mutex_lock(&priv->lock);
615	if (sms == 6 && !priv->enabled) {
616		clk_enable(priv->clk);
617		priv->enabled = true;
618	}
619	mutex_unlock(&priv->lock);
620
621	regmap_update_bits(priv->regmap, TIM_SMCR, TIM_SMCR_SMS, sms);
622
623	return 0;
624}
625
626static int stm32_sms2enable_mode(int mode)
627{
628	switch (mode) {
629	case 0:
630		return 0;
631	case 5:
632		return 1;
633	case 6:
634		return 2;
635	}
636
637	return -EINVAL;
638}
639
640static int stm32_get_enable_mode(struct iio_dev *indio_dev,
641				 const struct iio_chan_spec *chan)
642{
643	struct stm32_timer_trigger *priv = iio_priv(indio_dev);
644	u32 smcr;
645
646	regmap_read(priv->regmap, TIM_SMCR, &smcr);
647	smcr &= TIM_SMCR_SMS;
648
649	return stm32_sms2enable_mode(smcr);
650}
651
652static const struct iio_enum stm32_enable_mode_enum = {
653	.items = stm32_enable_modes,
654	.num_items = ARRAY_SIZE(stm32_enable_modes),
655	.set = stm32_set_enable_mode,
656	.get = stm32_get_enable_mode
657};
658
659static ssize_t stm32_count_get_preset(struct iio_dev *indio_dev,
660				      uintptr_t private,
661				      const struct iio_chan_spec *chan,
662				      char *buf)
663{
664	struct stm32_timer_trigger *priv = iio_priv(indio_dev);
665	u32 arr;
666
667	regmap_read(priv->regmap, TIM_ARR, &arr);
668
669	return snprintf(buf, PAGE_SIZE, "%u\n", arr);
670}
671
672static ssize_t stm32_count_set_preset(struct iio_dev *indio_dev,
673				      uintptr_t private,
674				      const struct iio_chan_spec *chan,
675				      const char *buf, size_t len)
676{
677	struct stm32_timer_trigger *priv = iio_priv(indio_dev);
678	unsigned int preset;
679	int ret;
680
681	ret = kstrtouint(buf, 0, &preset);
682	if (ret)
683		return ret;
684
685	/* TIMx_ARR register shouldn't be buffered (ARPE=0) */
686	regmap_update_bits(priv->regmap, TIM_CR1, TIM_CR1_ARPE, 0);
687	regmap_write(priv->regmap, TIM_ARR, preset);
688
689	return len;
690}
691
692static const struct iio_chan_spec_ext_info stm32_trigger_count_info[] = {
693	{
694		.name = "preset",
695		.shared = IIO_SEPARATE,
696		.read = stm32_count_get_preset,
697		.write = stm32_count_set_preset
698	},
699	IIO_ENUM("enable_mode", IIO_SEPARATE, &stm32_enable_mode_enum),
700	IIO_ENUM_AVAILABLE("enable_mode", IIO_SHARED_BY_TYPE, &stm32_enable_mode_enum),
701	IIO_ENUM("trigger_mode", IIO_SEPARATE, &stm32_trigger_mode_enum),
702	IIO_ENUM_AVAILABLE("trigger_mode", IIO_SHARED_BY_TYPE, &stm32_trigger_mode_enum),
703	{}
704};
705
706static const struct iio_chan_spec stm32_trigger_channel = {
707	.type = IIO_COUNT,
708	.channel = 0,
709	.info_mask_separate = BIT(IIO_CHAN_INFO_RAW) |
710			      BIT(IIO_CHAN_INFO_ENABLE) |
711			      BIT(IIO_CHAN_INFO_SCALE),
712	.ext_info = stm32_trigger_count_info,
713	.indexed = 1
714};
715
716static struct stm32_timer_trigger *stm32_setup_counter_device(struct device *dev)
717{
718	struct iio_dev *indio_dev;
719	int ret;
720
721	indio_dev = devm_iio_device_alloc(dev,
722					  sizeof(struct stm32_timer_trigger));
723	if (!indio_dev)
724		return NULL;
725
726	indio_dev->name = dev_name(dev);
727	indio_dev->info = &stm32_trigger_info;
728	indio_dev->modes = INDIO_HARDWARE_TRIGGERED;
729	indio_dev->num_channels = 1;
730	indio_dev->channels = &stm32_trigger_channel;
731
732	ret = devm_iio_device_register(dev, indio_dev);
733	if (ret)
734		return NULL;
735
736	return iio_priv(indio_dev);
737}
738
739/**
740 * is_stm32_timer_trigger
741 * @trig: trigger to be checked
742 *
743 * return true if the trigger is a valid stm32 iio timer trigger
744 * either return false
745 */
746bool is_stm32_timer_trigger(struct iio_trigger *trig)
747{
748	return (trig->ops == &timer_trigger_ops);
749}
750EXPORT_SYMBOL(is_stm32_timer_trigger);
751
752static void stm32_timer_detect_trgo2(struct stm32_timer_trigger *priv)
753{
754	u32 val;
755
756	/*
757	 * Master mode selection 2 bits can only be written and read back when
758	 * timer supports it.
759	 */
760	regmap_update_bits(priv->regmap, TIM_CR2, TIM_CR2_MMS2, TIM_CR2_MMS2);
761	regmap_read(priv->regmap, TIM_CR2, &val);
762	regmap_update_bits(priv->regmap, TIM_CR2, TIM_CR2_MMS2, 0);
763	priv->has_trgo2 = !!val;
764}
765
766static int stm32_timer_trigger_probe(struct platform_device *pdev)
767{
768	struct device *dev = &pdev->dev;
769	struct stm32_timer_trigger *priv;
770	struct stm32_timers *ddata = dev_get_drvdata(pdev->dev.parent);
771	const struct stm32_timer_trigger_cfg *cfg;
772	unsigned int index;
773	int ret;
774
775	ret = device_property_read_u32(dev, "reg", &index);
776	if (ret)
777		return ret;
778
779	cfg = device_get_match_data(dev);
780
781	if (index >= ARRAY_SIZE(triggers_table) ||
782	    index >= cfg->num_valids_table)
783		return -EINVAL;
784
785	/* Create an IIO device only if we have triggers to be validated */
786	if (*cfg->valids_table[index])
787		priv = stm32_setup_counter_device(dev);
788	else
789		priv = devm_kzalloc(dev, sizeof(*priv), GFP_KERNEL);
790
791	if (!priv)
792		return -ENOMEM;
793
794	priv->dev = dev;
795	priv->regmap = ddata->regmap;
796	priv->clk = ddata->clk;
797	priv->max_arr = ddata->max_arr;
798	priv->triggers = triggers_table[index];
799	priv->valids = cfg->valids_table[index];
800	stm32_timer_detect_trgo2(priv);
801	mutex_init(&priv->lock);
802
803	ret = stm32_register_iio_triggers(priv);
804	if (ret)
805		return ret;
806
807	platform_set_drvdata(pdev, priv);
808
809	return 0;
810}
811
812static void stm32_timer_trigger_remove(struct platform_device *pdev)
813{
814	struct stm32_timer_trigger *priv = platform_get_drvdata(pdev);
815	u32 val;
816
817	/* Unregister triggers before everything can be safely turned off */
818	stm32_unregister_iio_triggers(priv);
819
820	/* Check if nobody else use the timer, then disable it */
821	regmap_read(priv->regmap, TIM_CCER, &val);
822	if (!(val & TIM_CCER_CCXE))
823		regmap_update_bits(priv->regmap, TIM_CR1, TIM_CR1_CEN, 0);
824
825	if (priv->enabled)
826		clk_disable(priv->clk);
827}
828
829static int stm32_timer_trigger_suspend(struct device *dev)
830{
831	struct stm32_timer_trigger *priv = dev_get_drvdata(dev);
832
833	/* Only take care of enabled timer: don't disturb other MFD child */
834	if (priv->enabled) {
835		/* Backup registers that may get lost in low power mode */
836		regmap_read(priv->regmap, TIM_CR1, &priv->bak.cr1);
837		regmap_read(priv->regmap, TIM_CR2, &priv->bak.cr2);
838		regmap_read(priv->regmap, TIM_PSC, &priv->bak.psc);
839		regmap_read(priv->regmap, TIM_ARR, &priv->bak.arr);
840		regmap_read(priv->regmap, TIM_CNT, &priv->bak.cnt);
841		regmap_read(priv->regmap, TIM_SMCR, &priv->bak.smcr);
842
843		/* Disable the timer */
844		regmap_update_bits(priv->regmap, TIM_CR1, TIM_CR1_CEN, 0);
845		clk_disable(priv->clk);
846	}
847
848	return 0;
849}
850
851static int stm32_timer_trigger_resume(struct device *dev)
852{
853	struct stm32_timer_trigger *priv = dev_get_drvdata(dev);
854	int ret;
855
856	if (priv->enabled) {
857		ret = clk_enable(priv->clk);
858		if (ret)
859			return ret;
860
861		/* restore master/slave modes */
862		regmap_write(priv->regmap, TIM_SMCR, priv->bak.smcr);
863		regmap_write(priv->regmap, TIM_CR2, priv->bak.cr2);
864
865		/* restore sampling_frequency (trgo / trgo2 triggers) */
866		regmap_write(priv->regmap, TIM_PSC, priv->bak.psc);
867		regmap_write(priv->regmap, TIM_ARR, priv->bak.arr);
868		regmap_write(priv->regmap, TIM_CNT, priv->bak.cnt);
869
870		/* Also re-enables the timer */
871		regmap_write(priv->regmap, TIM_CR1, priv->bak.cr1);
872	}
873
874	return 0;
875}
876
877static DEFINE_SIMPLE_DEV_PM_OPS(stm32_timer_trigger_pm_ops,
878				stm32_timer_trigger_suspend,
879				stm32_timer_trigger_resume);
880
881static const struct stm32_timer_trigger_cfg stm32_timer_trg_cfg = {
882	.valids_table = valids_table,
883	.num_valids_table = ARRAY_SIZE(valids_table),
884};
885
886static const struct stm32_timer_trigger_cfg stm32h7_timer_trg_cfg = {
887	.valids_table = stm32h7_valids_table,
888	.num_valids_table = ARRAY_SIZE(stm32h7_valids_table),
889};
890
891static const struct of_device_id stm32_trig_of_match[] = {
892	{
893		.compatible = "st,stm32-timer-trigger",
894		.data = (void *)&stm32_timer_trg_cfg,
895	}, {
896		.compatible = "st,stm32h7-timer-trigger",
897		.data = (void *)&stm32h7_timer_trg_cfg,
898	},
899	{ /* end node */ },
900};
901MODULE_DEVICE_TABLE(of, stm32_trig_of_match);
902
903static struct platform_driver stm32_timer_trigger_driver = {
904	.probe = stm32_timer_trigger_probe,
905	.remove_new = stm32_timer_trigger_remove,
906	.driver = {
907		.name = "stm32-timer-trigger",
908		.of_match_table = stm32_trig_of_match,
909		.pm = pm_sleep_ptr(&stm32_timer_trigger_pm_ops),
910	},
911};
912module_platform_driver(stm32_timer_trigger_driver);
913
914MODULE_ALIAS("platform:stm32-timer-trigger");
915MODULE_DESCRIPTION("STMicroelectronics STM32 Timer Trigger driver");
916MODULE_LICENSE("GPL v2");
917