root/drivers/regulator/max77620-regulator.c
// SPDX-License-Identifier: GPL-2.0-only
/*
 * Maxim MAX77620 Regulator driver
 *
 * Copyright (c) 2016-2018, NVIDIA CORPORATION.  All rights reserved.
 *
 * Author: Mallikarjun Kasoju <mkasoju@nvidia.com>
 *      Laxman Dewangan <ldewangan@nvidia.com>
 */

#include <linux/init.h>
#include <linux/mfd/max77620.h>
#include <linux/module.h>
#include <linux/of.h>
#include <linux/platform_device.h>
#include <linux/regmap.h>
#include <linux/regulator/driver.h>
#include <linux/regulator/machine.h>
#include <linux/regulator/of_regulator.h>

#define max77620_rails(_name)   "max77620-"#_name

/* Power Mode */
#define MAX77620_POWER_MODE_NORMAL              3
#define MAX77620_POWER_MODE_LPM                 2
#define MAX77620_POWER_MODE_GLPM                1
#define MAX77620_POWER_MODE_DISABLE             0

/* SD Slew Rate */
#define MAX77620_SD_SR_13_75                    0
#define MAX77620_SD_SR_27_5                     1
#define MAX77620_SD_SR_55                       2
#define MAX77620_SD_SR_100                      3

enum max77620_regulators {
        MAX77620_REGULATOR_ID_SD0,
        MAX77620_REGULATOR_ID_SD1,
        MAX77620_REGULATOR_ID_SD2,
        MAX77620_REGULATOR_ID_SD3,
        MAX77620_REGULATOR_ID_SD4,
        MAX77620_REGULATOR_ID_LDO0,
        MAX77620_REGULATOR_ID_LDO1,
        MAX77620_REGULATOR_ID_LDO2,
        MAX77620_REGULATOR_ID_LDO3,
        MAX77620_REGULATOR_ID_LDO4,
        MAX77620_REGULATOR_ID_LDO5,
        MAX77620_REGULATOR_ID_LDO6,
        MAX77620_REGULATOR_ID_LDO7,
        MAX77620_REGULATOR_ID_LDO8,
        MAX77620_NUM_REGS,
};

/* Regulator types */
enum max77620_regulator_type {
        MAX77620_REGULATOR_TYPE_SD,
        MAX77620_REGULATOR_TYPE_LDO_N,
        MAX77620_REGULATOR_TYPE_LDO_P,
};

struct max77620_regulator_info {
        u8 type;
        u8 fps_addr;
        u8 volt_addr;
        u8 cfg_addr;
        u8 power_mode_mask;
        u8 power_mode_shift;
        u8 remote_sense_addr;
        u8 remote_sense_mask;
        struct regulator_desc desc;
};

struct max77620_regulator_pdata {
        int active_fps_src;
        int active_fps_pd_slot;
        int active_fps_pu_slot;
        int suspend_fps_src;
        int suspend_fps_pd_slot;
        int suspend_fps_pu_slot;
        int current_mode;
        int power_ok;
        int ramp_rate_setting;
};

struct max77620_regulator {
        struct device *dev;
        struct regmap *rmap;
        struct max77620_regulator_info *rinfo[MAX77620_NUM_REGS];
        struct max77620_regulator_pdata reg_pdata[MAX77620_NUM_REGS];
        int enable_power_mode[MAX77620_NUM_REGS];
        int current_power_mode[MAX77620_NUM_REGS];
        int active_fps_src[MAX77620_NUM_REGS];
};

#define fps_src_name(fps_src)   \
        (fps_src == MAX77620_FPS_SRC_0 ? "FPS_SRC_0" :  \
        fps_src == MAX77620_FPS_SRC_1 ? "FPS_SRC_1" :   \
        fps_src == MAX77620_FPS_SRC_2 ? "FPS_SRC_2" : "FPS_SRC_NONE")

static int max77620_regulator_get_fps_src(struct max77620_regulator *pmic,
                                          int id)
{
        struct max77620_regulator_info *rinfo = pmic->rinfo[id];
        unsigned int val;
        int ret;

        ret = regmap_read(pmic->rmap, rinfo->fps_addr, &val);
        if (ret < 0) {
                dev_err(pmic->dev, "Reg 0x%02x read failed %d\n",
                        rinfo->fps_addr, ret);
                return ret;
        }

        return (val & MAX77620_FPS_SRC_MASK) >> MAX77620_FPS_SRC_SHIFT;
}

