root/sound/soc/tegra/tegra210_adx.c
// SPDX-License-Identifier: GPL-2.0-only
// SPDX-FileCopyrightText: Copyright (c) 2021-2025 NVIDIA CORPORATION & AFFILIATES.
// All rights reserved.
//
// tegra210_adx.c - Tegra210 ADX driver

#include <linux/bits.h>
#include <linux/clk.h>
#include <linux/device.h>
#include <linux/io.h>
#include <linux/module.h>
#include <linux/of_device.h>
#include <linux/platform_device.h>
#include <linux/pm_runtime.h>
#include <linux/regmap.h>
#include <sound/core.h>
#include <sound/pcm.h>
#include <sound/pcm_params.h>
#include <sound/soc.h>

#include "tegra210_adx.h"
#include "tegra_cif.h"

static const struct reg_default tegra210_adx_reg_defaults[] = {
        { TEGRA210_ADX_RX_INT_MASK, 0x00000001},
        { TEGRA210_ADX_RX_CIF_CTRL, 0x00007000},
        { TEGRA210_ADX_TX_INT_MASK, 0x0000000f },
        { TEGRA210_ADX_TX1_CIF_CTRL, 0x00007000},
        { TEGRA210_ADX_TX2_CIF_CTRL, 0x00007000},
        { TEGRA210_ADX_TX3_CIF_CTRL, 0x00007000},
        { TEGRA210_ADX_TX4_CIF_CTRL, 0x00007000},
        { TEGRA210_ADX_CG, 0x1},
        { TEGRA210_ADX_CFG_RAM_CTRL, 0x00004000},
};

static const struct reg_default tegra264_adx_reg_defaults[] = {
        { TEGRA210_ADX_RX_INT_MASK, 0x00000001},
        { TEGRA210_ADX_RX_CIF_CTRL, 0x00003800},
        { TEGRA210_ADX_TX_INT_MASK, 0x0000000f },
        { TEGRA210_ADX_TX1_CIF_CTRL, 0x00003800},
        { TEGRA210_ADX_TX2_CIF_CTRL, 0x00003800},
        { TEGRA210_ADX_TX3_CIF_CTRL, 0x00003800},
        { TEGRA210_ADX_TX4_CIF_CTRL, 0x00003800},
        { TEGRA210_ADX_CG, 0x1},
        { TEGRA264_ADX_CFG_RAM_CTRL, 0x00004000},
};

static void tegra210_adx_write_map_ram(struct tegra210_adx *adx)
{
        const unsigned int bits_per_mask = BITS_PER_TYPE(*adx->byte_mask);
        int i;

        memset(adx->byte_mask, 0,
               adx->soc_data->byte_mask_size * sizeof(*adx->byte_mask));

        regmap_write(adx->regmap, TEGRA210_ADX_CFG_RAM_CTRL +
                        adx->soc_data->cya_offset,
                     TEGRA210_ADX_CFG_RAM_CTRL_SEQ_ACCESS_EN |
                     TEGRA210_ADX_CFG_RAM_CTRL_ADDR_INIT_EN |
                     TEGRA210_ADX_CFG_RAM_CTRL_RW_WRITE);

        for (i = 0; i < adx->soc_data->ram_depth; i++) {
                u32 word = 0;
                int b;

                for (b = 0; b < TEGRA_ADX_SLOTS_PER_WORD; b++) {
                        unsigned int slot = i * TEGRA_ADX_SLOTS_PER_WORD + b;
                        u16 val = adx->map[slot];

                        if (val >= 256)
                                continue;

                        word |= (u32)val << (b * BITS_PER_BYTE);
                        adx->byte_mask[slot / bits_per_mask] |=
                                1U << (slot % bits_per_mask);
                }
                regmap_write(adx->regmap, TEGRA210_ADX_CFG_RAM_DATA +
                                adx->soc_data->cya_offset, word);
        }

        for (i = 0; i < adx->soc_data->byte_mask_size; i++)
                regmap_write(adx->regmap,
                             TEGRA210_ADX_IN_BYTE_EN0 + (i * TEGRA210_ADX_AUDIOCIF_CH_STRIDE),
                             adx->byte_mask[i]);
}

static int tegra210_adx_startup(struct snd_pcm_substream *substream,
                                struct snd_soc_dai *dai)
{
        struct tegra210_adx *adx = snd_soc_dai_get_drvdata(dai);
        unsigned int val;
        int err;

