// SPDX-License-Identifier: GPL-2.0
//
// TI SRC4xxx Audio Codec driver
//
// Copyright 2021-2022 Deqx Pty Ltd
// Author: Matt Flax <flatmax@flatmax.com>

#include <linux/module.h>

#include <sound/soc.h>
#include <sound/tlv.h>

#include "src4xxx.h"

struct src4xxx {
	struct regmap *regmap;
	bool master[2];
	int mclk_hz;
	struct device *dev;
};

enum {SRC4XXX_PORTA, SRC4XXX_PORTB};

/* SRC attenuation */
static const DECLARE_TLV_DB_SCALE(src_tlv, -12750, 50, 0);

static const struct snd_kcontrol_new src4xxx_controls[] = {
	SOC_DOUBLE_R_TLV("SRC Volume",
		SRC4XXX_SCR_CTL_30, SRC4XXX_SCR_CTL_31, 0, 255, 1, src_tlv),
};

/* I2S port control */
static const char * const port_out_src_text[] = {
	"loopback", "other_port", "DIR", "SRC"
};
static SOC_ENUM_SINGLE_DECL(porta_out_src_enum, SRC4XXX_PORTA_CTL_03, 4,
	port_out_src_text);
static SOC_ENUM_SINGLE_DECL(portb_out_src_enum, SRC4XXX_PORTB_CTL_05, 4,
	port_out_src_text);
static const struct snd_kcontrol_new porta_out_control =
	SOC_DAPM_ENUM("Port A source select", porta_out_src_enum);
static const struct snd_kcontrol_new portb_out_control =
	SOC_DAPM_ENUM("Port B source select", portb_out_src_enum);

/* Digital audio transmitter control */
static const char * const dit_mux_text[] = {"Port A", "Port B", "DIR", "SRC"};
static SOC_ENUM_SINGLE_DECL(dit_mux_enum, SRC4XXX_TX_CTL_07, 3, dit_mux_text);
static const struct snd_kcontrol_new dit_mux_control =
	SOC_DAPM_ENUM("DIT source", dit_mux_enum);

/* SRC control */
static const char * const src_in_text[] = {"Port A", "Port B", "DIR"};
static SOC_ENUM_SINGLE_DECL(src_in_enum, SRC4XXX_SCR_CTL_2D, 0, src_in_text);
static const struct snd_kcontrol_new src_in_control =
	SOC_DAPM_ENUM("SRC source select", src_in_enum);

/* DIR control */
static const char * const dir_in_text[] = {"Ch 1", "Ch 2", "Ch 3", "Ch 4"};
static SOC_ENUM_SINGLE_DECL(dir_in_enum, SRC4XXX_RCV_CTL_0D, 0, dir_in_text);
static const struct snd_kcontrol_new dir_in_control =
	SOC_DAPM_ENUM("Digital Input", dir_in_enum);

