// SPDX-License-Identifier: GPL-2.0-only
/*
 * HDMI Channel map support helpers
 */

#include <linux/module.h>
#include <sound/control.h>
#include <sound/tlv.h>
#include <sound/hda_chmap.h>

/*
 * CEA speaker placement:
 *
 *        FLH       FCH        FRH
 *  FLW    FL  FLC   FC   FRC   FR   FRW
 *
 *                                  LFE
 *                     TC
 *
 *          RL  RLC   RC   RRC   RR
 *
 * The Left/Right Surround channel _notions_ LS/RS in SMPTE 320M corresponds to
 * CEA RL/RR; The SMPTE channel _assignment_ C/LFE is swapped to CEA LFE/FC.
 */
enum cea_speaker_placement {
        FL  = (1 <<  0),        /* Front Left           */
        FC  = (1 <<  1),        /* Front Center         */
        FR  = (1 <<  2),        /* Front Right          */
        FLC = (1 <<  3),        /* Front Left Center    */
        FRC = (1 <<  4),        /* Front Right Center   */
        RL  = (1 <<  5),        /* Rear Left            */
        RC  = (1 <<  6),        /* Rear Center          */
        RR  = (1 <<  7),        /* Rear Right           */
        RLC = (1 <<  8),        /* Rear Left Center     */
        RRC = (1 <<  9),        /* Rear Right Center    */
        LFE = (1 << 10),        /* Low Frequency Effect */
        FLW = (1 << 11),        /* Front Left Wide      */
        FRW = (1 << 12),        /* Front Right Wide     */
        FLH = (1 << 13),        /* Front Left High      */
        FCH = (1 << 14),        /* Front Center High    */
        FRH = (1 << 15),        /* Front Right High     */
        TC  = (1 << 16),        /* Top Center           */
};

static const char * const cea_speaker_allocation_names[] = {
        /*  0 */ "FL/FR",
        /*  1 */ "LFE",
        /*  2 */ "FC",
        /*  3 */ "RL/RR",
        /*  4 */ "RC",
        /*  5 */ "FLC/FRC",
        /*  6 */ "RLC/RRC",
        /*  7 */ "FLW/FRW",
        /*  8 */ "FLH/FRH",
        /*  9 */ "TC",
        /* 10 */ "FCH",
};

/*
 * ELD SA bits in the CEA Speaker Allocation data block
 */
static const int eld_speaker_allocation_bits[] = {
        [0] = FL | FR,
        [1] = LFE,
        [2] = FC,
        [3] = RL | RR,
        [4] = RC,
        [5] = FLC | FRC,
        [6] = RLC | RRC,
        /* the following are not defined in ELD yet */
        [7] = FLW | FRW,
        [8] = FLH | FRH,
        [9] = TC,
        [10] = FCH,
};

/*
 * ALSA sequence is:
 *
 *       surround40   surround41   surround50   surround51   surround71
 * ch0   front left   =            =            =            =
 * ch1   front right  =            =            =            =
 * ch2   rear left    =            =            =            =
 * ch3   rear right   =            =            =            =
 * ch4                LFE          center       center       center
 * ch5                                          LFE          LFE
 * ch6                                                       side left
 * ch7                                                       side right
 *
 * surround71 = {FL, FR, RLC, RRC, FC, LFE, RL, RR}
 */
static int hdmi_channel_mapping[0x32][8] = {
        /* stereo */
        [0x00] = { 0x00, 0x11, 0xf2, 0xf3, 0xf4, 0xf5, 0xf6, 0xf7 },
        /* 2.1 */
        [0x01] = { 0x00, 0x11, 0x22, 0xf3, 0xf4, 0xf5, 0xf6, 0xf7 },
        /* Dolby Surround */
        [0x02] = { 0x00, 0x11, 0x23, 0xf2, 0xf4, 0xf5, 0xf6, 0xf7 },
        /* surround40 */
        [0x08] = { 0x00, 0x11, 0x24, 0x35, 0xf3, 0xf2, 0xf6, 0xf7 },
        /* 4ch */
        [0x03] = { 0x00, 0x11, 0x23, 0x32, 0x44, 0xf5, 0xf6, 0xf7 },
        /* surround41 */
        [0x09] = { 0x00, 0x11, 0x24, 0x35, 0x42, 0xf3, 0xf6, 0xf7 },
        /* surround50 */
        [0x0a] = { 0x00, 0x11, 0x24, 0x35, 0x43, 0xf2, 0xf6, 0xf7 },
        /* surround51 */
        [0x0b] = { 0x00, 0x11, 0x24, 0x35, 0x43, 0x52, 0xf6, 0xf7 },
        /* 7.1 */
        [0x13] = { 0x00, 0x11, 0x26, 0x37, 0x43, 0x52, 0x64, 0x75 },
};

