root/sound/soc/codecs/tas2764.c
// SPDX-License-Identifier: GPL-2.0
//
// Driver for the Texas Instruments TAS2764 CODEC
// Copyright (C) 2020 Texas Instruments Inc.

#include <linux/module.h>
#include <linux/moduleparam.h>
#include <linux/err.h>
#include <linux/init.h>
#include <linux/delay.h>
#include <linux/hwmon.h>
#include <linux/pm.h>
#include <linux/i2c.h>
#include <linux/gpio/consumer.h>
#include <linux/regulator/consumer.h>
#include <linux/regmap.h>
#include <linux/of.h>
#include <linux/of_device.h>
#include <linux/slab.h>
#include <sound/soc.h>
#include <sound/pcm.h>
#include <sound/pcm_params.h>
#include <sound/initval.h>
#include <sound/tlv.h>

#include "tas2764.h"

enum tas2764_devid {
        DEVID_TAS2764  = 0,
        DEVID_SN012776 = 1
};

struct tas2764_priv {
        struct snd_soc_component *component;
        struct gpio_desc *reset_gpio;
        struct gpio_desc *sdz_gpio;
        struct regmap *regmap;
        struct device *dev;
        int irq;
        enum tas2764_devid devid;

        int v_sense_slot;
        int i_sense_slot;

        bool dac_powered;
        bool unmuted;
};

#include "tas2764-quirks.h"

static const char *tas2764_int_ltch0_msgs[8] = {
        "fault: over temperature", /* INT_LTCH0 & BIT(0) */
        "fault: over current",
        "fault: bad TDM clock",
        "limiter active",
        "fault: PVDD below limiter inflection point",
        "fault: limiter max attenuation",
        "fault: BOP infinite hold",
        "fault: BOP mute", /* INT_LTCH0 & BIT(7) */
};

static const unsigned int tas2764_int_readout_regs[6] = {
        TAS2764_INT_LTCH0,
        TAS2764_INT_LTCH1,
        TAS2764_INT_LTCH1_0,
        TAS2764_INT_LTCH2,
        TAS2764_INT_LTCH3,
        TAS2764_INT_LTCH4,
};

static irqreturn_t tas2764_irq(int irq, void *data)
{
        struct tas2764_priv *tas2764 = data;
        u8 latched[6] = {0, 0, 0, 0, 0, 0};
        int ret = IRQ_NONE;
        int i;

        for (i = 0; i < ARRAY_SIZE(latched); i++)
                latched[i] = snd_soc_component_read(tas2764->component,
                                                    tas2764_int_readout_regs[i]);

        for (i = 0; i < 8; i++) {
                if (latched[0] & BIT(i)) {
                        dev_crit_ratelimited(tas2764->dev, "%s\n",
                                             tas2764_int_ltch0_msgs[i]);
                        ret = IRQ_HANDLED;
                }
        }

        if (latched[0]) {
                dev_err_ratelimited(tas2764->dev, "other context to the fault: %02x,%02x,%02x,%02x,%02x",
                                    latched[1], latched[2], latched[3], latched[4], latched[5]);
                snd_soc_component_update_bits(tas2764->component,
                                              TAS2764_INT_CLK_CFG,
                                              TAS2764_INT_CLK_CFG_IRQZ_CLR,
                                              TAS2764_INT_CLK_CFG_IRQZ_CLR);
        }

        return ret;
}

static void tas2764_reset(struct tas2764_priv *tas2764)
{
        if (tas2764->reset_gpio) {
                gpiod_set_value_cansleep(tas2764->reset_gpio, 0);
                msleep(20);
                gpiod_set_value_cansleep(tas2764->reset_gpio, 1);
                usleep_range(1000, 2000);
        }

        snd_soc_component_write(tas2764->component, TAS2764_SW_RST,
                                TAS2764_RST);
        usleep_range(1000, 2000);
}

static int tas2764_update_pwr_ctrl(struct tas2764_priv *tas2764)
{
        struct snd_soc_component *component = tas2764->component;
        unsigned int val;
        int ret;

        if (tas2764->dac_powered)
                val = tas2764->unmuted ?
                        TAS2764_PWR_CTRL_ACTIVE : TAS2764_PWR_CTRL_MUTE;
        else
                val = TAS2764_PWR_CTRL_SHUTDOWN;

        if (ENABLED_APPLE_QUIRKS & TAS2764_SHUTDOWN_DANCE)
                return tas2764_do_quirky_pwr_ctrl_change(tas2764, val);

        ret = snd_soc_component_update_bits(component, TAS2764_PWR_CTRL,
                                            TAS2764_PWR_CTRL_MASK, val);
        if (ret < 0)
                return ret;

        return 0;
}

