// SPDX-License-Identifier: GPL-2.0-only
//
// rt715-sdca.c -- rt715 ALSA SoC audio driver
//
// Copyright(c) 2020 Realtek Semiconductor Corp.
//
//
//

#include <linux/module.h>
#include <linux/moduleparam.h>
#include <linux/kernel.h>
#include <linux/init.h>
#include <linux/pm_runtime.h>
#include <linux/pm.h>
#include <linux/soundwire/sdw.h>
#include <linux/regmap.h>
#include <linux/slab.h>
#include <linux/platform_device.h>
#include <sound/core.h>
#include <sound/pcm.h>
#include <sound/pcm_params.h>
#include <sound/sdw.h>
#include <sound/soc.h>
#include <sound/soc-dapm.h>
#include <sound/initval.h>
#include <sound/tlv.h>
#include <linux/soundwire/sdw_registers.h>

#include "rt715-sdca.h"

static int rt715_sdca_index_write(struct rt715_sdca_priv *rt715,
                unsigned int nid, unsigned int reg, unsigned int value)
{
        struct regmap *regmap = rt715->mbq_regmap;
        unsigned int addr;
        int ret;

        addr = (nid << 20) | reg;

        ret = regmap_write(regmap, addr, value);
        if (ret < 0)
                dev_err(&rt715->slave->dev,
                        "%s: Failed to set private value: %08x <= %04x %d\n",
                        __func__, addr, value, ret);

        return ret;
}

static int rt715_sdca_index_read(struct rt715_sdca_priv *rt715,
                unsigned int nid, unsigned int reg, unsigned int *value)
{
        struct regmap *regmap = rt715->mbq_regmap;
        unsigned int addr;
        int ret;

        addr = (nid << 20) | reg;

        ret = regmap_read(regmap, addr, value);
        if (ret < 0)
                dev_err(&rt715->slave->dev,
                        "%s: Failed to get private value: %06x => %04x ret=%d\n",
                        __func__, addr, *value, ret);

        return ret;
}

static int rt715_sdca_index_update_bits(struct rt715_sdca_priv *rt715,
        unsigned int nid, unsigned int reg, unsigned int mask, unsigned int val)
{
        unsigned int tmp;
        int ret;

        ret = rt715_sdca_index_read(rt715, nid, reg, &tmp);
        if (ret < 0)
                return ret;

        set_mask_bits(&tmp, mask, val);

        return rt715_sdca_index_write(rt715, nid, reg, tmp);
}

static inline unsigned int rt715_sdca_vol_gain(unsigned int u_ctrl_val,
                unsigned int vol_max, unsigned int vol_gain_sft)
{
        unsigned int val;

        if (u_ctrl_val > vol_max)
                u_ctrl_val = vol_max;
        val = u_ctrl_val;
        u_ctrl_val =
                ((abs(u_ctrl_val - vol_gain_sft) * RT715_SDCA_DB_STEP) << 8) / 1000;
        if (val <= vol_gain_sft) {
                u_ctrl_val = ~u_ctrl_val;
                u_ctrl_val += 1;
        }
        u_ctrl_val &= 0xffff;

        return u_ctrl_val;
}

static inline unsigned int rt715_sdca_boost_gain(unsigned int u_ctrl_val,
                unsigned int b_max, unsigned int b_gain_sft)
{
        if (u_ctrl_val > b_max)
                u_ctrl_val = b_max;

        return (u_ctrl_val * 10) << b_gain_sft;
}

static inline unsigned int rt715_sdca_get_gain(unsigned int reg_val,
                unsigned int gain_sft)
{
        unsigned int neg_flag = 0;

        if (reg_val & BIT(15)) {
                reg_val = ~(reg_val - 1) & 0xffff;
                neg_flag = 1;
        }
        reg_val *= 1000;
        reg_val >>= 8;
        if (neg_flag)
                reg_val = gain_sft - reg_val / RT715_SDCA_DB_STEP;
        else
                reg_val = gain_sft + reg_val / RT715_SDCA_DB_STEP;

        return reg_val;
}

/* SDCA Volume/Boost control */
static int rt715_sdca_set_amp_gain_put(struct snd_kcontrol *kcontrol,
                struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct soc_mixer_control *mc =
                (struct soc_mixer_control *)kcontrol->private_value;
        struct rt715_sdca_priv *rt715 = snd_soc_component_get_drvdata(component);
        unsigned int gain_val, i, k_changed = 0;
        int ret;

        for (i = 0; i < 2; i++) {
                if (ucontrol->value.integer.value[i] != rt715->kctl_2ch_orig[i]) {
                        k_changed = 1;
                        break;
                }
        }

        for (i = 0; i < 2; i++) {
                rt715->kctl_2ch_orig[i] = ucontrol->value.integer.value[i];
                gain_val =
                        rt715_sdca_vol_gain(ucontrol->value.integer.value[i], mc->max,
                                mc->shift);
                ret = regmap_write(rt715->mbq_regmap, mc->reg + i, gain_val);
                if (ret != 0) {
                        dev_err(component->dev, "%s: Failed to write 0x%x=0x%x\n",
                                __func__, mc->reg + i, gain_val);
                        return ret;
                }
        }

        return k_changed;
}

