root/sound/soc/codecs/pcm512x.c
// SPDX-License-Identifier: GPL-2.0-only
/*
 * Driver for the PCM512x CODECs
 *
 * Author:      Mark Brown <broonie@kernel.org>
 *              Copyright 2014 Linaro Ltd
 */


#include <linux/init.h>
#include <linux/module.h>
#include <linux/clk.h>
#include <linux/kernel.h>
#include <linux/pm_runtime.h>
#include <linux/regmap.h>
#include <linux/regulator/consumer.h>
#include <linux/gcd.h>
#include <sound/soc.h>
#include <sound/soc-dapm.h>
#include <sound/pcm_params.h>
#include <sound/tlv.h>

#include "pcm512x.h"

#define PCM512x_NUM_SUPPLIES 3
static const char * const pcm512x_supply_names[PCM512x_NUM_SUPPLIES] = {
        "AVDD",
        "DVDD",
        "CPVDD",
};

struct pcm512x_priv {
        struct regmap *regmap;
        struct clk *sclk;
        struct regulator_bulk_data supplies[PCM512x_NUM_SUPPLIES];
        struct notifier_block supply_nb[PCM512x_NUM_SUPPLIES];
        int fmt;
        int pll_in;
        int pll_out;
        int pll_r;
        int pll_j;
        int pll_d;
        int pll_p;
        unsigned long real_pll;
        unsigned long overclock_pll;
        unsigned long overclock_dac;
        unsigned long overclock_dsp;
        int mute;
        struct mutex mutex;
        unsigned int bclk_ratio;
        int force_pll_on;
};

/*
 * We can't use the same notifier block for more than one supply and
 * there's no way I can see to get from a callback to the caller
 * except container_of().
 */
#define PCM512x_REGULATOR_EVENT(n) \
static int pcm512x_regulator_event_##n(struct notifier_block *nb, \
                                      unsigned long event, void *data)    \
{ \
        struct pcm512x_priv *pcm512x = container_of(nb, struct pcm512x_priv, \
                                                    supply_nb[n]); \
        if (event & REGULATOR_EVENT_DISABLE) { \
                regcache_mark_dirty(pcm512x->regmap);   \
                regcache_cache_only(pcm512x->regmap, true);     \
        } \
        return 0; \
}

PCM512x_REGULATOR_EVENT(0)
PCM512x_REGULATOR_EVENT(1)
PCM512x_REGULATOR_EVENT(2)

static const struct reg_default pcm512x_reg_defaults[] = {
        { PCM512x_RESET,             0x00 },
        { PCM512x_POWER,             0x00 },
        { PCM512x_MUTE,              0x00 },
        { PCM512x_DSP,               0x00 },
        { PCM512x_PLL_REF,           0x00 },
        { PCM512x_DAC_REF,           0x00 },
        { PCM512x_DAC_ROUTING,       0x11 },
        { PCM512x_DSP_PROGRAM,       0x01 },
        { PCM512x_CLKDET,            0x00 },
        { PCM512x_AUTO_MUTE,         0x00 },
        { PCM512x_ERROR_DETECT,      0x00 },
        { PCM512x_DIGITAL_VOLUME_1,  0x00 },
        { PCM512x_DIGITAL_VOLUME_2,  0x30 },
        { PCM512x_DIGITAL_VOLUME_3,  0x30 },
        { PCM512x_DIGITAL_MUTE_1,    0x22 },
        { PCM512x_DIGITAL_MUTE_2,    0x00 },
        { PCM512x_DIGITAL_MUTE_3,    0x07 },
        { PCM512x_OUTPUT_AMPLITUDE,  0x00 },
        { PCM512x_ANALOG_GAIN_CTRL,  0x00 },
        { PCM512x_UNDERVOLTAGE_PROT, 0x00 },
        { PCM512x_ANALOG_MUTE_CTRL,  0x00 },
        { PCM512x_ANALOG_GAIN_BOOST, 0x00 },
        { PCM512x_VCOM_CTRL_1,       0x00 },
        { PCM512x_VCOM_CTRL_2,       0x01 },
        { PCM512x_BCLK_LRCLK_CFG,    0x00 },
        { PCM512x_MASTER_MODE,       0x7c },
        { PCM512x_GPIO_DACIN,        0x00 },
        { PCM512x_GPIO_PLLIN,        0x00 },
        { PCM512x_SYNCHRONIZE,       0x10 },
        { PCM512x_PLL_COEFF_0,       0x00 },
        { PCM512x_PLL_COEFF_1,       0x00 },
        { PCM512x_PLL_COEFF_2,       0x00 },
        { PCM512x_PLL_COEFF_3,       0x00 },
        { PCM512x_PLL_COEFF_4,       0x00 },
        { PCM512x_DSP_CLKDIV,        0x00 },
        { PCM512x_DAC_CLKDIV,        0x00 },
        { PCM512x_NCP_CLKDIV,        0x00 },
        { PCM512x_OSR_CLKDIV,        0x00 },
        { PCM512x_MASTER_CLKDIV_1,   0x00 },
        { PCM512x_MASTER_CLKDIV_2,   0x00 },
        { PCM512x_FS_SPEED_MODE,     0x00 },
        { PCM512x_IDAC_1,            0x01 },
        { PCM512x_IDAC_2,            0x00 },
        { PCM512x_I2S_1,             0x02 },
        { PCM512x_I2S_2,             0x00 },
};

static bool pcm512x_readable(struct device *dev, unsigned int reg)
{
        switch (reg) {
        case PCM512x_RESET:
        case PCM512x_POWER:
        case PCM512x_MUTE:
        case PCM512x_PLL_EN:
        case PCM512x_SPI_MISO_FUNCTION:
        case PCM512x_DSP:
        case PCM512x_GPIO_EN:
        case PCM512x_BCLK_LRCLK_CFG:
        case PCM512x_DSP_GPIO_INPUT:
        case PCM512x_MASTER_MODE:
        case PCM512x_PLL_REF:
        case PCM512x_DAC_REF:
        case PCM512x_GPIO_DACIN:
        case PCM512x_GPIO_PLLIN:
        case PCM512x_SYNCHRONIZE:
        case PCM512x_PLL_COEFF_0:
        case PCM512x_PLL_COEFF_1:
        case PCM512x_PLL_COEFF_2:
        case PCM512x_PLL_COEFF_3:
        case PCM512x_PLL_COEFF_4:
        case PCM512x_DSP_CLKDIV:
        case PCM512x_DAC_CLKDIV:
        case PCM512x_NCP_CLKDIV:
        case PCM512x_OSR_CLKDIV:
        case PCM512x_MASTER_CLKDIV_1:
        case PCM512x_MASTER_CLKDIV_2:
        case PCM512x_FS_SPEED_MODE:
        case PCM512x_IDAC_1:
        case PCM512x_IDAC_2:
        case PCM512x_ERROR_DETECT:
        case PCM512x_I2S_1:
        case PCM512x_I2S_2:
        case PCM512x_DAC_ROUTING:
        case PCM512x_DSP_PROGRAM:
        case PCM512x_CLKDET:
        case PCM512x_AUTO_MUTE:
        case PCM512x_DIGITAL_VOLUME_1:
        case PCM512x_DIGITAL_VOLUME_2:
        case PCM512x_DIGITAL_VOLUME_3:
        case PCM512x_DIGITAL_MUTE_1:
        case PCM512x_DIGITAL_MUTE_2:
        case PCM512x_DIGITAL_MUTE_3:
        case PCM512x_GPIO_OUTPUT_1:
        case PCM512x_GPIO_OUTPUT_2:
        case PCM512x_GPIO_OUTPUT_3:
        case PCM512x_GPIO_OUTPUT_4:
        case PCM512x_GPIO_OUTPUT_5:
        case PCM512x_GPIO_OUTPUT_6:
        case PCM512x_GPIO_CONTROL_1:
        case PCM512x_GPIO_CONTROL_2:
        case PCM512x_OVERFLOW:
        case PCM512x_RATE_DET_1:
        case PCM512x_RATE_DET_2:
        case PCM512x_RATE_DET_3:
        case PCM512x_RATE_DET_4:
        case PCM512x_CLOCK_STATUS:
        case PCM512x_ANALOG_MUTE_DET:
        case PCM512x_GPIN:
        case PCM512x_DIGITAL_MUTE_DET:
        case PCM512x_OUTPUT_AMPLITUDE:
        case PCM512x_ANALOG_GAIN_CTRL:
        case PCM512x_UNDERVOLTAGE_PROT:
        case PCM512x_ANALOG_MUTE_CTRL:
        case PCM512x_ANALOG_GAIN_BOOST:
        case PCM512x_VCOM_CTRL_1:
        case PCM512x_VCOM_CTRL_2:
        case PCM512x_CRAM_CTRL:
        case PCM512x_FLEX_A:
        case PCM512x_FLEX_B:
                return true;
        default:
                /* There are 256 raw register addresses */
                return reg < 0xff;
        }
}

