162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-or-later
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * MMP PMU power island support
462306a36Sopenharmony_ci *
562306a36Sopenharmony_ci * Copyright (C) 2020 Lubomir Rintel <lkundrak@v3.sk>
662306a36Sopenharmony_ci */
762306a36Sopenharmony_ci
862306a36Sopenharmony_ci#include <linux/pm_domain.h>
962306a36Sopenharmony_ci#include <linux/slab.h>
1062306a36Sopenharmony_ci#include <linux/io.h>
1162306a36Sopenharmony_ci
1262306a36Sopenharmony_ci#include "clk.h"
1362306a36Sopenharmony_ci
1462306a36Sopenharmony_ci#define to_mmp_pm_domain(genpd) container_of(genpd, struct mmp_pm_domain, genpd)
1562306a36Sopenharmony_ci
1662306a36Sopenharmony_cistruct mmp_pm_domain {
1762306a36Sopenharmony_ci	struct generic_pm_domain genpd;
1862306a36Sopenharmony_ci	void __iomem *reg;
1962306a36Sopenharmony_ci	spinlock_t *lock;
2062306a36Sopenharmony_ci	u32 power_on;
2162306a36Sopenharmony_ci	u32 reset;
2262306a36Sopenharmony_ci	u32 clock_enable;
2362306a36Sopenharmony_ci	unsigned int flags;
2462306a36Sopenharmony_ci};
2562306a36Sopenharmony_ci
2662306a36Sopenharmony_cistatic int mmp_pm_domain_power_on(struct generic_pm_domain *genpd)
2762306a36Sopenharmony_ci{
2862306a36Sopenharmony_ci	struct mmp_pm_domain *pm_domain = to_mmp_pm_domain(genpd);
2962306a36Sopenharmony_ci	unsigned long flags = 0;
3062306a36Sopenharmony_ci	u32 val;
3162306a36Sopenharmony_ci
3262306a36Sopenharmony_ci	if (pm_domain->lock)
3362306a36Sopenharmony_ci		spin_lock_irqsave(pm_domain->lock, flags);
3462306a36Sopenharmony_ci
3562306a36Sopenharmony_ci	val = readl(pm_domain->reg);
3662306a36Sopenharmony_ci
3762306a36Sopenharmony_ci	/* Turn on the power island */
3862306a36Sopenharmony_ci	val |= pm_domain->power_on;
3962306a36Sopenharmony_ci	writel(val, pm_domain->reg);
4062306a36Sopenharmony_ci
4162306a36Sopenharmony_ci	/* Disable isolation */
4262306a36Sopenharmony_ci	val |= 0x100;
4362306a36Sopenharmony_ci	writel(val, pm_domain->reg);
4462306a36Sopenharmony_ci
4562306a36Sopenharmony_ci	/* Some blocks need to be reset after a power up */
4662306a36Sopenharmony_ci	if (pm_domain->reset || pm_domain->clock_enable) {
4762306a36Sopenharmony_ci		u32 after_power_on = val;
4862306a36Sopenharmony_ci
4962306a36Sopenharmony_ci		val &= ~pm_domain->reset;
5062306a36Sopenharmony_ci		writel(val, pm_domain->reg);
5162306a36Sopenharmony_ci
5262306a36Sopenharmony_ci		val |= pm_domain->clock_enable;
5362306a36Sopenharmony_ci		writel(val, pm_domain->reg);
5462306a36Sopenharmony_ci
5562306a36Sopenharmony_ci		val |= pm_domain->reset;
5662306a36Sopenharmony_ci		writel(val, pm_domain->reg);
5762306a36Sopenharmony_ci
5862306a36Sopenharmony_ci		writel(after_power_on, pm_domain->reg);
5962306a36Sopenharmony_ci	}
6062306a36Sopenharmony_ci
6162306a36Sopenharmony_ci	if (pm_domain->lock)
6262306a36Sopenharmony_ci		spin_unlock_irqrestore(pm_domain->lock, flags);
6362306a36Sopenharmony_ci
6462306a36Sopenharmony_ci	return 0;
6562306a36Sopenharmony_ci}
6662306a36Sopenharmony_ci
6762306a36Sopenharmony_cistatic int mmp_pm_domain_power_off(struct generic_pm_domain *genpd)
6862306a36Sopenharmony_ci{
6962306a36Sopenharmony_ci	struct mmp_pm_domain *pm_domain = to_mmp_pm_domain(genpd);
7062306a36Sopenharmony_ci	unsigned long flags = 0;
7162306a36Sopenharmony_ci	u32 val;
7262306a36Sopenharmony_ci
7362306a36Sopenharmony_ci	if (pm_domain->flags & MMP_PM_DOMAIN_NO_DISABLE)
7462306a36Sopenharmony_ci		return 0;
7562306a36Sopenharmony_ci
7662306a36Sopenharmony_ci	if (pm_domain->lock)
7762306a36Sopenharmony_ci		spin_lock_irqsave(pm_domain->lock, flags);
7862306a36Sopenharmony_ci
7962306a36Sopenharmony_ci	/* Turn off and isolate the power island. */
8062306a36Sopenharmony_ci	val = readl(pm_domain->reg);
8162306a36Sopenharmony_ci	val &= ~pm_domain->power_on;
8262306a36Sopenharmony_ci	val &= ~0x100;
8362306a36Sopenharmony_ci	writel(val, pm_domain->reg);
8462306a36Sopenharmony_ci
8562306a36Sopenharmony_ci	if (pm_domain->lock)
8662306a36Sopenharmony_ci		spin_unlock_irqrestore(pm_domain->lock, flags);
8762306a36Sopenharmony_ci
8862306a36Sopenharmony_ci	return 0;
8962306a36Sopenharmony_ci}
9062306a36Sopenharmony_ci
9162306a36Sopenharmony_cistruct generic_pm_domain *mmp_pm_domain_register(const char *name,
9262306a36Sopenharmony_ci		void __iomem *reg,
9362306a36Sopenharmony_ci		u32 power_on, u32 reset, u32 clock_enable,
9462306a36Sopenharmony_ci		unsigned int flags, spinlock_t *lock)
9562306a36Sopenharmony_ci{
9662306a36Sopenharmony_ci	struct mmp_pm_domain *pm_domain;
9762306a36Sopenharmony_ci
9862306a36Sopenharmony_ci	pm_domain = kzalloc(sizeof(*pm_domain), GFP_KERNEL);
9962306a36Sopenharmony_ci	if (!pm_domain)
10062306a36Sopenharmony_ci		return ERR_PTR(-ENOMEM);
10162306a36Sopenharmony_ci
10262306a36Sopenharmony_ci	pm_domain->reg = reg;
10362306a36Sopenharmony_ci	pm_domain->power_on = power_on;
10462306a36Sopenharmony_ci	pm_domain->reset = reset;
10562306a36Sopenharmony_ci	pm_domain->clock_enable = clock_enable;
10662306a36Sopenharmony_ci	pm_domain->flags = flags;
10762306a36Sopenharmony_ci	pm_domain->lock = lock;
10862306a36Sopenharmony_ci
10962306a36Sopenharmony_ci	pm_genpd_init(&pm_domain->genpd, NULL, true);
11062306a36Sopenharmony_ci	pm_domain->genpd.name = name;
11162306a36Sopenharmony_ci	pm_domain->genpd.power_on = mmp_pm_domain_power_on;
11262306a36Sopenharmony_ci	pm_domain->genpd.power_off = mmp_pm_domain_power_off;
11362306a36Sopenharmony_ci
11462306a36Sopenharmony_ci	return &pm_domain->genpd;
11562306a36Sopenharmony_ci}
116