// SPDX-License-Identifier: GPL-2.0
/*
 * (C) COPYRIGHT 2018 ARM Limited. All rights reserved.
 * Author: James.Qian.Wang <james.qian.wang@arm.com>
 *
 */
#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/of.h>
#include <linux/platform_device.h>
#include <linux/pm_runtime.h>
#include <drm/clients/drm_client_setup.h>
#include <drm/drm_module.h>
#include <drm/drm_of.h>
#include "komeda_dev.h"
#include "komeda_kms.h"

struct komeda_drv {
        struct komeda_dev *mdev;
        struct komeda_kms_dev *kms;
};

struct komeda_dev *dev_to_mdev(struct device *dev)
{
        struct komeda_drv *mdrv = dev_get_drvdata(dev);

        return mdrv ? mdrv->mdev : NULL;
}

static void komeda_platform_remove(struct platform_device *pdev)
{
        struct device *dev = &pdev->dev;
        struct komeda_drv *mdrv = dev_get_drvdata(dev);

        komeda_kms_detach(mdrv->kms);

        if (pm_runtime_enabled(dev))
                pm_runtime_disable(dev);
        else
                komeda_dev_suspend(mdrv->mdev);

        komeda_dev_destroy(mdrv->mdev);

        dev_set_drvdata(dev, NULL);
        devm_kfree(dev, mdrv);
}

static void komeda_platform_shutdown(struct platform_device *pdev)
{
        struct device *dev = &pdev->dev;
        struct komeda_drv *mdrv = dev_get_drvdata(dev);

        komeda_kms_shutdown(mdrv->kms);
}

static int komeda_platform_probe(struct platform_device *pdev)
{
        struct device *dev = &pdev->dev;
        struct komeda_drv *mdrv;
        int err;

        err = dma_set_mask_and_coherent(dev, DMA_BIT_MASK(40));
        if (err)
                return dev_err_probe(dev, err, "DMA mask error\n");

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

        mdrv->mdev = komeda_dev_create(dev);
        if (IS_ERR(mdrv->mdev)) {
                err = PTR_ERR(mdrv->mdev);
                goto free_mdrv;
        }

        pm_runtime_enable(dev);
        if (!pm_runtime_enabled(dev))
                komeda_dev_resume(mdrv->mdev);

        mdrv->kms = komeda_kms_attach(mdrv->mdev);
        if (IS_ERR(mdrv->kms)) {
                err = PTR_ERR(mdrv->kms);
                goto destroy_mdev;
        }

        dev_set_drvdata(dev, mdrv);
        drm_client_setup(&mdrv->kms->base, NULL);

        return 0;

destroy_mdev:
        if (pm_runtime_enabled(dev))
                pm_runtime_disable(dev);
        else
                komeda_dev_suspend(mdrv->mdev);

        komeda_dev_destroy(mdrv->mdev);

free_mdrv:
        devm_kfree(dev, mdrv);
        return err;
}

static const struct of_device_id komeda_of_match[] = {
        { .compatible = "arm,mali-d71", .data = d71_identify, },
        { .compatible = "arm,mali-d32", .data = d71_identify, },
        {},
};

MODULE_DEVICE_TABLE(of, komeda_of_match);

static int __maybe_unused komeda_rt_pm_suspend(struct device *dev)
{
        struct komeda_drv *mdrv = dev_get_drvdata(dev);

        return komeda_dev_suspend(mdrv->mdev);
}

static int __maybe_unused komeda_rt_pm_resume(struct device *dev)
{
        struct komeda_drv *mdrv = dev_get_drvdata(dev);

        return komeda_dev_resume(mdrv->mdev);
}

static int __maybe_unused komeda_pm_suspend(struct device *dev)
{
        struct komeda_drv *mdrv = dev_get_drvdata(dev);
        int res;

        res = drm_mode_config_helper_suspend(&mdrv->kms->base);

        if (!pm_runtime_status_suspended(dev))
                komeda_dev_suspend(mdrv->mdev);

        return res;
}

static int __maybe_unused komeda_pm_resume(struct device *dev)
{
        struct komeda_drv *mdrv = dev_get_drvdata(dev);

        if (!pm_runtime_status_suspended(dev))
                komeda_dev_resume(mdrv->mdev);

        return drm_mode_config_helper_resume(&mdrv->kms->base);
}

static const struct dev_pm_ops komeda_pm_ops = {
        SET_SYSTEM_SLEEP_PM_OPS(komeda_pm_suspend, komeda_pm_resume)
        SET_RUNTIME_PM_OPS(komeda_rt_pm_suspend, komeda_rt_pm_resume, NULL)
};

static struct platform_driver komeda_platform_driver = {
        .probe  = komeda_platform_probe,
        .remove = komeda_platform_remove,
        .shutdown = komeda_platform_shutdown,
        .driver = {
                .name = "komeda",
                .of_match_table = komeda_of_match,
                .pm = &komeda_pm_ops,
        },
};

drm_module_platform_driver(komeda_platform_driver);

MODULE_AUTHOR("James.Qian.Wang <james.qian.wang@arm.com>");
MODULE_DESCRIPTION("Komeda KMS driver");
MODULE_LICENSE("GPL v2");