/*
 * This is an ordered list!
 *
 * The preceding ones have better chances to be selected by
 * hdmi_channel_allocation().
 */
static struct hdac_cea_channel_speaker_allocation channel_allocations[] = {
/*                        channel:   7     6    5    4    3     2    1    0  */
{ .ca_index = 0x00,  .speakers = {   0,    0,   0,   0,   0,    0,  FR,  FL } },
                                 /* 2.1 */
{ .ca_index = 0x01,  .speakers = {   0,    0,   0,   0,   0,  LFE,  FR,  FL } },
                                 /* Dolby Surround */
{ .ca_index = 0x02,  .speakers = {   0,    0,   0,   0,  FC,    0,  FR,  FL } },
                                 /* surround40 */
{ .ca_index = 0x08,  .speakers = {   0,    0,  RR,  RL,   0,    0,  FR,  FL } },
                                 /* surround41 */
{ .ca_index = 0x09,  .speakers = {   0,    0,  RR,  RL,   0,  LFE,  FR,  FL } },
                                 /* surround50 */
{ .ca_index = 0x0a,  .speakers = {   0,    0,  RR,  RL,  FC,    0,  FR,  FL } },
                                 /* surround51 */
{ .ca_index = 0x0b,  .speakers = {   0,    0,  RR,  RL,  FC,  LFE,  FR,  FL } },
                                 /* 6.1 */
{ .ca_index = 0x0f,  .speakers = {   0,   RC,  RR,  RL,  FC,  LFE,  FR,  FL } },
                                 /* surround71 */
{ .ca_index = 0x13,  .speakers = { RRC,  RLC,  RR,  RL,  FC,  LFE,  FR,  FL } },

{ .ca_index = 0x03,  .speakers = {   0,    0,   0,   0,  FC,  LFE,  FR,  FL } },
{ .ca_index = 0x04,  .speakers = {   0,    0,   0,  RC,   0,    0,  FR,  FL } },
{ .ca_index = 0x05,  .speakers = {   0,    0,   0,  RC,   0,  LFE,  FR,  FL } },
{ .ca_index = 0x06,  .speakers = {   0,    0,   0,  RC,  FC,    0,  FR,  FL } },
{ .ca_index = 0x07,  .speakers = {   0,    0,   0,  RC,  FC,  LFE,  FR,  FL } },
{ .ca_index = 0x0c,  .speakers = {   0,   RC,  RR,  RL,   0,    0,  FR,  FL } },
{ .ca_index = 0x0d,  .speakers = {   0,   RC,  RR,  RL,   0,  LFE,  FR,  FL } },
{ .ca_index = 0x0e,  .speakers = {   0,   RC,  RR,  RL,  FC,    0,  FR,  FL } },
{ .ca_index = 0x10,  .speakers = { RRC,  RLC,  RR,  RL,   0,    0,  FR,  FL } },
{ .ca_index = 0x11,  .speakers = { RRC,  RLC,  RR,  RL,   0,  LFE,  FR,  FL } },
{ .ca_index = 0x12,  .speakers = { RRC,  RLC,  RR,  RL,  FC,    0,  FR,  FL } },
{ .ca_index = 0x14,  .speakers = { FRC,  FLC,   0,   0,   0,    0,  FR,  FL } },
{ .ca_index = 0x15,  .speakers = { FRC,  FLC,   0,   0,   0,  LFE,  FR,  FL } },
{ .ca_index = 0x16,  .speakers = { FRC,  FLC,   0,   0,  FC,    0,  FR,  FL } },
{ .ca_index = 0x17,  .speakers = { FRC,  FLC,   0,   0,  FC,  LFE,  FR,  FL } },
{ .ca_index = 0x18,  .speakers = { FRC,  FLC,   0,  RC,   0,    0,  FR,  FL } },
{ .ca_index = 0x19,  .speakers = { FRC,  FLC,   0,  RC,   0,  LFE,  FR,  FL } },
{ .ca_index = 0x1a,  .speakers = { FRC,  FLC,   0,  RC,  FC,    0,  FR,  FL } },
{ .ca_index = 0x1b,  .speakers = { FRC,  FLC,   0,  RC,  FC,  LFE,  FR,  FL } },
{ .ca_index = 0x1c,  .speakers = { FRC,  FLC,  RR,  RL,   0,    0,  FR,  FL } },
{ .ca_index = 0x1d,  .speakers = { FRC,  FLC,  RR,  RL,   0,  LFE,  FR,  FL } },
{ .ca_index = 0x1e,  .speakers = { FRC,  FLC,  RR,  RL,  FC,    0,  FR,  FL } },
{ .ca_index = 0x1f,  .speakers = { FRC,  FLC,  RR,  RL,  FC,  LFE,  FR,  FL } },
{ .ca_index = 0x20,  .speakers = {   0,  FCH,  RR,  RL,  FC,    0,  FR,  FL } },
{ .ca_index = 0x21,  .speakers = {   0,  FCH,  RR,  RL,  FC,  LFE,  FR,  FL } },
{ .ca_index = 0x22,  .speakers = {  TC,    0,  RR,  RL,  FC,    0,  FR,  FL } },
{ .ca_index = 0x23,  .speakers = {  TC,    0,  RR,  RL,  FC,  LFE,  FR,  FL } },
{ .ca_index = 0x24,  .speakers = { FRH,  FLH,  RR,  RL,   0,    0,  FR,  FL } },
{ .ca_index = 0x25,  .speakers = { FRH,  FLH,  RR,  RL,   0,  LFE,  FR,  FL } },
{ .ca_index = 0x26,  .speakers = { FRW,  FLW,  RR,  RL,   0,    0,  FR,  FL } },
{ .ca_index = 0x27,  .speakers = { FRW,  FLW,  RR,  RL,   0,  LFE,  FR,  FL } },
{ .ca_index = 0x28,  .speakers = {  TC,   RC,  RR,  RL,  FC,    0,  FR,  FL } },
{ .ca_index = 0x29,  .speakers = {  TC,   RC,  RR,  RL,  FC,  LFE,  FR,  FL } },
{ .ca_index = 0x2a,  .speakers = { FCH,   RC,  RR,  RL,  FC,    0,  FR,  FL } },
{ .ca_index = 0x2b,  .speakers = { FCH,   RC,  RR,  RL,  FC,  LFE,  FR,  FL } },
{ .ca_index = 0x2c,  .speakers = {  TC,  FCH,  RR,  RL,  FC,    0,  FR,  FL } },
{ .ca_index = 0x2d,  .speakers = {  TC,  FCH,  RR,  RL,  FC,  LFE,  FR,  FL } },
{ .ca_index = 0x2e,  .speakers = { FRH,  FLH,  RR,  RL,  FC,    0,  FR,  FL } },
{ .ca_index = 0x2f,  .speakers = { FRH,  FLH,  RR,  RL,  FC,  LFE,  FR,  FL } },
{ .ca_index = 0x30,  .speakers = { FRW,  FLW,  RR,  RL,  FC,    0,  FR,  FL } },
{ .ca_index = 0x31,  .speakers = { FRW,  FLW,  RR,  RL,  FC,  LFE,  FR,  FL } },
};