static int max77620_regulator_set_fps_src(struct max77620_regulator *pmic,
                                          int fps_src, int id)
{
        struct max77620_regulator_info *rinfo = pmic->rinfo[id];
        unsigned int val;
        int ret;

        if (!rinfo)
                return 0;

        switch (fps_src) {
        case MAX77620_FPS_SRC_0:
        case MAX77620_FPS_SRC_1:
        case MAX77620_FPS_SRC_2:
        case MAX77620_FPS_SRC_NONE:
                break;

        case MAX77620_FPS_SRC_DEF:
                ret = regmap_read(pmic->rmap, rinfo->fps_addr, &val);
                if (ret < 0) {
                        dev_err(pmic->dev, "Reg 0x%02x read failed %d\n",
                                rinfo->fps_addr, ret);
                        return ret;
                }
                ret = (val & MAX77620_FPS_SRC_MASK) >> MAX77620_FPS_SRC_SHIFT;
                pmic->active_fps_src[id] = ret;
                return 0;

        default:
                dev_err(pmic->dev, "Invalid FPS %d for regulator %d\n",
                        fps_src, id);
                return -EINVAL;
        }

        ret = regmap_update_bits(pmic->rmap, rinfo->fps_addr,
                                 MAX77620_FPS_SRC_MASK,
                                 fps_src << MAX77620_FPS_SRC_SHIFT);
        if (ret < 0) {
                dev_err(pmic->dev, "Reg 0x%02x update failed %d\n",
                        rinfo->fps_addr, ret);
                return ret;
        }
        pmic->active_fps_src[id] = fps_src;

        return 0;
}

static int max77620_regulator_set_fps_slots(struct max77620_regulator *pmic,
                                            int id, bool is_suspend)
{
        struct max77620_regulator_pdata *rpdata = &pmic->reg_pdata[id];
        struct max77620_regulator_info *rinfo = pmic->rinfo[id];
        unsigned int val = 0;
        unsigned int mask = 0;
        int pu = rpdata->active_fps_pu_slot;
        int pd = rpdata->active_fps_pd_slot;
        int ret = 0;

        if (!rinfo)
                return 0;

        if (is_suspend) {
                pu = rpdata->suspend_fps_pu_slot;
                pd = rpdata->suspend_fps_pd_slot;
        }

        /* FPS power up period setting */
        if (pu >= 0) {
                val |= (pu << MAX77620_FPS_PU_PERIOD_SHIFT);
                mask |= MAX77620_FPS_PU_PERIOD_MASK;
        }

        /* FPS power down period setting */
        if (pd >= 0) {
                val |= (pd << MAX77620_FPS_PD_PERIOD_SHIFT);
                mask |= MAX77620_FPS_PD_PERIOD_MASK;
        }

        if (mask) {
                ret = regmap_update_bits(pmic->rmap, rinfo->fps_addr,
                                         mask, val);
                if (ret < 0) {
                        dev_err(pmic->dev, "Reg 0x%02x update failed: %d\n",
                                rinfo->fps_addr, ret);
                        return ret;
                }
        }

        return ret;
}

static int max77620_regulator_set_power_mode(struct max77620_regulator *pmic,
                                             int power_mode, int id)
{
        struct max77620_regulator_info *rinfo = pmic->rinfo[id];
        u8 mask = rinfo->power_mode_mask;
        u8 shift = rinfo->power_mode_shift;
        u8 addr;
        int ret;

        switch (rinfo->type) {
        case MAX77620_REGULATOR_TYPE_SD:
                addr = rinfo->cfg_addr;
                break;
        default:
                addr = rinfo->volt_addr;
                break;
        }

        ret = regmap_update_bits(pmic->rmap, addr, mask, power_mode << shift);
        if (ret < 0) {
                dev_err(pmic->dev, "Regulator %d mode set failed: %d\n",
                        id, ret);
                return ret;
        }
        pmic->current_power_mode[id] = power_mode;

        return ret;
}

static int max77620_regulator_get_power_mode(struct max77620_regulator *pmic,
                                             int id)
{
        struct max77620_regulator_info *rinfo = pmic->rinfo[id];
        unsigned int val, addr;
        u8 mask = rinfo->power_mode_mask;
        u8 shift = rinfo->power_mode_shift;
        int ret;

        switch (rinfo->type) {
        case MAX77620_REGULATOR_TYPE_SD:
                addr = rinfo->cfg_addr;
                break;
        default:
                addr = rinfo->volt_addr;
                break;
        }

        ret = regmap_read(pmic->rmap, addr, &val);
        if (ret < 0) {
                dev_err(pmic->dev, "Regulator %d: Reg 0x%02x read failed: %d\n",
                        id, addr, ret);
                return ret;
        }

        return (val & mask) >> shift;
}

