162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0+
262306a36Sopenharmony_ci//
362306a36Sopenharmony_ci//  Copyright (C) 2000-2001 Deep Blue Solutions
462306a36Sopenharmony_ci//  Copyright (C) 2002 Shane Nay (shane@minirl.com)
562306a36Sopenharmony_ci//  Copyright (C) 2006-2007 Pavel Pisa (ppisa@pikron.com)
662306a36Sopenharmony_ci//  Copyright (C) 2008 Juergen Beisert (kernel@pengutronix.de)
762306a36Sopenharmony_ci//  Copyright (C) 2010 Freescale Semiconductor, Inc. All Rights Reserved.
862306a36Sopenharmony_ci
962306a36Sopenharmony_ci#include <linux/err.h>
1062306a36Sopenharmony_ci#include <linux/interrupt.h>
1162306a36Sopenharmony_ci#include <linux/irq.h>
1262306a36Sopenharmony_ci#include <linux/clockchips.h>
1362306a36Sopenharmony_ci#include <linux/clk.h>
1462306a36Sopenharmony_ci#include <linux/of.h>
1562306a36Sopenharmony_ci#include <linux/of_address.h>
1662306a36Sopenharmony_ci#include <linux/of_irq.h>
1762306a36Sopenharmony_ci#include <linux/stmp_device.h>
1862306a36Sopenharmony_ci#include <linux/sched_clock.h>
1962306a36Sopenharmony_ci
2062306a36Sopenharmony_ci/*
2162306a36Sopenharmony_ci * There are 2 versions of the timrot on Freescale MXS-based SoCs.
2262306a36Sopenharmony_ci * The v1 on MX23 only gets 16 bits counter, while v2 on MX28
2362306a36Sopenharmony_ci * extends the counter to 32 bits.
2462306a36Sopenharmony_ci *
2562306a36Sopenharmony_ci * The implementation uses two timers, one for clock_event and
2662306a36Sopenharmony_ci * another for clocksource. MX28 uses timrot 0 and 1, while MX23
2762306a36Sopenharmony_ci * uses 0 and 2.
2862306a36Sopenharmony_ci */
2962306a36Sopenharmony_ci
3062306a36Sopenharmony_ci#define MX23_TIMROT_VERSION_OFFSET	0x0a0
3162306a36Sopenharmony_ci#define MX28_TIMROT_VERSION_OFFSET	0x120
3262306a36Sopenharmony_ci#define BP_TIMROT_MAJOR_VERSION		24
3362306a36Sopenharmony_ci#define BV_TIMROT_VERSION_1		0x01
3462306a36Sopenharmony_ci#define BV_TIMROT_VERSION_2		0x02
3562306a36Sopenharmony_ci#define timrot_is_v1()	(timrot_major_version == BV_TIMROT_VERSION_1)
3662306a36Sopenharmony_ci
3762306a36Sopenharmony_ci/*
3862306a36Sopenharmony_ci * There are 4 registers for each timrotv2 instance, and 2 registers
3962306a36Sopenharmony_ci * for each timrotv1. So address step 0x40 in macros below strides
4062306a36Sopenharmony_ci * one instance of timrotv2 while two instances of timrotv1.
4162306a36Sopenharmony_ci *
4262306a36Sopenharmony_ci * As the result, HW_TIMROT_XXXn(1) defines the address of timrot1
4362306a36Sopenharmony_ci * on MX28 while timrot2 on MX23.
4462306a36Sopenharmony_ci */
4562306a36Sopenharmony_ci/* common between v1 and v2 */
4662306a36Sopenharmony_ci#define HW_TIMROT_ROTCTRL		0x00
4762306a36Sopenharmony_ci#define HW_TIMROT_TIMCTRLn(n)		(0x20 + (n) * 0x40)
4862306a36Sopenharmony_ci/* v1 only */
4962306a36Sopenharmony_ci#define HW_TIMROT_TIMCOUNTn(n)		(0x30 + (n) * 0x40)
5062306a36Sopenharmony_ci/* v2 only */
5162306a36Sopenharmony_ci#define HW_TIMROT_RUNNING_COUNTn(n)	(0x30 + (n) * 0x40)
5262306a36Sopenharmony_ci#define HW_TIMROT_FIXED_COUNTn(n)	(0x40 + (n) * 0x40)
5362306a36Sopenharmony_ci
5462306a36Sopenharmony_ci#define BM_TIMROT_TIMCTRLn_RELOAD	(1 << 6)
5562306a36Sopenharmony_ci#define BM_TIMROT_TIMCTRLn_UPDATE	(1 << 7)
5662306a36Sopenharmony_ci#define BM_TIMROT_TIMCTRLn_IRQ_EN	(1 << 14)
5762306a36Sopenharmony_ci#define BM_TIMROT_TIMCTRLn_IRQ		(1 << 15)
5862306a36Sopenharmony_ci#define BP_TIMROT_TIMCTRLn_SELECT	0
5962306a36Sopenharmony_ci#define BV_TIMROTv1_TIMCTRLn_SELECT__32KHZ_XTAL		0x8
6062306a36Sopenharmony_ci#define BV_TIMROTv2_TIMCTRLn_SELECT__32KHZ_XTAL		0xb
6162306a36Sopenharmony_ci#define BV_TIMROTv2_TIMCTRLn_SELECT__TICK_ALWAYS	0xf
6262306a36Sopenharmony_ci
6362306a36Sopenharmony_cistatic struct clock_event_device mxs_clockevent_device;
6462306a36Sopenharmony_ci
6562306a36Sopenharmony_cistatic void __iomem *mxs_timrot_base;
6662306a36Sopenharmony_cistatic u32 timrot_major_version;
6762306a36Sopenharmony_ci
6862306a36Sopenharmony_cistatic inline void timrot_irq_disable(void)
6962306a36Sopenharmony_ci{
7062306a36Sopenharmony_ci	__raw_writel(BM_TIMROT_TIMCTRLn_IRQ_EN, mxs_timrot_base +
7162306a36Sopenharmony_ci		     HW_TIMROT_TIMCTRLn(0) + STMP_OFFSET_REG_CLR);
7262306a36Sopenharmony_ci}
7362306a36Sopenharmony_ci
7462306a36Sopenharmony_cistatic inline void timrot_irq_enable(void)
7562306a36Sopenharmony_ci{
7662306a36Sopenharmony_ci	__raw_writel(BM_TIMROT_TIMCTRLn_IRQ_EN, mxs_timrot_base +
7762306a36Sopenharmony_ci		     HW_TIMROT_TIMCTRLn(0) + STMP_OFFSET_REG_SET);
7862306a36Sopenharmony_ci}
7962306a36Sopenharmony_ci
8062306a36Sopenharmony_cistatic void timrot_irq_acknowledge(void)
8162306a36Sopenharmony_ci{
8262306a36Sopenharmony_ci	__raw_writel(BM_TIMROT_TIMCTRLn_IRQ, mxs_timrot_base +
8362306a36Sopenharmony_ci		     HW_TIMROT_TIMCTRLn(0) + STMP_OFFSET_REG_CLR);
8462306a36Sopenharmony_ci}
8562306a36Sopenharmony_ci
8662306a36Sopenharmony_cistatic u64 timrotv1_get_cycles(struct clocksource *cs)
8762306a36Sopenharmony_ci{
8862306a36Sopenharmony_ci	return ~((__raw_readl(mxs_timrot_base + HW_TIMROT_TIMCOUNTn(1))
8962306a36Sopenharmony_ci			& 0xffff0000) >> 16);
9062306a36Sopenharmony_ci}
9162306a36Sopenharmony_ci
9262306a36Sopenharmony_cistatic int timrotv1_set_next_event(unsigned long evt,
9362306a36Sopenharmony_ci					struct clock_event_device *dev)
9462306a36Sopenharmony_ci{
9562306a36Sopenharmony_ci	/* timrot decrements the count */
9662306a36Sopenharmony_ci	__raw_writel(evt, mxs_timrot_base + HW_TIMROT_TIMCOUNTn(0));
9762306a36Sopenharmony_ci
9862306a36Sopenharmony_ci	return 0;
9962306a36Sopenharmony_ci}
10062306a36Sopenharmony_ci
10162306a36Sopenharmony_cistatic int timrotv2_set_next_event(unsigned long evt,
10262306a36Sopenharmony_ci					struct clock_event_device *dev)
10362306a36Sopenharmony_ci{
10462306a36Sopenharmony_ci	/* timrot decrements the count */
10562306a36Sopenharmony_ci	__raw_writel(evt, mxs_timrot_base + HW_TIMROT_FIXED_COUNTn(0));
10662306a36Sopenharmony_ci
10762306a36Sopenharmony_ci	return 0;
10862306a36Sopenharmony_ci}
10962306a36Sopenharmony_ci
11062306a36Sopenharmony_cistatic irqreturn_t mxs_timer_interrupt(int irq, void *dev_id)
11162306a36Sopenharmony_ci{
11262306a36Sopenharmony_ci	struct clock_event_device *evt = dev_id;
11362306a36Sopenharmony_ci
11462306a36Sopenharmony_ci	timrot_irq_acknowledge();
11562306a36Sopenharmony_ci	evt->event_handler(evt);
11662306a36Sopenharmony_ci
11762306a36Sopenharmony_ci	return IRQ_HANDLED;
11862306a36Sopenharmony_ci}
11962306a36Sopenharmony_ci
12062306a36Sopenharmony_cistatic void mxs_irq_clear(char *state)
12162306a36Sopenharmony_ci{
12262306a36Sopenharmony_ci	/* Disable interrupt in timer module */
12362306a36Sopenharmony_ci	timrot_irq_disable();
12462306a36Sopenharmony_ci
12562306a36Sopenharmony_ci	/* Set event time into the furthest future */
12662306a36Sopenharmony_ci	if (timrot_is_v1())
12762306a36Sopenharmony_ci		__raw_writel(0xffff, mxs_timrot_base + HW_TIMROT_TIMCOUNTn(1));
12862306a36Sopenharmony_ci	else
12962306a36Sopenharmony_ci		__raw_writel(0xffffffff,
13062306a36Sopenharmony_ci			     mxs_timrot_base + HW_TIMROT_FIXED_COUNTn(1));
13162306a36Sopenharmony_ci
13262306a36Sopenharmony_ci	/* Clear pending interrupt */
13362306a36Sopenharmony_ci	timrot_irq_acknowledge();
13462306a36Sopenharmony_ci	pr_debug("%s: changing mode to %s\n", __func__, state);
13562306a36Sopenharmony_ci}
13662306a36Sopenharmony_ci
13762306a36Sopenharmony_cistatic int mxs_shutdown(struct clock_event_device *evt)
13862306a36Sopenharmony_ci{
13962306a36Sopenharmony_ci	mxs_irq_clear("shutdown");
14062306a36Sopenharmony_ci
14162306a36Sopenharmony_ci	return 0;
14262306a36Sopenharmony_ci}
14362306a36Sopenharmony_ci
14462306a36Sopenharmony_cistatic int mxs_set_oneshot(struct clock_event_device *evt)
14562306a36Sopenharmony_ci{
14662306a36Sopenharmony_ci	if (clockevent_state_oneshot(evt))
14762306a36Sopenharmony_ci		mxs_irq_clear("oneshot");
14862306a36Sopenharmony_ci	timrot_irq_enable();
14962306a36Sopenharmony_ci	return 0;
15062306a36Sopenharmony_ci}
15162306a36Sopenharmony_ci
15262306a36Sopenharmony_cistatic struct clock_event_device mxs_clockevent_device = {
15362306a36Sopenharmony_ci	.name			= "mxs_timrot",
15462306a36Sopenharmony_ci	.features		= CLOCK_EVT_FEAT_ONESHOT,
15562306a36Sopenharmony_ci	.set_state_shutdown	= mxs_shutdown,
15662306a36Sopenharmony_ci	.set_state_oneshot	= mxs_set_oneshot,
15762306a36Sopenharmony_ci	.tick_resume		= mxs_shutdown,
15862306a36Sopenharmony_ci	.set_next_event		= timrotv2_set_next_event,
15962306a36Sopenharmony_ci	.rating			= 200,
16062306a36Sopenharmony_ci};
16162306a36Sopenharmony_ci
16262306a36Sopenharmony_cistatic int __init mxs_clockevent_init(struct clk *timer_clk)
16362306a36Sopenharmony_ci{
16462306a36Sopenharmony_ci	if (timrot_is_v1())
16562306a36Sopenharmony_ci		mxs_clockevent_device.set_next_event = timrotv1_set_next_event;
16662306a36Sopenharmony_ci	mxs_clockevent_device.cpumask = cpumask_of(0);
16762306a36Sopenharmony_ci	clockevents_config_and_register(&mxs_clockevent_device,
16862306a36Sopenharmony_ci					clk_get_rate(timer_clk),
16962306a36Sopenharmony_ci					timrot_is_v1() ? 0xf : 0x2,
17062306a36Sopenharmony_ci					timrot_is_v1() ? 0xfffe : 0xfffffffe);
17162306a36Sopenharmony_ci
17262306a36Sopenharmony_ci	return 0;
17362306a36Sopenharmony_ci}
17462306a36Sopenharmony_ci
17562306a36Sopenharmony_cistatic struct clocksource clocksource_mxs = {
17662306a36Sopenharmony_ci	.name		= "mxs_timer",
17762306a36Sopenharmony_ci	.rating		= 200,
17862306a36Sopenharmony_ci	.read		= timrotv1_get_cycles,
17962306a36Sopenharmony_ci	.mask		= CLOCKSOURCE_MASK(16),
18062306a36Sopenharmony_ci	.flags		= CLOCK_SOURCE_IS_CONTINUOUS,
18162306a36Sopenharmony_ci};
18262306a36Sopenharmony_ci
18362306a36Sopenharmony_cistatic u64 notrace mxs_read_sched_clock_v2(void)
18462306a36Sopenharmony_ci{
18562306a36Sopenharmony_ci	return ~readl_relaxed(mxs_timrot_base + HW_TIMROT_RUNNING_COUNTn(1));
18662306a36Sopenharmony_ci}
18762306a36Sopenharmony_ci
18862306a36Sopenharmony_cistatic int __init mxs_clocksource_init(struct clk *timer_clk)
18962306a36Sopenharmony_ci{
19062306a36Sopenharmony_ci	unsigned int c = clk_get_rate(timer_clk);
19162306a36Sopenharmony_ci
19262306a36Sopenharmony_ci	if (timrot_is_v1())
19362306a36Sopenharmony_ci		clocksource_register_hz(&clocksource_mxs, c);
19462306a36Sopenharmony_ci	else {
19562306a36Sopenharmony_ci		clocksource_mmio_init(mxs_timrot_base + HW_TIMROT_RUNNING_COUNTn(1),
19662306a36Sopenharmony_ci			"mxs_timer", c, 200, 32, clocksource_mmio_readl_down);
19762306a36Sopenharmony_ci		sched_clock_register(mxs_read_sched_clock_v2, 32, c);
19862306a36Sopenharmony_ci	}
19962306a36Sopenharmony_ci
20062306a36Sopenharmony_ci	return 0;
20162306a36Sopenharmony_ci}
20262306a36Sopenharmony_ci
20362306a36Sopenharmony_cistatic int __init mxs_timer_init(struct device_node *np)
20462306a36Sopenharmony_ci{
20562306a36Sopenharmony_ci	struct clk *timer_clk;
20662306a36Sopenharmony_ci	int irq, ret;
20762306a36Sopenharmony_ci
20862306a36Sopenharmony_ci	mxs_timrot_base = of_iomap(np, 0);
20962306a36Sopenharmony_ci	WARN_ON(!mxs_timrot_base);
21062306a36Sopenharmony_ci
21162306a36Sopenharmony_ci	timer_clk = of_clk_get(np, 0);
21262306a36Sopenharmony_ci	if (IS_ERR(timer_clk)) {
21362306a36Sopenharmony_ci		pr_err("%s: failed to get clk\n", __func__);
21462306a36Sopenharmony_ci		return PTR_ERR(timer_clk);
21562306a36Sopenharmony_ci	}
21662306a36Sopenharmony_ci
21762306a36Sopenharmony_ci	ret = clk_prepare_enable(timer_clk);
21862306a36Sopenharmony_ci	if (ret)
21962306a36Sopenharmony_ci		return ret;
22062306a36Sopenharmony_ci
22162306a36Sopenharmony_ci	/*
22262306a36Sopenharmony_ci	 * Initialize timers to a known state
22362306a36Sopenharmony_ci	 */
22462306a36Sopenharmony_ci	stmp_reset_block(mxs_timrot_base + HW_TIMROT_ROTCTRL);
22562306a36Sopenharmony_ci
22662306a36Sopenharmony_ci	/* get timrot version */
22762306a36Sopenharmony_ci	timrot_major_version = __raw_readl(mxs_timrot_base +
22862306a36Sopenharmony_ci			(of_device_is_compatible(np, "fsl,imx23-timrot") ?
22962306a36Sopenharmony_ci						MX23_TIMROT_VERSION_OFFSET :
23062306a36Sopenharmony_ci						MX28_TIMROT_VERSION_OFFSET));
23162306a36Sopenharmony_ci	timrot_major_version >>= BP_TIMROT_MAJOR_VERSION;
23262306a36Sopenharmony_ci
23362306a36Sopenharmony_ci	/* one for clock_event */
23462306a36Sopenharmony_ci	__raw_writel((timrot_is_v1() ?
23562306a36Sopenharmony_ci			BV_TIMROTv1_TIMCTRLn_SELECT__32KHZ_XTAL :
23662306a36Sopenharmony_ci			BV_TIMROTv2_TIMCTRLn_SELECT__TICK_ALWAYS) |
23762306a36Sopenharmony_ci			BM_TIMROT_TIMCTRLn_UPDATE |
23862306a36Sopenharmony_ci			BM_TIMROT_TIMCTRLn_IRQ_EN,
23962306a36Sopenharmony_ci			mxs_timrot_base + HW_TIMROT_TIMCTRLn(0));
24062306a36Sopenharmony_ci
24162306a36Sopenharmony_ci	/* another for clocksource */
24262306a36Sopenharmony_ci	__raw_writel((timrot_is_v1() ?
24362306a36Sopenharmony_ci			BV_TIMROTv1_TIMCTRLn_SELECT__32KHZ_XTAL :
24462306a36Sopenharmony_ci			BV_TIMROTv2_TIMCTRLn_SELECT__TICK_ALWAYS) |
24562306a36Sopenharmony_ci			BM_TIMROT_TIMCTRLn_RELOAD,
24662306a36Sopenharmony_ci			mxs_timrot_base + HW_TIMROT_TIMCTRLn(1));
24762306a36Sopenharmony_ci
24862306a36Sopenharmony_ci	/* set clocksource timer fixed count to the maximum */
24962306a36Sopenharmony_ci	if (timrot_is_v1())
25062306a36Sopenharmony_ci		__raw_writel(0xffff,
25162306a36Sopenharmony_ci			mxs_timrot_base + HW_TIMROT_TIMCOUNTn(1));
25262306a36Sopenharmony_ci	else
25362306a36Sopenharmony_ci		__raw_writel(0xffffffff,
25462306a36Sopenharmony_ci			mxs_timrot_base + HW_TIMROT_FIXED_COUNTn(1));
25562306a36Sopenharmony_ci
25662306a36Sopenharmony_ci	/* init and register the timer to the framework */
25762306a36Sopenharmony_ci	ret = mxs_clocksource_init(timer_clk);
25862306a36Sopenharmony_ci	if (ret)
25962306a36Sopenharmony_ci		return ret;
26062306a36Sopenharmony_ci
26162306a36Sopenharmony_ci	ret = mxs_clockevent_init(timer_clk);
26262306a36Sopenharmony_ci	if (ret)
26362306a36Sopenharmony_ci		return ret;
26462306a36Sopenharmony_ci
26562306a36Sopenharmony_ci	/* Make irqs happen */
26662306a36Sopenharmony_ci	irq = irq_of_parse_and_map(np, 0);
26762306a36Sopenharmony_ci	if (irq <= 0)
26862306a36Sopenharmony_ci		return -EINVAL;
26962306a36Sopenharmony_ci
27062306a36Sopenharmony_ci	return request_irq(irq, mxs_timer_interrupt, IRQF_TIMER | IRQF_IRQPOLL,
27162306a36Sopenharmony_ci			   "MXS Timer Tick", &mxs_clockevent_device);
27262306a36Sopenharmony_ci}
27362306a36Sopenharmony_ciTIMER_OF_DECLARE(mxs, "fsl,timrot", mxs_timer_init);
274