static int hdmi_pin_set_slot_channel(struct hdac_device *codec,
                hda_nid_t pin_nid, int asp_slot, int channel)
{
        return snd_hdac_codec_write(codec, pin_nid, 0,
                                AC_VERB_SET_HDMI_CHAN_SLOT,
                                (channel << 4) | asp_slot);
}

static int hdmi_pin_get_slot_channel(struct hdac_device *codec,
                        hda_nid_t pin_nid, int asp_slot)
{
        return (snd_hdac_codec_read(codec, pin_nid, 0,
                                   AC_VERB_GET_HDMI_CHAN_SLOT,
                                   asp_slot) & 0xf0) >> 4;
}

static int hdmi_get_channel_count(struct hdac_device *codec, hda_nid_t cvt_nid)
{
        return 1 + snd_hdac_codec_read(codec, cvt_nid, 0,
                                        AC_VERB_GET_CVT_CHAN_COUNT, 0);
}

static void hdmi_set_channel_count(struct hdac_device *codec,
                                   hda_nid_t cvt_nid, int chs)
{
        if (chs != hdmi_get_channel_count(codec, cvt_nid))
                snd_hdac_codec_write(codec, cvt_nid, 0,
                                    AC_VERB_SET_CVT_CHAN_COUNT, chs - 1);
}

/*
 * Channel mapping routines
 */

/*
 * Compute derived values in channel_allocations[].
 */