        /* Ensure if ADX status is disabled */
        err = regmap_read_poll_timeout(adx->regmap, TEGRA210_ADX_STATUS,
                                       val, !(val & 0x1), 10, 10000);
        if (err < 0) {
                dev_err(dai->dev, "failed to stop ADX, err = %d\n", err);
                return err;
        }

        /*
         * Soft Reset: Below performs module soft reset which clears
         * all FSM logic, flushes flow control of FIFO and resets the
         * state register. It also brings module back to disabled
         * state (without flushing the data in the pipe).
         */
        regmap_update_bits(adx->regmap, TEGRA210_ADX_SOFT_RESET,
                           TEGRA210_ADX_SOFT_RESET_SOFT_RESET_MASK,
                           TEGRA210_ADX_SOFT_RESET_SOFT_EN);

        err = regmap_read_poll_timeout(adx->regmap, TEGRA210_ADX_SOFT_RESET,
                                       val, !(val & 0x1), 10, 10000);
        if (err < 0) {
                dev_err(dai->dev, "failed to reset ADX, err = %d\n", err);
                return err;
        }

        return 0;
}

static int tegra210_adx_runtime_suspend(struct device *dev)
{
        struct tegra210_adx *adx = dev_get_drvdata(dev);

        regcache_cache_only(adx->regmap, true);
        regcache_mark_dirty(adx->regmap);

        return 0;
}

static int tegra210_adx_runtime_resume(struct device *dev)
{
        struct tegra210_adx *adx = dev_get_drvdata(dev);

        regcache_cache_only(adx->regmap, false);
        regcache_sync(adx->regmap);

        tegra210_adx_write_map_ram(adx);

        return 0;
}

static int tegra210_adx_set_audio_cif(struct snd_soc_dai *dai,
                                      unsigned int channels,
                                      snd_pcm_format_t format,
                                      unsigned int reg)
{
        struct tegra210_adx *adx = snd_soc_dai_get_drvdata(dai);
        struct tegra_cif_conf cif_conf;
        int audio_bits;

        memset(&cif_conf, 0, sizeof(struct tegra_cif_conf));

        if (channels < 1 || channels > adx->soc_data->max_ch) {
                dev_err(dai->dev, "invalid channels: %u (max %u)\n",
                        channels, adx->soc_data->max_ch);
                return -EINVAL;
        }

        switch (format) {
        case SNDRV_PCM_FORMAT_S8:
                audio_bits = TEGRA_ACIF_BITS_8;
                break;
        case SNDRV_PCM_FORMAT_S16_LE:
                audio_bits = TEGRA_ACIF_BITS_16;
                break;
        case SNDRV_PCM_FORMAT_S24_LE:
        case SNDRV_PCM_FORMAT_S32_LE:
                audio_bits = TEGRA_ACIF_BITS_32;
                break;
        default:
                dev_err(dai->dev, "unsupported format: %d\n", format);
                return -EINVAL;
        }

        cif_conf.audio_ch = channels;
        cif_conf.client_ch = channels;
        cif_conf.audio_bits = audio_bits;
        cif_conf.client_bits = audio_bits;

        if (adx->soc_data->max_ch == 32)
                tegra264_set_cif(adx->regmap, reg, &cif_conf);
        else
                tegra_set_cif(adx->regmap, reg, &cif_conf);

        return 0;
}

static int tegra210_adx_out_hw_params(struct snd_pcm_substream *substream,
                                      struct snd_pcm_hw_params *params,
                                      struct snd_soc_dai *dai)
{
        return tegra210_adx_set_audio_cif(dai, params_channels(params),
                        params_format(params),
                        TEGRA210_ADX_TX1_CIF_CTRL + ((dai->id - 1) * TEGRA210_ADX_AUDIOCIF_CH_STRIDE));
}

static int tegra210_adx_in_hw_params(struct snd_pcm_substream *substream,
                                     struct snd_pcm_hw_params *params,
                                     struct snd_soc_dai *dai)
{
        return tegra210_adx_set_audio_cif(dai, params_channels(params),
                                          params_format(params),
                                          TEGRA210_ADX_RX_CIF_CTRL);
}

static int tegra210_adx_get_byte_map(struct snd_kcontrol *kcontrol,
                                     struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *cmpnt = snd_kcontrol_chip(kcontrol);
        struct tegra210_adx *adx = snd_soc_component_get_drvdata(cmpnt);
        struct soc_mixer_control *mc =
                (struct soc_mixer_control *)kcontrol->private_value;

        ucontrol->value.integer.value[0] = adx->map[mc->reg];

        return 0;
}

