// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright 2019 NXP.
 */

#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/of.h>
#include <linux/platform_device.h>
#include <drm/drm_module.h>
#include <drm/drm_of.h>

#include "dcss-dev.h"
#include "dcss-kms.h"

struct dcss_drv {
        struct dcss_dev *dcss;
        struct dcss_kms_dev *kms;
};

struct dcss_dev *dcss_drv_dev_to_dcss(struct device *dev)
{
        struct dcss_drv *mdrv = dev_get_drvdata(dev);

        return mdrv ? mdrv->dcss : NULL;
}

struct drm_device *dcss_drv_dev_to_drm(struct device *dev)
{
        struct dcss_drv *mdrv = dev_get_drvdata(dev);

        return mdrv ? &mdrv->kms->base : NULL;
}

static int dcss_drv_platform_probe(struct platform_device *pdev)
{
        struct device *dev = &pdev->dev;
        struct device_node *remote;
        struct dcss_drv *mdrv;
        int err = 0;
        bool hdmi_output = true;

        if (!dev->of_node)
                return -ENODEV;

        remote = of_graph_get_remote_node(dev->of_node, 0, 0);
        if (!remote)
                return -ENODEV;

        hdmi_output = !of_device_is_compatible(remote, "fsl,imx8mq-nwl-dsi");

        of_node_put(remote);

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

        mdrv->dcss = dcss_dev_create(dev, hdmi_output);
        if (IS_ERR(mdrv->dcss))
                return PTR_ERR(mdrv->dcss);

        dev_set_drvdata(dev, mdrv);

        mdrv->kms = dcss_kms_attach(mdrv->dcss);
        if (IS_ERR(mdrv->kms)) {
                err = PTR_ERR(mdrv->kms);
                dev_err_probe(dev, err, "Failed to initialize KMS\n");
                goto dcss_shutoff;
        }

        return 0;

dcss_shutoff:
        dcss_dev_destroy(mdrv->dcss);

        return err;
}

static void dcss_drv_platform_remove(struct platform_device *pdev)
{
        struct dcss_drv *mdrv = dev_get_drvdata(&pdev->dev);

        dcss_kms_detach(mdrv->kms);
        dcss_dev_destroy(mdrv->dcss);
}

static void dcss_drv_platform_shutdown(struct platform_device *pdev)
{
        struct dcss_drv *mdrv = dev_get_drvdata(&pdev->dev);

        dcss_kms_shutdown(mdrv->kms);
}

static struct dcss_type_data dcss_types[] = {
        [DCSS_IMX8MQ] = {
                .name = "DCSS_IMX8MQ",
                .blkctl_ofs = 0x2F000,
                .ctxld_ofs = 0x23000,
                .dtg_ofs = 0x20000,
                .scaler_ofs = 0x1C000,
                .ss_ofs = 0x1B000,
                .dpr_ofs = 0x18000,
        },
};

static const struct of_device_id dcss_of_match[] = {
        { .compatible = "nxp,imx8mq-dcss", .data = &dcss_types[DCSS_IMX8MQ], },
        {},
};

MODULE_DEVICE_TABLE(of, dcss_of_match);

static struct platform_driver dcss_platform_driver = {
        .probe  = dcss_drv_platform_probe,
        .remove = dcss_drv_platform_remove,
        .shutdown = dcss_drv_platform_shutdown,
        .driver = {
                .name = "imx-dcss",
                .of_match_table = dcss_of_match,
                .pm = pm_ptr(&dcss_dev_pm_ops),
        },
};

drm_module_platform_driver(dcss_platform_driver);

MODULE_AUTHOR("Laurentiu Palcu <laurentiu.palcu@nxp.com>");
MODULE_DESCRIPTION("DCSS driver for i.MX8MQ");
MODULE_LICENSE("GPL v2");