162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * MStar timer driver
462306a36Sopenharmony_ci *
562306a36Sopenharmony_ci * Copyright (C) 2021 Daniel Palmer
662306a36Sopenharmony_ci * Copyright (C) 2021 Romain Perier
762306a36Sopenharmony_ci *
862306a36Sopenharmony_ci */
962306a36Sopenharmony_ci
1062306a36Sopenharmony_ci#include <linux/clk.h>
1162306a36Sopenharmony_ci#include <linux/clockchips.h>
1262306a36Sopenharmony_ci#include <linux/interrupt.h>
1362306a36Sopenharmony_ci#include <linux/irq.h>
1462306a36Sopenharmony_ci#include <linux/irqreturn.h>
1562306a36Sopenharmony_ci#include <linux/sched_clock.h>
1662306a36Sopenharmony_ci#include <linux/of.h>
1762306a36Sopenharmony_ci#include <linux/of_address.h>
1862306a36Sopenharmony_ci#include <linux/of_irq.h>
1962306a36Sopenharmony_ci
2062306a36Sopenharmony_ci#ifdef CONFIG_ARM
2162306a36Sopenharmony_ci#include <linux/delay.h>
2262306a36Sopenharmony_ci#endif
2362306a36Sopenharmony_ci
2462306a36Sopenharmony_ci#include "timer-of.h"
2562306a36Sopenharmony_ci
2662306a36Sopenharmony_ci#define TIMER_NAME "msc313e_timer"
2762306a36Sopenharmony_ci
2862306a36Sopenharmony_ci#define MSC313E_REG_CTRL		0x00
2962306a36Sopenharmony_ci#define MSC313E_REG_CTRL_TIMER_EN	BIT(0)
3062306a36Sopenharmony_ci#define MSC313E_REG_CTRL_TIMER_TRIG	BIT(1)
3162306a36Sopenharmony_ci#define MSC313E_REG_CTRL_TIMER_INT_EN	BIT(8)
3262306a36Sopenharmony_ci#define MSC313E_REG_TIMER_MAX_LOW	0x08
3362306a36Sopenharmony_ci#define MSC313E_REG_TIMER_MAX_HIGH	0x0c
3462306a36Sopenharmony_ci#define MSC313E_REG_COUNTER_LOW		0x10
3562306a36Sopenharmony_ci#define MSC313E_REG_COUNTER_HIGH	0x14
3662306a36Sopenharmony_ci#define MSC313E_REG_TIMER_DIVIDE	0x18
3762306a36Sopenharmony_ci
3862306a36Sopenharmony_ci#define MSC313E_CLK_DIVIDER		9
3962306a36Sopenharmony_ci#define TIMER_SYNC_TICKS		3
4062306a36Sopenharmony_ci
4162306a36Sopenharmony_ci#ifdef CONFIG_ARM
4262306a36Sopenharmony_cistruct msc313e_delay {
4362306a36Sopenharmony_ci	void __iomem *base;
4462306a36Sopenharmony_ci	struct delay_timer delay;
4562306a36Sopenharmony_ci};
4662306a36Sopenharmony_cistatic struct msc313e_delay msc313e_delay;
4762306a36Sopenharmony_ci#endif
4862306a36Sopenharmony_ci
4962306a36Sopenharmony_cistatic void __iomem *msc313e_clksrc;
5062306a36Sopenharmony_ci
5162306a36Sopenharmony_cistatic void msc313e_timer_stop(void __iomem *base)
5262306a36Sopenharmony_ci{
5362306a36Sopenharmony_ci	writew(0, base + MSC313E_REG_CTRL);
5462306a36Sopenharmony_ci}
5562306a36Sopenharmony_ci
5662306a36Sopenharmony_cistatic void msc313e_timer_start(void __iomem *base, bool periodic)
5762306a36Sopenharmony_ci{
5862306a36Sopenharmony_ci	u16 reg;
5962306a36Sopenharmony_ci
6062306a36Sopenharmony_ci	reg = readw(base + MSC313E_REG_CTRL);
6162306a36Sopenharmony_ci	if (periodic)
6262306a36Sopenharmony_ci		reg |= MSC313E_REG_CTRL_TIMER_EN;
6362306a36Sopenharmony_ci	else
6462306a36Sopenharmony_ci		reg |= MSC313E_REG_CTRL_TIMER_TRIG;
6562306a36Sopenharmony_ci	writew(reg | MSC313E_REG_CTRL_TIMER_INT_EN, base + MSC313E_REG_CTRL);
6662306a36Sopenharmony_ci}
6762306a36Sopenharmony_ci
6862306a36Sopenharmony_cistatic void msc313e_timer_setup(void __iomem *base, unsigned long delay)
6962306a36Sopenharmony_ci{
7062306a36Sopenharmony_ci	unsigned long flags;
7162306a36Sopenharmony_ci
7262306a36Sopenharmony_ci	local_irq_save(flags);
7362306a36Sopenharmony_ci	writew(delay >> 16, base + MSC313E_REG_TIMER_MAX_HIGH);
7462306a36Sopenharmony_ci	writew(delay & 0xffff, base + MSC313E_REG_TIMER_MAX_LOW);
7562306a36Sopenharmony_ci	local_irq_restore(flags);
7662306a36Sopenharmony_ci}
7762306a36Sopenharmony_ci
7862306a36Sopenharmony_cistatic unsigned long msc313e_timer_current_value(void __iomem *base)
7962306a36Sopenharmony_ci{
8062306a36Sopenharmony_ci	unsigned long flags;
8162306a36Sopenharmony_ci	u16 l, h;
8262306a36Sopenharmony_ci
8362306a36Sopenharmony_ci	local_irq_save(flags);
8462306a36Sopenharmony_ci	l = readw(base + MSC313E_REG_COUNTER_LOW);
8562306a36Sopenharmony_ci	h = readw(base + MSC313E_REG_COUNTER_HIGH);
8662306a36Sopenharmony_ci	local_irq_restore(flags);
8762306a36Sopenharmony_ci
8862306a36Sopenharmony_ci	return (((u32)h) << 16 | l);
8962306a36Sopenharmony_ci}
9062306a36Sopenharmony_ci
9162306a36Sopenharmony_cistatic int msc313e_timer_clkevt_shutdown(struct clock_event_device *evt)
9262306a36Sopenharmony_ci{
9362306a36Sopenharmony_ci	struct timer_of *timer = to_timer_of(evt);
9462306a36Sopenharmony_ci
9562306a36Sopenharmony_ci	msc313e_timer_stop(timer_of_base(timer));
9662306a36Sopenharmony_ci
9762306a36Sopenharmony_ci	return 0;
9862306a36Sopenharmony_ci}
9962306a36Sopenharmony_ci
10062306a36Sopenharmony_cistatic int msc313e_timer_clkevt_set_oneshot(struct clock_event_device *evt)
10162306a36Sopenharmony_ci{
10262306a36Sopenharmony_ci	struct timer_of *timer = to_timer_of(evt);
10362306a36Sopenharmony_ci
10462306a36Sopenharmony_ci	msc313e_timer_stop(timer_of_base(timer));
10562306a36Sopenharmony_ci	msc313e_timer_start(timer_of_base(timer), false);
10662306a36Sopenharmony_ci
10762306a36Sopenharmony_ci	return 0;
10862306a36Sopenharmony_ci}
10962306a36Sopenharmony_ci
11062306a36Sopenharmony_cistatic int msc313e_timer_clkevt_set_periodic(struct clock_event_device *evt)
11162306a36Sopenharmony_ci{
11262306a36Sopenharmony_ci	struct timer_of *timer = to_timer_of(evt);
11362306a36Sopenharmony_ci
11462306a36Sopenharmony_ci	msc313e_timer_stop(timer_of_base(timer));
11562306a36Sopenharmony_ci	msc313e_timer_setup(timer_of_base(timer), timer_of_period(timer));
11662306a36Sopenharmony_ci	msc313e_timer_start(timer_of_base(timer), true);
11762306a36Sopenharmony_ci
11862306a36Sopenharmony_ci	return 0;
11962306a36Sopenharmony_ci}
12062306a36Sopenharmony_ci
12162306a36Sopenharmony_cistatic int msc313e_timer_clkevt_next_event(unsigned long evt, struct clock_event_device *clkevt)
12262306a36Sopenharmony_ci{
12362306a36Sopenharmony_ci	struct timer_of *timer = to_timer_of(clkevt);
12462306a36Sopenharmony_ci
12562306a36Sopenharmony_ci	msc313e_timer_stop(timer_of_base(timer));
12662306a36Sopenharmony_ci	msc313e_timer_setup(timer_of_base(timer), evt);
12762306a36Sopenharmony_ci	msc313e_timer_start(timer_of_base(timer), false);
12862306a36Sopenharmony_ci
12962306a36Sopenharmony_ci	return 0;
13062306a36Sopenharmony_ci}
13162306a36Sopenharmony_ci
13262306a36Sopenharmony_cistatic irqreturn_t msc313e_timer_clkevt_irq(int irq, void *dev_id)
13362306a36Sopenharmony_ci{
13462306a36Sopenharmony_ci	struct clock_event_device *evt = dev_id;
13562306a36Sopenharmony_ci
13662306a36Sopenharmony_ci	evt->event_handler(evt);
13762306a36Sopenharmony_ci
13862306a36Sopenharmony_ci	return IRQ_HANDLED;
13962306a36Sopenharmony_ci}
14062306a36Sopenharmony_ci
14162306a36Sopenharmony_cistatic u64 msc313e_timer_clksrc_read(struct clocksource *cs)
14262306a36Sopenharmony_ci{
14362306a36Sopenharmony_ci	return msc313e_timer_current_value(msc313e_clksrc) & cs->mask;
14462306a36Sopenharmony_ci}
14562306a36Sopenharmony_ci
14662306a36Sopenharmony_ci#ifdef CONFIG_ARM
14762306a36Sopenharmony_cistatic unsigned long msc313e_read_delay_timer_read(void)
14862306a36Sopenharmony_ci{
14962306a36Sopenharmony_ci	return msc313e_timer_current_value(msc313e_delay.base);
15062306a36Sopenharmony_ci}
15162306a36Sopenharmony_ci#endif
15262306a36Sopenharmony_ci
15362306a36Sopenharmony_cistatic u64 msc313e_timer_sched_clock_read(void)
15462306a36Sopenharmony_ci{
15562306a36Sopenharmony_ci	return msc313e_timer_current_value(msc313e_clksrc);
15662306a36Sopenharmony_ci}
15762306a36Sopenharmony_ci
15862306a36Sopenharmony_cistatic struct clock_event_device msc313e_clkevt = {
15962306a36Sopenharmony_ci	.name = TIMER_NAME,
16062306a36Sopenharmony_ci	.rating = 300,
16162306a36Sopenharmony_ci	.features = CLOCK_EVT_FEAT_PERIODIC | CLOCK_EVT_FEAT_ONESHOT,
16262306a36Sopenharmony_ci	.set_state_shutdown = msc313e_timer_clkevt_shutdown,
16362306a36Sopenharmony_ci	.set_state_periodic = msc313e_timer_clkevt_set_periodic,
16462306a36Sopenharmony_ci	.set_state_oneshot = msc313e_timer_clkevt_set_oneshot,
16562306a36Sopenharmony_ci	.tick_resume = msc313e_timer_clkevt_shutdown,
16662306a36Sopenharmony_ci	.set_next_event = msc313e_timer_clkevt_next_event,
16762306a36Sopenharmony_ci};
16862306a36Sopenharmony_ci
16962306a36Sopenharmony_cistatic int __init msc313e_clkevt_init(struct device_node *np)
17062306a36Sopenharmony_ci{
17162306a36Sopenharmony_ci	int ret;
17262306a36Sopenharmony_ci	struct timer_of *to;
17362306a36Sopenharmony_ci
17462306a36Sopenharmony_ci	to = kzalloc(sizeof(*to), GFP_KERNEL);
17562306a36Sopenharmony_ci	if (!to)
17662306a36Sopenharmony_ci		return -ENOMEM;
17762306a36Sopenharmony_ci
17862306a36Sopenharmony_ci	to->flags = TIMER_OF_IRQ | TIMER_OF_CLOCK | TIMER_OF_BASE;
17962306a36Sopenharmony_ci	to->of_irq.handler = msc313e_timer_clkevt_irq;
18062306a36Sopenharmony_ci	ret = timer_of_init(np, to);
18162306a36Sopenharmony_ci	if (ret)
18262306a36Sopenharmony_ci		return ret;
18362306a36Sopenharmony_ci
18462306a36Sopenharmony_ci	if (of_device_is_compatible(np, "sstar,ssd20xd-timer")) {
18562306a36Sopenharmony_ci		to->of_clk.rate = clk_get_rate(to->of_clk.clk) / MSC313E_CLK_DIVIDER;
18662306a36Sopenharmony_ci		to->of_clk.period = DIV_ROUND_UP(to->of_clk.rate, HZ);
18762306a36Sopenharmony_ci		writew(MSC313E_CLK_DIVIDER - 1, timer_of_base(to) + MSC313E_REG_TIMER_DIVIDE);
18862306a36Sopenharmony_ci	}
18962306a36Sopenharmony_ci
19062306a36Sopenharmony_ci	msc313e_clkevt.cpumask = cpu_possible_mask;
19162306a36Sopenharmony_ci	msc313e_clkevt.irq = to->of_irq.irq;
19262306a36Sopenharmony_ci	to->clkevt = msc313e_clkevt;
19362306a36Sopenharmony_ci
19462306a36Sopenharmony_ci	clockevents_config_and_register(&to->clkevt, timer_of_rate(to),
19562306a36Sopenharmony_ci					TIMER_SYNC_TICKS, 0xffffffff);
19662306a36Sopenharmony_ci	return 0;
19762306a36Sopenharmony_ci}
19862306a36Sopenharmony_ci
19962306a36Sopenharmony_cistatic int __init msc313e_clksrc_init(struct device_node *np)
20062306a36Sopenharmony_ci{
20162306a36Sopenharmony_ci	struct timer_of to = { 0 };
20262306a36Sopenharmony_ci	int ret;
20362306a36Sopenharmony_ci	u16 reg;
20462306a36Sopenharmony_ci
20562306a36Sopenharmony_ci	to.flags = TIMER_OF_BASE | TIMER_OF_CLOCK;
20662306a36Sopenharmony_ci	ret = timer_of_init(np, &to);
20762306a36Sopenharmony_ci	if (ret)
20862306a36Sopenharmony_ci		return ret;
20962306a36Sopenharmony_ci
21062306a36Sopenharmony_ci	msc313e_clksrc = timer_of_base(&to);
21162306a36Sopenharmony_ci	reg = readw(msc313e_clksrc + MSC313E_REG_CTRL);
21262306a36Sopenharmony_ci	reg |= MSC313E_REG_CTRL_TIMER_EN;
21362306a36Sopenharmony_ci	writew(reg, msc313e_clksrc + MSC313E_REG_CTRL);
21462306a36Sopenharmony_ci
21562306a36Sopenharmony_ci#ifdef CONFIG_ARM
21662306a36Sopenharmony_ci	msc313e_delay.base = timer_of_base(&to);
21762306a36Sopenharmony_ci	msc313e_delay.delay.read_current_timer = msc313e_read_delay_timer_read;
21862306a36Sopenharmony_ci	msc313e_delay.delay.freq = timer_of_rate(&to);
21962306a36Sopenharmony_ci
22062306a36Sopenharmony_ci	register_current_timer_delay(&msc313e_delay.delay);
22162306a36Sopenharmony_ci#endif
22262306a36Sopenharmony_ci
22362306a36Sopenharmony_ci	sched_clock_register(msc313e_timer_sched_clock_read, 32, timer_of_rate(&to));
22462306a36Sopenharmony_ci	return clocksource_mmio_init(timer_of_base(&to), TIMER_NAME, timer_of_rate(&to), 300, 32,
22562306a36Sopenharmony_ci				     msc313e_timer_clksrc_read);
22662306a36Sopenharmony_ci}
22762306a36Sopenharmony_ci
22862306a36Sopenharmony_cistatic int __init msc313e_timer_init(struct device_node *np)
22962306a36Sopenharmony_ci{
23062306a36Sopenharmony_ci	int ret = 0;
23162306a36Sopenharmony_ci	static int num_called;
23262306a36Sopenharmony_ci
23362306a36Sopenharmony_ci	switch (num_called) {
23462306a36Sopenharmony_ci	case 0:
23562306a36Sopenharmony_ci		ret = msc313e_clksrc_init(np);
23662306a36Sopenharmony_ci		if (ret)
23762306a36Sopenharmony_ci			return ret;
23862306a36Sopenharmony_ci		break;
23962306a36Sopenharmony_ci
24062306a36Sopenharmony_ci	default:
24162306a36Sopenharmony_ci		ret = msc313e_clkevt_init(np);
24262306a36Sopenharmony_ci		if (ret)
24362306a36Sopenharmony_ci			return ret;
24462306a36Sopenharmony_ci		break;
24562306a36Sopenharmony_ci	}
24662306a36Sopenharmony_ci
24762306a36Sopenharmony_ci	num_called++;
24862306a36Sopenharmony_ci
24962306a36Sopenharmony_ci	return 0;
25062306a36Sopenharmony_ci}
25162306a36Sopenharmony_ci
25262306a36Sopenharmony_ciTIMER_OF_DECLARE(msc313, "mstar,msc313e-timer", msc313e_timer_init);
25362306a36Sopenharmony_ciTIMER_OF_DECLARE(ssd20xd, "sstar,ssd20xd-timer", msc313e_timer_init);
254