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

#include <linux/of.h>
#include <linux/of_graph.h>

#include <drm/drm_atomic_helper.h>
#include <drm/drm_bridge.h>
#include <drm/drm_bridge_connector.h>
#include <drm/drm_connector.h>
#include <drm/drm_crtc.h>
#include <drm/drm_device.h>
#include <drm/drm_encoder.h>
#include <drm/drm_gem_framebuffer_helper.h>
#include <drm/drm_mode_config.h>
#include <drm/drm_print.h>
#include <drm/drm_probe_helper.h>
#include <drm/drm_simple_kms_helper.h>
#include <drm/drm_vblank.h>

#include "dc-de.h"
#include "dc-drv.h"
#include "dc-kms.h"

static const struct drm_mode_config_funcs dc_drm_mode_config_funcs = {
        .fb_create = drm_gem_fb_create,
        .atomic_check = drm_atomic_helper_check,
        .atomic_commit = drm_atomic_helper_commit,
};

static int dc_kms_init_encoder_per_crtc(struct dc_drm_device *dc_drm,
                                        int crtc_index)
{
        struct dc_crtc *dc_crtc = &dc_drm->dc_crtc[crtc_index];
        struct drm_device *drm = &dc_drm->base;
        struct drm_crtc *crtc = &dc_crtc->base;
        struct drm_connector *connector;
        struct device *dev = drm->dev;
        struct drm_encoder *encoder;
        struct drm_bridge *bridge;
        int ret;

        bridge = devm_drm_of_get_bridge(dev, dc_crtc->de->tc->dev->of_node,
                                        0, 0);
        if (IS_ERR(bridge)) {
                ret = PTR_ERR(bridge);
                if (ret == -ENODEV)
                        return 0;

                return dev_err_probe(dev, ret,
                                     "failed to find bridge for CRTC%u\n",
                                     crtc->index);
        }

        encoder = &dc_drm->encoder[crtc_index];
        ret = drm_simple_encoder_init(drm, encoder, DRM_MODE_ENCODER_NONE);
        if (ret) {
                dev_err(dev, "failed to initialize encoder for CRTC%u: %d\n",
                        crtc->index, ret);
                return ret;
        }

        encoder->possible_crtcs = drm_crtc_mask(crtc);

        ret = drm_bridge_attach(encoder, bridge, NULL,
                                DRM_BRIDGE_ATTACH_NO_CONNECTOR);
        if (ret) {
                dev_err(dev,
                        "failed to attach bridge to encoder for CRTC%u: %d\n",
                        crtc->index, ret);
                return ret;
        }

        connector = drm_bridge_connector_init(drm, encoder);
        if (IS_ERR(connector)) {
                ret = PTR_ERR(connector);
                dev_err(dev, "failed to init bridge connector for CRTC%u: %d\n",
                        crtc->index, ret);
                return ret;
        }

        ret = drm_connector_attach_encoder(connector, encoder);
        if (ret)
                dev_err(dev,
                        "failed to attach encoder to connector for CRTC%u: %d\n",
                        crtc->index, ret);

        return ret;
}

int dc_kms_init(struct dc_drm_device *dc_drm)
{
        struct drm_device *drm = &dc_drm->base;
        int ret, i;

        ret = drmm_mode_config_init(drm);
        if (ret)
                return ret;

        drm->mode_config.min_width = 60;
        drm->mode_config.min_height = 60;
        drm->mode_config.max_width = 8192;
        drm->mode_config.max_height = 8192;
        drm->mode_config.funcs = &dc_drm_mode_config_funcs;

        drm->vblank_disable_immediate = true;
        drm->max_vblank_count = DC_FRAMEGEN_MAX_FRAME_INDEX;

        for (i = 0; i < DC_DISPLAYS; i++) {
                ret = dc_crtc_init(dc_drm, i);
                if (ret)
                        return ret;

                ret = dc_kms_init_encoder_per_crtc(dc_drm, i);
                if (ret)
                        return ret;
        }

        for (i = 0; i < DC_DISPLAYS; i++) {
                ret = dc_crtc_post_init(dc_drm, i);
                if (ret)
                        return ret;
        }

        ret = drm_vblank_init(drm, DC_DISPLAYS);
        if (ret) {
                dev_err(drm->dev, "failed to init vblank support: %d\n", ret);
                return ret;
        }

        drm_mode_config_reset(drm);

        drm_kms_helper_poll_init(drm);

        return 0;
}

void dc_kms_uninit(struct dc_drm_device *dc_drm)
{
        drm_kms_helper_poll_fini(&dc_drm->base);
}