root/drivers/gpu/drm/imx/dc/dc-lb.c
// SPDX-License-Identifier: GPL-2.0+
/*
 * Copyright 2024 NXP
 */

#include <linux/bitfield.h>
#include <linux/bits.h>
#include <linux/component.h>
#include <linux/mod_devicetable.h>
#include <linux/module.h>
#include <linux/platform_device.h>
#include <linux/regmap.h>

#include <drm/drm_blend.h>

#include "dc-drv.h"
#include "dc-pe.h"

#define PIXENGCFG_DYNAMIC                       0x8
#define  PIXENGCFG_DYNAMIC_SEC_SEL_MASK         GENMASK(13, 8)
#define  PIXENGCFG_DYNAMIC_SEC_SEL(x)           \
                FIELD_PREP(PIXENGCFG_DYNAMIC_SEC_SEL_MASK, (x))
#define  PIXENGCFG_DYNAMIC_PRIM_SEL_MASK        GENMASK(5, 0)
#define  PIXENGCFG_DYNAMIC_PRIM_SEL(x)          \
                FIELD_PREP(PIXENGCFG_DYNAMIC_PRIM_SEL_MASK, (x))

#define STATICCONTROL                           0x8
#define  SHDTOKSEL_MASK                         GENMASK(4, 3)
#define  SHDTOKSEL(x)                           FIELD_PREP(SHDTOKSEL_MASK, (x))
#define  SHDLDSEL_MASK                          GENMASK(2, 1)
#define  SHDLDSEL(x)                            FIELD_PREP(SHDLDSEL_MASK, (x))

#define CONTROL                                 0xc
#define  CTRL_MODE_MASK                         BIT(0)
#define  CTRL_MODE(x)                           FIELD_PREP(CTRL_MODE_MASK, (x))

#define BLENDCONTROL                            0x10
#define  ALPHA_MASK                             GENMASK(23, 16)
#define  ALPHA(x)                               FIELD_PREP(ALPHA_MASK, (x))
#define  SEC_A_BLD_FUNC_MASK                    GENMASK(14, 12)
#define  SEC_A_BLD_FUNC(x)                      \
                FIELD_PREP(SEC_A_BLD_FUNC_MASK, (x))
#define  PRIM_A_BLD_FUNC_MASK                   GENMASK(10, 8)
#define  PRIM_A_BLD_FUNC(x)                     \
                FIELD_PREP(PRIM_A_BLD_FUNC_MASK, (x))
#define  SEC_C_BLD_FUNC_MASK                    GENMASK(6, 4)
#define  SEC_C_BLD_FUNC(x)                      \
                FIELD_PREP(SEC_C_BLD_FUNC_MASK, (x))
#define  PRIM_C_BLD_FUNC_MASK                   GENMASK(2, 0)
#define  PRIM_C_BLD_FUNC(x)                     \
                FIELD_PREP(PRIM_C_BLD_FUNC_MASK, (x))

#define POSITION                                0x14
#define  YPOS_MASK                              GENMASK(31, 16)
#define  YPOS(x)                                FIELD_PREP(YPOS_MASK, (x))
#define  XPOS_MASK                              GENMASK(15, 0)
#define  XPOS(x)                                FIELD_PREP(XPOS_MASK, (x))

enum dc_lb_blend_func {
        DC_LAYERBLEND_BLEND_ZERO,
        DC_LAYERBLEND_BLEND_ONE,
        DC_LAYERBLEND_BLEND_PRIM_ALPHA,
        DC_LAYERBLEND_BLEND_ONE_MINUS_PRIM_ALPHA,
        DC_LAYERBLEND_BLEND_SEC_ALPHA,
        DC_LAYERBLEND_BLEND_ONE_MINUS_SEC_ALPHA,
        DC_LAYERBLEND_BLEND_CONST_ALPHA,
        DC_LAYERBLEND_BLEND_ONE_MINUS_CONST_ALPHA,
};

enum dc_lb_shadow_sel {
        BOTH = 0x2,
};

static const struct dc_subdev_info dc_lb_info[] = {
        { .reg_start = 0x56180ba0, .id = 0, },
        { .reg_start = 0x56180bc0, .id = 1, },
        { .reg_start = 0x56180be0, .id = 2, },
        { .reg_start = 0x56180c00, .id = 3, },
};

static const struct regmap_range dc_lb_pec_regmap_access_ranges[] = {
        regmap_reg_range(PIXENGCFG_DYNAMIC, PIXENGCFG_DYNAMIC),
};