static bool pcm512x_volatile(struct device *dev, unsigned int reg)
{
        switch (reg) {
        case PCM512x_PLL_EN:
        case PCM512x_OVERFLOW:
        case PCM512x_RATE_DET_1:
        case PCM512x_RATE_DET_2:
        case PCM512x_RATE_DET_3:
        case PCM512x_RATE_DET_4:
        case PCM512x_CLOCK_STATUS:
        case PCM512x_ANALOG_MUTE_DET:
        case PCM512x_GPIN:
        case PCM512x_DIGITAL_MUTE_DET:
        case PCM512x_CRAM_CTRL:
                return true;
        default:
                /* There are 256 raw register addresses */
                return reg < 0xff;
        }
}

static int pcm512x_overclock_pll_get(struct snd_kcontrol *kcontrol,
                                     struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct pcm512x_priv *pcm512x = snd_soc_component_get_drvdata(component);

        ucontrol->value.integer.value[0] = pcm512x->overclock_pll;
        return 0;
}

static int pcm512x_overclock_pll_put(struct snd_kcontrol *kcontrol,
                                     struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct snd_soc_dapm_context *dapm = snd_soc_component_to_dapm(component);
        struct pcm512x_priv *pcm512x = snd_soc_component_get_drvdata(component);

        switch (snd_soc_dapm_get_bias_level(dapm)) {
        case SND_SOC_BIAS_OFF:
        case SND_SOC_BIAS_STANDBY:
                break;
        default:
                return -EBUSY;
        }

        pcm512x->overclock_pll = ucontrol->value.integer.value[0];
        return 0;
}

static int pcm512x_overclock_dsp_get(struct snd_kcontrol *kcontrol,
                                     struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct pcm512x_priv *pcm512x = snd_soc_component_get_drvdata(component);

        ucontrol->value.integer.value[0] = pcm512x->overclock_dsp;
        return 0;
}

static int pcm512x_overclock_dsp_put(struct snd_kcontrol *kcontrol,
                                     struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct snd_soc_dapm_context *dapm = snd_soc_component_to_dapm(component);
        struct pcm512x_priv *pcm512x = snd_soc_component_get_drvdata(component);

        switch (snd_soc_dapm_get_bias_level(dapm)) {
        case SND_SOC_BIAS_OFF:
        case SND_SOC_BIAS_STANDBY:
                break;
        default:
                return -EBUSY;
        }

        pcm512x->overclock_dsp = ucontrol->value.integer.value[0];
        return 0;
}

static int pcm512x_overclock_dac_get(struct snd_kcontrol *kcontrol,
                                     struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct pcm512x_priv *pcm512x = snd_soc_component_get_drvdata(component);

        ucontrol->value.integer.value[0] = pcm512x->overclock_dac;
        return 0;
}

static int pcm512x_overclock_dac_put(struct snd_kcontrol *kcontrol,
                                     struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct snd_soc_dapm_context *dapm = snd_soc_component_to_dapm(component);
        struct pcm512x_priv *pcm512x = snd_soc_component_get_drvdata(component);

        switch (snd_soc_dapm_get_bias_level(dapm)) {
        case SND_SOC_BIAS_OFF:
        case SND_SOC_BIAS_STANDBY:
                break;
        default:
                return -EBUSY;
        }

        pcm512x->overclock_dac = ucontrol->value.integer.value[0];
        return 0;
}

static const DECLARE_TLV_DB_SCALE(digital_tlv, -10350, 50, 1);
static const DECLARE_TLV_DB_SCALE(analog_tlv, -600, 600, 0);
static const DECLARE_TLV_DB_SCALE(boost_tlv, 0, 80, 0);

static const char * const pcm512x_dsp_program_texts[] = {
        "FIR interpolation with de-emphasis",
        "Low latency IIR with de-emphasis",
        "High attenuation with de-emphasis",
        "Fixed process flow",
        "Ringing-less low latency FIR",
};

static const unsigned int pcm512x_dsp_program_values[] = {
        1,
        2,
        3,
        5,
        7,
};

static SOC_VALUE_ENUM_SINGLE_DECL(pcm512x_dsp_program,
                                  PCM512x_DSP_PROGRAM, 0, 0x1f,
                                  pcm512x_dsp_program_texts,
                                  pcm512x_dsp_program_values);

static const char * const pcm512x_clk_missing_text[] = {
        "1s", "2s", "3s", "4s", "5s", "6s", "7s", "8s"
};

static const struct soc_enum pcm512x_clk_missing =
        SOC_ENUM_SINGLE(PCM512x_CLKDET, 0,  8, pcm512x_clk_missing_text);

static const char * const pcm512x_autom_text[] = {
        "21ms", "106ms", "213ms", "533ms", "1.07s", "2.13s", "5.33s", "10.66s"
};

static const struct soc_enum pcm512x_autom_l =
        SOC_ENUM_SINGLE(PCM512x_AUTO_MUTE, PCM512x_ATML_SHIFT, 8,
                        pcm512x_autom_text);

static const struct soc_enum pcm512x_autom_r =
        SOC_ENUM_SINGLE(PCM512x_AUTO_MUTE, PCM512x_ATMR_SHIFT, 8,
                        pcm512x_autom_text);

static const char * const pcm512x_ramp_rate_text[] = {
        "1 sample/update", "2 samples/update", "4 samples/update",
        "Immediate"
};

static const struct soc_enum pcm512x_vndf =
        SOC_ENUM_SINGLE(PCM512x_DIGITAL_MUTE_1, PCM512x_VNDF_SHIFT, 4,
                        pcm512x_ramp_rate_text);

static const struct soc_enum pcm512x_vnuf =
        SOC_ENUM_SINGLE(PCM512x_DIGITAL_MUTE_1, PCM512x_VNUF_SHIFT, 4,
                        pcm512x_ramp_rate_text);

static const struct soc_enum pcm512x_vedf =
        SOC_ENUM_SINGLE(PCM512x_DIGITAL_MUTE_2, PCM512x_VEDF_SHIFT, 4,
                        pcm512x_ramp_rate_text);

static const char * const pcm512x_ramp_step_text[] = {
        "4dB/step", "2dB/step", "1dB/step", "0.5dB/step"
};

static const struct soc_enum pcm512x_vnds =
        SOC_ENUM_SINGLE(PCM512x_DIGITAL_MUTE_1, PCM512x_VNDS_SHIFT, 4,
                        pcm512x_ramp_step_text);

static const struct soc_enum pcm512x_vnus =
        SOC_ENUM_SINGLE(PCM512x_DIGITAL_MUTE_1, PCM512x_VNUS_SHIFT, 4,
                        pcm512x_ramp_step_text);

static const struct soc_enum pcm512x_veds =
        SOC_ENUM_SINGLE(PCM512x_DIGITAL_MUTE_2, PCM512x_VEDS_SHIFT, 4,
                        pcm512x_ramp_step_text);

static int pcm512x_update_mute(struct pcm512x_priv *pcm512x)
{
        return regmap_update_bits(
                pcm512x->regmap, PCM512x_MUTE, PCM512x_RQML | PCM512x_RQMR,
                (!!(pcm512x->mute & 0x5) << PCM512x_RQML_SHIFT)
                | (!!(pcm512x->mute & 0x3) << PCM512x_RQMR_SHIFT));
}

static int pcm512x_digital_playback_switch_get(struct snd_kcontrol *kcontrol,
                                               struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct pcm512x_priv *pcm512x = snd_soc_component_get_drvdata(component);

        mutex_lock(&pcm512x->mutex);
        ucontrol->value.integer.value[0] = !(pcm512x->mute & 0x4);
        ucontrol->value.integer.value[1] = !(pcm512x->mute & 0x2);
        mutex_unlock(&pcm512x->mutex);

        return 0;
}

