root/sound/soc/codecs/rt1019.c
// SPDX-License-Identifier: GPL-2.0-only
//
// rt1019.c  --  RT1019 ALSA SoC audio amplifier driver
// Author: Jack Yu <jack.yu@realtek.com>
//
// Copyright(c) 2021 Realtek Semiconductor Corp.
//
//

#include <linux/acpi.h>
#include <linux/fs.h>
#include <linux/module.h>
#include <linux/moduleparam.h>
#include <linux/init.h>
#include <linux/delay.h>
#include <linux/pm.h>
#include <linux/regmap.h>
#include <linux/i2c.h>
#include <linux/platform_device.h>
#include <linux/firmware.h>
#include <sound/core.h>
#include <sound/pcm.h>
#include <sound/pcm_params.h>
#include <sound/soc.h>
#include <sound/soc-dapm.h>
#include <sound/initval.h>
#include <sound/tlv.h>

#include "rl6231.h"
#include "rt1019.h"

static const struct reg_default rt1019_reg[] = {
        { 0x0000, 0x00 },
        { 0x0011, 0x04 },
        { 0x0013, 0x00 },
        { 0x0019, 0x30 },
        { 0x001b, 0x01 },
        { 0x005c, 0x00 },
        { 0x005e, 0x10 },
        { 0x005f, 0xec },
        { 0x0061, 0x10 },
        { 0x0062, 0x19 },
        { 0x0066, 0x08 },
        { 0x0100, 0x80 },
        { 0x0100, 0x51 },
        { 0x0102, 0x23 },
        { 0x0311, 0x00 },
        { 0x0312, 0x3e },
        { 0x0313, 0x86 },
        { 0x0400, 0x03 },
        { 0x0401, 0x02 },
        { 0x0402, 0x01 },
        { 0x0504, 0xff },
        { 0x0505, 0x24 },
        { 0x0b00, 0x50 },
        { 0x0b01, 0xc3 },
};

static bool rt1019_volatile_register(struct device *dev, unsigned int reg)
{
        switch (reg) {
        case RT1019_PWR_STRP_2:
        case RT1019_VER_ID:
        case RT1019_VEND_ID_1:
        case RT1019_VEND_ID_2:
        case RT1019_DEV_ID_1:
        case RT1019_DEV_ID_2:
                return true;

        default:
                return false;
        }
}

static bool rt1019_readable_register(struct device *dev, unsigned int reg)
{
        switch (reg) {
        case RT1019_RESET:
        case RT1019_IDS_CTRL:
        case RT1019_ASEL_CTRL:
        case RT1019_PWR_STRP_2:
        case RT1019_BEEP_TONE:
        case RT1019_VER_ID:
        case RT1019_VEND_ID_1:
        case RT1019_VEND_ID_2:
        case RT1019_DEV_ID_1:
        case RT1019_DEV_ID_2:
        case RT1019_SDB_CTRL:
        case RT1019_CLK_TREE_1:
        case RT1019_CLK_TREE_2:
        case RT1019_CLK_TREE_3:
        case RT1019_PLL_1:
        case RT1019_PLL_2:
        case RT1019_PLL_3:
        case RT1019_TDM_1:
        case RT1019_TDM_2:
        case RT1019_TDM_3:
        case RT1019_DMIX_MONO_1:
        case RT1019_DMIX_MONO_2:
        case RT1019_BEEP_1:
        case RT1019_BEEP_2:
                return true;
        default:
                return false;
        }
}

static const DECLARE_TLV_DB_SCALE(dac_vol_tlv, -9525, 75, 0);

static const char * const rt1019_din_source_select[] = {
        "Left",
        "Right",
        "Left + Right average",
};

static SOC_ENUM_SINGLE_DECL(rt1019_mono_lr_sel, RT1019_IDS_CTRL, 0,
        rt1019_din_source_select);

static const struct snd_kcontrol_new rt1019_snd_controls[] = {
        SOC_SINGLE_TLV("DAC Playback Volume", RT1019_DMIX_MONO_1, 0,
                127, 0, dac_vol_tlv),
        SOC_ENUM("Mono LR Select", rt1019_mono_lr_sel),
};

static int r1019_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);

        switch (event) {
        case SND_SOC_DAPM_PRE_PMU:
                snd_soc_component_write(component, RT1019_SDB_CTRL, 0xb);
                break;
        case SND_SOC_DAPM_POST_PMD:
                snd_soc_component_write(component, RT1019_SDB_CTRL, 0xa);
                break;
        default:
                break;
        }

        return 0;
}

