// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright (C) 2018 Texas Instruments Incorporated - https://www.ti.com/
 * Author: Tomi Valkeinen <tomi.valkeinen@ti.com>
 */

#include <drm/drm_atomic.h>
#include <drm/drm_atomic_helper.h>
#include <drm/drm_crtc.h>
#include <drm/drm_crtc_helper.h>
#include <drm/drm_fourcc.h>
#include <drm/drm_fb_cma_helper.h>

#include "tidss_crtc.h"
#include "tidss_dispc.h"
#include "tidss_drv.h"
#include "tidss_plane.h"

/* drm_plane_helper_funcs */

static int tidss_plane_atomic_check(struct drm_plane *plane,
                                    struct drm_plane_state *state)
{
        struct drm_device *ddev = plane->dev;
        struct tidss_device *tidss = to_tidss(ddev);
        struct tidss_plane *tplane = to_tidss_plane(plane);
        const struct drm_format_info *finfo;
        struct drm_crtc_state *crtc_state;
        u32 hw_plane = tplane->hw_plane_id;
        u32 hw_videoport;
        int ret;

        dev_dbg(ddev->dev, "%s\n", __func__);

        if (!state->crtc) {
                /*
                 * The visible field is not reset by the DRM core but only
                 * updated by drm_plane_helper_check_state(), set it manually.
                 */
                state->visible = false;
                return 0;
        }

        crtc_state = drm_atomic_get_crtc_state(state->state, state->crtc);
        if (IS_ERR(crtc_state))
                return PTR_ERR(crtc_state);

        ret = drm_atomic_helper_check_plane_state(state, crtc_state, 0,
                                                  INT_MAX, true, true);
        if (ret < 0)
                return ret;

        /*
         * The HW is only able to start drawing at subpixel boundary
         * (the two first checks bellow). At the end of a row the HW
         * can only jump integer number of subpixels forward to the
         * beginning of the next row. So we can only show picture with
         * integer subpixel width (the third check). However, after
         * reaching the end of the drawn picture the drawing starts
         * again at the absolute memory address where top left corner
         * position of the drawn picture is (so there is no need to
         * check for odd height).
         */

        finfo = drm_format_info(state->fb->format->format);

        if ((state->src_x >> 16) % finfo->hsub != 0) {
                dev_dbg(ddev->dev,
                        "%s: x-position %u not divisible subpixel size %u\n",
                        __func__, (state->src_x >> 16), finfo->hsub);
                return -EINVAL;
        }

        if ((state->src_y >> 16) % finfo->vsub != 0) {
                dev_dbg(ddev->dev,
                        "%s: y-position %u not divisible subpixel size %u\n",
                        __func__, (state->src_y >> 16), finfo->vsub);
                return -EINVAL;
        }

        if ((state->src_w >> 16) % finfo->hsub != 0) {
                dev_dbg(ddev->dev,
                        "%s: src width %u not divisible by subpixel size %u\n",
                         __func__, (state->src_w >> 16), finfo->hsub);
                return -EINVAL;
        }

        if (!state->visible)
                return 0;

        hw_videoport = to_tidss_crtc(state->crtc)->hw_videoport;

        ret = dispc_plane_check(tidss->dispc, hw_plane, state, hw_videoport);
        if (ret)
                return ret;

        return 0;
}

static void tidss_plane_atomic_update(struct drm_plane *plane,
                                      struct drm_plane_state *old_state)
{
        struct drm_device *ddev = plane->dev;
        struct tidss_device *tidss = to_tidss(ddev);
        struct tidss_plane *tplane = to_tidss_plane(plane);
        struct drm_plane_state *state = plane->state;
        u32 hw_videoport;
        int ret;

        dev_dbg(ddev->dev, "%s\n", __func__);

        if (!state->visible) {
                dispc_plane_enable(tidss->dispc, tplane->hw_plane_id, false);
                return;
        }

        hw_videoport = to_tidss_crtc(state->crtc)->hw_videoport;

        ret = dispc_plane_setup(tidss->dispc, tplane->hw_plane_id,
                                state, hw_videoport);

