// SPDX-License-Identifier: GPL-2.0
//
// ALSA SoC Texas Instruments TAS2770 20-W Digital Input Mono Class-D
// Audio Amplifier with Speaker I/V Sense
//
// Copyright (C) 2016-2017 Texas Instruments Incorporated - https://www.ti.com/
//      Author: Tracy Yi <tracy-yi@ti.com>
//      Frank Shi <shifu0704@thundersoft.com>

#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/firmware.h>
#include <linux/regmap.h>
#include <linux/of.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 "tas2770.h"

#define TAS2770_MDELAY 0xFFFFFFFE

static void tas2770_reset(struct tas2770_priv *tas2770)
{
        if (tas2770->reset_gpio) {
                gpiod_set_value_cansleep(tas2770->reset_gpio, 0);
                msleep(20);
                gpiod_set_value_cansleep(tas2770->reset_gpio, 1);
                usleep_range(1000, 2000);
        }

        snd_soc_component_write(tas2770->component, TAS2770_SW_RST,
                TAS2770_RST);
        usleep_range(1000, 2000);
}

static int tas2770_update_pwr_ctrl(struct tas2770_priv *tas2770)
{
        struct snd_soc_component *component = tas2770->component;
        unsigned int val;
        int ret;

        if (tas2770->dac_powered)
                val = tas2770->unmuted ?
                        TAS2770_PWR_CTRL_ACTIVE : TAS2770_PWR_CTRL_MUTE;
        else
                val = TAS2770_PWR_CTRL_SHUTDOWN;

        ret = snd_soc_component_update_bits(component, TAS2770_PWR_CTRL,
                                            TAS2770_PWR_CTRL_MASK, val);
        if (ret < 0)
                return ret;

        return 0;
}

#ifdef CONFIG_PM
static int tas2770_codec_suspend(struct snd_soc_component *component)
{
        struct tas2770_priv *tas2770 = snd_soc_component_get_drvdata(component);
        int ret = 0;

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

        if (tas2770->sdz_gpio) {
                gpiod_set_value_cansleep(tas2770->sdz_gpio, 0);
        } else {
                ret = snd_soc_component_update_bits(component, TAS2770_PWR_CTRL,
                                                    TAS2770_PWR_CTRL_MASK,
                                                    TAS2770_PWR_CTRL_SHUTDOWN);
                if (ret < 0) {
                        regcache_cache_only(tas2770->regmap, false);
                        regcache_sync(tas2770->regmap);
                        return ret;
                }

                ret = 0;
        }

        return ret;
}

static int tas2770_codec_resume(struct snd_soc_component *component)
{
        struct tas2770_priv *tas2770 = snd_soc_component_get_drvdata(component);
        int ret;

        if (tas2770->sdz_gpio) {
                gpiod_set_value_cansleep(tas2770->sdz_gpio, 1);
                usleep_range(1000, 2000);
        } else {
                ret = tas2770_update_pwr_ctrl(tas2770);
                if (ret < 0)
                        return ret;
        }

        regcache_cache_only(tas2770->regmap, false);

        return regcache_sync(tas2770->regmap);
}
#else
#define tas2770_codec_suspend NULL
#define tas2770_codec_resume NULL
#endif

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

static SOC_ENUM_SINGLE_DECL(
        tas2770_ASI1_src_enum, TAS2770_TDM_CFG_REG2,
        4, tas2770_ASI1_src);

static const struct snd_kcontrol_new tas2770_asi1_mux =
        SOC_DAPM_ENUM("ASI1 Source", tas2770_ASI1_src_enum);

static int tas2770_dac_event(struct snd_soc_dapm_widget *w,
                             struct snd_kcontrol *kcontrol, int event)
{
        struct snd_soc_component *component =
                        snd_soc_dapm_to_component(w->dapm);
        struct tas2770_priv *tas2770 =
                        snd_soc_component_get_drvdata(component);
        int ret;