static void init_channel_allocations(void)
{
        int i, j;
        struct hdac_cea_channel_speaker_allocation *p;

        for (i = 0; i < ARRAY_SIZE(channel_allocations); i++) {
                p = channel_allocations + i;
                p->channels = 0;
                p->spk_mask = 0;
                for (j = 0; j < ARRAY_SIZE(p->speakers); j++)
                        if (p->speakers[j]) {
                                p->channels++;
                                p->spk_mask |= p->speakers[j];
                        }
        }
}

static int get_channel_allocation_order(int ca)
{
        int i;

        for (i = 0; i < ARRAY_SIZE(channel_allocations); i++) {
                if (channel_allocations[i].ca_index == ca)
                        break;
        }
        return i;
}

void snd_hdac_print_channel_allocation(int spk_alloc, char *buf, int buflen)
{
        int i, j;

        for (i = 0, j = 0; i < ARRAY_SIZE(cea_speaker_allocation_names); i++) {
                if (spk_alloc & (1 << i))
                        j += scnprintf(buf + j, buflen - j,  " %s",
                                        cea_speaker_allocation_names[i]);
        }
        buf[j] = '\0';  /* necessary when j == 0 */
}
EXPORT_SYMBOL_GPL(snd_hdac_print_channel_allocation);

/*
 * The transformation takes two steps:
 *
 *      eld->spk_alloc => (eld_speaker_allocation_bits[]) => spk_mask
 *            spk_mask => (channel_allocations[])         => ai->CA
 *
 * TODO: it could select the wrong CA from multiple candidates.
*/
static int hdmi_channel_allocation_spk_alloc_blk(struct hdac_device *codec,
                                   int spk_alloc, int channels)
{
        int i;
        int ca = 0;
        int spk_mask = 0;
        char buf[SND_PRINT_CHANNEL_ALLOCATION_ADVISED_BUFSIZE];

        /*
         * CA defaults to 0 for basic stereo audio
         */
        if (channels <= 2)
                return 0;

        /*
         * expand ELD's speaker allocation mask
         *
         * ELD tells the speaker mask in a compact(paired) form,
         * expand ELD's notions to match the ones used by Audio InfoFrame.
         */
        for (i = 0; i < ARRAY_SIZE(eld_speaker_allocation_bits); i++) {
                if (spk_alloc & (1 << i))
                        spk_mask |= eld_speaker_allocation_bits[i];
        }

        /* search for the first working match in the CA table */
        for (i = 0; i < ARRAY_SIZE(channel_allocations); i++) {
                if (channels == channel_allocations[i].channels &&
                    (spk_mask & channel_allocations[i].spk_mask) ==
                                channel_allocations[i].spk_mask) {
                        ca = channel_allocations[i].ca_index;
                        break;
                }
        }

        if (!ca) {
                /*
                 * if there was no match, select the regular ALSA channel
                 * allocation with the matching number of channels
                 */
                for (i = 0; i < ARRAY_SIZE(channel_allocations); i++) {
                        if (channels == channel_allocations[i].channels) {
                                ca = channel_allocations[i].ca_index;
                                break;
                        }
                }
        }

        snd_hdac_print_channel_allocation(spk_alloc, buf, sizeof(buf));
        dev_dbg(&codec->dev, "HDMI: select CA 0x%x for %d-channel allocation: %s\n",
                    ca, channels, buf);

        return ca;
}

static void hdmi_debug_channel_mapping(struct hdac_chmap *chmap,
                                       hda_nid_t pin_nid)
{
#ifdef CONFIG_SND_DEBUG_VERBOSE
        int i;
        int channel;

        for (i = 0; i < 8; i++) {
                channel = chmap->ops.pin_get_slot_channel(
                                chmap->hdac, pin_nid, i);
                dev_dbg(&chmap->hdac->dev, "HDMI: ASP channel %d => slot %d\n",
                                                channel, i);
        }
#endif
}

