// SPDX-License-Identifier: GPL-2.0-only
/*
 * Arizona core driver
 *
 * Copyright 2012 Wolfson Microelectronics plc
 *
 * Author: Mark Brown <broonie@opensource.wolfsonmicro.com>
 */

#include <linux/clk.h>
#include <linux/delay.h>
#include <linux/err.h>
#include <linux/gpio/consumer.h>
#include <linux/interrupt.h>
#include <linux/mfd/core.h>
#include <linux/module.h>
#include <linux/of.h>
#include <linux/pm_runtime.h>
#include <linux/regmap.h>
#include <linux/regulator/consumer.h>
#include <linux/regulator/machine.h>
#include <linux/slab.h>
#include <linux/ktime.h>
#include <linux/platform_device.h>

#include <linux/mfd/arizona/core.h>
#include <linux/mfd/arizona/registers.h>

#include "arizona.h"

static const char * const wm5102_core_supplies[] = {
        "AVDD",
        "DBVDD1",
};

int arizona_clk32k_enable(struct arizona *arizona)
{
        int ret = 0;

        mutex_lock(&arizona->clk_lock);

        arizona->clk32k_ref++;

        if (arizona->clk32k_ref == 1) {
                switch (arizona->pdata.clk32k_src) {
                case ARIZONA_32KZ_MCLK1:
                        ret = pm_runtime_resume_and_get(arizona->dev);
                        if (ret != 0)
                                goto err_ref;
                        ret = clk_prepare_enable(arizona->mclk[ARIZONA_MCLK1]);
                        if (ret != 0) {
                                pm_runtime_put_sync(arizona->dev);
                                goto err_ref;
                        }
                        break;
                case ARIZONA_32KZ_MCLK2:
                        ret = clk_prepare_enable(arizona->mclk[ARIZONA_MCLK2]);
                        if (ret != 0)
                                goto err_ref;
                        break;
                }

                ret = regmap_update_bits(arizona->regmap, ARIZONA_CLOCK_32K_1,
                                         ARIZONA_CLK_32K_ENA,
                                         ARIZONA_CLK_32K_ENA);
        }

err_ref:
        if (ret != 0)
                arizona->clk32k_ref--;

        mutex_unlock(&arizona->clk_lock);

        return ret;
}
EXPORT_SYMBOL_GPL(arizona_clk32k_enable);

int arizona_clk32k_disable(struct arizona *arizona)
{
        mutex_lock(&arizona->clk_lock);

        WARN_ON(arizona->clk32k_ref <= 0);

        arizona->clk32k_ref--;

        if (arizona->clk32k_ref == 0) {
                regmap_update_bits(arizona->regmap, ARIZONA_CLOCK_32K_1,
                                   ARIZONA_CLK_32K_ENA, 0);

                switch (arizona->pdata.clk32k_src) {
                case ARIZONA_32KZ_MCLK1:
                        pm_runtime_put_sync(arizona->dev);
                        clk_disable_unprepare(arizona->mclk[ARIZONA_MCLK1]);
                        break;
                case ARIZONA_32KZ_MCLK2:
                        clk_disable_unprepare(arizona->mclk[ARIZONA_MCLK2]);
                        break;
                }
        }

        mutex_unlock(&arizona->clk_lock);

        return 0;
}
EXPORT_SYMBOL_GPL(arizona_clk32k_disable);

static irqreturn_t arizona_clkgen_err(int irq, void *data)
{
        struct arizona *arizona = data;

        dev_err(arizona->dev, "CLKGEN error\n");

        return IRQ_HANDLED;
}

static irqreturn_t arizona_underclocked(int irq, void *data)
{
        struct arizona *arizona = data;
        unsigned int val;
        int ret;

        ret = regmap_read(arizona->regmap, ARIZONA_INTERRUPT_RAW_STATUS_8,
                          &val);
        if (ret != 0) {
                dev_err(arizona->dev, "Failed to read underclock status: %d\n",
                        ret);
                return IRQ_NONE;
        }

        if (val & ARIZONA_AIF3_UNDERCLOCKED_STS)
                dev_err(arizona->dev, "AIF3 underclocked\n");
        if (val & ARIZONA_AIF2_UNDERCLOCKED_STS)
                dev_err(arizona->dev, "AIF2 underclocked\n");
        if (val & ARIZONA_AIF1_UNDERCLOCKED_STS)
                dev_err(arizona->dev, "AIF1 underclocked\n");
        if (val & ARIZONA_ISRC3_UNDERCLOCKED_STS)
                dev_err(arizona->dev, "ISRC3 underclocked\n");
        if (val & ARIZONA_ISRC2_UNDERCLOCKED_STS)
                dev_err(arizona->dev, "ISRC2 underclocked\n");
        if (val & ARIZONA_ISRC1_UNDERCLOCKED_STS)
                dev_err(arizona->dev, "ISRC1 underclocked\n");
        if (val & ARIZONA_FX_UNDERCLOCKED_STS)
                dev_err(arizona->dev, "FX underclocked\n");
        if (val & ARIZONA_ASRC_UNDERCLOCKED_STS)
                dev_err(arizona->dev, "ASRC underclocked\n");
        if (val & ARIZONA_DAC_UNDERCLOCKED_STS)
                dev_err(arizona->dev, "DAC underclocked\n");
        if (val & ARIZONA_ADC_UNDERCLOCKED_STS)
                dev_err(arizona->dev, "ADC underclocked\n");
        if (val & ARIZONA_MIXER_UNDERCLOCKED_STS)
                dev_err(arizona->dev, "Mixer dropped sample\n");

        return IRQ_HANDLED;
}