static int rt715_sdca_set_amp_gain_4ch_put(struct snd_kcontrol *kcontrol,
                struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct rt715_sdca_priv *rt715 = snd_soc_component_get_drvdata(component);
        struct rt715_sdca_kcontrol_private *p =
                (struct rt715_sdca_kcontrol_private *)kcontrol->private_value;
        unsigned int reg_base = p->reg_base, k_changed = 0;
        const unsigned int gain_sft = 0x2f;
        unsigned int gain_val, i;
        int ret;

        for (i = 0; i < 4; i++) {
                if (ucontrol->value.integer.value[i] != rt715->kctl_4ch_orig[i]) {
                        k_changed = 1;
                        break;
                }
        }

        for (i = 0; i < 4; i++) {
                rt715->kctl_4ch_orig[i] = ucontrol->value.integer.value[i];
                gain_val =
                        rt715_sdca_vol_gain(ucontrol->value.integer.value[i], p->max,
                                gain_sft);
                ret = regmap_write(rt715->mbq_regmap, reg_base + i,
                                gain_val);
                if (ret != 0) {
                        dev_err(component->dev, "%s: Failed to write 0x%x=0x%x\n",
                                __func__, reg_base + i, gain_val);
                        return ret;
                }
        }

        return k_changed;
}

static int rt715_sdca_set_amp_gain_8ch_put(struct snd_kcontrol *kcontrol,
                struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct rt715_sdca_priv *rt715 = snd_soc_component_get_drvdata(component);
        struct rt715_sdca_kcontrol_private *p =
                (struct rt715_sdca_kcontrol_private *)kcontrol->private_value;
        unsigned int reg_base = p->reg_base, i, k_changed = 0;
        const unsigned int gain_sft = 8;
        unsigned int gain_val, reg;
        int ret;

        for (i = 0; i < 8; i++) {
                if (ucontrol->value.integer.value[i] != rt715->kctl_8ch_orig[i]) {
                        k_changed = 1;
                        break;
                }
        }

        for (i = 0; i < 8; i++) {
                rt715->kctl_8ch_orig[i] = ucontrol->value.integer.value[i];
                gain_val =
                        rt715_sdca_boost_gain(ucontrol->value.integer.value[i], p->max,
                                gain_sft);
                reg = i < 7 ? reg_base + i : (reg_base - 1) | BIT(15);
                ret = regmap_write(rt715->mbq_regmap, reg, gain_val);
                if (ret != 0) {
                        dev_err(component->dev, "%s: Failed to write 0x%x=0x%x\n",
                                __func__, reg, gain_val);
                        return ret;
                }
        }

        return k_changed;
}

static int rt715_sdca_set_amp_gain_get(struct snd_kcontrol *kcontrol,
                struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct soc_mixer_control *mc =
                (struct soc_mixer_control *)kcontrol->private_value;
        struct rt715_sdca_priv *rt715 = snd_soc_component_get_drvdata(component);
        unsigned int val, i;
        int ret;

        for (i = 0; i < 2; i++) {
                ret = regmap_read(rt715->mbq_regmap, mc->reg + i, &val);
                if (ret < 0) {
                        dev_err(component->dev, "%s: Failed to read 0x%x, ret=%d\n",
                                __func__, mc->reg + i, ret);
                        return ret;
                }
                ucontrol->value.integer.value[i] = rt715_sdca_get_gain(val, mc->shift);
        }

        return 0;
}

static int rt715_sdca_set_amp_gain_4ch_get(struct snd_kcontrol *kcontrol,
                struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct rt715_sdca_priv *rt715 = snd_soc_component_get_drvdata(component);
        struct rt715_sdca_kcontrol_private *p =
                (struct rt715_sdca_kcontrol_private *)kcontrol->private_value;
        unsigned int reg_base = p->reg_base, i;
        const unsigned int gain_sft = 0x2f;
        unsigned int val;
        int ret;

        for (i = 0; i < 4; i++) {
                ret = regmap_read(rt715->mbq_regmap, reg_base + i, &val);
                if (ret < 0) {
                        dev_err(component->dev, "%s: Failed to read 0x%x, ret=%d\n",
                                __func__, reg_base + i, ret);
                        return ret;
                }
                ucontrol->value.integer.value[i] = rt715_sdca_get_gain(val, gain_sft);
        }

        return 0;
}

static int rt715_sdca_set_amp_gain_8ch_get(struct snd_kcontrol *kcontrol,
                struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct rt715_sdca_priv *rt715 = snd_soc_component_get_drvdata(component);
        struct rt715_sdca_kcontrol_private *p =
                (struct rt715_sdca_kcontrol_private *)kcontrol->private_value;
        unsigned int reg_base = p->reg_base;
        const unsigned int gain_sft = 8;
        unsigned int val_l, val_r;
        unsigned int i, reg;
        int ret;

        for (i = 0; i < 8; i += 2) {
                ret = regmap_read(rt715->mbq_regmap, reg_base + i, &val_l);
                if (ret < 0) {
                        dev_err(component->dev, "%s: Failed to read 0x%x, ret=%d\n",
                                __func__, reg_base + i, ret);
                        return ret;
                }
                ucontrol->value.integer.value[i] = (val_l >> gain_sft) / 10;

                reg = (i == 6) ? (reg_base - 1) | BIT(15) : reg_base + 1 + i;
                ret = regmap_read(rt715->mbq_regmap, reg, &val_r);
                if (ret < 0) {
                        dev_err(component->dev, "%s: Failed to read 0x%x, ret=%d\n",
                                __func__, reg, ret);
                        return ret;
                }
                ucontrol->value.integer.value[i + 1] = (val_r >> gain_sft) / 10;
        }

        return 0;
}