static void hdmi_std_setup_channel_mapping(struct hdac_chmap *chmap,
                                       hda_nid_t pin_nid,
                                       bool non_pcm,
                                       int ca)
{
        struct hdac_cea_channel_speaker_allocation *ch_alloc;
        int i;
        int err;
        int order;
        int non_pcm_mapping[8];

        order = get_channel_allocation_order(ca);
        ch_alloc = &channel_allocations[order];

        if (hdmi_channel_mapping[ca][1] == 0) {
                int hdmi_slot = 0;
                /* fill actual channel mappings in ALSA channel (i) order */
                for (i = 0; i < ch_alloc->channels; i++) {
                        while (!WARN_ON(hdmi_slot >= 8) &&
                               !ch_alloc->speakers[7 - hdmi_slot])
                                hdmi_slot++; /* skip zero slots */

                        hdmi_channel_mapping[ca][i] = (i << 4) | hdmi_slot++;
                }
                /* fill the rest of the slots with ALSA channel 0xf */
                for (hdmi_slot = 0; hdmi_slot < 8; hdmi_slot++)
                        if (!ch_alloc->speakers[7 - hdmi_slot])
                                hdmi_channel_mapping[ca][i++] = (0xf << 4) | hdmi_slot;
        }

        if (non_pcm) {
                for (i = 0; i < ch_alloc->channels; i++)
                        non_pcm_mapping[i] = (i << 4) | i;
                for (; i < 8; i++)
                        non_pcm_mapping[i] = (0xf << 4) | i;
        }

        for (i = 0; i < 8; i++) {
                int slotsetup = non_pcm ? non_pcm_mapping[i] : hdmi_channel_mapping[ca][i];
                int hdmi_slot = slotsetup & 0x0f;
                int channel = (slotsetup & 0xf0) >> 4;

                err = chmap->ops.pin_set_slot_channel(chmap->hdac,
                                pin_nid, hdmi_slot, channel);
                if (err) {
                        dev_dbg(&chmap->hdac->dev, "HDMI: channel mapping failed\n");
                        break;
                }
        }
}

struct channel_map_table {
        unsigned char map;              /* ALSA API channel map position */
        int spk_mask;                   /* speaker position bit mask */
};

static struct channel_map_table map_tables[] = {
        { SNDRV_CHMAP_FL,       FL },
        { SNDRV_CHMAP_FR,       FR },
        { SNDRV_CHMAP_RL,       RL },
        { SNDRV_CHMAP_RR,       RR },
        { SNDRV_CHMAP_LFE,      LFE },
        { SNDRV_CHMAP_FC,       FC },
        { SNDRV_CHMAP_RLC,      RLC },
        { SNDRV_CHMAP_RRC,      RRC },
        { SNDRV_CHMAP_RC,       RC },
        { SNDRV_CHMAP_FLC,      FLC },
        { SNDRV_CHMAP_FRC,      FRC },
        { SNDRV_CHMAP_TFL,      FLH },
        { SNDRV_CHMAP_TFR,      FRH },
        { SNDRV_CHMAP_FLW,      FLW },
        { SNDRV_CHMAP_FRW,      FRW },
        { SNDRV_CHMAP_TC,       TC },
        { SNDRV_CHMAP_TFC,      FCH },
        {} /* terminator */
};

/* from ALSA API channel position to speaker bit mask */
int snd_hdac_chmap_to_spk_mask(unsigned char c)
{
        struct channel_map_table *t = map_tables;

        for (; t->map; t++) {
                if (t->map == c)
                        return t->spk_mask;
        }
        return 0;
}
EXPORT_SYMBOL_GPL(snd_hdac_chmap_to_spk_mask);

/* from ALSA API channel position to CEA slot */
static int to_cea_slot(int ordered_ca, unsigned char pos)
{
        int mask = snd_hdac_chmap_to_spk_mask(pos);
        int i;

        /* Add sanity check to pass klockwork check.
         * This should never happen.
         */
        if (ordered_ca >= ARRAY_SIZE(channel_allocations))
                return -1;

        if (mask) {
                for (i = 0; i < 8; i++) {
                        if (channel_allocations[ordered_ca].speakers[7 - i] == mask)
                                return i;
                }
        }

        return -1;
}

/* from speaker bit mask to ALSA API channel position */
int snd_hdac_spk_to_chmap(int spk)
{
        struct channel_map_table *t = map_tables;

        for (; t->map; t++) {
                if (t->spk_mask == spk)
                        return t->map;
        }
        return 0;
}
EXPORT_SYMBOL_GPL(snd_hdac_spk_to_chmap);

/* from CEA slot to ALSA API channel position */
static int from_cea_slot(int ordered_ca, unsigned char slot)
{
        int mask;

        /* Add sanity check to pass klockwork check.
         * This should never happen.
         */
        if (slot >= 8)
                return 0;

        mask = channel_allocations[ordered_ca].speakers[7 - slot];

        return snd_hdac_spk_to_chmap(mask);
}