static irqreturn_t arizona_overclocked(int irq, void *data)
{
        struct arizona *arizona = data;
        unsigned int val[3];
        int ret;

        ret = regmap_bulk_read(arizona->regmap, ARIZONA_INTERRUPT_RAW_STATUS_6,
                               &val[0], 3);
        if (ret != 0) {
                dev_err(arizona->dev, "Failed to read overclock status: %d\n",
                        ret);
                return IRQ_NONE;
        }

        switch (arizona->type) {
        case WM8998:
        case WM1814:
                /* Some bits are shifted on WM8998,
                 * rearrange to match the standard bit layout
                 */
                val[0] = ((val[0] & 0x60e0) >> 1) |
                         ((val[0] & 0x1e00) >> 2) |
                         (val[0] & 0x000f);
                break;
        default:
                break;
        }

        if (val[0] & ARIZONA_PWM_OVERCLOCKED_STS)
                dev_err(arizona->dev, "PWM overclocked\n");
        if (val[0] & ARIZONA_FX_CORE_OVERCLOCKED_STS)
                dev_err(arizona->dev, "FX core overclocked\n");
        if (val[0] & ARIZONA_DAC_SYS_OVERCLOCKED_STS)
                dev_err(arizona->dev, "DAC SYS overclocked\n");
        if (val[0] & ARIZONA_DAC_WARP_OVERCLOCKED_STS)
                dev_err(arizona->dev, "DAC WARP overclocked\n");
        if (val[0] & ARIZONA_ADC_OVERCLOCKED_STS)
                dev_err(arizona->dev, "ADC overclocked\n");
        if (val[0] & ARIZONA_MIXER_OVERCLOCKED_STS)
                dev_err(arizona->dev, "Mixer overclocked\n");
        if (val[0] & ARIZONA_AIF3_SYNC_OVERCLOCKED_STS)
                dev_err(arizona->dev, "AIF3 overclocked\n");
        if (val[0] & ARIZONA_AIF2_SYNC_OVERCLOCKED_STS)
                dev_err(arizona->dev, "AIF2 overclocked\n");
        if (val[0] & ARIZONA_AIF1_SYNC_OVERCLOCKED_STS)
                dev_err(arizona->dev, "AIF1 overclocked\n");
        if (val[0] & ARIZONA_PAD_CTRL_OVERCLOCKED_STS)
                dev_err(arizona->dev, "Pad control overclocked\n");

        if (val[1] & ARIZONA_SLIMBUS_SUBSYS_OVERCLOCKED_STS)
                dev_err(arizona->dev, "Slimbus subsystem overclocked\n");
        if (val[1] & ARIZONA_SLIMBUS_ASYNC_OVERCLOCKED_STS)
                dev_err(arizona->dev, "Slimbus async overclocked\n");
        if (val[1] & ARIZONA_SLIMBUS_SYNC_OVERCLOCKED_STS)
                dev_err(arizona->dev, "Slimbus sync overclocked\n");
        if (val[1] & ARIZONA_ASRC_ASYNC_SYS_OVERCLOCKED_STS)
                dev_err(arizona->dev, "ASRC async system overclocked\n");
        if (val[1] & ARIZONA_ASRC_ASYNC_WARP_OVERCLOCKED_STS)
                dev_err(arizona->dev, "ASRC async WARP overclocked\n");
        if (val[1] & ARIZONA_ASRC_SYNC_SYS_OVERCLOCKED_STS)
                dev_err(arizona->dev, "ASRC sync system overclocked\n");
        if (val[1] & ARIZONA_ASRC_SYNC_WARP_OVERCLOCKED_STS)
                dev_err(arizona->dev, "ASRC sync WARP overclocked\n");
        if (val[1] & ARIZONA_ADSP2_1_OVERCLOCKED_STS)
                dev_err(arizona->dev, "DSP1 overclocked\n");
        if (val[1] & ARIZONA_ISRC3_OVERCLOCKED_STS)
                dev_err(arizona->dev, "ISRC3 overclocked\n");
        if (val[1] & ARIZONA_ISRC2_OVERCLOCKED_STS)
                dev_err(arizona->dev, "ISRC2 overclocked\n");
        if (val[1] & ARIZONA_ISRC1_OVERCLOCKED_STS)
                dev_err(arizona->dev, "ISRC1 overclocked\n");

        if (val[2] & ARIZONA_SPDIF_OVERCLOCKED_STS)
                dev_err(arizona->dev, "SPDIF overclocked\n");

        return IRQ_HANDLED;
}

#define ARIZONA_REG_POLL_DELAY_US 7500

static inline bool arizona_poll_reg_delay(ktime_t timeout)
{
        if (ktime_compare(ktime_get(), timeout) > 0)
                return false;

        usleep_range(ARIZONA_REG_POLL_DELAY_US / 2, ARIZONA_REG_POLL_DELAY_US);

        return true;
}

static int arizona_poll_reg(struct arizona *arizona,
                            int timeout_ms, unsigned int reg,
                            unsigned int mask, unsigned int target)
{
        ktime_t timeout = ktime_add_us(ktime_get(), timeout_ms * USEC_PER_MSEC);
        unsigned int val = 0;
        int ret;

        do {
                ret = regmap_read(arizona->regmap, reg, &val);

                if ((val & mask) == target)
                        return 0;
        } while (arizona_poll_reg_delay(timeout));

        if (ret) {
                dev_err(arizona->dev, "Failed polling reg 0x%x: %d\n",
                        reg, ret);
                return ret;
        }

        dev_err(arizona->dev, "Polling reg 0x%x timed out: %x\n", reg, val);
        return -ETIMEDOUT;
}

static int arizona_wait_for_boot(struct arizona *arizona)
{
        int ret;

        /*
         * We can't use an interrupt as we need to runtime resume to do so,
         * we won't race with the interrupt handler as it'll be blocked on
         * runtime resume.
         */
        ret = arizona_poll_reg(arizona, 30, ARIZONA_INTERRUPT_RAW_STATUS_5,
                               ARIZONA_BOOT_DONE_STS, ARIZONA_BOOT_DONE_STS);

        if (!ret)
                regmap_write(arizona->regmap, ARIZONA_INTERRUPT_STATUS_5,
                             ARIZONA_BOOT_DONE_STS);

        pm_runtime_mark_last_busy(arizona->dev);

        return ret;
}