static int pcm512x_digital_playback_switch_put(struct snd_kcontrol *kcontrol,
                                               struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct pcm512x_priv *pcm512x = snd_soc_component_get_drvdata(component);
        int ret, changed = 0;

        mutex_lock(&pcm512x->mutex);

        if ((pcm512x->mute & 0x4) == (ucontrol->value.integer.value[0] << 2)) {
                pcm512x->mute ^= 0x4;
                changed = 1;
        }
        if ((pcm512x->mute & 0x2) == (ucontrol->value.integer.value[1] << 1)) {
                pcm512x->mute ^= 0x2;
                changed = 1;
        }

        if (changed) {
                ret = pcm512x_update_mute(pcm512x);
                if (ret != 0) {
                        dev_err(component->dev,
                                "Failed to update digital mute: %d\n", ret);
                        mutex_unlock(&pcm512x->mutex);
                        return ret;
                }
        }

        mutex_unlock(&pcm512x->mutex);

        return changed;
}

static const struct snd_kcontrol_new pcm512x_controls[] = {
SOC_DOUBLE_R_TLV("Digital Playback Volume", PCM512x_DIGITAL_VOLUME_2,
                 PCM512x_DIGITAL_VOLUME_3, 0, 255, 1, digital_tlv),
SOC_DOUBLE_TLV("Analogue Playback Volume", PCM512x_ANALOG_GAIN_CTRL,
               PCM512x_LAGN_SHIFT, PCM512x_RAGN_SHIFT, 1, 1, analog_tlv),
SOC_DOUBLE_TLV("Analogue Playback Boost Volume", PCM512x_ANALOG_GAIN_BOOST,
               PCM512x_AGBL_SHIFT, PCM512x_AGBR_SHIFT, 1, 0, boost_tlv),
{
        .iface = SNDRV_CTL_ELEM_IFACE_MIXER,
        .name = "Digital Playback Switch",
        .index = 0,
        .access = SNDRV_CTL_ELEM_ACCESS_READWRITE,
        .info = snd_ctl_boolean_stereo_info,
        .get = pcm512x_digital_playback_switch_get,
        .put = pcm512x_digital_playback_switch_put
},

SOC_SINGLE("Deemphasis Switch", PCM512x_DSP, PCM512x_DEMP_SHIFT, 1, 1),
SOC_ENUM("DSP Program", pcm512x_dsp_program),

SOC_ENUM("Clock Missing Period", pcm512x_clk_missing),
SOC_ENUM("Auto Mute Time Left", pcm512x_autom_l),
SOC_ENUM("Auto Mute Time Right", pcm512x_autom_r),
SOC_SINGLE("Auto Mute Mono Switch", PCM512x_DIGITAL_MUTE_3,
           PCM512x_ACTL_SHIFT, 1, 0),
SOC_DOUBLE("Auto Mute Switch", PCM512x_DIGITAL_MUTE_3, PCM512x_AMLE_SHIFT,
           PCM512x_AMRE_SHIFT, 1, 0),

SOC_ENUM("Volume Ramp Down Rate", pcm512x_vndf),
SOC_ENUM("Volume Ramp Down Step", pcm512x_vnds),
SOC_ENUM("Volume Ramp Up Rate", pcm512x_vnuf),
SOC_ENUM("Volume Ramp Up Step", pcm512x_vnus),
SOC_ENUM("Volume Ramp Down Emergency Rate", pcm512x_vedf),
SOC_ENUM("Volume Ramp Down Emergency Step", pcm512x_veds),

SOC_SINGLE_EXT("Max Overclock PLL", SND_SOC_NOPM, 0, 20, 0,
               pcm512x_overclock_pll_get, pcm512x_overclock_pll_put),
SOC_SINGLE_EXT("Max Overclock DSP", SND_SOC_NOPM, 0, 40, 0,
               pcm512x_overclock_dsp_get, pcm512x_overclock_dsp_put),
SOC_SINGLE_EXT("Max Overclock DAC", SND_SOC_NOPM, 0, 40, 0,
               pcm512x_overclock_dac_get, pcm512x_overclock_dac_put),
};

static const struct snd_soc_dapm_widget pcm512x_dapm_widgets[] = {
SND_SOC_DAPM_DAC("DACL", NULL, SND_SOC_NOPM, 0, 0),
SND_SOC_DAPM_DAC("DACR", NULL, SND_SOC_NOPM, 0, 0),

SND_SOC_DAPM_OUTPUT("OUTL"),
SND_SOC_DAPM_OUTPUT("OUTR"),
};

static const struct snd_soc_dapm_route pcm512x_dapm_routes[] = {
        { "DACL", NULL, "Playback" },
        { "DACR", NULL, "Playback" },

        { "OUTL", NULL, "DACL" },
        { "OUTR", NULL, "DACR" },
};

static unsigned long pcm512x_pll_max(struct pcm512x_priv *pcm512x)
{
        return 25000000 + 25000000 * pcm512x->overclock_pll / 100;
}

static unsigned long pcm512x_dsp_max(struct pcm512x_priv *pcm512x)
{
        return 50000000 + 50000000 * pcm512x->overclock_dsp / 100;
}

static unsigned long pcm512x_dac_max(struct pcm512x_priv *pcm512x,
                                     unsigned long rate)
{
        return rate + rate * pcm512x->overclock_dac / 100;
}

static unsigned long pcm512x_sck_max(struct pcm512x_priv *pcm512x)
{
        if (!pcm512x->pll_out)
                return 25000000;
        return pcm512x_pll_max(pcm512x);
}

static unsigned long pcm512x_ncp_target(struct pcm512x_priv *pcm512x,
                                        unsigned long dac_rate)
{
        /*
         * If the DAC is not actually overclocked, use the good old
         * NCP target rate...
         */
        if (dac_rate <= 6144000)
                return 1536000;
        /*
         * ...but if the DAC is in fact overclocked, bump the NCP target
         * rate to get the recommended dividers even when overclocking.
         */
        return pcm512x_dac_max(pcm512x, 1536000);
}

static const u32 pcm512x_dai_rates[] = {
        8000, 11025, 16000, 22050, 32000, 44100, 48000, 64000,
        88200, 96000, 176400, 192000, 384000,
};

static const struct snd_pcm_hw_constraint_list constraints_slave = {
        .count = ARRAY_SIZE(pcm512x_dai_rates),
        .list  = pcm512x_dai_rates,
};

static int pcm512x_hw_rule_rate(struct snd_pcm_hw_params *params,
                                struct snd_pcm_hw_rule *rule)
{
        struct pcm512x_priv *pcm512x = rule->private;
        struct snd_interval ranges[2];
        int frame_size;

        frame_size = snd_soc_params_to_frame_size(params);
        if (frame_size < 0)
                return frame_size;

        switch (frame_size) {
        case 32:
                /* No hole when the frame size is 32. */
                return 0;
        case 48:
        case 64:
                /* There is only one hole in the range of supported
                 * rates, but it moves with the frame size.
                 */
                memset(ranges, 0, sizeof(ranges));
                ranges[0].min = 8000;
                ranges[0].max = pcm512x_sck_max(pcm512x) / frame_size / 2;
                ranges[1].min = DIV_ROUND_UP(16000000, frame_size);
                ranges[1].max = 384000;
                break;
        default:
                return -EINVAL;
        }

        return snd_interval_ranges(hw_param_interval(params, rule->var),
                                   ARRAY_SIZE(ranges), ranges, 0);
}

static int pcm512x_dai_startup_master(struct snd_pcm_substream *substream,
                                      struct snd_soc_dai *dai)
{
        struct snd_soc_component *component = dai->component;
        struct pcm512x_priv *pcm512x = snd_soc_component_get_drvdata(component);
        struct device *dev = dai->dev;
        struct snd_pcm_hw_constraint_ratnums *constraints_no_pll;
        struct snd_ratnum *rats_no_pll;

        if (IS_ERR(pcm512x->sclk)) {
                dev_err(dev, "Need SCLK for master mode: %ld\n",
                        PTR_ERR(pcm512x->sclk));
                return PTR_ERR(pcm512x->sclk);
        }

        if (pcm512x->pll_out)
                return snd_pcm_hw_rule_add(substream->runtime, 0,
                                           SNDRV_PCM_HW_PARAM_RATE,
                                           pcm512x_hw_rule_rate,
                                           pcm512x,
                                           SNDRV_PCM_HW_PARAM_FRAME_BITS,
                                           SNDRV_PCM_HW_PARAM_CHANNELS, -1);

        constraints_no_pll = devm_kzalloc(dev, sizeof(*constraints_no_pll),
                                          GFP_KERNEL);
        if (!constraints_no_pll)
                return -ENOMEM;
        constraints_no_pll->nrats = 1;
        rats_no_pll = devm_kzalloc(dev, sizeof(*rats_no_pll), GFP_KERNEL);
        if (!rats_no_pll)
                return -ENOMEM;
        constraints_no_pll->rats = rats_no_pll;
        rats_no_pll->num = clk_get_rate(pcm512x->sclk) / 64;
        rats_no_pll->den_min = 1;
        rats_no_pll->den_max = 128;
        rats_no_pll->den_step = 1;