static const struct snd_soc_dapm_widget rt1019_dapm_widgets[] = {
        SND_SOC_DAPM_AIF_IN("AIFRX", "AIF Playback", 0, SND_SOC_NOPM, 0, 0),
        SND_SOC_DAPM_DAC_E("DAC", NULL, SND_SOC_NOPM, 0, 0,
                r1019_dac_event, SND_SOC_DAPM_PRE_PMU | SND_SOC_DAPM_POST_PMD),
        SND_SOC_DAPM_OUTPUT("SPO"),
};

static const struct snd_soc_dapm_route rt1019_dapm_routes[] = {
        { "DAC", NULL, "AIFRX" },
        { "SPO", NULL, "DAC" },
};

static int rt1019_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 rt1019_priv *rt1019 = snd_soc_component_get_drvdata(component);
        int pre_div, bclk_ms, frame_size;
        unsigned int val_len = 0, sys_div_da_filter = 0;
        unsigned int sys_dac_osr = 0, sys_fifo_clk = 0;
        unsigned int sys_clk_cal = 0, sys_asrc_in = 0;

        rt1019->lrck = params_rate(params);
        pre_div = rl6231_get_clk_info(rt1019->sysclk, rt1019->lrck);
        if (pre_div < 0) {
                dev_err(component->dev, "Unsupported clock setting\n");
                return -EINVAL;
        }

        frame_size = snd_soc_params_to_frame_size(params);
        if (frame_size < 0) {
                dev_err(component->dev, "Unsupported frame size: %d\n", frame_size);
                return -EINVAL;
        }

        bclk_ms = frame_size > 32;
        rt1019->bclk = rt1019->lrck * (32 << bclk_ms);

        dev_dbg(dai->dev, "bclk is %dHz and lrck is %dHz\n",
                rt1019->bclk, rt1019->lrck);
        dev_dbg(dai->dev, "bclk_ms is %d and pre_div is %d for iis %d\n",
                                bclk_ms, pre_div, dai->id);

        switch (pre_div) {
        case 0:
                sys_div_da_filter = RT1019_SYS_DIV_DA_FIL_DIV1;
                sys_dac_osr = RT1019_SYS_DA_OSR_DIV1;
                sys_asrc_in = RT1019_ASRC_256FS_DIV1;
                sys_fifo_clk = RT1019_SEL_FIFO_DIV1;
                sys_clk_cal = RT1019_SEL_CLK_CAL_DIV1;
                break;
        case 1:
                sys_div_da_filter = RT1019_SYS_DIV_DA_FIL_DIV2;
                sys_dac_osr = RT1019_SYS_DA_OSR_DIV2;
                sys_asrc_in = RT1019_ASRC_256FS_DIV2;
                sys_fifo_clk = RT1019_SEL_FIFO_DIV2;
                sys_clk_cal = RT1019_SEL_CLK_CAL_DIV2;
                break;
        case 3:
                sys_div_da_filter = RT1019_SYS_DIV_DA_FIL_DIV4;
                sys_dac_osr = RT1019_SYS_DA_OSR_DIV4;
                sys_asrc_in = RT1019_ASRC_256FS_DIV4;
                sys_fifo_clk = RT1019_SEL_FIFO_DIV4;
                sys_clk_cal = RT1019_SEL_CLK_CAL_DIV4;
                break;
        default:
                return -EINVAL;
        }

        switch (params_width(params)) {
        case 16:
                break;
        case 20:
                val_len = RT1019_I2S_DL_20;
                break;
        case 24:
                val_len = RT1019_I2S_DL_24;
                break;
        case 32:
                val_len = RT1019_I2S_DL_32;
                break;
        case 8:
                val_len = RT1019_I2S_DL_8;
                break;
        default:
                return -EINVAL;
        }

        snd_soc_component_update_bits(component, RT1019_TDM_2, RT1019_I2S_DL_MASK,
                        val_len);
        snd_soc_component_update_bits(component, RT1019_CLK_TREE_1,
                        RT1019_SEL_FIFO_MASK, sys_fifo_clk);
        snd_soc_component_update_bits(component, RT1019_CLK_TREE_2,
                        RT1019_SYS_DIV_DA_FIL_MASK | RT1019_SYS_DA_OSR_MASK |
                        RT1019_ASRC_256FS_MASK, sys_div_da_filter | sys_dac_osr |
                        sys_asrc_in);
        snd_soc_component_update_bits(component, RT1019_CLK_TREE_3,
                        RT1019_SEL_CLK_CAL_MASK, sys_clk_cal);

        return 0;
}