static inline void arizona_enable_reset(struct arizona *arizona)
{
        if (arizona->pdata.reset)
                gpiod_set_raw_value_cansleep(arizona->pdata.reset, 0);
}

static void arizona_disable_reset(struct arizona *arizona)
{
        if (arizona->pdata.reset) {
                switch (arizona->type) {
                case WM5110:
                case WM8280:
                        /* Meet requirements for minimum reset duration */
                        usleep_range(5000, 10000);
                        break;
                default:
                        break;
                }

                gpiod_set_raw_value_cansleep(arizona->pdata.reset, 1);
                usleep_range(1000, 5000);
        }
}

struct arizona_sysclk_state {
        unsigned int fll;
        unsigned int sysclk;
};

static int arizona_enable_freerun_sysclk(struct arizona *arizona,
                                         struct arizona_sysclk_state *state)
{
        int ret, err;

        /* Cache existing FLL and SYSCLK settings */
        ret = regmap_read(arizona->regmap, ARIZONA_FLL1_CONTROL_1, &state->fll);
        if (ret) {
                dev_err(arizona->dev, "Failed to cache FLL settings: %d\n",
                        ret);
                return ret;
        }
        ret = regmap_read(arizona->regmap, ARIZONA_SYSTEM_CLOCK_1,
                          &state->sysclk);
        if (ret) {
                dev_err(arizona->dev, "Failed to cache SYSCLK settings: %d\n",
                        ret);
                return ret;
        }

        /* Start up SYSCLK using the FLL in free running mode */
        ret = regmap_write(arizona->regmap, ARIZONA_FLL1_CONTROL_1,
                        ARIZONA_FLL1_ENA | ARIZONA_FLL1_FREERUN);
        if (ret) {
                dev_err(arizona->dev,
                        "Failed to start FLL in freerunning mode: %d\n",
                        ret);
                return ret;
        }
        ret = arizona_poll_reg(arizona, 180, ARIZONA_INTERRUPT_RAW_STATUS_5,
                               ARIZONA_FLL1_CLOCK_OK_STS,
                               ARIZONA_FLL1_CLOCK_OK_STS);
        if (ret)
                goto err_fll;

        ret = regmap_write(arizona->regmap, ARIZONA_SYSTEM_CLOCK_1, 0x0144);
        if (ret) {
                dev_err(arizona->dev, "Failed to start SYSCLK: %d\n", ret);
                goto err_fll;
        }

        return 0;

err_fll:
        err = regmap_write(arizona->regmap, ARIZONA_FLL1_CONTROL_1, state->fll);
        if (err)
                dev_err(arizona->dev,
                        "Failed to re-apply old FLL settings: %d\n", err);

        return ret;
}

static int arizona_disable_freerun_sysclk(struct arizona *arizona,
                                          struct arizona_sysclk_state *state)
{
        int ret;

        ret = regmap_write(arizona->regmap, ARIZONA_SYSTEM_CLOCK_1,
                           state->sysclk);
        if (ret) {
                dev_err(arizona->dev,
                        "Failed to re-apply old SYSCLK settings: %d\n", ret);
                return ret;
        }

        ret = regmap_write(arizona->regmap, ARIZONA_FLL1_CONTROL_1, state->fll);
        if (ret) {
                dev_err(arizona->dev,
                        "Failed to re-apply old FLL settings: %d\n", ret);
                return ret;
        }

        return 0;
}

static int wm5102_apply_hardware_patch(struct arizona *arizona)
{
        struct arizona_sysclk_state state;
        int err, ret;

        ret = arizona_enable_freerun_sysclk(arizona, &state);
        if (ret)
                return ret;

        /* Start the write sequencer and wait for it to finish */
        ret = regmap_write(arizona->regmap, ARIZONA_WRITE_SEQUENCER_CTRL_0,
                           ARIZONA_WSEQ_ENA | ARIZONA_WSEQ_START | 160);
        if (ret) {
                dev_err(arizona->dev, "Failed to start write sequencer: %d\n",
                        ret);
                goto err;
        }

        ret = arizona_poll_reg(arizona, 30, ARIZONA_WRITE_SEQUENCER_CTRL_1,
                               ARIZONA_WSEQ_BUSY, 0);
        if (ret)
                regmap_write(arizona->regmap, ARIZONA_WRITE_SEQUENCER_CTRL_0,
                             ARIZONA_WSEQ_ABORT);

err:
        err = arizona_disable_freerun_sysclk(arizona, &state);

        return ret ?: err;
}

/*
 * Register patch to some of the CODECs internal write sequences
 * to ensure a clean exit from the low power sleep state.
 */
static const struct reg_sequence wm5110_sleep_patch[] = {
        { 0x337A, 0xC100 },
        { 0x337B, 0x0041 },
        { 0x3300, 0xA210 },
        { 0x3301, 0x050C },
};

static int wm5110_apply_sleep_patch(struct arizona *arizona)
{
        struct arizona_sysclk_state state;
        int err, ret;

        ret = arizona_enable_freerun_sysclk(arizona, &state);
        if (ret)
                return ret;

        ret = regmap_multi_reg_write_bypassed(arizona->regmap,
                                              wm5110_sleep_patch,
                                              ARRAY_SIZE(wm5110_sleep_patch));

        err = arizona_disable_freerun_sysclk(arizona, &state);

        return ret ?: err;
}

static int wm5102_clear_write_sequencer(struct arizona *arizona)
{
        int ret;

        ret = regmap_write(arizona->regmap, ARIZONA_WRITE_SEQUENCER_CTRL_3,
                           0x0);
        if (ret) {
                dev_err(arizona->dev,
                        "Failed to clear write sequencer state: %d\n", ret);
                return ret;
        }

        arizona_enable_reset(arizona);
        regulator_disable(arizona->dcvdd);

        msleep(20);

        ret = regulator_enable(arizona->dcvdd);
        if (ret) {
                dev_err(arizona->dev, "Failed to re-enable DCVDD: %d\n", ret);
                return ret;
        }
        arizona_disable_reset(arizona);

        return 0;
}

