// SPDX-License-Identifier: GPL-2.0-only
/*
 * wm8958-dsp2.c  --  WM8958 DSP2 support
 *
 * Copyright 2011 Wolfson Microelectronics plc
 *
 * Author: Mark Brown <broonie@opensource.wolfsonmicro.com>
 */

#include <linux/module.h>
#include <linux/moduleparam.h>
#include <linux/init.h>
#include <linux/delay.h>
#include <linux/pm.h>
#include <linux/i2c.h>
#include <linux/platform_device.h>
#include <linux/slab.h>
#include <sound/soc.h>
#include <sound/initval.h>
#include <sound/tlv.h>
#include <trace/events/asoc.h>

#include <linux/mfd/wm8994/core.h>
#include <linux/mfd/wm8994/registers.h>
#include <linux/mfd/wm8994/pdata.h>
#include <linux/mfd/wm8994/gpio.h>

#include <linux/unaligned.h>

#include "wm8994.h"

#define WM_FW_BLOCK_INFO 0xff
#define WM_FW_BLOCK_PM   0x00
#define WM_FW_BLOCK_X    0x01
#define WM_FW_BLOCK_Y    0x02
#define WM_FW_BLOCK_Z    0x03
#define WM_FW_BLOCK_I    0x06
#define WM_FW_BLOCK_A    0x08
#define WM_FW_BLOCK_C    0x0c

static int wm8958_dsp2_fw(struct snd_soc_component *component, const char *name,
                          const struct firmware *fw, bool check)
{
        struct wm8994_priv *wm8994 = snd_soc_component_get_drvdata(component);
        u64 data64;
        u32 data32;
        const u8 *data;
        char *str;
        size_t block_len, len;
        int ret = 0;

        /* Suppress unneeded downloads */
        if (wm8994->cur_fw == fw)
                return 0;

        if (fw->size < 32) {
                dev_err(component->dev, "%s: firmware too short (%zd bytes)\n",
                        name, fw->size);
                goto err;
        }

        if (memcmp(fw->data, "WMFW", 4) != 0) {
                data32 = get_unaligned_be32(fw->data);
                dev_err(component->dev, "%s: firmware has bad file magic %08x\n",
                        name, data32);
                goto err;
        }

        len = get_unaligned_be32(fw->data + 4);
        data32 = get_unaligned_be32(fw->data + 8);

        if ((data32 >> 24) & 0xff) {
                dev_err(component->dev, "%s: unsupported firmware version %d\n",
                        name, (data32 >> 24) & 0xff);
                goto err;
        }
        if ((data32 & 0xffff) != 8958) {
                dev_err(component->dev, "%s: unsupported target device %d\n",
                        name, data32 & 0xffff);
                goto err;
        }
        if (((data32 >> 16) & 0xff) != 0xc) {
                dev_err(component->dev, "%s: unsupported target core %d\n",
                        name, (data32 >> 16) & 0xff);
                goto err;
        }

        if (check) {
                data64 = get_unaligned_be64(fw->data + 24);
                dev_info(component->dev, "%s timestamp %llx\n",  name, data64);
        } else {
                snd_soc_component_write(component, 0x102, 0x2);
                snd_soc_component_write(component, 0x900, 0x2);
        }

        data = fw->data + len;
        len = fw->size - len;
        while (len) {
                if (len < 12) {
                        dev_err(component->dev, "%s short data block of %zd\n",
                                name, len);
                        goto err;
                }

                block_len = get_unaligned_be32(data + 4);
                if (block_len + 8 > len) {
                        dev_err(component->dev, "%zd byte block longer than file\n",
                                block_len);
                        goto err;
                }
                if (block_len == 0) {
                        dev_err(component->dev, "Zero length block\n");
                        goto err;
                }

                data32 = get_unaligned_be32(data);

                switch ((data32 >> 24) & 0xff) {
                case WM_FW_BLOCK_INFO:
                        /* Informational text */
                        if (!check)
                                break;

                        str = kzalloc(block_len + 1, GFP_KERNEL);
                        if (str) {
                                memcpy(str, data + 8, block_len);
                                dev_info(component->dev, "%s: %s\n", name, str);
                                kfree(str);
                        } else {
                                dev_err(component->dev, "Out of memory\n");
                        }
                        break;
                case WM_FW_BLOCK_PM:
                case WM_FW_BLOCK_X:
                case WM_FW_BLOCK_Y:
                case WM_FW_BLOCK_Z:
                case WM_FW_BLOCK_I:
                case WM_FW_BLOCK_A:
                case WM_FW_BLOCK_C:
                        dev_dbg(component->dev, "%s: %zd bytes of %x@%x\n", name,
                                block_len, (data32 >> 24) & 0xff,
                                data32 & 0xffffff);

                        if (check)
                                break;

                        data32 &= 0xffffff;

                        wm8994_bulk_write(wm8994->wm8994,
                                          data32 & 0xffffff,
                                          block_len / 2,
                                          (void *)(data + 8));

                        break;
                default:
                        dev_warn(component->dev, "%s: unknown block type %d\n",
                                 name, (data32 >> 24) & 0xff);
                        break;
                }

                /* Round up to the next 32 bit word */
                block_len += block_len % 4;

                data += block_len + 8;
                len -= block_len + 8;
        }

        if (!check) {
                dev_dbg(component->dev, "%s: download done\n", name);
                wm8994->cur_fw = fw;
        } else {
                dev_info(component->dev, "%s: got firmware\n", name);
        }

        goto ok;

err:
        ret = -EINVAL;
ok:
        if (!check) {
                snd_soc_component_write(component, 0x900, 0x0);
                snd_soc_component_write(component, 0x102, 0x0);
        }

        return ret;
}