static int max77620_read_slew_rate(struct max77620_regulator *pmic, int id)
{
        struct max77620_regulator_info *rinfo = pmic->rinfo[id];
        unsigned int rval;
        int slew_rate;
        int ret;

        ret = regmap_read(pmic->rmap, rinfo->cfg_addr, &rval);
        if (ret < 0) {
                dev_err(pmic->dev, "Register 0x%02x read failed: %d\n",
                        rinfo->cfg_addr, ret);
                return ret;
        }

        switch (rinfo->type) {
        case MAX77620_REGULATOR_TYPE_SD:
                slew_rate = (rval >> MAX77620_SD_SR_SHIFT) & 0x3;
                switch (slew_rate) {
                case 0:
                        slew_rate = 13750;
                        break;
                case 1:
                        slew_rate = 27500;
                        break;
                case 2:
                        slew_rate = 55000;
                        break;
                case 3:
                        slew_rate = 100000;
                        break;
                }
                rinfo->desc.ramp_delay = slew_rate;
                break;
        default:
                slew_rate = rval & 0x1;
                switch (slew_rate) {
                case 0:
                        slew_rate = 100000;
                        break;
                case 1:
                        slew_rate = 5000;
                        break;
                }
                rinfo->desc.ramp_delay = slew_rate;
                break;
        }

        return 0;
}

static int max77620_set_slew_rate(struct max77620_regulator *pmic, int id,
                                  int slew_rate)
{
        struct max77620_regulator_info *rinfo = pmic->rinfo[id];
        unsigned int val;
        int ret;
        u8 mask;

        if (rinfo->type == MAX77620_REGULATOR_TYPE_SD) {
                if (slew_rate <= 13750)
                        val = 0;
                else if (slew_rate <= 27500)
                        val = 1;
                else if (slew_rate <= 55000)
                        val = 2;
                else
                        val = 3;
                val <<= MAX77620_SD_SR_SHIFT;
                mask = MAX77620_SD_SR_MASK;
        } else {
                if (slew_rate <= 5000)
                        val = 1;
                else
                        val = 0;
                mask = MAX77620_LDO_SLEW_RATE_MASK;
        }

        ret = regmap_update_bits(pmic->rmap, rinfo->cfg_addr, mask, val);
        if (ret < 0) {
                dev_err(pmic->dev, "Regulator %d slew rate set failed: %d\n",
                        id, ret);
                return ret;
        }

        return 0;
}

static int max77620_config_power_ok(struct max77620_regulator *pmic, int id)
{
        struct max77620_regulator_pdata *rpdata = &pmic->reg_pdata[id];
        struct max77620_regulator_info *rinfo = pmic->rinfo[id];
        struct max77620_chip *chip = dev_get_drvdata(pmic->dev->parent);
        u8 val, mask;
        int ret;

        switch (chip->chip_id) {
        case MAX20024:
                if (rpdata->power_ok >= 0) {
                        if (rinfo->type == MAX77620_REGULATOR_TYPE_SD)
                                mask = MAX20024_SD_CFG1_MPOK_MASK;
                        else
                                mask = MAX20024_LDO_CFG2_MPOK_MASK;

                        val = rpdata->power_ok ? mask : 0;

                        ret = regmap_update_bits(pmic->rmap, rinfo->cfg_addr,
                                                 mask, val);
                        if (ret < 0) {
                                dev_err(pmic->dev, "Reg 0x%02x update failed %d\n",
                                        rinfo->cfg_addr, ret);
                                return ret;
                        }
                }
                break;

        default:
                break;
        }

        return 0;
}

static int max77620_init_pmic(struct max77620_regulator *pmic, int id)
{
        struct max77620_regulator_pdata *rpdata = &pmic->reg_pdata[id];
        int ret;

        max77620_config_power_ok(pmic, id);

        /* Update power mode */
        ret = max77620_regulator_get_power_mode(pmic, id);
        if (ret < 0)
                return ret;

        pmic->current_power_mode[id] = ret;
        pmic->enable_power_mode[id] = MAX77620_POWER_MODE_NORMAL;

        if (rpdata->active_fps_src == MAX77620_FPS_SRC_DEF) {
                ret = max77620_regulator_get_fps_src(pmic, id);
                if (ret < 0)
                        return ret;
                rpdata->active_fps_src = ret;
        }

         /* If rails are externally control of FPS then enable it always. */
        if (rpdata->active_fps_src == MAX77620_FPS_SRC_NONE) {
                ret = max77620_regulator_set_power_mode(pmic,
                                        pmic->enable_power_mode[id], id);
                if (ret < 0)
                        return ret;
        } else {
                if (pmic->current_power_mode[id] !=
                     pmic->enable_power_mode[id]) {
                        ret = max77620_regulator_set_power_mode(pmic,
                                        pmic->enable_power_mode[id], id);
                        if (ret < 0)
                                return ret;
                }
        }

        ret = max77620_regulator_set_fps_src(pmic, rpdata->active_fps_src, id);
        if (ret < 0)
                return ret;

        ret = max77620_regulator_set_fps_slots(pmic, id, false);
        if (ret < 0)
                return ret;

        if (rpdata->ramp_rate_setting) {
                ret = max77620_set_slew_rate(pmic, id,
                                             rpdata->ramp_rate_setting);
                if (ret < 0)
                        return ret;
        }

        return 0;
}