static const struct snd_soc_dapm_widget src4xxx_dapm_widgets[] = {
	SND_SOC_DAPM_INPUT("loopback_A"),
	SND_SOC_DAPM_INPUT("other_port_A"),
	SND_SOC_DAPM_INPUT("DIR_A"),
	SND_SOC_DAPM_INPUT("SRC_A"),
	SND_SOC_DAPM_MUX("Port A source",
		SND_SOC_NOPM, 0, 0, &porta_out_control),

	SND_SOC_DAPM_INPUT("loopback_B"),
	SND_SOC_DAPM_INPUT("other_port_B"),
	SND_SOC_DAPM_INPUT("DIR_B"),
	SND_SOC_DAPM_INPUT("SRC_B"),
	SND_SOC_DAPM_MUX("Port B source",
		SND_SOC_NOPM, 0, 0, &portb_out_control),

	SND_SOC_DAPM_INPUT("Port_A"),
	SND_SOC_DAPM_INPUT("Port_B"),
	SND_SOC_DAPM_INPUT("DIR_"),

	/* Digital audio receivers and transmitters */
	SND_SOC_DAPM_OUTPUT("DIR_OUT"),
	SND_SOC_DAPM_OUTPUT("SRC_OUT"),
	SND_SOC_DAPM_MUX("DIT Out Src", SRC4XXX_PWR_RST_01,
		SRC4XXX_ENABLE_DIT_SHIFT, 1, &dit_mux_control),

	/* Audio Interface */
	SND_SOC_DAPM_AIF_IN("AIF_A_RX", "Playback A", 0,
		SRC4XXX_PWR_RST_01, SRC4XXX_ENABLE_PORT_A_SHIFT, 1),
	SND_SOC_DAPM_AIF_OUT("AIF_A_TX", "Capture A", 0,
		SRC4XXX_PWR_RST_01, SRC4XXX_ENABLE_PORT_A_SHIFT, 1),
	SND_SOC_DAPM_AIF_IN("AIF_B_RX", "Playback B", 0,
		SRC4XXX_PWR_RST_01, SRC4XXX_ENABLE_PORT_B_SHIFT, 1),
	SND_SOC_DAPM_AIF_OUT("AIF_B_TX", "Capture B", 0,
		SRC4XXX_PWR_RST_01, SRC4XXX_ENABLE_PORT_B_SHIFT, 1),

	SND_SOC_DAPM_MUX("SRC source", SND_SOC_NOPM, 0, 0, &src_in_control),

	SND_SOC_DAPM_INPUT("MCLK"),
	SND_SOC_DAPM_INPUT("RXMCLKI"),
	SND_SOC_DAPM_INPUT("RXMCLKO"),

	SND_SOC_DAPM_INPUT("RX1"),
	SND_SOC_DAPM_INPUT("RX2"),
	SND_SOC_DAPM_INPUT("RX3"),
	SND_SOC_DAPM_INPUT("RX4"),
	SND_SOC_DAPM_MUX("Digital Input", SRC4XXX_PWR_RST_01,
		SRC4XXX_ENABLE_DIR_SHIFT, 1, &dir_in_control),
};

static const struct snd_soc_dapm_route src4xxx_audio_routes[] = {
	/* I2S Input to Output Routing */
	{"Port A source", "loopback", "loopback_A"},
	{"Port A source", "other_port", "other_port_A"},
	{"Port A source", "DIR", "DIR_A"},
	{"Port A source", "SRC", "SRC_A"},
	{"Port B source", "loopback", "loopback_B"},
	{"Port B source", "other_port", "other_port_B"},
	{"Port B source", "DIR", "DIR_B"},
	{"Port B source", "SRC", "SRC_B"},
	/* DIT muxing */
	{"DIT Out Src", "Port A", "Capture A"},
	{"DIT Out Src", "Port B", "Capture B"},
	{"DIT Out Src", "DIR", "DIR_OUT"},
	{"DIT Out Src", "SRC", "SRC_OUT"},

	/* SRC input selection */
	{"SRC source", "Port A", "Port_A"},
	{"SRC source", "Port B", "Port_B"},
	{"SRC source", "DIR", "DIR_"},
	/* SRC mclk selection */
	{"SRC mclk source", "Master (MCLK)", "MCLK"},
	{"SRC mclk source", "Master (RXCLKI)", "RXMCLKI"},
	{"SRC mclk source", "Recovered receiver clk", "RXMCLKO"},
	/* DIR input selection */
	{"Digital Input", "Ch 1", "RX1"},
	{"Digital Input", "Ch 2", "RX2"},
	{"Digital Input", "Ch 3", "RX3"},
	{"Digital Input", "Ch 4", "RX4"},
};