        return snd_pcm_hw_constraint_ratnums(substream->runtime, 0,
                                             SNDRV_PCM_HW_PARAM_RATE,
                                             constraints_no_pll);
}

static int pcm512x_dai_startup_slave(struct snd_pcm_substream *substream,
                                     struct snd_soc_dai *dai)
{
        struct snd_soc_component *component = dai->component;
        struct pcm512x_priv *pcm512x = snd_soc_component_get_drvdata(component);
        struct device *dev = dai->dev;
        struct regmap *regmap = pcm512x->regmap;

        if (IS_ERR(pcm512x->sclk)) {
                dev_info_once(dev, "No SCLK, using BCLK: %ld\n",
                         PTR_ERR(pcm512x->sclk));

                /* Disable reporting of missing SCLK as an error */
                regmap_update_bits(regmap, PCM512x_ERROR_DETECT,
                                   PCM512x_IDCH, PCM512x_IDCH);

                /* Switch PLL input to BCLK */
                regmap_update_bits(regmap, PCM512x_PLL_REF,
                                   PCM512x_SREF, PCM512x_SREF_BCK);
        }

        return snd_pcm_hw_constraint_list(substream->runtime, 0,
                                          SNDRV_PCM_HW_PARAM_RATE,
                                          &constraints_slave);
}

static int pcm512x_dai_startup(struct snd_pcm_substream *substream,
                               struct snd_soc_dai *dai)
{
        struct snd_soc_component *component = dai->component;
        struct pcm512x_priv *pcm512x = snd_soc_component_get_drvdata(component);

        switch (pcm512x->fmt & SND_SOC_DAIFMT_CLOCK_PROVIDER_MASK) {
        case SND_SOC_DAIFMT_CBP_CFP:
        case SND_SOC_DAIFMT_CBP_CFC:
                return pcm512x_dai_startup_master(substream, dai);

        case SND_SOC_DAIFMT_CBC_CFC:
                return pcm512x_dai_startup_slave(substream, dai);

        default:
                return -EINVAL;
        }
}

static int pcm512x_set_bias_level(struct snd_soc_component *component,
                                  enum snd_soc_bias_level level)
{
        struct pcm512x_priv *pcm512x = dev_get_drvdata(component->dev);
        int ret;

        switch (level) {
        case SND_SOC_BIAS_ON:
        case SND_SOC_BIAS_PREPARE:
                break;

        case SND_SOC_BIAS_STANDBY:
                ret = regmap_update_bits(pcm512x->regmap, PCM512x_POWER,
                                         PCM512x_RQST, 0);
                if (ret != 0) {
                        dev_err(component->dev, "Failed to remove standby: %d\n",
                                ret);
                        return ret;
                }
                break;

        case SND_SOC_BIAS_OFF:
                ret = regmap_update_bits(pcm512x->regmap, PCM512x_POWER,
                                         PCM512x_RQST, PCM512x_RQST);
                if (ret != 0) {
                        dev_err(component->dev, "Failed to request standby: %d\n",
                                ret);
                        return ret;
                }
                break;
        }

        return 0;
}

static unsigned long pcm512x_find_sck(struct snd_soc_dai *dai,
                                      unsigned long bclk_rate)
{
        struct device *dev = dai->dev;
        struct snd_soc_component *component = dai->component;
        struct pcm512x_priv *pcm512x = snd_soc_component_get_drvdata(component);
        unsigned long sck_rate;
        int pow2;

        /* 64 MHz <= pll_rate <= 100 MHz, VREF mode */
        /* 16 MHz <= sck_rate <=  25 MHz, VREF mode */

        /* select sck_rate as a multiple of bclk_rate but still with
         * as many factors of 2 as possible, as that makes it easier
         * to find a fast DAC rate
         */
        pow2 = 1 << fls((pcm512x_pll_max(pcm512x) - 16000000) / bclk_rate);
        for (; pow2; pow2 >>= 1) {
                sck_rate = rounddown(pcm512x_pll_max(pcm512x),
                                     bclk_rate * pow2);
                if (sck_rate >= 16000000)
                        break;
        }
        if (!pow2) {
                dev_err(dev, "Impossible to generate a suitable SCK\n");
                return 0;
        }

        dev_dbg(dev, "sck_rate %lu\n", sck_rate);
        return sck_rate;
}

/* pll_rate = pllin_rate * R * J.D / P
 * 1 <= R <= 16
 * 1 <= J <= 63
 * 0 <= D <= 9999
 * 1 <= P <= 15
 * 64 MHz <= pll_rate <= 100 MHz
 * if D == 0
 *     1 MHz <= pllin_rate / P <= 20 MHz
 * else if D > 0
 *     6.667 MHz <= pllin_rate / P <= 20 MHz
 *     4 <= J <= 11
 *     R = 1
 */
static int pcm512x_find_pll_coeff(struct snd_soc_dai *dai,
                                  unsigned long pllin_rate,
                                  unsigned long pll_rate)
{
        struct device *dev = dai->dev;
        struct snd_soc_component *component = dai->component;
        struct pcm512x_priv *pcm512x = snd_soc_component_get_drvdata(component);
        unsigned long common;
        int R, J, D, P;
        unsigned long K; /* 10000 * J.D */
        unsigned long num;
        unsigned long den;

        common = gcd(pll_rate, pllin_rate);
        dev_dbg(dev, "pll %lu pllin %lu common %lu\n",
                pll_rate, pllin_rate, common);
        num = pll_rate / common;
        den = pllin_rate / common;

        /* pllin_rate / P (or here, den) cannot be greater than 20 MHz */
        if (pllin_rate / den > 20000000 && num < 8) {
                num *= DIV_ROUND_UP(pllin_rate / den, 20000000);
                den *= DIV_ROUND_UP(pllin_rate / den, 20000000);
        }
        dev_dbg(dev, "num / den = %lu / %lu\n", num, den);

        P = den;
        if (den <= 15 && num <= 16 * 63
            && 1000000 <= pllin_rate / P && pllin_rate / P <= 20000000) {
                /* Try the case with D = 0 */
                D = 0;
                /* factor 'num' into J and R, such that R <= 16 and J <= 63 */
                for (R = 16; R; R--) {
                        if (num % R)
                                continue;
                        J = num / R;
                        if (J == 0 || J > 63)
                                continue;

                        dev_dbg(dev, "R * J / P = %d * %d / %d\n", R, J, P);
                        pcm512x->real_pll = pll_rate;
                        goto done;
                }
                /* no luck */
        }

        R = 1;

        if (num > 0xffffffffUL / 10000)
                goto fallback;

        /* Try to find an exact pll_rate using the D > 0 case */
        common = gcd(10000 * num, den);
        num = 10000 * num / common;
        den /= common;
        dev_dbg(dev, "num %lu den %lu common %lu\n", num, den, common);

        for (P = den; P <= 15; P++) {
                if (pllin_rate / P < 6667000 || 200000000 < pllin_rate / P)
                        continue;
                if (num * P % den)
                        continue;
                K = num * P / den;
                /* J == 12 is ok if D == 0 */
                if (K < 40000 || K > 120000)
                        continue;

                J = K / 10000;
                D = K % 10000;
                dev_dbg(dev, "J.D / P = %d.%04d / %d\n", J, D, P);
                pcm512x->real_pll = pll_rate;
                goto done;
        }

        /* Fall back to an approximate pll_rate */

fallback:
        /* find smallest possible P */
        P = DIV_ROUND_UP(pllin_rate, 20000000);
        if (!P)
                P = 1;
        else if (P > 15) {
                dev_err(dev, "Need a slower clock as pll-input\n");
                return -EINVAL;
        }
        if (pllin_rate / P < 6667000) {
                dev_err(dev, "Need a faster clock as pll-input\n");
                return -EINVAL;
        }
        K = DIV_ROUND_CLOSEST_ULL(10000ULL * pll_rate * P, pllin_rate);
        if (K < 40000)
                K = 40000;
        /* J == 12 is ok if D == 0 */
        if (K > 120000)
                K = 120000;
        J = K / 10000;
        D = K % 10000;
        dev_dbg(dev, "J.D / P ~ %d.%04d / %d\n", J, D, P);
        pcm512x->real_pll = DIV_ROUND_DOWN_ULL((u64)K * pllin_rate, 10000 * P);

done:
        pcm512x->pll_r = R;
        pcm512x->pll_j = J;
        pcm512x->pll_d = D;
        pcm512x->pll_p = P;
        return 0;
}

