162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Copyright 2019 NXP.
462306a36Sopenharmony_ci */
562306a36Sopenharmony_ci
662306a36Sopenharmony_ci#include <linux/delay.h>
762306a36Sopenharmony_ci#include <linux/dma-mapping.h>
862306a36Sopenharmony_ci#include <linux/interrupt.h>
962306a36Sopenharmony_ci#include <linux/platform_device.h>
1062306a36Sopenharmony_ci#include <linux/slab.h>
1162306a36Sopenharmony_ci
1262306a36Sopenharmony_ci#include "dcss-dev.h"
1362306a36Sopenharmony_ci
1462306a36Sopenharmony_ci#define DCSS_CTXLD_CONTROL_STATUS	0x0
1562306a36Sopenharmony_ci#define   CTXLD_ENABLE			BIT(0)
1662306a36Sopenharmony_ci#define   ARB_SEL			BIT(1)
1762306a36Sopenharmony_ci#define   RD_ERR_EN			BIT(2)
1862306a36Sopenharmony_ci#define   DB_COMP_EN			BIT(3)
1962306a36Sopenharmony_ci#define   SB_HP_COMP_EN			BIT(4)
2062306a36Sopenharmony_ci#define   SB_LP_COMP_EN			BIT(5)
2162306a36Sopenharmony_ci#define   DB_PEND_SB_REC_EN		BIT(6)
2262306a36Sopenharmony_ci#define   SB_PEND_DISP_ACTIVE_EN	BIT(7)
2362306a36Sopenharmony_ci#define   AHB_ERR_EN			BIT(8)
2462306a36Sopenharmony_ci#define   RD_ERR			BIT(16)
2562306a36Sopenharmony_ci#define   DB_COMP			BIT(17)
2662306a36Sopenharmony_ci#define   SB_HP_COMP			BIT(18)
2762306a36Sopenharmony_ci#define   SB_LP_COMP			BIT(19)
2862306a36Sopenharmony_ci#define   DB_PEND_SB_REC		BIT(20)
2962306a36Sopenharmony_ci#define   SB_PEND_DISP_ACTIVE		BIT(21)
3062306a36Sopenharmony_ci#define   AHB_ERR			BIT(22)
3162306a36Sopenharmony_ci#define DCSS_CTXLD_DB_BASE_ADDR		0x10
3262306a36Sopenharmony_ci#define DCSS_CTXLD_DB_COUNT		0x14
3362306a36Sopenharmony_ci#define DCSS_CTXLD_SB_BASE_ADDR		0x18
3462306a36Sopenharmony_ci#define DCSS_CTXLD_SB_COUNT		0x1C
3562306a36Sopenharmony_ci#define   SB_HP_COUNT_POS		0
3662306a36Sopenharmony_ci#define   SB_HP_COUNT_MASK		0xffff
3762306a36Sopenharmony_ci#define   SB_LP_COUNT_POS		16
3862306a36Sopenharmony_ci#define   SB_LP_COUNT_MASK		0xffff0000
3962306a36Sopenharmony_ci#define DCSS_AHB_ERR_ADDR		0x20
4062306a36Sopenharmony_ci
4162306a36Sopenharmony_ci#define CTXLD_IRQ_COMPLETION		(DB_COMP | SB_HP_COMP | SB_LP_COMP)
4262306a36Sopenharmony_ci#define CTXLD_IRQ_ERROR			(RD_ERR | DB_PEND_SB_REC | AHB_ERR)
4362306a36Sopenharmony_ci
4462306a36Sopenharmony_ci/* The following sizes are in context loader entries, 8 bytes each. */
4562306a36Sopenharmony_ci#define CTXLD_DB_CTX_ENTRIES		1024	/* max 65536 */
4662306a36Sopenharmony_ci#define CTXLD_SB_LP_CTX_ENTRIES		10240	/* max 65536 */
4762306a36Sopenharmony_ci#define CTXLD_SB_HP_CTX_ENTRIES		20000	/* max 65536 */
4862306a36Sopenharmony_ci#define CTXLD_SB_CTX_ENTRIES		(CTXLD_SB_LP_CTX_ENTRIES + \
4962306a36Sopenharmony_ci					 CTXLD_SB_HP_CTX_ENTRIES)
5062306a36Sopenharmony_ci
5162306a36Sopenharmony_ci/* Sizes, in entries, of the DB, SB_HP and SB_LP context regions. */
5262306a36Sopenharmony_cistatic u16 dcss_ctxld_ctx_size[3] = {
5362306a36Sopenharmony_ci	CTXLD_DB_CTX_ENTRIES,
5462306a36Sopenharmony_ci	CTXLD_SB_HP_CTX_ENTRIES,
5562306a36Sopenharmony_ci	CTXLD_SB_LP_CTX_ENTRIES
5662306a36Sopenharmony_ci};
5762306a36Sopenharmony_ci
5862306a36Sopenharmony_ci/* this represents an entry in the context loader map */
5962306a36Sopenharmony_cistruct dcss_ctxld_item {
6062306a36Sopenharmony_ci	u32 val;
6162306a36Sopenharmony_ci	u32 ofs;
6262306a36Sopenharmony_ci};
6362306a36Sopenharmony_ci
6462306a36Sopenharmony_ci#define CTX_ITEM_SIZE			sizeof(struct dcss_ctxld_item)
6562306a36Sopenharmony_ci
6662306a36Sopenharmony_cistruct dcss_ctxld {
6762306a36Sopenharmony_ci	struct device *dev;
6862306a36Sopenharmony_ci	void __iomem *ctxld_reg;
6962306a36Sopenharmony_ci	int irq;
7062306a36Sopenharmony_ci	bool irq_en;
7162306a36Sopenharmony_ci
7262306a36Sopenharmony_ci	struct dcss_ctxld_item *db[2];
7362306a36Sopenharmony_ci	struct dcss_ctxld_item *sb_hp[2];
7462306a36Sopenharmony_ci	struct dcss_ctxld_item *sb_lp[2];
7562306a36Sopenharmony_ci
7662306a36Sopenharmony_ci	dma_addr_t db_paddr[2];
7762306a36Sopenharmony_ci	dma_addr_t sb_paddr[2];
7862306a36Sopenharmony_ci
7962306a36Sopenharmony_ci	u16 ctx_size[2][3]; /* holds the sizes of DB, SB_HP and SB_LP ctx */
8062306a36Sopenharmony_ci	u8 current_ctx;
8162306a36Sopenharmony_ci
8262306a36Sopenharmony_ci	bool in_use;
8362306a36Sopenharmony_ci	bool armed;
8462306a36Sopenharmony_ci
8562306a36Sopenharmony_ci	spinlock_t lock; /* protects concurent access to private data */
8662306a36Sopenharmony_ci};
8762306a36Sopenharmony_ci
8862306a36Sopenharmony_cistatic irqreturn_t dcss_ctxld_irq_handler(int irq, void *data)
8962306a36Sopenharmony_ci{
9062306a36Sopenharmony_ci	struct dcss_ctxld *ctxld = data;
9162306a36Sopenharmony_ci	struct dcss_dev *dcss = dcss_drv_dev_to_dcss(ctxld->dev);
9262306a36Sopenharmony_ci	u32 irq_status;
9362306a36Sopenharmony_ci
9462306a36Sopenharmony_ci	irq_status = dcss_readl(ctxld->ctxld_reg + DCSS_CTXLD_CONTROL_STATUS);
9562306a36Sopenharmony_ci
9662306a36Sopenharmony_ci	if (irq_status & CTXLD_IRQ_COMPLETION &&
9762306a36Sopenharmony_ci	    !(irq_status & CTXLD_ENABLE) && ctxld->in_use) {
9862306a36Sopenharmony_ci		ctxld->in_use = false;
9962306a36Sopenharmony_ci
10062306a36Sopenharmony_ci		if (dcss && dcss->disable_callback)
10162306a36Sopenharmony_ci			dcss->disable_callback(dcss);
10262306a36Sopenharmony_ci	} else if (irq_status & CTXLD_IRQ_ERROR) {
10362306a36Sopenharmony_ci		/*
10462306a36Sopenharmony_ci		 * Except for throwing an error message and clearing the status
10562306a36Sopenharmony_ci		 * register, there's not much we can do here.
10662306a36Sopenharmony_ci		 */
10762306a36Sopenharmony_ci		dev_err(ctxld->dev, "ctxld: error encountered: %08x\n",
10862306a36Sopenharmony_ci			irq_status);
10962306a36Sopenharmony_ci		dev_err(ctxld->dev, "ctxld: db=%d, sb_hp=%d, sb_lp=%d\n",
11062306a36Sopenharmony_ci			ctxld->ctx_size[ctxld->current_ctx ^ 1][CTX_DB],
11162306a36Sopenharmony_ci			ctxld->ctx_size[ctxld->current_ctx ^ 1][CTX_SB_HP],
11262306a36Sopenharmony_ci			ctxld->ctx_size[ctxld->current_ctx ^ 1][CTX_SB_LP]);
11362306a36Sopenharmony_ci	}
11462306a36Sopenharmony_ci
11562306a36Sopenharmony_ci	dcss_clr(irq_status & (CTXLD_IRQ_ERROR | CTXLD_IRQ_COMPLETION),
11662306a36Sopenharmony_ci		 ctxld->ctxld_reg + DCSS_CTXLD_CONTROL_STATUS);
11762306a36Sopenharmony_ci
11862306a36Sopenharmony_ci	return IRQ_HANDLED;
11962306a36Sopenharmony_ci}
12062306a36Sopenharmony_ci
12162306a36Sopenharmony_cistatic int dcss_ctxld_irq_config(struct dcss_ctxld *ctxld,
12262306a36Sopenharmony_ci				 struct platform_device *pdev)
12362306a36Sopenharmony_ci{
12462306a36Sopenharmony_ci	int ret;
12562306a36Sopenharmony_ci
12662306a36Sopenharmony_ci	ctxld->irq = platform_get_irq_byname(pdev, "ctxld");
12762306a36Sopenharmony_ci	if (ctxld->irq < 0)
12862306a36Sopenharmony_ci		return ctxld->irq;
12962306a36Sopenharmony_ci
13062306a36Sopenharmony_ci	ret = request_irq(ctxld->irq, dcss_ctxld_irq_handler,
13162306a36Sopenharmony_ci			  0, "dcss_ctxld", ctxld);
13262306a36Sopenharmony_ci	if (ret) {
13362306a36Sopenharmony_ci		dev_err(ctxld->dev, "ctxld: irq request failed.\n");
13462306a36Sopenharmony_ci		return ret;
13562306a36Sopenharmony_ci	}
13662306a36Sopenharmony_ci
13762306a36Sopenharmony_ci	ctxld->irq_en = true;
13862306a36Sopenharmony_ci
13962306a36Sopenharmony_ci	return 0;
14062306a36Sopenharmony_ci}
14162306a36Sopenharmony_ci
14262306a36Sopenharmony_cistatic void dcss_ctxld_hw_cfg(struct dcss_ctxld *ctxld)
14362306a36Sopenharmony_ci{
14462306a36Sopenharmony_ci	dcss_writel(RD_ERR_EN | SB_HP_COMP_EN |
14562306a36Sopenharmony_ci		    DB_PEND_SB_REC_EN | AHB_ERR_EN | RD_ERR | AHB_ERR,
14662306a36Sopenharmony_ci		    ctxld->ctxld_reg + DCSS_CTXLD_CONTROL_STATUS);
14762306a36Sopenharmony_ci}
14862306a36Sopenharmony_ci
14962306a36Sopenharmony_cistatic void dcss_ctxld_free_ctx(struct dcss_ctxld *ctxld)
15062306a36Sopenharmony_ci{
15162306a36Sopenharmony_ci	struct dcss_ctxld_item *ctx;
15262306a36Sopenharmony_ci	int i;
15362306a36Sopenharmony_ci
15462306a36Sopenharmony_ci	for (i = 0; i < 2; i++) {
15562306a36Sopenharmony_ci		if (ctxld->db[i]) {
15662306a36Sopenharmony_ci			dma_free_coherent(ctxld->dev,
15762306a36Sopenharmony_ci					  CTXLD_DB_CTX_ENTRIES * sizeof(*ctx),
15862306a36Sopenharmony_ci					  ctxld->db[i], ctxld->db_paddr[i]);
15962306a36Sopenharmony_ci			ctxld->db[i] = NULL;
16062306a36Sopenharmony_ci			ctxld->db_paddr[i] = 0;
16162306a36Sopenharmony_ci		}
16262306a36Sopenharmony_ci
16362306a36Sopenharmony_ci		if (ctxld->sb_hp[i]) {
16462306a36Sopenharmony_ci			dma_free_coherent(ctxld->dev,
16562306a36Sopenharmony_ci					  CTXLD_SB_CTX_ENTRIES * sizeof(*ctx),
16662306a36Sopenharmony_ci					  ctxld->sb_hp[i], ctxld->sb_paddr[i]);
16762306a36Sopenharmony_ci			ctxld->sb_hp[i] = NULL;
16862306a36Sopenharmony_ci			ctxld->sb_paddr[i] = 0;
16962306a36Sopenharmony_ci		}
17062306a36Sopenharmony_ci	}
17162306a36Sopenharmony_ci}
17262306a36Sopenharmony_ci
17362306a36Sopenharmony_cistatic int dcss_ctxld_alloc_ctx(struct dcss_ctxld *ctxld)
17462306a36Sopenharmony_ci{
17562306a36Sopenharmony_ci	struct dcss_ctxld_item *ctx;
17662306a36Sopenharmony_ci	int i;
17762306a36Sopenharmony_ci
17862306a36Sopenharmony_ci	for (i = 0; i < 2; i++) {
17962306a36Sopenharmony_ci		ctx = dma_alloc_coherent(ctxld->dev,
18062306a36Sopenharmony_ci					 CTXLD_DB_CTX_ENTRIES * sizeof(*ctx),
18162306a36Sopenharmony_ci					 &ctxld->db_paddr[i], GFP_KERNEL);
18262306a36Sopenharmony_ci		if (!ctx)
18362306a36Sopenharmony_ci			return -ENOMEM;
18462306a36Sopenharmony_ci
18562306a36Sopenharmony_ci		ctxld->db[i] = ctx;
18662306a36Sopenharmony_ci
18762306a36Sopenharmony_ci		ctx = dma_alloc_coherent(ctxld->dev,
18862306a36Sopenharmony_ci					 CTXLD_SB_CTX_ENTRIES * sizeof(*ctx),
18962306a36Sopenharmony_ci					 &ctxld->sb_paddr[i], GFP_KERNEL);
19062306a36Sopenharmony_ci		if (!ctx)
19162306a36Sopenharmony_ci			return -ENOMEM;
19262306a36Sopenharmony_ci
19362306a36Sopenharmony_ci		ctxld->sb_hp[i] = ctx;
19462306a36Sopenharmony_ci		ctxld->sb_lp[i] = ctx + CTXLD_SB_HP_CTX_ENTRIES;
19562306a36Sopenharmony_ci	}
19662306a36Sopenharmony_ci
19762306a36Sopenharmony_ci	return 0;
19862306a36Sopenharmony_ci}
19962306a36Sopenharmony_ci
20062306a36Sopenharmony_ciint dcss_ctxld_init(struct dcss_dev *dcss, unsigned long ctxld_base)
20162306a36Sopenharmony_ci{
20262306a36Sopenharmony_ci	struct dcss_ctxld *ctxld;
20362306a36Sopenharmony_ci	int ret;
20462306a36Sopenharmony_ci
20562306a36Sopenharmony_ci	ctxld = kzalloc(sizeof(*ctxld), GFP_KERNEL);
20662306a36Sopenharmony_ci	if (!ctxld)
20762306a36Sopenharmony_ci		return -ENOMEM;
20862306a36Sopenharmony_ci
20962306a36Sopenharmony_ci	dcss->ctxld = ctxld;
21062306a36Sopenharmony_ci	ctxld->dev = dcss->dev;
21162306a36Sopenharmony_ci
21262306a36Sopenharmony_ci	spin_lock_init(&ctxld->lock);
21362306a36Sopenharmony_ci
21462306a36Sopenharmony_ci	ret = dcss_ctxld_alloc_ctx(ctxld);
21562306a36Sopenharmony_ci	if (ret) {
21662306a36Sopenharmony_ci		dev_err(dcss->dev, "ctxld: cannot allocate context memory.\n");
21762306a36Sopenharmony_ci		goto err;
21862306a36Sopenharmony_ci	}
21962306a36Sopenharmony_ci
22062306a36Sopenharmony_ci	ctxld->ctxld_reg = ioremap(ctxld_base, SZ_4K);
22162306a36Sopenharmony_ci	if (!ctxld->ctxld_reg) {
22262306a36Sopenharmony_ci		dev_err(dcss->dev, "ctxld: unable to remap ctxld base\n");
22362306a36Sopenharmony_ci		ret = -ENOMEM;
22462306a36Sopenharmony_ci		goto err;
22562306a36Sopenharmony_ci	}
22662306a36Sopenharmony_ci
22762306a36Sopenharmony_ci	ret = dcss_ctxld_irq_config(ctxld, to_platform_device(dcss->dev));
22862306a36Sopenharmony_ci	if (ret)
22962306a36Sopenharmony_ci		goto err_irq;
23062306a36Sopenharmony_ci
23162306a36Sopenharmony_ci	dcss_ctxld_hw_cfg(ctxld);
23262306a36Sopenharmony_ci
23362306a36Sopenharmony_ci	return 0;
23462306a36Sopenharmony_ci
23562306a36Sopenharmony_cierr_irq:
23662306a36Sopenharmony_ci	iounmap(ctxld->ctxld_reg);
23762306a36Sopenharmony_ci
23862306a36Sopenharmony_cierr:
23962306a36Sopenharmony_ci	dcss_ctxld_free_ctx(ctxld);
24062306a36Sopenharmony_ci	kfree(ctxld);
24162306a36Sopenharmony_ci
24262306a36Sopenharmony_ci	return ret;
24362306a36Sopenharmony_ci}
24462306a36Sopenharmony_ci
24562306a36Sopenharmony_civoid dcss_ctxld_exit(struct dcss_ctxld *ctxld)
24662306a36Sopenharmony_ci{
24762306a36Sopenharmony_ci	free_irq(ctxld->irq, ctxld);
24862306a36Sopenharmony_ci
24962306a36Sopenharmony_ci	if (ctxld->ctxld_reg)
25062306a36Sopenharmony_ci		iounmap(ctxld->ctxld_reg);
25162306a36Sopenharmony_ci
25262306a36Sopenharmony_ci	dcss_ctxld_free_ctx(ctxld);
25362306a36Sopenharmony_ci	kfree(ctxld);
25462306a36Sopenharmony_ci}
25562306a36Sopenharmony_ci
25662306a36Sopenharmony_cistatic int dcss_ctxld_enable_locked(struct dcss_ctxld *ctxld)
25762306a36Sopenharmony_ci{
25862306a36Sopenharmony_ci	int curr_ctx = ctxld->current_ctx;
25962306a36Sopenharmony_ci	u32 db_base, sb_base, sb_count;
26062306a36Sopenharmony_ci	u32 sb_hp_cnt, sb_lp_cnt, db_cnt;
26162306a36Sopenharmony_ci	struct dcss_dev *dcss = dcss_drv_dev_to_dcss(ctxld->dev);
26262306a36Sopenharmony_ci
26362306a36Sopenharmony_ci	if (!dcss)
26462306a36Sopenharmony_ci		return 0;
26562306a36Sopenharmony_ci
26662306a36Sopenharmony_ci	dcss_dpr_write_sysctrl(dcss->dpr);
26762306a36Sopenharmony_ci
26862306a36Sopenharmony_ci	dcss_scaler_write_sclctrl(dcss->scaler);
26962306a36Sopenharmony_ci
27062306a36Sopenharmony_ci	sb_hp_cnt = ctxld->ctx_size[curr_ctx][CTX_SB_HP];
27162306a36Sopenharmony_ci	sb_lp_cnt = ctxld->ctx_size[curr_ctx][CTX_SB_LP];
27262306a36Sopenharmony_ci	db_cnt = ctxld->ctx_size[curr_ctx][CTX_DB];
27362306a36Sopenharmony_ci
27462306a36Sopenharmony_ci	/* make sure SB_LP context area comes after SB_HP */
27562306a36Sopenharmony_ci	if (sb_lp_cnt &&
27662306a36Sopenharmony_ci	    ctxld->sb_lp[curr_ctx] != ctxld->sb_hp[curr_ctx] + sb_hp_cnt) {
27762306a36Sopenharmony_ci		struct dcss_ctxld_item *sb_lp_adjusted;
27862306a36Sopenharmony_ci
27962306a36Sopenharmony_ci		sb_lp_adjusted = ctxld->sb_hp[curr_ctx] + sb_hp_cnt;
28062306a36Sopenharmony_ci
28162306a36Sopenharmony_ci		memcpy(sb_lp_adjusted, ctxld->sb_lp[curr_ctx],
28262306a36Sopenharmony_ci		       sb_lp_cnt * CTX_ITEM_SIZE);
28362306a36Sopenharmony_ci	}
28462306a36Sopenharmony_ci
28562306a36Sopenharmony_ci	db_base = db_cnt ? ctxld->db_paddr[curr_ctx] : 0;
28662306a36Sopenharmony_ci
28762306a36Sopenharmony_ci	dcss_writel(db_base, ctxld->ctxld_reg + DCSS_CTXLD_DB_BASE_ADDR);
28862306a36Sopenharmony_ci	dcss_writel(db_cnt, ctxld->ctxld_reg + DCSS_CTXLD_DB_COUNT);
28962306a36Sopenharmony_ci
29062306a36Sopenharmony_ci	if (sb_hp_cnt)
29162306a36Sopenharmony_ci		sb_count = ((sb_hp_cnt << SB_HP_COUNT_POS) & SB_HP_COUNT_MASK) |
29262306a36Sopenharmony_ci			   ((sb_lp_cnt << SB_LP_COUNT_POS) & SB_LP_COUNT_MASK);
29362306a36Sopenharmony_ci	else
29462306a36Sopenharmony_ci		sb_count = (sb_lp_cnt << SB_HP_COUNT_POS) & SB_HP_COUNT_MASK;
29562306a36Sopenharmony_ci
29662306a36Sopenharmony_ci	sb_base = sb_count ? ctxld->sb_paddr[curr_ctx] : 0;
29762306a36Sopenharmony_ci
29862306a36Sopenharmony_ci	dcss_writel(sb_base, ctxld->ctxld_reg + DCSS_CTXLD_SB_BASE_ADDR);
29962306a36Sopenharmony_ci	dcss_writel(sb_count, ctxld->ctxld_reg + DCSS_CTXLD_SB_COUNT);
30062306a36Sopenharmony_ci
30162306a36Sopenharmony_ci	/* enable the context loader */
30262306a36Sopenharmony_ci	dcss_set(CTXLD_ENABLE, ctxld->ctxld_reg + DCSS_CTXLD_CONTROL_STATUS);
30362306a36Sopenharmony_ci
30462306a36Sopenharmony_ci	ctxld->in_use = true;
30562306a36Sopenharmony_ci
30662306a36Sopenharmony_ci	/*
30762306a36Sopenharmony_ci	 * Toggle the current context to the alternate one so that any updates
30862306a36Sopenharmony_ci	 * in the modules' settings take place there.
30962306a36Sopenharmony_ci	 */
31062306a36Sopenharmony_ci	ctxld->current_ctx ^= 1;
31162306a36Sopenharmony_ci
31262306a36Sopenharmony_ci	ctxld->ctx_size[ctxld->current_ctx][CTX_DB] = 0;
31362306a36Sopenharmony_ci	ctxld->ctx_size[ctxld->current_ctx][CTX_SB_HP] = 0;
31462306a36Sopenharmony_ci	ctxld->ctx_size[ctxld->current_ctx][CTX_SB_LP] = 0;
31562306a36Sopenharmony_ci
31662306a36Sopenharmony_ci	return 0;
31762306a36Sopenharmony_ci}
31862306a36Sopenharmony_ci
31962306a36Sopenharmony_ciint dcss_ctxld_enable(struct dcss_ctxld *ctxld)
32062306a36Sopenharmony_ci{
32162306a36Sopenharmony_ci	spin_lock_irq(&ctxld->lock);
32262306a36Sopenharmony_ci	ctxld->armed = true;
32362306a36Sopenharmony_ci	spin_unlock_irq(&ctxld->lock);
32462306a36Sopenharmony_ci
32562306a36Sopenharmony_ci	return 0;
32662306a36Sopenharmony_ci}
32762306a36Sopenharmony_ci
32862306a36Sopenharmony_civoid dcss_ctxld_kick(struct dcss_ctxld *ctxld)
32962306a36Sopenharmony_ci{
33062306a36Sopenharmony_ci	unsigned long flags;
33162306a36Sopenharmony_ci
33262306a36Sopenharmony_ci	spin_lock_irqsave(&ctxld->lock, flags);
33362306a36Sopenharmony_ci	if (ctxld->armed && !ctxld->in_use) {
33462306a36Sopenharmony_ci		ctxld->armed = false;
33562306a36Sopenharmony_ci		dcss_ctxld_enable_locked(ctxld);
33662306a36Sopenharmony_ci	}
33762306a36Sopenharmony_ci	spin_unlock_irqrestore(&ctxld->lock, flags);
33862306a36Sopenharmony_ci}
33962306a36Sopenharmony_ci
34062306a36Sopenharmony_civoid dcss_ctxld_write_irqsafe(struct dcss_ctxld *ctxld, u32 ctx_id, u32 val,
34162306a36Sopenharmony_ci			      u32 reg_ofs)
34262306a36Sopenharmony_ci{
34362306a36Sopenharmony_ci	int curr_ctx = ctxld->current_ctx;
34462306a36Sopenharmony_ci	struct dcss_ctxld_item *ctx[] = {
34562306a36Sopenharmony_ci		[CTX_DB] = ctxld->db[curr_ctx],
34662306a36Sopenharmony_ci		[CTX_SB_HP] = ctxld->sb_hp[curr_ctx],
34762306a36Sopenharmony_ci		[CTX_SB_LP] = ctxld->sb_lp[curr_ctx]
34862306a36Sopenharmony_ci	};
34962306a36Sopenharmony_ci	int item_idx = ctxld->ctx_size[curr_ctx][ctx_id];
35062306a36Sopenharmony_ci
35162306a36Sopenharmony_ci	if (item_idx + 1 > dcss_ctxld_ctx_size[ctx_id]) {
35262306a36Sopenharmony_ci		WARN_ON(1);
35362306a36Sopenharmony_ci		return;
35462306a36Sopenharmony_ci	}
35562306a36Sopenharmony_ci
35662306a36Sopenharmony_ci	ctx[ctx_id][item_idx].val = val;
35762306a36Sopenharmony_ci	ctx[ctx_id][item_idx].ofs = reg_ofs;
35862306a36Sopenharmony_ci	ctxld->ctx_size[curr_ctx][ctx_id] += 1;
35962306a36Sopenharmony_ci}
36062306a36Sopenharmony_ci
36162306a36Sopenharmony_civoid dcss_ctxld_write(struct dcss_ctxld *ctxld, u32 ctx_id,
36262306a36Sopenharmony_ci		      u32 val, u32 reg_ofs)
36362306a36Sopenharmony_ci{
36462306a36Sopenharmony_ci	spin_lock_irq(&ctxld->lock);
36562306a36Sopenharmony_ci	dcss_ctxld_write_irqsafe(ctxld, ctx_id, val, reg_ofs);
36662306a36Sopenharmony_ci	spin_unlock_irq(&ctxld->lock);
36762306a36Sopenharmony_ci}
36862306a36Sopenharmony_ci
36962306a36Sopenharmony_cibool dcss_ctxld_is_flushed(struct dcss_ctxld *ctxld)
37062306a36Sopenharmony_ci{
37162306a36Sopenharmony_ci	return ctxld->ctx_size[ctxld->current_ctx][CTX_DB] == 0 &&
37262306a36Sopenharmony_ci		ctxld->ctx_size[ctxld->current_ctx][CTX_SB_HP] == 0 &&
37362306a36Sopenharmony_ci		ctxld->ctx_size[ctxld->current_ctx][CTX_SB_LP] == 0;
37462306a36Sopenharmony_ci}
37562306a36Sopenharmony_ci
37662306a36Sopenharmony_ciint dcss_ctxld_resume(struct dcss_ctxld *ctxld)
37762306a36Sopenharmony_ci{
37862306a36Sopenharmony_ci	dcss_ctxld_hw_cfg(ctxld);
37962306a36Sopenharmony_ci
38062306a36Sopenharmony_ci	if (!ctxld->irq_en) {
38162306a36Sopenharmony_ci		enable_irq(ctxld->irq);
38262306a36Sopenharmony_ci		ctxld->irq_en = true;
38362306a36Sopenharmony_ci	}
38462306a36Sopenharmony_ci
38562306a36Sopenharmony_ci	return 0;
38662306a36Sopenharmony_ci}
38762306a36Sopenharmony_ci
38862306a36Sopenharmony_ciint dcss_ctxld_suspend(struct dcss_ctxld *ctxld)
38962306a36Sopenharmony_ci{
39062306a36Sopenharmony_ci	int ret = 0;
39162306a36Sopenharmony_ci	unsigned long timeout = jiffies + msecs_to_jiffies(500);
39262306a36Sopenharmony_ci
39362306a36Sopenharmony_ci	if (!dcss_ctxld_is_flushed(ctxld)) {
39462306a36Sopenharmony_ci		dcss_ctxld_kick(ctxld);
39562306a36Sopenharmony_ci
39662306a36Sopenharmony_ci		while (!time_after(jiffies, timeout) && ctxld->in_use)
39762306a36Sopenharmony_ci			msleep(20);
39862306a36Sopenharmony_ci
39962306a36Sopenharmony_ci		if (time_after(jiffies, timeout))
40062306a36Sopenharmony_ci			return -ETIMEDOUT;
40162306a36Sopenharmony_ci	}
40262306a36Sopenharmony_ci
40362306a36Sopenharmony_ci	spin_lock_irq(&ctxld->lock);
40462306a36Sopenharmony_ci
40562306a36Sopenharmony_ci	if (ctxld->irq_en) {
40662306a36Sopenharmony_ci		disable_irq_nosync(ctxld->irq);
40762306a36Sopenharmony_ci		ctxld->irq_en = false;
40862306a36Sopenharmony_ci	}
40962306a36Sopenharmony_ci
41062306a36Sopenharmony_ci	/* reset context region and sizes */
41162306a36Sopenharmony_ci	ctxld->current_ctx = 0;
41262306a36Sopenharmony_ci	ctxld->ctx_size[0][CTX_DB] = 0;
41362306a36Sopenharmony_ci	ctxld->ctx_size[0][CTX_SB_HP] = 0;
41462306a36Sopenharmony_ci	ctxld->ctx_size[0][CTX_SB_LP] = 0;
41562306a36Sopenharmony_ci
41662306a36Sopenharmony_ci	spin_unlock_irq(&ctxld->lock);
41762306a36Sopenharmony_ci
41862306a36Sopenharmony_ci	return ret;
41962306a36Sopenharmony_ci}
42062306a36Sopenharmony_ci
42162306a36Sopenharmony_civoid dcss_ctxld_assert_locked(struct dcss_ctxld *ctxld)
42262306a36Sopenharmony_ci{
42362306a36Sopenharmony_ci	lockdep_assert_held(&ctxld->lock);
42462306a36Sopenharmony_ci}
425