root/drivers/gpu/drm/hisilicon/hibmc/hibmc_drm_dp.c
// SPDX-License-Identifier: GPL-2.0-or-later
// Copyright (c) 2024 Hisilicon Limited.

#include <linux/io.h>

#include <drm/drm_probe_helper.h>
#include <drm/drm_simple_kms_helper.h>
#include <drm/drm_atomic_helper.h>
#include <drm/drm_modes.h>
#include <drm/drm_drv.h>
#include <drm/drm_edid.h>

#include "hibmc_drm_drv.h"
#include "dp/dp_hw.h"
#include "dp/dp_comm.h"
#include "dp/dp_config.h"

#define DP_MASKED_SINK_HPD_PLUG_INT     BIT(2)

static int hibmc_dp_connector_get_modes(struct drm_connector *connector)
{
        const struct drm_edid *drm_edid;
        int count;

        drm_edid = drm_edid_read(connector);

        drm_edid_connector_update(connector, drm_edid);

        count = drm_edid_connector_add_modes(connector);

        drm_edid_free(drm_edid);

        return count;
}

static bool hibmc_dp_get_dpcd(struct hibmc_dp_dev *dp_dev)
{
        int ret;

        ret = drm_dp_read_dpcd_caps(dp_dev->aux, dp_dev->dpcd);
        if (ret)
                return false;

        dp_dev->is_branch = drm_dp_is_branch(dp_dev->dpcd);

        ret = drm_dp_read_desc(dp_dev->aux, &dp_dev->desc, dp_dev->is_branch);
        if (ret)
                return false;

        ret = drm_dp_read_downstream_info(dp_dev->aux, dp_dev->dpcd, dp_dev->downstream_ports);
        if (ret)
                return false;

        return true;
}

static int hibmc_dp_detect(struct drm_connector *connector,
                           struct drm_modeset_acquire_ctx *ctx, bool force)
{
        struct hibmc_dp *dp = to_hibmc_dp(connector);
        struct hibmc_dp_dev *dp_dev = dp->dp_dev;
        int ret;

        if (dp->irq_status) {
                if (dp_dev->hpd_status != HIBMC_HPD_IN)
                        return connector_status_disconnected;
        }

        if (!hibmc_dp_get_dpcd(dp_dev))
                return connector_status_disconnected;

        if (!dp_dev->is_branch)
                return connector_status_connected;

        if (drm_dp_read_sink_count_cap(connector, dp_dev->dpcd, &dp_dev->desc) &&
            dp_dev->downstream_ports[0] & DP_DS_PORT_HPD) {
                ret = drm_dp_read_sink_count(dp_dev->aux);
                if (ret > 0)
                        return connector_status_connected;
        }

        return connector_status_disconnected;
}

static int hibmc_dp_mode_valid(struct drm_connector *connector,
                               const struct drm_display_mode *mode,
                               struct drm_modeset_acquire_ctx *ctx,
                               enum drm_mode_status *status)
{
        struct hibmc_dp *dp = to_hibmc_dp(connector);
        u64 cur_val, max_val;

        /* check DP link BW */
        cur_val = (u64)mode->clock * HIBMC_DP_BPP;
        max_val = (u64)hibmc_dp_get_link_rate(dp) * DP_MODE_VALI_CAL * hibmc_dp_get_lanes(dp);

        *status = cur_val > max_val ? MODE_CLOCK_HIGH : MODE_OK;

        return 0;
}

static const struct drm_connector_helper_funcs hibmc_dp_conn_helper_funcs = {
        .get_modes = hibmc_dp_connector_get_modes,
        .detect_ctx = hibmc_dp_detect,
        .mode_valid_ctx = hibmc_dp_mode_valid,
};

static int hibmc_dp_late_register(struct drm_connector *connector)
{
        struct hibmc_dp *dp = to_hibmc_dp(connector);

        hibmc_dp_enable_int(dp);

        return drm_dp_aux_register(&dp->aux);
}

static void hibmc_dp_early_unregister(struct drm_connector *connector)
{
        struct hibmc_dp *dp = to_hibmc_dp(connector);

        drm_dp_aux_unregister(&dp->aux);

        hibmc_dp_disable_int(dp);
}

