// SPDX-License-Identifier: GPL-2.0-only
//
// Copyright 2014 Embest Technology Co. Ltd. Inc.
// bd71815-regulator.c ROHM BD71815 regulator driver
//
// Author: Tony Luo <luofc@embedinfo.com>
//
// Partially rewritten at 2021 by
// Matti Vaittinen <matti.vaitinen@fi.rohmeurope.com>

#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/init.h>
#include <linux/err.h>
#include <linux/platform_device.h>
#include <linux/of.h>
#include <linux/gpio/consumer.h>
#include <linux/regulator/driver.h>
#include <linux/delay.h>
#include <linux/slab.h>
#include <linux/mfd/rohm-generic.h>
#include <linux/mfd/rohm-bd71815.h>
#include <linux/regulator/of_regulator.h>

struct bd71815_regulator {
        struct regulator_desc desc;
        const struct rohm_dvs_config *dvs;
};

static const int bd7181x_wled_currents[] = {
        10, 20, 30, 50, 70, 100, 200, 300, 500, 700, 1000, 2000, 3000, 4000,
        5000, 6000, 7000, 8000, 9000, 10000, 11000, 12000, 13000, 14000, 15000,
        16000, 17000, 18000, 19000, 20000, 21000, 22000, 23000, 24000, 25000,
};

static const struct rohm_dvs_config buck1_dvs = {
        .level_map              = ROHM_DVS_LEVEL_RUN | ROHM_DVS_LEVEL_SNVS |
                                  ROHM_DVS_LEVEL_SUSPEND | ROHM_DVS_LEVEL_LPSR,
        .run_reg                = BD71815_REG_BUCK1_VOLT_H,
        .run_mask               = BD71815_VOLT_MASK,
        .run_on_mask            = BD71815_BUCK_RUN_ON,
        .snvs_on_mask           = BD71815_BUCK_SNVS_ON,
        .suspend_reg            = BD71815_REG_BUCK1_VOLT_L,
        .suspend_mask           = BD71815_VOLT_MASK,
        .suspend_on_mask        = BD71815_BUCK_SUSP_ON,
        .lpsr_reg               = BD71815_REG_BUCK1_VOLT_L,
        .lpsr_mask              = BD71815_VOLT_MASK,
        .lpsr_on_mask           = BD71815_BUCK_LPSR_ON,
};

static const struct rohm_dvs_config buck2_dvs = {
        .level_map              = ROHM_DVS_LEVEL_RUN | ROHM_DVS_LEVEL_SNVS |
                                  ROHM_DVS_LEVEL_SUSPEND | ROHM_DVS_LEVEL_LPSR,
        .run_reg                = BD71815_REG_BUCK2_VOLT_H,
        .run_mask               = BD71815_VOLT_MASK,
        .run_on_mask            = BD71815_BUCK_RUN_ON,
        .snvs_on_mask           = BD71815_BUCK_SNVS_ON,
        .suspend_reg            = BD71815_REG_BUCK2_VOLT_L,
        .suspend_mask           = BD71815_VOLT_MASK,
        .suspend_on_mask        = BD71815_BUCK_SUSP_ON,
        .lpsr_reg               = BD71815_REG_BUCK2_VOLT_L,
        .lpsr_mask              = BD71815_VOLT_MASK,
        .lpsr_on_mask           = BD71815_BUCK_LPSR_ON,
};

static const struct rohm_dvs_config buck3_dvs = {
        .level_map              = ROHM_DVS_LEVEL_RUN | ROHM_DVS_LEVEL_SNVS |
                                  ROHM_DVS_LEVEL_SUSPEND | ROHM_DVS_LEVEL_LPSR,
        .run_reg                = BD71815_REG_BUCK3_VOLT,
        .run_mask               = BD71815_VOLT_MASK,
        .run_on_mask            = BD71815_BUCK_RUN_ON,
        .snvs_on_mask           = BD71815_BUCK_SNVS_ON,
        .suspend_on_mask        = BD71815_BUCK_SUSP_ON,
        .lpsr_on_mask           = BD71815_BUCK_LPSR_ON,
};

