// SPDX-License-Identifier: GPL-2.0-or-later
 /*
 * Audio Codec driver supporting:
 *  AD1835A, AD1836, AD1837A, AD1838A, AD1839A
 *
 * Copyright 2009-2011 Analog Devices Inc.
 */

#include <linux/init.h>
#include <linux/slab.h>
#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/device.h>
#include <sound/core.h>
#include <sound/pcm.h>
#include <sound/pcm_params.h>
#include <sound/initval.h>
#include <sound/soc.h>
#include <sound/tlv.h>
#include <linux/spi/spi.h>
#include <linux/regmap.h>

#include "ad1836.h"

enum ad1836_type {
        AD1835,
        AD1836,
        AD1838,
};

/* codec private data */
struct ad1836_priv {
        enum ad1836_type type;
        struct regmap *regmap;
};

/*
 * AD1836 volume/mute/de-emphasis etc. controls
 */
static const char *ad1836_deemp[] = {"None", "44.1kHz", "32kHz", "48kHz"};

static SOC_ENUM_SINGLE_DECL(ad1836_deemp_enum,
                            AD1836_DAC_CTRL1, 8, ad1836_deemp);

#define AD1836_DAC_VOLUME(x) \
        SOC_DOUBLE_R("DAC" #x " Playback Volume", AD1836_DAC_L_VOL(x), \
                        AD1836_DAC_R_VOL(x), 0, 0x3FF, 0)

#define AD1836_DAC_SWITCH(x) \
        SOC_DOUBLE("DAC" #x " Playback Switch", AD1836_DAC_CTRL2, \
                        AD1836_MUTE_LEFT(x), AD1836_MUTE_RIGHT(x), 1, 1)

#define AD1836_ADC_SWITCH(x) \
        SOC_DOUBLE("ADC" #x " Capture Switch", AD1836_ADC_CTRL2, \
                AD1836_MUTE_LEFT(x), AD1836_MUTE_RIGHT(x), 1, 1)

static const struct snd_kcontrol_new ad183x_dac_controls[] = {
        AD1836_DAC_VOLUME(1),
        AD1836_DAC_SWITCH(1),
        AD1836_DAC_VOLUME(2),
        AD1836_DAC_SWITCH(2),
        AD1836_DAC_VOLUME(3),
        AD1836_DAC_SWITCH(3),
        AD1836_DAC_VOLUME(4),
        AD1836_DAC_SWITCH(4),
};

static const struct snd_soc_dapm_widget ad183x_dac_dapm_widgets[] = {
        SND_SOC_DAPM_OUTPUT("DAC1OUT"),
        SND_SOC_DAPM_OUTPUT("DAC2OUT"),
        SND_SOC_DAPM_OUTPUT("DAC3OUT"),
        SND_SOC_DAPM_OUTPUT("DAC4OUT"),
};

static const struct snd_soc_dapm_route ad183x_dac_routes[] = {
        { "DAC1OUT", NULL, "DAC" },
        { "DAC2OUT", NULL, "DAC" },
        { "DAC3OUT", NULL, "DAC" },
        { "DAC4OUT", NULL, "DAC" },
};

static const struct snd_kcontrol_new ad183x_adc_controls[] = {
        AD1836_ADC_SWITCH(1),
        AD1836_ADC_SWITCH(2),
        AD1836_ADC_SWITCH(3),
};

static const struct snd_soc_dapm_widget ad183x_adc_dapm_widgets[] = {
        SND_SOC_DAPM_INPUT("ADC1IN"),
        SND_SOC_DAPM_INPUT("ADC2IN"),
};

static const struct snd_soc_dapm_route ad183x_adc_routes[] = {
        { "ADC", NULL, "ADC1IN" },
        { "ADC", NULL, "ADC2IN" },
};

static const struct snd_kcontrol_new ad183x_controls[] = {
        /* ADC high-pass filter */
        SOC_SINGLE("ADC High Pass Filter Switch", AD1836_ADC_CTRL1,
                        AD1836_ADC_HIGHPASS_FILTER, 1, 0),

        /* DAC de-emphasis */
        SOC_ENUM("Playback Deemphasis", ad1836_deemp_enum),
};

