// SPDX-License-Identifier: GPL-2.0-only
/*
 * Copyright (c) 2014 The Linux Foundation. All rights reserved.
 * Copyright (C) 2013 Red Hat
 * Author: Rob Clark <robdclark@gmail.com>
 */

#include <linux/gpio/consumer.h>
#include <linux/of_irq.h>
#include <linux/of_platform.h>
#include <linux/pinctrl/consumer.h>
#include <linux/platform_device.h>

#include <drm/drm_bridge_connector.h>
#include <drm/drm_of.h>
#include <drm/display/drm_hdmi_state_helper.h>

#include "msm_kms.h"
#include "hdmi.h"

void msm_hdmi_set_mode(struct hdmi *hdmi, bool power_on)
{
        u32 ctrl = 0;
        unsigned long flags;

        spin_lock_irqsave(&hdmi->reg_lock, flags);
        if (power_on) {
                ctrl |= HDMI_CTRL_ENABLE;
                if (!hdmi->connector->display_info.is_hdmi) {
                        ctrl |= HDMI_CTRL_HDMI;
                        hdmi_write(hdmi, REG_HDMI_CTRL, ctrl);
                        ctrl &= ~HDMI_CTRL_HDMI;
                } else {
                        ctrl |= HDMI_CTRL_HDMI;
                }
        } else {
                ctrl = HDMI_CTRL_HDMI;
        }

        hdmi_write(hdmi, REG_HDMI_CTRL, ctrl);
        spin_unlock_irqrestore(&hdmi->reg_lock, flags);
        DBG("HDMI Core: %s, HDMI_CTRL=0x%08x",
                        power_on ? "Enable" : "Disable", ctrl);
}

static irqreturn_t msm_hdmi_irq(int irq, void *dev_id)
{
        struct hdmi *hdmi = dev_id;

        /* Process HPD: */
        msm_hdmi_hpd_irq(hdmi->bridge);

        /* Process DDC: */
        msm_hdmi_i2c_irq(hdmi->i2c);

        /* Process HDCP: */
        if (hdmi->hdcp_ctrl)
                msm_hdmi_hdcp_irq(hdmi->hdcp_ctrl);

        /* TODO audio.. */

        return IRQ_HANDLED;
}

static void msm_hdmi_destroy(struct hdmi *hdmi)
{
        /*
         * at this point, hpd has been disabled,
         * after flush workq, it's safe to deinit hdcp
         */
        if (hdmi->workq)
                destroy_workqueue(hdmi->workq);
        msm_hdmi_hdcp_destroy(hdmi);

        if (hdmi->i2c)
                msm_hdmi_i2c_destroy(hdmi->i2c);
}

static void msm_hdmi_put_phy(struct hdmi *hdmi)
{
        if (hdmi->phy_dev) {
                put_device(hdmi->phy_dev);
                hdmi->phy = NULL;
                hdmi->phy_dev = NULL;
        }
}

static int msm_hdmi_get_phy(struct hdmi *hdmi)
{
        struct platform_device *pdev = hdmi->pdev;
        struct platform_device *phy_pdev;
        struct device_node *phy_node;

        phy_node = of_parse_phandle(dev_of_node(&pdev->dev), "phys", 0);
        if (!phy_node) {
                DRM_DEV_ERROR(&pdev->dev, "cannot find phy device\n");
                return -ENXIO;
        }

        phy_pdev = of_find_device_by_node(phy_node);
        of_node_put(phy_node);

        if (!phy_pdev)
                return dev_err_probe(&pdev->dev, -EPROBE_DEFER, "phy driver is not ready\n");

        hdmi->phy = platform_get_drvdata(phy_pdev);
        if (!hdmi->phy) {
                put_device(&phy_pdev->dev);
                return dev_err_probe(&pdev->dev, -EPROBE_DEFER, "phy driver is not ready\n");
        }

        hdmi->phy_dev = &phy_pdev->dev;

        return 0;
}

/* construct hdmi at bind/probe time, grab all the resources.  If
 * we are to EPROBE_DEFER we want to do it here, rather than later
 * at modeset_init() time
 */