static void wm8958_dsp_start_mbc(struct snd_soc_component *component, int path)
{
        struct wm8994_priv *wm8994 = snd_soc_component_get_drvdata(component);
        struct wm8994 *control = wm8994->wm8994;
        int i;

        /* If the DSP is already running then noop */
        if (snd_soc_component_read(component, WM8958_DSP2_PROGRAM) & WM8958_DSP2_ENA)
                return;

        /* If we have MBC firmware download it */
        if (wm8994->mbc)
                wm8958_dsp2_fw(component, "MBC", wm8994->mbc, false);

        snd_soc_component_update_bits(component, WM8958_DSP2_PROGRAM,
                            WM8958_DSP2_ENA, WM8958_DSP2_ENA);

        /* If we've got user supplied MBC settings use them */
        if (control->pdata.num_mbc_cfgs) {
                struct wm8958_mbc_cfg *cfg
                        = &control->pdata.mbc_cfgs[wm8994->mbc_cfg];

                for (i = 0; i < ARRAY_SIZE(cfg->coeff_regs); i++)
                        snd_soc_component_write(component, i + WM8958_MBC_BAND_1_K_1,
                                      cfg->coeff_regs[i]);

                for (i = 0; i < ARRAY_SIZE(cfg->cutoff_regs); i++)
                        snd_soc_component_write(component,
                                      i + WM8958_MBC_BAND_2_LOWER_CUTOFF_C1_1,
                                      cfg->cutoff_regs[i]);
        }

        /* Run the DSP */
        snd_soc_component_write(component, WM8958_DSP2_EXECCONTROL,
                      WM8958_DSP2_RUNR);

        /* And we're off! */
        snd_soc_component_update_bits(component, WM8958_DSP2_CONFIG,
                            WM8958_MBC_ENA |
                            WM8958_MBC_SEL_MASK,
                            path << WM8958_MBC_SEL_SHIFT |
                            WM8958_MBC_ENA);
}

static void wm8958_dsp_start_vss(struct snd_soc_component *component, int path)
{
        struct wm8994_priv *wm8994 = snd_soc_component_get_drvdata(component);
        struct wm8994 *control = wm8994->wm8994;
        int i, ena;

        if (wm8994->mbc_vss)
                wm8958_dsp2_fw(component, "MBC+VSS", wm8994->mbc_vss, false);

        snd_soc_component_update_bits(component, WM8958_DSP2_PROGRAM,
                            WM8958_DSP2_ENA, WM8958_DSP2_ENA);

        /* If we've got user supplied settings use them */
        if (control->pdata.num_mbc_cfgs) {
                struct wm8958_mbc_cfg *cfg
                        = &control->pdata.mbc_cfgs[wm8994->mbc_cfg];

                for (i = 0; i < ARRAY_SIZE(cfg->combined_regs); i++)
                        snd_soc_component_write(component, i + 0x2800,
                                      cfg->combined_regs[i]);
        }

        if (control->pdata.num_vss_cfgs) {
                struct wm8958_vss_cfg *cfg
                        = &control->pdata.vss_cfgs[wm8994->vss_cfg];

                for (i = 0; i < ARRAY_SIZE(cfg->regs); i++)
                        snd_soc_component_write(component, i + 0x2600, cfg->regs[i]);
        }

        if (control->pdata.num_vss_hpf_cfgs) {
                struct wm8958_vss_hpf_cfg *cfg
                        = &control->pdata.vss_hpf_cfgs[wm8994->vss_hpf_cfg];

                for (i = 0; i < ARRAY_SIZE(cfg->regs); i++)
                        snd_soc_component_write(component, i + 0x2400, cfg->regs[i]);
        }

        /* Run the DSP */
        snd_soc_component_write(component, WM8958_DSP2_EXECCONTROL,
                      WM8958_DSP2_RUNR);

        /* Enable the algorithms we've selected */
        ena = 0;
        if (wm8994->mbc_ena[path])
                ena |= 0x8;
        if (wm8994->hpf2_ena[path])
                ena |= 0x4;
        if (wm8994->hpf1_ena[path])
                ena |= 0x2;
        if (wm8994->vss_ena[path])
                ena |= 0x1;

        snd_soc_component_write(component, 0x2201, ena);

        /* Switch the DSP into the data path */
        snd_soc_component_update_bits(component, WM8958_DSP2_CONFIG,
                            WM8958_MBC_SEL_MASK | WM8958_MBC_ENA,
                            path << WM8958_MBC_SEL_SHIFT | WM8958_MBC_ENA);
}