/* get the CA index corresponding to the given ALSA API channel map */
static int hdmi_manual_channel_allocation(int chs, unsigned char *map)
{
        int i, spks = 0, spk_mask = 0;

        for (i = 0; i < chs; i++) {
                int mask = snd_hdac_chmap_to_spk_mask(map[i]);

                if (mask) {
                        spk_mask |= mask;
                        spks++;
                }
        }

        for (i = 0; i < ARRAY_SIZE(channel_allocations); i++) {
                if ((chs == channel_allocations[i].channels ||
                     spks == channel_allocations[i].channels) &&
                    (spk_mask & channel_allocations[i].spk_mask) ==
                                channel_allocations[i].spk_mask)
                        return channel_allocations[i].ca_index;
        }
        return -1;
}

/* set up the channel slots for the given ALSA API channel map */
static int hdmi_manual_setup_channel_mapping(struct hdac_chmap *chmap,
                                             hda_nid_t pin_nid,
                                             int chs, unsigned char *map,
                                             int ca)
{
        int ordered_ca = get_channel_allocation_order(ca);
        int alsa_pos, hdmi_slot;
        int assignments[8] = {[0 ... 7] = 0xf};

        for (alsa_pos = 0; alsa_pos < chs; alsa_pos++) {

                hdmi_slot = to_cea_slot(ordered_ca, map[alsa_pos]);

                if (hdmi_slot < 0)
                        continue; /* unassigned channel */

                assignments[hdmi_slot] = alsa_pos;
        }

        for (hdmi_slot = 0; hdmi_slot < 8; hdmi_slot++) {
                int err;

                err = chmap->ops.pin_set_slot_channel(chmap->hdac,
                                pin_nid, hdmi_slot, assignments[hdmi_slot]);
                if (err)
                        return -EINVAL;
        }
        return 0;
}

/* store ALSA API channel map from the current default map */
static void hdmi_setup_fake_chmap(unsigned char *map, int ca)
{
        int i;
        int ordered_ca = get_channel_allocation_order(ca);

        for (i = 0; i < 8; i++) {
                if (ordered_ca < ARRAY_SIZE(channel_allocations) &&
                    i < channel_allocations[ordered_ca].channels)
                        map[i] = from_cea_slot(ordered_ca, hdmi_channel_mapping[ca][i] & 0x0f);
                else
                        map[i] = 0;
        }
}

void snd_hdac_setup_channel_mapping(struct hdac_chmap *chmap,
                                       hda_nid_t pin_nid, bool non_pcm, int ca,
                                       int channels, unsigned char *map,
                                       bool chmap_set)
{
        if (!non_pcm && chmap_set) {
                hdmi_manual_setup_channel_mapping(chmap, pin_nid,
                                                  channels, map, ca);
        } else {
                hdmi_std_setup_channel_mapping(chmap, pin_nid, non_pcm, ca);
                hdmi_setup_fake_chmap(map, ca);
        }

        hdmi_debug_channel_mapping(chmap, pin_nid);
}
EXPORT_SYMBOL_GPL(snd_hdac_setup_channel_mapping);

int snd_hdac_get_active_channels(int ca)
{
        int ordered_ca = get_channel_allocation_order(ca);

        /* Add sanity check to pass klockwork check.
         * This should never happen.
         */
        if (ordered_ca >= ARRAY_SIZE(channel_allocations))
                ordered_ca = 0;

        return channel_allocations[ordered_ca].channels;
}
EXPORT_SYMBOL_GPL(snd_hdac_get_active_channels);

struct hdac_cea_channel_speaker_allocation *snd_hdac_get_ch_alloc_from_ca(int ca)
{
        return &channel_allocations[get_channel_allocation_order(ca)];
}
EXPORT_SYMBOL_GPL(snd_hdac_get_ch_alloc_from_ca);

int snd_hdac_channel_allocation(struct hdac_device *hdac, int spk_alloc,
                int channels, bool chmap_set, bool non_pcm, unsigned char *map)
{
        int ca;

        if (!non_pcm && chmap_set)
                ca = hdmi_manual_channel_allocation(channels, map);
        else
                ca = hdmi_channel_allocation_spk_alloc_blk(hdac,
                                        spk_alloc, channels);

        if (ca < 0)
                ca = 0;

        return ca;
}
EXPORT_SYMBOL_GPL(snd_hdac_channel_allocation);

/*
 * ALSA API channel-map control callbacks
 */
static int hdmi_chmap_ctl_info(struct snd_kcontrol *kcontrol,
                               struct snd_ctl_elem_info *uinfo)
{
        struct snd_pcm_chmap *info = snd_kcontrol_chip(kcontrol);
        struct hdac_chmap *chmap = info->private_data;

        uinfo->type = SNDRV_CTL_ELEM_TYPE_INTEGER;
        uinfo->count = chmap->channels_max;
        uinfo->value.integer.min = 0;
        uinfo->value.integer.max = SNDRV_CHMAP_LAST;
        return 0;
}

