root/sound/soc/codecs/rtq9124.c
// SPDX-License-Identifier: GPL-2.0-only
//
// rtq9124.c -- RTQ9124 ALSA SoC Codec driver
//
// Author: ChiYuan Huang <cy_huang@richtek.com>

#include <linux/bitfield.h>
#include <linux/bits.h>
#include <linux/byteorder/generic.h>
#include <linux/delay.h>
#include <linux/err.h>
#include <linux/gpio/consumer.h>
#include <linux/i2c.h>
#include <linux/kernel.h>
#include <linux/mod_devicetable.h>
#include <linux/module.h>
#include <linux/of.h>
#include <linux/pm_runtime.h>
#include <linux/property.h>
#include <linux/regmap.h>
#include <sound/pcm.h>
#include <sound/pcm_params.h>
#include <sound/soc.h>
#include <sound/soc-dai.h>
#include <sound/soc-dapm.h>
#include <sound/tlv.h>

#define RTQ9124_REG_SDI_SEL             0x00
#define RTQ9124_REG_SDO_SEL             0x01
#define RTQ9124_REG_I2S_OPT             0x02
#define RTQ9124_REG_AMP_OPT             0x03
#define RTQ9124_REG_STATE_CTRL          0x04
#define RTQ9124_REG_PWM_PHASE           0x05
#define RTQ9124_REG_SIL_CTRL            0x06
#define RTQ9124_REG_PWM_SS_OPT          0x07
#define RTQ9124_REG_ERR_INT_0           0x10
#define RTQ9124_REG_ERR_MASK6           0x26
#define RTQ9124_REG_TDM_TX_CH0          0x32
#define RTQ9124_REG_TDM_RX_CH0          0x34
#define RTQ9124_REG_VOL_OPT             0x38
#define RTQ9124_REG_DCR_TH              0x4B
#define RTQ9124_REG_ERR_TH              0x4C
#define RTQ9124_REG_PROT_EN             0x5B
#define RTQ9124_REG_PRJ_CODE            0xF9

#define RTQ9124_MASK_CS_DATA_INV        BIT(9)
#define RTQ9124_MASK_VDDIO_SDO_SEL      BIT(8)
#define RTQ9124_MASK_AUD_BITS           GENMASK(5, 4)
#define RTQ9124_MASK_AUD_FMT            GENMASK(3, 0)
#define RTQ9124_MASK_CH_STATE           GENMASK(1, 0)
#define RTQ9124_MASK_SF_RESET           BIT(15)

#define RTQ9124_FIXED_VENID             0x9124

struct rtq9124_priv {
        struct gpio_desc *enable;
        unsigned int dai_fmt;
        int tdm_slots;
        int tdm_slot_width;
};

static int rtq9124_enable_event(struct snd_soc_dapm_widget *w, struct snd_kcontrol *kcontrol,
                                int event)
{
        struct snd_soc_component *comp = snd_soc_dapm_to_component(w->dapm);
        unsigned int i, chan_state;

        switch (event) {
        case SND_SOC_DAPM_POST_PMU:
                /* Change state to normal */
                chan_state = 0;
                break;
        case SND_SOC_DAPM_POST_PMD:
                /* Change state to HiZ */
                chan_state = 1;
                break;
        default:
                return -EINVAL;
        }

        /* Before amp turn on, clear old events first */
        for (i = 0; !chan_state && i < 8; i++)
                snd_soc_component_write(comp, RTQ9124_REG_ERR_INT_0 + i, 0xffff);

        snd_soc_component_write_field(comp, RTQ9124_REG_STATE_CTRL, RTQ9124_MASK_CH_STATE,
                                      chan_state);

        return 0;
}

static const struct snd_soc_dapm_widget rtq9124_dapm_widgets[] = {
        SND_SOC_DAPM_OUTPUT("SPK"),
        SND_SOC_DAPM_OUT_DRV_E("Amp Drv", SND_SOC_NOPM, 0, 0, NULL, 0, rtq9124_enable_event,
                               SND_SOC_DAPM_POST_PMU | SND_SOC_DAPM_POST_PMD),
};

