// SPDX-License-Identifier: GPL-2.0-only
/*
 * Copyright (c) 2021-2022 Qualcomm Innovation Center, Inc. All rights reserved.
 * Copyright (c) 2017-2021, The Linux Foundation. All rights reserved.
 */

#define pr_fmt(fmt)	"[drm:%s:%d] " fmt, __func__, __LINE__
#include "sde_hw_mdss.h"
#include "sde_hwio.h"
#include "sde_hw_catalog.h"
#include "sde_hw_dsc.h"
#include "sde_hw_pingpong.h"
#include "sde_dbg.h"
#include "sde_kms.h"
#include "sde_hw_dsc_1_2.h"

#define DSC_COMMON_MODE	                0x000
#define DSC_ENC                         0X004
#define DSC_PICTURE                     0x008
#define DSC_SLICE                       0x00C
#define DSC_CHUNK_SIZE                  0x010
#define DSC_DELAY                       0x014
#define DSC_SCALE_INITIAL               0x018
#define DSC_SCALE_DEC_INTERVAL          0x01C
#define DSC_SCALE_INC_INTERVAL          0x020
#define DSC_FIRST_LINE_BPG_OFFSET       0x024
#define DSC_BPG_OFFSET                  0x028
#define DSC_DSC_OFFSET                  0x02C
#define DSC_FLATNESS                    0x030
#define DSC_RC_MODEL_SIZE               0x034
#define DSC_RC                          0x038
#define DSC_RC_BUF_THRESH               0x03C
#define DSC_RANGE_MIN_QP                0x074
#define DSC_RANGE_MAX_QP                0x0B0
#define DSC_RANGE_BPG_OFFSET            0x0EC

#define DSC_CTL_BLOCK_SIZE              0x300
#define DSC_CTL(m)     \
	(((m == DSC_NONE) || (m >= DSC_MAX)) ? 0 : (0x1800 - 0x3FC * (m - 1)))

static void sde_hw_dsc_disable(struct sde_hw_dsc *dsc)
{
	struct sde_hw_blk_reg_map *dsc_c = &dsc->hw;

	SDE_REG_WRITE(dsc_c, DSC_COMMON_MODE, 0);
}

static void sde_hw_dsc_config(struct sde_hw_dsc *hw_dsc,
		struct msm_display_dsc_info *dsc, u32 mode,
		bool ich_reset_override)
{
	u32 data;
	u32 initial_lines = dsc->initial_lines;
	struct sde_hw_blk_reg_map *dsc_c = &hw_dsc->hw;

	SDE_REG_WRITE(dsc_c, DSC_COMMON_MODE, mode);

	data = 0;
	if (ich_reset_override)
		data = 3 << 28;

	data |= (initial_lines << 20);
	data |= (dsc->slice_last_group_size << 18);
	/* integer bpp support only */
	data |= (dsc->config.bits_per_pixel << 8);
	data |= (dsc->config.block_pred_enable << 7);
	data |= (dsc->config.line_buf_depth << 3);
	data |= (dsc->config.simple_422 << 2);
	data |= (dsc->config.convert_rgb << 1);
	if (dsc->config.bits_per_component == 10)
		data |= BIT(0);

	SDE_REG_WRITE(dsc_c, DSC_ENC, data);

	data = dsc->config.pic_width << 16;
	data |= dsc->config.pic_height;
	SDE_REG_WRITE(dsc_c, DSC_PICTURE, data);

	data = dsc->config.slice_width << 16;
	data |= dsc->config.slice_height;
	SDE_REG_WRITE(dsc_c, DSC_SLICE, data);

	data = dsc->config.slice_chunk_size << 16;
	SDE_REG_WRITE(dsc_c, DSC_CHUNK_SIZE, data);

	data = dsc->config.initial_dec_delay << 16;
	data |= dsc->config.initial_xmit_delay;
	SDE_REG_WRITE(dsc_c, DSC_DELAY, data);

	data = dsc->config.initial_scale_value;
	SDE_REG_WRITE(dsc_c, DSC_SCALE_INITIAL, data);

	data = dsc->config.scale_decrement_interval;
	SDE_REG_WRITE(dsc_c, DSC_SCALE_DEC_INTERVAL, data);

	data = dsc->config.scale_increment_interval;
	SDE_REG_WRITE(dsc_c, DSC_SCALE_INC_INTERVAL, data);

	data = dsc->config.first_line_bpg_offset;
	SDE_REG_WRITE(dsc_c, DSC_FIRST_LINE_BPG_OFFSET, data);

	data = dsc->config.nfl_bpg_offset << 16;
	data |= dsc->config.slice_bpg_offset;
	SDE_REG_WRITE(dsc_c, DSC_BPG_OFFSET, data);

	data = dsc->config.initial_offset << 16;
	data |= dsc->config.final_offset;
	SDE_REG_WRITE(dsc_c, DSC_DSC_OFFSET, data);

	data = dsc->det_thresh_flatness << 10;
	data |= dsc->config.flatness_max_qp << 5;
	data |= dsc->config.flatness_min_qp;
	SDE_REG_WRITE(dsc_c, DSC_FLATNESS, data);

	data = dsc->config.rc_model_size;
	SDE_REG_WRITE(dsc_c, DSC_RC_MODEL_SIZE, data);

	data = dsc->config.rc_tgt_offset_low << 18;
	data |= dsc->config.rc_tgt_offset_high << 14;
	data |= dsc->config.rc_quant_incr_limit1 << 9;
	data |= dsc->config.rc_quant_incr_limit0 << 4;
	data |= dsc->config.rc_edge_factor;
	SDE_REG_WRITE(dsc_c, DSC_RC, data);
}