#ifdef CONFIG_PM
static int tas2764_codec_suspend(struct snd_soc_component *component)
{
        struct tas2764_priv *tas2764 = snd_soc_component_get_drvdata(component);
        int ret;

        ret = snd_soc_component_update_bits(component, TAS2764_PWR_CTRL,
                                            TAS2764_PWR_CTRL_MASK,
                                            TAS2764_PWR_CTRL_SHUTDOWN);

        if (ret < 0)
                return ret;

        if (tas2764->sdz_gpio)
                gpiod_set_value_cansleep(tas2764->sdz_gpio, 0);

        regcache_cache_only(tas2764->regmap, true);
        regcache_mark_dirty(tas2764->regmap);

        usleep_range(6000, 7000);

        return 0;
}

static int tas2764_codec_resume(struct snd_soc_component *component)
{
        struct tas2764_priv *tas2764 = snd_soc_component_get_drvdata(component);
        int ret;

        if (tas2764->sdz_gpio) {
                gpiod_set_value_cansleep(tas2764->sdz_gpio, 1);
                usleep_range(1000, 2000);
        }

        ret = tas2764_update_pwr_ctrl(tas2764);

        if (ret < 0)
                return ret;

        regcache_cache_only(tas2764->regmap, false);

        return regcache_sync(tas2764->regmap);
}
#else
#define tas2764_codec_suspend NULL
#define tas2764_codec_resume NULL
#endif

static const char * const tas2764_ASI1_src[] = {
        "I2C offset", "Left", "Right", "LeftRightDiv2",
};

static SOC_ENUM_SINGLE_DECL(
        tas2764_ASI1_src_enum, TAS2764_TDM_CFG2, TAS2764_TDM_CFG2_SCFG_SHIFT,
        tas2764_ASI1_src);

static const struct snd_kcontrol_new tas2764_asi1_mux =
        SOC_DAPM_ENUM("ASI1 Source", tas2764_ASI1_src_enum);

static const struct snd_kcontrol_new isense_switch =
        SOC_DAPM_SINGLE("Switch", TAS2764_PWR_CTRL, TAS2764_ISENSE_POWER_EN, 1, 1);
static const struct snd_kcontrol_new vsense_switch =
        SOC_DAPM_SINGLE("Switch", TAS2764_PWR_CTRL, TAS2764_VSENSE_POWER_EN, 1, 1);

static const struct snd_soc_dapm_widget tas2764_dapm_widgets[] = {
        SND_SOC_DAPM_AIF_IN("ASI1", "ASI1 Playback", 0, SND_SOC_NOPM, 0, 0),
        SND_SOC_DAPM_MUX("ASI1 Sel", SND_SOC_NOPM, 0, 0, &tas2764_asi1_mux),
        SND_SOC_DAPM_SWITCH("ISENSE", TAS2764_PWR_CTRL, TAS2764_ISENSE_POWER_EN,
                            1, &isense_switch),
        SND_SOC_DAPM_SWITCH("VSENSE", TAS2764_PWR_CTRL, TAS2764_VSENSE_POWER_EN,
                            1, &vsense_switch),
        SND_SOC_DAPM_DAC("DAC", NULL, SND_SOC_NOPM, 0, 0),
        SND_SOC_DAPM_OUTPUT("OUT"),
        SND_SOC_DAPM_SIGGEN("VMON"),
        SND_SOC_DAPM_SIGGEN("IMON")
};

static const struct snd_soc_dapm_route tas2764_audio_map[] = {
        {"ASI1 Sel", "I2C offset", "ASI1"},
        {"ASI1 Sel", "Left", "ASI1"},
        {"ASI1 Sel", "Right", "ASI1"},
        {"ASI1 Sel", "LeftRightDiv2", "ASI1"},
        {"DAC", NULL, "ASI1 Sel"},
        {"OUT", NULL, "DAC"},
        {"ISENSE", "Switch", "IMON"},
        {"VSENSE", "Switch", "VMON"},
};