static const struct snd_soc_dapm_route rtq9124_dapm_routes[] = {
        { "Amp Drv", NULL, "HiFi Playback" },
        { "SPK", NULL, "Amp Drv" },
};

static const DECLARE_TLV_DB_SCALE(dig_tlv, -10375, 25, 0);
static const DECLARE_TLV_DB_RANGE(ana_tlv,
                                  0, 3, TLV_DB_SCALE_ITEM(-600, 600, 0),
                                  4, 6, TLV_DB_SCALE_ITEM(1400, 200, 0));
static const char * const i2sch_text[] = { "(L+R)/2", "LCH", "RCH", "(L+R)/2" };
static const struct soc_enum rtq9124_i2sch_select_enum =
        SOC_ENUM_SINGLE(RTQ9124_REG_SDI_SEL, 0, ARRAY_SIZE(i2sch_text), i2sch_text);
static const char * const sdo_vsel_text[] = { "1.8V", "3.3V" };
static const struct soc_enum rtq9124_sdo_vselect_enum =
        SOC_ENUM_SINGLE(RTQ9124_REG_SDO_SEL, 8, ARRAY_SIZE(sdo_vsel_text), sdo_vsel_text);
static const char * const pwmfreq_text[] = { "8*fs", "10*fs", "40*fs", "44*fs", "48*fs" };
static const struct soc_enum rtq9124_pwm_freq_enum =
        SOC_ENUM_SINGLE(RTQ9124_REG_AMP_OPT, 4, ARRAY_SIZE(pwmfreq_text), pwmfreq_text);
static const char * const out_angle_text[] = { "0", "45", "90", "135", "180", "225", "270", "315" };
static const struct soc_enum rtq9124_out_angle_enum =
        SOC_ENUM_SINGLE(RTQ9124_REG_PWM_PHASE, 0, ARRAY_SIZE(out_angle_text), out_angle_text);
static const char * const sdo_select_text[] = {
        "None", "I2S DataI", "Interface", "DSP", "DF", "ISense", "ACLoad Cos", "ACLoad Sin",
        "DCR",
};
static const struct soc_enum rtq9124_sdo_select_enum =
        SOC_ENUM_DOUBLE(RTQ9124_REG_SDO_SEL, 4, 0, ARRAY_SIZE(sdo_select_text), sdo_select_text);
static const char * const ulqm_dcvt_text[] = { "Disable", "DC", "VT", "DC+VT" };
static const struct soc_enum rtq9124_ulqm_dcvt_select_enum =
        SOC_ENUM_SINGLE(RTQ9124_REG_STATE_CTRL, 10, ARRAY_SIZE(ulqm_dcvt_text), ulqm_dcvt_text);

static const struct snd_kcontrol_new rtq9124_controls[] = {
        SOC_SINGLE_TLV("Master Volume", RTQ9124_REG_VOL_OPT, 2, 511, 1, dig_tlv),
        SOC_SINGLE_TLV("Speaker Volume", RTQ9124_REG_AMP_OPT, 0, 6, 0, ana_tlv),
        SOC_ENUM("I2S CH Select", rtq9124_i2sch_select_enum),
        SOC_ENUM("SDO VDDIO Select", rtq9124_sdo_vselect_enum),
        SOC_ENUM("PWM Frequency Select", rtq9124_pwm_freq_enum),
        SOC_ENUM("PWM Output Phase Select", rtq9124_out_angle_enum),
        SOC_ENUM("SDO Select", rtq9124_sdo_select_enum),
        SOC_ENUM("ULQM DCVT Select", rtq9124_ulqm_dcvt_select_enum),
        SOC_SINGLE("Silence Detect Enable Switch", RTQ9124_REG_SIL_CTRL, 7, 1, 0),
        SOC_SINGLE("Spread Spectrum Enable Switch", RTQ9124_REG_PWM_SS_OPT, 7, 1, 0),
};