static int arizona_isolate_dcvdd(struct arizona *arizona)
{
        int ret;

        ret = regmap_update_bits(arizona->regmap,
                                 ARIZONA_ISOLATION_CONTROL,
                                 ARIZONA_ISOLATE_DCVDD1,
                                 ARIZONA_ISOLATE_DCVDD1);
        if (ret != 0)
                dev_err(arizona->dev, "Failed to isolate DCVDD: %d\n", ret);

        return ret;
}

static int arizona_connect_dcvdd(struct arizona *arizona)
{
        int ret;

        ret = regmap_update_bits(arizona->regmap,
                                 ARIZONA_ISOLATION_CONTROL,
                                 ARIZONA_ISOLATE_DCVDD1, 0);
        if (ret != 0)
                dev_err(arizona->dev, "Failed to connect DCVDD: %d\n", ret);

        return ret;
}

static int arizona_is_jack_det_active(struct arizona *arizona)
{
        unsigned int val;
        int ret;

        ret = regmap_read(arizona->regmap, ARIZONA_JACK_DETECT_ANALOGUE, &val);
        if (ret) {
                dev_err(arizona->dev,
                        "Failed to check jack det status: %d\n", ret);
                return ret;
        } else if (val & ARIZONA_JD1_ENA) {
                return 1;
        } else {
                return 0;
        }
}

static int arizona_runtime_resume(struct device *dev)
{
        struct arizona *arizona = dev_get_drvdata(dev);
        int ret;

        dev_dbg(arizona->dev, "Leaving AoD mode\n");

        if (arizona->has_fully_powered_off) {
                dev_dbg(arizona->dev, "Re-enabling core supplies\n");

                ret = regulator_bulk_enable(arizona->num_core_supplies,
                                            arizona->core_supplies);
                if (ret) {
                        dev_err(dev, "Failed to enable core supplies: %d\n",
                                ret);
                        return ret;
                }
        }

        ret = regulator_enable(arizona->dcvdd);
        if (ret != 0) {
                dev_err(arizona->dev, "Failed to enable DCVDD: %d\n", ret);
                if (arizona->has_fully_powered_off)
                        regulator_bulk_disable(arizona->num_core_supplies,
                                               arizona->core_supplies);
                return ret;
        }

        if (arizona->has_fully_powered_off) {
                arizona_disable_reset(arizona);
                enable_irq(arizona->irq);
                arizona->has_fully_powered_off = false;
        }

        regcache_cache_only(arizona->regmap, false);

        switch (arizona->type) {
        case WM5102:
                if (arizona->external_dcvdd) {
                        ret = arizona_connect_dcvdd(arizona);
                        if (ret != 0)
                                goto err;
                }

                ret = wm5102_patch(arizona);
                if (ret != 0) {
                        dev_err(arizona->dev, "Failed to apply patch: %d\n",
                                ret);
                        goto err;
                }

                ret = wm5102_apply_hardware_patch(arizona);
                if (ret) {
                        dev_err(arizona->dev,
                                "Failed to apply hardware patch: %d\n",
                                ret);
                        goto err;
                }
                break;
        case WM5110:
        case WM8280:
                ret = arizona_wait_for_boot(arizona);
                if (ret)
                        goto err;

                if (arizona->external_dcvdd) {
                        ret = arizona_connect_dcvdd(arizona);
                        if (ret != 0)
                                goto err;
                } else {
                        /*
                         * As this is only called for the internal regulator
                         * (where we know voltage ranges available) it is ok
                         * to request an exact range.
                         */
                        ret = regulator_set_voltage(arizona->dcvdd,
                                                    1200000, 1200000);
                        if (ret < 0) {
                                dev_err(arizona->dev,
                                        "Failed to set resume voltage: %d\n",
                                        ret);
                                goto err;
                        }
                }

                ret = wm5110_apply_sleep_patch(arizona);
                if (ret) {
                        dev_err(arizona->dev,
                                "Failed to re-apply sleep patch: %d\n",
                                ret);
                        goto err;
                }
                break;
        case WM1831:
        case CS47L24:
                ret = arizona_wait_for_boot(arizona);
                if (ret != 0)
                        goto err;
                break;
        default:
                ret = arizona_wait_for_boot(arizona);
                if (ret != 0)
                        goto err;

                if (arizona->external_dcvdd) {
                        ret = arizona_connect_dcvdd(arizona);
                        if (ret != 0)
                                goto err;
                }
                break;
        }

        ret = regcache_sync(arizona->regmap);
        if (ret != 0) {
                dev_err(arizona->dev, "Failed to restore register cache\n");
                goto err;
        }

        return 0;

err:
        regcache_cache_only(arizona->regmap, true);
        regulator_disable(arizona->dcvdd);
        return ret;
}