static int tas2764_mute(struct snd_soc_dai *dai, int mute, int direction)
{
        struct tas2764_priv *tas2764 =
                        snd_soc_component_get_drvdata(dai->component);
        int ret;

        if (!mute) {
                tas2764->dac_powered = true;
                ret = tas2764_update_pwr_ctrl(tas2764);
                if (ret)
                        return ret;
        }

        tas2764->unmuted = !mute;
        ret = tas2764_update_pwr_ctrl(tas2764);
        if (ret)
                return ret;

        if (mute) {
                /* Wait for ramp-down */
                usleep_range(6000, 7000);

                tas2764->dac_powered = false;
                ret = tas2764_update_pwr_ctrl(tas2764);
                if (ret)
                        return ret;

                /* Wait a bit after shutdown */
                usleep_range(2000, 3000);
        }

        return 0;
}

static int tas2764_set_bitwidth(struct tas2764_priv *tas2764, int bitwidth)
{
        struct snd_soc_component *component = tas2764->component;
        int sense_en;
        int val;
        int ret;

        switch (bitwidth) {
        case SNDRV_PCM_FORMAT_S16_LE:
                ret = snd_soc_component_update_bits(component,
                                                    TAS2764_TDM_CFG2,
                                                    TAS2764_TDM_CFG2_RXW_MASK,
                                                    TAS2764_TDM_CFG2_RXW_16BITS);
                break;
        case SNDRV_PCM_FORMAT_S24_LE:
                ret = snd_soc_component_update_bits(component,
                                                    TAS2764_TDM_CFG2,
                                                    TAS2764_TDM_CFG2_RXW_MASK,
                                                    TAS2764_TDM_CFG2_RXW_24BITS);
                break;
        case SNDRV_PCM_FORMAT_S32_LE:
                ret = snd_soc_component_update_bits(component,
                                                    TAS2764_TDM_CFG2,
                                                    TAS2764_TDM_CFG2_RXW_MASK,
                                                    TAS2764_TDM_CFG2_RXW_32BITS);
                break;

        default:
                return -EINVAL;
        }

        if (ret < 0)
                return ret;

        val = snd_soc_component_read(tas2764->component, TAS2764_PWR_CTRL);
        if (val < 0)
                return val;

        if (val & (1 << TAS2764_VSENSE_POWER_EN))
                sense_en = 0;
        else
                sense_en = TAS2764_TDM_CFG5_VSNS_ENABLE;

        ret = snd_soc_component_update_bits(tas2764->component, TAS2764_TDM_CFG5,
                                            TAS2764_TDM_CFG5_VSNS_ENABLE,
                                            sense_en);
        if (ret < 0)
                return ret;

        if (val & (1 << TAS2764_ISENSE_POWER_EN))
                sense_en = 0;
        else
                sense_en = TAS2764_TDM_CFG6_ISNS_ENABLE;

        ret = snd_soc_component_update_bits(tas2764->component, TAS2764_TDM_CFG6,
                                            TAS2764_TDM_CFG6_ISNS_ENABLE,
                                            sense_en);
        if (ret < 0)
                return ret;

        return 0;
}

static int tas2764_set_samplerate(struct tas2764_priv *tas2764, int samplerate)
{
        struct snd_soc_component *component = tas2764->component;
        int ramp_rate_val;
        int ret;

        switch (samplerate) {
        case 48000:
                ramp_rate_val = TAS2764_TDM_CFG0_SMP_48KHZ |
                                TAS2764_TDM_CFG0_44_1_48KHZ;
                break;
        case 44100:
                ramp_rate_val = TAS2764_TDM_CFG0_SMP_44_1KHZ |
                                TAS2764_TDM_CFG0_44_1_48KHZ;
                break;
        case 96000:
                ramp_rate_val = TAS2764_TDM_CFG0_SMP_48KHZ |
                                TAS2764_TDM_CFG0_88_2_96KHZ;
                break;
        case 88200:
                ramp_rate_val = TAS2764_TDM_CFG0_SMP_44_1KHZ |
                                TAS2764_TDM_CFG0_88_2_96KHZ;
                break;
        default:
                return -EINVAL;
        }

        ret = snd_soc_component_update_bits(component, TAS2764_TDM_CFG0,
                                            TAS2764_TDM_CFG0_SMP_MASK |
                                            TAS2764_TDM_CFG0_MASK,
                                            ramp_rate_val);
        if (ret < 0)
                return ret;

        return 0;
}