static void wm8958_dsp_start_enh_eq(struct snd_soc_component *component, int path)
{
        struct wm8994_priv *wm8994 = snd_soc_component_get_drvdata(component);
        struct wm8994 *control = wm8994->wm8994;
        int i;

        wm8958_dsp2_fw(component, "ENH_EQ", wm8994->enh_eq, false);

        snd_soc_component_update_bits(component, WM8958_DSP2_PROGRAM,
                            WM8958_DSP2_ENA, WM8958_DSP2_ENA);

        /* If we've got user supplied settings use them */
        if (control->pdata.num_enh_eq_cfgs) {
                struct wm8958_enh_eq_cfg *cfg
                        = &control->pdata.enh_eq_cfgs[wm8994->enh_eq_cfg];

                for (i = 0; i < ARRAY_SIZE(cfg->regs); i++)
                        snd_soc_component_write(component, i + 0x2200,
                                      cfg->regs[i]);
        }

        /* Run the DSP */
        snd_soc_component_write(component, WM8958_DSP2_EXECCONTROL,
                      WM8958_DSP2_RUNR);

        /* Switch the DSP into the data path */
        snd_soc_component_update_bits(component, WM8958_DSP2_CONFIG,
                            WM8958_MBC_SEL_MASK | WM8958_MBC_ENA,
                            path << WM8958_MBC_SEL_SHIFT | WM8958_MBC_ENA);
}

static void wm8958_dsp_apply(struct snd_soc_component *component, int path, int start)
{
        struct wm8994_priv *wm8994 = snd_soc_component_get_drvdata(component);
        int pwr_reg = snd_soc_component_read(component, WM8994_POWER_MANAGEMENT_5);
        int ena, reg, aif;

        switch (path) {
        case 0:
                pwr_reg &= (WM8994_AIF1DAC1L_ENA | WM8994_AIF1DAC1R_ENA);
                aif = 0;
                break;
        case 1:
                pwr_reg &= (WM8994_AIF1DAC2L_ENA | WM8994_AIF1DAC2R_ENA);
                aif = 0;
                break;
        case 2:
                pwr_reg &= (WM8994_AIF2DACL_ENA | WM8994_AIF2DACR_ENA);
                aif = 1;
                break;
        default:
                WARN(1, "Invalid path %d\n", path);
                return;
        }

        /* Do we have both an active AIF and an active algorithm? */
        ena = wm8994->mbc_ena[path] || wm8994->vss_ena[path] ||
                wm8994->hpf1_ena[path] || wm8994->hpf2_ena[path] ||
                wm8994->enh_eq_ena[path];
        if (!pwr_reg)
                ena = 0;

        reg = snd_soc_component_read(component, WM8958_DSP2_PROGRAM);

        dev_dbg(component->dev, "DSP path %d %d startup: %d, power: %x, DSP: %x\n",
                path, wm8994->dsp_active, start, pwr_reg, reg);

        if (start && ena) {
                /* If the DSP is already running then noop */
                if (reg & WM8958_DSP2_ENA)
                        return;

                /* If either AIFnCLK is not yet enabled postpone */
                if (!(snd_soc_component_read(component, WM8994_AIF1_CLOCKING_1)
                      & WM8994_AIF1CLK_ENA_MASK) &&
                    !(snd_soc_component_read(component, WM8994_AIF2_CLOCKING_1)
                      & WM8994_AIF2CLK_ENA_MASK))
                        return;

                /* Switch the clock over to the appropriate AIF */
                snd_soc_component_update_bits(component, WM8994_CLOCKING_1,
                                    WM8958_DSP2CLK_SRC | WM8958_DSP2CLK_ENA,
                                    aif << WM8958_DSP2CLK_SRC_SHIFT |
                                    WM8958_DSP2CLK_ENA);

                if (wm8994->enh_eq_ena[path])
                        wm8958_dsp_start_enh_eq(component, path);
                else if (wm8994->vss_ena[path] || wm8994->hpf1_ena[path] ||
                    wm8994->hpf2_ena[path])
                        wm8958_dsp_start_vss(component, path);
                else if (wm8994->mbc_ena[path])
                        wm8958_dsp_start_mbc(component, path);

                wm8994->dsp_active = path;

                dev_dbg(component->dev, "DSP running in path %d\n", path);
        }

        if (!start && wm8994->dsp_active == path) {
                /* If the DSP is already stopped then noop */
                if (!(reg & WM8958_DSP2_ENA))
                        return;

                snd_soc_component_update_bits(component, WM8958_DSP2_CONFIG,
                                    WM8958_MBC_ENA, 0); 
                snd_soc_component_write(component, WM8958_DSP2_EXECCONTROL,
                              WM8958_DSP2_STOP);
                snd_soc_component_update_bits(component, WM8958_DSP2_PROGRAM,
                                    WM8958_DSP2_ENA, 0);
                snd_soc_component_update_bits(component, WM8994_CLOCKING_1,
                                    WM8958_DSP2CLK_ENA, 0);

                wm8994->dsp_active = -1;

                dev_dbg(component->dev, "DSP stopped\n");
        }
}

