// SPDX-License-Identifier: GPL-2.0-only
//
// Copyright(c) 2020 Intel Corporation
#include <linux/module.h>
#include <linux/string.h>
#include <sound/pcm.h>
#include <sound/pcm_params.h>
#include <sound/soc.h>
#include <sound/soc-acpi.h>
#include <sound/soc-dai.h>
#include <sound/soc-dapm.h>
#include <sound/sof.h>
#include <uapi/sound/asound.h>
#include "../common/soc-intel-quirks.h"
#include "sof_maxim_common.h"

/*
 * Common structures and functions
 */
static const struct snd_kcontrol_new maxim_2spk_kcontrols[] = {
        SOC_DAPM_PIN_SWITCH("Left Spk"),
        SOC_DAPM_PIN_SWITCH("Right Spk"),

};

static const struct snd_soc_dapm_widget maxim_2spk_widgets[] = {
        SND_SOC_DAPM_SPK("Left Spk", NULL),
        SND_SOC_DAPM_SPK("Right Spk", NULL),
};

/* helper function to get the number of specific codec */
static unsigned int get_num_codecs(const char *hid)
{
        struct acpi_device *adev;
        unsigned int dev_num = 0;

        for_each_acpi_dev_match(adev, hid, NULL, -1)
                dev_num++;

        return dev_num;
}

/*
 * Maxim MAX98373
 */
#define MAX_98373_PIN_NAME 16

static const struct snd_soc_dapm_route max_98373_dapm_routes[] = {
        /* speaker */
        { "Left Spk", NULL, "Left BE_OUT" },
        { "Right Spk", NULL, "Right BE_OUT" },
};

static struct snd_soc_codec_conf max_98373_codec_conf[] = {
        {
                .dlc = COMP_CODEC_CONF(MAX_98373_DEV0_NAME),
                .name_prefix = "Right",
        },
        {
                .dlc = COMP_CODEC_CONF(MAX_98373_DEV1_NAME),
                .name_prefix = "Left",
        },
};

static struct snd_soc_dai_link_component max_98373_components[] = {
        {  /* For Right */
                .name = MAX_98373_DEV0_NAME,
                .dai_name = MAX_98373_CODEC_DAI,
        },
        {  /* For Left */
                .name = MAX_98373_DEV1_NAME,
                .dai_name = MAX_98373_CODEC_DAI,
        },
};

/*
 * According to the definition of 'DAI Sel Mux' mixer in max98373.c, rx mask
 * should choose two channels from TDM slots, the LSB of rx mask is left channel
 * and the other one is right channel.
 */
static const struct {
        unsigned int rx;
} max_98373_tdm_mask[] = {
        {.rx = 0x3},
        {.rx = 0x3},
};

/*
 * The tx mask indicates which channel(s) contains output IV-sense data and
 * others should set to Hi-Z. Here we get the channel number from codec's ACPI
 * device property "maxim,vmon-slot-no" and "maxim,imon-slot-no" to generate the
 * mask. Refer to the max98373_slot_config() function in max98373.c codec driver.
 */
static unsigned int max_98373_get_tx_mask(struct device *dev)
{
        int vmon_slot;
        int imon_slot;

        if (device_property_read_u32(dev, "maxim,vmon-slot-no", &vmon_slot))
                vmon_slot = 0;

        if (device_property_read_u32(dev, "maxim,imon-slot-no", &imon_slot))
                imon_slot = 1;

        dev_dbg(dev, "vmon_slot %d imon_slot %d\n", vmon_slot, imon_slot);

        return (0x1 << vmon_slot) | (0x1 << imon_slot);
}

static int max_98373_hw_params(struct snd_pcm_substream *substream,
                               struct snd_pcm_hw_params *params)
{
        struct snd_soc_pcm_runtime *rtd = snd_soc_substream_to_rtd(substream);
        struct snd_soc_dai_link *dai_link = rtd->dai_link;
        struct snd_soc_dai *codec_dai;
        int i;
        int tdm_slots;
        unsigned int tx_mask;
        unsigned int tx_mask_used = 0x0;
        int ret = 0;