static unsigned long pcm512x_pllin_dac_rate(struct snd_soc_dai *dai,
                                            unsigned long osr_rate,
                                            unsigned long pllin_rate)
{
        struct snd_soc_component *component = dai->component;
        struct pcm512x_priv *pcm512x = snd_soc_component_get_drvdata(component);
        unsigned long dac_rate;

        if (!pcm512x->pll_out)
                return 0; /* no PLL to bypass, force SCK as DAC input */

        if (pllin_rate % osr_rate)
                return 0; /* futile, quit early */

        /* run DAC no faster than 6144000 Hz */
        for (dac_rate = rounddown(pcm512x_dac_max(pcm512x, 6144000), osr_rate);
             dac_rate;
             dac_rate -= osr_rate) {

                if (pllin_rate / dac_rate > 128)
                        return 0; /* DAC divider would be too big */

                if (!(pllin_rate % dac_rate))
                        return dac_rate;

                dac_rate -= osr_rate;
        }

        return 0;
}

static int pcm512x_set_dividers(struct snd_soc_dai *dai,
                                struct snd_pcm_hw_params *params)
{
        struct device *dev = dai->dev;
        struct snd_soc_component *component = dai->component;
        struct pcm512x_priv *pcm512x = snd_soc_component_get_drvdata(component);
        unsigned long pllin_rate = 0;
        unsigned long pll_rate;
        unsigned long sck_rate;
        unsigned long mck_rate;
        unsigned long bclk_rate;
        unsigned long sample_rate;
        unsigned long osr_rate;
        unsigned long dacsrc_rate;
        int bclk_div;
        int lrclk_div;
        int dsp_div;
        int dac_div;
        unsigned long dac_rate;
        int ncp_div;
        int osr_div;
        int ret;
        int idac;
        int fssp;
        int gpio;

        if (pcm512x->bclk_ratio > 0) {
                lrclk_div = pcm512x->bclk_ratio;
        } else {
                lrclk_div = snd_soc_params_to_frame_size(params);

                if (lrclk_div == 0) {
                        dev_err(dev, "No LRCLK?\n");
                        return -EINVAL;
                }
        }

        if (!pcm512x->pll_out) {
                sck_rate = clk_get_rate(pcm512x->sclk);
                bclk_rate = params_rate(params) * lrclk_div;
                bclk_div = DIV_ROUND_CLOSEST(sck_rate, bclk_rate);

                mck_rate = sck_rate;
        } else {
                ret = snd_soc_params_to_bclk(params);
                if (ret < 0) {
                        dev_err(dev, "Failed to find suitable BCLK: %d\n", ret);
                        return ret;
                }
                if (ret == 0) {
                        dev_err(dev, "No BCLK?\n");
                        return -EINVAL;
                }
                bclk_rate = ret;

                pllin_rate = clk_get_rate(pcm512x->sclk);

                sck_rate = pcm512x_find_sck(dai, bclk_rate);
                if (!sck_rate)
                        return -EINVAL;
                pll_rate = 4 * sck_rate;

                ret = pcm512x_find_pll_coeff(dai, pllin_rate, pll_rate);
                if (ret != 0)
                        return ret;

                ret = regmap_write(pcm512x->regmap,
                                   PCM512x_PLL_COEFF_0, pcm512x->pll_p - 1);
                if (ret != 0) {
                        dev_err(dev, "Failed to write PLL P: %d\n", ret);
                        return ret;
                }

                ret = regmap_write(pcm512x->regmap,
                                   PCM512x_PLL_COEFF_1, pcm512x->pll_j);
                if (ret != 0) {
                        dev_err(dev, "Failed to write PLL J: %d\n", ret);
                        return ret;
                }

                ret = regmap_write(pcm512x->regmap,
                                   PCM512x_PLL_COEFF_2, pcm512x->pll_d >> 8);
                if (ret != 0) {
                        dev_err(dev, "Failed to write PLL D msb: %d\n", ret);
                        return ret;
                }

                ret = regmap_write(pcm512x->regmap,
                                   PCM512x_PLL_COEFF_3, pcm512x->pll_d & 0xff);
                if (ret != 0) {
                        dev_err(dev, "Failed to write PLL D lsb: %d\n", ret);
                        return ret;
                }

                ret = regmap_write(pcm512x->regmap,
                                   PCM512x_PLL_COEFF_4, pcm512x->pll_r - 1);
                if (ret != 0) {
                        dev_err(dev, "Failed to write PLL R: %d\n", ret);
                        return ret;
                }

                mck_rate = pcm512x->real_pll;

                bclk_div = DIV_ROUND_CLOSEST(sck_rate, bclk_rate);
        }

        if (bclk_div > 128) {
                dev_err(dev, "Failed to find BCLK divider\n");
                return -EINVAL;
        }

        /* the actual rate */
        sample_rate = sck_rate / bclk_div / lrclk_div;
        osr_rate = 16 * sample_rate;

        /* run DSP no faster than 50 MHz */
        dsp_div = mck_rate > pcm512x_dsp_max(pcm512x) ? 2 : 1;

        dac_rate = pcm512x_pllin_dac_rate(dai, osr_rate, pllin_rate);
        if (dac_rate) {
                /* the desired clock rate is "compatible" with the pll input
                 * clock, so use that clock as dac input instead of the pll
                 * output clock since the pll will introduce jitter and thus
                 * noise.
                 */
                dev_dbg(dev, "using pll input as dac input\n");
                ret = regmap_update_bits(pcm512x->regmap, PCM512x_DAC_REF,
                                         PCM512x_SDAC, PCM512x_SDAC_GPIO);
                if (ret != 0) {
                        dev_err(component->dev,
                                "Failed to set gpio as dacref: %d\n", ret);
                        return ret;
                }

                gpio = PCM512x_GREF_GPIO1 + pcm512x->pll_in - 1;
                ret = regmap_update_bits(pcm512x->regmap, PCM512x_GPIO_DACIN,
                                         PCM512x_GREF, gpio);
                if (ret != 0) {
                        dev_err(component->dev,
                                "Failed to set gpio %d as dacin: %d\n",
                                pcm512x->pll_in, ret);
                        return ret;
                }

                dacsrc_rate = pllin_rate;
        } else {
                /* run DAC no faster than 6144000 Hz */
                unsigned long dac_mul = pcm512x_dac_max(pcm512x, 6144000)
                        / osr_rate;
                unsigned long sck_mul = sck_rate / osr_rate;

                for (; dac_mul; dac_mul--) {
                        if (!(sck_mul % dac_mul))
                                break;
                }
                if (!dac_mul) {
                        dev_err(dev, "Failed to find DAC rate\n");
                        return -EINVAL;
                }

                dac_rate = dac_mul * osr_rate;
                dev_dbg(dev, "dac_rate %lu sample_rate %lu\n",
                        dac_rate, sample_rate);

                ret = regmap_update_bits(pcm512x->regmap, PCM512x_DAC_REF,
                                         PCM512x_SDAC, PCM512x_SDAC_SCK);
                if (ret != 0) {
                        dev_err(component->dev,
                                "Failed to set sck as dacref: %d\n", ret);
                        return ret;
                }

                dacsrc_rate = sck_rate;
        }

        osr_div = DIV_ROUND_CLOSEST(dac_rate, osr_rate);
        if (osr_div > 128) {
                dev_err(dev, "Failed to find OSR divider\n");
                return -EINVAL;
        }

        dac_div = DIV_ROUND_CLOSEST(dacsrc_rate, dac_rate);
        if (dac_div > 128) {
                dev_err(dev, "Failed to find DAC divider\n");
                return -EINVAL;
        }
        dac_rate = dacsrc_rate / dac_div;

        ncp_div = DIV_ROUND_CLOSEST(dac_rate,
                                    pcm512x_ncp_target(pcm512x, dac_rate));
        if (ncp_div > 128 || dac_rate / ncp_div > 2048000) {
                /* run NCP no faster than 2048000 Hz, but why? */
                ncp_div = DIV_ROUND_UP(dac_rate, 2048000);
                if (ncp_div > 128) {
                        dev_err(dev, "Failed to find NCP divider\n");
                        return -EINVAL;
                }
        }

        idac = mck_rate / (dsp_div * sample_rate);

        ret = regmap_write(pcm512x->regmap, PCM512x_DSP_CLKDIV, dsp_div - 1);
        if (ret != 0) {
                dev_err(dev, "Failed to write DSP divider: %d\n", ret);
                return ret;
        }