int wm8958_aif_ev(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 wm8994 *control = dev_get_drvdata(component->dev->parent);
        int i;

        if (control->type != WM8958)
                return 0;

        switch (event) {
        case SND_SOC_DAPM_POST_PMU:
        case SND_SOC_DAPM_PRE_PMU:
                for (i = 0; i < 3; i++)
                        wm8958_dsp_apply(component, i, 1);
                break;
        case SND_SOC_DAPM_POST_PMD:
        case SND_SOC_DAPM_PRE_PMD:
                for (i = 0; i < 3; i++)
                        wm8958_dsp_apply(component, i, 0);
                break;
        }

        return 0;
}

/* Check if DSP2 is in use on another AIF */
static int wm8958_dsp2_busy(struct wm8994_priv *wm8994, int aif)
{
        int i;

        for (i = 0; i < ARRAY_SIZE(wm8994->mbc_ena); i++) {
                if (i == aif)
                        continue;
                if (wm8994->mbc_ena[i] || wm8994->vss_ena[i] ||
                    wm8994->hpf1_ena[i] || wm8994->hpf2_ena[i])
                        return 1;
        }

        return 0;
}

static int wm8958_put_mbc_enum(struct snd_kcontrol *kcontrol,
                               struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct wm8994_priv *wm8994 = snd_soc_component_get_drvdata(component);
        struct wm8994 *control = wm8994->wm8994;
        int value = ucontrol->value.enumerated.item[0];
        int reg;

        /* Don't allow on the fly reconfiguration */
        reg = snd_soc_component_read(component, WM8994_CLOCKING_1);
        if (reg < 0 || reg & WM8958_DSP2CLK_ENA)
                return -EBUSY;

        if (value >= control->pdata.num_mbc_cfgs)
                return -EINVAL;

        wm8994->mbc_cfg = value;

        return 0;
}

static int wm8958_get_mbc_enum(struct snd_kcontrol *kcontrol,
                               struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct wm8994_priv *wm8994 = snd_soc_component_get_drvdata(component);

        ucontrol->value.enumerated.item[0] = wm8994->mbc_cfg;

        return 0;
}

static int wm8958_mbc_info(struct snd_kcontrol *kcontrol,
                           struct snd_ctl_elem_info *uinfo)
{
        uinfo->type = SNDRV_CTL_ELEM_TYPE_BOOLEAN;
        uinfo->count = 1;
        uinfo->value.integer.min = 0;
        uinfo->value.integer.max = 1;
        return 0;
}

static int wm8958_mbc_get(struct snd_kcontrol *kcontrol,
                          struct snd_ctl_elem_value *ucontrol)
{
        int mbc = kcontrol->private_value;
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct wm8994_priv *wm8994 = snd_soc_component_get_drvdata(component);

        ucontrol->value.integer.value[0] = wm8994->mbc_ena[mbc];

        return 0;
}

static int wm8958_mbc_put(struct snd_kcontrol *kcontrol,
                          struct snd_ctl_elem_value *ucontrol)
{
        int mbc = kcontrol->private_value;
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct wm8994_priv *wm8994 = snd_soc_component_get_drvdata(component);

        if (wm8994->mbc_ena[mbc] == ucontrol->value.integer.value[0])
                return 0;

        if (ucontrol->value.integer.value[0] > 1)
                return -EINVAL;

        if (wm8958_dsp2_busy(wm8994, mbc)) {
                dev_dbg(component->dev, "DSP2 active on %d already\n", mbc);
                return -EBUSY;
        }

        if (wm8994->enh_eq_ena[mbc])
                return -EBUSY;

        wm8994->mbc_ena[mbc] = ucontrol->value.integer.value[0];

        wm8958_dsp_apply(component, mbc, wm8994->mbc_ena[mbc]);

        return 1;
}

#define WM8958_MBC_SWITCH(xname, xval) {\
        .iface = SNDRV_CTL_ELEM_IFACE_MIXER, .name = (xname), \
        .access = SNDRV_CTL_ELEM_ACCESS_READWRITE,\
        .info = wm8958_mbc_info, \
        .get = wm8958_mbc_get, .put = wm8958_mbc_put, \
        .private_value = xval }