static int rtq9124_comp_probe(struct snd_soc_component *comp)
{
        /* CS Data INV */
        snd_soc_component_write_field(comp, RTQ9124_REG_SDO_SEL, RTQ9124_MASK_CS_DATA_INV, 1);

        /* RTLD */
        snd_soc_component_write(comp, RTQ9124_REG_DCR_TH, 0x5e30);
        snd_soc_component_write(comp, RTQ9124_REG_ERR_TH, 0x3ff);
        snd_soc_component_write(comp, RTQ9124_REG_PROT_EN, 0x3fc);
        snd_soc_component_write(comp, RTQ9124_REG_ERR_MASK6, 0);

        return 0;
}

static const struct snd_soc_component_driver rtq9124_comp_driver = {
        .probe                  = rtq9124_comp_probe,
        .controls               = rtq9124_controls,
        .num_controls           = ARRAY_SIZE(rtq9124_controls),
        .dapm_widgets           = rtq9124_dapm_widgets,
        .num_dapm_widgets       = ARRAY_SIZE(rtq9124_dapm_widgets),
        .dapm_routes            = rtq9124_dapm_routes,
        .num_dapm_routes        = ARRAY_SIZE(rtq9124_dapm_routes),
        .use_pmdown_time        = 1,
        .endianness             = 1,
};

static int rtq9124_dai_set_format(struct snd_soc_dai *dai, unsigned int fmt)
{
        struct rtq9124_priv *rtq9124 = snd_soc_dai_get_drvdata(dai);

        rtq9124->dai_fmt = fmt;
        return 0;
}

static int rtq9124_dai_set_tdm_slot(struct snd_soc_dai *dai, unsigned int tx_mask,
                                    unsigned int rx_mask, int slots, int slot_width)
{
        struct rtq9124_priv *rtq9124 = snd_soc_dai_get_drvdata(dai);
        struct snd_soc_component *comp = dai->component;
        struct device *dev = dai->dev;
        unsigned int byte_loc, i;

        dev_dbg(dev, "(slots, slot_width) = (%d, %d), (txmask, rxmask) = 0x%x, 0x%x\n", slots,
                slot_width, tx_mask, rx_mask);

        if (slots <= 0 || slots > 16 || slot_width <= 0 || slots % 2 || slot_width % 8) {
                dev_err(dev, "Invalid slot parameter (%d, %d)\n", slots, slot_width);
                return -EINVAL;
        }

        if (tx_mask && (hweight_long(tx_mask) > 2 || fls(tx_mask) > slots)) {
                dev_err(dev, "Invalid tx_mask 0x%08x, slots = %d\n", tx_mask, slots);
                return -EINVAL;
        }

        if (!rx_mask || hweight_long(rx_mask) > 1 || fls(rx_mask) > slots) {
                dev_err(dev, "Invalid rx_mask 0x%08x, slots = %d\n", rx_mask, slots);
                return -EINVAL;
        }

        /* Configure tx channel data location */
        for (i = 0; tx_mask; i++, tx_mask ^= BIT(ffs(tx_mask) - 1)) {
                byte_loc = (ffs(tx_mask) - 1) * slot_width / 8;
                snd_soc_component_write(comp, RTQ9124_REG_TDM_TX_CH0 + i, byte_loc);
        }

        /* Configure rx channel data location */
        byte_loc = (ffs(rx_mask) - 1) * slot_width / 8;
        snd_soc_component_write(comp, RTQ9124_REG_TDM_RX_CH0, byte_loc);

        rtq9124->tdm_slots = slots;
        rtq9124->tdm_slot_width = slot_width;

        return 0;
}

static int rtq9124_dai_hw_params(struct snd_pcm_substream *substream,
                                 struct snd_pcm_hw_params *param, struct snd_soc_dai *dai)
{
        struct rtq9124_priv *rtq9124 = snd_soc_dai_get_drvdata(dai);
        struct snd_soc_component *comp = dai->component;
        unsigned int fmtval, width, slot_width, bitrate;
        struct device *dev = dai->dev;
        unsigned int audfmt, audbit;