static const struct rohm_dvs_config buck4_dvs = {
        .level_map              = ROHM_DVS_LEVEL_RUN | ROHM_DVS_LEVEL_SNVS |
                                  ROHM_DVS_LEVEL_SUSPEND | ROHM_DVS_LEVEL_LPSR,
        .run_reg                = BD71815_REG_BUCK4_VOLT,
        .run_mask               = BD71815_VOLT_MASK,
        .run_on_mask            = BD71815_BUCK_RUN_ON,
        .snvs_on_mask           = BD71815_BUCK_SNVS_ON,
        .suspend_on_mask        = BD71815_BUCK_SUSP_ON,
        .lpsr_on_mask           = BD71815_BUCK_LPSR_ON,
};

static const struct rohm_dvs_config ldo1_dvs = {
        .level_map              = ROHM_DVS_LEVEL_RUN | ROHM_DVS_LEVEL_SNVS |
                                  ROHM_DVS_LEVEL_SUSPEND | ROHM_DVS_LEVEL_LPSR,
        .run_reg                = BD71815_REG_LDO_MODE1,
        .run_mask               = BD71815_VOLT_MASK,
        .run_on_mask            = LDO1_RUN_ON,
        .snvs_on_mask           = LDO1_SNVS_ON,
        .suspend_on_mask        = LDO1_SUSP_ON,
        .lpsr_on_mask           = LDO1_LPSR_ON,
};

static const struct rohm_dvs_config ldo2_dvs = {
        .level_map              = ROHM_DVS_LEVEL_RUN | ROHM_DVS_LEVEL_SNVS |
                                  ROHM_DVS_LEVEL_SUSPEND | ROHM_DVS_LEVEL_LPSR,
        .run_reg                = BD71815_REG_LDO_MODE2,
        .run_mask               = BD71815_VOLT_MASK,
        .run_on_mask            = LDO2_RUN_ON,
        .snvs_on_mask           = LDO2_SNVS_ON,
        .suspend_on_mask        = LDO2_SUSP_ON,
        .lpsr_on_mask           = LDO2_LPSR_ON,
};

static const struct rohm_dvs_config ldo3_dvs = {
        .level_map              = ROHM_DVS_LEVEL_RUN | ROHM_DVS_LEVEL_SNVS |
                                  ROHM_DVS_LEVEL_SUSPEND | ROHM_DVS_LEVEL_LPSR,
        .run_reg                = BD71815_REG_LDO_MODE2,
        .run_mask               = BD71815_VOLT_MASK,
        .run_on_mask            = LDO3_RUN_ON,
        .snvs_on_mask           = LDO3_SNVS_ON,
        .suspend_on_mask        = LDO3_SUSP_ON,
        .lpsr_on_mask           = LDO3_LPSR_ON,
};

static const struct rohm_dvs_config ldo4_dvs = {
        .level_map              = ROHM_DVS_LEVEL_RUN | ROHM_DVS_LEVEL_SNVS |
                                  ROHM_DVS_LEVEL_SUSPEND | ROHM_DVS_LEVEL_LPSR,
        .run_reg                = BD71815_REG_LDO_MODE3,
        .run_mask               = BD71815_VOLT_MASK,
        .run_on_mask            = LDO4_RUN_ON,
        .snvs_on_mask           = LDO4_SNVS_ON,
        .suspend_on_mask        = LDO4_SUSP_ON,
        .lpsr_on_mask           = LDO4_LPSR_ON,
};

static const struct rohm_dvs_config ldo5_dvs = {
        .level_map              = ROHM_DVS_LEVEL_RUN | ROHM_DVS_LEVEL_SNVS |
                                  ROHM_DVS_LEVEL_SUSPEND | ROHM_DVS_LEVEL_LPSR,
        .run_reg                = BD71815_REG_LDO_MODE3,
        .run_mask               = BD71815_VOLT_MASK,
        .run_on_mask            = LDO5_RUN_ON,
        .snvs_on_mask           = LDO5_SNVS_ON,
        .suspend_on_mask        = LDO5_SUSP_ON,
        .lpsr_on_mask           = LDO5_LPSR_ON,
};

static const struct rohm_dvs_config dvref_dvs = {
        .level_map              = ROHM_DVS_LEVEL_RUN | ROHM_DVS_LEVEL_SNVS |
                                  ROHM_DVS_LEVEL_SUSPEND | ROHM_DVS_LEVEL_LPSR,
        .run_on_mask            = DVREF_RUN_ON,
        .snvs_on_mask           = DVREF_SNVS_ON,
        .suspend_on_mask        = DVREF_SUSP_ON,
        .lpsr_on_mask           = DVREF_LPSR_ON,
};