static int wm8958_put_vss_enum(struct snd_kcontrol *kcontrol,
                               struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct wm8994_priv *wm8994 = snd_soc_component_get_drvdata(component);
        struct wm8994 *control = wm8994->wm8994;
        int value = ucontrol->value.enumerated.item[0];
        int reg;

        /* Don't allow on the fly reconfiguration */
        reg = snd_soc_component_read(component, WM8994_CLOCKING_1);
        if (reg < 0 || reg & WM8958_DSP2CLK_ENA)
                return -EBUSY;

        if (value >= control->pdata.num_vss_cfgs)
                return -EINVAL;

        wm8994->vss_cfg = value;

        return 0;
}

static int wm8958_get_vss_enum(struct snd_kcontrol *kcontrol,
                               struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct wm8994_priv *wm8994 = snd_soc_component_get_drvdata(component);

        ucontrol->value.enumerated.item[0] = wm8994->vss_cfg;

        return 0;
}

static int wm8958_put_vss_hpf_enum(struct snd_kcontrol *kcontrol,
                                   struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct wm8994_priv *wm8994 = snd_soc_component_get_drvdata(component);
        struct wm8994 *control = wm8994->wm8994;
        int value = ucontrol->value.enumerated.item[0];
        int reg;

        /* Don't allow on the fly reconfiguration */
        reg = snd_soc_component_read(component, WM8994_CLOCKING_1);
        if (reg < 0 || reg & WM8958_DSP2CLK_ENA)
                return -EBUSY;

        if (value >= control->pdata.num_vss_hpf_cfgs)
                return -EINVAL;

        wm8994->vss_hpf_cfg = value;

        return 0;
}

static int wm8958_get_vss_hpf_enum(struct snd_kcontrol *kcontrol,
                                   struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct wm8994_priv *wm8994 = snd_soc_component_get_drvdata(component);

        ucontrol->value.enumerated.item[0] = wm8994->vss_hpf_cfg;

        return 0;
}

static int wm8958_vss_info(struct snd_kcontrol *kcontrol,
                           struct snd_ctl_elem_info *uinfo)
{
        uinfo->type = SNDRV_CTL_ELEM_TYPE_BOOLEAN;
        uinfo->count = 1;
        uinfo->value.integer.min = 0;
        uinfo->value.integer.max = 1;
        return 0;
}

static int wm8958_vss_get(struct snd_kcontrol *kcontrol,
                          struct snd_ctl_elem_value *ucontrol)
{
        int vss = kcontrol->private_value;
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct wm8994_priv *wm8994 = snd_soc_component_get_drvdata(component);

        ucontrol->value.integer.value[0] = wm8994->vss_ena[vss];

        return 0;
}

static int wm8958_vss_put(struct snd_kcontrol *kcontrol,
                          struct snd_ctl_elem_value *ucontrol)
{
        int vss = kcontrol->private_value;
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct wm8994_priv *wm8994 = snd_soc_component_get_drvdata(component);

        if (wm8994->vss_ena[vss] == ucontrol->value.integer.value[0])
                return 0;

        if (ucontrol->value.integer.value[0] > 1)
                return -EINVAL;

        if (!wm8994->mbc_vss)
                return -ENODEV;

        if (wm8958_dsp2_busy(wm8994, vss)) {
                dev_dbg(component->dev, "DSP2 active on %d already\n", vss);
                return -EBUSY;
        }

        if (wm8994->enh_eq_ena[vss])
                return -EBUSY;

        wm8994->vss_ena[vss] = ucontrol->value.integer.value[0];

        wm8958_dsp_apply(component, vss, wm8994->vss_ena[vss]);

        return 1;
}


#define WM8958_VSS_SWITCH(xname, xval) {\
        .iface = SNDRV_CTL_ELEM_IFACE_MIXER, .name = (xname), \
        .access = SNDRV_CTL_ELEM_ACCESS_READWRITE,\
        .info = wm8958_vss_info, \
        .get = wm8958_vss_get, .put = wm8958_vss_put, \
        .private_value = xval }

static int wm8958_hpf_info(struct snd_kcontrol *kcontrol,
                           struct snd_ctl_elem_info *uinfo)
{
        uinfo->type = SNDRV_CTL_ELEM_TYPE_BOOLEAN;
        uinfo->count = 1;
        uinfo->value.integer.min = 0;
        uinfo->value.integer.max = 1;
        return 0;
}

static int wm8958_hpf_get(struct snd_kcontrol *kcontrol,
                          struct snd_ctl_elem_value *ucontrol)
{
        int hpf = kcontrol->private_value;
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct wm8994_priv *wm8994 = snd_soc_component_get_drvdata(component);