        switch (event) {
        case SND_SOC_DAPM_POST_PMU:
                tas2770->dac_powered = 1;
                ret = tas2770_update_pwr_ctrl(tas2770);
                break;
        case SND_SOC_DAPM_PRE_PMD:
                tas2770->dac_powered = 0;
                ret = tas2770_update_pwr_ctrl(tas2770);
                break;
        default:
                dev_err(tas2770->dev, "Not supported evevt\n");
                return -EINVAL;
        }

        return ret;
}

static const struct snd_kcontrol_new isense_switch =
        SOC_DAPM_SINGLE("Switch", TAS2770_PWR_CTRL, 3, 1, 1);
static const struct snd_kcontrol_new vsense_switch =
        SOC_DAPM_SINGLE("Switch", TAS2770_PWR_CTRL, 2, 1, 1);

static int sense_event(struct snd_soc_dapm_widget *w,
                        struct snd_kcontrol *kcontrol, int event)
{
        struct snd_soc_component *component = snd_soc_dapm_to_component(w->dapm);
        struct tas2770_priv *tas2770 = snd_soc_component_get_drvdata(component);

        /*
         * Powering up ISENSE/VSENSE requires a trip through the shutdown state.
         * Do that here to ensure that our changes are applied properly, otherwise
         * we might end up with non-functional IVSENSE if playback started earlier,
         * which would break software speaker protection.
         */
        switch (event) {
        case SND_SOC_DAPM_PRE_REG:
                return snd_soc_component_update_bits(component, TAS2770_PWR_CTRL,
                                                    TAS2770_PWR_CTRL_MASK,
                                                    TAS2770_PWR_CTRL_SHUTDOWN);
        case SND_SOC_DAPM_POST_REG:
                return tas2770_update_pwr_ctrl(tas2770);
        default:
                return 0;
        }
}

static const struct snd_soc_dapm_widget tas2770_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, &tas2770_asi1_mux),
        SND_SOC_DAPM_SWITCH_E("ISENSE", TAS2770_PWR_CTRL, 3, 1, &isense_switch,
                sense_event, SND_SOC_DAPM_PRE_REG | SND_SOC_DAPM_POST_REG),
        SND_SOC_DAPM_SWITCH_E("VSENSE", TAS2770_PWR_CTRL, 2, 1, &vsense_switch,
                sense_event, SND_SOC_DAPM_PRE_REG | SND_SOC_DAPM_POST_REG),
        SND_SOC_DAPM_DAC_E("DAC", NULL, SND_SOC_NOPM, 0, 0, tas2770_dac_event,
                           SND_SOC_DAPM_POST_PMU | SND_SOC_DAPM_PRE_PMD),
        SND_SOC_DAPM_OUTPUT("OUT"),
        SND_SOC_DAPM_SIGGEN("VMON"),
        SND_SOC_DAPM_SIGGEN("IMON")
};

static const struct snd_soc_dapm_route tas2770_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 tas2770_mute(struct snd_soc_dai *dai, int mute, int direction)
{
        struct snd_soc_component *component = dai->component;
        struct tas2770_priv *tas2770 =
                        snd_soc_component_get_drvdata(component);

        tas2770->unmuted = !mute;
        return tas2770_update_pwr_ctrl(tas2770);
}

static int tas2770_set_ivsense_transmit(struct tas2770_priv *tas2770,
                                        int i_slot, int v_slot)
{
        struct snd_soc_component *component = tas2770->component;
        int ret;

        ret = snd_soc_component_update_bits(component, TAS2770_TDM_CFG_REG5,
                                            TAS2770_TDM_CFG_REG5_VSNS_MASK |
                                            TAS2770_TDM_CFG_REG5_50_MASK,
                                            TAS2770_TDM_CFG_REG5_VSNS_ENABLE |
                                            v_slot);
        if (ret < 0)
                return ret;

        ret = snd_soc_component_update_bits(component, TAS2770_TDM_CFG_REG6,
                                            TAS2770_TDM_CFG_REG6_ISNS_MASK |
                                            TAS2770_TDM_CFG_REG6_50_MASK,
                                            TAS2770_TDM_CFG_REG6_ISNS_ENABLE |
                                            i_slot);
        if (ret < 0)
                return ret;

        return 0;
}

