// SPDX-License-Identifier: GPL-2.0
// Copyright (c) 2011-2017, The Linux Foundation. All rights reserved.
// Copyright (c) 2018, Linaro Limited

#include "q6dsp-common.h"
#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/string.h>
#include <linux/errno.h>

int q6dsp_map_channels(u8 ch_map[PCM_MAX_NUM_CHANNEL], int ch)
{
 memset(ch_map, 0, PCM_MAX_NUM_CHANNEL);

 switch (ch) {
 case 1:
  ch_map[0] = PCM_CHANNEL_FC;
  break;
 case 2:
  ch_map[0] = PCM_CHANNEL_FL;
  ch_map[1] = PCM_CHANNEL_FR;
  break;
 case 3:
  ch_map[0] = PCM_CHANNEL_FL;
  ch_map[1] = PCM_CHANNEL_FR;
  ch_map[2] = PCM_CHANNEL_FC;
  break;
 case 4:
  ch_map[0] = PCM_CHANNEL_FL;
  ch_map[1] = PCM_CHANNEL_FR;
  ch_map[2] = PCM_CHANNEL_LS;
  ch_map[3] = PCM_CHANNEL_RS;
  break;
 case 5:
  ch_map[0] = PCM_CHANNEL_FL;
  ch_map[1] = PCM_CHANNEL_FR;
  ch_map[2] = PCM_CHANNEL_FC;
  ch_map[3] = PCM_CHANNEL_LS;
  ch_map[4] = PCM_CHANNEL_RS;
  break;
 case 6:
  ch_map[0] = PCM_CHANNEL_FL;
  ch_map[1] = PCM_CHANNEL_FR;
  ch_map[2] = PCM_CHANNEL_LFE;
  ch_map[3] = PCM_CHANNEL_FC;
  ch_map[4] = PCM_CHANNEL_LS;
  ch_map[5] = PCM_CHANNEL_RS;
  break;
 case 8:
  ch_map[0] = PCM_CHANNEL_FL;
  ch_map[1] = PCM_CHANNEL_FR;
  ch_map[2] = PCM_CHANNEL_LFE;
  ch_map[3] = PCM_CHANNEL_FC;
  ch_map[4] = PCM_CHANNEL_LS;
  ch_map[5] = PCM_CHANNEL_RS;
  ch_map[6] = PCM_CHANNEL_LB;
  ch_map[7] = PCM_CHANNEL_RB;
  break;
 default:
  return -EINVAL;
 }

 return 0;
}
EXPORT_SYMBOL_GPL(q6dsp_map_channels);

int q6dsp_get_channel_allocation(int channels)
{
 int channel_allocation;

 /* HDMI spec CEA-861-E: Table 28 Audio InfoFrame Data Byte 4 */
 switch (channels) {
 case 2:
  channel_allocation = 0;
  break;
 case 3:
  channel_allocation = 0x02;
  break;
 case 4:
  channel_allocation = 0x06;
  break;
 case 5:
  channel_allocation = 0x0A;
  break;
 case 6:
  channel_allocation = 0x0B;
  break;
 case 7:
  channel_allocation = 0x12;
  break;
 case 8:
  channel_allocation = 0x13;
  break;
 default:
  return -EINVAL;
 }

 return channel_allocation;
}
EXPORT_SYMBOL_GPL(q6dsp_get_channel_allocation);

MODULE_DESCRIPTION("ASoC MSM QDSP6 helper functions");
MODULE_LICENSE("GPL v2");