        if (hpf < 3)
                ucontrol->value.integer.value[0] = wm8994->hpf1_ena[hpf % 3];
        else
                ucontrol->value.integer.value[0] = wm8994->hpf2_ena[hpf % 3];

        return 0;
}

static int wm8958_hpf_put(struct snd_kcontrol *kcontrol,
                          struct snd_ctl_elem_value *ucontrol)
{
        int hpf = kcontrol->private_value;
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct wm8994_priv *wm8994 = snd_soc_component_get_drvdata(component);

        if (hpf < 3) {
                if (wm8994->hpf1_ena[hpf % 3] ==
                    ucontrol->value.integer.value[0])
                        return 0;
        } else {
                if (wm8994->hpf2_ena[hpf % 3] ==
                    ucontrol->value.integer.value[0])
                        return 0;
        }

        if (ucontrol->value.integer.value[0] > 1)
                return -EINVAL;

        if (!wm8994->mbc_vss)
                return -ENODEV;

        if (wm8958_dsp2_busy(wm8994, hpf % 3)) {
                dev_dbg(component->dev, "DSP2 active on %d already\n", hpf);
                return -EBUSY;
        }

        if (wm8994->enh_eq_ena[hpf % 3])
                return -EBUSY;

        if (hpf < 3)
                wm8994->hpf1_ena[hpf % 3] = ucontrol->value.integer.value[0];
        else
                wm8994->hpf2_ena[hpf % 3] = ucontrol->value.integer.value[0];

        wm8958_dsp_apply(component, hpf % 3, ucontrol->value.integer.value[0]);

        return 1;
}

#define WM8958_HPF_SWITCH(xname, xval) {\
        .iface = SNDRV_CTL_ELEM_IFACE_MIXER, .name = (xname), \
        .access = SNDRV_CTL_ELEM_ACCESS_READWRITE,\
        .info = wm8958_hpf_info, \
        .get = wm8958_hpf_get, .put = wm8958_hpf_put, \
        .private_value = xval }

static int wm8958_put_enh_eq_enum(struct snd_kcontrol *kcontrol,
                                  struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct wm8994_priv *wm8994 = snd_soc_component_get_drvdata(component);
        struct wm8994 *control = wm8994->wm8994;
        int value = ucontrol->value.enumerated.item[0];
        int reg;

        /* Don't allow on the fly reconfiguration */
        reg = snd_soc_component_read(component, WM8994_CLOCKING_1);
        if (reg < 0 || reg & WM8958_DSP2CLK_ENA)
                return -EBUSY;

        if (value >= control->pdata.num_enh_eq_cfgs)
                return -EINVAL;

        wm8994->enh_eq_cfg = value;

        return 0;
}

static int wm8958_get_enh_eq_enum(struct snd_kcontrol *kcontrol,
                                  struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct wm8994_priv *wm8994 = snd_soc_component_get_drvdata(component);

        ucontrol->value.enumerated.item[0] = wm8994->enh_eq_cfg;

        return 0;
}

static int wm8958_enh_eq_info(struct snd_kcontrol *kcontrol,
                           struct snd_ctl_elem_info *uinfo)
{
        uinfo->type = SNDRV_CTL_ELEM_TYPE_BOOLEAN;
        uinfo->count = 1;
        uinfo->value.integer.min = 0;
        uinfo->value.integer.max = 1;
        return 0;
}

static int wm8958_enh_eq_get(struct snd_kcontrol *kcontrol,
                          struct snd_ctl_elem_value *ucontrol)
{
        int eq = kcontrol->private_value;
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct wm8994_priv *wm8994 = snd_soc_component_get_drvdata(component);

        ucontrol->value.integer.value[0] = wm8994->enh_eq_ena[eq];

        return 0;
}

static int wm8958_enh_eq_put(struct snd_kcontrol *kcontrol,
                          struct snd_ctl_elem_value *ucontrol)
{
        int eq = kcontrol->private_value;
        struct snd_soc_component *component = snd_kcontrol_chip(kcontrol);
        struct wm8994_priv *wm8994 = snd_soc_component_get_drvdata(component);

        if (wm8994->enh_eq_ena[eq] == ucontrol->value.integer.value[0])
                return 0;

        if (ucontrol->value.integer.value[0] > 1)
                return -EINVAL;

        if (!wm8994->enh_eq)
                return -ENODEV;

        if (wm8958_dsp2_busy(wm8994, eq)) {
                dev_dbg(component->dev, "DSP2 active on %d already\n", eq);
                return -EBUSY;
        }

        if (wm8994->mbc_ena[eq] || wm8994->vss_ena[eq] ||
            wm8994->hpf1_ena[eq] || wm8994->hpf2_ena[eq])
                return -EBUSY;

        wm8994->enh_eq_ena[eq] = ucontrol->value.integer.value[0];

        wm8958_dsp_apply(component, eq, ucontrol->value.integer.value[0]);