static const DECLARE_TLV_DB_SCALE(in_vol_tlv, -1725, 75, 0);
static const DECLARE_TLV_DB_SCALE(mic_vol_tlv, 0, 1000, 0);

static int rt715_sdca_get_volsw(struct snd_kcontrol *kcontrol,
        struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct rt715_sdca_kcontrol_private *p =
                (struct rt715_sdca_kcontrol_private *)kcontrol->private_value;
        unsigned int reg_base = p->reg_base;
        unsigned int invert = p->invert, i;
        int val;

        for (i = 0; i < p->count; i += 2) {
                val = snd_soc_component_read(component, reg_base + i);
                if (val < 0)
                        return -EINVAL;
                ucontrol->value.integer.value[i] = invert ? p->max - val : val;

                val = snd_soc_component_read(component, reg_base + 1 + i);
                if (val < 0)
                        return -EINVAL;
                ucontrol->value.integer.value[i + 1] =
                        invert ? p->max - val : val;
        }

        return 0;
}

static int rt715_sdca_put_volsw(struct snd_kcontrol *kcontrol,
        struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct rt715_sdca_priv *rt715 = snd_soc_component_get_drvdata(component);
        struct rt715_sdca_kcontrol_private *p =
                (struct rt715_sdca_kcontrol_private *)kcontrol->private_value;
        unsigned int val[4] = {0}, val_mask, i, k_changed = 0;
        unsigned int reg = p->reg_base;
        unsigned int shift = p->shift;
        unsigned int max = p->max;
        unsigned int mask = (1 << fls(max)) - 1;
        unsigned int invert = p->invert;
        int err;

        for (i = 0; i < 4; i++) {
                if (ucontrol->value.integer.value[i] != rt715->kctl_switch_orig[i]) {
                        k_changed = 1;
                        break;
                }
        }

        for (i = 0; i < 2; i++) {
                rt715->kctl_switch_orig[i * 2] = ucontrol->value.integer.value[i * 2];
                val[i * 2] = ucontrol->value.integer.value[i * 2] & mask;
                if (invert)
                        val[i * 2] = max - val[i * 2];
                val_mask = mask << shift;
                val[i * 2] <<= shift;

                rt715->kctl_switch_orig[i * 2 + 1] =
                        ucontrol->value.integer.value[i * 2 + 1];
                val[i * 2 + 1] =
                        ucontrol->value.integer.value[i * 2 + 1] & mask;
                if (invert)
                        val[i * 2 + 1] = max - val[i * 2 + 1];

                val[i * 2 + 1] <<=  shift;

                err = snd_soc_component_update_bits(component, reg + i * 2, val_mask,
                                val[i * 2]);
                if (err < 0)
                        return err;

                err = snd_soc_component_update_bits(component, reg + 1 + i * 2,
                        val_mask, val[i * 2 + 1]);
                if (err < 0)
                        return err;
        }

        return k_changed;
}

static int rt715_sdca_fu_info(struct snd_kcontrol *kcontrol,
        struct snd_ctl_elem_info *uinfo)
{
        struct rt715_sdca_kcontrol_private *p =
                (struct rt715_sdca_kcontrol_private *)kcontrol->private_value;

        if (p->max == 1)
                uinfo->type = SNDRV_CTL_ELEM_TYPE_BOOLEAN;
        else
                uinfo->type = SNDRV_CTL_ELEM_TYPE_INTEGER;
        uinfo->count = p->count;
        uinfo->value.integer.min = 0;
        uinfo->value.integer.max = p->max;
        return 0;
}

#define RT715_SDCA_PR_VALUE(xreg_base, xcount, xmax, xshift, xinvert) \
        ((unsigned long)&(struct rt715_sdca_kcontrol_private) \
                {.reg_base = xreg_base, .count = xcount, .max = xmax, \
                .shift = xshift, .invert = xinvert})

#define RT715_SDCA_FU_CTRL(xname, reg_base, xshift, xmax, xinvert, xcount) \
{       .iface = SNDRV_CTL_ELEM_IFACE_MIXER, .name = (xname), \
        .info = rt715_sdca_fu_info, \
        .get = rt715_sdca_get_volsw, \
        .put = rt715_sdca_put_volsw, \
        .private_value = RT715_SDCA_PR_VALUE(reg_base, xcount, xmax, \
                                        xshift, xinvert)}

#define RT715_SDCA_EXT_TLV(xname, reg_base, xhandler_get,\
         xhandler_put, tlv_array, xcount, xmax) \
{       .iface = SNDRV_CTL_ELEM_IFACE_MIXER, .name = (xname), \
        .access = SNDRV_CTL_ELEM_ACCESS_TLV_READ | \
                 SNDRV_CTL_ELEM_ACCESS_READWRITE, \
        .tlv.p = (tlv_array), \
        .info = rt715_sdca_fu_info, \
        .get = xhandler_get, .put = xhandler_put, \
        .private_value = RT715_SDCA_PR_VALUE(reg_base, xcount, xmax, 0, 0) }