static int rt1019_set_dai_fmt(struct snd_soc_dai *dai, unsigned int fmt)
{
        struct snd_soc_component *component = dai->component;
        unsigned int reg_val = 0, reg_val2 = 0;

        switch (fmt & SND_SOC_DAIFMT_INV_MASK) {
        case SND_SOC_DAIFMT_NB_NF:
                break;
        case SND_SOC_DAIFMT_IB_NF:
                reg_val2 |= RT1019_TDM_BCLK_INV;
                break;
        default:
                return -EINVAL;
        }

        switch (fmt & SND_SOC_DAIFMT_FORMAT_MASK) {
        case SND_SOC_DAIFMT_I2S:
                break;

        case SND_SOC_DAIFMT_LEFT_J:
                reg_val |= RT1019_I2S_DF_LEFT;
                break;

        case SND_SOC_DAIFMT_DSP_A:
                reg_val |= RT1019_I2S_DF_PCM_A_R;
                break;

        case SND_SOC_DAIFMT_DSP_B:
                reg_val |= RT1019_I2S_DF_PCM_B_R;
                break;

        default:
                return -EINVAL;
        }

        snd_soc_component_update_bits(component, RT1019_TDM_2,
                RT1019_I2S_DF_MASK, reg_val);
        snd_soc_component_update_bits(component, RT1019_TDM_1,
                RT1019_TDM_BCLK_MASK, reg_val2);

        return 0;
}

static int rt1019_set_dai_sysclk(struct snd_soc_dai *dai,
                int clk_id, unsigned int freq, int dir)
{
        struct snd_soc_component *component = dai->component;
        struct rt1019_priv *rt1019 = snd_soc_component_get_drvdata(component);
        unsigned int reg_val = 0;

        if (freq == rt1019->sysclk && clk_id == rt1019->sysclk_src)
                return 0;

        switch (clk_id) {
        case RT1019_SCLK_S_BCLK:
                reg_val |= RT1019_CLK_SYS_PRE_SEL_BCLK;
                break;

        case RT1019_SCLK_S_PLL:
                reg_val |= RT1019_CLK_SYS_PRE_SEL_PLL;
                break;

        default:
                dev_err(component->dev, "Invalid clock id (%d)\n", clk_id);
                return -EINVAL;
        }

        rt1019->sysclk = freq;
        rt1019->sysclk_src = clk_id;

        dev_dbg(dai->dev, "Sysclk is %dHz and clock id is %d\n", freq, clk_id);

        snd_soc_component_update_bits(component, RT1019_CLK_TREE_1,
                RT1019_CLK_SYS_PRE_SEL_MASK, reg_val);

        return 0;
}

static int rt1019_set_dai_pll(struct snd_soc_dai *dai, int pll_id, int source,
                        unsigned int freq_in, unsigned int freq_out)
{
        struct snd_soc_component *component = dai->component;
        struct rt1019_priv *rt1019 = snd_soc_component_get_drvdata(component);
        struct rl6231_pll_code pll_code;
        int ret;

        if (!freq_in || !freq_out) {
                dev_dbg(component->dev, "PLL disabled\n");
                rt1019->pll_in = 0;
                rt1019->pll_out = 0;
                return 0;
        }

        if (source == rt1019->pll_src && freq_in == rt1019->pll_in &&
                freq_out == rt1019->pll_out)
                return 0;

        switch (source) {
        case RT1019_PLL_S_BCLK:
                snd_soc_component_update_bits(component, RT1019_CLK_TREE_1,
                        RT1019_PLL_SRC_MASK, RT1019_PLL_SRC_SEL_BCLK);
                break;

        case RT1019_PLL_S_RC25M:
                snd_soc_component_update_bits(component, RT1019_CLK_TREE_1,
                        RT1019_PLL_SRC_MASK, RT1019_PLL_SRC_SEL_RC);
                break;

        default:
                dev_err(component->dev, "Unknown PLL source %d\n", source);
                return -EINVAL;
        }

        ret = rl6231_pll_calc(freq_in, freq_out, &pll_code);
        if (ret < 0) {
                dev_err(component->dev, "Unsupported input clock %d\n", freq_in);
                return ret;
        }

        dev_dbg(component->dev, "bypass=%d m=%d n=%d k=%d\n",
                pll_code.m_bp, (pll_code.m_bp ? 0 : pll_code.m_code),
                pll_code.n_code, pll_code.k_code);

        snd_soc_component_update_bits(component, RT1019_PWR_STRP_2,
                RT1019_AUTO_BITS_SEL_MASK | RT1019_AUTO_CLK_SEL_MASK,
                RT1019_AUTO_BITS_SEL_MANU | RT1019_AUTO_CLK_SEL_MANU);
        snd_soc_component_update_bits(component, RT1019_PLL_1,
                RT1019_PLL_M_MASK | RT1019_PLL_M_BP_MASK | RT1019_PLL_Q_8_8_MASK,
                ((pll_code.m_bp ? 0 : pll_code.m_code) << RT1019_PLL_M_SFT) |
                (pll_code.m_bp << RT1019_PLL_M_BP_SFT) |
                ((pll_code.n_code >> 8) & RT1019_PLL_Q_8_8_MASK));
        snd_soc_component_update_bits(component, RT1019_PLL_2,
                RT1019_PLL_Q_7_0_MASK, pll_code.n_code & RT1019_PLL_Q_7_0_MASK);
        snd_soc_component_update_bits(component, RT1019_PLL_3,
                RT1019_PLL_K_MASK, pll_code.k_code);

        rt1019->pll_in = freq_in;
        rt1019->pll_out = freq_out;
        rt1019->pll_src = source;

        return 0;
}