static int tas2764_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 tas2764_priv *tas2764 = snd_soc_component_get_drvdata(component);
        int ret;

        ret = tas2764_set_bitwidth(tas2764, params_format(params));
        if (ret < 0)
                return ret;

        return tas2764_set_samplerate(tas2764, params_rate(params));
}

static int tas2764_set_fmt(struct snd_soc_dai *dai, unsigned int fmt)
{
        struct snd_soc_component *component = dai->component;
        struct tas2764_priv *tas2764 = snd_soc_component_get_drvdata(component);
        u8 tdm_rx_start_slot = 0, asi_cfg_0 = 0, asi_cfg_1 = 0, asi_cfg_4 = 0;
        int ret;

        switch (fmt & SND_SOC_DAIFMT_INV_MASK) {
        case SND_SOC_DAIFMT_NB_IF:
                asi_cfg_0 ^= TAS2764_TDM_CFG0_FRAME_START;
                fallthrough;
        case SND_SOC_DAIFMT_NB_NF:
                asi_cfg_1 = TAS2764_TDM_CFG1_RX_RISING;
                asi_cfg_4 = TAS2764_TDM_CFG4_TX_FALLING;
                break;
        case SND_SOC_DAIFMT_IB_IF:
                asi_cfg_0 ^= TAS2764_TDM_CFG0_FRAME_START;
                fallthrough;
        case SND_SOC_DAIFMT_IB_NF:
                asi_cfg_1 = TAS2764_TDM_CFG1_RX_FALLING;
                asi_cfg_4 = TAS2764_TDM_CFG4_TX_RISING;
                break;
        }

        ret = snd_soc_component_update_bits(component, TAS2764_TDM_CFG1,
                                            TAS2764_TDM_CFG1_RX_MASK,
                                            asi_cfg_1);
        if (ret < 0)
                return ret;

        ret = snd_soc_component_update_bits(component, TAS2764_TDM_CFG4,
                                            TAS2764_TDM_CFG4_TX_MASK,
                                            asi_cfg_4);
        if (ret < 0)
                return ret;

        switch (fmt & SND_SOC_DAIFMT_FORMAT_MASK) {
        case SND_SOC_DAIFMT_I2S:
                asi_cfg_0 ^= TAS2764_TDM_CFG0_FRAME_START;
                fallthrough;
        case SND_SOC_DAIFMT_DSP_A:
                tdm_rx_start_slot = 1;
                break;
        case SND_SOC_DAIFMT_DSP_B:
        case SND_SOC_DAIFMT_LEFT_J:
                tdm_rx_start_slot = 0;
                break;
        default:
                dev_err(tas2764->dev,
                        "DAI Format is not found, fmt=0x%x\n", fmt);
                return -EINVAL;
        }

        ret = snd_soc_component_update_bits(component, TAS2764_TDM_CFG0,
                                            TAS2764_TDM_CFG0_FRAME_START,
                                            asi_cfg_0);
        if (ret < 0)
                return ret;

        ret = snd_soc_component_update_bits(component, TAS2764_TDM_CFG1,
                                            TAS2764_TDM_CFG1_MASK,
                                            (tdm_rx_start_slot << TAS2764_TDM_CFG1_51_SHIFT));
        if (ret < 0)
                return ret;

        return 0;
}