static int tegra210_adx_put_byte_map(struct snd_kcontrol *kcontrol,
                                     struct snd_ctl_elem_value *ucontrol)
{
        struct snd_soc_component *cmpnt = snd_kcontrol_chip(kcontrol);
        struct tegra210_adx *adx = snd_soc_component_get_drvdata(cmpnt);
        struct soc_mixer_control *mc =
                (struct soc_mixer_control *)kcontrol->private_value;
        unsigned int value = ucontrol->value.integer.value[0];

        /*
         * Match the previous behaviour: any value outside [0, 255] is
         * treated as the "disabled" sentinel (256). Negative values from
         * userspace fold in through the unsigned cast and are caught here.
         */
        if (value > 255)
                value = 256;

        if (adx->map[mc->reg] == value)
                return 0;

        adx->map[mc->reg] = value;

        return 1;
}

static const struct snd_soc_dai_ops tegra210_adx_in_dai_ops = {
        .hw_params      = tegra210_adx_in_hw_params,
        .startup        = tegra210_adx_startup,
};

static const struct snd_soc_dai_ops tegra210_adx_out_dai_ops = {
        .hw_params      = tegra210_adx_out_hw_params,
};

#define IN_DAI                                                  \
        {                                                       \
                .name = "ADX-RX-CIF",                           \
                .playback = {                                   \
                        .stream_name = "RX-CIF-Playback",       \
                        .channels_min = 1,                      \
                        .channels_max = 16,                     \
                        .rates = SNDRV_PCM_RATE_8000_192000,    \
                        .formats = SNDRV_PCM_FMTBIT_S8 |        \
                                   SNDRV_PCM_FMTBIT_S16_LE |    \
                                   SNDRV_PCM_FMTBIT_S24_LE |    \
                                   SNDRV_PCM_FMTBIT_S32_LE,     \
                },                                              \
                .capture = {                                    \
                        .stream_name = "RX-CIF-Capture",        \
                        .channels_min = 1,                      \
                        .channels_max = 16,                     \
                        .rates = SNDRV_PCM_RATE_8000_192000,    \
                        .formats = SNDRV_PCM_FMTBIT_S8 |        \
                                   SNDRV_PCM_FMTBIT_S16_LE |    \
                                   SNDRV_PCM_FMTBIT_S24_LE |    \
                                   SNDRV_PCM_FMTBIT_S32_LE,     \
                },                                              \
                .ops = &tegra210_adx_in_dai_ops,                \
        }

#define OUT_DAI(id)                                             \
        {                                                       \
                .name = "ADX-TX" #id "-CIF",                    \
                .playback = {                                   \
                        .stream_name = "TX" #id "-CIF-Playback",\
                        .channels_min = 1,                      \
                        .channels_max = 16,                     \
                        .rates = SNDRV_PCM_RATE_8000_192000,    \
                        .formats = SNDRV_PCM_FMTBIT_S8 |        \
                                   SNDRV_PCM_FMTBIT_S16_LE |    \
                                   SNDRV_PCM_FMTBIT_S24_LE |    \
                                   SNDRV_PCM_FMTBIT_S32_LE,     \
                },                                              \
                .capture = {                                    \
                        .stream_name = "TX" #id "-CIF-Capture", \
                        .channels_min = 1,                      \
                        .channels_max = 16,                     \
                        .rates = SNDRV_PCM_RATE_8000_192000,    \
                        .formats = SNDRV_PCM_FMTBIT_S8 |        \
                                   SNDRV_PCM_FMTBIT_S16_LE |    \
                                   SNDRV_PCM_FMTBIT_S24_LE |    \
                                   SNDRV_PCM_FMTBIT_S32_LE,     \
                },                                              \
                .ops = &tegra210_adx_out_dai_ops,               \
        }

static struct snd_soc_dai_driver tegra210_adx_dais[] = {
        IN_DAI,
        OUT_DAI(1),
        OUT_DAI(2),
        OUT_DAI(3),
        OUT_DAI(4),
};

static const struct snd_soc_dapm_widget tegra210_adx_widgets[] = {
        SND_SOC_DAPM_AIF_IN("RX", NULL, 0, TEGRA210_ADX_ENABLE,
                            TEGRA210_ADX_ENABLE_SHIFT, 0),
        SND_SOC_DAPM_AIF_OUT("TX1", NULL, 0, TEGRA210_ADX_CTRL, 0, 0),
        SND_SOC_DAPM_AIF_OUT("TX2", NULL, 0, TEGRA210_ADX_CTRL, 1, 0),
        SND_SOC_DAPM_AIF_OUT("TX3", NULL, 0, TEGRA210_ADX_CTRL, 2, 0),
        SND_SOC_DAPM_AIF_OUT("TX4", NULL, 0, TEGRA210_ADX_CTRL, 3, 0),
};