static int arizona_runtime_suspend(struct device *dev)
{
        struct arizona *arizona = dev_get_drvdata(dev);
        int jd_active = 0;
        int ret;

        dev_dbg(arizona->dev, "Entering AoD mode\n");

        switch (arizona->type) {
        case WM5110:
        case WM8280:
                jd_active = arizona_is_jack_det_active(arizona);
                if (jd_active < 0)
                        return jd_active;

                if (arizona->external_dcvdd) {
                        ret = arizona_isolate_dcvdd(arizona);
                        if (ret != 0)
                                return ret;
                } else {
                        /*
                         * As this is only called for the internal regulator
                         * (where we know voltage ranges available) it is ok
                         * to request an exact range.
                         */
                        ret = regulator_set_voltage(arizona->dcvdd,
                                                    1175000, 1175000);
                        if (ret < 0) {
                                dev_err(arizona->dev,
                                        "Failed to set suspend voltage: %d\n",
                                        ret);
                                return ret;
                        }
                }
                break;
        case WM5102:
                jd_active = arizona_is_jack_det_active(arizona);
                if (jd_active < 0)
                        return jd_active;

                if (arizona->external_dcvdd) {
                        ret = arizona_isolate_dcvdd(arizona);
                        if (ret != 0)
                                return ret;
                }

                if (!jd_active) {
                        ret = regmap_write(arizona->regmap,
                                           ARIZONA_WRITE_SEQUENCER_CTRL_3, 0x0);
                        if (ret) {
                                dev_err(arizona->dev,
                                        "Failed to clear write sequencer: %d\n",
                                        ret);
                                return ret;
                        }
                }
                break;
        case WM1831:
        case CS47L24:
                break;
        default:
                jd_active = arizona_is_jack_det_active(arizona);
                if (jd_active < 0)
                        return jd_active;

                if (arizona->external_dcvdd) {
                        ret = arizona_isolate_dcvdd(arizona);
                        if (ret != 0)
                                return ret;
                }
                break;
        }

        regcache_cache_only(arizona->regmap, true);
        regcache_mark_dirty(arizona->regmap);
        regulator_disable(arizona->dcvdd);

        /* Allow us to completely power down if no jack detection */
        if (!jd_active) {
                dev_dbg(arizona->dev, "Fully powering off\n");

                arizona->has_fully_powered_off = true;

                disable_irq_nosync(arizona->irq);
                arizona_enable_reset(arizona);
                regulator_bulk_disable(arizona->num_core_supplies,
                                       arizona->core_supplies);
        }

        return 0;
}

static int arizona_suspend(struct device *dev)
{
        struct arizona *arizona = dev_get_drvdata(dev);

        dev_dbg(arizona->dev, "Suspend, disabling IRQ\n");
        disable_irq(arizona->irq);

        return 0;
}

static int arizona_suspend_noirq(struct device *dev)
{
        struct arizona *arizona = dev_get_drvdata(dev);

        dev_dbg(arizona->dev, "Late suspend, reenabling IRQ\n");
        enable_irq(arizona->irq);

        return 0;
}

static int arizona_resume_noirq(struct device *dev)
{
        struct arizona *arizona = dev_get_drvdata(dev);

        dev_dbg(arizona->dev, "Early resume, disabling IRQ\n");
        disable_irq(arizona->irq);

        return 0;
}

static int arizona_resume(struct device *dev)
{
        struct arizona *arizona = dev_get_drvdata(dev);

        dev_dbg(arizona->dev, "Resume, reenabling IRQ\n");
        enable_irq(arizona->irq);

        return 0;
}

EXPORT_GPL_DEV_PM_OPS(arizona_pm_ops) = {
        RUNTIME_PM_OPS(arizona_runtime_suspend,
                       arizona_runtime_resume,
                       NULL)
        SYSTEM_SLEEP_PM_OPS(arizona_suspend, arizona_resume)
        NOIRQ_SYSTEM_SLEEP_PM_OPS(arizona_suspend_noirq,
                                  arizona_resume_noirq)
};

#ifdef CONFIG_OF
static int arizona_of_get_core_pdata(struct arizona *arizona)
{
        struct arizona_pdata *pdata = &arizona->pdata;
        int ret, i;

        /* Handle old non-standard DT binding */
        pdata->reset = devm_gpiod_get(arizona->dev, "wlf,reset", GPIOD_OUT_LOW);
        if (IS_ERR(pdata->reset)) {
                ret = PTR_ERR(pdata->reset);

                /*
                 * Reset missing will be caught when other binding is read
                 * but all other errors imply this binding is in use but has
                 * encountered a problem so should be handled.
                 */
                if (ret == -EPROBE_DEFER)
                        return ret;
                else if (ret != -ENOENT && ret != -ENOSYS)
                        dev_err(arizona->dev, "Reset GPIO malformed: %d\n",
                                ret);

                pdata->reset = NULL;
        }

        ret = of_property_read_u32_array(arizona->dev->of_node,
                                         "wlf,gpio-defaults",
                                         pdata->gpio_defaults,
                                         ARRAY_SIZE(pdata->gpio_defaults));
        if (ret >= 0) {
                /*
                 * All values are literal except out of range values
                 * which are chip default, translate into platform
                 * data which uses 0 as chip default and out of range
                 * as zero.
                 */
                for (i = 0; i < ARRAY_SIZE(pdata->gpio_defaults); i++) {
                        if (pdata->gpio_defaults[i] > 0xffff)
                                pdata->gpio_defaults[i] = 0;
                        else if (pdata->gpio_defaults[i] == 0)
                                pdata->gpio_defaults[i] = 0x10000;
                }
        } else {
                dev_err(arizona->dev, "Failed to parse GPIO defaults: %d\n",
                        ret);
        }

        return 0;
}
#else
static inline int arizona_of_get_core_pdata(struct arizona *arizona)
{
        return 0;
}
#endif

static const struct mfd_cell early_devs[] = {
        { .name = "arizona-ldo1" },
};

static const char * const wm5102_supplies[] = {
        "MICVDD",
        "DBVDD2",
        "DBVDD3",
        "CPVDD",
        "SPKVDDL",
        "SPKVDDR",
};

static const struct mfd_cell wm5102_devs[] = {
        { .name = "arizona-micsupp" },
        { .name = "arizona-gpio" },
        { .name = "arizona-haptics" },
        { .name = "arizona-pwm" },
        {
                .name = "wm5102-codec",
                .parent_supplies = wm5102_supplies,
                .num_parent_supplies = ARRAY_SIZE(wm5102_supplies),
        },
};

static const struct mfd_cell wm5110_devs[] = {
        { .name = "arizona-micsupp" },
        { .name = "arizona-gpio" },
        { .name = "arizona-haptics" },
        { .name = "arizona-pwm" },
        {
                .name = "wm5110-codec",
                .parent_supplies = wm5102_supplies,
                .num_parent_supplies = ARRAY_SIZE(wm5102_supplies),
        },
};

static const char * const cs47l24_supplies[] = {
        "MICVDD",
        "CPVDD",
        "SPKVDD",
};