static const struct regmap_access_table dc_lb_pec_regmap_access_table = {
        .yes_ranges = dc_lb_pec_regmap_access_ranges,
        .n_yes_ranges = ARRAY_SIZE(dc_lb_pec_regmap_access_ranges),
};

static const struct regmap_config dc_lb_pec_regmap_config = {
        .name = "pec",
        .reg_bits = 32,
        .reg_stride = 4,
        .val_bits = 32,
        .fast_io = true,
        .wr_table = &dc_lb_pec_regmap_access_table,
        .rd_table = &dc_lb_pec_regmap_access_table,
        .max_register = PIXENGCFG_DYNAMIC,
};

static const struct regmap_range dc_lb_regmap_ranges[] = {
        regmap_reg_range(STATICCONTROL, POSITION),
};

static const struct regmap_access_table dc_lb_regmap_access_table = {
        .yes_ranges = dc_lb_regmap_ranges,
        .n_yes_ranges = ARRAY_SIZE(dc_lb_regmap_ranges),
};

static const struct regmap_config dc_lb_cfg_regmap_config = {
        .name = "cfg",
        .reg_bits = 32,
        .reg_stride = 4,
        .val_bits = 32,
        .fast_io = true,
        .wr_table = &dc_lb_regmap_access_table,
        .rd_table = &dc_lb_regmap_access_table,
        .max_register = POSITION,
};

static const enum dc_link_id prim_sels[] = {
        /* common options */
        LINK_ID_NONE,
        LINK_ID_CONSTFRAME0,
        LINK_ID_CONSTFRAME1,
        LINK_ID_CONSTFRAME4,
        LINK_ID_CONSTFRAME5,
        /*
         * special options:
         * layerblend(n) has n special options,
         * from layerblend0 to layerblend(n - 1), e.g.,
         * layerblend3 has 3 special options -
         * layerblend0/1/2.
         */
        LINK_ID_LAYERBLEND0,
        LINK_ID_LAYERBLEND1,
        LINK_ID_LAYERBLEND2,
        LINK_ID_LAYERBLEND3,
};

static const enum dc_link_id sec_sels[] = {
        LINK_ID_NONE,
        LINK_ID_FETCHWARP2,
        LINK_ID_FETCHLAYER0,
};

enum dc_link_id dc_lb_get_link_id(struct dc_lb *lb)
{
        return lb->link;
}

void dc_lb_pec_dynamic_prim_sel(struct dc_lb *lb, enum dc_link_id prim)
{
        int fixed_sels_num = ARRAY_SIZE(prim_sels) - 4;
        int i;

        for (i = 0; i < fixed_sels_num + lb->id; i++) {
                if (prim_sels[i] == prim) {
                        regmap_write_bits(lb->reg_pec, PIXENGCFG_DYNAMIC,
                                          PIXENGCFG_DYNAMIC_PRIM_SEL_MASK,
                                          PIXENGCFG_DYNAMIC_PRIM_SEL(prim));
                        return;
                }
        }

        dev_warn(lb->dev, "invalid primary input selection:%d\n", prim);
}

void dc_lb_pec_dynamic_sec_sel(struct dc_lb *lb, enum dc_link_id sec)
{
        int i;

        for (i = 0; i < ARRAY_SIZE(sec_sels); i++) {
                if (sec_sels[i] == sec) {
                        regmap_write_bits(lb->reg_pec, PIXENGCFG_DYNAMIC,
                                          PIXENGCFG_DYNAMIC_SEC_SEL_MASK,
                                          PIXENGCFG_DYNAMIC_SEC_SEL(sec));
                        return;
                }
        }

        dev_warn(lb->dev, "invalid secondary input selection:%d\n", sec);
}

void dc_lb_pec_clken(struct dc_lb *lb, enum dc_pec_clken clken)
{
        regmap_write_bits(lb->reg_pec, PIXENGCFG_DYNAMIC, CLKEN_MASK,
                          CLKEN(clken));
}

static inline void dc_lb_enable_shden(struct dc_lb *lb)
{
        regmap_write_bits(lb->reg_cfg, STATICCONTROL, SHDEN, SHDEN);
}

static inline void dc_lb_shdtoksel(struct dc_lb *lb, enum dc_lb_shadow_sel sel)
{
        regmap_write_bits(lb->reg_cfg, STATICCONTROL, SHDTOKSEL_MASK,
                          SHDTOKSEL(sel));
}

static inline void dc_lb_shdldsel(struct dc_lb *lb, enum dc_lb_shadow_sel sel)
{
        regmap_write_bits(lb->reg_cfg, STATICCONTROL, SHDLDSEL_MASK,
                          SHDLDSEL(sel));
}

