drivers/iio/trigger/stm32-timer-trigger.c

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