static int tas2764_set_dai_tdm_slot(struct snd_soc_dai *dai,
                                unsigned int tx_mask,
                                unsigned int rx_mask,
                                int slots, int slot_width)
{
        struct snd_soc_component *component = dai->component;
        struct tas2764_priv *tas2764 = snd_soc_component_get_drvdata(component);
        int left_slot, right_slot;
        int slots_cfg;
        int slot_size;
        int ret;

        if (tx_mask == 0 || rx_mask != 0)
                return -EINVAL;

        left_slot = __ffs(tx_mask);
        tx_mask &= ~(1 << left_slot);
        if (tx_mask == 0) {
                right_slot = left_slot;
        } else {
                right_slot = __ffs(tx_mask);
                tx_mask &= ~(1 << right_slot);
        }

        if (tx_mask != 0 || left_slot >= slots || right_slot >= slots)
                return -EINVAL;

        slots_cfg = (right_slot << TAS2764_TDM_CFG3_RXS_SHIFT) | left_slot;

        ret = snd_soc_component_write(component, TAS2764_TDM_CFG3, slots_cfg);
        if (ret)
                return ret;

        switch (slot_width) {
        case 16:
                slot_size = TAS2764_TDM_CFG2_RXS_16BITS;
                break;
        case 24:
                slot_size = TAS2764_TDM_CFG2_RXS_24BITS;
                break;
        case 32:
                slot_size = TAS2764_TDM_CFG2_RXS_32BITS;
                break;
        default:
                return -EINVAL;
        }

        ret = snd_soc_component_update_bits(component, TAS2764_TDM_CFG2,
                                            TAS2764_TDM_CFG2_RXS_MASK,
                                            slot_size);
        if (ret < 0)
                return ret;

        ret = snd_soc_component_update_bits(component, TAS2764_TDM_CFG5,
                                            TAS2764_TDM_CFG5_50_MASK,
                                            tas2764->v_sense_slot);
        if (ret < 0)
                return ret;

        ret = snd_soc_component_update_bits(component, TAS2764_TDM_CFG6,
                                            TAS2764_TDM_CFG6_50_MASK,
                                            tas2764->i_sense_slot);
        if (ret < 0)
                return ret;

        return 0;
}

static const struct snd_soc_dai_ops tas2764_dai_ops = {
        .mute_stream = tas2764_mute,
        .hw_params  = tas2764_hw_params,
        .set_fmt    = tas2764_set_fmt,
        .set_tdm_slot = tas2764_set_dai_tdm_slot,
        .no_capture_mute = 1,
};

#define TAS2764_FORMATS (SNDRV_PCM_FMTBIT_S16_LE | SNDRV_PCM_FMTBIT_S20_3LE |\
                         SNDRV_PCM_FMTBIT_S24_LE | SNDRV_PCM_FMTBIT_S32_LE)

#define TAS2764_RATES (SNDRV_PCM_RATE_44100 | SNDRV_PCM_RATE_48000 |\
                       SNDRV_PCM_RATE_96000 | SNDRV_PCM_RATE_88200)

static struct snd_soc_dai_driver tas2764_dai_driver[] = {
        {
                .name = "tas2764 ASI1",
                .id = 0,
                .playback = {
                        .stream_name    = "ASI1 Playback",
                        .channels_min   = 1,
                        .channels_max   = 2,
                        .rates      = TAS2764_RATES,
                        .formats    = TAS2764_FORMATS,
                },
                .capture = {
                        .stream_name    = "ASI1 Capture",
                        .channels_min   = 0,
                        .channels_max   = 2,
                        .rates = TAS2764_RATES,
                        .formats = TAS2764_FORMATS,
                },
                .ops = &tas2764_dai_ops,
                .symmetric_rate = 1,
        },
};

static uint8_t sn012776_bop_presets[] = {
        0x01, 0x32, 0x02, 0x22, 0x83, 0x2d, 0x80, 0x02, 0x06,
        0x32, 0x46, 0x30, 0x02, 0x06, 0x38, 0x40, 0x30, 0x02,
        0x06, 0x3e, 0x37, 0x30, 0xff, 0xe6
};

static const struct regmap_config tas2764_i2c_regmap;

static int tas2764_apply_init_quirks(struct tas2764_priv *tas2764)
{
        int ret, i;

        for (i = 0; i < ARRAY_SIZE(tas2764_quirk_init_sequences); i++) {
                const struct tas2764_quirk_init_sequence *init_seq =
                                                &tas2764_quirk_init_sequences[i];

                if (!init_seq->seq)
                        continue;

                if (!(BIT(i) & ENABLED_APPLE_QUIRKS))
                        continue;

                ret = regmap_multi_reg_write(tas2764->regmap, init_seq->seq,
                                             init_seq->len);

                if (ret < 0)
                        return ret;
        }

        return 0;
}

static int tas2764_read_die_temp(struct tas2764_priv *tas2764, long *result)
{
        int ret, reg;

        ret = regmap_read(tas2764->regmap, TAS2764_TEMP, &reg);
        if (ret)
                return ret;
        /*
         * As per datasheet, subtract 93 from raw value to get degrees
         * Celsius. hwmon wants millidegrees.
         *
         * NOTE: The chip will initialise the TAS2764_TEMP register to
         * 2.6 *C to avoid triggering temperature protection. Since the
         * ADC is powered down during software shutdown, this value will
         * persist until the chip is fully powered up (e.g. the PCM it's
         * attached to is opened). The ADC will power down again when
         * the chip is put back into software shutdown, with the last
         * value sampled persisting in the ADC's register.
         */
        *result = (reg - 93) * 1000;
        return 0;
}

