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

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

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

static int dc_pe_bind(struct device *dev, struct device *master, void *data)
{
        struct dc_drm_device *dc_drm = data;
        struct dc_pe *pe;
        int ret;

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

        pe->clk_axi = devm_clk_get(dev, NULL);
        if (IS_ERR(pe->clk_axi))
                return dev_err_probe(dev, PTR_ERR(pe->clk_axi),
                                     "failed to get AXI clock\n");

        pe->dev = dev;

        dev_set_drvdata(dev, pe);

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

        dc_drm->pe = pe;

        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 pixel engine component first.  To avoid the dependency, post bind to
 * get the pointers from dc_drm in a safe manner.
 */
void dc_pe_post_bind(struct dc_drm_device *dc_drm)
{
        struct dc_pe *pe = dc_drm->pe;
        int i;

        for (i = 0; i < DC_DISPLAYS; i++) {
                pe->cf_safe[i] = dc_drm->cf_safe[i];
                pe->cf_cont[i] = dc_drm->cf_cont[i];
                pe->ed_safe[i] = dc_drm->ed_safe[i];
                pe->ed_cont[i] = dc_drm->ed_cont[i];
        }

        for (i = 0; i < DC_DISP_FU_CNT; i++)
                pe->fu_disp[i] = dc_drm->fu_disp[i];

        for (i = 0; i < DC_LB_CNT; i++)
                pe->lb[i] = dc_drm->lb[i];
}

static const struct component_ops dc_pe_ops = {
        .bind = dc_pe_bind,
};

static int dc_pe_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_pe_ops);
        if (ret)
                return dev_err_probe(&pdev->dev, ret,
                                     "failed to add component\n");

        return 0;
}

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

static int dc_pe_runtime_suspend(struct device *dev)
{
        struct dc_pe *pe = dev_get_drvdata(dev);

        clk_disable_unprepare(pe->clk_axi);

        return 0;
}

static int dc_pe_runtime_resume(struct device *dev)
{
        struct dc_pe *pe = dev_get_drvdata(dev);
        int i, ret;

        ret = clk_prepare_enable(pe->clk_axi);
        if (ret) {
                dev_err(dev, "failed to enable AXI clock: %d\n", ret);
                return ret;
        }

        for (i = 0; i < ARRAY_SIZE(pe->cf_safe); i++)
                dc_cf_init(pe->cf_safe[i]);

        for (i = 0; i < ARRAY_SIZE(pe->cf_cont); i++)
                dc_cf_init(pe->cf_cont[i]);

        for (i = 0; i < ARRAY_SIZE(pe->ed_safe); i++)
                dc_ed_init(pe->ed_safe[i]);

        for (i = 0; i < ARRAY_SIZE(pe->ed_cont); i++)
                dc_ed_init(pe->ed_cont[i]);

        for (i = 0; i < ARRAY_SIZE(pe->fu_disp); i++)
                pe->fu_disp[i]->ops.init(pe->fu_disp[i]);

        for (i = 0; i < ARRAY_SIZE(pe->lb); i++)
                dc_lb_init(pe->lb[i]);

        return 0;
}

static const struct dev_pm_ops dc_pe_pm_ops = {
        RUNTIME_PM_OPS(dc_pe_runtime_suspend, dc_pe_runtime_resume, NULL)
};

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

struct platform_driver dc_pe_driver = {
        .probe = dc_pe_probe,
        .remove = dc_pe_remove,
        .driver = {
                .name = "imx8-dc-pixel-engine",
                .suppress_bind_attrs = true,
                .of_match_table = dc_pe_dt_ids,
                .pm = pm_sleep_ptr(&dc_pe_pm_ops),
        },
};