static int hdmi_chmap_cea_alloc_validate_get_type(struct hdac_chmap *chmap,
                struct hdac_cea_channel_speaker_allocation *cap, int channels)
{
        /* If the speaker allocation matches the channel count, it is OK.*/
        if (cap->channels != channels)
                return -1;

        /* all channels are remappable freely */
        return SNDRV_CTL_TLVT_CHMAP_VAR;
}

static void hdmi_cea_alloc_to_tlv_chmap(struct hdac_chmap *hchmap,
                struct hdac_cea_channel_speaker_allocation *cap,
                unsigned int *chmap, int channels)
{
        int count = 0;
        int c;

        for (c = 7; c >= 0; c--) {
                int spk = cap->speakers[c];

                if (!spk)
                        continue;

                chmap[count++] = snd_hdac_spk_to_chmap(spk);
        }

        WARN_ON(count != channels);
}

static int spk_mask_from_spk_alloc(int spk_alloc)
{
        int i;
        int spk_mask = eld_speaker_allocation_bits[0];

        for (i = 0; i < ARRAY_SIZE(eld_speaker_allocation_bits); i++) {
                if (spk_alloc & (1 << i))
                        spk_mask |= eld_speaker_allocation_bits[i];
        }

        return spk_mask;
}

static int hdmi_chmap_ctl_tlv(struct snd_kcontrol *kcontrol, int op_flag,
                              unsigned int size, unsigned int __user *tlv)
{
        struct snd_pcm_chmap *info = snd_kcontrol_chip(kcontrol);
        struct hdac_chmap *chmap = info->private_data;
        int pcm_idx = kcontrol->private_value;
        unsigned int __user *dst;
        int chs, count = 0;
        unsigned long max_chs;
        int type;
        int spk_alloc, spk_mask;

        if (size < 8)
                return -ENOMEM;
        if (put_user(SNDRV_CTL_TLVT_CONTAINER, tlv))
                return -EFAULT;
        size -= 8;
        dst = tlv + 2;

        spk_alloc = chmap->ops.get_spk_alloc(chmap->hdac, pcm_idx);
        spk_mask = spk_mask_from_spk_alloc(spk_alloc);

        max_chs = hweight_long(spk_mask);

        for (chs = 2; chs <= max_chs; chs++) {
                int i;
                struct hdac_cea_channel_speaker_allocation *cap;

                cap = channel_allocations;
                for (i = 0; i < ARRAY_SIZE(channel_allocations); i++, cap++) {
                        int chs_bytes = chs * 4;
                        unsigned int tlv_chmap[8];

                        if (cap->channels != chs)
                                continue;

                        if (!(cap->spk_mask == (spk_mask & cap->spk_mask)))
                                continue;

                        type = chmap->ops.chmap_cea_alloc_validate_get_type(
                                                        chmap, cap, chs);
                        if (type < 0)
                                return -ENODEV;
                        if (size < 8)
                                return -ENOMEM;

                        if (put_user(type, dst) ||
                            put_user(chs_bytes, dst + 1))
                                return -EFAULT;

                        dst += 2;
                        size -= 8;
                        count += 8;

                        if (size < chs_bytes)
                                return -ENOMEM;

                        size -= chs_bytes;
                        count += chs_bytes;
                        chmap->ops.cea_alloc_to_tlv_chmap(chmap, cap,
                                                tlv_chmap, chs);

                        if (copy_to_user(dst, tlv_chmap, chs_bytes))
                                return -EFAULT;
                        dst += chs;
                }
        }

        if (put_user(count, tlv + 1))
                return -EFAULT;

        return 0;
}

static int hdmi_chmap_ctl_get(struct snd_kcontrol *kcontrol,
                              struct snd_ctl_elem_value *ucontrol)
{
        struct snd_pcm_chmap *info = snd_kcontrol_chip(kcontrol);
        struct hdac_chmap *chmap = info->private_data;
        int pcm_idx = kcontrol->private_value;
        unsigned char pcm_chmap[8];
        int i;

        memset(pcm_chmap, 0, sizeof(pcm_chmap));
        chmap->ops.get_chmap(chmap->hdac, pcm_idx, pcm_chmap);

        for (i = 0; i < ARRAY_SIZE(pcm_chmap); i++)
                ucontrol->value.integer.value[i] = pcm_chmap[i];

        return 0;
}