        for_each_rtd_codec_dais(rtd, i, codec_dai) {
                if (i >= ARRAY_SIZE(max_98373_tdm_mask)) {
                        dev_err(codec_dai->dev, "only 2 amps are supported\n");
                        return -EINVAL;
                }

                switch (dai_link->dai_fmt & SND_SOC_DAIFMT_FORMAT_MASK) {
                case SND_SOC_DAIFMT_DSP_A:
                case SND_SOC_DAIFMT_DSP_B:
                        /* get the tplg configured tdm slot number */
                        tdm_slots = sof_dai_get_tdm_slots(rtd);
                        if (tdm_slots <= 0) {
                                dev_err(rtd->dev, "invalid tdm slots %d\n",
                                        tdm_slots);
                                return -EINVAL;
                        }

                        /* get the tx mask from ACPI device properties */
                        tx_mask = max_98373_get_tx_mask(codec_dai->dev);
                        if (!tx_mask)
                                return -EINVAL;

                        if (tx_mask & tx_mask_used) {
                                dev_err(codec_dai->dev, "invalid tx mask 0x%x, used 0x%x\n",
                                        tx_mask, tx_mask_used);
                                return -EINVAL;
                        }

                        tx_mask_used |= tx_mask;

                        /*
                         * check if tdm slot number is too small for channel
                         * allocation
                         */
                        if (fls(tx_mask) > tdm_slots) {
                                dev_err(codec_dai->dev, "slot mismatch, tx %d slots %d\n",
                                        fls(tx_mask), tdm_slots);
                                return -EINVAL;
                        }

                        if (fls(max_98373_tdm_mask[i].rx) > tdm_slots) {
                                dev_err(codec_dai->dev, "slot mismatch, rx %d slots %d\n",
                                        fls(max_98373_tdm_mask[i].rx), tdm_slots);
                                return -EINVAL;
                        }

                        dev_dbg(codec_dai->dev, "set tdm slot: tx 0x%x rx 0x%x slots %d width %d\n",
                                tx_mask, max_98373_tdm_mask[i].rx,
                                tdm_slots, params_width(params));

                        ret = snd_soc_dai_set_tdm_slot(codec_dai, tx_mask,
                                                       max_98373_tdm_mask[i].rx,
                                                       tdm_slots,
                                                       params_width(params));
                        if (ret < 0) {
                                dev_err(codec_dai->dev, "fail to set tdm slot, ret %d\n",
                                        ret);
                                return ret;
                        }
                        break;
                default:
                        dev_dbg(codec_dai->dev, "codec is in I2S mode\n");
                        break;
                }
        }
        return 0;
}

static int max_98373_trigger(struct snd_pcm_substream *substream, int cmd)
{
        struct snd_soc_pcm_runtime *rtd = snd_soc_substream_to_rtd(substream);
        struct snd_soc_dai *codec_dai;
        struct snd_soc_dai *cpu_dai;
        int j;
        int ret = 0;

        /* set spk pin by playback only */
        if (substream->stream == SNDRV_PCM_STREAM_CAPTURE)
                return 0;

        cpu_dai = snd_soc_rtd_to_cpu(rtd, 0);
        for_each_rtd_codec_dais(rtd, j, codec_dai) {
                struct snd_soc_dapm_context *dapm = snd_soc_component_to_dapm(cpu_dai->component);
                char pin_name[MAX_98373_PIN_NAME];

                snprintf(pin_name, ARRAY_SIZE(pin_name), "%s Spk",
                         codec_dai->component->name_prefix);

                switch (cmd) {
                case SNDRV_PCM_TRIGGER_START:
                case SNDRV_PCM_TRIGGER_RESUME:
                case SNDRV_PCM_TRIGGER_PAUSE_RELEASE:
                        ret = snd_soc_dapm_enable_pin(dapm, pin_name);
                        if (!ret)
                                snd_soc_dapm_sync(dapm);
                        break;
                case SNDRV_PCM_TRIGGER_STOP:
                case SNDRV_PCM_TRIGGER_SUSPEND:
                case SNDRV_PCM_TRIGGER_PAUSE_PUSH:
                        ret = snd_soc_dapm_disable_pin(dapm, pin_name);
                        if (!ret)
                                snd_soc_dapm_sync(dapm);
                        break;
                default:
                        break;
                }
        }

        return ret;
}

static const struct snd_soc_ops max_98373_ops = {
        .hw_params = max_98373_hw_params,
        .trigger = max_98373_trigger,
};