#define RT715_SDCA_BOOST_EXT_TLV(xname, reg_base, xhandler_get,\
         xhandler_put, tlv_array, xcount, xmax) \
{       .iface = SNDRV_CTL_ELEM_IFACE_MIXER, .name = (xname), \
        .access = SNDRV_CTL_ELEM_ACCESS_TLV_READ | \
                 SNDRV_CTL_ELEM_ACCESS_READWRITE, \
        .tlv.p = (tlv_array), \
        .info = rt715_sdca_fu_info, \
        .get = xhandler_get, .put = xhandler_put, \
        .private_value = RT715_SDCA_PR_VALUE(reg_base, xcount, xmax, 0, 0) }

static const struct snd_kcontrol_new rt715_sdca_snd_controls[] = {
        /* Capture switch */
        SOC_DOUBLE_R("FU0A Capture Switch",
                SDW_SDCA_CTL(FUN_MIC_ARRAY, RT715_SDCA_FU_ADC7_27_VOL,
                        RT715_SDCA_FU_MUTE_CTRL, CH_01),
                SDW_SDCA_CTL(FUN_MIC_ARRAY, RT715_SDCA_FU_ADC7_27_VOL,
                        RT715_SDCA_FU_MUTE_CTRL, CH_02),
                        0, 1, 1),
        RT715_SDCA_FU_CTRL("FU02 Capture Switch",
                SDW_SDCA_CTL(FUN_MIC_ARRAY, RT715_SDCA_FU_ADC8_9_VOL,
                        RT715_SDCA_FU_MUTE_CTRL, CH_01),
                        0, 1, 1, 4),
        RT715_SDCA_FU_CTRL("FU06 Capture Switch",
                SDW_SDCA_CTL(FUN_MIC_ARRAY, RT715_SDCA_FU_ADC10_11_VOL,
                        RT715_SDCA_FU_MUTE_CTRL, CH_01),
                        0, 1, 1, 4),
        /* Volume Control */
        SOC_DOUBLE_R_EXT_TLV("FU0A Capture Volume",
                SDW_SDCA_CTL(FUN_MIC_ARRAY, RT715_SDCA_FU_ADC7_27_VOL,
                        RT715_SDCA_FU_VOL_CTRL, CH_01),
                SDW_SDCA_CTL(FUN_MIC_ARRAY, RT715_SDCA_FU_ADC7_27_VOL,
                        RT715_SDCA_FU_VOL_CTRL, CH_02),
                        0x2f, 0x3f, 0,
                rt715_sdca_set_amp_gain_get, rt715_sdca_set_amp_gain_put,
                in_vol_tlv),
        RT715_SDCA_EXT_TLV("FU02 Capture Volume",
                SDW_SDCA_CTL(FUN_MIC_ARRAY, RT715_SDCA_FU_ADC8_9_VOL,
                        RT715_SDCA_FU_VOL_CTRL, CH_01),
                rt715_sdca_set_amp_gain_4ch_get,
                rt715_sdca_set_amp_gain_4ch_put,
                in_vol_tlv, 4, 0x3f),
        RT715_SDCA_EXT_TLV("FU06 Capture Volume",
                SDW_SDCA_CTL(FUN_MIC_ARRAY, RT715_SDCA_FU_ADC10_11_VOL,
                        RT715_SDCA_FU_VOL_CTRL, CH_01),
                rt715_sdca_set_amp_gain_4ch_get,
                rt715_sdca_set_amp_gain_4ch_put,
                in_vol_tlv, 4, 0x3f),
        /* MIC Boost Control */
        RT715_SDCA_BOOST_EXT_TLV("FU0E Boost",
                SDW_SDCA_CTL(FUN_MIC_ARRAY, RT715_SDCA_FU_DMIC_GAIN_EN,
                        RT715_SDCA_FU_DMIC_GAIN_CTRL, CH_01),
                        rt715_sdca_set_amp_gain_8ch_get,
                        rt715_sdca_set_amp_gain_8ch_put,
                        mic_vol_tlv, 8, 3),
        RT715_SDCA_BOOST_EXT_TLV("FU0C Boost",
                SDW_SDCA_CTL(FUN_MIC_ARRAY, RT715_SDCA_FU_AMIC_GAIN_EN,
                        RT715_SDCA_FU_DMIC_GAIN_CTRL, CH_01),
                        rt715_sdca_set_amp_gain_8ch_get,
                        rt715_sdca_set_amp_gain_8ch_put,
                        mic_vol_tlv, 8, 3),
};

static int rt715_sdca_mux_get(struct snd_kcontrol *kcontrol,
                        struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_soc_dapm_kcontrol_to_component(kcontrol);
        struct rt715_sdca_priv *rt715 = snd_soc_component_get_drvdata(component);
        unsigned int val, mask_sft;

        if (strstr(ucontrol->id.name, "ADC 22 Mux"))
                mask_sft = 12;
        else if (strstr(ucontrol->id.name, "ADC 23 Mux"))
                mask_sft = 8;
        else if (strstr(ucontrol->id.name, "ADC 24 Mux"))
                mask_sft = 4;
        else if (strstr(ucontrol->id.name, "ADC 25 Mux"))
                mask_sft = 0;
        else
                return -EINVAL;

        rt715_sdca_index_read(rt715, RT715_VENDOR_HDA_CTL,
                RT715_HDA_LEGACY_MUX_CTL1, &val);
        val = (val >> mask_sft) & 0xf;

        /*
         * The first two indices of ADC Mux 24/25 are routed to the same
         * hardware source. ie, ADC Mux 24 0/1 will both connect to MIC2.
         * To have a unique set of inputs, we skip the index1 of the muxes.
         */
        if ((strstr(ucontrol->id.name, "ADC 24 Mux") ||
                strstr(ucontrol->id.name, "ADC 25 Mux")) && val > 0)
                val -= 1;
        ucontrol->value.enumerated.item[0] = val;

        return 0;
}