static int tas2770_set_pdm_transmit(struct tas2770_priv *tas2770, int slot)
{
        struct snd_soc_component *component = tas2770->component;
        int ret;

        ret = snd_soc_component_update_bits(component, TAS2770_TDM_CFG_REG7,
                                            TAS2770_TDM_CFG_REG7_PDM_MASK |
                                            TAS2770_TDM_CFG_REG7_50_MASK,
                                            TAS2770_TDM_CFG_REG7_PDM_ENABLE |
                                            slot);
        return ret;
}

static int tas2770_set_bitwidth(struct tas2770_priv *tas2770, int bitwidth)
{
        int ret;
        struct snd_soc_component *component = tas2770->component;

        switch (bitwidth) {
        case SNDRV_PCM_FORMAT_S16_LE:
                ret = snd_soc_component_update_bits(component, TAS2770_TDM_CFG_REG2,
                                                    TAS2770_TDM_CFG_REG2_RXW_MASK,
                                                    TAS2770_TDM_CFG_REG2_RXW_16BITS);
                break;
        case SNDRV_PCM_FORMAT_S24_LE:
                ret = snd_soc_component_update_bits(component, TAS2770_TDM_CFG_REG2,
                                                    TAS2770_TDM_CFG_REG2_RXW_MASK,
                                                    TAS2770_TDM_CFG_REG2_RXW_24BITS);
                break;
        case SNDRV_PCM_FORMAT_S32_LE:
                ret = snd_soc_component_update_bits(component, TAS2770_TDM_CFG_REG2,
                                                    TAS2770_TDM_CFG_REG2_RXW_MASK,
                                                    TAS2770_TDM_CFG_REG2_RXW_32BITS);
                break;

        default:
                return -EINVAL;
        }

        if (ret < 0)
                return ret;

        return 0;
}

static int tas2770_set_samplerate(struct tas2770_priv *tas2770, int samplerate)
{
        struct snd_soc_component *component = tas2770->component;
        int ramp_rate_val;
        int ret;

        switch (samplerate) {
        case 48000:
                ramp_rate_val = TAS2770_TDM_CFG_REG0_SMP_48KHZ |
                                TAS2770_TDM_CFG_REG0_31_44_1_48KHZ;
                break;
        case 44100:
                ramp_rate_val = TAS2770_TDM_CFG_REG0_SMP_44_1KHZ |
                                TAS2770_TDM_CFG_REG0_31_44_1_48KHZ;
                break;
        case 96000:
                ramp_rate_val = TAS2770_TDM_CFG_REG0_SMP_48KHZ |
                                TAS2770_TDM_CFG_REG0_31_88_2_96KHZ;
                break;
        case 88200:
                ramp_rate_val = TAS2770_TDM_CFG_REG0_SMP_44_1KHZ |
                                TAS2770_TDM_CFG_REG0_31_88_2_96KHZ;
                break;
        case 192000:
                ramp_rate_val = TAS2770_TDM_CFG_REG0_SMP_48KHZ |
                                TAS2770_TDM_CFG_REG0_31_176_4_192KHZ;
                break;
        case 176400:
                ramp_rate_val = TAS2770_TDM_CFG_REG0_SMP_44_1KHZ |
                                TAS2770_TDM_CFG_REG0_31_176_4_192KHZ;
                break;
        default:
                return -EINVAL;
        }

        ret = snd_soc_component_update_bits(component, TAS2770_TDM_CFG_REG0,
                                            TAS2770_TDM_CFG_REG0_SMP_MASK |
                                            TAS2770_TDM_CFG_REG0_31_MASK,
                                            ramp_rate_val);
        if (ret < 0)
                return ret;

        return 0;
}

