// SPDX-License-Identifier: GPL-2.0+
/*
 * Copyright 2024 NXP
 */

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

#include <drm/drm_fourcc.h>

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

#define BASEADDRESS(x)                  (0x10 + FRAC_OFFSET * (x))
#define SOURCEBUFFERATTRIBUTES(x)       (0x14 + FRAC_OFFSET * (x))
#define SOURCEBUFFERDIMENSION(x)        (0x18 + FRAC_OFFSET * (x))
#define COLORCOMPONENTBITS(x)           (0x1c + FRAC_OFFSET * (x))
#define COLORCOMPONENTSHIFT(x)          (0x20 + FRAC_OFFSET * (x))
#define LAYEROFFSET(x)                  (0x24 + FRAC_OFFSET * (x))
#define CLIPWINDOWOFFSET(x)             (0x28 + FRAC_OFFSET * (x))
#define CLIPWINDOWDIMENSIONS(x)         (0x2c + FRAC_OFFSET * (x))
#define CONSTANTCOLOR(x)                (0x30 + FRAC_OFFSET * (x))
#define LAYERPROPERTY(x)                (0x34 + FRAC_OFFSET * (x))
#define FRAMEDIMENSIONS                 0x150

struct dc_fl {
        struct dc_fu fu;
};

static const struct dc_subdev_info dc_fl_info[] = {
        { .reg_start = 0x56180ac0, .id = 0, },
};

static const struct regmap_range dc_fl_regmap_ranges[] = {
        regmap_reg_range(STATICCONTROL, FRAMEDIMENSIONS),
};

static const struct regmap_access_table dc_fl_regmap_access_table = {
        .yes_ranges = dc_fl_regmap_ranges,
        .n_yes_ranges = ARRAY_SIZE(dc_fl_regmap_ranges),
};

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

static void dc_fl_set_fmt(struct dc_fu *fu, enum dc_fu_frac frac,
                          const struct drm_format_info *format)
{
        u32 bits = 0, shifts = 0;

        dc_fu_set_src_bpp(fu, frac, format->cpp[0] * 8);

        regmap_write_bits(fu->reg_cfg, LAYERPROPERTY(frac),
                          YUVCONVERSIONMODE_MASK,
                          YUVCONVERSIONMODE(YUVCONVERSIONMODE_OFF));

        dc_fu_get_pixel_format_bits(fu, format->format, &bits);
        dc_fu_get_pixel_format_shifts(fu, format->format, &shifts);

        regmap_write(fu->reg_cfg, COLORCOMPONENTBITS(frac), bits);
        regmap_write(fu->reg_cfg, COLORCOMPONENTSHIFT(frac), shifts);
}

static void dc_fl_set_framedimensions(struct dc_fu *fu, int w, int h)
{
        regmap_write(fu->reg_cfg, FRAMEDIMENSIONS,
                     FRAMEWIDTH(w) | FRAMEHEIGHT(h));
}

static void dc_fl_init(struct dc_fu *fu)
{
        dc_fu_common_hw_init(fu);
        dc_fu_shdldreq_sticky(fu, 0xff);
}

static void dc_fl_set_ops(struct dc_fu *fu)
{
        memcpy(&fu->ops, &dc_fu_common_ops, sizeof(dc_fu_common_ops));
        fu->ops.init = dc_fl_init;
        fu->ops.set_fmt = dc_fl_set_fmt;
        fu->ops.set_framedimensions = dc_fl_set_framedimensions;
}

static int dc_fl_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_cfg;
        struct dc_fl *fl;
        struct dc_fu *fu;
        int i, id;

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

        fu = &fl->fu;

        res_pec = platform_get_resource(pdev, IORESOURCE_MEM, 0);

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

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

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

        fu->link_id = LINK_ID_FETCHLAYER0;
        fu->id = DC_FETCHUNIT_FL0;
        for (i = 0; i < DC_FETCHUNIT_FRAC_NUM; i++) {
                fu->reg_baseaddr[i]               = BASEADDRESS(i);
                fu->reg_sourcebufferattributes[i] = SOURCEBUFFERATTRIBUTES(i);
                fu->reg_sourcebufferdimension[i]  = SOURCEBUFFERDIMENSION(i);
                fu->reg_layeroffset[i]            = LAYEROFFSET(i);
                fu->reg_clipwindowoffset[i]       = CLIPWINDOWOFFSET(i);
                fu->reg_clipwindowdimensions[i]   = CLIPWINDOWDIMENSIONS(i);
                fu->reg_constantcolor[i]          = CONSTANTCOLOR(i);
                fu->reg_layerproperty[i]          = LAYERPROPERTY(i);
        }
        snprintf(fu->name, sizeof(fu->name), "FetchLayer%d", id);

        dc_fl_set_ops(fu);

        dc_drm->fu_disp[fu->id] = fu;

        return 0;
}

static const struct component_ops dc_fl_ops = {
        .bind = dc_fl_bind,
};

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

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

        return 0;
}

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

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

struct platform_driver dc_fl_driver = {
        .probe = dc_fl_probe,
        .remove = dc_fl_remove,
        .driver = {
                .name = "imx8-dc-fetchlayer",
                .suppress_bind_attrs = true,
                .of_match_table = dc_fl_dt_ids,
        },
};