static int rt715_sdca_mux_put(struct snd_kcontrol *kcontrol,
                        struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_soc_dapm_kcontrol_to_component(kcontrol);
        struct snd_soc_dapm_context *dapm = snd_soc_dapm_kcontrol_to_dapm(kcontrol);
        struct rt715_sdca_priv *rt715 = snd_soc_component_get_drvdata(component);
        struct soc_enum *e = (struct soc_enum *)kcontrol->private_value;
        unsigned int *item = ucontrol->value.enumerated.item;
        unsigned int val, val2 = 0, change, mask_sft;

        if (item[0] >= e->items)
                return -EINVAL;

        if (strstr(ucontrol->id.name, "ADC 22 Mux"))
                mask_sft = 12;
        else if (strstr(ucontrol->id.name, "ADC 23 Mux"))
                mask_sft = 8;
        else if (strstr(ucontrol->id.name, "ADC 24 Mux"))
                mask_sft = 4;
        else if (strstr(ucontrol->id.name, "ADC 25 Mux"))
                mask_sft = 0;
        else
                return -EINVAL;

        /* Verb ID = 0x701h, nid = e->reg */
        val = snd_soc_enum_item_to_val(e, item[0]) << e->shift_l;

        rt715_sdca_index_read(rt715, RT715_VENDOR_HDA_CTL,
                RT715_HDA_LEGACY_MUX_CTL1, &val2);
        val2 = (val2 >> mask_sft) & 0xf;

        change = val != val2;

        if (change)
                rt715_sdca_index_update_bits(rt715, RT715_VENDOR_HDA_CTL,
                        RT715_HDA_LEGACY_MUX_CTL1, 0xf << mask_sft, val << mask_sft);

        snd_soc_dapm_mux_update_power(dapm, kcontrol, item[0], e, NULL);

        return change;
}

static const char * const adc_22_23_mux_text[] = {
        "MIC1",
        "MIC2",
        "LINE1",
        "LINE2",
        "DMIC1",
        "DMIC2",
        "DMIC3",
        "DMIC4",
};

/*
 * Due to mux design for nid 24 (MUX_IN3)/25 (MUX_IN4), connection index 0 and
 * 1 will be connected to the same dmic source, therefore we skip index 1 to
 * avoid misunderstanding on usage of dapm routing.
 */
static int rt715_adc_24_25_values[] = {
        0,
        2,
        3,
        4,
        5,
};

static const char * const adc_24_mux_text[] = {
        "MIC2",
        "DMIC1",
        "DMIC2",
        "DMIC3",
        "DMIC4",
};

static const char * const adc_25_mux_text[] = {
        "MIC1",
        "DMIC1",
        "DMIC2",
        "DMIC3",
        "DMIC4",
};

static SOC_ENUM_SINGLE_DECL(rt715_adc22_enum, SND_SOC_NOPM, 0,
        adc_22_23_mux_text);

static SOC_ENUM_SINGLE_DECL(rt715_adc23_enum, SND_SOC_NOPM, 0,
        adc_22_23_mux_text);

static SOC_VALUE_ENUM_SINGLE_DECL(rt715_adc24_enum,
        SND_SOC_NOPM, 0, 0xf,
        adc_24_mux_text, rt715_adc_24_25_values);
static SOC_VALUE_ENUM_SINGLE_DECL(rt715_adc25_enum,
        SND_SOC_NOPM, 0, 0xf,
        adc_25_mux_text, rt715_adc_24_25_values);

static const struct snd_kcontrol_new rt715_adc22_mux =
        SOC_DAPM_ENUM_EXT("ADC 22 Mux", rt715_adc22_enum,
                        rt715_sdca_mux_get, rt715_sdca_mux_put);

static const struct snd_kcontrol_new rt715_adc23_mux =
        SOC_DAPM_ENUM_EXT("ADC 23 Mux", rt715_adc23_enum,
                        rt715_sdca_mux_get, rt715_sdca_mux_put);

static const struct snd_kcontrol_new rt715_adc24_mux =
        SOC_DAPM_ENUM_EXT("ADC 24 Mux", rt715_adc24_enum,
                        rt715_sdca_mux_get, rt715_sdca_mux_put);

static const struct snd_kcontrol_new rt715_adc25_mux =
        SOC_DAPM_ENUM_EXT("ADC 25 Mux", rt715_adc25_enum,
                        rt715_sdca_mux_get, rt715_sdca_mux_put);

static int rt715_sdca_pde23_24_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 rt715_sdca_priv *rt715 = snd_soc_component_get_drvdata(component);

        switch (event) {
        case SND_SOC_DAPM_POST_PMU:
                regmap_write(rt715->regmap,
                        SDW_SDCA_CTL(FUN_MIC_ARRAY, RT715_SDCA_CREQ_POW_EN,
                                RT715_SDCA_REQ_POW_CTRL,
                                CH_00), 0x00);
                break;
        case SND_SOC_DAPM_PRE_PMD:
                regmap_write(rt715->regmap,
                        SDW_SDCA_CTL(FUN_MIC_ARRAY, RT715_SDCA_CREQ_POW_EN,
                                RT715_SDCA_REQ_POW_CTRL,
                                CH_00), 0x03);
                break;
        }
        return 0;
}