        ret = regmap_write(pcm512x->regmap, PCM512x_DAC_CLKDIV, dac_div - 1);
        if (ret != 0) {
                dev_err(dev, "Failed to write DAC divider: %d\n", ret);
                return ret;
        }

        ret = regmap_write(pcm512x->regmap, PCM512x_NCP_CLKDIV, ncp_div - 1);
        if (ret != 0) {
                dev_err(dev, "Failed to write NCP divider: %d\n", ret);
                return ret;
        }

        ret = regmap_write(pcm512x->regmap, PCM512x_OSR_CLKDIV, osr_div - 1);
        if (ret != 0) {
                dev_err(dev, "Failed to write OSR divider: %d\n", ret);
                return ret;
        }

        ret = regmap_write(pcm512x->regmap,
                           PCM512x_MASTER_CLKDIV_1, bclk_div - 1);
        if (ret != 0) {
                dev_err(dev, "Failed to write BCLK divider: %d\n", ret);
                return ret;
        }

        ret = regmap_write(pcm512x->regmap,
                           PCM512x_MASTER_CLKDIV_2, lrclk_div - 1);
        if (ret != 0) {
                dev_err(dev, "Failed to write LRCLK divider: %d\n", ret);
                return ret;
        }

        ret = regmap_write(pcm512x->regmap, PCM512x_IDAC_1, idac >> 8);
        if (ret != 0) {
                dev_err(dev, "Failed to write IDAC msb divider: %d\n", ret);
                return ret;
        }

        ret = regmap_write(pcm512x->regmap, PCM512x_IDAC_2, idac & 0xff);
        if (ret != 0) {
                dev_err(dev, "Failed to write IDAC lsb divider: %d\n", ret);
                return ret;
        }

        if (sample_rate <= pcm512x_dac_max(pcm512x, 48000))
                fssp = PCM512x_FSSP_48KHZ;
        else if (sample_rate <= pcm512x_dac_max(pcm512x, 96000))
                fssp = PCM512x_FSSP_96KHZ;
        else if (sample_rate <= pcm512x_dac_max(pcm512x, 192000))
                fssp = PCM512x_FSSP_192KHZ;
        else
                fssp = PCM512x_FSSP_384KHZ;
        ret = regmap_update_bits(pcm512x->regmap, PCM512x_FS_SPEED_MODE,
                                 PCM512x_FSSP, fssp);
        if (ret != 0) {
                dev_err(component->dev, "Failed to set fs speed: %d\n", ret);
                return ret;
        }

        dev_dbg(component->dev, "DSP divider %d\n", dsp_div);
        dev_dbg(component->dev, "DAC divider %d\n", dac_div);
        dev_dbg(component->dev, "NCP divider %d\n", ncp_div);
        dev_dbg(component->dev, "OSR divider %d\n", osr_div);
        dev_dbg(component->dev, "BCK divider %d\n", bclk_div);
        dev_dbg(component->dev, "LRCK divider %d\n", lrclk_div);
        dev_dbg(component->dev, "IDAC %d\n", idac);
        dev_dbg(component->dev, "1<<FSSP %d\n", 1 << fssp);

        return 0;
}

static int pcm512x_hw_params(struct snd_pcm_substream *substream,
                             struct snd_pcm_hw_params *params,
                             struct snd_soc_dai *dai)
{
        struct snd_soc_component *component = dai->component;
        struct pcm512x_priv *pcm512x = snd_soc_component_get_drvdata(component);
        int alen;
        int gpio;
        int ret;

        dev_dbg(component->dev, "hw_params %u Hz, %u channels\n",
                params_rate(params),
                params_channels(params));

        switch (params_width(params)) {
        case 16:
                alen = PCM512x_ALEN_16;
                break;
        case 20:
                alen = PCM512x_ALEN_20;
                break;
        case 24:
                alen = PCM512x_ALEN_24;
                break;
        case 32:
                alen = PCM512x_ALEN_32;
                break;
        default:
                dev_err(component->dev, "Bad frame size: %d\n",
                        params_width(params));
                return -EINVAL;
        }

        ret = regmap_update_bits(pcm512x->regmap, PCM512x_I2S_1,
                                 PCM512x_ALEN, alen);
        if (ret != 0) {
                dev_err(component->dev, "Failed to set frame size: %d\n", ret);
                return ret;
        }

        if ((pcm512x->fmt & SND_SOC_DAIFMT_CLOCK_PROVIDER_MASK) ==
            SND_SOC_DAIFMT_CBC_CFC) {
                ret = regmap_update_bits(pcm512x->regmap, PCM512x_ERROR_DETECT,
                                         PCM512x_DCAS, 0);
                if (ret != 0) {
                        dev_err(component->dev,
                                "Failed to enable clock divider autoset: %d\n",
                                ret);
                        return ret;
                }
                goto skip_pll;
        }

        if (pcm512x->pll_out) {
                ret = regmap_write(pcm512x->regmap, PCM512x_FLEX_A, 0x11);
                if (ret != 0) {
                        dev_err(component->dev, "Failed to set FLEX_A: %d\n", ret);
                        return ret;
                }

                ret = regmap_write(pcm512x->regmap, PCM512x_FLEX_B, 0xff);
                if (ret != 0) {
                        dev_err(component->dev, "Failed to set FLEX_B: %d\n", ret);
                        return ret;
                }

                ret = regmap_update_bits(pcm512x->regmap, PCM512x_ERROR_DETECT,
                                         PCM512x_IDFS | PCM512x_IDBK
                                         | PCM512x_IDSK | PCM512x_IDCH
                                         | PCM512x_IDCM | PCM512x_DCAS
                                         | PCM512x_IPLK,
                                         PCM512x_IDFS | PCM512x_IDBK
                                         | PCM512x_IDSK | PCM512x_IDCH
                                         | PCM512x_DCAS);
                if (ret != 0) {
                        dev_err(component->dev,
                                "Failed to ignore auto-clock failures: %d\n",
                                ret);
                        return ret;
                }
        } else {
                ret = regmap_update_bits(pcm512x->regmap, PCM512x_ERROR_DETECT,
                                         PCM512x_IDFS | PCM512x_IDBK
                                         | PCM512x_IDSK | PCM512x_IDCH
                                         | PCM512x_IDCM | PCM512x_DCAS
                                         | PCM512x_IPLK,
                                         PCM512x_IDFS | PCM512x_IDBK
                                         | PCM512x_IDSK | PCM512x_IDCH
                                         | PCM512x_DCAS | PCM512x_IPLK);
                if (ret != 0) {
                        dev_err(component->dev,
                                "Failed to ignore auto-clock failures: %d\n",
                                ret);
                        return ret;
                }

                if (!pcm512x->force_pll_on) {
                        ret = regmap_update_bits(pcm512x->regmap,
                                                 PCM512x_PLL_EN, PCM512x_PLLE, 0);
                } else {
                        /* provide minimum PLL config for TAS575x clocking
                         * and leave PLL enabled
                         */
                        ret = regmap_write(pcm512x->regmap,
                                           PCM512x_PLL_COEFF_0, 0x01);
                        if (ret != 0) {
                                dev_err(component->dev,
                                        "Failed to set pll coefficient: %d\n", ret);
                                return ret;
                        }
                        ret = regmap_write(pcm512x->regmap,
                                           PCM512x_PLL_COEFF_1, 0x04);
                        if (ret != 0) {
                                dev_err(component->dev,
                                        "Failed to set pll coefficient: %d\n", ret);
                                return ret;
                        }
                        ret = regmap_write(pcm512x->regmap,
                                           PCM512x_PLL_EN, 0x01);
                        dev_dbg(component->dev, "Enabling PLL for TAS575x\n");
                }

                if (ret != 0) {
                        dev_err(component->dev, "Failed to set pll mode: %d\n", ret);
                        return ret;
                }
        }

        ret = pcm512x_set_dividers(dai, params);
        if (ret != 0)
                return ret;