static int tas2770_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 tas2770_priv *tas2770 =
                        snd_soc_component_get_drvdata(component);
        int ret;

        ret = tas2770_set_bitwidth(tas2770, params_format(params));
        if (ret)
                return ret;

        return tas2770_set_samplerate(tas2770, params_rate(params));
}

static int tas2770_set_fmt(struct snd_soc_dai *dai, unsigned int fmt)
{
        struct snd_soc_component *component = dai->component;
        struct tas2770_priv *tas2770 =
                        snd_soc_component_get_drvdata(component);
        u8 tdm_rx_start_slot = 0, invert_fpol = 0, fpol_preinv = 0, asi_cfg_1 = 0;
        int ret;

        switch (fmt & SND_SOC_DAIFMT_CLOCK_PROVIDER_MASK) {
        case SND_SOC_DAIFMT_CBC_CFC:
                break;
        default:
                dev_err(tas2770->dev, "ASI invalid DAI clocking\n");
                return -EINVAL;
        }

        switch (fmt & SND_SOC_DAIFMT_INV_MASK) {
        case SND_SOC_DAIFMT_NB_IF:
                invert_fpol = 1;
                fallthrough;
        case SND_SOC_DAIFMT_NB_NF:
                asi_cfg_1 |= TAS2770_TDM_CFG_REG1_RX_RSING;
                break;
        case SND_SOC_DAIFMT_IB_IF:
                invert_fpol = 1;
                fallthrough;
        case SND_SOC_DAIFMT_IB_NF:
                asi_cfg_1 |= TAS2770_TDM_CFG_REG1_RX_FALING;
                break;
        default:
                dev_err(tas2770->dev, "ASI format Inverse is not found\n");
                return -EINVAL;
        }

        ret = snd_soc_component_update_bits(component, TAS2770_TDM_CFG_REG1,
                                            TAS2770_TDM_CFG_REG1_RX_MASK,
                                            asi_cfg_1);
        if (ret < 0)
                return ret;

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

        ret = snd_soc_component_update_bits(component, TAS2770_TDM_CFG_REG1,
                                            TAS2770_TDM_CFG_REG1_MASK,
                                            (tdm_rx_start_slot << TAS2770_TDM_CFG_REG1_51_SHIFT));
        if (ret < 0)
                return ret;

        ret = snd_soc_component_update_bits(component, TAS2770_TDM_CFG_REG0,
                                            TAS2770_TDM_CFG_REG0_FPOL_MASK,
                                            (fpol_preinv ^ invert_fpol)
                                             ? TAS2770_TDM_CFG_REG0_FPOL_RSING
                                             : TAS2770_TDM_CFG_REG0_FPOL_FALING);
        if (ret < 0)
                return ret;

        return 0;
}

static int tas2770_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;
        int left_slot, right_slot;
        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;

        ret = snd_soc_component_update_bits(component, TAS2770_TDM_CFG_REG3,
                                            TAS2770_TDM_CFG_REG3_30_MASK,
                                            (left_slot << TAS2770_TDM_CFG_REG3_30_SHIFT));
        if (ret < 0)
                return ret;
        ret = snd_soc_component_update_bits(component, TAS2770_TDM_CFG_REG3,
                                            TAS2770_TDM_CFG_REG3_RXS_MASK,
                                            (right_slot << TAS2770_TDM_CFG_REG3_RXS_SHIFT));
        if (ret < 0)
                return ret;

        switch (slot_width) {
        case 16:
                ret = snd_soc_component_update_bits(component, TAS2770_TDM_CFG_REG2,
                                                    TAS2770_TDM_CFG_REG2_RXS_MASK,
                                                    TAS2770_TDM_CFG_REG2_RXS_16BITS);
                break;
        case 24:
                ret = snd_soc_component_update_bits(component, TAS2770_TDM_CFG_REG2,
                                                    TAS2770_TDM_CFG_REG2_RXS_MASK,
                                                    TAS2770_TDM_CFG_REG2_RXS_24BITS);
                break;
        case 32:
                ret = snd_soc_component_update_bits(component, TAS2770_TDM_CFG_REG2,
                                                    TAS2770_TDM_CFG_REG2_RXS_MASK,
                                                    TAS2770_TDM_CFG_REG2_RXS_32BITS);
                break;
        case 0:
                /* Do not change slot width */
                ret = 0;
                break;
        default:
                ret = -EINVAL;
        }

        if (ret < 0)
                return ret;

        return 0;
}