static const struct snd_soc_dapm_widget ad183x_dapm_widgets[] = {
        SND_SOC_DAPM_DAC("DAC", "Playback", AD1836_DAC_CTRL1,
                                AD1836_DAC_POWERDOWN, 1),
        SND_SOC_DAPM_ADC("ADC", "Capture", SND_SOC_NOPM, 0, 0),
        SND_SOC_DAPM_SUPPLY("ADC_PWR", AD1836_ADC_CTRL1,
                                AD1836_ADC_POWERDOWN, 1, NULL, 0),
};

static const struct snd_soc_dapm_route ad183x_dapm_routes[] = {
        { "DAC", NULL, "ADC_PWR" },
        { "ADC", NULL, "ADC_PWR" },
};

static const DECLARE_TLV_DB_SCALE(ad1836_in_tlv, 0, 300, 0);

static const struct snd_kcontrol_new ad1836_controls[] = {
        SOC_DOUBLE_TLV("ADC2 Capture Volume", AD1836_ADC_CTRL1, 3, 0, 4, 0,
            ad1836_in_tlv),
};

/*
 * DAI ops entries
 */

static int ad1836_set_dai_fmt(struct snd_soc_dai *codec_dai,
                unsigned int fmt)
{
        switch (fmt & SND_SOC_DAIFMT_FORMAT_MASK) {
        /* at present, we support adc aux mode to interface with
         * blackfin sport tdm mode
         */
        case SND_SOC_DAIFMT_DSP_A:
                break;
        default:
                return -EINVAL;
        }

        switch (fmt & SND_SOC_DAIFMT_INV_MASK) {
        case SND_SOC_DAIFMT_IB_IF:
                break;
        default:
                return -EINVAL;
        }

        switch (fmt & SND_SOC_DAIFMT_CLOCK_PROVIDER_MASK) {
        /* ALCLK,ABCLK are both output, AD1836 can only be provider */
        case SND_SOC_DAIFMT_CBP_CFP:
                break;
        default:
                return -EINVAL;
        }

        return 0;
}

static int ad1836_hw_params(struct snd_pcm_substream *substream,
                struct snd_pcm_hw_params *params,
                struct snd_soc_dai *dai)
{
        struct ad1836_priv *ad1836 = snd_soc_component_get_drvdata(dai->component);
        int word_len = 0;

        /* bit size */
        switch (params_width(params)) {
        case 16:
                word_len = AD1836_WORD_LEN_16;
                break;
        case 20:
                word_len = AD1836_WORD_LEN_20;
                break;
        case 24:
        case 32:
                word_len = AD1836_WORD_LEN_24;
                break;
        default:
                return -EINVAL;
        }

        regmap_update_bits(ad1836->regmap, AD1836_DAC_CTRL1,
                AD1836_DAC_WORD_LEN_MASK,
                word_len << AD1836_DAC_WORD_LEN_OFFSET);

        regmap_update_bits(ad1836->regmap, AD1836_ADC_CTRL2,
                AD1836_ADC_WORD_LEN_MASK,
                word_len << AD1836_ADC_WORD_OFFSET);

        return 0;
}

static const struct snd_soc_dai_ops ad1836_dai_ops = {
        .hw_params = ad1836_hw_params,
        .set_fmt = ad1836_set_dai_fmt,
};

#define AD183X_DAI(_name, num_dacs, num_adcs) \
{ \
        .name = _name "-hifi", \
        .playback = { \
                .stream_name = "Playback", \
                .channels_min = 2, \
                .channels_max = (num_dacs) * 2, \
                .rates = SNDRV_PCM_RATE_48000,  \
                .formats = SNDRV_PCM_FMTBIT_S32_LE | SNDRV_PCM_FMTBIT_S16_LE | \
                        SNDRV_PCM_FMTBIT_S20_3LE | SNDRV_PCM_FMTBIT_S24_LE, \
        }, \
        .capture = { \
                .stream_name = "Capture", \
                .channels_min = 2, \
                .channels_max = (num_adcs) * 2, \
                .rates = SNDRV_PCM_RATE_48000, \
                .formats = SNDRV_PCM_FMTBIT_S32_LE | SNDRV_PCM_FMTBIT_S16_LE | \
                        SNDRV_PCM_FMTBIT_S20_3LE | SNDRV_PCM_FMTBIT_S24_LE, \
        }, \
        .ops = &ad1836_dai_ops, \
}