static int max_98373_spk_codec_init(struct snd_soc_pcm_runtime *rtd)
{
        struct snd_soc_card *card = rtd->card;
        struct snd_soc_dapm_context *dapm = snd_soc_card_to_dapm(card);
        unsigned int num_codecs = get_num_codecs(MAX_98373_ACPI_HID);
        int ret;

        switch (num_codecs) {
        case 2:
                ret = snd_soc_dapm_new_controls(dapm, maxim_2spk_widgets,
                                                ARRAY_SIZE(maxim_2spk_widgets));
                if (ret) {
                        dev_err(rtd->dev, "fail to add max98373 widgets, ret %d\n",
                                ret);
                        return ret;
                }

                ret = snd_soc_add_card_controls(card, maxim_2spk_kcontrols,
                                                ARRAY_SIZE(maxim_2spk_kcontrols));
                if (ret) {
                        dev_err(rtd->dev, "fail to add max98373 kcontrols, ret %d\n",
                                ret);
                        return ret;
                }

                ret = snd_soc_dapm_add_routes(dapm, max_98373_dapm_routes,
                                              ARRAY_SIZE(max_98373_dapm_routes));
                if (ret) {
                        dev_err(rtd->dev, "fail to add max98373 routes, ret %d\n",
                                ret);
                        return ret;
                }
                break;
        default:
                dev_err(rtd->dev, "max98373: invalid num_codecs %d\n", num_codecs);
                return -EINVAL;
        }

        return ret;
}

void max_98373_dai_link(struct device *dev, struct snd_soc_dai_link *link)
{
        link->codecs = max_98373_components;
        link->num_codecs = ARRAY_SIZE(max_98373_components);
        link->init = max_98373_spk_codec_init;
        link->ops = &max_98373_ops;
}
EXPORT_SYMBOL_NS(max_98373_dai_link, "SND_SOC_INTEL_SOF_MAXIM_COMMON");

void max_98373_set_codec_conf(struct snd_soc_card *card)
{
        card->codec_conf = max_98373_codec_conf;
        card->num_configs = ARRAY_SIZE(max_98373_codec_conf);
}
EXPORT_SYMBOL_NS(max_98373_set_codec_conf, "SND_SOC_INTEL_SOF_MAXIM_COMMON");

/*
 * Maxim MAX98390
 */
static const struct snd_soc_dapm_route max_98390_dapm_routes[] = {
        /* speaker */
        { "Left Spk", NULL, "Left BE_OUT" },
        { "Right Spk", NULL, "Right BE_OUT" },
};

static const struct snd_kcontrol_new max_98390_tt_kcontrols[] = {
        SOC_DAPM_PIN_SWITCH("TL Spk"),
        SOC_DAPM_PIN_SWITCH("TR Spk"),
};

static const struct snd_soc_dapm_widget max_98390_tt_dapm_widgets[] = {
        SND_SOC_DAPM_SPK("TL Spk", NULL),
        SND_SOC_DAPM_SPK("TR Spk", NULL),
};

static const struct snd_soc_dapm_route max_98390_tt_dapm_routes[] = {
        /* Tweeter speaker */
        { "TL Spk", NULL, "Tweeter Left BE_OUT" },
        { "TR Spk", NULL, "Tweeter Right BE_OUT" },
};

static struct snd_soc_codec_conf max_98390_cml_codec_conf[] = {
        {
                .dlc = COMP_CODEC_CONF(MAX_98390_DEV0_NAME),
                .name_prefix = "Left",
        },
        {
                .dlc = COMP_CODEC_CONF(MAX_98390_DEV1_NAME),
                .name_prefix = "Right",
        },
};

static struct snd_soc_codec_conf max_98390_codec_conf[] = {
        {
                .dlc = COMP_CODEC_CONF(MAX_98390_DEV0_NAME),
                .name_prefix = "Right",
        },
        {
                .dlc = COMP_CODEC_CONF(MAX_98390_DEV1_NAME),
                .name_prefix = "Left",
        },
        {
                .dlc = COMP_CODEC_CONF(MAX_98390_DEV2_NAME),
                .name_prefix = "Tweeter Right",
        },
        {
                .dlc = COMP_CODEC_CONF(MAX_98390_DEV3_NAME),
                .name_prefix = "Tweeter Left",
        },
};

static struct snd_soc_dai_link_component max_98390_components[] = {
        {
                .name = MAX_98390_DEV0_NAME,
                .dai_name = MAX_98390_CODEC_DAI,
        },
        {
                .name = MAX_98390_DEV1_NAME,
                .dai_name = MAX_98390_CODEC_DAI,
        },
        {
                .name = MAX_98390_DEV2_NAME,
                .dai_name = MAX_98390_CODEC_DAI,
        },
        {
                .name = MAX_98390_DEV3_NAME,
                .dai_name = MAX_98390_CODEC_DAI,
        },
};