static const struct mfd_cell cs47l24_devs[] = {
        { .name = "arizona-gpio" },
        { .name = "arizona-haptics" },
        { .name = "arizona-pwm" },
        {
                .name = "cs47l24-codec",
                .parent_supplies = cs47l24_supplies,
                .num_parent_supplies = ARRAY_SIZE(cs47l24_supplies),
        },
};

static const char * const wm8997_supplies[] = {
        "MICVDD",
        "DBVDD2",
        "CPVDD",
        "SPKVDD",
};

static const struct mfd_cell wm8997_devs[] = {
        { .name = "arizona-micsupp" },
        { .name = "arizona-gpio" },
        { .name = "arizona-haptics" },
        { .name = "arizona-pwm" },
        {
                .name = "wm8997-codec",
                .parent_supplies = wm8997_supplies,
                .num_parent_supplies = ARRAY_SIZE(wm8997_supplies),
        },
};

static const struct mfd_cell wm8998_devs[] = {
        { .name = "arizona-micsupp" },
        { .name = "arizona-gpio" },
        { .name = "arizona-haptics" },
        { .name = "arizona-pwm" },
        {
                .name = "wm8998-codec",
                .parent_supplies = wm5102_supplies,
                .num_parent_supplies = ARRAY_SIZE(wm5102_supplies),
        },
};