/* a simple sanity check for input values to chmap kcontrol */
static int chmap_value_check(struct hdac_chmap *hchmap,
                             const struct snd_ctl_elem_value *ucontrol)
{
        int i;

        for (i = 0; i < hchmap->channels_max; i++) {
                if (ucontrol->value.integer.value[i] < 0 ||
                    ucontrol->value.integer.value[i] > SNDRV_CHMAP_LAST)
                        return -EINVAL;
        }
        return 0;
}

static int hdmi_chmap_ctl_put(struct snd_kcontrol *kcontrol,
                              struct snd_ctl_elem_value *ucontrol)
{
        struct snd_pcm_chmap *info = snd_kcontrol_chip(kcontrol);
        struct hdac_chmap *hchmap = info->private_data;
        int pcm_idx = kcontrol->private_value;
        unsigned int ctl_idx;
        struct snd_pcm_substream *substream;
        unsigned char chmap[8], per_pin_chmap[8];
        int i, err, ca, prepared = 0;

        err = chmap_value_check(hchmap, ucontrol);
        if (err < 0)
                return err;

        /* No monitor is connected in dyn_pcm_assign.
         * It's invalid to setup the chmap
         */
        if (!hchmap->ops.is_pcm_attached(hchmap->hdac, pcm_idx))
                return 0;

        ctl_idx = snd_ctl_get_ioffidx(kcontrol, &ucontrol->id);
        substream = snd_pcm_chmap_substream(info, ctl_idx);
        if (!substream || !substream->runtime)
                return 0; /* just for avoiding error from alsactl restore */
        switch (substream->runtime->state) {
        case SNDRV_PCM_STATE_OPEN:
        case SNDRV_PCM_STATE_SETUP:
                break;
        case SNDRV_PCM_STATE_PREPARED:
                prepared = 1;
                break;
        default:
                return -EBUSY;
        }
        memset(chmap, 0, sizeof(chmap));
        for (i = 0; i < ARRAY_SIZE(chmap); i++)
                chmap[i] = ucontrol->value.integer.value[i];

        hchmap->ops.get_chmap(hchmap->hdac, pcm_idx, per_pin_chmap);
        if (!memcmp(chmap, per_pin_chmap, sizeof(chmap)))
                return 0;
        ca = hdmi_manual_channel_allocation(ARRAY_SIZE(chmap), chmap);
        if (ca < 0)
                return -EINVAL;
        if (hchmap->ops.chmap_validate) {
                err = hchmap->ops.chmap_validate(hchmap, ca,
                                ARRAY_SIZE(chmap), chmap);
                if (err)
                        return err;
        }

        hchmap->ops.set_chmap(hchmap->hdac, pcm_idx, chmap, prepared);

        return 0;
}

static const struct hdac_chmap_ops chmap_ops = {
        .chmap_cea_alloc_validate_get_type      = hdmi_chmap_cea_alloc_validate_get_type,
        .cea_alloc_to_tlv_chmap                 = hdmi_cea_alloc_to_tlv_chmap,
        .pin_get_slot_channel                   = hdmi_pin_get_slot_channel,
        .pin_set_slot_channel                   = hdmi_pin_set_slot_channel,
        .set_channel_count                      = hdmi_set_channel_count,
};

void snd_hdac_register_chmap_ops(struct hdac_device *hdac,
                                struct hdac_chmap *chmap)
{
        chmap->ops = chmap_ops;
        chmap->hdac = hdac;
        init_channel_allocations();
}
EXPORT_SYMBOL_GPL(snd_hdac_register_chmap_ops);

int snd_hdac_add_chmap_ctls(struct snd_pcm *pcm, int pcm_idx,
                                struct hdac_chmap *hchmap)
{
        struct snd_pcm_chmap *chmap;
        struct snd_kcontrol *kctl;
        int err, i;

        err = snd_pcm_add_chmap_ctls(pcm,
                                     SNDRV_PCM_STREAM_PLAYBACK,
                                     NULL, 0, pcm_idx, &chmap);
        if (err < 0)
                return err;
        /* override handlers */
        chmap->private_data = hchmap;
        kctl = chmap->kctl;
        for (i = 0; i < kctl->count; i++)
                kctl->vd[i].access |= SNDRV_CTL_ELEM_ACCESS_WRITE;
        kctl->info = hdmi_chmap_ctl_info;
        kctl->get = hdmi_chmap_ctl_get;
        kctl->put = hdmi_chmap_ctl_put;
        kctl->tlv.c = hdmi_chmap_ctl_tlv;

        return 0;
}
EXPORT_SYMBOL_GPL(snd_hdac_add_chmap_ctls);