static const struct rohm_dvs_config ldolpsr_dvs = {
        .level_map              = ROHM_DVS_LEVEL_RUN | ROHM_DVS_LEVEL_SNVS |
                                  ROHM_DVS_LEVEL_SUSPEND | ROHM_DVS_LEVEL_LPSR,
        .run_on_mask            = DVREF_RUN_ON,
        .snvs_on_mask           = DVREF_SNVS_ON,
        .suspend_on_mask        = DVREF_SUSP_ON,
        .lpsr_on_mask           = DVREF_LPSR_ON,
};

static const struct rohm_dvs_config buck5_dvs = {
        .level_map              = ROHM_DVS_LEVEL_RUN | ROHM_DVS_LEVEL_SNVS |
                                  ROHM_DVS_LEVEL_SUSPEND | ROHM_DVS_LEVEL_LPSR,
        .run_reg                = BD71815_REG_BUCK5_VOLT,
        .run_mask               = BD71815_VOLT_MASK,
        .run_on_mask            = BD71815_BUCK_RUN_ON,
        .snvs_on_mask           = BD71815_BUCK_SNVS_ON,
        .suspend_on_mask        = BD71815_BUCK_SUSP_ON,
        .lpsr_on_mask           = BD71815_BUCK_LPSR_ON,
};

static int set_hw_dvs_levels(struct device_node *np,
                             const struct regulator_desc *desc,
                             struct regulator_config *cfg)
{
        const struct bd71815_regulator *data;

        data = container_of_const(desc, struct bd71815_regulator, desc);
        return rohm_regulator_set_dvs_levels(data->dvs, np, desc, cfg->regmap);
}

/*
 * Bucks 1 and 2 have two voltage selection registers where selected
 * voltage can be set. Which of the registers is used can be either controlled
 * by a control bit in register - or by HW state. If HW state specific voltages
 * are given - then we assume HW state based control should be used.
 *
 * If volatge value is updated to currently selected register - then output
 * voltage is immediately changed no matter what is set as ramp rate. Thus we
 * default changing voltage by writing new value to inactive register and
 * then updating the 'register selection' bit. This naturally only works when
 * HW state machine is not used to select the voltage.
 */
static int buck12_set_hw_dvs_levels(struct device_node *np,
                                    const struct regulator_desc *desc,
                                    struct regulator_config *cfg)
{
        const struct bd71815_regulator *data;
        int ret = 0, val;

        data = container_of_const(desc, struct bd71815_regulator, desc);

        if (of_property_present(np, "rohm,dvs-run-voltage") ||
            of_property_present(np, "rohm,dvs-suspend-voltage") ||
            of_property_present(np, "rohm,dvs-lpsr-voltage") ||
            of_property_present(np, "rohm,dvs-snvs-voltage")) {
                ret = regmap_read(cfg->regmap, desc->vsel_reg, &val);
                if (ret)
                        return ret;

                if (!(BD71815_BUCK_STBY_DVS & val) &&
                    !(BD71815_BUCK_DVSSEL & val)) {
                        int val2;

                        /*
                         * We are currently using voltage from _L.
                         * We'd better copy it to _H and switch to it to
                         * avoid shutting us down if LPSR or SUSPEND is set to
                         * disabled. _L value is at reg _H + 1
                         */
                        ret = regmap_read(cfg->regmap, desc->vsel_reg + 1,
                                          &val2);
                        if (ret)
                                return ret;

                        ret = regmap_update_bits(cfg->regmap, desc->vsel_reg,
                                                 BD71815_VOLT_MASK |
                                                 BD71815_BUCK_DVSSEL,
                                                 val2 | BD71815_BUCK_DVSSEL);
                        if (ret)
                                return ret;
                }
                ret = rohm_regulator_set_dvs_levels(data->dvs, np, desc,
                                                    cfg->regmap);
                if (ret)
                        return ret;
                /*
                 * DVS levels were given => use HW-state machine for voltage
                 * controls. NOTE: AFAIK, This means that if voltage is changed
                 * by SW the ramp-rate is not respected. Should we disable
                 * SW voltage control when the HW state machine is used?
                 */
                ret = regmap_update_bits(cfg->regmap, desc->vsel_reg,
                                         BD71815_BUCK_STBY_DVS,
                                         BD71815_BUCK_STBY_DVS);
        }