static int rt1019_set_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;
        unsigned int cn = 0, cl = 0, rx_slotnum;
        int ret = 0, first_bit;

        switch (slots) {
        case 4:
                cn = RT1019_I2S_TX_4CH;
                break;
        case 6:
                cn = RT1019_I2S_TX_6CH;
                break;
        case 8:
                cn = RT1019_I2S_TX_8CH;
                break;
        case 2:
                break;
        default:
                return -EINVAL;
        }

        switch (slot_width) {
        case 20:
                cl = RT1019_TDM_CL_20;
                break;
        case 24:
                cl = RT1019_TDM_CL_24;
                break;
        case 32:
                cl = RT1019_TDM_CL_32;
                break;
        case 8:
                cl = RT1019_TDM_CL_8;
                break;
        case 16:
                break;
        default:
                return -EINVAL;
        }

        /* Rx slot configuration */
        rx_slotnum = hweight_long(rx_mask);
        if (rx_slotnum != 1) {
                ret = -EINVAL;
                dev_err(component->dev, "too many rx slots or zero slot\n");
                goto _set_tdm_err_;
        }
        /* This is an assumption that the system sends stereo audio to the
         * amplifier typically. And the stereo audio is placed in slot 0/2/4/6
         * as the starting slot. The users could select the channel from
         * L/R/L+R by "Mono LR Select" control.
         */
        first_bit = __ffs(rx_mask);
        switch (first_bit) {
        case 0:
        case 2:
        case 4:
        case 6:
                snd_soc_component_update_bits(component,
                        RT1019_TDM_3,
                        RT1019_TDM_I2S_TX_L_DAC1_1_MASK |
                        RT1019_TDM_I2S_TX_R_DAC1_1_MASK,
                        (first_bit << RT1019_TDM_I2S_TX_L_DAC1_1_SFT) |
                        ((first_bit + 1) << RT1019_TDM_I2S_TX_R_DAC1_1_SFT));
                break;
        case 1:
        case 3:
        case 5:
        case 7:
                snd_soc_component_update_bits(component,
                        RT1019_TDM_3,
                        RT1019_TDM_I2S_TX_L_DAC1_1_MASK |
                        RT1019_TDM_I2S_TX_R_DAC1_1_MASK,
                        ((first_bit - 1) << RT1019_TDM_I2S_TX_L_DAC1_1_SFT) |
                        (first_bit << RT1019_TDM_I2S_TX_R_DAC1_1_SFT));
                break;
        default:
                ret = -EINVAL;
                goto _set_tdm_err_;
        }

        snd_soc_component_update_bits(component, RT1019_TDM_1,
                RT1019_TDM_CL_MASK, cl);
        snd_soc_component_update_bits(component, RT1019_TDM_2,
                RT1019_I2S_CH_TX_MASK, cn);

_set_tdm_err_:
        return ret;
}

static int rt1019_probe(struct snd_soc_component *component)
{
        struct rt1019_priv *rt1019 = snd_soc_component_get_drvdata(component);

        rt1019->component = component;
        snd_soc_component_write(component, RT1019_SDB_CTRL, 0xa);

        return 0;
}