static const struct snd_soc_component_driver src4xxx_driver = {
	.controls = src4xxx_controls,
	.num_controls = ARRAY_SIZE(src4xxx_controls),

	.dapm_widgets = src4xxx_dapm_widgets,
	.num_dapm_widgets = ARRAY_SIZE(src4xxx_dapm_widgets),
	.dapm_routes = src4xxx_audio_routes,
	.num_dapm_routes = ARRAY_SIZE(src4xxx_audio_routes),
};

static int src4xxx_set_dai_fmt(struct snd_soc_dai *dai, unsigned int fmt)
{
	struct snd_soc_component *component = dai->component;
	struct src4xxx *src4xxx = snd_soc_component_get_drvdata(component);
	unsigned int ctrl;

	switch (fmt & SND_SOC_DAIFMT_MASTER_MASK) {
	case SND_SOC_DAIFMT_CBM_CFM:
		ctrl = SRC4XXX_BUS_MASTER;
		src4xxx->master[dai->id] = true;
		break;
	case SND_SOC_DAIFMT_CBS_CFS:
		ctrl = 0;
		src4xxx->master[dai->id] = false;
		break;
	default:
		return -EINVAL;
		break;
	}

	switch (fmt & SND_SOC_DAIFMT_FORMAT_MASK) {
	case SND_SOC_DAIFMT_I2S:
		ctrl |= SRC4XXX_BUS_I2S;
		break;
	case SND_SOC_DAIFMT_LEFT_J:
		ctrl |= SRC4XXX_BUS_LEFT_J;
		break;
	case SND_SOC_DAIFMT_RIGHT_J:
		ctrl |= SRC4XXX_BUS_RIGHT_J_24;
		break;
	default:
		return -EINVAL;
		break;
	}

	switch (fmt & SND_SOC_DAIFMT_INV_MASK) {
	case SND_SOC_DAIFMT_NB_NF:
		break;
	default:
		return -EINVAL;
		break;
	}

	regmap_update_bits(src4xxx->regmap, SRC4XXX_BUS_FMT(dai->id),
		SRC4XXX_BUS_FMT_MS_MASK, ctrl);

	return 0;
}

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

	dev_info(component->dev, "changing mclk rate from %d to %d Hz\n",
		src4xxx->mclk_hz, freq);
	src4xxx->mclk_hz = freq;

	return 0;
}