        if (pcm512x->pll_out) {
                ret = regmap_update_bits(pcm512x->regmap, PCM512x_PLL_REF,
                                         PCM512x_SREF, PCM512x_SREF_GPIO);
                if (ret != 0) {
                        dev_err(component->dev,
                                "Failed to set gpio as pllref: %d\n", ret);
                        return ret;
                }

                gpio = PCM512x_GREF_GPIO1 + pcm512x->pll_in - 1;
                ret = regmap_update_bits(pcm512x->regmap, PCM512x_GPIO_PLLIN,
                                         PCM512x_GREF, gpio);
                if (ret != 0) {
                        dev_err(component->dev,
                                "Failed to set gpio %d as pllin: %d\n",
                                pcm512x->pll_in, ret);
                        return ret;
                }

                ret = regmap_update_bits(pcm512x->regmap, PCM512x_PLL_EN,
                                         PCM512x_PLLE, PCM512x_PLLE);
                if (ret != 0) {
                        dev_err(component->dev, "Failed to enable pll: %d\n", ret);
                        return ret;
                }

                gpio = PCM512x_G1OE << (pcm512x->pll_out - 1);
                ret = regmap_update_bits(pcm512x->regmap, PCM512x_GPIO_EN,
                                         gpio, gpio);
                if (ret != 0) {
                        dev_err(component->dev, "Failed to enable gpio %d: %d\n",
                                pcm512x->pll_out, ret);
                        return ret;
                }

                gpio = PCM512x_GPIO_OUTPUT_1 + pcm512x->pll_out - 1;
                ret = regmap_update_bits(pcm512x->regmap, gpio,
                                         PCM512x_GxSL, PCM512x_GxSL_PLLCK);
                if (ret != 0) {
                        dev_err(component->dev, "Failed to output pll on %d: %d\n",
                                ret, pcm512x->pll_out);
                        return ret;
                }
        }

        ret = regmap_update_bits(pcm512x->regmap, PCM512x_SYNCHRONIZE,
                                 PCM512x_RQSY, PCM512x_RQSY_HALT);
        if (ret != 0) {
                dev_err(component->dev, "Failed to halt clocks: %d\n", ret);
                return ret;
        }

        ret = regmap_update_bits(pcm512x->regmap, PCM512x_SYNCHRONIZE,
                                 PCM512x_RQSY, PCM512x_RQSY_RESUME);
        if (ret != 0) {
                dev_err(component->dev, "Failed to resume clocks: %d\n", ret);
                return ret;
        }

skip_pll:
        return 0;
}

static int pcm512x_set_fmt(struct snd_soc_dai *dai, unsigned int fmt)
{
        struct snd_soc_component *component = dai->component;
        struct pcm512x_priv *pcm512x = snd_soc_component_get_drvdata(component);
        int afmt;
        int offset = 0;
        int clock_output;
        int provider_mode;
        int ret;

        switch (fmt & SND_SOC_DAIFMT_CLOCK_PROVIDER_MASK) {
        case SND_SOC_DAIFMT_CBC_CFC:
                clock_output = 0;
                provider_mode = 0;
                break;
        case SND_SOC_DAIFMT_CBP_CFP:
                clock_output = PCM512x_BCKO | PCM512x_LRKO;
                provider_mode = PCM512x_RLRK | PCM512x_RBCK;
                break;
        case SND_SOC_DAIFMT_CBP_CFC:
                clock_output = PCM512x_BCKO;
                provider_mode = PCM512x_RBCK;
                break;
        default:
                return -EINVAL;
        }

        ret = regmap_update_bits(pcm512x->regmap, PCM512x_BCLK_LRCLK_CFG,
                                 PCM512x_BCKP | PCM512x_BCKO | PCM512x_LRKO,
                                 clock_output);
        if (ret != 0) {
                dev_err(component->dev, "Failed to enable clock output: %d\n", ret);
                return ret;
        }

        ret = regmap_update_bits(pcm512x->regmap, PCM512x_MASTER_MODE,
                                 PCM512x_RLRK | PCM512x_RBCK,
                                 provider_mode);
        if (ret != 0) {
                dev_err(component->dev, "Failed to enable provider mode: %d\n", ret);
                return ret;
        }

        switch (fmt & SND_SOC_DAIFMT_FORMAT_MASK) {
        case SND_SOC_DAIFMT_I2S:
                afmt = PCM512x_AFMT_I2S;
                break;
        case SND_SOC_DAIFMT_RIGHT_J:
                afmt = PCM512x_AFMT_RTJ;
                break;
        case SND_SOC_DAIFMT_LEFT_J:
                afmt = PCM512x_AFMT_LTJ;
                break;
        case SND_SOC_DAIFMT_DSP_A:
                offset = 1;
                fallthrough;
        case SND_SOC_DAIFMT_DSP_B:
                afmt = PCM512x_AFMT_DSP;
                break;
        default:
                dev_err(component->dev, "unsupported DAI format: 0x%x\n",
                        pcm512x->fmt);
                return -EINVAL;
        }

        ret = regmap_update_bits(pcm512x->regmap, PCM512x_I2S_1,
                                 PCM512x_AFMT, afmt);
        if (ret != 0) {
                dev_err(component->dev, "Failed to set data format: %d\n", ret);
                return ret;
        }

        ret = regmap_update_bits(pcm512x->regmap, PCM512x_I2S_2,
                                 0xFF, offset);
        if (ret != 0) {
                dev_err(component->dev, "Failed to set data offset: %d\n", ret);
                return ret;
        }

        pcm512x->fmt = fmt;

        return 0;
}

static int pcm512x_set_bclk_ratio(struct snd_soc_dai *dai, unsigned int ratio)
{
        struct snd_soc_component *component = dai->component;
        struct pcm512x_priv *pcm512x = snd_soc_component_get_drvdata(component);

        if (ratio > 256)
                return -EINVAL;

        pcm512x->bclk_ratio = ratio;

        return 0;
}

static int pcm512x_mute(struct snd_soc_dai *dai, int mute, int direction)
{
        struct snd_soc_component *component = dai->component;
        struct pcm512x_priv *pcm512x = snd_soc_component_get_drvdata(component);
        int ret;
        unsigned int mute_det;

        mutex_lock(&pcm512x->mutex);

        if (mute) {
                pcm512x->mute |= 0x1;
                ret = regmap_update_bits(pcm512x->regmap, PCM512x_MUTE,
                                         PCM512x_RQML | PCM512x_RQMR,
                                         PCM512x_RQML | PCM512x_RQMR);
                if (ret != 0) {
                        dev_err(component->dev,
                                "Failed to set digital mute: %d\n", ret);
                        goto unlock;
                }

                regmap_read_poll_timeout(pcm512x->regmap,
                                         PCM512x_ANALOG_MUTE_DET,
                                         mute_det, (mute_det & 0x3) == 0,
                                         200, 10000);
        } else {
                pcm512x->mute &= ~0x1;
                ret = pcm512x_update_mute(pcm512x);
                if (ret != 0) {
                        dev_err(component->dev,
                                "Failed to update digital mute: %d\n", ret);
                        goto unlock;
                }

                regmap_read_poll_timeout(pcm512x->regmap,
                                         PCM512x_ANALOG_MUTE_DET,
                                         mute_det,
                                         (mute_det & 0x3)
                                         == ((~pcm512x->mute >> 1) & 0x3),
                                         200, 10000);
        }

unlock:
        mutex_unlock(&pcm512x->mutex);

        return ret;
}

static const struct snd_soc_dai_ops pcm512x_dai_ops = {
        .startup = pcm512x_dai_startup,
        .hw_params = pcm512x_hw_params,
        .set_fmt = pcm512x_set_fmt,
        .mute_stream = pcm512x_mute,
        .set_bclk_ratio = pcm512x_set_bclk_ratio,
        .no_capture_mute = 1,
};

static struct snd_soc_dai_driver pcm512x_dai = {
        .name = "pcm512x-hifi",
        .playback = {
                .stream_name = "Playback",
                .channels_min = 2,
                .channels_max = 2,
                .rates = SNDRV_PCM_RATE_CONTINUOUS,
                .rate_min = 8000,
                .rate_max = 384000,
                .formats = SNDRV_PCM_FMTBIT_S16_LE |
                           SNDRV_PCM_FMTBIT_S24_LE |
                           SNDRV_PCM_FMTBIT_S32_LE
        },
        .ops = &pcm512x_dai_ops,
};

static const struct snd_soc_component_driver pcm512x_component_driver = {
        .set_bias_level         = pcm512x_set_bias_level,
        .controls               = pcm512x_controls,
        .num_controls           = ARRAY_SIZE(pcm512x_controls),
        .dapm_widgets           = pcm512x_dapm_widgets,
        .num_dapm_widgets       = ARRAY_SIZE(pcm512x_dapm_widgets),
        .dapm_routes            = pcm512x_dapm_routes,
        .num_dapm_routes        = ARRAY_SIZE(pcm512x_dapm_routes),
        .use_pmdown_time        = 1,
        .endianness             = 1,
};

static const struct regmap_range_cfg pcm512x_range = {
        .name = "Pages", .range_min = PCM512x_VIRT_BASE,
        .range_max = PCM512x_MAX_REGISTER,
        .selector_reg = PCM512x_PAGE,
        .selector_mask = 0xff,
        .window_start = 0, .window_len = 0x100,
};

