3 files changed, 201 insertions, 1 deletions

diff --git a/drivers/gpu/drm/arm/display/komeda/Makefile b/drivers/gpu/drm/arm/display/komeda/Makefile
index 394fc2aa434a..25beae900ed2 100644
--- a/drivers/gpu/drm/arm/display/komeda/Makefile
+++ b/drivers/gpu/drm/arm/display/komeda/Makefile
@@ -8,7 +8,8 @@ komeda-y := \
         komeda_drv.o \
         komeda_dev.o \
         komeda_format_caps.o \
-        komeda_pipeline.o
+        komeda_pipeline.o \
+        komeda_framebuffer.o
 
 komeda-y += \
         d71/d71_dev.o
diff --git a/drivers/gpu/drm/arm/display/komeda/komeda_framebuffer.c b/drivers/gpu/drm/arm/display/komeda/komeda_framebuffer.c
new file mode 100644
index 000000000000..4ddd5314ca23
--- /dev/null
+++ b/drivers/gpu/drm/arm/display/komeda/komeda_framebuffer.c
@@ -0,0 +1,165 @@
+// SPDX-License-Identifier: GPL-2.0
+/*
+ * (C) COPYRIGHT 2018 ARM Limited. All rights reserved.
+ * Author: James.Qian.Wang <james.qian.wang@arm.com>
+ *
+ */
+#include <drm/drm_gem.h>
+#include <drm/drm_gem_framebuffer_helper.h>
+#include <drm/drm_fb_cma_helper.h>
+#include <drm/drm_gem_cma_helper.h>
+#include "komeda_framebuffer.h"
+#include "komeda_dev.h"
+
+static void komeda_fb_destroy(struct drm_framebuffer *fb)
+{
+        struct komeda_fb *kfb = to_kfb(fb);
+        u32 i;
+
+        for (i = 0; i < fb->format->num_planes; i++)
+                drm_gem_object_put_unlocked(fb->obj[i]);
+
+        drm_framebuffer_cleanup(fb);
+        kfree(kfb);
+}
+
+static int komeda_fb_create_handle(struct drm_framebuffer *fb,
+                                   struct drm_file *file, u32 *handle)
+{
+        return drm_gem_handle_create(file, fb->obj[0], handle);
+}
+
+static const struct drm_framebuffer_funcs komeda_fb_funcs = {
+        .destroy        = komeda_fb_destroy,
+        .create_handle  = komeda_fb_create_handle,
+};
+
+static int
+komeda_fb_none_afbc_size_check(struct komeda_dev *mdev, struct komeda_fb *kfb,
+                               struct drm_file *file,
+                               const struct drm_mode_fb_cmd2 *mode_cmd)
+{
+        struct drm_framebuffer *fb = &kfb->base;
+        struct drm_gem_object *obj;
+        u32 min_size = 0;
+        u32 i;
+
+        for (i = 0; i < fb->format->num_planes; i++) {
+                obj = drm_gem_object_lookup(file, mode_cmd->handles[i]);
+                if (!obj) {
+                        DRM_DEBUG_KMS("Failed to lookup GEM object\n");
+                        fb->obj[i] = NULL;
+
+                        return -ENOENT;
+                }
+
+                kfb->aligned_w = fb->width / (i ? fb->format->hsub : 1);
+                kfb->aligned_h = fb->height / (i ? fb->format->vsub : 1);
+
+                if (fb->pitches[i] % mdev->chip.bus_width) {
+                        DRM_DEBUG_KMS("Pitch[%d]: 0x%x doesn't align to 0x%x\n",
+                                      i, fb->pitches[i], mdev->chip.bus_width);
+                        drm_gem_object_put_unlocked(obj);
+                        fb->obj[i] = NULL;
+
+                        return -EINVAL;
+                }
+
+                min_size = ((kfb->aligned_h / kfb->format_caps->tile_size - 1)
+                            * fb->pitches[i])
+                            + (kfb->aligned_w * fb->format->cpp[i]
+                               * kfb->format_caps->tile_size)
+                            + fb->offsets[i];
+
+                if (obj->size < min_size) {
+                        DRM_DEBUG_KMS("Fail to check none afbc fb size.\n");
+                        drm_gem_object_put_unlocked(obj);
+                        fb->obj[i] = NULL;
+
+                        return -EINVAL;
+                }
+
+                fb->obj[i] = obj;
+        }
+
+        if (fb->format->num_planes == 3) {
+                if (fb->pitches[1] != fb->pitches[2]) {
+                        DRM_DEBUG_KMS("The pitch[1] and [2] are not same\n");
+                        return -EINVAL;
+                }
+        }
+
+        return 0;
+}
+
+struct drm_framebuffer *
+komeda_fb_create(struct drm_device *dev, struct drm_file *file,
+                 const struct drm_mode_fb_cmd2 *mode_cmd)
+{
+        struct komeda_dev *mdev = dev->dev_private;
+        struct komeda_fb *kfb;
+        int ret = 0, i;
+
+        kfb = kzalloc(sizeof(*kfb), GFP_KERNEL);
+        if (!kfb)
+                return ERR_PTR(-ENOMEM);
+
+        kfb->format_caps = komeda_get_format_caps(&mdev->fmt_tbl,
+                                                  mode_cmd->pixel_format,
+                                                  mode_cmd->modifier[0]);
+        if (!kfb->format_caps) {
+                DRM_DEBUG_KMS("FMT %x is not supported.\n",
+                              mode_cmd->pixel_format);
+                kfree(kfb);
+                return ERR_PTR(-EINVAL);
+        }
+
+        drm_helper_mode_fill_fb_struct(dev, &kfb->base, mode_cmd);
+
+        ret = komeda_fb_none_afbc_size_check(mdev, kfb, file, mode_cmd);
+        if (ret < 0)
+                goto err_cleanup;
+
+        ret = drm_framebuffer_init(dev, &kfb->base, &komeda_fb_funcs);
+        if (ret < 0) {
+                DRM_DEBUG_KMS("failed to initialize fb\n");
+
+                goto err_cleanup;
+        }
+
+        return &kfb->base;
+
+err_cleanup:
+        for (i = 0; i < kfb->base.format->num_planes; i++)
+                drm_gem_object_put_unlocked(kfb->base.obj[i]);
+
+        kfree(kfb);
+        return ERR_PTR(ret);
+}
+
+dma_addr_t
+komeda_fb_get_pixel_addr(struct komeda_fb *kfb, int x, int y, int plane)
+{
+        struct drm_framebuffer *fb = &kfb->base;
+        const struct drm_gem_cma_object *obj;
+        u32 plane_x, plane_y, cpp, pitch, offset;
+
+        if (plane > fb->format->num_planes) {
+                DRM_DEBUG_KMS("Out of max plane num.\n");
+                return -EINVAL;
+        }
+
+        obj = drm_fb_cma_get_gem_obj(fb, plane);
+
+        offset = fb->offsets[plane];
+        if (!fb->modifier) {
+                plane_x = x / (plane ? fb->format->hsub : 1);
+                plane_y = y / (plane ? fb->format->vsub : 1);
+                cpp = fb->format->cpp[plane];
+                pitch = fb->pitches[plane];
+                offset += plane_x * cpp *  kfb->format_caps->tile_size +
+                                (plane_y * pitch) / kfb->format_caps->tile_size;
+        }
+
+        return obj->paddr + offset;
+}
diff --git a/drivers/gpu/drm/arm/display/komeda/komeda_framebuffer.h b/drivers/gpu/drm/arm/display/komeda/komeda_framebuffer.h
new file mode 100644
index 000000000000..0de2e4a2afd2
--- /dev/null
+++ b/drivers/gpu/drm/arm/display/komeda/komeda_framebuffer.h
@@ -0,0 +1,34 @@
+/* SPDX-License-Identifier: GPL-2.0 */
+/*
+ * (C) COPYRIGHT 2018 ARM Limited. All rights reserved.
+ * Author: James.Qian.Wang <james.qian.wang@arm.com>
+ *
+ */
+#ifndef _KOMEDA_FRAMEBUFFER_H_
+#define _KOMEDA_FRAMEBUFFER_H_
+
+#include <drm/drm_framebuffer.h>
+#include "komeda_format_caps.h"
+
+/** struct komeda_fb - entend drm_framebuffer with komeda attribute */
+struct komeda_fb {
+        /** @base: &drm_framebuffer */
+        struct drm_framebuffer base;
+        /* @format_caps: &komeda_format_caps */
+        const struct komeda_format_caps *format_caps;
+        /** @aligned_w: aligned frame buffer width */
+        u32 aligned_w;
+        /** @aligned_h: aligned frame buffer height */
+        u32 aligned_h;
+};
+
+#define to_kfb(dfb)     container_of(dfb, struct komeda_fb, base)
+
+struct drm_framebuffer *
+komeda_fb_create(struct drm_device *dev, struct drm_file *file,
+                 const struct drm_mode_fb_cmd2 *mode_cmd);
+dma_addr_t
+komeda_fb_get_pixel_addr(struct komeda_fb *kfb, int x, int y, int plane);
+bool komeda_fb_is_layer_supported(struct komeda_fb *kfb, u32 layer_type);
+
+#endif