static struct snd_soc_dai_driver ad183x_dais[] = {
        [AD1835] = AD183X_DAI("ad1835", 4, 1),
        [AD1836] = AD183X_DAI("ad1836", 3, 2),
        [AD1838] = AD183X_DAI("ad1838", 3, 1),
};

#ifdef CONFIG_PM
static int ad1836_suspend(struct snd_soc_component *component)
{
        struct ad1836_priv *ad1836 = snd_soc_component_get_drvdata(component);
        /* reset clock control mode */
        return regmap_update_bits(ad1836->regmap, AD1836_ADC_CTRL2,
                AD1836_ADC_SERFMT_MASK, 0);
}

static int ad1836_resume(struct snd_soc_component *component)
{
        struct ad1836_priv *ad1836 = snd_soc_component_get_drvdata(component);
        /* restore clock control mode */
        return regmap_update_bits(ad1836->regmap, AD1836_ADC_CTRL2,
                AD1836_ADC_SERFMT_MASK, AD1836_ADC_AUX);
}
#else
#define ad1836_suspend NULL
#define ad1836_resume  NULL
#endif

static int ad1836_probe(struct snd_soc_component *component)
{
        struct ad1836_priv *ad1836 = snd_soc_component_get_drvdata(component);
        struct snd_soc_dapm_context *dapm = snd_soc_component_to_dapm(component);
        int num_dacs, num_adcs;
        int ret = 0;
        int i;

        num_dacs = ad183x_dais[ad1836->type].playback.channels_max / 2;
        num_adcs = ad183x_dais[ad1836->type].capture.channels_max / 2;

        /* default setting for ad1836 */
        /* de-emphasis: 48kHz, power-on dac */
        regmap_write(ad1836->regmap, AD1836_DAC_CTRL1, 0x300);
        /* unmute dac channels */
        regmap_write(ad1836->regmap, AD1836_DAC_CTRL2, 0x0);
        /* high-pass filter enable, power-on adc */
        regmap_write(ad1836->regmap, AD1836_ADC_CTRL1, 0x100);
        /* unmute adc channles, adc aux mode */
        regmap_write(ad1836->regmap, AD1836_ADC_CTRL2, 0x180);
        /* volume */
        for (i = 1; i <= num_dacs; ++i) {
                regmap_write(ad1836->regmap, AD1836_DAC_L_VOL(i), 0x3FF);
                regmap_write(ad1836->regmap, AD1836_DAC_R_VOL(i), 0x3FF);
        }

        if (ad1836->type == AD1836) {
                /* left/right diff:PGA/MUX */
                regmap_write(ad1836->regmap, AD1836_ADC_CTRL3, 0x3A);
                ret = snd_soc_add_component_controls(component, ad1836_controls,
                                ARRAY_SIZE(ad1836_controls));
                if (ret)
                        return ret;
        } else {
                regmap_write(ad1836->regmap, AD1836_ADC_CTRL3, 0x00);
        }

        ret = snd_soc_add_component_controls(component, ad183x_dac_controls, num_dacs * 2);
        if (ret)
                return ret;

        ret = snd_soc_add_component_controls(component, ad183x_adc_controls, num_adcs);
        if (ret)
                return ret;

        ret = snd_soc_dapm_new_controls(dapm, ad183x_dac_dapm_widgets, num_dacs);
        if (ret)
                return ret;

        ret = snd_soc_dapm_new_controls(dapm, ad183x_adc_dapm_widgets, num_adcs);
        if (ret)
                return ret;

        ret = snd_soc_dapm_add_routes(dapm, ad183x_dac_routes, num_dacs);
        if (ret)
                return ret;

        ret = snd_soc_dapm_add_routes(dapm, ad183x_adc_routes, num_adcs);

        return ret;
}