#define RT1019_STEREO_RATES SNDRV_PCM_RATE_8000_192000
#define RT1019_FORMATS (SNDRV_PCM_FMTBIT_S16_LE | SNDRV_PCM_FMTBIT_S20_3LE | \
                        SNDRV_PCM_FMTBIT_S24_LE | SNDRV_PCM_FMTBIT_S8)

static const struct snd_soc_dai_ops rt1019_aif_dai_ops = {
        .hw_params = rt1019_hw_params,
        .set_fmt = rt1019_set_dai_fmt,
        .set_sysclk = rt1019_set_dai_sysclk,
        .set_pll = rt1019_set_dai_pll,
        .set_tdm_slot = rt1019_set_tdm_slot,
};

static struct snd_soc_dai_driver rt1019_dai[] = {
        {
                .name = "rt1019-aif",
                .id = 0,
                .playback = {
                        .stream_name = "AIF Playback",
                        .channels_min = 1,
                        .channels_max = 2,
                        .rates = RT1019_STEREO_RATES,
                        .formats = RT1019_FORMATS,
                },
                .ops = &rt1019_aif_dai_ops,
        }
};

static const struct snd_soc_component_driver soc_component_dev_rt1019 = {
        .probe                  = rt1019_probe,
        .controls               = rt1019_snd_controls,
        .num_controls           = ARRAY_SIZE(rt1019_snd_controls),
        .dapm_widgets           = rt1019_dapm_widgets,
        .num_dapm_widgets       = ARRAY_SIZE(rt1019_dapm_widgets),
        .dapm_routes            = rt1019_dapm_routes,
        .num_dapm_routes        = ARRAY_SIZE(rt1019_dapm_routes),
        .endianness             = 1,
};

static const struct regmap_config rt1019_regmap = {
        .reg_bits = 16,
        .val_bits = 8,
        .use_single_read = true,
        .use_single_write = true,
        .max_register = RT1019_BEEP_2,
        .volatile_reg = rt1019_volatile_register,
        .readable_reg = rt1019_readable_register,
        .cache_type = REGCACHE_MAPLE,
        .reg_defaults = rt1019_reg,
        .num_reg_defaults = ARRAY_SIZE(rt1019_reg),
};

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

static const struct of_device_id rt1019_of_match[] __maybe_unused = {
        { .compatible = "realtek,rt1019", },
        { }
};
MODULE_DEVICE_TABLE(of, rt1019_of_match);

#ifdef CONFIG_ACPI
static const struct acpi_device_id rt1019_acpi_match[] = {
        { "10EC1019" },
        { }
};
MODULE_DEVICE_TABLE(acpi, rt1019_acpi_match);
#endif

static int rt1019_i2c_probe(struct i2c_client *i2c)
{
        struct rt1019_priv *rt1019;
        int ret;
        unsigned int val, val2, dev_id;

        rt1019 = devm_kzalloc(&i2c->dev, sizeof(struct rt1019_priv),
                                GFP_KERNEL);
        if (!rt1019)
                return -ENOMEM;

        i2c_set_clientdata(i2c, rt1019);

        rt1019->regmap = devm_regmap_init_i2c(i2c, &rt1019_regmap);
        if (IS_ERR(rt1019->regmap)) {
                ret = PTR_ERR(rt1019->regmap);
                dev_err(&i2c->dev, "Failed to allocate register map: %d\n",
                        ret);
                return ret;
        }

        regmap_read(rt1019->regmap, RT1019_DEV_ID_1, &val);
        regmap_read(rt1019->regmap, RT1019_DEV_ID_2, &val2);
        dev_id = val << 8 | val2;
        if (dev_id != RT1019_DEVICE_ID_VAL && dev_id != RT1019_DEVICE_ID_VAL2) {
                dev_err(&i2c->dev,
                        "Device with ID register 0x%x is not rt1019\n", dev_id);
                return -ENODEV;
        }

        return devm_snd_soc_register_component(&i2c->dev,
                &soc_component_dev_rt1019, rt1019_dai, ARRAY_SIZE(rt1019_dai));
}

static struct i2c_driver rt1019_i2c_driver = {
        .driver = {
                .name = "rt1019",
                .of_match_table = of_match_ptr(rt1019_of_match),
                .acpi_match_table = ACPI_PTR(rt1019_acpi_match),
        },
        .probe = rt1019_i2c_probe,
        .id_table = rt1019_i2c_id,
};
module_i2c_driver(rt1019_i2c_driver);

MODULE_DESCRIPTION("ASoC RT1019 driver");
MODULE_AUTHOR("Jack Yu <jack.yu@realtek.com>");
MODULE_LICENSE("GPL v2");