#define STREAM_ROUTES(id, sname)                                          \
        { "XBAR-" sname,                NULL,   "XBAR-TX" },              \
        { "RX-CIF-" sname,              NULL,   "XBAR-" sname },          \
        { "RX",                         NULL,   "RX-CIF-" sname },        \
        { "TX" #id,                     NULL,   "RX" },                   \
        { "TX" #id "-CIF-" sname,       NULL,   "TX" #id },               \
        { "TX" #id " XBAR-" sname,      NULL,   "TX" #id "-CIF-" sname }, \
        { "TX" #id " XBAR-RX",          NULL,   "TX" #id " XBAR-" sname }

#define ADX_ROUTES(id)                  \
        STREAM_ROUTES(id, "Playback"),  \
        STREAM_ROUTES(id, "Capture")

#define STREAM_ROUTES(id, sname)                                          \
        { "XBAR-" sname,                NULL,   "XBAR-TX" },              \
        { "RX-CIF-" sname,              NULL,   "XBAR-" sname },          \
        { "RX",                         NULL,   "RX-CIF-" sname },        \
        { "TX" #id,                     NULL,   "RX" },                   \
        { "TX" #id "-CIF-" sname,       NULL,   "TX" #id },               \
        { "TX" #id " XBAR-" sname,      NULL,   "TX" #id "-CIF-" sname }, \
        { "TX" #id " XBAR-RX",          NULL,   "TX" #id " XBAR-" sname }

#define ADX_ROUTES(id)                  \
        STREAM_ROUTES(id, "Playback"),  \
        STREAM_ROUTES(id, "Capture")

static const struct snd_soc_dapm_route tegra210_adx_routes[] = {
        ADX_ROUTES(1),
        ADX_ROUTES(2),
        ADX_ROUTES(3),
        ADX_ROUTES(4),
};

#define TEGRA210_ADX_BYTE_MAP_CTRL(reg)                  \
        SOC_SINGLE_EXT("Byte Map " #reg, reg, 0, 256, 0, \
                       tegra210_adx_get_byte_map,        \
                       tegra210_adx_put_byte_map)

static struct snd_kcontrol_new tegra210_adx_controls[] = {
        TEGRA210_ADX_BYTE_MAP_CTRL(0),
        TEGRA210_ADX_BYTE_MAP_CTRL(1),
        TEGRA210_ADX_BYTE_MAP_CTRL(2),
        TEGRA210_ADX_BYTE_MAP_CTRL(3),
        TEGRA210_ADX_BYTE_MAP_CTRL(4),
        TEGRA210_ADX_BYTE_MAP_CTRL(5),
        TEGRA210_ADX_BYTE_MAP_CTRL(6),
        TEGRA210_ADX_BYTE_MAP_CTRL(7),
        TEGRA210_ADX_BYTE_MAP_CTRL(8),
        TEGRA210_ADX_BYTE_MAP_CTRL(9),
        TEGRA210_ADX_BYTE_MAP_CTRL(10),
        TEGRA210_ADX_BYTE_MAP_CTRL(11),
        TEGRA210_ADX_BYTE_MAP_CTRL(12),
        TEGRA210_ADX_BYTE_MAP_CTRL(13),
        TEGRA210_ADX_BYTE_MAP_CTRL(14),
        TEGRA210_ADX_BYTE_MAP_CTRL(15),
        TEGRA210_ADX_BYTE_MAP_CTRL(16),
        TEGRA210_ADX_BYTE_MAP_CTRL(17),
        TEGRA210_ADX_BYTE_MAP_CTRL(18),
        TEGRA210_ADX_BYTE_MAP_CTRL(19),
        TEGRA210_ADX_BYTE_MAP_CTRL(20),
        TEGRA210_ADX_BYTE_MAP_CTRL(21),
        TEGRA210_ADX_BYTE_MAP_CTRL(22),
        TEGRA210_ADX_BYTE_MAP_CTRL(23),
        TEGRA210_ADX_BYTE_MAP_CTRL(24),
        TEGRA210_ADX_BYTE_MAP_CTRL(25),
        TEGRA210_ADX_BYTE_MAP_CTRL(26),
        TEGRA210_ADX_BYTE_MAP_CTRL(27),
        TEGRA210_ADX_BYTE_MAP_CTRL(28),
        TEGRA210_ADX_BYTE_MAP_CTRL(29),
        TEGRA210_ADX_BYTE_MAP_CTRL(30),
        TEGRA210_ADX_BYTE_MAP_CTRL(31),
        TEGRA210_ADX_BYTE_MAP_CTRL(32),
        TEGRA210_ADX_BYTE_MAP_CTRL(33),
        TEGRA210_ADX_BYTE_MAP_CTRL(34),
        TEGRA210_ADX_BYTE_MAP_CTRL(35),
        TEGRA210_ADX_BYTE_MAP_CTRL(36),
        TEGRA210_ADX_BYTE_MAP_CTRL(37),
        TEGRA210_ADX_BYTE_MAP_CTRL(38),
        TEGRA210_ADX_BYTE_MAP_CTRL(39),
        TEGRA210_ADX_BYTE_MAP_CTRL(40),
        TEGRA210_ADX_BYTE_MAP_CTRL(41),
        TEGRA210_ADX_BYTE_MAP_CTRL(42),
        TEGRA210_ADX_BYTE_MAP_CTRL(43),
        TEGRA210_ADX_BYTE_MAP_CTRL(44),
        TEGRA210_ADX_BYTE_MAP_CTRL(45),
        TEGRA210_ADX_BYTE_MAP_CTRL(46),
        TEGRA210_ADX_BYTE_MAP_CTRL(47),
        TEGRA210_ADX_BYTE_MAP_CTRL(48),
        TEGRA210_ADX_BYTE_MAP_CTRL(49),
        TEGRA210_ADX_BYTE_MAP_CTRL(50),
        TEGRA210_ADX_BYTE_MAP_CTRL(51),
        TEGRA210_ADX_BYTE_MAP_CTRL(52),
        TEGRA210_ADX_BYTE_MAP_CTRL(53),
        TEGRA210_ADX_BYTE_MAP_CTRL(54),
        TEGRA210_ADX_BYTE_MAP_CTRL(55),
        TEGRA210_ADX_BYTE_MAP_CTRL(56),
        TEGRA210_ADX_BYTE_MAP_CTRL(57),
        TEGRA210_ADX_BYTE_MAP_CTRL(58),
        TEGRA210_ADX_BYTE_MAP_CTRL(59),
        TEGRA210_ADX_BYTE_MAP_CTRL(60),
        TEGRA210_ADX_BYTE_MAP_CTRL(61),
        TEGRA210_ADX_BYTE_MAP_CTRL(62),
        TEGRA210_ADX_BYTE_MAP_CTRL(63),
};

static struct snd_kcontrol_new tegra264_adx_controls[] = {
        TEGRA210_ADX_BYTE_MAP_CTRL(64),
        TEGRA210_ADX_BYTE_MAP_CTRL(65),
        TEGRA210_ADX_BYTE_MAP_CTRL(66),
        TEGRA210_ADX_BYTE_MAP_CTRL(67),
        TEGRA210_ADX_BYTE_MAP_CTRL(68),
        TEGRA210_ADX_BYTE_MAP_CTRL(69),
        TEGRA210_ADX_BYTE_MAP_CTRL(70),
        TEGRA210_ADX_BYTE_MAP_CTRL(71),
        TEGRA210_ADX_BYTE_MAP_CTRL(72),
        TEGRA210_ADX_BYTE_MAP_CTRL(73),
        TEGRA210_ADX_BYTE_MAP_CTRL(74),
        TEGRA210_ADX_BYTE_MAP_CTRL(75),
        TEGRA210_ADX_BYTE_MAP_CTRL(76),
        TEGRA210_ADX_BYTE_MAP_CTRL(77),
        TEGRA210_ADX_BYTE_MAP_CTRL(78),
        TEGRA210_ADX_BYTE_MAP_CTRL(79),
        TEGRA210_ADX_BYTE_MAP_CTRL(80),
        TEGRA210_ADX_BYTE_MAP_CTRL(81),
        TEGRA210_ADX_BYTE_MAP_CTRL(82),
        TEGRA210_ADX_BYTE_MAP_CTRL(83),
        TEGRA210_ADX_BYTE_MAP_CTRL(84),
        TEGRA210_ADX_BYTE_MAP_CTRL(85),
        TEGRA210_ADX_BYTE_MAP_CTRL(86),
        TEGRA210_ADX_BYTE_MAP_CTRL(87),
        TEGRA210_ADX_BYTE_MAP_CTRL(88),
        TEGRA210_ADX_BYTE_MAP_CTRL(89),
        TEGRA210_ADX_BYTE_MAP_CTRL(90),
        TEGRA210_ADX_BYTE_MAP_CTRL(91),
        TEGRA210_ADX_BYTE_MAP_CTRL(92),
        TEGRA210_ADX_BYTE_MAP_CTRL(93),
        TEGRA210_ADX_BYTE_MAP_CTRL(94),
        TEGRA210_ADX_BYTE_MAP_CTRL(95),
        TEGRA210_ADX_BYTE_MAP_CTRL(96),
        TEGRA210_ADX_BYTE_MAP_CTRL(97),
        TEGRA210_ADX_BYTE_MAP_CTRL(98),
        TEGRA210_ADX_BYTE_MAP_CTRL(99),
        TEGRA210_ADX_BYTE_MAP_CTRL(100),
        TEGRA210_ADX_BYTE_MAP_CTRL(101),
        TEGRA210_ADX_BYTE_MAP_CTRL(102),
        TEGRA210_ADX_BYTE_MAP_CTRL(103),
        TEGRA210_ADX_BYTE_MAP_CTRL(104),
        TEGRA210_ADX_BYTE_MAP_CTRL(105),
        TEGRA210_ADX_BYTE_MAP_CTRL(106),
        TEGRA210_ADX_BYTE_MAP_CTRL(107),
        TEGRA210_ADX_BYTE_MAP_CTRL(108),
        TEGRA210_ADX_BYTE_MAP_CTRL(109),
        TEGRA210_ADX_BYTE_MAP_CTRL(110),
        TEGRA210_ADX_BYTE_MAP_CTRL(111),
        TEGRA210_ADX_BYTE_MAP_CTRL(112),
        TEGRA210_ADX_BYTE_MAP_CTRL(113),
        TEGRA210_ADX_BYTE_MAP_CTRL(114),
        TEGRA210_ADX_BYTE_MAP_CTRL(115),
        TEGRA210_ADX_BYTE_MAP_CTRL(116),
        TEGRA210_ADX_BYTE_MAP_CTRL(117),
        TEGRA210_ADX_BYTE_MAP_CTRL(118),
        TEGRA210_ADX_BYTE_MAP_CTRL(119),
        TEGRA210_ADX_BYTE_MAP_CTRL(120),
        TEGRA210_ADX_BYTE_MAP_CTRL(121),
        TEGRA210_ADX_BYTE_MAP_CTRL(122),
        TEGRA210_ADX_BYTE_MAP_CTRL(123),
        TEGRA210_ADX_BYTE_MAP_CTRL(124),
        TEGRA210_ADX_BYTE_MAP_CTRL(125),
        TEGRA210_ADX_BYTE_MAP_CTRL(126),
        TEGRA210_ADX_BYTE_MAP_CTRL(127),
};

static int tegra210_adx_component_probe(struct snd_soc_component *component)
{
        struct tegra210_adx *adx = snd_soc_component_get_drvdata(component);
        int err = 0;

        if (adx->soc_data->num_controls) {
                err = snd_soc_add_component_controls(component, adx->soc_data->controls,
                                                     adx->soc_data->num_controls);
                if (err)
                        dev_err(component->dev, "can't add ADX controls, err: %d\n", err);
        }

        return err;
}

static const struct snd_soc_component_driver tegra210_adx_cmpnt = {
        .probe                  = tegra210_adx_component_probe,
        .dapm_widgets           = tegra210_adx_widgets,
        .num_dapm_widgets       = ARRAY_SIZE(tegra210_adx_widgets),
        .dapm_routes            = tegra210_adx_routes,
        .num_dapm_routes        = ARRAY_SIZE(tegra210_adx_routes),
        .controls               = tegra210_adx_controls,
        .num_controls           = ARRAY_SIZE(tegra210_adx_controls),
};

static bool tegra210_adx_wr_reg(struct device *dev,
                                unsigned int reg)
{
        switch (reg) {
        case TEGRA210_ADX_TX_INT_MASK ... TEGRA210_ADX_TX4_CIF_CTRL:
        case TEGRA210_ADX_RX_INT_MASK ... TEGRA210_ADX_RX_CIF_CTRL:
        case TEGRA210_ADX_ENABLE ... TEGRA210_ADX_CG:
        case TEGRA210_ADX_CTRL ... TEGRA210_ADX_IN_BYTE_EN1:
        case TEGRA210_ADX_CFG_RAM_CTRL ... TEGRA210_ADX_CFG_RAM_DATA:
                return true;
        default:
                return false;
        }
}

static bool tegra210_adx_rd_reg(struct device *dev,
                                unsigned int reg)
{
        switch (reg) {
        case TEGRA210_ADX_RX_STATUS ... TEGRA210_ADX_CFG_RAM_DATA:
                return true;
        default:
                return false;
        }
}

static bool tegra210_adx_volatile_reg(struct device *dev,
                                unsigned int reg)
{
        switch (reg) {
        case TEGRA210_ADX_RX_STATUS:
        case TEGRA210_ADX_RX_INT_STATUS:
        case TEGRA210_ADX_RX_INT_SET:
        case TEGRA210_ADX_TX_STATUS:
        case TEGRA210_ADX_TX_INT_STATUS:
        case TEGRA210_ADX_TX_INT_SET:
        case TEGRA210_ADX_SOFT_RESET:
        case TEGRA210_ADX_STATUS:
        case TEGRA210_ADX_INT_STATUS:
        case TEGRA210_ADX_CFG_RAM_CTRL:
        case TEGRA210_ADX_CFG_RAM_DATA:
                return true;
        default:
                break;
        }

        return false;
}

static bool tegra264_adx_wr_reg(struct device *dev,
                                unsigned int reg)
{
        switch (reg) {
        case TEGRA210_ADX_TX_INT_MASK ... TEGRA210_ADX_TX4_CIF_CTRL:
        case TEGRA210_ADX_RX_INT_MASK ... TEGRA210_ADX_RX_CIF_CTRL:
        case TEGRA210_ADX_ENABLE ... TEGRA210_ADX_CG:
        case TEGRA210_ADX_CTRL ... TEGRA264_ADX_CYA:
        case TEGRA264_ADX_CFG_RAM_CTRL ... TEGRA264_ADX_CFG_RAM_DATA:
                return true;
        default:
                return false;
        }
}

static bool tegra264_adx_rd_reg(struct device *dev,
                                unsigned int reg)
{
        switch (reg) {
        case TEGRA210_ADX_RX_STATUS ... TEGRA210_ADX_RX_CIF_CTRL:
        case TEGRA210_ADX_TX_STATUS ... TEGRA210_ADX_TX4_CIF_CTRL:
        case TEGRA210_ADX_ENABLE ... TEGRA210_ADX_INT_STATUS:
        case TEGRA210_ADX_CTRL ... TEGRA264_ADX_CFG_RAM_DATA:
                return true;
        default:
                return false;
        }
}

static bool tegra264_adx_volatile_reg(struct device *dev,
                                      unsigned int reg)
{
        switch (reg) {
        case TEGRA210_ADX_RX_STATUS:
        case TEGRA210_ADX_RX_INT_STATUS:
        case TEGRA210_ADX_RX_INT_SET:
        case TEGRA210_ADX_TX_STATUS:
        case TEGRA210_ADX_TX_INT_STATUS:
        case TEGRA210_ADX_TX_INT_SET:
        case TEGRA210_ADX_SOFT_RESET:
        case TEGRA210_ADX_STATUS:
        case TEGRA210_ADX_INT_STATUS:
        case TEGRA264_ADX_CFG_RAM_CTRL:
        case TEGRA264_ADX_CFG_RAM_DATA:
                return true;
        default:
                break;
        }

        return false;
}

static const struct regmap_config tegra210_adx_regmap_config = {
        .reg_bits               = 32,
        .reg_stride             = 4,
        .val_bits               = 32,
        .max_register           = TEGRA210_ADX_CFG_RAM_DATA,
        .writeable_reg          = tegra210_adx_wr_reg,
        .readable_reg           = tegra210_adx_rd_reg,
        .volatile_reg           = tegra210_adx_volatile_reg,
        .reg_defaults           = tegra210_adx_reg_defaults,
        .num_reg_defaults       = ARRAY_SIZE(tegra210_adx_reg_defaults),
        .reg_default_cb         = regmap_default_zero_cb,
        .cache_type             = REGCACHE_FLAT,
};

static const struct regmap_config tegra264_adx_regmap_config = {
        .reg_bits               = 32,
        .reg_stride             = 4,
        .val_bits               = 32,
        .max_register           = TEGRA264_ADX_CFG_RAM_DATA,
        .writeable_reg          = tegra264_adx_wr_reg,
        .readable_reg           = tegra264_adx_rd_reg,
        .volatile_reg           = tegra264_adx_volatile_reg,
        .reg_defaults           = tegra264_adx_reg_defaults,
        .num_reg_defaults       = ARRAY_SIZE(tegra264_adx_reg_defaults),
        .reg_default_cb         = regmap_default_zero_cb,
        .cache_type             = REGCACHE_FLAT,
};

static const struct tegra210_adx_soc_data soc_data_tegra210 = {
        .regmap_conf            = &tegra210_adx_regmap_config,
        .max_ch                 = TEGRA210_ADX_MAX_CHANNEL,
        .ram_depth              = TEGRA210_ADX_RAM_DEPTH,
        .byte_mask_size         = TEGRA210_ADX_BYTE_MASK_COUNT,
        .cya_offset             = TEGRA210_ADX_CYA_OFFSET,
};

static const struct tegra210_adx_soc_data soc_data_tegra264 = {
        .regmap_conf            = &tegra264_adx_regmap_config,
        .max_ch                 = TEGRA264_ADX_MAX_CHANNEL,
        .ram_depth              = TEGRA264_ADX_RAM_DEPTH,
        .byte_mask_size         = TEGRA264_ADX_BYTE_MASK_COUNT,
        .cya_offset             = TEGRA264_ADX_CYA_OFFSET,
        .controls               = tegra264_adx_controls,
        .num_controls           = ARRAY_SIZE(tegra264_adx_controls),
};

static const struct of_device_id tegra210_adx_of_match[] = {
        { .compatible = "nvidia,tegra210-adx", .data = &soc_data_tegra210 },
        { .compatible = "nvidia,tegra264-adx", .data = &soc_data_tegra264 },
        {},
};
MODULE_DEVICE_TABLE(of, tegra210_adx_of_match);

static int tegra210_adx_platform_probe(struct platform_device *pdev)
{
        struct device *dev = &pdev->dev;
        struct tegra210_adx *adx;
        const struct tegra210_adx_soc_data *soc_data;
        void __iomem *regs;
        int err, i;

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

        soc_data = of_device_get_match_data(dev);
        adx->soc_data = soc_data;

        dev_set_drvdata(dev, adx);

        regs = devm_platform_ioremap_resource(pdev, 0);
        if (IS_ERR(regs))
                return PTR_ERR(regs);

        adx->regmap = devm_regmap_init_mmio(dev, regs,
                                            soc_data->regmap_conf);
        if (IS_ERR(adx->regmap))
                return dev_err_probe(dev, PTR_ERR(adx->regmap),
                                     "regmap init failed\n");

        regcache_cache_only(adx->regmap, true);

        adx->map = devm_kcalloc(dev,
                                soc_data->ram_depth * TEGRA_ADX_SLOTS_PER_WORD,
                                sizeof(*adx->map), GFP_KERNEL);
        if (!adx->map)
                return -ENOMEM;

        adx->byte_mask = devm_kcalloc(dev, soc_data->byte_mask_size,
                                      sizeof(*adx->byte_mask), GFP_KERNEL);
        if (!adx->byte_mask)
                return -ENOMEM;

        /* Initialise all byte map slots as disabled (value 256). */
        for (i = 0; i < soc_data->ram_depth * TEGRA_ADX_SLOTS_PER_WORD; i++)
                adx->map[i] = 256;

        tegra210_adx_dais[TEGRA_ADX_IN_DAI_ID].playback.channels_max =
                        adx->soc_data->max_ch;

        err = devm_snd_soc_register_component(dev, &tegra210_adx_cmpnt,
                                              tegra210_adx_dais,
                                              ARRAY_SIZE(tegra210_adx_dais));
        if (err)
                return dev_err_probe(dev, err,
                                     "can't register ADX component\n");

        pm_runtime_enable(dev);

        return 0;
}

static void tegra210_adx_platform_remove(struct platform_device *pdev)
{
        pm_runtime_disable(&pdev->dev);
}

static const struct dev_pm_ops tegra210_adx_pm_ops = {
        RUNTIME_PM_OPS(tegra210_adx_runtime_suspend,
                       tegra210_adx_runtime_resume, NULL)
        SYSTEM_SLEEP_PM_OPS(pm_runtime_force_suspend, pm_runtime_force_resume)
};

static struct platform_driver tegra210_adx_driver = {
        .driver = {
                .name = "tegra210-adx",
                .of_match_table = tegra210_adx_of_match,
                .pm = pm_ptr(&tegra210_adx_pm_ops),
        },
        .probe = tegra210_adx_platform_probe,
        .remove = tegra210_adx_platform_remove,
};
module_platform_driver(tegra210_adx_driver);

MODULE_AUTHOR("Arun Shamanna Lakshmi <aruns@nvidia.com>");
MODULE_DESCRIPTION("Tegra210 ADX ASoC driver");
MODULE_LICENSE("GPL v2");