static umode_t tas2764_hwmon_is_visible(const void *data,
                                        enum hwmon_sensor_types type, u32 attr,
                                        int channel)
{
        if (type != hwmon_temp)
                return 0;

        switch (attr) {
        case hwmon_temp_input:
                return 0444;
        default:
                break;
        }

        return 0;
}

static int tas2764_hwmon_read(struct device *dev,
                              enum hwmon_sensor_types type,
                              u32 attr, int channel, long *val)
{
        struct tas2764_priv *tas2764 = dev_get_drvdata(dev);
        int ret;

        switch (attr) {
        case hwmon_temp_input:
                ret = tas2764_read_die_temp(tas2764, val);
                break;
        default:
                ret = -EOPNOTSUPP;
                break;
        }

        return ret;
}

static const struct hwmon_channel_info *const tas2764_hwmon_info[] = {
        HWMON_CHANNEL_INFO(temp, HWMON_T_INPUT),
        NULL
};

static const struct hwmon_ops tas2764_hwmon_ops = {
        .is_visible     = tas2764_hwmon_is_visible,
        .read           = tas2764_hwmon_read,
};

static const struct hwmon_chip_info tas2764_hwmon_chip_info = {
        .ops    = &tas2764_hwmon_ops,
        .info   = tas2764_hwmon_info,
};

static int tas2764_codec_probe(struct snd_soc_component *component)
{
        struct tas2764_priv *tas2764 = snd_soc_component_get_drvdata(component);
        int ret, i;

        tas2764->component = component;

        if (tas2764->sdz_gpio) {
                gpiod_set_value_cansleep(tas2764->sdz_gpio, 1);
                usleep_range(1000, 2000);
        }

        tas2764_reset(tas2764);
        regmap_reinit_cache(tas2764->regmap, &tas2764_i2c_regmap);

        if (tas2764->irq) {
                ret = snd_soc_component_write(tas2764->component, TAS2764_INT_MASK0, 0x00);
                if (ret < 0)
                        return ret;

                ret = snd_soc_component_write(tas2764->component, TAS2764_INT_MASK1, 0xff);
                if (ret < 0)
                        return ret;

                ret = snd_soc_component_write(tas2764->component, TAS2764_INT_MASK2, 0xff);
                if (ret < 0)
                        return ret;

                ret = snd_soc_component_write(tas2764->component, TAS2764_INT_MASK3, 0xff);
                if (ret < 0)
                        return ret;

                ret = snd_soc_component_write(tas2764->component, TAS2764_INT_MASK4, 0xff);
                if (ret < 0)
                        return ret;

                ret = devm_request_threaded_irq(tas2764->dev, tas2764->irq, NULL, tas2764_irq,
                                                IRQF_ONESHOT | IRQF_SHARED | IRQF_TRIGGER_LOW,
                                                "tas2764", tas2764);
                if (ret)
                        dev_warn(tas2764->dev, "failed to request IRQ: %d\n", ret);
        }

        ret = snd_soc_component_update_bits(tas2764->component, TAS2764_TDM_CFG5,
                                            TAS2764_TDM_CFG5_VSNS_ENABLE, 0);
        if (ret < 0)
                return ret;

        ret = snd_soc_component_update_bits(tas2764->component, TAS2764_TDM_CFG6,
                                            TAS2764_TDM_CFG6_ISNS_ENABLE, 0);
        if (ret < 0)
                return ret;

        switch (tas2764->devid) {
        case DEVID_SN012776:
                ret = snd_soc_component_update_bits(component, TAS2764_PWR_CTRL,
                                        TAS2764_PWR_CTRL_BOP_SRC,
                                        TAS2764_PWR_CTRL_BOP_SRC);
                if (ret < 0)
                        return ret;

                for (i = 0; i < ARRAY_SIZE(sn012776_bop_presets); i++) {
                        ret = snd_soc_component_write(component,
                                                TAS2764_BOP_CFG0 + i,
                                                sn012776_bop_presets[i]);

                        if (ret < 0)
                                return ret;
                }

                /* Apply all enabled Apple quirks */
                ret = tas2764_apply_init_quirks(tas2764);

                if (ret < 0)
                        return ret;

                break;
        default:
                break;
        }

        return 0;
}