static const struct snd_soc_dapm_widget rt715_sdca_dapm_widgets[] = {
        SND_SOC_DAPM_INPUT("DMIC1"),
        SND_SOC_DAPM_INPUT("DMIC2"),
        SND_SOC_DAPM_INPUT("DMIC3"),
        SND_SOC_DAPM_INPUT("DMIC4"),
        SND_SOC_DAPM_INPUT("MIC1"),
        SND_SOC_DAPM_INPUT("MIC2"),
        SND_SOC_DAPM_INPUT("LINE1"),
        SND_SOC_DAPM_INPUT("LINE2"),

        SND_SOC_DAPM_SUPPLY("PDE23_24", SND_SOC_NOPM, 0, 0,
                rt715_sdca_pde23_24_event,
                SND_SOC_DAPM_POST_PMU | SND_SOC_DAPM_PRE_PMD),

        SND_SOC_DAPM_ADC("ADC 07", NULL, SND_SOC_NOPM, 4, 0),
        SND_SOC_DAPM_ADC("ADC 08", NULL, SND_SOC_NOPM, 4, 0),
        SND_SOC_DAPM_ADC("ADC 09", NULL, SND_SOC_NOPM, 4, 0),
        SND_SOC_DAPM_ADC("ADC 27", NULL, SND_SOC_NOPM, 4, 0),
        SND_SOC_DAPM_MUX("ADC 22 Mux", SND_SOC_NOPM, 0, 0,
                &rt715_adc22_mux),
        SND_SOC_DAPM_MUX("ADC 23 Mux", SND_SOC_NOPM, 0, 0,
                &rt715_adc23_mux),
        SND_SOC_DAPM_MUX("ADC 24 Mux", SND_SOC_NOPM, 0, 0,
                &rt715_adc24_mux),
        SND_SOC_DAPM_MUX("ADC 25 Mux", SND_SOC_NOPM, 0, 0,
                &rt715_adc25_mux),
        SND_SOC_DAPM_AIF_OUT("DP4TX", "DP4 Capture", 0, SND_SOC_NOPM, 0, 0),
        SND_SOC_DAPM_AIF_OUT("DP6TX", "DP6 Capture", 0, SND_SOC_NOPM, 0, 0),
};

static const struct snd_soc_dapm_route rt715_sdca_audio_map[] = {
        {"DP6TX", NULL, "ADC 09"},
        {"DP6TX", NULL, "ADC 08"},
        {"DP4TX", NULL, "ADC 07"},
        {"DP4TX", NULL, "ADC 27"},
        {"DP4TX", NULL, "ADC 09"},
        {"DP4TX", NULL, "ADC 08"},

        {"LINE1", NULL, "PDE23_24"},
        {"LINE2", NULL, "PDE23_24"},
        {"MIC1", NULL, "PDE23_24"},
        {"MIC2", NULL, "PDE23_24"},
        {"DMIC1", NULL, "PDE23_24"},
        {"DMIC2", NULL, "PDE23_24"},
        {"DMIC3", NULL, "PDE23_24"},
        {"DMIC4", NULL, "PDE23_24"},

        {"ADC 09", NULL, "ADC 22 Mux"},
        {"ADC 08", NULL, "ADC 23 Mux"},
        {"ADC 07", NULL, "ADC 24 Mux"},
        {"ADC 27", NULL, "ADC 25 Mux"},
        {"ADC 22 Mux", "MIC1", "MIC1"},
        {"ADC 22 Mux", "MIC2", "MIC2"},
        {"ADC 22 Mux", "LINE1", "LINE1"},
        {"ADC 22 Mux", "LINE2", "LINE2"},
        {"ADC 22 Mux", "DMIC1", "DMIC1"},
        {"ADC 22 Mux", "DMIC2", "DMIC2"},
        {"ADC 22 Mux", "DMIC3", "DMIC3"},
        {"ADC 22 Mux", "DMIC4", "DMIC4"},
        {"ADC 23 Mux", "MIC1", "MIC1"},
        {"ADC 23 Mux", "MIC2", "MIC2"},
        {"ADC 23 Mux", "LINE1", "LINE1"},
        {"ADC 23 Mux", "LINE2", "LINE2"},
        {"ADC 23 Mux", "DMIC1", "DMIC1"},
        {"ADC 23 Mux", "DMIC2", "DMIC2"},
        {"ADC 23 Mux", "DMIC3", "DMIC3"},
        {"ADC 23 Mux", "DMIC4", "DMIC4"},
        {"ADC 24 Mux", "MIC2", "MIC2"},
        {"ADC 24 Mux", "DMIC1", "DMIC1"},
        {"ADC 24 Mux", "DMIC2", "DMIC2"},
        {"ADC 24 Mux", "DMIC3", "DMIC3"},
        {"ADC 24 Mux", "DMIC4", "DMIC4"},
        {"ADC 25 Mux", "MIC1", "MIC1"},
        {"ADC 25 Mux", "DMIC1", "DMIC1"},
        {"ADC 25 Mux", "DMIC2", "DMIC2"},
        {"ADC 25 Mux", "DMIC3", "DMIC3"},
        {"ADC 25 Mux", "DMIC4", "DMIC4"},
};