const struct regmap_config pcm512x_regmap = {
        .reg_bits = 8,
        .val_bits = 8,

        .readable_reg = pcm512x_readable,
        .volatile_reg = pcm512x_volatile,

        .ranges = &pcm512x_range,
        .num_ranges = 1,

        .max_register = PCM512x_MAX_REGISTER,
        .reg_defaults = pcm512x_reg_defaults,
        .num_reg_defaults = ARRAY_SIZE(pcm512x_reg_defaults),
        .cache_type = REGCACHE_RBTREE,
};
EXPORT_SYMBOL_GPL(pcm512x_regmap);

int pcm512x_probe(struct device *dev, struct regmap *regmap)
{
        struct pcm512x_priv *pcm512x;
        int i, ret;

        pcm512x = devm_kzalloc(dev, sizeof(struct pcm512x_priv), GFP_KERNEL);
        if (!pcm512x)
                return -ENOMEM;

        mutex_init(&pcm512x->mutex);

        dev_set_drvdata(dev, pcm512x);
        pcm512x->regmap = regmap;

        for (i = 0; i < ARRAY_SIZE(pcm512x->supplies); i++)
                pcm512x->supplies[i].supply = pcm512x_supply_names[i];

        ret = devm_regulator_bulk_get(dev, ARRAY_SIZE(pcm512x->supplies),
                                      pcm512x->supplies);
        if (ret != 0) {
                dev_err(dev, "Failed to get supplies: %d\n", ret);
                return ret;
        }

        pcm512x->supply_nb[0].notifier_call = pcm512x_regulator_event_0;
        pcm512x->supply_nb[1].notifier_call = pcm512x_regulator_event_1;
        pcm512x->supply_nb[2].notifier_call = pcm512x_regulator_event_2;

        for (i = 0; i < ARRAY_SIZE(pcm512x->supplies); i++) {
                ret = devm_regulator_register_notifier(
                                                pcm512x->supplies[i].consumer,
                                                &pcm512x->supply_nb[i]);
                if (ret != 0) {
                        dev_err(dev,
                                "Failed to register regulator notifier: %d\n",
                                ret);
                }
        }

        ret = regulator_bulk_enable(ARRAY_SIZE(pcm512x->supplies),
                                    pcm512x->supplies);
        if (ret != 0) {
                dev_err(dev, "Failed to enable supplies: %d\n", ret);
                return ret;
        }

        /* Reset the device, verifying I/O in the process for I2C */
        ret = regmap_write(regmap, PCM512x_RESET,
                           PCM512x_RSTM | PCM512x_RSTR);
        if (ret != 0) {
                dev_err(dev, "Failed to reset device: %d\n", ret);
                goto err;
        }

        ret = regmap_write(regmap, PCM512x_RESET, 0);
        if (ret != 0) {
                dev_err(dev, "Failed to reset device: %d\n", ret);
                goto err;
        }

        pcm512x->sclk = devm_clk_get(dev, NULL);
        if (PTR_ERR(pcm512x->sclk) == -EPROBE_DEFER) {
                ret = -EPROBE_DEFER;
                goto err;
        }
        if (!IS_ERR(pcm512x->sclk)) {
                ret = clk_prepare_enable(pcm512x->sclk);
                if (ret != 0) {
                        dev_err(dev, "Failed to enable SCLK: %d\n", ret);
                        goto err;
                }
        }

        /* Default to standby mode */
        ret = regmap_update_bits(pcm512x->regmap, PCM512x_POWER,
                                 PCM512x_RQST, PCM512x_RQST);
        if (ret != 0) {
                dev_err(dev, "Failed to request standby: %d\n",
                        ret);
                goto err_clk;
        }

        pm_runtime_set_active(dev);
        pm_runtime_enable(dev);
        pm_runtime_idle(dev);

#ifdef CONFIG_OF
        if (dev->of_node) {
                const struct device_node *np = dev->of_node;
                u32 val;

                if (of_property_read_u32(np, "pll-in", &val) >= 0) {
                        if (val > 6) {
                                dev_err(dev, "Invalid pll-in\n");
                                ret = -EINVAL;
                                goto err_pm;
                        }
                        pcm512x->pll_in = val;
                }

                if (of_property_read_u32(np, "pll-out", &val) >= 0) {
                        if (val > 6) {
                                dev_err(dev, "Invalid pll-out\n");
                                ret = -EINVAL;
                                goto err_pm;
                        }
                        pcm512x->pll_out = val;
                }

                if (!pcm512x->pll_in != !pcm512x->pll_out) {
                        dev_err(dev,
                                "Error: both pll-in and pll-out, or none\n");
                        ret = -EINVAL;
                        goto err_pm;
                }
                if (pcm512x->pll_in && pcm512x->pll_in == pcm512x->pll_out) {
                        dev_err(dev, "Error: pll-in == pll-out\n");
                        ret = -EINVAL;
                        goto err_pm;
                }

                if (!strcmp(np->name, "tas5756") ||
                    !strcmp(np->name, "tas5754"))
                        pcm512x->force_pll_on = 1;
                dev_dbg(dev, "Device ID: %s\n", np->name);
        }
#endif

        ret = devm_snd_soc_register_component(dev, &pcm512x_component_driver,
                                    &pcm512x_dai, 1);
        if (ret != 0) {
                dev_err(dev, "Failed to register CODEC: %d\n", ret);
                goto err_pm;
        }

        return 0;

err_pm:
        pm_runtime_disable(dev);
err_clk:
        if (!IS_ERR(pcm512x->sclk))
                clk_disable_unprepare(pcm512x->sclk);
err:
        regulator_bulk_disable(ARRAY_SIZE(pcm512x->supplies),
                                     pcm512x->supplies);
        return ret;
}
EXPORT_SYMBOL_GPL(pcm512x_probe);

void pcm512x_remove(struct device *dev)
{
        struct pcm512x_priv *pcm512x = dev_get_drvdata(dev);

        pm_runtime_disable(dev);
        if (!IS_ERR(pcm512x->sclk))
                clk_disable_unprepare(pcm512x->sclk);
        regulator_bulk_disable(ARRAY_SIZE(pcm512x->supplies),
                               pcm512x->supplies);
}
EXPORT_SYMBOL_GPL(pcm512x_remove);

static int pcm512x_suspend(struct device *dev)
{
        struct pcm512x_priv *pcm512x = dev_get_drvdata(dev);
        int ret;

        ret = regmap_update_bits(pcm512x->regmap, PCM512x_POWER,
                                 PCM512x_RQPD, PCM512x_RQPD);
        if (ret != 0) {
                dev_err(dev, "Failed to request power down: %d\n", ret);
                return ret;
        }

        ret = regulator_bulk_disable(ARRAY_SIZE(pcm512x->supplies),
                                     pcm512x->supplies);
        if (ret != 0) {
                dev_err(dev, "Failed to disable supplies: %d\n", ret);
                return ret;
        }

        if (!IS_ERR(pcm512x->sclk))
                clk_disable_unprepare(pcm512x->sclk);

        return 0;
}

static int pcm512x_resume(struct device *dev)
{
        struct pcm512x_priv *pcm512x = dev_get_drvdata(dev);
        int ret;

        if (!IS_ERR(pcm512x->sclk)) {
                ret = clk_prepare_enable(pcm512x->sclk);
                if (ret != 0) {
                        dev_err(dev, "Failed to enable SCLK: %d\n", ret);
                        return ret;
                }
        }

        ret = regulator_bulk_enable(ARRAY_SIZE(pcm512x->supplies),
                                    pcm512x->supplies);
        if (ret != 0) {
                dev_err(dev, "Failed to enable supplies: %d\n", ret);
                return ret;
        }

        regcache_cache_only(pcm512x->regmap, false);
        ret = regcache_sync(pcm512x->regmap);
        if (ret != 0) {
                dev_err(dev, "Failed to sync cache: %d\n", ret);
                return ret;
        }

        ret = regmap_update_bits(pcm512x->regmap, PCM512x_POWER,
                                 PCM512x_RQPD, 0);
        if (ret != 0) {
                dev_err(dev, "Failed to remove power down: %d\n", ret);
                return ret;
        }

        return 0;
}

EXPORT_GPL_DEV_PM_OPS(pcm512x_pm_ops) = {
        RUNTIME_PM_OPS(pcm512x_suspend, pcm512x_resume, NULL)
};

MODULE_DESCRIPTION("ASoC PCM512x codec driver");
MODULE_AUTHOR("Mark Brown <broonie@kernel.org>");
MODULE_LICENSE("GPL v2");