static const struct {
        unsigned int tx;
        unsigned int rx;
} max_98390_tdm_mask[] = {
        {.tx = 0x01, .rx = 0x3},
        {.tx = 0x02, .rx = 0x3},
        {.tx = 0x04, .rx = 0x3},
        {.tx = 0x08, .rx = 0x3},
};

static int max_98390_hw_params(struct snd_pcm_substream *substream,
                               struct snd_pcm_hw_params *params)
{
        struct snd_soc_pcm_runtime *rtd = snd_soc_substream_to_rtd(substream);
        struct snd_soc_dai_link *dai_link = rtd->dai_link;
        struct snd_soc_dai *codec_dai;
        int i, ret;

        for_each_rtd_codec_dais(rtd, i, codec_dai) {
                if (i >= ARRAY_SIZE(max_98390_tdm_mask)) {
                        dev_err(codec_dai->dev, "invalid codec index %d\n", i);
                        return -ENODEV;
                }

                switch (dai_link->dai_fmt & SND_SOC_DAIFMT_FORMAT_MASK) {
                case SND_SOC_DAIFMT_DSP_A:
                case SND_SOC_DAIFMT_DSP_B:
                        /* 4-slot TDM */
                        ret = snd_soc_dai_set_tdm_slot(codec_dai,
                                                       max_98390_tdm_mask[i].tx,
                                                       max_98390_tdm_mask[i].rx,
                                                       4,
                                                       params_width(params));
                        if (ret < 0) {
                                dev_err(codec_dai->dev, "fail to set tdm slot, ret %d\n",
                                        ret);
                                return ret;
                        }
                        break;
                default:
                        dev_dbg(codec_dai->dev, "codec is in I2S mode\n");
                        break;
                }
        }
        return 0;
}

static int max_98390_init(struct snd_soc_pcm_runtime *rtd)
{
        struct snd_soc_card *card = rtd->card;
        struct snd_soc_dapm_context *dapm = snd_soc_card_to_dapm(card);
        unsigned int num_codecs = get_num_codecs(MAX_98390_ACPI_HID);
        int ret;

        switch (num_codecs) {
        case 4:
                /* add widgets/controls/dapm for tweeter speakers */
                ret = snd_soc_dapm_new_controls(dapm, max_98390_tt_dapm_widgets,
                                                ARRAY_SIZE(max_98390_tt_dapm_widgets));
                if (ret) {
                        dev_err(rtd->dev, "unable to add tweeter dapm widgets, ret %d\n",
                                ret);
                        /* Don't need to add routes if widget addition failed */
                        return ret;
                }

                ret = snd_soc_add_card_controls(card, max_98390_tt_kcontrols,
                                                ARRAY_SIZE(max_98390_tt_kcontrols));
                if (ret) {
                        dev_err(rtd->dev, "unable to add tweeter controls, ret %d\n",
                                ret);
                        return ret;
                }

                ret = snd_soc_dapm_add_routes(dapm, max_98390_tt_dapm_routes,
                                              ARRAY_SIZE(max_98390_tt_dapm_routes));
                if (ret) {
                        dev_err(rtd->dev, "unable to add tweeter dapm routes, ret %d\n",
                                ret);
                        return ret;
                }

                fallthrough;
        case 2:
                /* add regular speakers dapm route */
                ret = snd_soc_dapm_new_controls(dapm, maxim_2spk_widgets,
                                                ARRAY_SIZE(maxim_2spk_widgets));
                if (ret) {
                        dev_err(rtd->dev, "fail to add max98390 woofer widgets, ret %d\n",
                                ret);
                        return ret;
                }

                ret = snd_soc_add_card_controls(card, maxim_2spk_kcontrols,
                                                ARRAY_SIZE(maxim_2spk_kcontrols));
                if (ret) {
                        dev_err(rtd->dev, "fail to add max98390 woofer kcontrols, ret %d\n",
                                ret);
                        return ret;
                }

                ret = snd_soc_dapm_add_routes(dapm, max_98390_dapm_routes,
                                              ARRAY_SIZE(max_98390_dapm_routes));
                if (ret) {
                        dev_err(rtd->dev, "unable to add dapm routes, ret %d\n",
                                ret);
                        return ret;
                }
                break;
        default:
                dev_err(rtd->dev, "invalid codec number %d\n", num_codecs);
                return -EINVAL;
        }

        return ret;
}