static int rt715_sdca_probe(struct snd_soc_component *component)
{
        struct rt715_sdca_priv *rt715 = snd_soc_component_get_drvdata(component);
        int ret;

        if (!rt715->first_hw_init)
                return 0;

        ret = pm_runtime_resume(component->dev);
        if (ret < 0 && ret != -EACCES)
                return ret;

        return 0;
}

static const struct snd_soc_component_driver soc_codec_dev_rt715_sdca = {
        .probe = rt715_sdca_probe,
        .controls = rt715_sdca_snd_controls,
        .num_controls = ARRAY_SIZE(rt715_sdca_snd_controls),
        .dapm_widgets = rt715_sdca_dapm_widgets,
        .num_dapm_widgets = ARRAY_SIZE(rt715_sdca_dapm_widgets),
        .dapm_routes = rt715_sdca_audio_map,
        .num_dapm_routes = ARRAY_SIZE(rt715_sdca_audio_map),
        .endianness = 1,
};

static int rt715_sdca_set_sdw_stream(struct snd_soc_dai *dai, void *sdw_stream,
                                int direction)
{
        snd_soc_dai_dma_data_set(dai, direction, sdw_stream);

        return 0;
}

static void rt715_sdca_shutdown(struct snd_pcm_substream *substream,
                                struct snd_soc_dai *dai)

{
        snd_soc_dai_set_dma_data(dai, substream, NULL);
}

static int rt715_sdca_pcm_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 rt715_sdca_priv *rt715 = snd_soc_component_get_drvdata(component);
        struct sdw_stream_config stream_config = {0};
        struct sdw_port_config port_config = {0};
        struct sdw_stream_runtime *sdw_stream;
        int retval;
        unsigned int val;

        sdw_stream = snd_soc_dai_get_dma_data(dai, substream);

        if (!sdw_stream)
                return -EINVAL;

        if (!rt715->slave)
                return -EINVAL;

        snd_sdw_params_to_config(substream, params, &stream_config, &port_config);

        switch (dai->id) {
        case RT715_AIF1:
                port_config.num = 6;
                rt715_sdca_index_write(rt715, RT715_VENDOR_REG, RT715_SDW_INPUT_SEL,
                        0xa500);
                break;
        case RT715_AIF2:
                port_config.num = 4;
                rt715_sdca_index_write(rt715, RT715_VENDOR_REG, RT715_SDW_INPUT_SEL,
                        0xaf00);
                break;
        default:
                dev_err(component->dev, "%s: Invalid DAI id %d\n", __func__, dai->id);
                return -EINVAL;
        }

        retval = sdw_stream_add_slave(rt715->slave, &stream_config,
                                        &port_config, 1, sdw_stream);
        if (retval) {
                dev_err(component->dev, "%s: Unable to configure port, retval:%d\n",
                        __func__, retval);
                return retval;
        }

        switch (params_rate(params)) {
        case 8000:
                val = 0x1;
                break;
        case 11025:
                val = 0x2;
                break;
        case 12000:
                val = 0x3;
                break;
        case 16000:
                val = 0x4;
                break;
        case 22050:
                val = 0x5;
                break;
        case 24000:
                val = 0x6;
                break;
        case 32000:
                val = 0x7;
                break;
        case 44100:
                val = 0x8;
                break;
        case 48000:
                val = 0x9;
                break;
        case 88200:
                val = 0xa;
                break;
        case 96000:
                val = 0xb;
                break;
        case 176400:
                val = 0xc;
                break;
        case 192000:
                val = 0xd;
                break;
        case 384000:
                val = 0xe;
                break;
        case 768000:
                val = 0xf;
                break;
        default:
                dev_err(component->dev, "%s: Unsupported sample rate %d\n",
                        __func__, params_rate(params));
                return -EINVAL;
        }

        regmap_write(rt715->regmap,
                SDW_SDCA_CTL(FUN_MIC_ARRAY, RT715_SDCA_CS_FREQ_IND_EN,
                        RT715_SDCA_FREQ_IND_CTRL, CH_00), val);

        return 0;
}

static int rt715_sdca_pcm_hw_free(struct snd_pcm_substream *substream,
                                struct snd_soc_dai *dai)
{
        struct snd_soc_component *component = dai->component;
        struct rt715_sdca_priv *rt715 = snd_soc_component_get_drvdata(component);
        struct sdw_stream_runtime *sdw_stream =
                snd_soc_dai_get_dma_data(dai, substream);

        if (!rt715->slave)
                return -EINVAL;

        sdw_stream_remove_slave(rt715->slave, sdw_stream);
        return 0;
}

#define RT715_STEREO_RATES (SNDRV_PCM_RATE_44100 | SNDRV_PCM_RATE_48000)
#define RT715_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 rt715_sdca_ops = {
        .hw_params      = rt715_sdca_pcm_hw_params,
        .hw_free        = rt715_sdca_pcm_hw_free,
        .set_stream     = rt715_sdca_set_sdw_stream,
        .shutdown       = rt715_sdca_shutdown,
};