static int src4xxx_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 src4xxx *src4xxx = snd_soc_component_get_drvdata(component);
	unsigned int mclk_div;
	int val, pj, jd, d;
	int reg;
	int ret;

	switch (dai->id) {
	case SRC4XXX_PORTB:
		reg = SRC4XXX_PORTB_CTL_06;
		break;
	default:
		reg = SRC4XXX_PORTA_CTL_04;
		break;
	}

	if (src4xxx->master[dai->id]) {
		mclk_div = src4xxx->mclk_hz/params_rate(params);
		if (src4xxx->mclk_hz != mclk_div*params_rate(params)) {
			dev_err(component->dev,
				"mclk %d / rate %d has a remainder.\n",
				src4xxx->mclk_hz, params_rate(params));
			return -EINVAL;
		}

		val = ((int)mclk_div - 128) / 128;
		if ((val < 0) | (val > 3)) {
			dev_err(component->dev,
				"div register setting %d is out of range\n",
				val);
			dev_err(component->dev,
				"unsupported sample rate %d Hz for the master clock of %d Hz\n",
				params_rate(params), src4xxx->mclk_hz);
			return -EINVAL;
		}

		/* set the TX DIV */
		ret = regmap_update_bits(src4xxx->regmap,
			SRC4XXX_TX_CTL_07, SRC4XXX_TX_MCLK_DIV_MASK,
			val<<SRC4XXX_TX_MCLK_DIV_SHIFT);
		if (ret) {
			dev_err(component->dev,
				"Couldn't set the TX's div register to %d << %d = 0x%x\n",
				val, SRC4XXX_TX_MCLK_DIV_SHIFT,
				val<<SRC4XXX_TX_MCLK_DIV_SHIFT);
			return ret;
		}

		/* set the PLL for the digital receiver */
		switch (src4xxx->mclk_hz) {
		case 24576000:
			pj = 0x22;
			jd = 0x00;
			d = 0x00;
			break;
		case 22579200:
			pj = 0x22;
			jd = 0x1b;
			d = 0xa3;
			break;
		default:
			/* don't error out here,
			 * other parts of the chip are still functional
			 * Dummy initialize variables to avoid
			 * -Wsometimes-uninitialized from clang.
			 */
			dev_info(component->dev,
				"Couldn't set the RCV PLL as this master clock rate is unknown. Chosen regmap values may not match real world values.\n");
			pj = 0x0;
			jd = 0xff;
			d = 0xff;
			break;
		}
		ret = regmap_write(src4xxx->regmap, SRC4XXX_RCV_PLL_0F, pj);
		if (ret < 0)
			dev_err(component->dev,
				"Failed to update PLL register 0x%x\n",
				SRC4XXX_RCV_PLL_0F);
		ret = regmap_write(src4xxx->regmap, SRC4XXX_RCV_PLL_10, jd);
		if (ret < 0)
			dev_err(component->dev,
				"Failed to update PLL register 0x%x\n",
				SRC4XXX_RCV_PLL_10);
		ret = regmap_write(src4xxx->regmap, SRC4XXX_RCV_PLL_11, d);
		if (ret < 0)
			dev_err(component->dev,
				"Failed to update PLL register 0x%x\n",
				SRC4XXX_RCV_PLL_11);

		ret = regmap_update_bits(src4xxx->regmap,
			SRC4XXX_TX_CTL_07, SRC4XXX_TX_MCLK_DIV_MASK,
			val<<SRC4XXX_TX_MCLK_DIV_SHIFT);
		if (ret < 0) {
			dev_err(component->dev,
				"Couldn't set the TX's div register to %d << %d = 0x%x\n",
				val, SRC4XXX_TX_MCLK_DIV_SHIFT,
				val<<SRC4XXX_TX_MCLK_DIV_SHIFT);
			return ret;
		}

		return regmap_update_bits(src4xxx->regmap, reg,
					SRC4XXX_MCLK_DIV_MASK, val);
	} else {
		dev_info(dai->dev, "not setting up MCLK as not master\n");
	}

	return 0;
};

static const struct snd_soc_dai_ops src4xxx_dai_ops = {
	.hw_params	= src4xxx_hw_params,
	.set_sysclk	= src4xxx_set_mclk_hz,
	.set_fmt	= src4xxx_set_dai_fmt,
};

#define SRC4XXX_FORMATS (SNDRV_PCM_FMTBIT_S16_LE |	SNDRV_PCM_FMTBIT_S32_LE)
#define SRC4XXX_RATES (SNDRV_PCM_RATE_44100|SNDRV_PCM_RATE_48000|\
				SNDRV_PCM_RATE_88200|\
				SNDRV_PCM_RATE_96000|\
				SNDRV_PCM_RATE_176400|\
				SNDRV_PCM_RATE_192000)

static struct snd_soc_dai_driver src4xxx_dai_driver[] = {
	{
		.id = SRC4XXX_PORTA,
		.name = "src4xxx-portA",
		.playback = {
			.stream_name = "Playback A",
			.channels_min = 2,
			.channels_max = 2,
			.rates = SRC4XXX_RATES,
			.formats = SRC4XXX_FORMATS,
		},
		.capture = {
			.stream_name = "Capture A",
			.channels_min = 2,
			.channels_max = 2,
			.rates = SRC4XXX_RATES,
			.formats = SRC4XXX_FORMATS,
		},
		.ops = &src4xxx_dai_ops,
	},
	{
		.id = SRC4XXX_PORTB,
		.name = "src4xxx-portB",
		.playback = {
			.stream_name = "Playback B",
			.channels_min = 2,
			.channels_max = 2,
			.rates = SRC4XXX_RATES,
			.formats = SRC4XXX_FORMATS,
		},
		.capture = {
			.stream_name = "Capture B",
			.channels_min = 2,
			.channels_max = 2,
			.rates = SRC4XXX_RATES,
			.formats = SRC4XXX_FORMATS,
		},
		.ops = &src4xxx_dai_ops,
	},
};