static DECLARE_TLV_DB_SCALE(tas2764_digital_tlv, 1100, 50, 0);
static DECLARE_TLV_DB_SCALE(tas2764_playback_volume, -10050, 50, 1);

static const char * const tas2764_hpf_texts[] = {
        "Disabled", "2 Hz", "50 Hz", "100 Hz", "200 Hz",
        "400 Hz", "800 Hz"
};

static SOC_ENUM_SINGLE_DECL(
        tas2764_hpf_enum, TAS2764_DC_BLK0,
        TAS2764_DC_BLK0_HPF_FREQ_PB_SHIFT, tas2764_hpf_texts);

static const char * const tas2764_oce_texts[] = {
        "Disable", "Retry",
};

static SOC_ENUM_SINGLE_DECL(
        tas2764_oce_enum, TAS2764_MISC_CFG1,
        TAS2764_MISC_CFG1_OCE_RETRY_SHIFT, tas2764_oce_texts);

static const struct snd_kcontrol_new tas2764_snd_controls[] = {
        SOC_SINGLE_TLV("Speaker Volume", TAS2764_DVC, 0,
                       TAS2764_DVC_MAX, 1, tas2764_playback_volume),
        SOC_SINGLE_TLV("Amp Gain Volume", TAS2764_CHNL_0, 1, 0x14, 0,
                       tas2764_digital_tlv),
        SOC_ENUM("HPF Corner Frequency", tas2764_hpf_enum),
        SOC_ENUM("OCE Handling", tas2764_oce_enum),
};

static const struct snd_soc_component_driver soc_component_driver_tas2764 = {
        .probe                  = tas2764_codec_probe,
        .suspend                = tas2764_codec_suspend,
        .resume                 = tas2764_codec_resume,
        .controls               = tas2764_snd_controls,
        .num_controls           = ARRAY_SIZE(tas2764_snd_controls),
        .dapm_widgets           = tas2764_dapm_widgets,
        .num_dapm_widgets       = ARRAY_SIZE(tas2764_dapm_widgets),
        .dapm_routes            = tas2764_audio_map,
        .num_dapm_routes        = ARRAY_SIZE(tas2764_audio_map),
        .idle_bias_on           = 1,
        .endianness             = 1,
};

static const struct reg_default tas2764_reg_defaults[] = {
        { TAS2764_PAGE, 0x00 },
        { TAS2764_SW_RST, 0x00 },
        { TAS2764_PWR_CTRL, 0x1a },
        { TAS2764_DVC, 0x00 },
        { TAS2764_CHNL_0, 0x28 },
        { TAS2764_TDM_CFG0, 0x09 },
        { TAS2764_TDM_CFG1, 0x02 },
        { TAS2764_TDM_CFG2, 0x0a },
        { TAS2764_TDM_CFG3, 0x10 },
        { TAS2764_TDM_CFG5, 0x42 },
        { TAS2764_INT_CLK_CFG, 0x19 },
};

static const struct regmap_range_cfg tas2764_regmap_ranges[] = {
        {
                .range_min = 0,
                .range_max = 0xffff,
                .selector_reg = TAS2764_PAGE,
                .selector_mask = 0xff,
                .selector_shift = 0,
                .window_start = 0,
                .window_len = 128,
        },
};

static bool tas2764_volatile_register(struct device *dev, unsigned int reg)
{
        switch (reg) {
        case TAS2764_SW_RST:
        case TAS2764_INT_LTCH0 ... TAS2764_INT_LTCH4:
        case TAS2764_INT_CLK_CFG:
                return true;
        case TAS2764_REG(0xf0, 0x0) ... TAS2764_REG(0xff, 0x0):
                /* TI's undocumented registers for the application of quirks */
                return true;
        default:
                return false;
        }
}

static const struct regmap_config tas2764_i2c_regmap = {
        .reg_bits = 8,
        .val_bits = 8,
        .volatile_reg = tas2764_volatile_register,
        .reg_defaults = tas2764_reg_defaults,
        .num_reg_defaults = ARRAY_SIZE(tas2764_reg_defaults),
        .cache_type = REGCACHE_RBTREE,
        .ranges = tas2764_regmap_ranges,
        .num_ranges = ARRAY_SIZE(tas2764_regmap_ranges),
        .max_register = 0xffff,
};