static int max77620_regulator_enable(struct regulator_dev *rdev)
{
        struct max77620_regulator *pmic = rdev_get_drvdata(rdev);
        int id = rdev_get_id(rdev);

        if (pmic->active_fps_src[id] != MAX77620_FPS_SRC_NONE)
                return 0;

        return max77620_regulator_set_power_mode(pmic,
                        pmic->enable_power_mode[id], id);
}

static int max77620_regulator_disable(struct regulator_dev *rdev)
{
        struct max77620_regulator *pmic = rdev_get_drvdata(rdev);
        int id = rdev_get_id(rdev);

        if (pmic->active_fps_src[id] != MAX77620_FPS_SRC_NONE)
                return 0;

        return max77620_regulator_set_power_mode(pmic,
                        MAX77620_POWER_MODE_DISABLE, id);
}

static int max77620_regulator_is_enabled(struct regulator_dev *rdev)
{
        struct max77620_regulator *pmic = rdev_get_drvdata(rdev);
        int id = rdev_get_id(rdev);
        int ret;

        if (pmic->active_fps_src[id] != MAX77620_FPS_SRC_NONE)
                return 1;

        ret = max77620_regulator_get_power_mode(pmic, id);
        if (ret < 0)
                return ret;

        if (ret != MAX77620_POWER_MODE_DISABLE)
                return 1;

        return 0;
}

static int max77620_regulator_set_mode(struct regulator_dev *rdev,
                                       unsigned int mode)
{
        struct max77620_regulator *pmic = rdev_get_drvdata(rdev);
        int id = rdev_get_id(rdev);
        struct max77620_regulator_info *rinfo = pmic->rinfo[id];
        struct max77620_regulator_pdata *rpdata = &pmic->reg_pdata[id];
        bool fpwm = false;
        int power_mode;
        int ret;
        u8 val;

        switch (mode) {
        case REGULATOR_MODE_FAST:
                fpwm = true;
                power_mode = MAX77620_POWER_MODE_NORMAL;
                break;

        case REGULATOR_MODE_NORMAL:
                power_mode = MAX77620_POWER_MODE_NORMAL;
                break;

        case REGULATOR_MODE_IDLE:
                power_mode = MAX77620_POWER_MODE_LPM;
                break;

        default:
                dev_err(pmic->dev, "Regulator %d mode %d is invalid\n",
                        id, mode);
                return -EINVAL;
        }

        if (rinfo->type != MAX77620_REGULATOR_TYPE_SD)
                goto skip_fpwm;

        val = (fpwm) ? MAX77620_SD_FPWM_MASK : 0;
        ret = regmap_update_bits(pmic->rmap, rinfo->cfg_addr,
                                 MAX77620_SD_FPWM_MASK, val);
        if (ret < 0) {
                dev_err(pmic->dev, "Reg 0x%02x update failed: %d\n",
                        rinfo->cfg_addr, ret);
                return ret;
        }
        rpdata->current_mode = mode;

skip_fpwm:
        ret = max77620_regulator_set_power_mode(pmic, power_mode, id);
        if (ret < 0)
                return ret;

        pmic->enable_power_mode[id] = power_mode;

        return 0;
}

static unsigned int max77620_regulator_get_mode(struct regulator_dev *rdev)
{
        struct max77620_regulator *pmic = rdev_get_drvdata(rdev);
        int id = rdev_get_id(rdev);
        struct max77620_regulator_info *rinfo = pmic->rinfo[id];
        int fpwm = 0;
        int ret;
        int pm_mode, reg_mode;
        unsigned int val;

        ret = max77620_regulator_get_power_mode(pmic, id);
        if (ret < 0)
                return 0;

        pm_mode = ret;

        if (rinfo->type == MAX77620_REGULATOR_TYPE_SD) {
                ret = regmap_read(pmic->rmap, rinfo->cfg_addr, &val);
                if (ret < 0) {
                        dev_err(pmic->dev, "Reg 0x%02x read failed: %d\n",
                                rinfo->cfg_addr, ret);
                        return ret;
                }
                fpwm = !!(val & MAX77620_SD_FPWM_MASK);
        }

        switch (pm_mode) {
        case MAX77620_POWER_MODE_NORMAL:
        case MAX77620_POWER_MODE_DISABLE:
                if (fpwm)
                        reg_mode = REGULATOR_MODE_FAST;
                else
                        reg_mode = REGULATOR_MODE_NORMAL;
                break;
        case MAX77620_POWER_MODE_LPM:
        case MAX77620_POWER_MODE_GLPM:
                reg_mode = REGULATOR_MODE_IDLE;
                break;
        default:
                return 0;
        }

        return reg_mode;
}

