162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * JZ47xx SoCs TCU Operating System Timer driver
462306a36Sopenharmony_ci *
562306a36Sopenharmony_ci * Copyright (C) 2016 Maarten ter Huurne <maarten@treewalker.org>
662306a36Sopenharmony_ci * Copyright (C) 2020 Paul Cercueil <paul@crapouillou.net>
762306a36Sopenharmony_ci */
862306a36Sopenharmony_ci
962306a36Sopenharmony_ci#include <linux/clk.h>
1062306a36Sopenharmony_ci#include <linux/clocksource.h>
1162306a36Sopenharmony_ci#include <linux/mfd/ingenic-tcu.h>
1262306a36Sopenharmony_ci#include <linux/mfd/syscon.h>
1362306a36Sopenharmony_ci#include <linux/of.h>
1462306a36Sopenharmony_ci#include <linux/platform_device.h>
1562306a36Sopenharmony_ci#include <linux/pm.h>
1662306a36Sopenharmony_ci#include <linux/regmap.h>
1762306a36Sopenharmony_ci#include <linux/sched_clock.h>
1862306a36Sopenharmony_ci
1962306a36Sopenharmony_ci#define TCU_OST_TCSR_MASK	0xffc0
2062306a36Sopenharmony_ci#define TCU_OST_TCSR_CNT_MD	BIT(15)
2162306a36Sopenharmony_ci
2262306a36Sopenharmony_ci#define TCU_OST_CHANNEL		15
2362306a36Sopenharmony_ci
2462306a36Sopenharmony_ci/*
2562306a36Sopenharmony_ci * The TCU_REG_OST_CNT{L,R} from <linux/mfd/ingenic-tcu.h> are only for the
2662306a36Sopenharmony_ci * regmap; these are for use with the __iomem pointer.
2762306a36Sopenharmony_ci */
2862306a36Sopenharmony_ci#define OST_REG_CNTL		0x4
2962306a36Sopenharmony_ci#define OST_REG_CNTH		0x8
3062306a36Sopenharmony_ci
3162306a36Sopenharmony_cistruct ingenic_ost_soc_info {
3262306a36Sopenharmony_ci	bool is64bit;
3362306a36Sopenharmony_ci};
3462306a36Sopenharmony_ci
3562306a36Sopenharmony_cistruct ingenic_ost {
3662306a36Sopenharmony_ci	void __iomem *regs;
3762306a36Sopenharmony_ci	struct clk *clk;
3862306a36Sopenharmony_ci
3962306a36Sopenharmony_ci	struct clocksource cs;
4062306a36Sopenharmony_ci};
4162306a36Sopenharmony_ci
4262306a36Sopenharmony_cistatic struct ingenic_ost *ingenic_ost;
4362306a36Sopenharmony_ci
4462306a36Sopenharmony_cistatic u64 notrace ingenic_ost_read_cntl(void)
4562306a36Sopenharmony_ci{
4662306a36Sopenharmony_ci	/* Read using __iomem pointer instead of regmap to avoid locking */
4762306a36Sopenharmony_ci	return readl(ingenic_ost->regs + OST_REG_CNTL);
4862306a36Sopenharmony_ci}
4962306a36Sopenharmony_ci
5062306a36Sopenharmony_cistatic u64 notrace ingenic_ost_read_cnth(void)
5162306a36Sopenharmony_ci{
5262306a36Sopenharmony_ci	/* Read using __iomem pointer instead of regmap to avoid locking */
5362306a36Sopenharmony_ci	return readl(ingenic_ost->regs + OST_REG_CNTH);
5462306a36Sopenharmony_ci}
5562306a36Sopenharmony_ci
5662306a36Sopenharmony_cistatic u64 notrace ingenic_ost_clocksource_readl(struct clocksource *cs)
5762306a36Sopenharmony_ci{
5862306a36Sopenharmony_ci	return ingenic_ost_read_cntl();
5962306a36Sopenharmony_ci}
6062306a36Sopenharmony_ci
6162306a36Sopenharmony_cistatic u64 notrace ingenic_ost_clocksource_readh(struct clocksource *cs)
6262306a36Sopenharmony_ci{
6362306a36Sopenharmony_ci	return ingenic_ost_read_cnth();
6462306a36Sopenharmony_ci}
6562306a36Sopenharmony_ci
6662306a36Sopenharmony_cistatic int __init ingenic_ost_probe(struct platform_device *pdev)
6762306a36Sopenharmony_ci{
6862306a36Sopenharmony_ci	const struct ingenic_ost_soc_info *soc_info;
6962306a36Sopenharmony_ci	struct device *dev = &pdev->dev;
7062306a36Sopenharmony_ci	struct ingenic_ost *ost;
7162306a36Sopenharmony_ci	struct clocksource *cs;
7262306a36Sopenharmony_ci	struct regmap *map;
7362306a36Sopenharmony_ci	unsigned long rate;
7462306a36Sopenharmony_ci	int err;
7562306a36Sopenharmony_ci
7662306a36Sopenharmony_ci	soc_info = device_get_match_data(dev);
7762306a36Sopenharmony_ci	if (!soc_info)
7862306a36Sopenharmony_ci		return -EINVAL;
7962306a36Sopenharmony_ci
8062306a36Sopenharmony_ci	ost = devm_kzalloc(dev, sizeof(*ost), GFP_KERNEL);
8162306a36Sopenharmony_ci	if (!ost)
8262306a36Sopenharmony_ci		return -ENOMEM;
8362306a36Sopenharmony_ci
8462306a36Sopenharmony_ci	ingenic_ost = ost;
8562306a36Sopenharmony_ci
8662306a36Sopenharmony_ci	ost->regs = devm_platform_ioremap_resource(pdev, 0);
8762306a36Sopenharmony_ci	if (IS_ERR(ost->regs))
8862306a36Sopenharmony_ci		return PTR_ERR(ost->regs);
8962306a36Sopenharmony_ci
9062306a36Sopenharmony_ci	map = device_node_to_regmap(dev->parent->of_node);
9162306a36Sopenharmony_ci	if (IS_ERR(map)) {
9262306a36Sopenharmony_ci		dev_err(dev, "regmap not found");
9362306a36Sopenharmony_ci		return PTR_ERR(map);
9462306a36Sopenharmony_ci	}
9562306a36Sopenharmony_ci
9662306a36Sopenharmony_ci	ost->clk = devm_clk_get(dev, "ost");
9762306a36Sopenharmony_ci	if (IS_ERR(ost->clk))
9862306a36Sopenharmony_ci		return PTR_ERR(ost->clk);
9962306a36Sopenharmony_ci
10062306a36Sopenharmony_ci	err = clk_prepare_enable(ost->clk);
10162306a36Sopenharmony_ci	if (err)
10262306a36Sopenharmony_ci		return err;
10362306a36Sopenharmony_ci
10462306a36Sopenharmony_ci	/* Clear counter high/low registers */
10562306a36Sopenharmony_ci	if (soc_info->is64bit)
10662306a36Sopenharmony_ci		regmap_write(map, TCU_REG_OST_CNTL, 0);
10762306a36Sopenharmony_ci	regmap_write(map, TCU_REG_OST_CNTH, 0);
10862306a36Sopenharmony_ci
10962306a36Sopenharmony_ci	/* Don't reset counter at compare value. */
11062306a36Sopenharmony_ci	regmap_update_bits(map, TCU_REG_OST_TCSR,
11162306a36Sopenharmony_ci			   TCU_OST_TCSR_MASK, TCU_OST_TCSR_CNT_MD);
11262306a36Sopenharmony_ci
11362306a36Sopenharmony_ci	rate = clk_get_rate(ost->clk);
11462306a36Sopenharmony_ci
11562306a36Sopenharmony_ci	/* Enable OST TCU channel */
11662306a36Sopenharmony_ci	regmap_write(map, TCU_REG_TESR, BIT(TCU_OST_CHANNEL));
11762306a36Sopenharmony_ci
11862306a36Sopenharmony_ci	cs = &ost->cs;
11962306a36Sopenharmony_ci	cs->name	= "ingenic-ost";
12062306a36Sopenharmony_ci	cs->rating	= 320;
12162306a36Sopenharmony_ci	cs->flags	= CLOCK_SOURCE_IS_CONTINUOUS;
12262306a36Sopenharmony_ci	cs->mask	= CLOCKSOURCE_MASK(32);
12362306a36Sopenharmony_ci
12462306a36Sopenharmony_ci	if (soc_info->is64bit)
12562306a36Sopenharmony_ci		cs->read = ingenic_ost_clocksource_readl;
12662306a36Sopenharmony_ci	else
12762306a36Sopenharmony_ci		cs->read = ingenic_ost_clocksource_readh;
12862306a36Sopenharmony_ci
12962306a36Sopenharmony_ci	err = clocksource_register_hz(cs, rate);
13062306a36Sopenharmony_ci	if (err) {
13162306a36Sopenharmony_ci		dev_err(dev, "clocksource registration failed");
13262306a36Sopenharmony_ci		clk_disable_unprepare(ost->clk);
13362306a36Sopenharmony_ci		return err;
13462306a36Sopenharmony_ci	}
13562306a36Sopenharmony_ci
13662306a36Sopenharmony_ci	if (soc_info->is64bit)
13762306a36Sopenharmony_ci		sched_clock_register(ingenic_ost_read_cntl, 32, rate);
13862306a36Sopenharmony_ci	else
13962306a36Sopenharmony_ci		sched_clock_register(ingenic_ost_read_cnth, 32, rate);
14062306a36Sopenharmony_ci
14162306a36Sopenharmony_ci	return 0;
14262306a36Sopenharmony_ci}
14362306a36Sopenharmony_ci
14462306a36Sopenharmony_cistatic int ingenic_ost_suspend(struct device *dev)
14562306a36Sopenharmony_ci{
14662306a36Sopenharmony_ci	struct ingenic_ost *ost = dev_get_drvdata(dev);
14762306a36Sopenharmony_ci
14862306a36Sopenharmony_ci	clk_disable(ost->clk);
14962306a36Sopenharmony_ci
15062306a36Sopenharmony_ci	return 0;
15162306a36Sopenharmony_ci}
15262306a36Sopenharmony_ci
15362306a36Sopenharmony_cistatic int ingenic_ost_resume(struct device *dev)
15462306a36Sopenharmony_ci{
15562306a36Sopenharmony_ci	struct ingenic_ost *ost = dev_get_drvdata(dev);
15662306a36Sopenharmony_ci
15762306a36Sopenharmony_ci	return clk_enable(ost->clk);
15862306a36Sopenharmony_ci}
15962306a36Sopenharmony_ci
16062306a36Sopenharmony_cistatic const struct dev_pm_ops ingenic_ost_pm_ops = {
16162306a36Sopenharmony_ci	/* _noirq: We want the OST clock to be gated last / ungated first */
16262306a36Sopenharmony_ci	.suspend_noirq = ingenic_ost_suspend,
16362306a36Sopenharmony_ci	.resume_noirq  = ingenic_ost_resume,
16462306a36Sopenharmony_ci};
16562306a36Sopenharmony_ci
16662306a36Sopenharmony_cistatic const struct ingenic_ost_soc_info jz4725b_ost_soc_info = {
16762306a36Sopenharmony_ci	.is64bit = false,
16862306a36Sopenharmony_ci};
16962306a36Sopenharmony_ci
17062306a36Sopenharmony_cistatic const struct ingenic_ost_soc_info jz4760b_ost_soc_info = {
17162306a36Sopenharmony_ci	.is64bit = true,
17262306a36Sopenharmony_ci};
17362306a36Sopenharmony_ci
17462306a36Sopenharmony_cistatic const struct of_device_id ingenic_ost_of_match[] = {
17562306a36Sopenharmony_ci	{ .compatible = "ingenic,jz4725b-ost", .data = &jz4725b_ost_soc_info, },
17662306a36Sopenharmony_ci	{ .compatible = "ingenic,jz4760b-ost", .data = &jz4760b_ost_soc_info, },
17762306a36Sopenharmony_ci	{ .compatible = "ingenic,jz4770-ost", .data = &jz4760b_ost_soc_info, },
17862306a36Sopenharmony_ci	{ }
17962306a36Sopenharmony_ci};
18062306a36Sopenharmony_ci
18162306a36Sopenharmony_cistatic struct platform_driver ingenic_ost_driver = {
18262306a36Sopenharmony_ci	.driver = {
18362306a36Sopenharmony_ci		.name = "ingenic-ost",
18462306a36Sopenharmony_ci		.pm = pm_sleep_ptr(&ingenic_ost_pm_ops),
18562306a36Sopenharmony_ci		.of_match_table = ingenic_ost_of_match,
18662306a36Sopenharmony_ci	},
18762306a36Sopenharmony_ci};
18862306a36Sopenharmony_cibuiltin_platform_driver_probe(ingenic_ost_driver, ingenic_ost_probe);
189