/*
 * Tegra ACONNECT Bus Driver
 *
 * Copyright (C) 2016-2019, NVIDIA CORPORATION.  All rights reserved.
 *
 * This file is subject to the terms and conditions of the GNU General Public
 * License.  See the file "COPYING" in the main directory of this archive
 * for more details.
 */

#include <linux/clk.h>
#include <linux/module.h>
#include <linux/of_platform.h>
#include <linux/platform_device.h>
#include <linux/pm_runtime.h>

struct tegra_aconnect {
	struct clk	*ape_clk;
	struct clk	*apb2ape_clk;
	const struct tegra_aconnect_soc_data *soc_data;
};

struct tegra_aconnect_soc_data {
	bool is_hv;
};

static const struct tegra_aconnect_soc_data soc_data_tegra = {
	.is_hv = false,
};

static const struct tegra_aconnect_soc_data soc_data_tegra_hv = {
	.is_hv = true,
};

static const struct of_device_id tegra_aconnect_of_match[] = {
	{ .compatible = "nvidia,tegra210-aconnect", .data = &soc_data_tegra },
	{ .compatible = "nvidia,tegra186-aconnect-hv",
						.data = &soc_data_tegra_hv },
	{ }
};
static int tegra_aconnect_probe(struct platform_device *pdev)
{
	struct tegra_aconnect *aconnect;
	const struct of_device_id *match;

	if (!pdev->dev.of_node)
		return -EINVAL;

	aconnect = devm_kzalloc(&pdev->dev, sizeof(struct tegra_aconnect),
				GFP_KERNEL);
	if (!aconnect)
		return -ENOMEM;

	match = of_match_device(tegra_aconnect_of_match, &pdev->dev);
	if (!match) {
		dev_err(&pdev->dev, "Error: No device match found\n");
		return -ENODEV;
	}

	aconnect->soc_data = (struct tegra_aconnect_soc_data *)match->data;
	if (!aconnect->soc_data->is_hv) {
		aconnect->ape_clk = devm_clk_get(&pdev->dev, "ape");
		if (IS_ERR(aconnect->ape_clk)) {
			dev_err(&pdev->dev, "Can't retrieve ape clock\n");
			return PTR_ERR(aconnect->ape_clk);
		}

		aconnect->apb2ape_clk = devm_clk_get(&pdev->dev, "apb2ape");
		if (IS_ERR(aconnect->apb2ape_clk)) {
			dev_err(&pdev->dev, "Can't retrieve apb2ape clock\n");
			return PTR_ERR(aconnect->apb2ape_clk);
		}
	}
	dev_set_drvdata(&pdev->dev, aconnect);

	pm_runtime_enable(&pdev->dev);

	of_platform_populate(pdev->dev.of_node, NULL, NULL, &pdev->dev);

	dev_info(&pdev->dev, "Tegra ACONNECT bus registered\n");

	return 0;
}

#if IS_MODULE(CONFIG_TEGRA_ACONNECT)
static int tegra_aconnect_remove(struct platform_device *pdev)
{
	pm_runtime_disable(&pdev->dev);

	return 0;
}
#endif

static int tegra_aconnect_runtime_resume(struct device *dev)
{
	struct tegra_aconnect *aconnect = dev_get_drvdata(dev);
	int ret;

	if (!aconnect->soc_data->is_hv) {
		ret = clk_prepare_enable(aconnect->ape_clk);
		if (ret) {
			dev_err(dev, "ape clk_enable failed: %d\n", ret);
			return ret;
		}

		ret = clk_prepare_enable(aconnect->apb2ape_clk);
		if (ret) {
			clk_disable_unprepare(aconnect->ape_clk);
			dev_err(dev, "apb2ape clk_enable failed: %d\n", ret);
			return ret;
		}
	}

	return 0;
}

static int tegra_aconnect_runtime_suspend(struct device *dev)
{
	struct tegra_aconnect *aconnect = dev_get_drvdata(dev);
	if (!aconnect->soc_data->is_hv) {
		clk_disable_unprepare(aconnect->ape_clk);
		clk_disable_unprepare(aconnect->apb2ape_clk);
	}
	return 0;
}

static const struct dev_pm_ops tegra_aconnect_pm_ops = {
	SET_RUNTIME_PM_OPS(tegra_aconnect_runtime_suspend,
			tegra_aconnect_runtime_resume, NULL)
	SET_NOIRQ_SYSTEM_SLEEP_PM_OPS(pm_runtime_force_suspend,
				      pm_runtime_force_resume)

};

MODULE_DEVICE_TABLE(of, tegra_aconnect_of_match);

static struct platform_driver tegra_aconnect_driver = {
	.probe = tegra_aconnect_probe,
#if IS_MODULE(CONFIG_TEGRA_ACONNECT)
	.remove = tegra_aconnect_remove,
#endif
	.driver = {
		.name = "tegra-aconnect",
		.of_match_table = tegra_aconnect_of_match,
		.pm = &tegra_aconnect_pm_ops,
	},
};
module_platform_driver(tegra_aconnect_driver);

MODULE_DESCRIPTION("NVIDIA Tegra ACONNECT Bus Driver");
MODULE_AUTHOR("Jon Hunter <jonathanh@nvidia.com>");
MODULE_LICENSE("GPL v2");