        fmtval = FIELD_GET(SND_SOC_DAIFMT_FORMAT_MASK, rtq9124->dai_fmt);
        if (rtq9124->tdm_slots && fmtval != SND_SOC_DAIFMT_DSP_A &&
            fmtval != SND_SOC_DAIFMT_DSP_B) {
                dev_err(dev, "TDM only can support DSP_A or DSP_B format\n");
                return -EINVAL;
        }

        switch (fmtval) {
        case SND_SOC_DAIFMT_I2S:
                audfmt = 0;
                break;
        case SND_SOC_DAIFMT_LEFT_J:
                audfmt = 1;
                break;
        case SND_SOC_DAIFMT_RIGHT_J:
                audfmt = 2;
                break;
        case SND_SOC_DAIFMT_DSP_B:
                audfmt = rtq9124->tdm_slots ? 7 : 3;
                break;
        case SND_SOC_DAIFMT_DSP_A:
                audfmt = rtq9124->tdm_slots ? 15 : 11;
                break;
        default:
                dev_err(dev, "Unsupported format %d\n", fmtval);
                return -EINVAL;
        }

        switch (width = params_width(param)) {
        case 16:
                audbit = 0;
                break;
        case 20:
                audbit = 1;
                break;
        case 24:
        case 32:
                audbit = 3;
                break;
        default:
                dev_err(dev, "Unsupported width %d\n", width);
                return -EINVAL;
        }

        if (rtq9124->tdm_slots) {
                slot_width = params_physical_width(param);
                if (slot_width > rtq9124->tdm_slot_width) {
                        dev_err(dev, "Slot width is larger than TDM slot width\n");
                        return -EINVAL;
                }

                bitrate = rtq9124->tdm_slots * rtq9124->tdm_slot_width * params_rate(param);
                if (bitrate > 24576000) {
                        dev_err(dev, "Bitrate exceed the internal PLL 24.576MHz (%d)\n", bitrate);
                        return -EINVAL;
                }
        }

        snd_soc_component_write_field(comp, RTQ9124_REG_I2S_OPT, RTQ9124_MASK_AUD_FMT, audfmt);
        snd_soc_component_write_field(comp, RTQ9124_REG_I2S_OPT, RTQ9124_MASK_AUD_BITS, audbit);

        return 0;
}

static const struct snd_soc_dai_ops rtq9124_dai_ops = {
        .set_fmt        = rtq9124_dai_set_format,
        .set_tdm_slot   = rtq9124_dai_set_tdm_slot,
        .hw_params      = rtq9124_dai_hw_params,
};

static struct snd_soc_dai_driver rtq9124_dai_driver = {
        .name = "HiFi",
        .playback = {
                .stream_name    = "HiFi Playback",
                .formats        = SNDRV_PCM_FMTBIT_S16 | SNDRV_PCM_FMTBIT_S24 |
                                  SNDRV_PCM_FMTBIT_S32,
                .rates          = SNDRV_PCM_RATE_8000_48000 | SNDRV_PCM_RATE_24000 |
                                  SNDRV_PCM_RATE_88200 | SNDRV_PCM_RATE_96000 |
                                  SNDRV_PCM_RATE_192000,
                .rate_min       = 8000,
                .rate_max       = 192000,
                .channels_min   = 1,
                .channels_max   = 2,
        },
        .ops = &rtq9124_dai_ops,
        .symmetric_rate = 1,
        .symmetric_sample_bits = 1,
};

static bool rtq9124_readable_reg(struct device *dev, unsigned int reg)
{
        switch (reg) {
        case 0x00 ... 0x17:
        case 0x20 ... 0x27:
        case 0x30 ... 0x3D:
        case 0x40 ... 0x68:
        case 0x80 ... 0xBC:
        case 0xC0 ... 0xDE:
        case 0xE0 ... 0xE7:
        case 0xF0 ... 0xFD:
                return true;
        default:
                return false;
        }
}