static const struct snd_soc_dai_ops tas2770_dai_ops = {
        .mute_stream = tas2770_mute,
        .hw_params  = tas2770_hw_params,
        .set_fmt    = tas2770_set_fmt,
        .set_tdm_slot = tas2770_set_dai_tdm_slot,
        .no_capture_mute = 1,
};

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

#define TAS2770_RATES (SNDRV_PCM_RATE_44100 | SNDRV_PCM_RATE_48000 |\
                                           SNDRV_PCM_RATE_96000 |\
                                            SNDRV_PCM_RATE_192000\
                                          )

static struct snd_soc_dai_driver tas2770_dai_driver[] = {
        {
                .name = "tas2770 ASI1",
                .id = 0,
                .playback = {
                        .stream_name    = "ASI1 Playback",
                        .channels_min   = 1,
                        .channels_max   = 2,
                        .rates      = TAS2770_RATES,
                        .formats    = TAS2770_FORMATS,
                },
                .capture = {
                        .stream_name    = "ASI1 Capture",
                        .channels_min   = 0,
                        .channels_max   = 2,
                        .rates          = TAS2770_RATES,
                        .formats    = TAS2770_FORMATS,
                },
                .ops = &tas2770_dai_ops,
                .symmetric_rate = 1,
        },
};

static int tas2770_read_die_temp(struct tas2770_priv *tas2770, long *result)
{
        int ret = 0;
        int reading, msb, lsb;

        ret = regmap_read(tas2770->regmap, TAS2770_TEMP_MSB, &msb);
        if (ret)
                return ret;

        ret = regmap_read(tas2770->regmap, TAS2770_TEMP_LSB, &lsb);
        if (ret)
                return ret;

        reading = (msb << 4) | (lsb >> 4);

        /*
         * As per datasheet: divide register by 16 and subtract 93 to get
         * degrees Celsius. hwmon requires millidegrees. Let's avoid rounding
         * errors by subtracting 93 * 16 then multiplying by 1000 / 16.
         *
         * NOTE: The ADC registers are initialised to 0 on reset. This means
         * that the temperature will read -93 *C until the chip is brought out
         * of software shutdown (e.g. the PCM it's attached to is opened). The
         * ADC is also shut down in software shutdown/low-power mode, so the
         * value read back from its registers will be the last value sampled
         * before entering software shutdown.
         */
        *result = (reading - (93 * 16)) * (1000 / 16);
        return 0;
}

static umode_t tas2770_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 tas2770_hwmon_read(struct device *dev,
                              enum hwmon_sensor_types type,
                              u32 attr, int channel, long *val)
{
        struct tas2770_priv *tas2770 = dev_get_drvdata(dev);
        int ret;

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

        return ret;
}

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

static const struct hwmon_ops tas2770_hwmon_ops = {
        .is_visible     = tas2770_hwmon_is_visible,
        .read           = tas2770_hwmon_read,
};

static const struct hwmon_chip_info tas2770_hwmon_chip_info = {
        .ops    = &tas2770_hwmon_ops,
        .info   = tas2770_hwmon_info,
};

static const struct regmap_config tas2770_i2c_regmap;