static int max77620_regulator_set_ramp_delay(struct regulator_dev *rdev,
                                             int ramp_delay)
{
        struct max77620_regulator *pmic = rdev_get_drvdata(rdev);
        int id = rdev_get_id(rdev);
        struct max77620_regulator_pdata *rpdata = &pmic->reg_pdata[id];

        /* Device specific ramp rate setting tells that platform has
         * different ramp rate from advertised value. In this case,
         * do not configure anything and just return success.
         */
        if (rpdata->ramp_rate_setting)
                return 0;

        return max77620_set_slew_rate(pmic, id, ramp_delay);
}

static int max77620_of_parse_cb(struct device_node *np,
                                const struct regulator_desc *desc,
                                struct regulator_config *config)
{
        struct max77620_regulator *pmic = config->driver_data;
        struct max77620_regulator_pdata *rpdata = &pmic->reg_pdata[desc->id];
        u32 pval;
        int ret;

        ret = of_property_read_u32(np, "maxim,active-fps-source", &pval);
        rpdata->active_fps_src = (!ret) ? pval : MAX77620_FPS_SRC_DEF;

        ret = of_property_read_u32(np, "maxim,active-fps-power-up-slot", &pval);
        rpdata->active_fps_pu_slot = (!ret) ? pval : -1;

        ret = of_property_read_u32(
                        np, "maxim,active-fps-power-down-slot", &pval);
        rpdata->active_fps_pd_slot = (!ret) ? pval : -1;

        ret = of_property_read_u32(np, "maxim,suspend-fps-source", &pval);
        rpdata->suspend_fps_src = (!ret) ? pval : -1;

        ret = of_property_read_u32(
                        np, "maxim,suspend-fps-power-up-slot", &pval);
        rpdata->suspend_fps_pu_slot = (!ret) ? pval : -1;

        ret = of_property_read_u32(
                        np, "maxim,suspend-fps-power-down-slot", &pval);
        rpdata->suspend_fps_pd_slot = (!ret) ? pval : -1;

        ret = of_property_read_u32(np, "maxim,power-ok-control", &pval);
        if (!ret)
                rpdata->power_ok = pval;
        else
                rpdata->power_ok = -1;

        ret = of_property_read_u32(np, "maxim,ramp-rate-setting", &pval);
        rpdata->ramp_rate_setting = (!ret) ? pval : 0;

        return max77620_init_pmic(pmic, desc->id);
}

static const struct regulator_ops max77620_regulator_ops = {
        .is_enabled = max77620_regulator_is_enabled,
        .enable = max77620_regulator_enable,
        .disable = max77620_regulator_disable,
        .list_voltage = regulator_list_voltage_linear,
        .map_voltage = regulator_map_voltage_linear,
        .get_voltage_sel = regulator_get_voltage_sel_regmap,
        .set_voltage_sel = regulator_set_voltage_sel_regmap,
        .set_mode = max77620_regulator_set_mode,
        .get_mode = max77620_regulator_get_mode,
        .set_ramp_delay = max77620_regulator_set_ramp_delay,
        .set_voltage_time_sel = regulator_set_voltage_time_sel,
        .set_active_discharge = regulator_set_active_discharge_regmap,
};