static const struct snd_soc_ops max_98390_ops = {
        .hw_params = max_98390_hw_params,
};

void max_98390_dai_link(struct device *dev, struct snd_soc_dai_link *link)
{
        unsigned int num_codecs = get_num_codecs(MAX_98390_ACPI_HID);

        link->codecs = max_98390_components;

        switch (num_codecs) {
        case 2:
        case 4:
                link->num_codecs = num_codecs;
                break;
        default:
                dev_err(dev, "invalid codec number %d for %s\n", num_codecs,
                        MAX_98390_ACPI_HID);
                break;
        }

        link->init = max_98390_init;
        link->ops = &max_98390_ops;
}
EXPORT_SYMBOL_NS(max_98390_dai_link, "SND_SOC_INTEL_SOF_MAXIM_COMMON");

void max_98390_set_codec_conf(struct device *dev, struct snd_soc_card *card)
{
        unsigned int num_codecs = get_num_codecs(MAX_98390_ACPI_HID);

        card->codec_conf = max_98390_codec_conf;

        switch (num_codecs) {
        case 2:
                if (soc_intel_is_cml())
                        card->codec_conf = max_98390_cml_codec_conf;

                fallthrough;
        case 4:
                card->num_configs = num_codecs;
                break;
        default:
                dev_err(dev, "invalid codec number %d for %s\n", num_codecs,
                        MAX_98390_ACPI_HID);
                break;
        }
}
EXPORT_SYMBOL_NS(max_98390_set_codec_conf, "SND_SOC_INTEL_SOF_MAXIM_COMMON");

/*
 * Maxim MAX98357A/MAX98360A
 */
static const struct snd_kcontrol_new max_98357a_kcontrols[] = {
        SOC_DAPM_PIN_SWITCH("Spk"),
};

static const struct snd_soc_dapm_widget max_98357a_dapm_widgets[] = {
        SND_SOC_DAPM_SPK("Spk", NULL),
};

static const struct snd_soc_dapm_route max_98357a_dapm_routes[] = {
        /* speaker */
        {"Spk", NULL, "Speaker"},
};

static struct snd_soc_dai_link_component max_98357a_components[] = {
        {
                .name = MAX_98357A_DEV0_NAME,
                .dai_name = MAX_98357A_CODEC_DAI,
        }
};

static struct snd_soc_dai_link_component max_98360a_components[] = {
        {
                .name = MAX_98360A_DEV0_NAME,
                .dai_name = MAX_98357A_CODEC_DAI,
        }
};

static int max_98357a_init(struct snd_soc_pcm_runtime *rtd)
{
        struct snd_soc_card *card = rtd->card;
        struct snd_soc_dapm_context *dapm = snd_soc_card_to_dapm(card);
        int ret;

        ret = snd_soc_dapm_new_controls(dapm, max_98357a_dapm_widgets,
                                        ARRAY_SIZE(max_98357a_dapm_widgets));
        if (ret) {
                dev_err(rtd->dev, "unable to add dapm controls, ret %d\n", ret);
                /* Don't need to add routes if widget addition failed */
                return ret;
        }

        ret = snd_soc_add_card_controls(card, max_98357a_kcontrols,
                                        ARRAY_SIZE(max_98357a_kcontrols));
        if (ret) {
                dev_err(rtd->dev, "unable to add card controls, ret %d\n", ret);
                return ret;
        }

        ret = snd_soc_dapm_add_routes(dapm, max_98357a_dapm_routes,
                                      ARRAY_SIZE(max_98357a_dapm_routes));

        if (ret)
                dev_err(rtd->dev, "unable to add dapm routes, ret %d\n", ret);

        return ret;
}

void max_98357a_dai_link(struct snd_soc_dai_link *link)
{
        link->codecs = max_98357a_components;
        link->num_codecs = ARRAY_SIZE(max_98357a_components);
        link->init = max_98357a_init;
}
EXPORT_SYMBOL_NS(max_98357a_dai_link, "SND_SOC_INTEL_SOF_MAXIM_COMMON");

void max_98360a_dai_link(struct snd_soc_dai_link *link)
{
        link->codecs = max_98360a_components;
        link->num_codecs = ARRAY_SIZE(max_98360a_components);
        link->init = max_98357a_init;
}
EXPORT_SYMBOL_NS(max_98360a_dai_link, "SND_SOC_INTEL_SOF_MAXIM_COMMON");

MODULE_DESCRIPTION("ASoC Intel SOF Maxim helpers");
MODULE_LICENSE("GPL");