void dc_lb_mode(struct dc_lb *lb, enum dc_lb_mode mode)
{
        regmap_write_bits(lb->reg_cfg, CONTROL, CTRL_MODE_MASK, mode);
}

static inline void dc_lb_blendcontrol(struct dc_lb *lb)
{
        u32 val = PRIM_A_BLD_FUNC(DC_LAYERBLEND_BLEND_ZERO) |
                  SEC_A_BLD_FUNC(DC_LAYERBLEND_BLEND_ZERO) |
                  PRIM_C_BLD_FUNC(DC_LAYERBLEND_BLEND_ZERO) |
                  SEC_C_BLD_FUNC(DC_LAYERBLEND_BLEND_CONST_ALPHA) |
                  ALPHA(DRM_BLEND_ALPHA_OPAQUE >> 8);

        regmap_write(lb->reg_cfg, BLENDCONTROL, val);
}

void dc_lb_position(struct dc_lb *lb, int x, int y)
{
        regmap_write(lb->reg_cfg, POSITION, XPOS(x) | YPOS(y));
}

int dc_lb_get_id(struct dc_lb *lb)
{
        return lb->id;
}

void dc_lb_init(struct dc_lb *lb)
{
        dc_lb_pec_dynamic_prim_sel(lb, LINK_ID_NONE);
        dc_lb_pec_dynamic_sec_sel(lb, LINK_ID_NONE);
        dc_lb_pec_clken(lb, CLKEN_DISABLE);
        dc_lb_shdldsel(lb, BOTH);
        dc_lb_shdtoksel(lb, BOTH);
        dc_lb_blendcontrol(lb);
        dc_lb_enable_shden(lb);
}

static int dc_lb_bind(struct device *dev, struct device *master, void *data)
{
        struct platform_device *pdev = to_platform_device(dev);
        struct dc_drm_device *dc_drm = data;
        struct resource *res_pec;
        void __iomem *base_pec;
        void __iomem *base_cfg;
        struct dc_lb *lb;

        lb = devm_kzalloc(dev, sizeof(*lb), GFP_KERNEL);
        if (!lb)
                return -ENOMEM;

        base_pec = devm_platform_get_and_ioremap_resource(pdev, 0, &res_pec);
        if (IS_ERR(base_pec))
                return PTR_ERR(base_pec);

        base_cfg = devm_platform_ioremap_resource_byname(pdev, "cfg");
        if (IS_ERR(base_cfg))
                return PTR_ERR(base_cfg);

        lb->reg_pec = devm_regmap_init_mmio(dev, base_pec,
                                            &dc_lb_pec_regmap_config);
        if (IS_ERR(lb->reg_pec))
                return PTR_ERR(lb->reg_pec);

        lb->reg_cfg = devm_regmap_init_mmio(dev, base_cfg,
                                            &dc_lb_cfg_regmap_config);
        if (IS_ERR(lb->reg_cfg))
                return PTR_ERR(lb->reg_cfg);

        lb->id = dc_subdev_get_id(dc_lb_info, ARRAY_SIZE(dc_lb_info), res_pec);
        if (lb->id < 0) {
                dev_err(dev, "failed to get instance number: %d\n", lb->id);
                return lb->id;
        }

        lb->dev = dev;
        lb->link = LINK_ID_LAYERBLEND0 + lb->id;

        dc_drm->lb[lb->id] = lb;

        return 0;
}

static const struct component_ops dc_lb_ops = {
        .bind = dc_lb_bind,
};

static int dc_lb_probe(struct platform_device *pdev)
{
        int ret;

        ret = component_add(&pdev->dev, &dc_lb_ops);
        if (ret)
                return dev_err_probe(&pdev->dev, ret,
                                     "failed to add component\n");

        return 0;
}

static void dc_lb_remove(struct platform_device *pdev)
{
        component_del(&pdev->dev, &dc_lb_ops);
}

static const struct of_device_id dc_lb_dt_ids[] = {
        { .compatible = "fsl,imx8qxp-dc-layerblend" },
        { /* sentinel */ }
};
MODULE_DEVICE_TABLE(of, dc_lb_dt_ids);

struct platform_driver dc_lb_driver = {
        .probe = dc_lb_probe,
        .remove = dc_lb_remove,
        .driver = {
                .name = "imx8-dc-layerblend",
                .suppress_bind_attrs = true,
                .of_match_table = dc_lb_dt_ids,
        },
};