drivers/iio/adc/axp20x_adc.c

   1 // SPDX-License-Identifier: GPL-2.0-only
   2 /* ADC driver for AXP20X and AXP22X PMICs
   3  *
   4  * Copyright (c) 2016 Free Electrons NextThing Co.
   5  *      Quentin Schulz <quentin.schulz@free-electrons.com>
   6  */
   7
   8 #include <linux/completion.h>
   9 #include <linux/interrupt.h>
  10 #include <linux/io.h>
  11 #include <linux/module.h>
  12 #include <linux/mod_devicetable.h>
  13 #include <linux/platform_device.h>
  14 #include <linux/pm_runtime.h>
  15 #include <linux/property.h>
  16 #include <linux/regmap.h>
  17 #include <linux/thermal.h>
  18
  19 #include <linux/iio/iio.h>
  20 #include <linux/iio/driver.h>
  21 #include <linux/iio/machine.h>
  22 #include <linux/mfd/axp20x.h>
  23
  24 #define AXP20X_ADC_EN1_MASK                     GENMASK(7, 0)
  25
  26 #define AXP20X_ADC_EN2_MASK                     (GENMASK(3, 2) | BIT(7))
  27 #define AXP22X_ADC_EN1_MASK                     (GENMASK(7, 5) | BIT(0))
  28
  29 #define AXP20X_GPIO10_IN_RANGE_GPIO0            BIT(0)
  30 #define AXP20X_GPIO10_IN_RANGE_GPIO1            BIT(1)
  31 #define AXP20X_GPIO10_IN_RANGE_GPIO0_VAL(x)     ((x) & BIT(0))
  32 #define AXP20X_GPIO10_IN_RANGE_GPIO1_VAL(x)     (((x) & BIT(0)) << 1)
  33
  34 #define AXP20X_ADC_RATE_MASK                    GENMASK(7, 6)
  35 #define AXP813_V_I_ADC_RATE_MASK                GENMASK(5, 4)
  36 #define AXP813_ADC_RATE_MASK                    (AXP20X_ADC_RATE_MASK | AXP813_V_I_ADC_RATE_MASK)
  37 #define AXP20X_ADC_RATE_HZ(x)                   ((ilog2((x) / 25) << 6) & AXP20X_ADC_RATE_MASK)
  38 #define AXP22X_ADC_RATE_HZ(x)                   ((ilog2((x) / 100) << 6) & AXP20X_ADC_RATE_MASK)
  39 #define AXP813_TS_GPIO0_ADC_RATE_HZ(x)          AXP20X_ADC_RATE_HZ(x)
  40 #define AXP813_V_I_ADC_RATE_HZ(x)               ((ilog2((x) / 100) << 4) & AXP813_V_I_ADC_RATE_MASK)
  41 #define AXP813_ADC_RATE_HZ(x)                   (AXP20X_ADC_RATE_HZ(x) | AXP813_V_I_ADC_RATE_HZ(x))
  42
  43 #define AXP20X_ADC_CHANNEL(_channel, _name, _type, _reg)        \
  44         {                                                       \
  45                 .type = _type,                                  \
  46                 .indexed = 1,                                   \
  47                 .channel = _channel,                            \
  48                 .address = _reg,                                \
  49                 .info_mask_separate = BIT(IIO_CHAN_INFO_RAW) |  \
  50                                       BIT(IIO_CHAN_INFO_SCALE), \
  51                 .datasheet_name = _name,                        \
  52         }
  53
  54 #define AXP20X_ADC_CHANNEL_OFFSET(_channel, _name, _type, _reg) \
  55         {                                                       \
  56                 .type = _type,                                  \
  57                 .indexed = 1,                                   \
  58                 .channel = _channel,                            \
  59                 .address = _reg,                                \
  60                 .info_mask_separate = BIT(IIO_CHAN_INFO_RAW) |  \
  61                                       BIT(IIO_CHAN_INFO_SCALE) |\
  62                                       BIT(IIO_CHAN_INFO_OFFSET),\
  63                 .datasheet_name = _name,                        \
  64         }
  65
  66 struct axp_data;
  67
  68 struct axp20x_adc_iio {
  69         struct regmap           *regmap;
  70         const struct axp_data   *data;
  71 };
  72
  73 enum axp20x_adc_channel_v {
  74         AXP20X_ACIN_V = 0,
  75         AXP20X_VBUS_V,
  76         AXP20X_TS_IN,
  77         AXP20X_GPIO0_V,
  78         AXP20X_GPIO1_V,
  79         AXP20X_IPSOUT_V,
  80         AXP20X_BATT_V,
  81 };
  82
  83 enum axp20x_adc_channel_i {
  84         AXP20X_ACIN_I = 0,
  85         AXP20X_VBUS_I,
  86         AXP20X_BATT_CHRG_I,
  87         AXP20X_BATT_DISCHRG_I,
  88 };
  89
  90 enum axp22x_adc_channel_v {
  91         AXP22X_TS_IN = 0,
  92         AXP22X_BATT_V,
  93 };
  94
  95 enum axp22x_adc_channel_i {
  96         AXP22X_BATT_CHRG_I = 1,
  97         AXP22X_BATT_DISCHRG_I,
  98 };
  99
 100 enum axp813_adc_channel_v {
 101         AXP813_TS_IN = 0,
 102         AXP813_GPIO0_V,
 103         AXP813_BATT_V,
 104 };
 105
 106 static struct iio_map axp20x_maps[] = {
 107         {
 108                 .consumer_dev_name = "axp20x-usb-power-supply",
 109                 .consumer_channel = "vbus_v",
 110                 .adc_channel_label = "vbus_v",
 111         }, {
 112                 .consumer_dev_name = "axp20x-usb-power-supply",
 113                 .consumer_channel = "vbus_i",
 114                 .adc_channel_label = "vbus_i",
 115         }, {
 116                 .consumer_dev_name = "axp20x-ac-power-supply",
 117                 .consumer_channel = "acin_v",
 118                 .adc_channel_label = "acin_v",
 119         }, {
 120                 .consumer_dev_name = "axp20x-ac-power-supply",
 121                 .consumer_channel = "acin_i",
 122                 .adc_channel_label = "acin_i",
 123         }, {
 124                 .consumer_dev_name = "axp20x-battery-power-supply",
 125                 .consumer_channel = "batt_v",
 126                 .adc_channel_label = "batt_v",
 127         }, {
 128                 .consumer_dev_name = "axp20x-battery-power-supply",
 129                 .consumer_channel = "batt_chrg_i",
 130                 .adc_channel_label = "batt_chrg_i",
 131         }, {
 132                 .consumer_dev_name = "axp20x-battery-power-supply",
 133                 .consumer_channel = "batt_dischrg_i",
 134                 .adc_channel_label = "batt_dischrg_i",
 135         }, { /* sentinel */ }
 136 };
 137
 138 static struct iio_map axp22x_maps[] = {
 139         {
 140                 .consumer_dev_name = "axp20x-battery-power-supply",
 141                 .consumer_channel = "batt_v",
 142                 .adc_channel_label = "batt_v",
 143         }, {
 144                 .consumer_dev_name = "axp20x-battery-power-supply",
 145                 .consumer_channel = "batt_chrg_i",
 146                 .adc_channel_label = "batt_chrg_i",
 147         }, {
 148                 .consumer_dev_name = "axp20x-battery-power-supply",
 149                 .consumer_channel = "batt_dischrg_i",
 150                 .adc_channel_label = "batt_dischrg_i",
 151         }, { /* sentinel */ }
 152 };
 153
 154 /*
 155  * Channels are mapped by physical system. Their channels share the same index.
 156  * i.e. acin_i is in_current0_raw and acin_v is in_voltage0_raw.
 157  * The only exception is for the battery. batt_v will be in_voltage6_raw and
 158  * charge current in_current6_raw and discharge current will be in_current7_raw.
 159  */
 160 static const struct iio_chan_spec axp20x_adc_channels[] = {
 161         AXP20X_ADC_CHANNEL(AXP20X_ACIN_V, "acin_v", IIO_VOLTAGE,
 162                            AXP20X_ACIN_V_ADC_H),
 163         AXP20X_ADC_CHANNEL(AXP20X_ACIN_I, "acin_i", IIO_CURRENT,
 164                            AXP20X_ACIN_I_ADC_H),
 165         AXP20X_ADC_CHANNEL(AXP20X_VBUS_V, "vbus_v", IIO_VOLTAGE,
 166                            AXP20X_VBUS_V_ADC_H),
 167         AXP20X_ADC_CHANNEL(AXP20X_VBUS_I, "vbus_i", IIO_CURRENT,
 168                            AXP20X_VBUS_I_ADC_H),
 169         {
 170                 .type = IIO_TEMP,
 171                 .address = AXP20X_TEMP_ADC_H,
 172                 .info_mask_separate = BIT(IIO_CHAN_INFO_RAW) |
 173                                       BIT(IIO_CHAN_INFO_SCALE) |
 174                                       BIT(IIO_CHAN_INFO_OFFSET),
 175                 .datasheet_name = "pmic_temp",
 176         },
 177         AXP20X_ADC_CHANNEL_OFFSET(AXP20X_GPIO0_V, "gpio0_v", IIO_VOLTAGE,
 178                                   AXP20X_GPIO0_V_ADC_H),
 179         AXP20X_ADC_CHANNEL_OFFSET(AXP20X_GPIO1_V, "gpio1_v", IIO_VOLTAGE,
 180                                   AXP20X_GPIO1_V_ADC_H),
 181         AXP20X_ADC_CHANNEL(AXP20X_IPSOUT_V, "ipsout_v", IIO_VOLTAGE,
 182                            AXP20X_IPSOUT_V_HIGH_H),
 183         AXP20X_ADC_CHANNEL(AXP20X_BATT_V, "batt_v", IIO_VOLTAGE,
 184                            AXP20X_BATT_V_H),
 185         AXP20X_ADC_CHANNEL(AXP20X_BATT_CHRG_I, "batt_chrg_i", IIO_CURRENT,
 186                            AXP20X_BATT_CHRG_I_H),
 187         AXP20X_ADC_CHANNEL(AXP20X_BATT_DISCHRG_I, "batt_dischrg_i", IIO_CURRENT,
 188                            AXP20X_BATT_DISCHRG_I_H),
 189         AXP20X_ADC_CHANNEL(AXP20X_TS_IN, "ts_v", IIO_VOLTAGE,
 190                            AXP20X_TS_IN_H),
 191 };
 192
 193 static const struct iio_chan_spec axp22x_adc_channels[] = {
 194         {
 195                 .type = IIO_TEMP,
 196                 .address = AXP22X_PMIC_TEMP_H,
 197                 .info_mask_separate = BIT(IIO_CHAN_INFO_RAW) |
 198                                       BIT(IIO_CHAN_INFO_SCALE) |
 199                                       BIT(IIO_CHAN_INFO_OFFSET),
 200                 .datasheet_name = "pmic_temp",
 201         },
 202         AXP20X_ADC_CHANNEL(AXP22X_BATT_V, "batt_v", IIO_VOLTAGE,
 203                            AXP20X_BATT_V_H),
 204         AXP20X_ADC_CHANNEL(AXP22X_BATT_CHRG_I, "batt_chrg_i", IIO_CURRENT,
 205                            AXP20X_BATT_CHRG_I_H),
 206         AXP20X_ADC_CHANNEL(AXP22X_BATT_DISCHRG_I, "batt_dischrg_i", IIO_CURRENT,
 207                            AXP20X_BATT_DISCHRG_I_H),
 208         AXP20X_ADC_CHANNEL(AXP22X_TS_IN, "ts_v", IIO_VOLTAGE,
 209                            AXP22X_TS_ADC_H),
 210 };
 211
 212 static const struct iio_chan_spec axp813_adc_channels[] = {
 213         {
 214                 .type = IIO_TEMP,
 215                 .address = AXP22X_PMIC_TEMP_H,
 216                 .info_mask_separate = BIT(IIO_CHAN_INFO_RAW) |
 217                                       BIT(IIO_CHAN_INFO_SCALE) |
 218                                       BIT(IIO_CHAN_INFO_OFFSET),
 219                 .datasheet_name = "pmic_temp",
 220         },
 221         AXP20X_ADC_CHANNEL(AXP813_GPIO0_V, "gpio0_v", IIO_VOLTAGE,
 222                            AXP288_GP_ADC_H),
 223         AXP20X_ADC_CHANNEL(AXP813_BATT_V, "batt_v", IIO_VOLTAGE,
 224                            AXP20X_BATT_V_H),
 225         AXP20X_ADC_CHANNEL(AXP22X_BATT_CHRG_I, "batt_chrg_i", IIO_CURRENT,
 226                            AXP20X_BATT_CHRG_I_H),
 227         AXP20X_ADC_CHANNEL(AXP22X_BATT_DISCHRG_I, "batt_dischrg_i", IIO_CURRENT,
 228                            AXP20X_BATT_DISCHRG_I_H),
 229         AXP20X_ADC_CHANNEL(AXP813_TS_IN, "ts_v", IIO_VOLTAGE,
 230                            AXP288_TS_ADC_H),
 231 };
 232
 233 static int axp20x_adc_raw(struct iio_dev *indio_dev,
 234                           struct iio_chan_spec const *chan, int *val)
 235 {
 236         struct axp20x_adc_iio *info = iio_priv(indio_dev);
 237         int size = 12;
 238
 239         /*
 240          * N.B.:  Unlike the Chinese datasheets tell, the charging current is
 241          * stored on 12 bits, not 13 bits. Only discharging current is on 13
 242          * bits.
 243          */
 244         if (chan->type == IIO_CURRENT && chan->channel == AXP20X_BATT_DISCHRG_I)
 245                 size = 13;
 246         else
 247                 size = 12;
 248
 249         *val = axp20x_read_variable_width(info->regmap, chan->address, size);
 250         if (*val < 0)
 251                 return *val;
 252
 253         return IIO_VAL_INT;
 254 }
 255
 256 static int axp22x_adc_raw(struct iio_dev *indio_dev,
 257                           struct iio_chan_spec const *chan, int *val)
 258 {
 259         struct axp20x_adc_iio *info = iio_priv(indio_dev);
 260
 261         *val = axp20x_read_variable_width(info->regmap, chan->address, 12);
 262         if (*val < 0)
 263                 return *val;
 264
 265         return IIO_VAL_INT;
 266 }
 267
 268 static int axp813_adc_raw(struct iio_dev *indio_dev,
 269                           struct iio_chan_spec const *chan, int *val)
 270 {
 271         struct axp20x_adc_iio *info = iio_priv(indio_dev);
 272
 273         *val = axp20x_read_variable_width(info->regmap, chan->address, 12);
 274         if (*val < 0)
 275                 return *val;
 276
 277         return IIO_VAL_INT;
 278 }
 279
 280 static int axp20x_adc_scale_voltage(int channel, int *val, int *val2)
 281 {
 282         switch (channel) {
 283         case AXP20X_ACIN_V:
 284         case AXP20X_VBUS_V:
 285                 *val = 1;
 286                 *val2 = 700000;
 287                 return IIO_VAL_INT_PLUS_MICRO;
 288
 289         case AXP20X_GPIO0_V:
 290         case AXP20X_GPIO1_V:
 291                 *val = 0;
 292                 *val2 = 500000;
 293                 return IIO_VAL_INT_PLUS_MICRO;
 294
 295         case AXP20X_BATT_V:
 296                 *val = 1;
 297                 *val2 = 100000;
 298                 return IIO_VAL_INT_PLUS_MICRO;
 299
 300         case AXP20X_IPSOUT_V:
 301                 *val = 1;
 302                 *val2 = 400000;
 303                 return IIO_VAL_INT_PLUS_MICRO;
 304
 305         case AXP20X_TS_IN:
 306                 /* 0.8 mV per LSB */
 307                 *val = 0;
 308                 *val2 = 800000;
 309                 return IIO_VAL_INT_PLUS_MICRO;
 310
 311         default:
 312                 return -EINVAL;
 313         }
 314 }
 315
 316 static int axp22x_adc_scale_voltage(int channel, int *val, int *val2)
 317 {
 318         switch (channel) {
 319         case AXP22X_BATT_V:
 320                 /* 1.1 mV per LSB */
 321                 *val = 1;
 322                 *val2 = 100000;
 323                 return IIO_VAL_INT_PLUS_MICRO;
 324
 325         case AXP22X_TS_IN:
 326                 /* 0.8 mV per LSB */
 327                 *val = 0;
 328                 *val2 = 800000;
 329                 return IIO_VAL_INT_PLUS_MICRO;
 330
 331         default:
 332                 return -EINVAL;
 333         }
 334 }
 335 static int axp813_adc_scale_voltage(int channel, int *val, int *val2)
 336 {
 337         switch (channel) {
 338         case AXP813_GPIO0_V:
 339                 *val = 0;
 340                 *val2 = 800000;
 341                 return IIO_VAL_INT_PLUS_MICRO;
 342
 343         case AXP813_BATT_V:
 344                 *val = 1;
 345                 *val2 = 100000;
 346                 return IIO_VAL_INT_PLUS_MICRO;
 347
 348         case AXP813_TS_IN:
 349                 /* 0.8 mV per LSB */
 350                 *val = 0;
 351                 *val2 = 800000;
 352                 return IIO_VAL_INT_PLUS_MICRO;
 353
 354         default:
 355                 return -EINVAL;
 356         }
 357 }
 358
 359 static int axp20x_adc_scale_current(int channel, int *val, int *val2)
 360 {
 361         switch (channel) {
 362         case AXP20X_ACIN_I:
 363                 *val = 0;
 364                 *val2 = 625000;
 365                 return IIO_VAL_INT_PLUS_MICRO;
 366
 367         case AXP20X_VBUS_I:
 368                 *val = 0;
 369                 *val2 = 375000;
 370                 return IIO_VAL_INT_PLUS_MICRO;
 371
 372         case AXP20X_BATT_DISCHRG_I:
 373         case AXP20X_BATT_CHRG_I:
 374                 *val = 0;
 375                 *val2 = 500000;
 376                 return IIO_VAL_INT_PLUS_MICRO;
 377
 378         default:
 379                 return -EINVAL;
 380         }
 381 }
 382
 383 static int axp20x_adc_scale(struct iio_chan_spec const *chan, int *val,
 384                             int *val2)
 385 {
 386         switch (chan->type) {
 387         case IIO_VOLTAGE:
 388                 return axp20x_adc_scale_voltage(chan->channel, val, val2);
 389
 390         case IIO_CURRENT:
 391                 return axp20x_adc_scale_current(chan->channel, val, val2);
 392
 393         case IIO_TEMP:
 394                 *val = 100;
 395                 return IIO_VAL_INT;
 396
 397         default:
 398                 return -EINVAL;
 399         }
 400 }
 401
 402 static int axp22x_adc_scale(struct iio_chan_spec const *chan, int *val,
 403                             int *val2)
 404 {
 405         switch (chan->type) {
 406         case IIO_VOLTAGE:
 407                 return axp22x_adc_scale_voltage(chan->channel, val, val2);
 408
 409         case IIO_CURRENT:
 410                 *val = 1;
 411                 return IIO_VAL_INT;
 412
 413         case IIO_TEMP:
 414                 *val = 100;
 415                 return IIO_VAL_INT;
 416
 417         default:
 418                 return -EINVAL;
 419         }
 420 }
 421
 422 static int axp813_adc_scale(struct iio_chan_spec const *chan, int *val,
 423                             int *val2)
 424 {
 425         switch (chan->type) {
 426         case IIO_VOLTAGE:
 427                 return axp813_adc_scale_voltage(chan->channel, val, val2);
 428
 429         case IIO_CURRENT:
 430                 *val = 1;
 431                 return IIO_VAL_INT;
 432
 433         case IIO_TEMP:
 434                 *val = 100;
 435                 return IIO_VAL_INT;
 436
 437         default:
 438                 return -EINVAL;
 439         }
 440 }
 441
 442 static int axp20x_adc_offset_voltage(struct iio_dev *indio_dev, int channel,
 443                                      int *val)
 444 {
 445         struct axp20x_adc_iio *info = iio_priv(indio_dev);
 446         int ret;
 447
 448         ret = regmap_read(info->regmap, AXP20X_GPIO10_IN_RANGE, val);
 449         if (ret < 0)
 450                 return ret;
 451
 452         switch (channel) {
 453         case AXP20X_GPIO0_V:
 454                 *val &= AXP20X_GPIO10_IN_RANGE_GPIO0;
 455                 break;
 456
 457         case AXP20X_GPIO1_V:
 458                 *val &= AXP20X_GPIO10_IN_RANGE_GPIO1;
 459                 break;
 460
 461         default:
 462                 return -EINVAL;
 463         }
 464
 465         *val = *val ? 700000 : 0;
 466
 467         return IIO_VAL_INT;
 468 }
 469
 470 static int axp20x_adc_offset(struct iio_dev *indio_dev,
 471                              struct iio_chan_spec const *chan, int *val)
 472 {
 473         switch (chan->type) {
 474         case IIO_VOLTAGE:
 475                 return axp20x_adc_offset_voltage(indio_dev, chan->channel, val);
 476
 477         case IIO_TEMP:
 478                 *val = -1447;
 479                 return IIO_VAL_INT;
 480
 481         default:
 482                 return -EINVAL;
 483         }
 484 }
 485
 486 static int axp20x_read_raw(struct iio_dev *indio_dev,
 487                            struct iio_chan_spec const *chan, int *val,
 488                            int *val2, long mask)
 489 {
 490         switch (mask) {
 491         case IIO_CHAN_INFO_OFFSET:
 492                 return axp20x_adc_offset(indio_dev, chan, val);
 493
 494         case IIO_CHAN_INFO_SCALE:
 495                 return axp20x_adc_scale(chan, val, val2);
 496
 497         case IIO_CHAN_INFO_RAW:
 498                 return axp20x_adc_raw(indio_dev, chan, val);
 499
 500         default:
 501                 return -EINVAL;
 502         }
 503 }
 504
 505 static int axp22x_read_raw(struct iio_dev *indio_dev,
 506                            struct iio_chan_spec const *chan, int *val,
 507                            int *val2, long mask)
 508 {
 509         switch (mask) {
 510         case IIO_CHAN_INFO_OFFSET:
 511                 /* For PMIC temp only */
 512                 *val = -2677;
 513                 return IIO_VAL_INT;
 514
 515         case IIO_CHAN_INFO_SCALE:
 516                 return axp22x_adc_scale(chan, val, val2);
 517
 518         case IIO_CHAN_INFO_RAW:
 519                 return axp22x_adc_raw(indio_dev, chan, val);
 520
 521         default:
 522                 return -EINVAL;
 523         }
 524 }
 525
 526 static int axp813_read_raw(struct iio_dev *indio_dev,
 527                            struct iio_chan_spec const *chan, int *val,
 528                            int *val2, long mask)
 529 {
 530         switch (mask) {
 531         case IIO_CHAN_INFO_OFFSET:
 532                 *val = -2667;
 533                 return IIO_VAL_INT;
 534
 535         case IIO_CHAN_INFO_SCALE:
 536                 return axp813_adc_scale(chan, val, val2);
 537
 538         case IIO_CHAN_INFO_RAW:
 539                 return axp813_adc_raw(indio_dev, chan, val);
 540
 541         default:
 542                 return -EINVAL;
 543         }
 544 }
 545
 546 static int axp20x_write_raw(struct iio_dev *indio_dev,
 547                             struct iio_chan_spec const *chan, int val, int val2,
 548                             long mask)
 549 {
 550         struct axp20x_adc_iio *info = iio_priv(indio_dev);
 551         unsigned int reg, regval;
 552
 553         /*
 554          * The AXP20X PMIC allows the user to choose between 0V and 0.7V offsets
 555          * for (independently) GPIO0 and GPIO1 when in ADC mode.
 556          */
 557         if (mask != IIO_CHAN_INFO_OFFSET)
 558                 return -EINVAL;
 559
 560         if (val != 0 && val != 700000)
 561                 return -EINVAL;
 562
 563         val = val ? 1 : 0;
 564
 565         switch (chan->channel) {
 566         case AXP20X_GPIO0_V:
 567                 reg = AXP20X_GPIO10_IN_RANGE_GPIO0;
 568                 regval = AXP20X_GPIO10_IN_RANGE_GPIO0_VAL(val);
 569                 break;
 570
 571         case AXP20X_GPIO1_V:
 572                 reg = AXP20X_GPIO10_IN_RANGE_GPIO1;
 573                 regval = AXP20X_GPIO10_IN_RANGE_GPIO1_VAL(val);
 574                 break;
 575
 576         default:
 577                 return -EINVAL;
 578         }
 579
 580         return regmap_update_bits(info->regmap, AXP20X_GPIO10_IN_RANGE, reg,
 581                                   regval);
 582 }
 583
 584 static const struct iio_info axp20x_adc_iio_info = {
 585         .read_raw = axp20x_read_raw,
 586         .write_raw = axp20x_write_raw,
 587 };
 588
 589 static const struct iio_info axp22x_adc_iio_info = {
 590         .read_raw = axp22x_read_raw,
 591 };
 592
 593 static const struct iio_info axp813_adc_iio_info = {
 594         .read_raw = axp813_read_raw,
 595 };
 596
 597 static int axp20x_adc_rate(struct axp20x_adc_iio *info, int rate)
 598 {
 599         return regmap_update_bits(info->regmap, AXP20X_ADC_RATE,
 600                                   AXP20X_ADC_RATE_MASK,
 601                                   AXP20X_ADC_RATE_HZ(rate));
 602 }
 603
 604 static int axp22x_adc_rate(struct axp20x_adc_iio *info, int rate)
 605 {
 606         return regmap_update_bits(info->regmap, AXP20X_ADC_RATE,
 607                                   AXP20X_ADC_RATE_MASK,
 608                                   AXP22X_ADC_RATE_HZ(rate));
 609 }
 610
 611 static int axp813_adc_rate(struct axp20x_adc_iio *info, int rate)
 612 {
 613         return regmap_update_bits(info->regmap, AXP813_ADC_RATE,
 614                                  AXP813_ADC_RATE_MASK,
 615                                  AXP813_ADC_RATE_HZ(rate));
 616 }
 617
 618 struct axp_data {
 619         const struct iio_info           *iio_info;
 620         int                             num_channels;
 621         struct iio_chan_spec const      *channels;
 622         unsigned long                   adc_en1_mask;
 623         int                             (*adc_rate)(struct axp20x_adc_iio *info,
 624                                                     int rate);
 625         bool                            adc_en2;
 626         struct iio_map                  *maps;
 627 };
 628
 629 static const struct axp_data axp20x_data = {
 630         .iio_info = &axp20x_adc_iio_info,
 631         .num_channels = ARRAY_SIZE(axp20x_adc_channels),
 632         .channels = axp20x_adc_channels,
 633         .adc_en1_mask = AXP20X_ADC_EN1_MASK,
 634         .adc_rate = axp20x_adc_rate,
 635         .adc_en2 = true,
 636         .maps = axp20x_maps,
 637 };
 638
 639 static const struct axp_data axp22x_data = {
 640         .iio_info = &axp22x_adc_iio_info,
 641         .num_channels = ARRAY_SIZE(axp22x_adc_channels),
 642         .channels = axp22x_adc_channels,
 643         .adc_en1_mask = AXP22X_ADC_EN1_MASK,
 644         .adc_rate = axp22x_adc_rate,
 645         .adc_en2 = false,
 646         .maps = axp22x_maps,
 647 };
 648
 649 static const struct axp_data axp813_data = {
 650         .iio_info = &axp813_adc_iio_info,
 651         .num_channels = ARRAY_SIZE(axp813_adc_channels),
 652         .channels = axp813_adc_channels,
 653         .adc_en1_mask = AXP22X_ADC_EN1_MASK,
 654         .adc_rate = axp813_adc_rate,
 655         .adc_en2 = false,
 656         .maps = axp22x_maps,
 657 };
 658
 659 static const struct of_device_id axp20x_adc_of_match[] = {
 660         { .compatible = "x-powers,axp209-adc", .data = (void *)&axp20x_data, },
 661         { .compatible = "x-powers,axp221-adc", .data = (void *)&axp22x_data, },
 662         { .compatible = "x-powers,axp813-adc", .data = (void *)&axp813_data, },
 663         { /* sentinel */ }
 664 };
 665 MODULE_DEVICE_TABLE(of, axp20x_adc_of_match);
 666
 667 static const struct platform_device_id axp20x_adc_id_match[] = {
 668         { .name = "axp20x-adc", .driver_data = (kernel_ulong_t)&axp20x_data, },
 669         { .name = "axp22x-adc", .driver_data = (kernel_ulong_t)&axp22x_data, },
 670         { .name = "axp813-adc", .driver_data = (kernel_ulong_t)&axp813_data, },
 671         { /* sentinel */ },
 672 };
 673 MODULE_DEVICE_TABLE(platform, axp20x_adc_id_match);
 674
 675 static int axp20x_probe(struct platform_device *pdev)
 676 {
 677         struct axp20x_adc_iio *info;
 678         struct iio_dev *indio_dev;
 679         struct axp20x_dev *axp20x_dev;
 680         int ret;
 681
 682         axp20x_dev = dev_get_drvdata(pdev->dev.parent);
 683
 684         indio_dev = devm_iio_device_alloc(&pdev->dev, sizeof(*info));
 685         if (!indio_dev)
 686                 return -ENOMEM;
 687
 688         info = iio_priv(indio_dev);
 689         platform_set_drvdata(pdev, indio_dev);
 690
 691         info->regmap = axp20x_dev->regmap;
 692         indio_dev->modes = INDIO_DIRECT_MODE;
 693
 694         if (!dev_fwnode(&pdev->dev)) {
 695                 const struct platform_device_id *id;
 696
 697                 id = platform_get_device_id(pdev);
 698                 info->data = (const struct axp_data *)id->driver_data;
 699         } else {
 700                 struct device *dev = &pdev->dev;
 701
 702                 info->data = device_get_match_data(dev);
 703         }
 704
 705         indio_dev->name = platform_get_device_id(pdev)->name;
 706         indio_dev->info = info->data->iio_info;
 707         indio_dev->num_channels = info->data->num_channels;
 708         indio_dev->channels = info->data->channels;
 709
 710         /* Enable the ADCs on IP */
 711         regmap_write(info->regmap, AXP20X_ADC_EN1, info->data->adc_en1_mask);
 712
 713         if (info->data->adc_en2)
 714                 /* Enable GPIO0/1 and internal temperature ADCs */
 715                 regmap_update_bits(info->regmap, AXP20X_ADC_EN2,
 716                                    AXP20X_ADC_EN2_MASK, AXP20X_ADC_EN2_MASK);
 717
 718         /* Configure ADCs rate */
 719         info->data->adc_rate(info, 100);
 720
 721         ret = iio_map_array_register(indio_dev, info->data->maps);
 722         if (ret < 0) {
 723                 dev_err(&pdev->dev, "failed to register IIO maps: %d\n", ret);
 724                 goto fail_map;
 725         }
 726
 727         ret = iio_device_register(indio_dev);
 728         if (ret < 0) {
 729                 dev_err(&pdev->dev, "could not register the device\n");
 730                 goto fail_register;
 731         }
 732
 733         return 0;
 734
 735 fail_register:
 736         iio_map_array_unregister(indio_dev);
 737
 738 fail_map:
 739         regmap_write(info->regmap, AXP20X_ADC_EN1, 0);
 740
 741         if (info->data->adc_en2)
 742                 regmap_write(info->regmap, AXP20X_ADC_EN2, 0);
 743
 744         return ret;
 745 }
 746
 747 static int axp20x_remove(struct platform_device *pdev)
 748 {
 749         struct iio_dev *indio_dev = platform_get_drvdata(pdev);
 750         struct axp20x_adc_iio *info = iio_priv(indio_dev);
 751
 752         iio_device_unregister(indio_dev);
 753         iio_map_array_unregister(indio_dev);
 754
 755         regmap_write(info->regmap, AXP20X_ADC_EN1, 0);
 756
 757         if (info->data->adc_en2)
 758                 regmap_write(info->regmap, AXP20X_ADC_EN2, 0);
 759
 760         return 0;
 761 }
 762
 763 static struct platform_driver axp20x_adc_driver = {
 764         .driver = {
 765                 .name = "axp20x-adc",
 766                 .of_match_table = axp20x_adc_of_match,
 767         },
 768         .id_table = axp20x_adc_id_match,
 769         .probe = axp20x_probe,
 770         .remove = axp20x_remove,
 771 };
 772
 773 module_platform_driver(axp20x_adc_driver);
 774
 775 MODULE_DESCRIPTION("ADC driver for AXP20X and AXP22X PMICs");
 776 MODULE_AUTHOR("Quentin Schulz <quentin.schulz@free-electrons.com>");
 777 MODULE_LICENSE("GPL");