        if (ret) {
                dev_err(plane->dev->dev, "%s: Failed to setup plane %d\n",
                        __func__, tplane->hw_plane_id);
                dispc_plane_enable(tidss->dispc, tplane->hw_plane_id, false);
                return;
        }

        dispc_plane_enable(tidss->dispc, tplane->hw_plane_id, true);
}

static void tidss_plane_atomic_disable(struct drm_plane *plane,
                                       struct drm_plane_state *old_state)
{
        struct drm_device *ddev = plane->dev;
        struct tidss_device *tidss = to_tidss(ddev);
        struct tidss_plane *tplane = to_tidss_plane(plane);

        dev_dbg(ddev->dev, "%s\n", __func__);

        dispc_plane_enable(tidss->dispc, tplane->hw_plane_id, false);
}

static void drm_plane_destroy(struct drm_plane *plane)
{
        struct tidss_plane *tplane = to_tidss_plane(plane);

        drm_plane_cleanup(plane);
        kfree(tplane);
}

static const struct drm_plane_helper_funcs tidss_plane_helper_funcs = {
        .atomic_check = tidss_plane_atomic_check,
        .atomic_update = tidss_plane_atomic_update,
        .atomic_disable = tidss_plane_atomic_disable,
};

static const struct drm_plane_funcs tidss_plane_funcs = {
        .update_plane = drm_atomic_helper_update_plane,
        .disable_plane = drm_atomic_helper_disable_plane,
        .reset = drm_atomic_helper_plane_reset,
        .destroy = drm_plane_destroy,
        .atomic_duplicate_state = drm_atomic_helper_plane_duplicate_state,
        .atomic_destroy_state = drm_atomic_helper_plane_destroy_state,
};

struct tidss_plane *tidss_plane_create(struct tidss_device *tidss,
                                       u32 hw_plane_id, u32 plane_type,
                                       u32 crtc_mask, const u32 *formats,
                                       u32 num_formats)
{
        struct tidss_plane *tplane;
        enum drm_plane_type type;
        u32 possible_crtcs;
        u32 num_planes = tidss->feat->num_planes;
        u32 color_encodings = (BIT(DRM_COLOR_YCBCR_BT601) |
                               BIT(DRM_COLOR_YCBCR_BT709));
        u32 color_ranges = (BIT(DRM_COLOR_YCBCR_FULL_RANGE) |
                            BIT(DRM_COLOR_YCBCR_LIMITED_RANGE));
        u32 default_encoding = DRM_COLOR_YCBCR_BT601;
        u32 default_range = DRM_COLOR_YCBCR_FULL_RANGE;
        u32 blend_modes = (BIT(DRM_MODE_BLEND_PREMULTI) |
                           BIT(DRM_MODE_BLEND_COVERAGE));
        int ret;

        tplane = kzalloc(sizeof(*tplane), GFP_KERNEL);
        if (!tplane)
                return ERR_PTR(-ENOMEM);

        tplane->hw_plane_id = hw_plane_id;

        possible_crtcs = crtc_mask;
        type = plane_type;

        ret = drm_universal_plane_init(&tidss->ddev, &tplane->plane,
                                       possible_crtcs,
                                       &tidss_plane_funcs,
                                       formats, num_formats,
                                       NULL, type, NULL);
        if (ret < 0)
                goto err;

        drm_plane_helper_add(&tplane->plane, &tidss_plane_helper_funcs);

        drm_plane_create_zpos_property(&tplane->plane, hw_plane_id, 0,
                                       num_planes - 1);

        ret = drm_plane_create_color_properties(&tplane->plane,
                                                color_encodings,
                                                color_ranges,
                                                default_encoding,
                                                default_range);
        if (ret)
                goto err;

        ret = drm_plane_create_alpha_property(&tplane->plane);
        if (ret)
                goto err;

        ret = drm_plane_create_blend_mode_property(&tplane->plane, blend_modes);
        if (ret)
                goto err;

        return tplane;

err:
        kfree(tplane);
        return ERR_PTR(ret);
}