static int msm_hdmi_init(struct hdmi *hdmi)
{
        struct platform_device *pdev = hdmi->pdev;
        int ret;

        hdmi->workq = alloc_ordered_workqueue("msm_hdmi", 0);
        if (!hdmi->workq) {
                ret = -ENOMEM;
                goto fail;
        }

        hdmi->i2c = msm_hdmi_i2c_init(hdmi);
        if (IS_ERR(hdmi->i2c)) {
                ret = PTR_ERR(hdmi->i2c);
                DRM_DEV_ERROR(&pdev->dev, "failed to get i2c: %d\n", ret);
                hdmi->i2c = NULL;
                goto fail;
        }

        hdmi->hdcp_ctrl = msm_hdmi_hdcp_init(hdmi);
        if (IS_ERR(hdmi->hdcp_ctrl)) {
                dev_warn(&pdev->dev, "failed to init hdcp: disabled\n");
                hdmi->hdcp_ctrl = NULL;
        }

        return 0;

fail:
        msm_hdmi_destroy(hdmi);

        return ret;
}

/* Second part of initialization, the drm/kms level modeset_init,
 * constructs/initializes mode objects, etc, is called from master
 * driver (not hdmi sub-device's probe/bind!)
 *
 * Any resource (regulator/clk/etc) which could be missing at boot
 * should be handled in msm_hdmi_init() so that failure happens from
 * hdmi sub-device's probe.
 */
int msm_hdmi_modeset_init(struct hdmi *hdmi,
                struct drm_device *dev, struct drm_encoder *encoder)
{
        int ret;

        hdmi->dev = dev;
        hdmi->encoder = encoder;

        ret = msm_hdmi_bridge_init(hdmi);
        if (ret) {
                DRM_DEV_ERROR(dev->dev, "failed to create HDMI bridge: %d\n", ret);
                goto fail;
        }

        if (hdmi->next_bridge) {
                ret = drm_bridge_attach(hdmi->encoder, hdmi->next_bridge, hdmi->bridge,
                                        DRM_BRIDGE_ATTACH_NO_CONNECTOR);
                if (ret) {
                        DRM_DEV_ERROR(dev->dev, "failed to attach next HDMI bridge: %d\n", ret);
                        goto fail;
                }
        }

        hdmi->connector = drm_bridge_connector_init(hdmi->dev, encoder);
        if (IS_ERR(hdmi->connector)) {
                ret = PTR_ERR(hdmi->connector);
                DRM_DEV_ERROR(dev->dev, "failed to create HDMI connector: %d\n", ret);
                hdmi->connector = NULL;
                goto fail;
        }

        ret = devm_request_irq(dev->dev, hdmi->irq,
                        msm_hdmi_irq, IRQF_TRIGGER_HIGH,
                        "hdmi_isr", hdmi);
        if (ret < 0) {
                DRM_DEV_ERROR(dev->dev, "failed to request IRQ%u: %d\n",
                                hdmi->irq, ret);
                goto fail;
        }

        return 0;

fail:
        if (hdmi->connector) {
                hdmi->connector->funcs->destroy(hdmi->connector);
                hdmi->connector = NULL;
        }

        return ret;
}

/*
 * The hdmi device:
 */

static const char * const pwr_reg_names_8960[] = {"core-vdda"};
static const char * const pwr_clk_names_8960[] = {"core", "master_iface", "slave_iface"};

static const struct hdmi_platform_config hdmi_tx_8960_config = {
        .pwr_reg_names = pwr_reg_names_8960,
        .pwr_reg_cnt = ARRAY_SIZE(pwr_reg_names_8960),
        .pwr_clk_names = pwr_clk_names_8960,
        .pwr_clk_cnt = ARRAY_SIZE(pwr_clk_names_8960),
};

static const char * const pwr_reg_names_8x74[] = {"core-vdda", "core-vcc"};
static const char * const pwr_clk_names_8x74[] = {"iface", "core", "mdp_core", "alt_iface"};

static const struct hdmi_platform_config hdmi_tx_8974_config = {
        .pwr_reg_names = pwr_reg_names_8x74,
        .pwr_reg_cnt = ARRAY_SIZE(pwr_reg_names_8x74),
        .pwr_clk_names = pwr_clk_names_8x74,
        .pwr_clk_cnt = ARRAY_SIZE(pwr_clk_names_8x74),
};