        return ret;
}

/*
 * BUCK1/2
 * BUCK1RAMPRATE[1:0] BUCK1 DVS ramp rate setting
 * 00: 10.00mV/usec     10mV 1uS
 * 01: 5.00mV/usec      10mV 2uS
 * 10: 2.50mV/usec      10mV 4uS
 * 11: 1.25mV/usec      10mV 8uS
 */
static const unsigned int bd7181x_ramp_table[] = { 10000, 5000, 2500, 1250 };

static int bd7181x_led_set_current_limit(struct regulator_dev *rdev,
                                        int min_uA, int max_uA)
{
        int ret;
        int onstatus;

        onstatus = regulator_is_enabled_regmap(rdev);

        ret = regulator_set_current_limit_regmap(rdev, min_uA, max_uA);
        if (!ret) {
                int newstatus;

                newstatus = regulator_is_enabled_regmap(rdev);
                if (onstatus != newstatus) {
                        /*
                         * HW FIX: spurious led status change detected. Toggle
                         * state as a workaround
                         */
                        if (onstatus)
                                ret = regulator_enable_regmap(rdev);
                        else
                                ret = regulator_disable_regmap(rdev);

                        if (ret)
                                dev_err(rdev_get_dev(rdev),
                                        "failed to revert the LED state (%d)\n",
                                        ret);
                }
        }

        return ret;
}

static int bd7181x_buck12_get_voltage_sel(struct regulator_dev *rdev)
{
        int rid = rdev_get_id(rdev);
        int ret, regh, regl, val;

        regh = BD71815_REG_BUCK1_VOLT_H + rid * 0x2;
        regl = BD71815_REG_BUCK1_VOLT_L + rid * 0x2;

        ret = regmap_read(rdev->regmap, regh, &val);
        if (ret)
                return ret;

        /*
         * If we use HW state machine based voltage reg selection - then we
         * return BD71815_REG_BUCK1_VOLT_H which is used at RUN.
         * Else we do return the BD71815_REG_BUCK1_VOLT_H or
         * BD71815_REG_BUCK1_VOLT_L depending on which is selected to be used
         * by BD71815_BUCK_DVSSEL bit
         */
        if ((!(val & BD71815_BUCK_STBY_DVS)) && (!(val & BD71815_BUCK_DVSSEL)))
                ret = regmap_read(rdev->regmap, regl, &val);

        if (ret)
                return ret;

        return val & BD71815_VOLT_MASK;
}

/*
 * For Buck 1/2.
 */
static int bd7181x_buck12_set_voltage_sel(struct regulator_dev *rdev,
                                          unsigned int sel)
{
        int rid = rdev_get_id(rdev);
        int ret, val, reg, regh, regl;

        regh = BD71815_REG_BUCK1_VOLT_H + rid*0x2;
        regl = BD71815_REG_BUCK1_VOLT_L + rid*0x2;

        ret = regmap_read(rdev->regmap, regh, &val);
        if (ret)
                return ret;

        /*
         * If bucks 1 & 2 are controlled by state machine - then the RUN state
         * voltage is set to BD71815_REG_BUCK1_VOLT_H. Changing SUSPEND/LPSR
         * voltages at runtime is not supported by this driver.
         */
        if (((val & BD71815_BUCK_STBY_DVS))) {
                return regmap_update_bits(rdev->regmap, regh, BD71815_VOLT_MASK,
                                          sel);
        }
        /* Update new voltage to the register which is not selected now */
        if (val & BD71815_BUCK_DVSSEL)
                reg = regl;
        else
                reg = regh;

        ret = regmap_update_bits(rdev->regmap, reg, BD71815_VOLT_MASK, sel);
        if (ret)
                return ret;

        /* Select the other DVS register to be used */
        return regmap_update_bits(rdev->regmap, regh, BD71815_BUCK_DVSSEL,
                                  ~val);
}

static const struct regulator_ops bd7181x_ldo_regulator_ops = {
        .enable = regulator_enable_regmap,
        .disable = regulator_disable_regmap,
        .is_enabled = regulator_is_enabled_regmap,
        .list_voltage = regulator_list_voltage_linear,
        .set_voltage_sel = regulator_set_voltage_sel_regmap,
        .get_voltage_sel = regulator_get_voltage_sel_regmap,
};