static const struct reg_default src4xxx_reg_defaults[] = {
	{ SRC4XXX_PWR_RST_01,		0x00 }, /* all powered down intially */
	{ SRC4XXX_PORTA_CTL_03,		0x00 },
	{ SRC4XXX_PORTA_CTL_04,		0x00 },
	{ SRC4XXX_PORTB_CTL_05,		0x00 },
	{ SRC4XXX_PORTB_CTL_06,		0x00 },
	{ SRC4XXX_TX_CTL_07,		0x00 },
	{ SRC4XXX_TX_CTL_08,		0x00 },
	{ SRC4XXX_TX_CTL_09,		0x00 },
	{ SRC4XXX_SRC_DIT_IRQ_MSK_0B,	0x00 },
	{ SRC4XXX_SRC_DIT_IRQ_MODE_0C,	0x00 },
	{ SRC4XXX_RCV_CTL_0D,		0x00 },
	{ SRC4XXX_RCV_CTL_0E,		0x00 },
	{ SRC4XXX_RCV_PLL_0F,		0x00 }, /* not spec. in the datasheet */
	{ SRC4XXX_RCV_PLL_10,		0xff }, /* not spec. in the datasheet */
	{ SRC4XXX_RCV_PLL_11,		0xff }, /* not spec. in the datasheet */
	{ SRC4XXX_RVC_IRQ_MSK_16,	0x00 },
	{ SRC4XXX_RVC_IRQ_MSK_17,	0x00 },
	{ SRC4XXX_RVC_IRQ_MODE_18,	0x00 },
	{ SRC4XXX_RVC_IRQ_MODE_19,	0x00 },
	{ SRC4XXX_RVC_IRQ_MODE_1A,	0x00 },
	{ SRC4XXX_GPIO_1_1B,		0x00 },
	{ SRC4XXX_GPIO_2_1C,		0x00 },
	{ SRC4XXX_GPIO_3_1D,		0x00 },
	{ SRC4XXX_GPIO_4_1E,		0x00 },
	{ SRC4XXX_SCR_CTL_2D,		0x00 },
	{ SRC4XXX_SCR_CTL_2E,		0x00 },
	{ SRC4XXX_SCR_CTL_2F,		0x00 },
	{ SRC4XXX_SCR_CTL_30,		0x00 },
	{ SRC4XXX_SCR_CTL_31,		0x00 },
};