static int tas2770_codec_probe(struct snd_soc_component *component)
{
        struct tas2770_priv *tas2770 =
                        snd_soc_component_get_drvdata(component);
        int ret;

        tas2770->component = component;

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

        tas2770_reset(tas2770);
        regmap_reinit_cache(tas2770->regmap, &tas2770_i2c_regmap);

        if (tas2770->i_sense_slot != -1 && tas2770->v_sense_slot != -1) {
                ret = tas2770_set_ivsense_transmit(tas2770, tas2770->i_sense_slot,
                                                   tas2770->v_sense_slot);

                if (ret < 0)
                        return ret;
        }

        if (tas2770->pdm_slot != -1) {
                ret = tas2770_set_pdm_transmit(tas2770, tas2770->pdm_slot);

                if (ret < 0)
                        return ret;
        }

        return 0;
}

static DECLARE_TLV_DB_SCALE(tas2770_digital_tlv, 1100, 50, 0);
static DECLARE_TLV_DB_SCALE(tas2770_playback_volume, -10050, 50, 0);

static const struct snd_kcontrol_new tas2770_snd_controls[] = {
        SOC_SINGLE_TLV("Speaker Playback Volume", TAS2770_PLAY_CFG_REG2,
                       0, TAS2770_PLAY_CFG_REG2_VMAX, 1, tas2770_playback_volume),
        SOC_SINGLE_TLV("Amp Gain Volume", TAS2770_PLAY_CFG_REG0, 0, 0x14, 0,
                       tas2770_digital_tlv),
};

static const struct snd_soc_component_driver soc_component_driver_tas2770 = {
        .probe                  = tas2770_codec_probe,
        .suspend                = tas2770_codec_suspend,
        .resume                 = tas2770_codec_resume,
        .controls               = tas2770_snd_controls,
        .num_controls           = ARRAY_SIZE(tas2770_snd_controls),
        .dapm_widgets           = tas2770_dapm_widgets,
        .num_dapm_widgets       = ARRAY_SIZE(tas2770_dapm_widgets),
        .dapm_routes            = tas2770_audio_map,
        .num_dapm_routes        = ARRAY_SIZE(tas2770_audio_map),
        .idle_bias_on           = 1,
        .endianness             = 1,
};

static int tas2770_register_codec(struct tas2770_priv *tas2770)
{
        return devm_snd_soc_register_component(tas2770->dev,
                &soc_component_driver_tas2770,
                tas2770_dai_driver, ARRAY_SIZE(tas2770_dai_driver));
}

static const struct reg_default tas2770_reg_defaults[] = {
        { TAS2770_PAGE, 0x00 },
        { TAS2770_SW_RST, 0x00 },
        { TAS2770_PWR_CTRL, 0x0e },
        { TAS2770_PLAY_CFG_REG0, 0x10 },
        { TAS2770_PLAY_CFG_REG1, 0x01 },
        { TAS2770_PLAY_CFG_REG2, 0x00 },
        { TAS2770_MSC_CFG_REG0, 0x07 },
        { TAS2770_TDM_CFG_REG1, 0x02 },
        { TAS2770_TDM_CFG_REG2, 0x0a },
        { TAS2770_TDM_CFG_REG3, 0x10 },
        { TAS2770_INT_MASK_REG0, 0xfc },
        { TAS2770_INT_MASK_REG1, 0xb1 },
        { TAS2770_INT_CFG, 0x05 },
        { TAS2770_MISC_IRQ, 0x81 },
        { TAS2770_CLK_CGF, 0x0c },

};

static bool tas2770_volatile(struct device *dev, unsigned int reg)
{
        switch (reg) {
        case TAS2770_PAGE: /* regmap implementation requires this */
        case TAS2770_SW_RST: /* always clears after write */
        case TAS2770_BO_PRV_REG0:/* has a self clearing bit */
        case TAS2770_LVE_INT_REG0:
        case TAS2770_LVE_INT_REG1:
        case TAS2770_LAT_INT_REG0:/* Sticky interrupt flags */
        case TAS2770_LAT_INT_REG1:/* Sticky interrupt flags */
        case TAS2770_VBAT_MSB:
        case TAS2770_VBAT_LSB:
        case TAS2770_TEMP_MSB:
        case TAS2770_TEMP_LSB:
                return true;
        }

        return false;
}