        return 1;
}

#define WM8958_ENH_EQ_SWITCH(xname, xval) {\
        .iface = SNDRV_CTL_ELEM_IFACE_MIXER, .name = (xname), \
        .access = SNDRV_CTL_ELEM_ACCESS_READWRITE,\
        .info = wm8958_enh_eq_info, \
        .get = wm8958_enh_eq_get, .put = wm8958_enh_eq_put, \
        .private_value = xval }

static const struct snd_kcontrol_new wm8958_mbc_snd_controls[] = {
WM8958_MBC_SWITCH("AIF1DAC1 MBC Switch", 0),
WM8958_MBC_SWITCH("AIF1DAC2 MBC Switch", 1),
WM8958_MBC_SWITCH("AIF2DAC MBC Switch", 2),
};

static const struct snd_kcontrol_new wm8958_vss_snd_controls[] = {
WM8958_VSS_SWITCH("AIF1DAC1 VSS Switch", 0),
WM8958_VSS_SWITCH("AIF1DAC2 VSS Switch", 1),
WM8958_VSS_SWITCH("AIF2DAC VSS Switch", 2),
WM8958_HPF_SWITCH("AIF1DAC1 HPF1 Switch", 0),
WM8958_HPF_SWITCH("AIF1DAC2 HPF1 Switch", 1),
WM8958_HPF_SWITCH("AIF2DAC HPF1 Switch", 2),
WM8958_HPF_SWITCH("AIF1DAC1 HPF2 Switch", 3),
WM8958_HPF_SWITCH("AIF1DAC2 HPF2 Switch", 4),
WM8958_HPF_SWITCH("AIF2DAC HPF2 Switch", 5),
};

static const struct snd_kcontrol_new wm8958_enh_eq_snd_controls[] = {
WM8958_ENH_EQ_SWITCH("AIF1DAC1 Enhanced EQ Switch", 0),
WM8958_ENH_EQ_SWITCH("AIF1DAC2 Enhanced EQ Switch", 1),
WM8958_ENH_EQ_SWITCH("AIF2DAC Enhanced EQ Switch", 2),
};

static void wm8958_enh_eq_loaded(const struct firmware *fw, void *context)
{
        struct snd_soc_component *component = context;
        struct wm8994_priv *wm8994 = snd_soc_component_get_drvdata(component);

        if (fw && (wm8958_dsp2_fw(component, "ENH_EQ", fw, true) == 0)) {
                mutex_lock(&wm8994->fw_lock);
                wm8994->enh_eq = fw;
                mutex_unlock(&wm8994->fw_lock);
        }
}

static void wm8958_mbc_vss_loaded(const struct firmware *fw, void *context)
{
        struct snd_soc_component *component = context;
        struct wm8994_priv *wm8994 = snd_soc_component_get_drvdata(component);

        if (fw && (wm8958_dsp2_fw(component, "MBC+VSS", fw, true) == 0)) {
                mutex_lock(&wm8994->fw_lock);
                wm8994->mbc_vss = fw;
                mutex_unlock(&wm8994->fw_lock);
        }
}

static void wm8958_mbc_loaded(const struct firmware *fw, void *context)
{
        struct snd_soc_component *component = context;
        struct wm8994_priv *wm8994 = snd_soc_component_get_drvdata(component);

        if (fw && (wm8958_dsp2_fw(component, "MBC", fw, true) == 0)) {
                mutex_lock(&wm8994->fw_lock);
                wm8994->mbc = fw;
                mutex_unlock(&wm8994->fw_lock);
        }
}

