drivers/iio/adc/axp20x_adc.c

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