static bool rtq9124_writeable_reg(struct device *dev, unsigned int reg)
{
        switch (reg) {
        case 0x00 ... 0x09:
        case 0x0C ... 0x0E:
        case 0x10 ... 0x17:
        case 0x20 ... 0x27:
        case 0x30:
        case 0x32 ... 0x3D:
        case 0x40 ... 0x4E:
        case 0x50 ... 0x68:
        case 0x80 ... 0xBC:
        case 0xC0 ... 0xDE:
        case 0xE0 ... 0xE7:
        case 0xF0 ... 0xFD:
                return true;
        default:
                return false;
        }
}

static bool rtq9124_volatile_reg(struct device *dev, unsigned int reg)
{
        switch (reg) {
        case 0x0A ... 0x0B:
        case 0x0F ... 0x17:
        case 0x31:
        case 0x4F:
        case 0x51:
        case 0x53 ... 0x57:
        case 0x80 ... 0xBC:
        case 0xC0 ... 0xDE:
        case 0xE0 ... 0xE7:
        case 0xF0 ... 0xFD:
                return true;
        default:
                return false;
        }
}

static inline u8 rtq9124_get_reg_len(unsigned int reg)
{
        return (reg >= 0x40 && reg <= 0x47) ? 4 : 2;
}

static int rtq9124_regmap_read(void *context, const void *reg_buf, size_t reg_size, void *val_buf,
                               size_t val_size)
{
        struct i2c_client *i2c = context;
        u8 reg = *(u8 *)reg_buf;
        u8 size = rtq9124_get_reg_len(reg);
        u32 *val = val_buf;
        int ret;

        ret = i2c_smbus_read_i2c_block_data(i2c, reg, size, val_buf);
        if (ret < 0)
                return ret;
        else if (ret != size)
                return -EIO;

        *val = size == 4 ? be32_to_cpup(val_buf) : be16_to_cpup(val_buf);

        return 0;
}

static int rtq9124_regmap_write(void *context, const void *data, size_t count)
{
        struct i2c_client *i2c = context;
        u8 reg = *(u8 *)data, *vbuf;
        u8 size = rtq9124_get_reg_len(reg);
        __be16 val16 = cpu_to_be16p(data + 1);
        __be32 val32 = cpu_to_be32p(data + 1);

        vbuf = size == 4 ? (u8 *)&val32 : (u8 *)&val16;
        return i2c_smbus_write_i2c_block_data(i2c, reg, size, vbuf);
}

static const struct regmap_config rtq9124_regmap_config = {
        .name                   = "rtq9124",
        .reg_bits               = 8,
        .val_bits               = 32,
        .read                   = rtq9124_regmap_read,
        .write                  = rtq9124_regmap_write,
        .readable_reg           = rtq9124_readable_reg,
        .writeable_reg          = rtq9124_writeable_reg,
        .volatile_reg           = rtq9124_volatile_reg,
        .cache_type             = REGCACHE_MAPLE,
        .num_reg_defaults_raw   = 0xFD + 1,
        .use_single_read        = 1,
        .use_single_write       = 1,
};

static const struct reg_sequence rtq9124_init_regs[] = {
        { 0xfb, 0x0065 },
        { 0x93, 0x2000 },
        { 0xfb, 0x0000 },
};