int src4xxx_probe(struct device *dev, struct regmap *regmap,
			void (*switch_mode)(struct device *dev))
{
	struct src4xxx *src4xxx;
	int ret;

	if (IS_ERR(regmap))
		return PTR_ERR(regmap);

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

	src4xxx->regmap = regmap;
	src4xxx->dev = dev;
	src4xxx->mclk_hz = 0; /* mclk has not been configured yet */
	dev_set_drvdata(dev, src4xxx);

	ret = regmap_write(regmap, SRC4XXX_PWR_RST_01, SRC4XXX_RESET);
	if (ret < 0)
		dev_err(dev, "Failed to issue reset: %d\n", ret);
	usleep_range(1, 500); /* sleep for more then 500 ns */
	ret = regmap_write(regmap, SRC4XXX_PWR_RST_01, SRC4XXX_POWER_DOWN);
	if (ret < 0)
		dev_err(dev, "Failed to decommission reset: %d\n", ret);
	usleep_range(500, 1000); /* sleep for 500 us or more */

	ret = regmap_update_bits(src4xxx->regmap, SRC4XXX_PWR_RST_01,
		SRC4XXX_POWER_ENABLE, SRC4XXX_POWER_ENABLE);
	if (ret < 0)
		dev_err(dev, "Failed to port A and B : %d\n", ret);

	/* set receiver to use master clock (rcv mclk is most likely jittery) */
	ret = regmap_update_bits(src4xxx->regmap, SRC4XXX_RCV_CTL_0D,
		SRC4XXX_RXCLK_MCLK,	SRC4XXX_RXCLK_MCLK);
	if (ret < 0)
		dev_err(dev,
			"Failed to enable mclk as the PLL1 DIR reference : %d\n", ret);

	/* default to leaving the PLL2 running on loss of lock, divide by 8 */
	ret = regmap_update_bits(src4xxx->regmap, SRC4XXX_RCV_CTL_0E,
		SRC4XXX_PLL2_DIV_8 | SRC4XXX_REC_MCLK_EN | SRC4XXX_PLL2_LOL,
		SRC4XXX_PLL2_DIV_8 | SRC4XXX_REC_MCLK_EN | SRC4XXX_PLL2_LOL);
	if (ret < 0)
		dev_err(dev, "Failed to enable mclk rec and div : %d\n", ret);

	ret = devm_snd_soc_register_component(dev, &src4xxx_driver,
			src4xxx_dai_driver, ARRAY_SIZE(src4xxx_dai_driver));
	if (ret == 0)
		dev_info(dev, "src4392 probe ok %d\n", ret);
	return ret;
}
EXPORT_SYMBOL_GPL(src4xxx_probe);

static bool src4xxx_volatile_register(struct device *dev, unsigned int reg)
{
	switch (reg) {
	case SRC4XXX_RES_00:
	case SRC4XXX_GLOBAL_ITR_STS_02:
	case SRC4XXX_SRC_DIT_STS_0A:
	case SRC4XXX_NON_AUDIO_D_12:
	case SRC4XXX_RVC_STS_13:
	case SRC4XXX_RVC_STS_14:
	case SRC4XXX_RVC_STS_15:
	case SRC4XXX_SUB_CODE_1F:
	case SRC4XXX_SUB_CODE_20:
	case SRC4XXX_SUB_CODE_21:
	case SRC4XXX_SUB_CODE_22:
	case SRC4XXX_SUB_CODE_23:
	case SRC4XXX_SUB_CODE_24:
	case SRC4XXX_SUB_CODE_25:
	case SRC4XXX_SUB_CODE_26:
	case SRC4XXX_SUB_CODE_27:
	case SRC4XXX_SUB_CODE_28:
	case SRC4XXX_PC_PREAMBLE_HI_29:
	case SRC4XXX_PC_PREAMBLE_LO_2A:
	case SRC4XXX_PD_PREAMBLE_HI_2B:
	case SRC4XXX_PC_PREAMBLE_LO_2C:
	case SRC4XXX_IO_RATIO_32:
	case SRC4XXX_IO_RATIO_33:
		return true;
	}

	if (reg > SRC4XXX_IO_RATIO_33 && reg < SRC4XXX_PAGE_SEL_7F)
		return true;

	return false;
}

const struct regmap_config src4xxx_regmap_config = {
	.val_bits = 8,
	.reg_bits = 8,
	.max_register = SRC4XXX_IO_RATIO_33,

	.reg_defaults = src4xxx_reg_defaults,
	.num_reg_defaults = ARRAY_SIZE(src4xxx_reg_defaults),
	.volatile_reg = src4xxx_volatile_register,
	.cache_type = REGCACHE_RBTREE,
};
EXPORT_SYMBOL_GPL(src4xxx_regmap_config);

MODULE_DESCRIPTION("ASoC SRC4XXX CODEC driver");
MODULE_AUTHOR("Matt Flax <flatmax@flatmax.com>");
MODULE_LICENSE("GPL");