ARK-OS/arkos/kernel/drivers/reset/reset-imx8mp-audiomix.c

// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * Copyright 2024 NXP
 */

#include <dt-bindings/reset/fsl,imx8ulp-sim-lpav.h>
#include <dt-bindings/reset/imx8mp-reset-audiomix.h>

#include <linux/auxiliary_bus.h>
#include <linux/bits.h>
#include <linux/device.h>
#include <linux/io.h>
#include <linux/module.h>
#include <linux/of.h>
#include <linux/of_address.h>
#include <linux/regmap.h>
#include <linux/reset-controller.h>

#define IMX8MP_AUDIOMIX_EARC_RESET_OFFSET	0x200
#define IMX8MP_AUDIOMIX_DSP_RUNSTALL_OFFSET	0x108

#define IMX8ULP_SIM_LPAV_SYSCTRL0_OFFSET	0x8

struct imx8mp_reset_map {
	unsigned int offset;
	unsigned int mask;
	bool active_low;
};

struct imx8mp_reset_info {
	const struct imx8mp_reset_map *map;
	int num_lines;
};

static const struct imx8mp_reset_map imx8mp_reset_map[] = {
	[IMX8MP_AUDIOMIX_EARC_RESET] = {
		.offset	= IMX8MP_AUDIOMIX_EARC_RESET_OFFSET,
		.mask = BIT(0),
		.active_low = true,
	},
	[IMX8MP_AUDIOMIX_EARC_PHY_RESET] = {
		.offset	= IMX8MP_AUDIOMIX_EARC_RESET_OFFSET,
		.mask = BIT(1),
		.active_low = true,
	},
	[IMX8MP_AUDIOMIX_DSP_RUNSTALL] = {
		.offset	= IMX8MP_AUDIOMIX_DSP_RUNSTALL_OFFSET,
		.mask = BIT(5),
		.active_low = false,
	},
};

static const struct imx8mp_reset_info imx8mp_reset_info = {
	.map = imx8mp_reset_map,
	.num_lines = ARRAY_SIZE(imx8mp_reset_map),
};

static const struct imx8mp_reset_map imx8ulp_reset_map[] = {
	[IMX8ULP_SIM_LPAV_HIFI4_DSP_DBG_RST] = {
		.offset = IMX8ULP_SIM_LPAV_SYSCTRL0_OFFSET,
		.mask = BIT(25),
		.active_low = false,
	},
	[IMX8ULP_SIM_LPAV_HIFI4_DSP_RST] = {
		.offset = IMX8ULP_SIM_LPAV_SYSCTRL0_OFFSET,
		.mask = BIT(16),
		.active_low = false,
	},
	[IMX8ULP_SIM_LPAV_HIFI4_DSP_STALL] = {
		.offset = IMX8ULP_SIM_LPAV_SYSCTRL0_OFFSET,
		.mask = BIT(13),
		.active_low = false,
	},
	[IMX8ULP_SIM_LPAV_DSI_RST_BYTE_N] = {
		.offset = IMX8ULP_SIM_LPAV_SYSCTRL0_OFFSET,
		.mask = BIT(5),
		.active_low = true,
	},
	[IMX8ULP_SIM_LPAV_DSI_RST_ESC_N] = {
		.offset = IMX8ULP_SIM_LPAV_SYSCTRL0_OFFSET,
		.mask = BIT(4),
		.active_low = true,
	},
	[IMX8ULP_SIM_LPAV_DSI_RST_DPI_N] = {
		.offset = IMX8ULP_SIM_LPAV_SYSCTRL0_OFFSET,
		.mask = BIT(3),
		.active_low = true,
	},
};

static const struct imx8mp_reset_info imx8ulp_reset_info = {
	.map = imx8ulp_reset_map,
	.num_lines = ARRAY_SIZE(imx8ulp_reset_map),
};

struct imx8mp_audiomix_reset {
	struct reset_controller_dev rcdev;
	struct regmap *regmap;
	const struct imx8mp_reset_map *map;
};

static struct imx8mp_audiomix_reset *to_imx8mp_audiomix_reset(struct reset_controller_dev *rcdev)
{
	return container_of(rcdev, struct imx8mp_audiomix_reset, rcdev);
}

