// SPDX-License-Identifier: GPL-2.0-only
/*
 * Texas Instruments TLV320AIC26 low power audio CODEC
 * ALSA SoC CODEC driver
 *
 * Copyright (C) 2008 Secret Lab Technologies Ltd.
 */

#include <linux/module.h>
#include <linux/moduleparam.h>
#include <linux/init.h>
#include <linux/delay.h>
#include <linux/pm.h>
#include <linux/device.h>
#include <linux/sysfs.h>
#include <linux/spi/spi.h>
#include <linux/slab.h>
#include <sound/core.h>
#include <sound/pcm.h>
#include <sound/pcm_params.h>
#include <sound/soc.h>
#include <sound/initval.h>

#include "tlv320aic26.h"

MODULE_DESCRIPTION("ASoC TLV320AIC26 codec driver");
MODULE_AUTHOR("Grant Likely <grant.likely@secretlab.ca>");
MODULE_LICENSE("GPL");

/* AIC26 driver private data */
struct aic26 {
        struct spi_device *spi;
        struct regmap *regmap;
        struct snd_soc_component *component;
        int clock_provider;
        int datfm;
        int mclk;

        /* Keyclick parameters */
        int keyclick_amplitude;
        int keyclick_freq;
        int keyclick_len;
};

static const struct snd_soc_dapm_widget tlv320aic26_dapm_widgets[] = {
SND_SOC_DAPM_INPUT("MICIN"),
SND_SOC_DAPM_INPUT("AUX"),

SND_SOC_DAPM_OUTPUT("HPL"),
SND_SOC_DAPM_OUTPUT("HPR"),
};

static const struct snd_soc_dapm_route tlv320aic26_dapm_routes[] = {
        { "Capture", NULL, "MICIN" },
        { "Capture", NULL, "AUX" },

        { "HPL", NULL, "Playback" },
        { "HPR", NULL, "Playback" },
};

/* ---------------------------------------------------------------------
 * Digital Audio Interface Operations
 */
static int aic26_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 aic26 *aic26 = snd_soc_component_get_drvdata(component);
        int fsref, divisor, wlen, pval, jval, dval, qval;
        u16 reg;

        dev_dbg(&aic26->spi->dev, "aic26_hw_params(substream=%p, params=%p)\n",
                substream, params);
        dev_dbg(&aic26->spi->dev, "rate=%i width=%d\n", params_rate(params),
                params_width(params));

        switch (params_rate(params)) {
        case 8000:  fsref = 48000; divisor = AIC26_DIV_6; break;
        case 11025: fsref = 44100; divisor = AIC26_DIV_4; break;
        case 12000: fsref = 48000; divisor = AIC26_DIV_4; break;
        case 16000: fsref = 48000; divisor = AIC26_DIV_3; break;
        case 22050: fsref = 44100; divisor = AIC26_DIV_2; break;
        case 24000: fsref = 48000; divisor = AIC26_DIV_2; break;
        case 32000: fsref = 48000; divisor = AIC26_DIV_1_5; break;
        case 44100: fsref = 44100; divisor = AIC26_DIV_1; break;
        case 48000: fsref = 48000; divisor = AIC26_DIV_1; break;
        default:
                dev_dbg(&aic26->spi->dev, "bad rate\n"); return -EINVAL;
        }

        /* select data word length */
        switch (params_width(params)) {
        case 8:  wlen = AIC26_WLEN_16; break;
        case 16: wlen = AIC26_WLEN_16; break;
        case 24: wlen = AIC26_WLEN_24; break;
        case 32: wlen = AIC26_WLEN_32; break;
        default:
                dev_dbg(&aic26->spi->dev, "bad format\n"); return -EINVAL;
        }

        /**
         * Configure PLL
         * fsref = (mclk * PLLM) / 2048
         * where PLLM = J.DDDD (DDDD register ranges from 0 to 9999, decimal)
         */
        pval = 1;
        /* compute J portion of multiplier */
        jval = fsref / (aic26->mclk / 2048);
        /* compute fractional DDDD component of multiplier */
        dval = fsref - (jval * (aic26->mclk / 2048));
        dval = (10000 * dval) / (aic26->mclk / 2048);
        dev_dbg(&aic26->spi->dev, "Setting PLLM to %d.%04d\n", jval, dval);
        qval = 0;
        reg = 0x8000 | qval << 11 | pval << 8 | jval << 2;
        snd_soc_component_write(component, AIC26_REG_PLL_PROG1, reg);
        reg = dval << 2;
        snd_soc_component_write(component, AIC26_REG_PLL_PROG2, reg);

        /* Audio Control 3 (clock provider mode, fsref rate) */
        if (aic26->clock_provider)
                reg = 0x0800;
        if (fsref == 48000)
                reg = 0x2000;
        snd_soc_component_update_bits(component, AIC26_REG_AUDIO_CTRL3, 0xf800, reg);

        /* Audio Control 1 (FSref divisor) */
        reg = wlen | aic26->datfm | (divisor << 3) | divisor;
        snd_soc_component_update_bits(component, AIC26_REG_AUDIO_CTRL1, 0xfff, reg);

        return 0;
}