static struct snd_soc_dai_driver rt715_sdca_dai[] = {
        {
                .name = "rt715-sdca-aif1",
                .id = RT715_AIF1,
                .capture = {
                        .stream_name = "DP6 Capture",
                        .channels_min = 1,
                        .channels_max = 2,
                        .rates = RT715_STEREO_RATES,
                        .formats = RT715_FORMATS,
                },
                .ops = &rt715_sdca_ops,
        },
        {
                .name = "rt715-sdca-aif2",
                .id = RT715_AIF2,
                .capture = {
                        .stream_name = "DP4 Capture",
                        .channels_min = 1,
                        .channels_max = 2,
                        .rates = RT715_STEREO_RATES,
                        .formats = RT715_FORMATS,
                },
                .ops = &rt715_sdca_ops,
        },
};

/* Bus clock frequency */
#define RT715_CLK_FREQ_9600000HZ 9600000
#define RT715_CLK_FREQ_12000000HZ 12000000
#define RT715_CLK_FREQ_6000000HZ 6000000
#define RT715_CLK_FREQ_4800000HZ 4800000
#define RT715_CLK_FREQ_2400000HZ 2400000
#define RT715_CLK_FREQ_12288000HZ 12288000

int rt715_sdca_init(struct device *dev, struct regmap *mbq_regmap,
        struct regmap *regmap, struct sdw_slave *slave)
{
        struct rt715_sdca_priv *rt715;
        int ret;

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

        dev_set_drvdata(dev, rt715);
        rt715->slave = slave;
        rt715->regmap = regmap;
        rt715->mbq_regmap = mbq_regmap;
        rt715->hw_sdw_ver = slave->id.sdw_version;

        regcache_cache_only(rt715->regmap, true);
        regcache_cache_only(rt715->mbq_regmap, true);

        /*
         * Mark hw_init to false
         * HW init will be performed when device reports present
         */
        rt715->hw_init = false;
        rt715->first_hw_init = false;

        ret = devm_snd_soc_register_component(dev,
                        &soc_codec_dev_rt715_sdca,
                        rt715_sdca_dai,
                        ARRAY_SIZE(rt715_sdca_dai));
        if (ret < 0)
                return ret;

        /* set autosuspend parameters */
        pm_runtime_set_autosuspend_delay(dev, 3000);
        pm_runtime_use_autosuspend(dev);

        /* make sure the device does not suspend immediately */
        pm_runtime_mark_last_busy(dev);

        pm_runtime_enable(dev);

        /* important note: the device is NOT tagged as 'active' and will remain
         * 'suspended' until the hardware is enumerated/initialized. This is required
         * to make sure the ASoC framework use of pm_runtime_get_sync() does not silently
         * fail with -EACCESS because of race conditions between card creation and enumeration
         */

        dev_dbg(dev, "%s\n", __func__);

        return ret;
}

int rt715_sdca_io_init(struct device *dev, struct sdw_slave *slave)
{
        struct rt715_sdca_priv *rt715 = dev_get_drvdata(dev);
        unsigned int hw_ver;

        if (rt715->hw_init)
                return 0;

        regcache_cache_only(rt715->regmap, false);
        regcache_cache_only(rt715->mbq_regmap, false);

        /*
         * PM runtime status is marked as 'active' only when a Slave reports as Attached
         */
        if (!rt715->first_hw_init) {
                /* update count of parent 'active' children */
                pm_runtime_set_active(&slave->dev);

                rt715->first_hw_init = true;
        }

        pm_runtime_get_noresume(&slave->dev);

        rt715_sdca_index_read(rt715, RT715_VENDOR_REG,
                RT715_PRODUCT_NUM, &hw_ver);
        hw_ver = hw_ver & 0x000f;

        /* set clock selector = external */
        regmap_write(rt715->regmap,
                SDW_SDCA_CTL(FUN_MIC_ARRAY, RT715_SDCA_CX_CLK_SEL_EN,
                        RT715_SDCA_CX_CLK_SEL_CTRL, CH_00), 0x1);
        /* set GPIO_4/5/6 to be 3rd/4th DMIC usage */
        if (hw_ver == 0x0)
                rt715_sdca_index_update_bits(rt715, RT715_VENDOR_REG,
                        RT715_AD_FUNC_EN, 0x54, 0x54);
        else if (hw_ver == 0x1) {
                rt715_sdca_index_update_bits(rt715, RT715_VENDOR_REG,
                        RT715_AD_FUNC_EN, 0x55, 0x55);
                rt715_sdca_index_update_bits(rt715, RT715_VENDOR_REG,
                        RT715_REV_1, 0x40, 0x40);
        }
        /* DFLL Calibration trigger */
        rt715_sdca_index_update_bits(rt715, RT715_VENDOR_REG,
                        RT715_DFLL_VAD, 0x1, 0x1);
        /* trigger mode = VAD enable */
        regmap_write(rt715->regmap,
                SDW_SDCA_CTL(FUN_MIC_ARRAY, RT715_SDCA_SMPU_TRIG_ST_EN,
                        RT715_SDCA_SMPU_TRIG_EN_CTRL, CH_00), 0x2);
        /* SMPU-1 interrupt enable mask */
        regmap_update_bits(rt715->regmap, RT715_INT_MASK, 0x1, 0x1);

        /* Mark Slave initialization complete */
        rt715->hw_init = true;

        pm_runtime_put_autosuspend(&slave->dev);

        return 0;
}

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