/* power down chip */
static void ad1836_remove(struct snd_soc_component *component)
{
        struct ad1836_priv *ad1836 = snd_soc_component_get_drvdata(component);
        /* reset clock control mode */
        regmap_update_bits(ad1836->regmap, AD1836_ADC_CTRL2,
                AD1836_ADC_SERFMT_MASK, 0);
}

static const struct snd_soc_component_driver soc_component_dev_ad1836 = {
        .probe                  = ad1836_probe,
        .remove                 = ad1836_remove,
        .suspend                = ad1836_suspend,
        .resume                 = ad1836_resume,
        .controls               = ad183x_controls,
        .num_controls           = ARRAY_SIZE(ad183x_controls),
        .dapm_widgets           = ad183x_dapm_widgets,
        .num_dapm_widgets       = ARRAY_SIZE(ad183x_dapm_widgets),
        .dapm_routes            = ad183x_dapm_routes,
        .num_dapm_routes        = ARRAY_SIZE(ad183x_dapm_routes),
        .idle_bias_on           = 1,
        .use_pmdown_time        = 1,
        .endianness             = 1,
};

static const struct reg_default ad1836_reg_defaults[] = {
        { AD1836_DAC_CTRL1, 0x0000 },
        { AD1836_DAC_CTRL2, 0x0000 },
        { AD1836_DAC_L_VOL(0), 0x0000 },
        { AD1836_DAC_R_VOL(0), 0x0000 },
        { AD1836_DAC_L_VOL(1), 0x0000 },
        { AD1836_DAC_R_VOL(1), 0x0000 },
        { AD1836_DAC_L_VOL(2), 0x0000 },
        { AD1836_DAC_R_VOL(2), 0x0000 },
        { AD1836_DAC_L_VOL(3), 0x0000 },
        { AD1836_DAC_R_VOL(3), 0x0000 },
        { AD1836_ADC_CTRL1, 0x0000 },
        { AD1836_ADC_CTRL2, 0x0000 },
        { AD1836_ADC_CTRL3, 0x0000 },
};

static const struct regmap_config ad1836_regmap_config = {
        .val_bits = 12,
        .reg_bits = 4,
        .read_flag_mask = 0x08,

        .max_register = AD1836_ADC_CTRL3,
        .reg_defaults = ad1836_reg_defaults,
        .num_reg_defaults = ARRAY_SIZE(ad1836_reg_defaults),
        .cache_type = REGCACHE_MAPLE,
};

static int ad1836_spi_probe(struct spi_device *spi)
{
        struct ad1836_priv *ad1836;
        int ret;

        ad1836 = devm_kzalloc(&spi->dev, sizeof(struct ad1836_priv),
                              GFP_KERNEL);
        if (ad1836 == NULL)
                return -ENOMEM;

        ad1836->regmap = devm_regmap_init_spi(spi, &ad1836_regmap_config);
        if (IS_ERR(ad1836->regmap))
                return PTR_ERR(ad1836->regmap);

        ad1836->type = spi_get_device_id(spi)->driver_data;

        spi_set_drvdata(spi, ad1836);

        ret = devm_snd_soc_register_component(&spi->dev,
                        &soc_component_dev_ad1836, &ad183x_dais[ad1836->type], 1);
        return ret;
}

static const struct spi_device_id ad1836_ids[] = {
        { "ad1835", AD1835 },
        { "ad1836", AD1836 },
        { "ad1837", AD1835 },
        { "ad1838", AD1838 },
        { "ad1839", AD1838 },
        { },
};
MODULE_DEVICE_TABLE(spi, ad1836_ids);

static struct spi_driver ad1836_spi_driver = {
        .driver = {
                .name   = "ad1836",
        },
        .probe          = ad1836_spi_probe,
        .id_table       = ad1836_ids,
};

module_spi_driver(ad1836_spi_driver);

MODULE_DESCRIPTION("ASoC ad1836 driver");
MODULE_AUTHOR("Barry Song <21cnbao@gmail.com>");
MODULE_LICENSE("GPL");