/*
 * aic26_mute - Mute control to reduce noise when changing audio format
 */
static int aic26_mute(struct snd_soc_dai *dai, int mute, int direction)
{
        struct snd_soc_component *component = dai->component;
        struct aic26 *aic26 = snd_soc_component_get_drvdata(component);
        u16 reg;

        dev_dbg(&aic26->spi->dev, "aic26_mute(dai=%p, mute=%i)\n",
                dai, mute);

        if (mute)
                reg = 0x8080;
        else
                reg = 0;
        snd_soc_component_update_bits(component, AIC26_REG_DAC_GAIN, 0x8000, reg);

        return 0;
}

static int aic26_set_sysclk(struct snd_soc_dai *codec_dai,
                            int clk_id, unsigned int freq, int dir)
{
        struct snd_soc_component *component = codec_dai->component;
        struct aic26 *aic26 = snd_soc_component_get_drvdata(component);

        dev_dbg(&aic26->spi->dev, "aic26_set_sysclk(dai=%p, clk_id==%i,"
                " freq=%i, dir=%i)\n",
                codec_dai, clk_id, freq, dir);

        /* MCLK needs to fall between 2MHz and 50 MHz */
        if ((freq < 2000000) || (freq > 50000000))
                return -EINVAL;

        aic26->mclk = freq;
        return 0;
}

static int aic26_set_fmt(struct snd_soc_dai *codec_dai, unsigned int fmt)
{
        struct snd_soc_component *component = codec_dai->component;
        struct aic26 *aic26 = snd_soc_component_get_drvdata(component);

        dev_dbg(&aic26->spi->dev, "aic26_set_fmt(dai=%p, fmt==%i)\n",
                codec_dai, fmt);

        switch (fmt & SND_SOC_DAIFMT_CLOCK_PROVIDER_MASK) {
        case SND_SOC_DAIFMT_CBP_CFP: aic26->clock_provider = 1; break;
        case SND_SOC_DAIFMT_CBC_CFC: aic26->clock_provider = 0; break;
        default:
                dev_dbg(&aic26->spi->dev, "bad master\n"); return -EINVAL;
        }

        /* interface format */
        switch (fmt & SND_SOC_DAIFMT_FORMAT_MASK) {
        case SND_SOC_DAIFMT_I2S:     aic26->datfm = AIC26_DATFM_I2S; break;
        case SND_SOC_DAIFMT_DSP_A:   aic26->datfm = AIC26_DATFM_DSP; break;
        case SND_SOC_DAIFMT_RIGHT_J: aic26->datfm = AIC26_DATFM_RIGHTJ; break;
        case SND_SOC_DAIFMT_LEFT_J:  aic26->datfm = AIC26_DATFM_LEFTJ; break;
        default:
                dev_dbg(&aic26->spi->dev, "bad format\n"); return -EINVAL;
        }

        return 0;
}

/* ---------------------------------------------------------------------
 * Digital Audio Interface Definition
 */
#define AIC26_RATES     (SNDRV_PCM_RATE_8000  | SNDRV_PCM_RATE_11025 |\
                         SNDRV_PCM_RATE_16000 | SNDRV_PCM_RATE_22050 |\
                         SNDRV_PCM_RATE_32000 | SNDRV_PCM_RATE_44100 |\
                         SNDRV_PCM_RATE_48000)
#define AIC26_FORMATS   (SNDRV_PCM_FMTBIT_S8     | SNDRV_PCM_FMTBIT_S16_BE |\
                         SNDRV_PCM_FMTBIT_S24_BE | SNDRV_PCM_FMTBIT_S32_BE)

static const struct snd_soc_dai_ops aic26_dai_ops = {
        .hw_params      = aic26_hw_params,
        .mute_stream    = aic26_mute,
        .set_sysclk     = aic26_set_sysclk,
        .set_fmt        = aic26_set_fmt,
        .no_capture_mute = 1,
};

static struct snd_soc_dai_driver aic26_dai = {
        .name = "tlv320aic26-hifi",
        .playback = {
                .stream_name = "Playback",
                .channels_min = 2,
                .channels_max = 2,
                .rates = AIC26_RATES,
                .formats = AIC26_FORMATS,
        },
        .capture = {
                .stream_name = "Capture",
                .channels_min = 2,
                .channels_max = 2,
                .rates = AIC26_RATES,
                .formats = AIC26_FORMATS,
        },
        .ops = &aic26_dai_ops,
};

/* ---------------------------------------------------------------------
 * ALSA controls
 */
static const char *aic26_capture_src_text[] = {"Mic", "Aux"};
static SOC_ENUM_SINGLE_DECL(aic26_capture_src_enum,
                            AIC26_REG_AUDIO_CTRL1, 12,
                            aic26_capture_src_text);