static void sde_hw_dsc_config_thresh(struct sde_hw_dsc *hw_dsc,
		struct msm_display_dsc_info *dsc)
{
	u16 *lp;
	int i;

	struct sde_hw_blk_reg_map *dsc_c = &hw_dsc->hw;
	u32 off = 0x0;
	struct drm_dsc_rc_range_parameters *rc =
		dsc->config.rc_range_params;

	lp = dsc->config.rc_buf_thresh;
	off = DSC_RC_BUF_THRESH;
	for (i = 0; i < DSC_NUM_BUF_RANGES - 1; i++) {
		SDE_REG_WRITE(dsc_c, off, *lp++);
		off += 4;
	}

	off = DSC_RANGE_MIN_QP;
	for (i = 0; i < DSC_NUM_BUF_RANGES; i++) {
		SDE_REG_WRITE(dsc_c, off, rc[i].range_min_qp);
		off += 4;
	}

	off = DSC_RANGE_MAX_QP;
	for (i = 0; i < DSC_NUM_BUF_RANGES; i++) {
		SDE_REG_WRITE(dsc_c, off, rc[i].range_max_qp);
		off += 4;
	}

	off = DSC_RANGE_BPG_OFFSET;
	for (i = 0; i < DSC_NUM_BUF_RANGES; i++) {
		SDE_REG_WRITE(dsc_c, off, rc[i].range_bpg_offset);
		off += 4;
	}
}

static void sde_hw_dsc_bind_pingpong_blk(
		struct sde_hw_dsc *hw_dsc,
		bool enable,
		const enum sde_pingpong pp)
{
	struct sde_hw_blk_reg_map *c;
	int mux_cfg = 0xF;
	u32 dsc_ctl_offset;

	if (!hw_dsc)
		return;

	c = &hw_dsc->hw;
	dsc_ctl_offset = DSC_CTL(hw_dsc->idx);

	if (enable)
		mux_cfg = (pp - PINGPONG_0) & 0x7;

	if (dsc_ctl_offset)
		SDE_REG_WRITE(c, dsc_ctl_offset, mux_cfg);
}


static struct sde_dsc_cfg *_dsc_offset(enum sde_dsc dsc,
		struct sde_mdss_cfg *m,
		void __iomem *addr,
		struct sde_hw_blk_reg_map *b)
{
	int i;