static int msm_hdmi_bind(struct device *dev, struct device *master, void *data)
{
        struct msm_drm_private *priv = dev_get_drvdata(master);
        struct hdmi *hdmi = dev_get_drvdata(dev);
        int err;

        err = msm_hdmi_init(hdmi);
        if (err)
                return err;
        priv->kms->hdmi = hdmi;

        return 0;
}

static void msm_hdmi_unbind(struct device *dev, struct device *master,
                void *data)
{
        struct msm_drm_private *priv = dev_get_drvdata(master);

        if (priv->kms->hdmi) {
                msm_hdmi_destroy(priv->kms->hdmi);
                priv->kms->hdmi = NULL;
        }
}

static const struct component_ops msm_hdmi_ops = {
                .bind   = msm_hdmi_bind,
                .unbind = msm_hdmi_unbind,
};

static int msm_hdmi_dev_probe(struct platform_device *pdev)
{
        const struct hdmi_platform_config *config;
        struct device *dev = &pdev->dev;
        struct hdmi *hdmi;
        struct resource *res;
        int i, ret;

        config = of_device_get_match_data(dev);
        if (!config)
                return -EINVAL;

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

        hdmi->pdev = pdev;
        hdmi->config = config;
        spin_lock_init(&hdmi->reg_lock);
        mutex_init(&hdmi->state_mutex);

        hdmi->next_bridge = of_drm_get_bridge_by_endpoint(dev_of_node(dev), 1, 0);
        if (IS_ERR(hdmi->next_bridge)) {
                if (PTR_ERR(hdmi->next_bridge) != -ENODEV)
                        return PTR_ERR(hdmi->next_bridge);

                hdmi->next_bridge = NULL;
        }

        hdmi->mmio = msm_ioremap(pdev, "core_physical");
        if (IS_ERR(hdmi->mmio)) {
                ret = PTR_ERR(hdmi->mmio);
                goto err_put_bridge;
        }

        /* HDCP needs physical address of hdmi register */
        res = platform_get_resource_byname(pdev, IORESOURCE_MEM,
                "core_physical");
        if (!res) {
                ret = -EINVAL;
                goto err_put_bridge;
        }
        hdmi->mmio_phy_addr = res->start;

        hdmi->qfprom_mmio = msm_ioremap(pdev, "qfprom_physical");
        if (IS_ERR(hdmi->qfprom_mmio)) {
                DRM_DEV_INFO(dev, "can't find qfprom resource\n");
                hdmi->qfprom_mmio = NULL;
        }

        hdmi->irq = platform_get_irq(pdev, 0);
        if (hdmi->irq < 0) {
                ret = hdmi->irq;
                goto err_put_bridge;
        }

        hdmi->pwr_regs = devm_kcalloc(dev, config->pwr_reg_cnt,
                                      sizeof(hdmi->pwr_regs[0]),
                                      GFP_KERNEL);
        if (!hdmi->pwr_regs) {
                ret = -ENOMEM;
                goto err_put_bridge;
        }

        for (i = 0; i < config->pwr_reg_cnt; i++)
                hdmi->pwr_regs[i].supply = config->pwr_reg_names[i];

        ret = devm_regulator_bulk_get(dev, config->pwr_reg_cnt, hdmi->pwr_regs);
        if (ret) {
                dev_err_probe(dev, ret, "failed to get pwr regulators\n");
                goto err_put_bridge;
        }

        hdmi->pwr_clks = devm_kcalloc(dev, config->pwr_clk_cnt,
                                      sizeof(hdmi->pwr_clks[0]),
                                      GFP_KERNEL);
        if (!hdmi->pwr_clks) {
                ret = -ENOMEM;
                goto err_put_bridge;
        }

        for (i = 0; i < config->pwr_clk_cnt; i++)
                hdmi->pwr_clks[i].id = config->pwr_clk_names[i];

        ret = devm_clk_bulk_get(dev, config->pwr_clk_cnt, hdmi->pwr_clks);
        if (ret)
                goto err_put_bridge;


        hdmi->extp_clk = devm_clk_get_optional(dev, "extp");
        if (IS_ERR(hdmi->extp_clk)) {
                ret = dev_err_probe(dev, PTR_ERR(hdmi->extp_clk),
                                    "failed to get extp clock\n");
                goto err_put_bridge;
        }

        hdmi->hpd_gpiod = devm_gpiod_get_optional(dev, "hpd", GPIOD_IN);
        /* This will catch e.g. -EPROBE_DEFER */
        if (IS_ERR(hdmi->hpd_gpiod)) {
                ret = dev_err_probe(dev, PTR_ERR(hdmi->hpd_gpiod),
                                    "failed to get hpd gpio\n");
                goto err_put_bridge;
        }

        if (!hdmi->hpd_gpiod)
                DBG("failed to get HPD gpio");

        if (hdmi->hpd_gpiod)
                gpiod_set_consumer_name(hdmi->hpd_gpiod, "HDMI_HPD");

        ret = msm_hdmi_get_phy(hdmi);
        if (ret)
                goto err_put_bridge;

        ret = devm_pm_runtime_enable(dev);
        if (ret)
                goto err_put_phy;

        platform_set_drvdata(pdev, hdmi);

        ret = component_add(dev, &msm_hdmi_ops);
        if (ret)
                goto err_put_phy;

        return 0;

err_put_phy:
        msm_hdmi_put_phy(hdmi);
err_put_bridge:
        drm_bridge_put(hdmi->next_bridge);
        return ret;
}