#define MAX77620_SD_CNF2_ROVS_EN_NONE   0
#define RAIL_SD(_id, _name, _sname, _volt_mask, _min_uV, _max_uV,       \
                _step_uV, _rs_add, _rs_mask)                            \
        [MAX77620_REGULATOR_ID_##_id] = {                               \
                .type = MAX77620_REGULATOR_TYPE_SD,                     \
                .volt_addr = MAX77620_REG_##_id,                        \
                .cfg_addr = MAX77620_REG_##_id##_CFG,                   \
                .fps_addr = MAX77620_REG_FPS_##_id,                     \
                .remote_sense_addr = _rs_add,                           \
                .remote_sense_mask = MAX77620_SD_CNF2_ROVS_EN_##_rs_mask, \
                .power_mode_mask = MAX77620_SD_POWER_MODE_MASK,         \
                .power_mode_shift = MAX77620_SD_POWER_MODE_SHIFT,       \
                .desc = {                                               \
                        .name = max77620_rails(_name),                  \
                        .of_match = of_match_ptr(#_name),               \
                        .regulators_node = of_match_ptr("regulators"),  \
                        .of_parse_cb = max77620_of_parse_cb,            \
                        .supply_name = _sname,                          \
                        .id = MAX77620_REGULATOR_ID_##_id,              \
                        .ops = &max77620_regulator_ops,                 \
                        .n_voltages = ((_max_uV - _min_uV) / _step_uV) + 1, \
                        .min_uV = _min_uV,                              \
                        .uV_step = _step_uV,                            \
                        .enable_time = 500,                             \
                        .vsel_mask = MAX77620_##_volt_mask##_VOLT_MASK, \
                        .vsel_reg = MAX77620_REG_##_id,                 \
                        .active_discharge_off = 0,                      \
                        .active_discharge_on = MAX77620_SD_CFG1_ADE_ENABLE, \
                        .active_discharge_mask = MAX77620_SD_CFG1_ADE_MASK, \
                        .active_discharge_reg = MAX77620_REG_##_id##_CFG, \
                        .type = REGULATOR_VOLTAGE,                      \
                        .owner = THIS_MODULE,                           \
                },                                                      \
        }

#define RAIL_LDO(_id, _name, _sname, _type, _min_uV, _max_uV, _step_uV) \
        [MAX77620_REGULATOR_ID_##_id] = {                               \
                .type = MAX77620_REGULATOR_TYPE_LDO_##_type,            \
                .volt_addr = MAX77620_REG_##_id##_CFG,                  \
                .cfg_addr = MAX77620_REG_##_id##_CFG2,                  \
                .fps_addr = MAX77620_REG_FPS_##_id,                     \
                .remote_sense_addr = 0xFF,                              \
                .power_mode_mask = MAX77620_LDO_POWER_MODE_MASK,        \
                .power_mode_shift = MAX77620_LDO_POWER_MODE_SHIFT,      \
                .desc = {                                               \
                        .name = max77620_rails(_name),                  \
                        .of_match = of_match_ptr(#_name),               \
                        .regulators_node = of_match_ptr("regulators"),  \
                        .of_parse_cb = max77620_of_parse_cb,            \
                        .supply_name = _sname,                          \
                        .id = MAX77620_REGULATOR_ID_##_id,              \
                        .ops = &max77620_regulator_ops,                 \
                        .n_voltages = ((_max_uV - _min_uV) / _step_uV) + 1, \
                        .min_uV = _min_uV,                              \
                        .uV_step = _step_uV,                            \
                        .enable_time = 500,                             \
                        .vsel_mask = MAX77620_LDO_VOLT_MASK,            \
                        .vsel_reg = MAX77620_REG_##_id##_CFG,           \
                        .active_discharge_off = 0,                      \
                        .active_discharge_on = MAX77620_LDO_CFG2_ADE_ENABLE, \
                        .active_discharge_mask = MAX77620_LDO_CFG2_ADE_MASK, \
                        .active_discharge_reg = MAX77620_REG_##_id##_CFG2, \
                        .type = REGULATOR_VOLTAGE,                      \
                        .owner = THIS_MODULE,                           \
                },                                                      \
        }

static struct max77620_regulator_info max77620_regs_info[MAX77620_NUM_REGS] = {
        RAIL_SD(SD0, sd0, "in-sd0", SD0, 600000, 1400000, 12500, 0x22, SD0),
        RAIL_SD(SD1, sd1, "in-sd1", SD1, 600000, 1550000, 12500, 0x22, SD1),
        RAIL_SD(SD2, sd2, "in-sd2", SDX, 600000, 3787500, 12500, 0xFF, NONE),
        RAIL_SD(SD3, sd3, "in-sd3", SDX, 600000, 3787500, 12500, 0xFF, NONE),

        RAIL_LDO(LDO0, ldo0, "in-ldo0-1", N, 800000, 2375000, 25000),
        RAIL_LDO(LDO1, ldo1, "in-ldo0-1", N, 800000, 2375000, 25000),
        RAIL_LDO(LDO2, ldo2, "in-ldo2",   P, 800000, 3950000, 50000),
        RAIL_LDO(LDO3, ldo3, "in-ldo3-5", P, 800000, 3950000, 50000),
        RAIL_LDO(LDO4, ldo4, "in-ldo4-6", P, 800000, 1587500, 12500),
        RAIL_LDO(LDO5, ldo5, "in-ldo3-5", P, 800000, 3950000, 50000),
        RAIL_LDO(LDO6, ldo6, "in-ldo4-6", P, 800000, 3950000, 50000),
        RAIL_LDO(LDO7, ldo7, "in-ldo7-8", N, 800000, 3950000, 50000),
        RAIL_LDO(LDO8, ldo8, "in-ldo7-8", N, 800000, 3950000, 50000),
};

static struct max77620_regulator_info max20024_regs_info[MAX77620_NUM_REGS] = {
        RAIL_SD(SD0, sd0, "in-sd0", SD0, 800000, 1587500, 12500, 0x22, SD0),
        RAIL_SD(SD1, sd1, "in-sd1", SD1, 600000, 3387500, 12500, 0x22, SD1),
        RAIL_SD(SD2, sd2, "in-sd2", SDX, 600000, 3787500, 12500, 0xFF, NONE),
        RAIL_SD(SD3, sd3, "in-sd3", SDX, 600000, 3787500, 12500, 0xFF, NONE),
        RAIL_SD(SD4, sd4, "in-sd4", SDX, 600000, 3787500, 12500, 0xFF, NONE),

        RAIL_LDO(LDO0, ldo0, "in-ldo0-1", N, 800000, 2375000, 25000),
        RAIL_LDO(LDO1, ldo1, "in-ldo0-1", N, 800000, 2375000, 25000),
        RAIL_LDO(LDO2, ldo2, "in-ldo2",   P, 800000, 3950000, 50000),
        RAIL_LDO(LDO3, ldo3, "in-ldo3-5", P, 800000, 3950000, 50000),
        RAIL_LDO(LDO4, ldo4, "in-ldo4-6", P, 800000, 1587500, 12500),
        RAIL_LDO(LDO5, ldo5, "in-ldo3-5", P, 800000, 3950000, 50000),
        RAIL_LDO(LDO6, ldo6, "in-ldo4-6", P, 800000, 3950000, 50000),
        RAIL_LDO(LDO7, ldo7, "in-ldo7-8", N, 800000, 3950000, 50000),
        RAIL_LDO(LDO8, ldo8, "in-ldo7-8", N, 800000, 3950000, 50000),
};

static struct max77620_regulator_info max77663_regs_info[MAX77620_NUM_REGS] = {
        RAIL_SD(SD0, sd0, "in-sd0", SD0, 600000, 3387500, 12500, 0xFF, NONE),
        RAIL_SD(SD1, sd1, "in-sd1", SD1, 800000, 1587500, 12500, 0xFF, NONE),
        RAIL_SD(SD2, sd2, "in-sd2", SDX, 600000, 3787500, 12500, 0xFF, NONE),
        RAIL_SD(SD3, sd3, "in-sd3", SDX, 600000, 3787500, 12500, 0xFF, NONE),
        RAIL_SD(SD4, sd4, "in-sd4", SDX, 600000, 3787500, 12500, 0xFF, NONE),

        RAIL_LDO(LDO0, ldo0, "in-ldo0-1", N, 800000, 2375000, 25000),
        RAIL_LDO(LDO1, ldo1, "in-ldo0-1", N, 800000, 2375000, 25000),
        RAIL_LDO(LDO2, ldo2, "in-ldo2",   P, 800000, 3950000, 50000),
        RAIL_LDO(LDO3, ldo3, "in-ldo3-5", P, 800000, 3950000, 50000),
        RAIL_LDO(LDO4, ldo4, "in-ldo4-6", P, 800000, 1587500, 12500),
        RAIL_LDO(LDO5, ldo5, "in-ldo3-5", P, 800000, 3950000, 50000),
        RAIL_LDO(LDO6, ldo6, "in-ldo4-6", P, 800000, 3950000, 50000),
        RAIL_LDO(LDO7, ldo7, "in-ldo7-8", N, 800000, 3950000, 50000),
        RAIL_LDO(LDO8, ldo8, "in-ldo7-8", N, 800000, 3950000, 50000),
};

static int max77620_regulator_probe(struct platform_device *pdev)
{
        struct max77620_chip *max77620_chip = dev_get_drvdata(pdev->dev.parent);
        struct max77620_regulator_info *rinfo;
        struct device *dev = &pdev->dev;
        struct regulator_config config = { };
        struct max77620_regulator *pmic;
        int ret = 0;
        int id;

        pmic = devm_kzalloc(dev, sizeof(*pmic), GFP_KERNEL);
        if (!pmic)
                return -ENOMEM;

        platform_set_drvdata(pdev, pmic);
        pmic->dev = dev;
        pmic->rmap = max77620_chip->rmap;
        if (!dev->of_node)
                dev->of_node = pdev->dev.parent->of_node;

        switch (max77620_chip->chip_id) {
        case MAX77620:
                rinfo = max77620_regs_info;
                break;
        case MAX20024:
                rinfo = max20024_regs_info;
                break;
        case MAX77663:
                rinfo = max77663_regs_info;
                break;
        default:
                return -EINVAL;
        }

        config.regmap = pmic->rmap;
        config.dev = dev;
        config.driver_data = pmic;

        /*
         * Set of_node_reuse flag to prevent driver core from attempting to
         * claim any pinmux resources already claimed by the parent device.
         * Otherwise PMIC driver will fail to re-probe.
         */
        device_set_of_node_from_dev(&pdev->dev, pdev->dev.parent);

        for (id = 0; id < MAX77620_NUM_REGS; id++) {
                struct regulator_dev *rdev;
                struct regulator_desc *rdesc;

                if ((max77620_chip->chip_id == MAX77620) &&
                    (id == MAX77620_REGULATOR_ID_SD4))
                        continue;

                rdesc = &rinfo[id].desc;
                pmic->rinfo[id] = &rinfo[id];
                pmic->enable_power_mode[id] = MAX77620_POWER_MODE_NORMAL;
                pmic->reg_pdata[id].active_fps_src = -1;
                pmic->reg_pdata[id].active_fps_pd_slot = -1;
                pmic->reg_pdata[id].active_fps_pu_slot = -1;
                pmic->reg_pdata[id].suspend_fps_src = -1;
                pmic->reg_pdata[id].suspend_fps_pd_slot = -1;
                pmic->reg_pdata[id].suspend_fps_pu_slot = -1;
                pmic->reg_pdata[id].power_ok = -1;
                pmic->reg_pdata[id].ramp_rate_setting = -1;

                ret = max77620_read_slew_rate(pmic, id);
                if (ret < 0)
                        return ret;

                rdev = devm_regulator_register(dev, rdesc, &config);
                if (IS_ERR(rdev))
                        return dev_err_probe(dev, PTR_ERR(rdev),
                                             "Regulator registration %s failed\n",
                                             rdesc->name);
        }

        return 0;
}

#ifdef CONFIG_PM_SLEEP
static int max77620_regulator_suspend(struct device *dev)
{
        struct max77620_regulator *pmic = dev_get_drvdata(dev);
        struct max77620_regulator_pdata *reg_pdata;
        int id;

        for (id = 0; id < MAX77620_NUM_REGS; id++) {
                reg_pdata = &pmic->reg_pdata[id];

                max77620_regulator_set_fps_slots(pmic, id, true);
                if (reg_pdata->suspend_fps_src < 0)
                        continue;

                max77620_regulator_set_fps_src(pmic, reg_pdata->suspend_fps_src,
                                               id);
        }

        return 0;
}

static int max77620_regulator_resume(struct device *dev)
{
        struct max77620_regulator *pmic = dev_get_drvdata(dev);
        struct max77620_regulator_pdata *reg_pdata;
        int id;

        for (id = 0; id < MAX77620_NUM_REGS; id++) {
                reg_pdata = &pmic->reg_pdata[id];

                max77620_config_power_ok(pmic, id);

                max77620_regulator_set_fps_slots(pmic, id, false);
                if (reg_pdata->active_fps_src < 0)
                        continue;
                max77620_regulator_set_fps_src(pmic, reg_pdata->active_fps_src,
                                               id);
        }

        return 0;
}
#endif

static const struct dev_pm_ops max77620_regulator_pm_ops = {
        SET_SYSTEM_SLEEP_PM_OPS(max77620_regulator_suspend,
                                max77620_regulator_resume)
};

static const struct platform_device_id max77620_regulator_devtype[] = {
        { .name = "max77620-pmic", },
        { .name = "max20024-pmic", },
        { .name = "max77663-pmic", },
        {},
};
MODULE_DEVICE_TABLE(platform, max77620_regulator_devtype);

static struct platform_driver max77620_regulator_driver = {
        .probe = max77620_regulator_probe,
        .id_table = max77620_regulator_devtype,
        .driver = {
                .name = "max77620-pmic",
                .probe_type = PROBE_PREFER_ASYNCHRONOUS,
                .pm = &max77620_regulator_pm_ops,
        },
};

module_platform_driver(max77620_regulator_driver);

MODULE_DESCRIPTION("MAX77620/MAX20024 regulator driver");
MODULE_AUTHOR("Mallikarjun Kasoju <mkasoju@nvidia.com>");
MODULE_AUTHOR("Laxman Dewangan <ldewangan@nvidia.com>");
MODULE_LICENSE("GPL v2");