	for (i = 0; i < m->dsc_count; i++) {
		if (dsc == m->dsc[i].id) {
			b->base_off = addr;
			b->blk_off = m->dsc[i].base;
			b->length = m->dsc[i].len;
			b->hw_rev = m->hw_rev;
			b->log_mask = SDE_DBG_MASK_DSC;
			return &m->dsc[i];
		}
	}

	return NULL;
}

static void _setup_dsc_ops(struct sde_hw_dsc_ops *ops,
		unsigned long features)
{
	ops->dsc_disable = sde_hw_dsc_disable;
	ops->dsc_config = sde_hw_dsc_config;
	ops->dsc_config_thresh = sde_hw_dsc_config_thresh;
	if (test_bit(SDE_DSC_OUTPUT_CTRL, &features))
		ops->bind_pingpong_blk = sde_hw_dsc_bind_pingpong_blk;
};

struct sde_hw_blk_reg_map *sde_hw_dsc_init(enum sde_dsc idx,
		void __iomem *addr,
		struct sde_mdss_cfg *m)
{
	struct sde_hw_dsc *c;
	struct sde_dsc_cfg *cfg;
	u32 dsc_ctl_offset;

	c = kzalloc(sizeof(*c), GFP_KERNEL);
	if (!c)
		return ERR_PTR(-ENOMEM);

	cfg = _dsc_offset(idx, m, addr, &c->hw);
	if (IS_ERR_OR_NULL(cfg))
		goto error_inv;

	c->idx = idx;
	c->caps = cfg;
	if (test_bit(SDE_DSC_HW_REV_1_1, &c->caps->features)) {
		_setup_dsc_ops(&c->ops, c->caps->features);
		sde_dbg_reg_register_dump_range(SDE_DBG_NAME, cfg->name,
				c->hw.blk_off,
				c->hw.blk_off + c->hw.length,
				c->hw.xin_id);

		if ((c->idx == DSC_0) &&
				test_bit(SDE_DSC_OUTPUT_CTRL, &cfg->features)) {
			dsc_ctl_offset = DSC_CTL(c->idx);
			sde_dbg_reg_register_dump_range(SDE_DBG_NAME,
					"dsc_ctl",
					c->hw.blk_off + dsc_ctl_offset,
					c->hw.blk_off + dsc_ctl_offset +
						DSC_CTL_BLOCK_SIZE,
					c->hw.xin_id);
		}
	} else if (test_bit(SDE_DSC_HW_REV_1_2, &c->caps->features)) {
		char blk_name[32];

		sde_dsc1_2_setup_ops(&c->ops, c->caps->features);

		sde_dbg_reg_register_dump_range(SDE_DBG_NAME, cfg->name,
				c->hw.blk_off,
				c->hw.blk_off + c->hw.length,
				c->hw.xin_id);

		snprintf(blk_name, sizeof(blk_name), "dsc_enc_%u",
				c->idx - DSC_0);

		sde_dbg_reg_register_dump_range(SDE_DBG_NAME,
				blk_name,
				c->hw.blk_off + c->caps->sblk->enc.base,
				c->hw.blk_off + c->caps->sblk->enc.base +
				c->caps->sblk->enc.len,
				c->hw.xin_id);

		snprintf(blk_name, sizeof(blk_name), "dsc_ctl_%u",
				c->idx - DSC_0);

		sde_dbg_reg_register_dump_range(SDE_DBG_NAME,
				blk_name,
				c->hw.blk_off + c->caps->sblk->ctl.base,
				c->hw.blk_off + c->caps->sblk->ctl.base +
					c->caps->sblk->ctl.len,
				c->hw.xin_id);
	} else {
		SDE_ERROR("failed to setup ops\n");
		goto error_inv;
	}

	return &c->hw;

error_inv:
	kfree(c);

	return ERR_PTR(-EINVAL);
}

void sde_hw_dsc_destroy(struct sde_hw_blk_reg_map *hw)
{
	if (hw)
		kfree(to_sde_hw_dsc(hw));
}