static int tas2764_parse_dt(struct device *dev, struct tas2764_priv *tas2764)
{
        int ret = 0;

        tas2764->reset_gpio = devm_gpiod_get_optional(tas2764->dev, "reset",
                                                      GPIOD_OUT_HIGH);
        if (IS_ERR(tas2764->reset_gpio)) {
                if (PTR_ERR(tas2764->reset_gpio) == -EPROBE_DEFER) {
                        tas2764->reset_gpio = NULL;
                        return -EPROBE_DEFER;
                }
        }

        tas2764->sdz_gpio = devm_gpiod_get_optional(dev, "shutdown", GPIOD_OUT_HIGH);
        if (IS_ERR(tas2764->sdz_gpio)) {
                if (PTR_ERR(tas2764->sdz_gpio) == -EPROBE_DEFER)
                        return -EPROBE_DEFER;

                tas2764->sdz_gpio = NULL;
        }

        ret = fwnode_property_read_u32(dev->fwnode, "ti,imon-slot-no",
                                       &tas2764->i_sense_slot);
        if (ret)
                tas2764->i_sense_slot = 0;

        ret = fwnode_property_read_u32(dev->fwnode, "ti,vmon-slot-no",
                                       &tas2764->v_sense_slot);
        if (ret)
                tas2764->v_sense_slot = 2;

        return 0;
}

static int tas2764_i2c_probe(struct i2c_client *client)
{
        struct tas2764_priv *tas2764;
        int result;

        tas2764 = devm_kzalloc(&client->dev, sizeof(struct tas2764_priv),
                               GFP_KERNEL);
        if (!tas2764)
                return -ENOMEM;

        tas2764->devid = (kernel_ulong_t)of_device_get_match_data(&client->dev);

        tas2764->dev = &client->dev;
        tas2764->irq = client->irq;
        i2c_set_clientdata(client, tas2764);
        dev_set_drvdata(&client->dev, tas2764);

        tas2764->regmap = devm_regmap_init_i2c(client, &tas2764_i2c_regmap);
        if (IS_ERR(tas2764->regmap)) {
                result = PTR_ERR(tas2764->regmap);
                dev_err(&client->dev, "Failed to allocate register map: %d\n",
                                        result);
                return result;
        }

        if (client->dev.of_node) {
                result = tas2764_parse_dt(&client->dev, tas2764);
                if (result) {
                        dev_err(tas2764->dev, "%s: Failed to parse devicetree\n",
                                __func__);
                        return result;
                }
        }

        if (IS_REACHABLE(CONFIG_HWMON)) {
                struct device *hwmon;

                hwmon = devm_hwmon_device_register_with_info(&client->dev, "tas2764",
                                                        tas2764,
                                                        &tas2764_hwmon_chip_info,
                                                        NULL);
                if (IS_ERR(hwmon)) {
                        return dev_err_probe(&client->dev, PTR_ERR(hwmon),
                                             "Failed to register temp sensor\n");
                }
        }


        return devm_snd_soc_register_component(tas2764->dev,
                                               &soc_component_driver_tas2764,
                                               tas2764_dai_driver,
                                               ARRAY_SIZE(tas2764_dai_driver));
}

static const struct i2c_device_id tas2764_i2c_id[] = {
        { "tas2764"},
        { }
};
MODULE_DEVICE_TABLE(i2c, tas2764_i2c_id);

#if defined(CONFIG_OF)
static const struct of_device_id tas2764_of_match[] = {
        { .compatible = "ti,tas2764",  .data = (void *)DEVID_TAS2764 },
        { .compatible = "ti,sn012776", .data = (void *)DEVID_SN012776 },
        {},
};
MODULE_DEVICE_TABLE(of, tas2764_of_match);
#endif

static struct i2c_driver tas2764_i2c_driver = {
        .driver = {
                .name   = "tas2764",
                .of_match_table = of_match_ptr(tas2764_of_match),
        },
        .probe      = tas2764_i2c_probe,
        .id_table   = tas2764_i2c_id,
};
module_i2c_driver(tas2764_i2c_driver);

MODULE_AUTHOR("Dan Murphy <dmurphy@ti.com>");
MODULE_DESCRIPTION("TAS2764 I2C Smart Amplifier driver");
MODULE_LICENSE("GPL v2");