static void msm_hdmi_dev_remove(struct platform_device *pdev)
{
        struct hdmi *hdmi = dev_get_drvdata(&pdev->dev);

        component_del(&pdev->dev, &msm_hdmi_ops);

        msm_hdmi_put_phy(hdmi);
        drm_bridge_put(hdmi->next_bridge);
}

static int msm_hdmi_runtime_suspend(struct device *dev)
{
        struct hdmi *hdmi = dev_get_drvdata(dev);
        const struct hdmi_platform_config *config = hdmi->config;

        clk_bulk_disable_unprepare(config->pwr_clk_cnt, hdmi->pwr_clks);

        pinctrl_pm_select_sleep_state(dev);

        regulator_bulk_disable(config->pwr_reg_cnt, hdmi->pwr_regs);

        return 0;
}

static int msm_hdmi_runtime_resume(struct device *dev)
{
        struct hdmi *hdmi = dev_get_drvdata(dev);
        const struct hdmi_platform_config *config = hdmi->config;
        int ret;

        ret = regulator_bulk_enable(config->pwr_reg_cnt, hdmi->pwr_regs);
        if (ret)
                return ret;

        ret = pinctrl_pm_select_default_state(dev);
        if (ret)
                goto fail;

        ret = clk_bulk_prepare_enable(config->pwr_clk_cnt, hdmi->pwr_clks);
        if (ret)
                goto fail;

        return 0;

fail:
        pinctrl_pm_select_sleep_state(dev);

        return ret;
}

static DEFINE_RUNTIME_DEV_PM_OPS(msm_hdmi_pm_ops, msm_hdmi_runtime_suspend, msm_hdmi_runtime_resume, NULL);

static const struct of_device_id msm_hdmi_dt_match[] = {
        { .compatible = "qcom,hdmi-tx-8998", .data = &hdmi_tx_8974_config },
        { .compatible = "qcom,hdmi-tx-8996", .data = &hdmi_tx_8974_config },
        { .compatible = "qcom,hdmi-tx-8994", .data = &hdmi_tx_8974_config },
        { .compatible = "qcom,hdmi-tx-8084", .data = &hdmi_tx_8974_config },
        { .compatible = "qcom,hdmi-tx-8974", .data = &hdmi_tx_8974_config },
        { .compatible = "qcom,hdmi-tx-8960", .data = &hdmi_tx_8960_config },
        { .compatible = "qcom,hdmi-tx-8660", .data = &hdmi_tx_8960_config },
        {}
};
MODULE_DEVICE_TABLE(of, msm_hdmi_dt_match);

static struct platform_driver msm_hdmi_driver = {
        .probe = msm_hdmi_dev_probe,
        .remove = msm_hdmi_dev_remove,
        .driver = {
                .name = "hdmi_msm",
                .of_match_table = msm_hdmi_dt_match,
                .pm = &msm_hdmi_pm_ops,
        },
};

void __init msm_hdmi_register(void)
{
        msm_hdmi_phy_driver_register();
        platform_driver_register(&msm_hdmi_driver);
}

void __exit msm_hdmi_unregister(void)
{
        platform_driver_unregister(&msm_hdmi_driver);
        msm_hdmi_phy_driver_unregister();
}