static bool tas2770_writeable(struct device *dev, unsigned int reg)
{
        switch (reg) {
        case TAS2770_LVE_INT_REG0:
        case TAS2770_LVE_INT_REG1:
        case TAS2770_LAT_INT_REG0:
        case TAS2770_LAT_INT_REG1:
        case TAS2770_VBAT_MSB:
        case TAS2770_VBAT_LSB:
        case TAS2770_TEMP_MSB:
        case TAS2770_TEMP_LSB:
        case TAS2770_TDM_CLK_DETC:
        case TAS2770_REV_AND_GPID:
                return false;
        }

        return true;
}

static const struct regmap_range_cfg tas2770_regmap_ranges[] = {
        {
                .range_min = 0,
                .range_max = 1 * 128,
                .selector_reg = TAS2770_PAGE,
                .selector_mask = 0xff,
                .selector_shift = 0,
                .window_start = 0,
                .window_len = 128,
        },
};

static const struct regmap_config tas2770_i2c_regmap = {
        .reg_bits = 8,
        .val_bits = 8,
        .writeable_reg = tas2770_writeable,
        .volatile_reg = tas2770_volatile,
        .reg_defaults = tas2770_reg_defaults,
        .num_reg_defaults = ARRAY_SIZE(tas2770_reg_defaults),
        .cache_type = REGCACHE_RBTREE,
        .ranges = tas2770_regmap_ranges,
        .num_ranges = ARRAY_SIZE(tas2770_regmap_ranges),
        .max_register = 1 * 128,
};

static int tas2770_parse_dt(struct device *dev, struct tas2770_priv *tas2770)
{
        int rc = 0;

        rc = fwnode_property_read_u32(dev->fwnode, "ti,imon-slot-no",
                                      &tas2770->i_sense_slot);
        if (rc) {
                dev_info(tas2770->dev, "Property %s is missing setting default slot\n",
                         "ti,imon-slot-no");

                tas2770->i_sense_slot = -1;
        }

        rc = fwnode_property_read_u32(dev->fwnode, "ti,vmon-slot-no",
                                      &tas2770->v_sense_slot);
        if (rc) {
                dev_info(tas2770->dev, "Property %s is missing setting default slot\n",
                         "ti,vmon-slot-no");

                tas2770->v_sense_slot = -1;
        }

        rc = fwnode_property_read_u32(dev->fwnode, "ti,pdm-slot-no",
                                      &tas2770->pdm_slot);
        if (rc)
                tas2770->pdm_slot = -1;

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

                tas2770->sdz_gpio = NULL;
        }

        return 0;
}

static int tas2770_i2c_probe(struct i2c_client *client)
{
        struct tas2770_priv *tas2770;
        int result;

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

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

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

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

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

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

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

        result = tas2770_register_codec(tas2770);
        if (result)
                dev_err(tas2770->dev, "Register codec failed.\n");

        return result;
}

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

#if defined(CONFIG_OF)
static const struct of_device_id tas2770_of_match[] = {
        { .compatible = "ti,tas2770" },
        {},
};
MODULE_DEVICE_TABLE(of, tas2770_of_match);
#endif

static struct i2c_driver tas2770_i2c_driver = {
        .driver = {
                .name   = "tas2770",
                .of_match_table = of_match_ptr(tas2770_of_match),
        },
        .probe      = tas2770_i2c_probe,
        .id_table   = tas2770_i2c_id,
};
module_i2c_driver(tas2770_i2c_driver);

MODULE_AUTHOR("Shi Fu <shifu0704@thundersoft.com>");
MODULE_DESCRIPTION("TAS2770 I2C Smart Amplifier driver");
MODULE_LICENSE("GPL v2");