int arizona_dev_init(struct arizona *arizona)
{
        static const char * const mclk_name[] = { "mclk1", "mclk2" };
        struct device *dev = arizona->dev;
        const char *type_name = NULL;
        unsigned int reg, val;
        int (*apply_patch)(struct arizona *) = NULL;
        const struct mfd_cell *subdevs = NULL;
        int n_subdevs = 0, ret, i;

        dev_set_drvdata(arizona->dev, arizona);
        mutex_init(&arizona->clk_lock);

        if (dev_get_platdata(arizona->dev)) {
                memcpy(&arizona->pdata, dev_get_platdata(arizona->dev),
                       sizeof(arizona->pdata));
        } else {
                ret = arizona_of_get_core_pdata(arizona);
                if (ret < 0)
                        return ret;
        }

        BUILD_BUG_ON(ARRAY_SIZE(arizona->mclk) != ARRAY_SIZE(mclk_name));
        for (i = 0; i < ARRAY_SIZE(arizona->mclk); i++) {
                arizona->mclk[i] = devm_clk_get(arizona->dev, mclk_name[i]);
                if (IS_ERR(arizona->mclk[i])) {
                        dev_info(arizona->dev, "Failed to get %s: %ld\n",
                                 mclk_name[i], PTR_ERR(arizona->mclk[i]));
                        arizona->mclk[i] = NULL;
                }
        }

        regcache_cache_only(arizona->regmap, true);

        switch (arizona->type) {
        case WM5102:
        case WM5110:
        case WM8280:
        case WM8997:
        case WM8998:
        case WM1814:
        case WM1831:
        case CS47L24:
                for (i = 0; i < ARRAY_SIZE(wm5102_core_supplies); i++)
                        arizona->core_supplies[i].supply
                                = wm5102_core_supplies[i];
                arizona->num_core_supplies = ARRAY_SIZE(wm5102_core_supplies);
                break;
        default:
                dev_err(arizona->dev, "Unknown device type %d\n",
                        arizona->type);
                return -ENODEV;
        }

        /* Mark DCVDD as external, LDO1 driver will clear if internal */
        arizona->external_dcvdd = true;

        switch (arizona->type) {
        case WM1831:
        case CS47L24:
                break; /* No LDO1 regulator */
        default:
                ret = mfd_add_devices(arizona->dev, -1, early_devs,
                                      ARRAY_SIZE(early_devs), NULL, 0, NULL);
                if (ret != 0) {
                        dev_err(dev, "Failed to add early children: %d\n", ret);
                        return ret;
                }
                break;
        }

        ret = devm_regulator_bulk_get(dev, arizona->num_core_supplies,
                                      arizona->core_supplies);
        if (ret != 0) {
                dev_err(dev, "Failed to request core supplies: %d\n",
                        ret);
                goto err_early;
        }

        /**
         * Don't use devres here because the only device we have to get
         * against is the MFD device and DCVDD will likely be supplied by
         * one of its children. Meaning that the regulator will be
         * destroyed by the time devres calls regulator put.
         */
        arizona->dcvdd = regulator_get(arizona->dev, "DCVDD");
        if (IS_ERR(arizona->dcvdd)) {
                ret = PTR_ERR(arizona->dcvdd);
                dev_err(dev, "Failed to request DCVDD: %d\n", ret);
                goto err_early;
        }

        if (!arizona->pdata.reset) {
                /* Start out with /RESET low to put the chip into reset */
                arizona->pdata.reset = devm_gpiod_get(arizona->dev, "reset",
                                                      GPIOD_OUT_LOW);
                if (IS_ERR(arizona->pdata.reset)) {
                        ret = PTR_ERR(arizona->pdata.reset);
                        if (ret == -EPROBE_DEFER)
                                goto err_dcvdd;

                        dev_err(arizona->dev,
                                "Reset GPIO missing/malformed: %d\n", ret);

                        arizona->pdata.reset = NULL;
                }
        }

        ret = regulator_bulk_enable(arizona->num_core_supplies,
                                    arizona->core_supplies);
        if (ret != 0) {
                dev_err(dev, "Failed to enable core supplies: %d\n",
                        ret);
                goto err_dcvdd;
        }

        ret = regulator_enable(arizona->dcvdd);
        if (ret != 0) {
                dev_err(dev, "Failed to enable DCVDD: %d\n", ret);
                goto err_enable;
        }

        arizona_disable_reset(arizona);

        regcache_cache_only(arizona->regmap, false);

        /* Verify that this is a chip we know about */
        ret = regmap_read(arizona->regmap, ARIZONA_SOFTWARE_RESET, &reg);
        if (ret != 0) {
                dev_err(dev, "Failed to read ID register: %d\n", ret);
                goto err_reset;
        }

        switch (reg) {
        case 0x5102:
        case 0x5110:
        case 0x6349:
        case 0x6363:
        case 0x8997:
                break;
        default:
                dev_err(arizona->dev, "Unknown device ID: %x\n", reg);
                ret = -ENODEV;
                goto err_reset;
        }

        /* If we have a /RESET GPIO we'll already be reset */
        if (!arizona->pdata.reset) {
                ret = regmap_write(arizona->regmap, ARIZONA_SOFTWARE_RESET, 0);
                if (ret != 0) {
                        dev_err(dev, "Failed to reset device: %d\n", ret);
                        goto err_reset;
                }

                usleep_range(1000, 5000);
        }

        /* Ensure device startup is complete */
        switch (arizona->type) {
        case WM5102:
                ret = regmap_read(arizona->regmap,
                                  ARIZONA_WRITE_SEQUENCER_CTRL_3, &val);
                if (ret) {
                        dev_err(dev,
                                "Failed to check write sequencer state: %d\n",
                                ret);
                } else if (val & 0x01) {
                        ret = wm5102_clear_write_sequencer(arizona);
                        if (ret)
                                goto err_reset;
                }
                break;
        default:
                break;
        }

        ret = arizona_wait_for_boot(arizona);
        if (ret) {
                dev_err(arizona->dev, "Device failed initial boot: %d\n", ret);
                goto err_reset;
        }

        /* Read the device ID information & do device specific stuff */
        ret = regmap_read(arizona->regmap, ARIZONA_SOFTWARE_RESET, &reg);
        if (ret != 0) {
                dev_err(dev, "Failed to read ID register: %d\n", ret);
                goto err_reset;
        }

        ret = regmap_read(arizona->regmap, ARIZONA_DEVICE_REVISION,
                          &arizona->rev);
        if (ret != 0) {
                dev_err(dev, "Failed to read revision register: %d\n", ret);
                goto err_reset;
        }
        arizona->rev &= ARIZONA_DEVICE_REVISION_MASK;

        switch (reg) {
        case 0x5102:
                if (IS_ENABLED(CONFIG_MFD_WM5102)) {
                        type_name = "WM5102";
                        if (arizona->type != WM5102) {
                                dev_warn(arizona->dev,
                                         "WM5102 registered as %d\n",
                                         arizona->type);
                                arizona->type = WM5102;
                        }

                        apply_patch = wm5102_patch;
                        arizona->rev &= 0x7;
                        subdevs = wm5102_devs;
                        n_subdevs = ARRAY_SIZE(wm5102_devs);
                }
                break;
        case 0x5110:
                if (IS_ENABLED(CONFIG_MFD_WM5110)) {
                        switch (arizona->type) {
                        case WM5110:
                                type_name = "WM5110";
                                break;
                        case WM8280:
                                type_name = "WM8280";
                                break;
                        default:
                                type_name = "WM5110";
                                dev_warn(arizona->dev,
                                         "WM5110 registered as %d\n",
                                         arizona->type);
                                arizona->type = WM5110;
                                break;
                        }

                        apply_patch = wm5110_patch;
                        subdevs = wm5110_devs;
                        n_subdevs = ARRAY_SIZE(wm5110_devs);
                }
                break;
        case 0x6363:
                if (IS_ENABLED(CONFIG_MFD_CS47L24)) {
                        switch (arizona->type) {
                        case CS47L24:
                                type_name = "CS47L24";
                                break;

                        case WM1831:
                                type_name = "WM1831";
                                break;

                        default:
                                dev_warn(arizona->dev,
                                         "CS47L24 registered as %d\n",
                                         arizona->type);
                                arizona->type = CS47L24;
                                break;
                        }

                        apply_patch = cs47l24_patch;
                        subdevs = cs47l24_devs;
                        n_subdevs = ARRAY_SIZE(cs47l24_devs);
                }
                break;
        case 0x8997:
                if (IS_ENABLED(CONFIG_MFD_WM8997)) {
                        type_name = "WM8997";
                        if (arizona->type != WM8997) {
                                dev_warn(arizona->dev,
                                         "WM8997 registered as %d\n",
                                         arizona->type);
                                arizona->type = WM8997;
                        }

                        apply_patch = wm8997_patch;
                        subdevs = wm8997_devs;
                        n_subdevs = ARRAY_SIZE(wm8997_devs);
                }
                break;
        case 0x6349:
                if (IS_ENABLED(CONFIG_MFD_WM8998)) {
                        switch (arizona->type) {
                        case WM8998:
                                type_name = "WM8998";
                                break;

                        case WM1814:
                                type_name = "WM1814";
                                break;

                        default:
                                type_name = "WM8998";
                                dev_warn(arizona->dev,
                                         "WM8998 registered as %d\n",
                                         arizona->type);
                                arizona->type = WM8998;
                        }

                        apply_patch = wm8998_patch;
                        subdevs = wm8998_devs;
                        n_subdevs = ARRAY_SIZE(wm8998_devs);
                }
                break;
        default:
                dev_err(arizona->dev, "Unknown device ID %x\n", reg);
                ret = -ENODEV;
                goto err_reset;
        }

        if (!subdevs) {
                dev_err(arizona->dev,
                        "No kernel support for device ID %x\n", reg);
                ret = -ENODEV;
                goto err_reset;
        }

        dev_info(dev, "%s revision %c\n", type_name, arizona->rev + 'A');

        if (apply_patch) {
                ret = apply_patch(arizona);
                if (ret != 0) {
                        dev_err(arizona->dev, "Failed to apply patch: %d\n",
                                ret);
                        goto err_reset;
                }

                switch (arizona->type) {
                case WM5102:
                        ret = wm5102_apply_hardware_patch(arizona);
                        if (ret) {
                                dev_err(arizona->dev,
                                        "Failed to apply hardware patch: %d\n",
                                        ret);
                                goto err_reset;
                        }
                        break;
                case WM5110:
                case WM8280:
                        ret = wm5110_apply_sleep_patch(arizona);
                        if (ret) {
                                dev_err(arizona->dev,
                                        "Failed to apply sleep patch: %d\n",
                                        ret);
                                goto err_reset;
                        }
                        break;
                default:
                        break;
                }
        }

        for (i = 0; i < ARRAY_SIZE(arizona->pdata.gpio_defaults); i++) {
                if (!arizona->pdata.gpio_defaults[i])
                        continue;

                regmap_write(arizona->regmap, ARIZONA_GPIO1_CTRL + i,
                             arizona->pdata.gpio_defaults[i]);
        }

        /* Chip default */
        if (!arizona->pdata.clk32k_src)
                arizona->pdata.clk32k_src = ARIZONA_32KZ_MCLK2;

        switch (arizona->pdata.clk32k_src) {
        case ARIZONA_32KZ_MCLK1:
        case ARIZONA_32KZ_MCLK2:
                regmap_update_bits(arizona->regmap, ARIZONA_CLOCK_32K_1,
                                   ARIZONA_CLK_32K_SRC_MASK,
                                   arizona->pdata.clk32k_src - 1);
                arizona_clk32k_enable(arizona);
                break;
        case ARIZONA_32KZ_NONE:
                regmap_update_bits(arizona->regmap, ARIZONA_CLOCK_32K_1,
                                   ARIZONA_CLK_32K_SRC_MASK, 2);
                break;
        default:
                dev_err(arizona->dev, "Invalid 32kHz clock source: %d\n",
                        arizona->pdata.clk32k_src);
                ret = -EINVAL;
                goto err_reset;
        }

        for (i = 0; i < ARIZONA_MAX_MICBIAS; i++) {
                if (!arizona->pdata.micbias[i].mV &&
                    !arizona->pdata.micbias[i].bypass)
                        continue;

                /* Apply default for bypass mode */
                if (!arizona->pdata.micbias[i].mV)
                        arizona->pdata.micbias[i].mV = 2800;

                val = (arizona->pdata.micbias[i].mV - 1500) / 100;

                val <<= ARIZONA_MICB1_LVL_SHIFT;

                if (arizona->pdata.micbias[i].ext_cap)
                        val |= ARIZONA_MICB1_EXT_CAP;

                if (arizona->pdata.micbias[i].discharge)
                        val |= ARIZONA_MICB1_DISCH;

                if (arizona->pdata.micbias[i].soft_start)
                        val |= ARIZONA_MICB1_RATE;

                if (arizona->pdata.micbias[i].bypass)
                        val |= ARIZONA_MICB1_BYPASS;

                regmap_update_bits(arizona->regmap,
                                   ARIZONA_MIC_BIAS_CTRL_1 + i,
                                   ARIZONA_MICB1_LVL_MASK |
                                   ARIZONA_MICB1_EXT_CAP |
                                   ARIZONA_MICB1_DISCH |
                                   ARIZONA_MICB1_BYPASS |
                                   ARIZONA_MICB1_RATE, val);
        }

        pm_runtime_set_active(arizona->dev);
        pm_runtime_enable(arizona->dev);

        /* Set up for interrupts */
        ret = arizona_irq_init(arizona);
        if (ret != 0)
                goto err_pm;

        pm_runtime_set_autosuspend_delay(arizona->dev, 100);
        pm_runtime_use_autosuspend(arizona->dev);

        arizona_request_irq(arizona, ARIZONA_IRQ_CLKGEN_ERR, "CLKGEN error",
                            arizona_clkgen_err, arizona);
        arizona_request_irq(arizona, ARIZONA_IRQ_OVERCLOCKED, "Overclocked",
                            arizona_overclocked, arizona);
        arizona_request_irq(arizona, ARIZONA_IRQ_UNDERCLOCKED, "Underclocked",
                            arizona_underclocked, arizona);

        ret = mfd_add_devices(arizona->dev, PLATFORM_DEVID_NONE,
                              subdevs, n_subdevs, NULL, 0, NULL);

        if (ret) {
                dev_err(arizona->dev, "Failed to add subdevices: %d\n", ret);
                goto err_irq;
        }

        return 0;

err_irq:
        arizona_irq_exit(arizona);
err_pm:
        pm_runtime_disable(arizona->dev);

        switch (arizona->pdata.clk32k_src) {
        case ARIZONA_32KZ_MCLK1:
        case ARIZONA_32KZ_MCLK2:
                arizona_clk32k_disable(arizona);
                break;
        default:
                break;
        }
err_reset:
        arizona_enable_reset(arizona);
        regulator_disable(arizona->dcvdd);
err_enable:
        regulator_bulk_disable(arizona->num_core_supplies,
                               arizona->core_supplies);
err_dcvdd:
        regulator_put(arizona->dcvdd);
err_early:
        mfd_remove_devices(dev);
        return ret;
}
EXPORT_SYMBOL_GPL(arizona_dev_init);

int arizona_dev_exit(struct arizona *arizona)
{
        disable_irq(arizona->irq);
        pm_runtime_disable(arizona->dev);

        regulator_disable(arizona->dcvdd);
        regulator_put(arizona->dcvdd);

        switch (arizona->pdata.clk32k_src) {
        case ARIZONA_32KZ_MCLK1:
        case ARIZONA_32KZ_MCLK2:
                arizona_clk32k_disable(arizona);
                break;
        default:
                break;
        }

        mfd_remove_devices(arizona->dev);
        arizona_free_irq(arizona, ARIZONA_IRQ_UNDERCLOCKED, arizona);
        arizona_free_irq(arizona, ARIZONA_IRQ_OVERCLOCKED, arizona);
        arizona_free_irq(arizona, ARIZONA_IRQ_CLKGEN_ERR, arizona);
        arizona_irq_exit(arizona);
        arizona_enable_reset(arizona);

        regulator_bulk_disable(arizona->num_core_supplies,
                               arizona->core_supplies);
        return 0;
}
EXPORT_SYMBOL_GPL(arizona_dev_exit);

MODULE_DESCRIPTION("Wolfson Arizona core driver");
MODULE_LICENSE("GPL v2");