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

#include <linux/component.h>
#include <linux/mod_devicetable.h>
#include <linux/module.h>
#include <linux/of_platform.h>
#include <linux/platform_device.h>
#include <linux/pm.h>
#include <linux/pm_runtime.h>
#include <linux/regmap.h>

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

#define POLARITYCTRL            0xc
#define  POLEN_HIGH             BIT(2)

static const struct dc_subdev_info dc_de_info[] = {
        { .reg_start = 0x5618b400, .id = 0, },
        { .reg_start = 0x5618b420, .id = 1, },
};

static const struct regmap_range dc_de_regmap_ranges[] = {
        regmap_reg_range(POLARITYCTRL, POLARITYCTRL),
};

static const struct regmap_access_table dc_de_regmap_access_table = {
        .yes_ranges = dc_de_regmap_ranges,
        .n_yes_ranges = ARRAY_SIZE(dc_de_regmap_ranges),
};

static const struct regmap_config dc_de_top_regmap_config = {
        .name = "top",
        .reg_bits = 32,
        .reg_stride = 4,
        .val_bits = 32,
        .fast_io = true,
        .wr_table = &dc_de_regmap_access_table,
        .rd_table = &dc_de_regmap_access_table,
        .max_register = POLARITYCTRL,
};

static inline void dc_dec_init(struct dc_de *de)
{
        regmap_write_bits(de->reg_top, POLARITYCTRL, POLARITYCTRL, POLEN_HIGH);
}

static int dc_de_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_top;
        void __iomem *base_top;
        struct dc_de *de;
        int ret, id;

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

        base_top = devm_platform_get_and_ioremap_resource(pdev, 0, &res_top);
        if (IS_ERR(base_top))
                return PTR_ERR(base_top);

        de->reg_top = devm_regmap_init_mmio(dev, base_top,
                                            &dc_de_top_regmap_config);
        if (IS_ERR(de->reg_top))
                return PTR_ERR(de->reg_top);

        de->irq_shdload = platform_get_irq_byname(pdev, "shdload");
        if (de->irq_shdload < 0)
                return de->irq_shdload;

        de->irq_framecomplete = platform_get_irq_byname(pdev, "framecomplete");
        if (de->irq_framecomplete < 0)
                return de->irq_framecomplete;

        de->irq_seqcomplete = platform_get_irq_byname(pdev, "seqcomplete");
        if (de->irq_seqcomplete < 0)
                return de->irq_seqcomplete;

        de->dev = dev;

        dev_set_drvdata(dev, de);

        ret = devm_pm_runtime_enable(dev);
        if (ret)
                return ret;

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

        dc_drm->de[id] = de;

        return 0;
}

/*
 * It's possible to get the child device pointers from the child component
 * bind callbacks, but it depends on the component helper behavior to bind
 * the display engine component first.  To avoid the dependency, post bind
 * to get the pointers from dc_drm in a safe manner.
 */
void dc_de_post_bind(struct dc_drm_device *dc_drm)
{
        struct dc_de *de;
        int i;

        for (i = 0; i < DC_DISPLAYS; i++) {
                de = dc_drm->de[i];
                de->fg = dc_drm->fg[i];
                de->tc = dc_drm->tc[i];
        }
}

static const struct component_ops dc_de_ops = {
        .bind = dc_de_bind,
};

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

        ret = devm_of_platform_populate(&pdev->dev);
        if (ret < 0)
                return ret;

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

        return 0;
}

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

static int dc_de_runtime_resume(struct device *dev)
{
        struct dc_de *de = dev_get_drvdata(dev);

        dc_dec_init(de);
        dc_fg_init(de->fg);
        dc_tc_init(de->tc);

        return 0;
}

static const struct dev_pm_ops dc_de_pm_ops = {
        RUNTIME_PM_OPS(NULL, dc_de_runtime_resume, NULL)
};

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

struct platform_driver dc_de_driver = {
        .probe = dc_de_probe,
        .remove = dc_de_remove,
        .driver = {
                .name = "imx8-dc-display-engine",
                .suppress_bind_attrs = true,
                .of_match_table = dc_de_dt_ids,
                .pm = pm_sleep_ptr(&dc_de_pm_ops),
        },
};