void wm8958_dsp2_init(struct snd_soc_component *component)
{
        struct wm8994_priv *wm8994 = snd_soc_component_get_drvdata(component);
        struct wm8994 *control = wm8994->wm8994;
        struct wm8994_pdata *pdata = &control->pdata;
        int ret, i;

        wm8994->dsp_active = -1;

        snd_soc_add_component_controls(component, wm8958_mbc_snd_controls,
                             ARRAY_SIZE(wm8958_mbc_snd_controls));
        snd_soc_add_component_controls(component, wm8958_vss_snd_controls,
                             ARRAY_SIZE(wm8958_vss_snd_controls));
        snd_soc_add_component_controls(component, wm8958_enh_eq_snd_controls,
                             ARRAY_SIZE(wm8958_enh_eq_snd_controls));


        /* We don't *require* firmware and don't want to delay boot */
        request_firmware_nowait(THIS_MODULE, FW_ACTION_UEVENT,
                                "wm8958_mbc.wfw", component->dev, GFP_KERNEL,
                                component, wm8958_mbc_loaded);
        request_firmware_nowait(THIS_MODULE, FW_ACTION_UEVENT,
                                "wm8958_mbc_vss.wfw", component->dev, GFP_KERNEL,
                                component, wm8958_mbc_vss_loaded);
        request_firmware_nowait(THIS_MODULE, FW_ACTION_UEVENT,
                                "wm8958_enh_eq.wfw", component->dev, GFP_KERNEL,
                                component, wm8958_enh_eq_loaded);

        if (pdata->num_mbc_cfgs) {
                struct snd_kcontrol_new mbc_control[] = {
                        SOC_ENUM_EXT("MBC Mode", wm8994->mbc_enum,
                                     wm8958_get_mbc_enum, wm8958_put_mbc_enum),
                };

                /* We need an array of texts for the enum API */
                wm8994->mbc_texts = kmalloc_array(pdata->num_mbc_cfgs,
                                                  sizeof(char *),
                                                  GFP_KERNEL);
                if (!wm8994->mbc_texts)
                        return;

                for (i = 0; i < pdata->num_mbc_cfgs; i++)
                        wm8994->mbc_texts[i] = pdata->mbc_cfgs[i].name;

                wm8994->mbc_enum.items = pdata->num_mbc_cfgs;
                wm8994->mbc_enum.texts = wm8994->mbc_texts;

                ret = snd_soc_add_component_controls(wm8994->hubs.component,
                                                 mbc_control, 1);
                if (ret != 0)
                        dev_err(wm8994->hubs.component->dev,
                                "Failed to add MBC mode controls: %d\n", ret);
        }

        if (pdata->num_vss_cfgs) {
                struct snd_kcontrol_new vss_control[] = {
                        SOC_ENUM_EXT("VSS Mode", wm8994->vss_enum,
                                     wm8958_get_vss_enum, wm8958_put_vss_enum),
                };

                /* We need an array of texts for the enum API */
                wm8994->vss_texts = kmalloc_array(pdata->num_vss_cfgs,
                                                  sizeof(char *),
                                                  GFP_KERNEL);
                if (!wm8994->vss_texts)
                        return;

                for (i = 0; i < pdata->num_vss_cfgs; i++)
                        wm8994->vss_texts[i] = pdata->vss_cfgs[i].name;

                wm8994->vss_enum.items = pdata->num_vss_cfgs;
                wm8994->vss_enum.texts = wm8994->vss_texts;

                ret = snd_soc_add_component_controls(wm8994->hubs.component,
                                                 vss_control, 1);
                if (ret != 0)
                        dev_err(wm8994->hubs.component->dev,
                                "Failed to add VSS mode controls: %d\n", ret);
        }

        if (pdata->num_vss_hpf_cfgs) {
                struct snd_kcontrol_new hpf_control[] = {
                        SOC_ENUM_EXT("VSS HPF Mode", wm8994->vss_hpf_enum,
                                     wm8958_get_vss_hpf_enum,
                                     wm8958_put_vss_hpf_enum),
                };

                /* We need an array of texts for the enum API */
                wm8994->vss_hpf_texts = kmalloc_array(pdata->num_vss_hpf_cfgs,
                                                      sizeof(char *),
                                                      GFP_KERNEL);
                if (!wm8994->vss_hpf_texts)
                        return;

                for (i = 0; i < pdata->num_vss_hpf_cfgs; i++)
                        wm8994->vss_hpf_texts[i] = pdata->vss_hpf_cfgs[i].name;

                wm8994->vss_hpf_enum.items = pdata->num_vss_hpf_cfgs;
                wm8994->vss_hpf_enum.texts = wm8994->vss_hpf_texts;

                ret = snd_soc_add_component_controls(wm8994->hubs.component,
                                                 hpf_control, 1);
                if (ret != 0)
                        dev_err(wm8994->hubs.component->dev,
                                "Failed to add VSS HPFmode controls: %d\n",
                                ret);
        }

        if (pdata->num_enh_eq_cfgs) {
                struct snd_kcontrol_new eq_control[] = {
                        SOC_ENUM_EXT("Enhanced EQ Mode", wm8994->enh_eq_enum,
                                     wm8958_get_enh_eq_enum,
                                     wm8958_put_enh_eq_enum),
                };

                /* We need an array of texts for the enum API */
                wm8994->enh_eq_texts = kmalloc_array(pdata->num_enh_eq_cfgs,
                                                     sizeof(char *),
                                                     GFP_KERNEL);
                if (!wm8994->enh_eq_texts)
                        return;

                for (i = 0; i < pdata->num_enh_eq_cfgs; i++)
                        wm8994->enh_eq_texts[i] = pdata->enh_eq_cfgs[i].name;

                wm8994->enh_eq_enum.items = pdata->num_enh_eq_cfgs;
                wm8994->enh_eq_enum.texts = wm8994->enh_eq_texts;

                ret = snd_soc_add_component_controls(wm8994->hubs.component,
                                                 eq_control, 1);
                if (ret != 0)
                        dev_err(wm8994->hubs.component->dev,
                                "Failed to add enhanced EQ controls: %d\n",
                                ret);
        }
}