static const struct regulator_ops bd7181x_fixed_regulator_ops = {
        .enable = regulator_enable_regmap,
        .disable = regulator_disable_regmap,
        .is_enabled = regulator_is_enabled_regmap,
        .list_voltage = regulator_list_voltage_linear,
};

static const struct regulator_ops bd7181x_buck_regulator_ops = {
        .enable = regulator_enable_regmap,
        .disable = regulator_disable_regmap,
        .is_enabled = regulator_is_enabled_regmap,
        .list_voltage = regulator_list_voltage_linear,
        .set_voltage_sel = regulator_set_voltage_sel_regmap,
        .get_voltage_sel = regulator_get_voltage_sel_regmap,
        .set_voltage_time_sel = regulator_set_voltage_time_sel,
};

static const struct regulator_ops bd7181x_buck12_regulator_ops = {
        .enable = regulator_enable_regmap,
        .disable = regulator_disable_regmap,
        .is_enabled = regulator_is_enabled_regmap,
        .list_voltage = regulator_list_voltage_linear,
        .set_voltage_sel = bd7181x_buck12_set_voltage_sel,
        .get_voltage_sel = bd7181x_buck12_get_voltage_sel,
        .set_voltage_time_sel = regulator_set_voltage_time_sel,
        .set_ramp_delay = regulator_set_ramp_delay_regmap,
};

static const struct regulator_ops bd7181x_led_regulator_ops = {
        .enable = regulator_enable_regmap,
        .disable = regulator_disable_regmap,
        .is_enabled = regulator_is_enabled_regmap,
        .set_current_limit = bd7181x_led_set_current_limit,
        .get_current_limit = regulator_get_current_limit_regmap,
};