static const struct snd_kcontrol_new aic26_snd_controls[] = {
        /* Output */
        SOC_DOUBLE("PCM Playback Volume", AIC26_REG_DAC_GAIN, 8, 0, 0x7f, 1),
        SOC_DOUBLE("PCM Playback Switch", AIC26_REG_DAC_GAIN, 15, 7, 1, 1),
        SOC_SINGLE("PCM Capture Volume", AIC26_REG_ADC_GAIN, 8, 0x7f, 0),
        SOC_SINGLE("PCM Capture Mute", AIC26_REG_ADC_GAIN, 15, 1, 1),
        SOC_SINGLE("Keyclick activate", AIC26_REG_AUDIO_CTRL2, 15, 0x1, 0),
        SOC_SINGLE("Keyclick amplitude", AIC26_REG_AUDIO_CTRL2, 12, 0x7, 0),
        SOC_SINGLE("Keyclick frequency", AIC26_REG_AUDIO_CTRL2, 8, 0x7, 0),
        SOC_SINGLE("Keyclick period", AIC26_REG_AUDIO_CTRL2, 4, 0xf, 0),
        SOC_ENUM("Capture Source", aic26_capture_src_enum),
};

/* ---------------------------------------------------------------------
 * SPI device portion of driver: sysfs files for debugging
 */

static ssize_t keyclick_show(struct device *dev,
                             struct device_attribute *attr, char *buf)
{
        struct aic26 *aic26 = dev_get_drvdata(dev);
        int val, amp, freq, len;

        val = snd_soc_component_read(aic26->component, AIC26_REG_AUDIO_CTRL2);
        amp = (val >> 12) & 0x7;
        freq = (125 << ((val >> 8) & 0x7)) >> 1;
        len = 2 * (1 + ((val >> 4) & 0xf));

        return sysfs_emit(buf, "amp=%x freq=%iHz len=%iclks\n", amp, freq, len);
}

/* Any write to the keyclick attribute will trigger the keyclick event */
static ssize_t keyclick_store(struct device *dev,
                              struct device_attribute *attr,
                              const char *buf, size_t count)
{
        struct aic26 *aic26 = dev_get_drvdata(dev);

        snd_soc_component_update_bits(aic26->component, AIC26_REG_AUDIO_CTRL2,
                            0x8000, 0x800);

        return count;
}

static DEVICE_ATTR_RW(keyclick);

/* ---------------------------------------------------------------------
 * SoC CODEC portion of driver: probe and release routines
 */
static int aic26_probe(struct snd_soc_component *component)
{
        struct aic26 *aic26 = dev_get_drvdata(component->dev);
        int ret, reg;

        aic26->component = component;

        /* Reset the codec to power on defaults */
        snd_soc_component_write(component, AIC26_REG_RESET, 0xBB00);

        /* Power up CODEC */
        snd_soc_component_write(component, AIC26_REG_POWER_CTRL, 0);

        /* Audio Control 3 (master mode, fsref rate) */
        reg = snd_soc_component_read(component, AIC26_REG_AUDIO_CTRL3);
        reg &= ~0xf800;
        reg |= 0x0800; /* set master mode */
        snd_soc_component_write(component, AIC26_REG_AUDIO_CTRL3, reg);

        /* Register the sysfs files for debugging */
        /* Create SysFS files */
        ret = device_create_file(component->dev, &dev_attr_keyclick);
        if (ret)
                dev_info(component->dev, "error creating sysfs files\n");

        return 0;
}

static const struct snd_soc_component_driver aic26_soc_component_dev = {
        .probe                  = aic26_probe,
        .controls               = aic26_snd_controls,
        .num_controls           = ARRAY_SIZE(aic26_snd_controls),
        .dapm_widgets           = tlv320aic26_dapm_widgets,
        .num_dapm_widgets       = ARRAY_SIZE(tlv320aic26_dapm_widgets),
        .dapm_routes            = tlv320aic26_dapm_routes,
        .num_dapm_routes        = ARRAY_SIZE(tlv320aic26_dapm_routes),
        .idle_bias_on           = 1,
        .use_pmdown_time        = 1,
        .endianness             = 1,
};

static const struct regmap_config aic26_regmap = {
        .reg_bits = 16,
        .val_bits = 16,
};

/* ---------------------------------------------------------------------
 * SPI device portion of driver: probe and release routines and SPI
 *                               driver registration.
 */
static int aic26_spi_probe(struct spi_device *spi)
{
        struct aic26 *aic26;
        int ret;

        dev_dbg(&spi->dev, "probing tlv320aic26 spi device\n");

        /* Allocate driver data */
        aic26 = devm_kzalloc(&spi->dev, sizeof *aic26, GFP_KERNEL);
        if (!aic26)
                return -ENOMEM;

        aic26->regmap = devm_regmap_init_spi(spi, &aic26_regmap);
        if (IS_ERR(aic26->regmap))
                return PTR_ERR(aic26->regmap);

        /* Initialize the driver data */
        aic26->spi = spi;
        dev_set_drvdata(&spi->dev, aic26);
        aic26->clock_provider = 1;

        ret = devm_snd_soc_register_component(&spi->dev,
                        &aic26_soc_component_dev, &aic26_dai, 1);
        return ret;
}

static struct spi_driver aic26_spi = {
        .driver = {
                .name = "tlv320aic26-codec",
        },
        .probe = aic26_spi_probe,
};

module_spi_driver(aic26_spi);