static int imx8mp_audiomix_update(struct reset_controller_dev *rcdev,
				  unsigned long id, bool assert)
{
	struct imx8mp_audiomix_reset *priv = to_imx8mp_audiomix_reset(rcdev);
	const struct imx8mp_reset_map *reset_map = priv->map;
	unsigned int mask, offset, active_low, val;

	mask = reset_map[id].mask;
	offset = reset_map[id].offset;
	active_low = reset_map[id].active_low;
	val = (active_low ^ assert) ? mask : ~mask;

	return regmap_update_bits(priv->regmap, offset, mask, val);
}

static int imx8mp_audiomix_reset_assert(struct reset_controller_dev *rcdev,
					unsigned long id)
{
	return imx8mp_audiomix_update(rcdev, id, true);
}

static int imx8mp_audiomix_reset_deassert(struct reset_controller_dev *rcdev,
					  unsigned long id)
{
	return imx8mp_audiomix_update(rcdev, id, false);
}

static const struct reset_control_ops imx8mp_audiomix_reset_ops = {
	.assert   = imx8mp_audiomix_reset_assert,
	.deassert = imx8mp_audiomix_reset_deassert,
};

static const struct regmap_config regmap_config = {
	.reg_bits = 32,
	.val_bits = 32,
	.reg_stride = 4,
};

/* assumption: registered only if not using parent regmap */
static void imx8mp_audiomix_reset_iounmap(void *data)
{
	void __iomem *base = (void __iomem *)data;

	iounmap(base);
}

static int imx8mp_audiomix_reset_get_regmap(struct imx8mp_audiomix_reset *priv)
{
	void __iomem *base;
	struct device *dev;
	int ret;

	dev = priv->rcdev.dev;

	/* try to use the parent's regmap */
	priv->regmap = dev_get_regmap(dev->parent, NULL);
	if (priv->regmap)
		return 0;

	/* ... if that's not possible then initialize the regmap right now */
	base = of_iomap(dev->parent->of_node, 0);
	if (!base)
		return dev_err_probe(dev, -ENOMEM, "failed to iomap address space\n");

	ret = devm_add_action_or_reset(dev,
				       imx8mp_audiomix_reset_iounmap,
				       (void __force *)base);
	if (ret)
		return dev_err_probe(dev, ret, "failed to register action\n");

	priv->regmap = devm_regmap_init_mmio(dev, base, &regmap_config);
	if (IS_ERR(priv->regmap))
		return dev_err_probe(dev, PTR_ERR(priv->regmap),
				     "failed to initialize regmap\n");

	return 0;
}

static int imx8mp_audiomix_reset_probe(struct auxiliary_device *adev,
				       const struct auxiliary_device_id *id)
{
	const struct imx8mp_reset_info *rinfo;
	struct imx8mp_audiomix_reset *priv;
	struct device *dev = &adev->dev;
	int ret;

	rinfo = (void *)id->driver_data;

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

	priv->rcdev.owner     = THIS_MODULE;
	priv->map             = rinfo->map;
	priv->rcdev.nr_resets = rinfo->num_lines;
	priv->rcdev.ops       = &imx8mp_audiomix_reset_ops;
	priv->rcdev.of_node   = dev->parent->of_node;
	priv->rcdev.dev	      = dev;
	priv->rcdev.of_reset_n_cells = 1;

	dev_set_drvdata(dev, priv);

	ret = imx8mp_audiomix_reset_get_regmap(priv);
	if (ret)
		return dev_err_probe(dev, ret, "failed to get regmap\n");

	ret = devm_reset_controller_register(dev, &priv->rcdev);
	if (ret)
		return dev_err_probe(dev, ret,
				     "failed to register reset controller\n");

	return 0;
}

static const struct auxiliary_device_id imx8mp_audiomix_reset_ids[] = {
	{
		.name = "clk_imx8mp_audiomix.reset",
		.driver_data = (kernel_ulong_t)&imx8mp_reset_info,
	},
	{
		.name = "clk_imx8ulp_sim_lpav.reset",
		.driver_data = (kernel_ulong_t)&imx8ulp_reset_info,
	},
	{ }
};
MODULE_DEVICE_TABLE(auxiliary, imx8mp_audiomix_reset_ids);

static struct auxiliary_driver imx8mp_audiomix_reset_driver = {
	.probe		= imx8mp_audiomix_reset_probe,
	.id_table	= imx8mp_audiomix_reset_ids,
};

module_auxiliary_driver(imx8mp_audiomix_reset_driver);

MODULE_AUTHOR("Shengjiu Wang <shengjiu.wang@nxp.com>");
MODULE_DESCRIPTION("Freescale i.MX8MP Audio Block Controller reset driver");
MODULE_LICENSE("GPL");