#define BD71815_FIXED_REG(_name, _id, ereg, emsk, voltage, _dvs)        \
        [(_id)] = {                                                     \
                .desc = {                                               \
                        .name = #_name,                                 \
                        .of_match = of_match_ptr(#_name),               \
                        .regulators_node = of_match_ptr("regulators"),  \
                        .n_voltages = 1,                                \
                        .ops = &bd7181x_fixed_regulator_ops,            \
                        .type = REGULATOR_VOLTAGE,                      \
                        .id = (_id),                                    \
                        .owner = THIS_MODULE,                           \
                        .min_uV = (voltage),                            \
                        .enable_reg = (ereg),                           \
                        .enable_mask = (emsk),                          \
                        .of_parse_cb = set_hw_dvs_levels,               \
                },                                                      \
                .dvs = (_dvs),                                          \
        }

#define BD71815_BUCK_REG(_name, _id, vsel, ereg, min, max, step, _dvs)  \
        [(_id)] = {                                                     \
                .desc = {                                               \
                        .name = #_name,                                 \
                        .of_match = of_match_ptr(#_name),               \
                        .regulators_node = of_match_ptr("regulators"),  \
                        .n_voltages = ((max) - (min)) / (step) + 1,     \
                        .ops = &bd7181x_buck_regulator_ops,             \
                        .type = REGULATOR_VOLTAGE,                      \
                        .id = (_id),                                    \
                        .owner = THIS_MODULE,                           \
                        .min_uV = (min),                                \
                        .uV_step = (step),                              \
                        .vsel_reg = (vsel),                             \
                        .vsel_mask = BD71815_VOLT_MASK,                 \
                        .enable_reg = (ereg),                           \
                        .enable_mask = BD71815_BUCK_RUN_ON,             \
                        .of_parse_cb = set_hw_dvs_levels,               \
                },                                                      \
                .dvs = (_dvs),                                          \
        }

#define BD71815_BUCK12_REG(_name, _id, vsel, ereg, min, max, step,      \
                           _dvs)                                        \
        [(_id)] = {                                                     \
                .desc = {                                               \
                        .name = #_name,                                 \
                        .of_match = of_match_ptr(#_name),               \
                        .regulators_node = of_match_ptr("regulators"),  \
                        .n_voltages = ((max) - (min)) / (step) + 1,     \
                        .ops = &bd7181x_buck12_regulator_ops,           \
                        .type = REGULATOR_VOLTAGE,                      \
                        .id = (_id),                                    \
                        .owner = THIS_MODULE,                           \
                        .min_uV = (min),                                \
                        .uV_step = (step),                              \
                        .vsel_reg = (vsel),                             \
                        .vsel_mask = BD71815_VOLT_MASK,                 \
                        .enable_reg = (ereg),                           \
                        .enable_mask = BD71815_BUCK_RUN_ON,             \
                        .ramp_reg = (ereg),                             \
                        .ramp_mask = BD71815_BUCK_RAMPRATE_MASK,        \
                        .ramp_delay_table = bd7181x_ramp_table,         \
                        .n_ramp_values = ARRAY_SIZE(bd7181x_ramp_table),\
                        .of_parse_cb = buck12_set_hw_dvs_levels,        \
                },                                                      \
                .dvs = (_dvs),                                          \
        }

#define BD71815_LED_REG(_name, _id, csel, mask, ereg, emsk, currents)   \
        [(_id)] = {                                                     \
                .desc = {                                               \
                        .name = #_name,                                 \
                        .of_match = of_match_ptr(#_name),               \
                        .regulators_node = of_match_ptr("regulators"),  \
                        .n_current_limits = ARRAY_SIZE(currents),       \
                        .ops = &bd7181x_led_regulator_ops,              \
                        .type = REGULATOR_CURRENT,                      \
                        .id = (_id),                                    \
                        .owner = THIS_MODULE,                           \
                        .curr_table = currents,                         \
                        .csel_reg = (csel),                             \
                        .csel_mask = (mask),                            \
                        .enable_reg = (ereg),                           \
                        .enable_mask = (emsk),                          \
                },                                                      \
        }

#define BD71815_LDO_REG(_name, _id, vsel, ereg, emsk, min, max, step,   \
                        _dvs)                                           \
        [(_id)] = {                                                     \
                .desc = {                                               \
                        .name = #_name,                                 \
                        .of_match = of_match_ptr(#_name),               \
                        .regulators_node = of_match_ptr("regulators"),  \
                        .n_voltages = ((max) - (min)) / (step) + 1,     \
                        .ops = &bd7181x_ldo_regulator_ops,              \
                        .type = REGULATOR_VOLTAGE,                      \
                        .id = (_id),                                    \
                        .owner = THIS_MODULE,                           \
                        .min_uV = (min),                                \
                        .uV_step = (step),                              \
                        .vsel_reg = (vsel),                             \
                        .vsel_mask = BD71815_VOLT_MASK,                 \
                        .enable_reg = (ereg),                           \
                        .enable_mask = (emsk),                          \
                        .of_parse_cb = set_hw_dvs_levels,               \
                },                                                      \
                .dvs = (_dvs),                                          \
        }

static const struct bd71815_regulator bd71815_regulators[] = {
        BD71815_BUCK12_REG(buck1, BD71815_BUCK1, BD71815_REG_BUCK1_VOLT_H,
                           BD71815_REG_BUCK1_MODE, 800000, 2000000, 25000,
                           &buck1_dvs),
        BD71815_BUCK12_REG(buck2, BD71815_BUCK2, BD71815_REG_BUCK2_VOLT_H,
                           BD71815_REG_BUCK2_MODE, 800000, 2000000, 25000,
                           &buck2_dvs),
        BD71815_BUCK_REG(buck3, BD71815_BUCK3, BD71815_REG_BUCK3_VOLT,
                         BD71815_REG_BUCK3_MODE,  1200000, 2700000, 50000,
                         &buck3_dvs),
        BD71815_BUCK_REG(buck4, BD71815_BUCK4, BD71815_REG_BUCK4_VOLT,
                         BD71815_REG_BUCK4_MODE,  1100000, 1850000, 25000,
                         &buck4_dvs),
        BD71815_BUCK_REG(buck5, BD71815_BUCK5, BD71815_REG_BUCK5_VOLT,
                         BD71815_REG_BUCK5_MODE,  1800000, 3300000, 50000,
                         &buck5_dvs),
        BD71815_LDO_REG(ldo1, BD71815_LDO1, BD71815_REG_LDO1_VOLT,
                        BD71815_REG_LDO_MODE1, LDO1_RUN_ON, 800000, 3300000,
                        50000, &ldo1_dvs),
        BD71815_LDO_REG(ldo2, BD71815_LDO2, BD71815_REG_LDO2_VOLT,
                        BD71815_REG_LDO_MODE2, LDO2_RUN_ON, 800000, 3300000,
                        50000, &ldo2_dvs),
        /*
         * Let's default LDO3 to be enabled by SW. We can override ops if DT
         * says LDO3 should be enabled by HW when DCIN is connected.
         */
        BD71815_LDO_REG(ldo3, BD71815_LDO3, BD71815_REG_LDO3_VOLT,
                        BD71815_REG_LDO_MODE2, LDO3_RUN_ON, 800000, 3300000,
                        50000, &ldo3_dvs),
        BD71815_LDO_REG(ldo4, BD71815_LDO4, BD71815_REG_LDO4_VOLT,
                        BD71815_REG_LDO_MODE3, LDO4_RUN_ON, 800000, 3300000,
                        50000, &ldo4_dvs),
        BD71815_LDO_REG(ldo5, BD71815_LDO5, BD71815_REG_LDO5_VOLT_H,
                        BD71815_REG_LDO_MODE3, LDO5_RUN_ON, 800000, 3300000,
                        50000, &ldo5_dvs),
        BD71815_FIXED_REG(ldodvref, BD71815_LDODVREF, BD71815_REG_LDO_MODE4,
                          DVREF_RUN_ON, 3000000, &dvref_dvs),
        BD71815_FIXED_REG(ldolpsr, BD71815_LDOLPSR, BD71815_REG_LDO_MODE4,
                          LDO_LPSR_RUN_ON, 1800000, &ldolpsr_dvs),
        BD71815_LED_REG(wled, BD71815_WLED, BD71815_REG_LED_DIMM, LED_DIMM_MASK,
                        BD71815_REG_LED_CTRL, LED_RUN_ON,
                        bd7181x_wled_currents),
};

static int bd7181x_probe(struct platform_device *pdev)
{
        struct regulator_config config = {};
        int i, ret;
        struct gpio_desc *ldo4_en;
        struct regmap *regmap;

        regmap = dev_get_regmap(pdev->dev.parent, NULL);
        if (!regmap) {
                dev_err(&pdev->dev, "No parent regmap\n");
                return -ENODEV;
        }

        ldo4_en = devm_fwnode_gpiod_get_optional(&pdev->dev,
                                                 dev_fwnode(pdev->dev.parent),
                                                 "rohm,vsel", GPIOD_ASIS,
                                                 "ldo4-en");
        if (IS_ERR(ldo4_en))
                return PTR_ERR(ldo4_en);

        /* Disable to go to ship-mode */
        ret = regmap_update_bits(regmap, BD71815_REG_PWRCTRL, RESTARTEN, 0);
        if (ret)
                return ret;

        config.dev = pdev->dev.parent;
        config.regmap = regmap;

        for (i = 0; i < BD71815_REGULATOR_CNT; i++) {
                const struct regulator_desc *desc;
                struct regulator_dev *rdev;

                desc = &bd71815_regulators[i].desc;

                if (i == BD71815_LDO4)
                        config.ena_gpiod = ldo4_en;
                else
                        config.ena_gpiod = NULL;

                rdev = devm_regulator_register(&pdev->dev, desc, &config);
                if (IS_ERR(rdev))
                        return dev_err_probe(&pdev->dev, PTR_ERR(rdev),
                                             "failed to register %s regulator\n",
                                             desc->name);
        }
        return 0;
}

static const struct platform_device_id bd7181x_pmic_id[] = {
        { "bd71815-pmic", ROHM_CHIP_TYPE_BD71815 },
        { },
};
MODULE_DEVICE_TABLE(platform, bd7181x_pmic_id);

static struct platform_driver bd7181x_regulator = {
        .driver = {
                .name = "bd7181x-pmic",
                .probe_type = PROBE_PREFER_ASYNCHRONOUS,
        },
        .probe = bd7181x_probe,
        .id_table = bd7181x_pmic_id,
};
module_platform_driver(bd7181x_regulator);

MODULE_AUTHOR("Tony Luo <luofc@embedinfo.com>");
MODULE_DESCRIPTION("BD71815 voltage regulator driver");
MODULE_LICENSE("GPL v2");
MODULE_ALIAS("platform:bd7181x-pmic");