static const struct drm_connector_funcs hibmc_dp_conn_funcs = {
        .reset = drm_atomic_helper_connector_reset,
        .fill_modes = drm_helper_probe_single_connector_modes,
        .destroy = drm_connector_cleanup,
        .atomic_duplicate_state = drm_atomic_helper_connector_duplicate_state,
        .atomic_destroy_state = drm_atomic_helper_connector_destroy_state,
        .late_register = hibmc_dp_late_register,
        .early_unregister = hibmc_dp_early_unregister,
        .debugfs_init = hibmc_debugfs_init,
};

static inline int hibmc_dp_prepare(struct hibmc_dp *dp, struct drm_display_mode *mode)
{
        int ret;

        hibmc_dp_display_en(dp, false);

        ret = hibmc_dp_mode_set(dp, mode);
        if (ret)
                drm_err(dp->drm_dev, "hibmc dp mode set failed: %d\n", ret);

        return ret;
}

static void hibmc_dp_encoder_enable(struct drm_encoder *drm_encoder,
                                    struct drm_atomic_state *state)
{
        struct hibmc_dp *dp = container_of(drm_encoder, struct hibmc_dp, encoder);
        struct drm_display_mode *mode = &drm_encoder->crtc->state->mode;

        if (hibmc_dp_prepare(dp, mode))
                return;

        hibmc_dp_display_en(dp, true);
}

static void hibmc_dp_encoder_disable(struct drm_encoder *drm_encoder,
                                     struct drm_atomic_state *state)
{
        struct hibmc_dp *dp = container_of(drm_encoder, struct hibmc_dp, encoder);

        hibmc_dp_display_en(dp, false);
}

static const struct drm_encoder_helper_funcs hibmc_dp_encoder_helper_funcs = {
        .atomic_enable = hibmc_dp_encoder_enable,
        .atomic_disable = hibmc_dp_encoder_disable,
};

irqreturn_t hibmc_dp_hpd_isr(int irq, void *arg)
{
        struct drm_device *dev = (struct drm_device *)arg;
        struct hibmc_drm_private *priv = to_hibmc_drm_private(dev);
        int idx, exp_status;

        if (!drm_dev_enter(dev, &idx))
                return -ENODEV;

        if (priv->dp.irq_status & DP_MASKED_SINK_HPD_PLUG_INT) {
                drm_dbg_dp(&priv->dev, "HPD IN isr occur!\n");
                hibmc_dp_hpd_cfg(&priv->dp);
                exp_status = HIBMC_HPD_IN;
        } else {
                drm_dbg_dp(&priv->dev, "HPD OUT isr occur!\n");
                hibmc_dp_reset_link(&priv->dp);
                exp_status = HIBMC_HPD_OUT;
        }

        if (hibmc_dp_check_hpd_status(&priv->dp, exp_status))
                drm_connector_helper_hpd_irq_event(&priv->dp.connector);

        drm_dev_exit(idx);

        return IRQ_HANDLED;
}

int hibmc_dp_init(struct hibmc_drm_private *priv)
{
        struct drm_device *dev = &priv->dev;
        struct drm_crtc *crtc = &priv->crtc;
        struct hibmc_dp *dp = &priv->dp;
        struct drm_connector *connector = &dp->connector;
        struct drm_encoder *encoder = &dp->encoder;
        int ret;

        dp->mmio = priv->mmio;
        dp->drm_dev = dev;

        ret = hibmc_dp_hw_init(&priv->dp);
        if (ret) {
                drm_err(dev, "hibmc dp hw init failed: %d\n", ret);
                return ret;
        }

        hibmc_dp_display_en(&priv->dp, false);

        encoder->possible_crtcs = drm_crtc_mask(crtc);
        ret = drmm_encoder_init(dev, encoder, NULL, DRM_MODE_ENCODER_TMDS, NULL);
        if (ret) {
                drm_err(dev, "init dp encoder failed: %d\n", ret);
                return ret;
        }

        drm_encoder_helper_add(encoder, &hibmc_dp_encoder_helper_funcs);

        ret = drm_connector_init_with_ddc(dev, connector, &hibmc_dp_conn_funcs,
                                          DRM_MODE_CONNECTOR_DisplayPort, &dp->aux.ddc);
        if (ret) {
                drm_err(dev, "init dp connector failed: %d\n", ret);
                return ret;
        }

        drm_connector_helper_add(connector, &hibmc_dp_conn_helper_funcs);

        drm_connector_attach_encoder(connector, encoder);

        connector->polled = DRM_CONNECTOR_POLL_HPD;

        return 0;
}