// SPDX-License-Identifier: GPL-2.0+

#include "vkms_config.h"
#include "vkms_connector.h"
#include "vkms_drv.h"
#include <drm/drm_managed.h>
#include <drm/drm_print.h>

int vkms_output_init(struct vkms_device *vkmsdev)
{
	struct drm_device *dev = &vkmsdev->drm;
	struct vkms_config_plane *plane_cfg;
	struct vkms_config_crtc *crtc_cfg;
	struct vkms_config_encoder *encoder_cfg;
	struct vkms_config_connector *connector_cfg;
	int ret;
	int writeback;

	if (!vkms_config_is_valid(vkmsdev->config))
		return -EINVAL;

	vkms_config_for_each_plane(vkmsdev->config, plane_cfg) {
		plane_cfg->plane = vkms_plane_init(vkmsdev, plane_cfg);
		if (IS_ERR(plane_cfg->plane)) {
			DRM_DEV_ERROR(dev->dev, "Failed to init vkms plane\n");
			return PTR_ERR(plane_cfg->plane);
		}
	}

	vkms_config_for_each_crtc(vkmsdev->config, crtc_cfg) {
		struct vkms_config_plane *primary, *cursor;

		primary = vkms_config_crtc_primary_plane(vkmsdev->config, crtc_cfg);
		cursor = vkms_config_crtc_cursor_plane(vkmsdev->config, crtc_cfg);

		crtc_cfg->crtc = vkms_crtc_init(dev, &primary->plane->base,
						cursor ? &cursor->plane->base : NULL);
		if (IS_ERR(crtc_cfg->crtc)) {
			DRM_ERROR("Failed to allocate CRTC\n");
			return PTR_ERR(crtc_cfg->crtc);
		}

		/* Initialize the writeback component */
		if (vkms_config_crtc_get_writeback(crtc_cfg)) {
			writeback = vkms_enable_writeback_connector(vkmsdev, crtc_cfg->crtc);
			if (writeback)
				DRM_ERROR("Failed to init writeback connector\n");
		}
	}

	vkms_config_for_each_plane(vkmsdev->config, plane_cfg) {
		struct vkms_config_crtc *possible_crtc;
		unsigned long idx = 0;

		vkms_config_plane_for_each_possible_crtc(plane_cfg, idx, possible_crtc) {
			plane_cfg->plane->base.possible_crtcs |=
				drm_crtc_mask(&possible_crtc->crtc->crtc);
		}
	}

	vkms_config_for_each_encoder(vkmsdev->config, encoder_cfg) {
		struct vkms_config_crtc *possible_crtc;
		unsigned long idx = 0;

		encoder_cfg->encoder = drmm_kzalloc(dev, sizeof(*encoder_cfg->encoder), GFP_KERNEL);
		if (!encoder_cfg->encoder) {
			DRM_ERROR("Failed to allocate encoder\n");
			return -ENOMEM;
		}
		ret = drmm_encoder_init(dev, encoder_cfg->encoder, NULL,
					DRM_MODE_ENCODER_VIRTUAL, NULL);
		if (ret) {
			DRM_ERROR("Failed to init encoder\n");
			return ret;
		}

		encoder_cfg->encoder->possible_clones |=
			drm_encoder_mask(encoder_cfg->encoder);

		vkms_config_encoder_for_each_possible_crtc(encoder_cfg, idx, possible_crtc) {
			encoder_cfg->encoder->possible_crtcs |=
				drm_crtc_mask(&possible_crtc->crtc->crtc);

			if (vkms_config_crtc_get_writeback(possible_crtc)) {
				struct drm_encoder *wb_encoder =
					&possible_crtc->crtc->wb_encoder;

				encoder_cfg->encoder->possible_clones |=
					drm_encoder_mask(wb_encoder);
				wb_encoder->possible_clones |=
					drm_encoder_mask(encoder_cfg->encoder);
			}
		}
	}

	vkms_config_for_each_connector(vkmsdev->config, connector_cfg) {
		struct vkms_config_encoder *possible_encoder;
		unsigned long idx = 0;

		connector_cfg->connector = vkms_connector_init(vkmsdev);
		if (IS_ERR(connector_cfg->connector)) {
			DRM_ERROR("Failed to init connector\n");
			return PTR_ERR(connector_cfg->connector);
		}

		vkms_config_connector_for_each_possible_encoder(connector_cfg,
								idx,
								possible_encoder) {
			ret = drm_connector_attach_encoder(&connector_cfg->connector->base,
							   possible_encoder->encoder);
			if (ret) {
				DRM_ERROR("Failed to attach connector to encoder\n");
				return ret;
			}
		}
	}

	drm_mode_config_reset(dev);

	return 0;
}