static int rtq9124_probe(struct i2c_client *i2c)
{
        struct device *dev = &i2c->dev;
        struct rtq9124_priv *rtq9124;
        struct regmap *regmap;
        int ret;

        rtq9124 = devm_kzalloc(dev, sizeof(*rtq9124), GFP_KERNEL);
        if (!rtq9124)
                return -ENOMEM;

        rtq9124->enable = devm_gpiod_get_optional(dev, "enable", GPIOD_OUT_HIGH);
        if (IS_ERR(rtq9124->enable))
                return PTR_ERR(rtq9124->enable);
        else if (rtq9124->enable)
                usleep_range(6000, 7000);
        else
                dev_dbg(dev, "No 'enable' GPIO specified, treat it as default on\n");

        /* Check vendor id information */
        ret = i2c_smbus_read_word_swapped(i2c, RTQ9124_REG_PRJ_CODE);
        if (ret < 0)
                return dev_err_probe(dev, ret, "Failed to read project code\n");
        else if (ret != RTQ9124_FIXED_VENID)
                return dev_err_probe(dev, -ENODEV, "Incorrect project-code 0x%04x\n", ret);

        /* Trigger RG reset before regmap init */
        ret = i2c_smbus_write_word_swapped(i2c, RTQ9124_REG_STATE_CTRL, RTQ9124_MASK_SF_RESET);
        if (ret)
                return dev_err_probe(dev, ret, "Failed to trigger RG reset\n");

        /* Need to wait 10ms for the reset to complete */
        usleep_range(10000, 11000);

        regmap = devm_regmap_init(dev, NULL, i2c, &rtq9124_regmap_config);
        if (IS_ERR(regmap))
                return dev_err_probe(dev, PTR_ERR(regmap), "Failed to init regmap\n");

        ret = regmap_register_patch(regmap, rtq9124_init_regs, ARRAY_SIZE(rtq9124_init_regs));
        if (ret)
                return dev_err_probe(dev, ret, "Failed to register regmap patch\n");

        i2c_set_clientdata(i2c, rtq9124);

        pm_runtime_set_autosuspend_delay(dev, 1000);
        pm_runtime_use_autosuspend(dev);
        pm_runtime_set_active(dev);
        ret = devm_pm_runtime_enable(dev);
        if (ret)
                return dev_err_probe(dev, ret, "Failed to enable pm runtime\n");

        return devm_snd_soc_register_component(dev, &rtq9124_comp_driver, &rtq9124_dai_driver, 1);
}

#ifdef CONFIG_PM
static int rtq9124_runtime_suspend(struct device *dev)
{
        struct rtq9124_priv *rtq9124 = dev_get_drvdata(dev);
        struct regmap *regmap = dev_get_regmap(dev, NULL);

        if (rtq9124->enable) {
                regcache_cache_only(regmap, true);
                regcache_mark_dirty(regmap);
                gpiod_set_value(rtq9124->enable, 0);
        }

        return 0;
}

static int rtq9124_runtime_resume(struct device *dev)
{
        struct rtq9124_priv *rtq9124 = dev_get_drvdata(dev);
        struct regmap *regmap = dev_get_regmap(dev, NULL);
        int ret;

        if (rtq9124->enable) {
                gpiod_set_value(rtq9124->enable, 1);
                usleep_range(6000, 7000);

                regcache_cache_only(regmap, false);
                ret = regcache_sync(regmap);
                if (ret)
                        return ret;
        }

        return 0;
}
#endif

static const struct dev_pm_ops rtq9124_dev_pm_ops = {
        SET_RUNTIME_PM_OPS(rtq9124_runtime_suspend, rtq9124_runtime_resume, NULL)
};

#ifdef CONFIG_OF
static const struct of_device_id rtq9124_device_id[] = {
        { .compatible = "richtek,rtq9124" },
        {}
};
MODULE_DEVICE_TABLE(of, rtq9124_device_id);
#endif

static struct i2c_driver rtq9124_driver = {
        .driver = {
                .name = "rtq9124",
                .of_match_table = of_match_ptr(rtq9124_device_id),
                .pm = pm_ptr(&rtq9124_dev_pm_ops),
        },
        .probe  = rtq9124_probe,
};
module_i2c_driver(rtq9124_driver);

MODULE_AUTHOR("ChiYuan Huang <cy_huang@richtek.com>");
MODULE_DESCRIPTION("ASoC RTQ9124 Driver");
MODULE_LICENSE("GPL");