drivers/gpu/drm/rockchip/rockchip_drm_fb.c

   1 /*
   2  * Copyright (C) Fuzhou Rockchip Electronics Co.Ltd
   3  * Author:Mark Yao <mark.yao@rock-chips.com>
   4  *
   5  * This software is licensed under the terms of the GNU General Public
   6  * License version 2, as published by the Free Software Foundation, and
   7  * may be copied, distributed, and modified under those terms.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  */
  14
  15 #include <linux/kernel.h>
  16 #include <drm/drm.h>
  17 #include <drm/drmP.h>
  18 #include <drm/drm_atomic.h>
  19 #include <drm/drm_fb_helper.h>
  20 #include <drm/drm_crtc_helper.h>
  21
  22 #include "rockchip_drm_drv.h"
  23 #include "rockchip_drm_fb.h"
  24 #include "rockchip_drm_gem.h"
  25 #include "rockchip_drm_psr.h"
  26
  27 #define to_rockchip_fb(x) container_of(x, struct rockchip_drm_fb, fb)
  28
  29 struct rockchip_drm_fb {
  30         struct drm_framebuffer fb;
  31         struct drm_gem_object *obj[ROCKCHIP_MAX_FB_BUFFER];
  32 };
  33
  34 struct drm_gem_object *rockchip_fb_get_gem_obj(struct drm_framebuffer *fb,
  35                                                unsigned int plane)
  36 {
  37         struct rockchip_drm_fb *rk_fb = to_rockchip_fb(fb);
  38
  39         if (plane >= ROCKCHIP_MAX_FB_BUFFER)
  40                 return NULL;
  41
  42         return rk_fb->obj[plane];
  43 }
  44
  45 static void rockchip_drm_fb_destroy(struct drm_framebuffer *fb)
  46 {
  47         struct rockchip_drm_fb *rockchip_fb = to_rockchip_fb(fb);
  48         int i;
  49
  50         for (i = 0; i < ROCKCHIP_MAX_FB_BUFFER; i++)
  51                 drm_gem_object_put_unlocked(rockchip_fb->obj[i]);
  52
  53         drm_framebuffer_cleanup(fb);
  54         kfree(rockchip_fb);
  55 }
  56
  57 static int rockchip_drm_fb_create_handle(struct drm_framebuffer *fb,
  58                                          struct drm_file *file_priv,
  59                                          unsigned int *handle)
  60 {
  61         struct rockchip_drm_fb *rockchip_fb = to_rockchip_fb(fb);
  62
  63         return drm_gem_handle_create(file_priv,
  64                                      rockchip_fb->obj[0], handle);
  65 }
  66
  67 static int rockchip_drm_fb_dirty(struct drm_framebuffer *fb,
  68                                  struct drm_file *file,
  69                                  unsigned int flags, unsigned int color,
  70                                  struct drm_clip_rect *clips,
  71                                  unsigned int num_clips)
  72 {
  73         rockchip_drm_psr_flush_all(fb->dev);
  74         return 0;
  75 }
  76
  77 static const struct drm_framebuffer_funcs rockchip_drm_fb_funcs = {
  78         .destroy        = rockchip_drm_fb_destroy,
  79         .create_handle  = rockchip_drm_fb_create_handle,
  80         .dirty          = rockchip_drm_fb_dirty,
  81 };
  82
  83 static struct rockchip_drm_fb *
  84 rockchip_fb_alloc(struct drm_device *dev, const struct drm_mode_fb_cmd2 *mode_cmd,
  85                   struct drm_gem_object **obj, unsigned int num_planes)
  86 {
  87         struct rockchip_drm_fb *rockchip_fb;
  88         int ret;
  89         int i;
  90
  91         rockchip_fb = kzalloc(sizeof(*rockchip_fb), GFP_KERNEL);
  92         if (!rockchip_fb)
  93                 return ERR_PTR(-ENOMEM);
  94
  95         drm_helper_mode_fill_fb_struct(dev, &rockchip_fb->fb, mode_cmd);
  96
  97         for (i = 0; i < num_planes; i++)
  98                 rockchip_fb->obj[i] = obj[i];
  99
 100         ret = drm_framebuffer_init(dev, &rockchip_fb->fb,
 101                                    &rockchip_drm_fb_funcs);
 102         if (ret) {
 103                 DRM_DEV_ERROR(dev->dev,
 104                               "Failed to initialize framebuffer: %d\n",
 105                               ret);
 106                 kfree(rockchip_fb);
 107                 return ERR_PTR(ret);
 108         }
 109
 110         return rockchip_fb;
 111 }
 112
 113 static struct drm_framebuffer *
 114 rockchip_user_fb_create(struct drm_device *dev, struct drm_file *file_priv,
 115                         const struct drm_mode_fb_cmd2 *mode_cmd)
 116 {
 117         struct rockchip_drm_fb *rockchip_fb;
 118         struct drm_gem_object *objs[ROCKCHIP_MAX_FB_BUFFER];
 119         struct drm_gem_object *obj;
 120         unsigned int hsub;
 121         unsigned int vsub;
 122         int num_planes;
 123         int ret;
 124         int i;
 125
 126         hsub = drm_format_horz_chroma_subsampling(mode_cmd->pixel_format);
 127         vsub = drm_format_vert_chroma_subsampling(mode_cmd->pixel_format);
 128         num_planes = min(drm_format_num_planes(mode_cmd->pixel_format),
 129                          ROCKCHIP_MAX_FB_BUFFER);
 130
 131         for (i = 0; i < num_planes; i++) {
 132                 unsigned int width = mode_cmd->width / (i ? hsub : 1);
 133                 unsigned int height = mode_cmd->height / (i ? vsub : 1);
 134                 unsigned int min_size;
 135
 136                 obj = drm_gem_object_lookup(file_priv, mode_cmd->handles[i]);
 137                 if (!obj) {
 138                         DRM_DEV_ERROR(dev->dev,
 139                                       "Failed to lookup GEM object\n");
 140                         ret = -ENXIO;
 141                         goto err_gem_object_unreference;
 142                 }
 143
 144                 min_size = (height - 1) * mode_cmd->pitches[i] +
 145                         mode_cmd->offsets[i] +
 146                         width * drm_format_plane_cpp(mode_cmd->pixel_format, i);
 147
 148                 if (obj->size < min_size) {
 149                         drm_gem_object_put_unlocked(obj);
 150                         ret = -EINVAL;
 151                         goto err_gem_object_unreference;
 152                 }
 153                 objs[i] = obj;
 154         }
 155
 156         rockchip_fb = rockchip_fb_alloc(dev, mode_cmd, objs, i);
 157         if (IS_ERR(rockchip_fb)) {
 158                 ret = PTR_ERR(rockchip_fb);
 159                 goto err_gem_object_unreference;
 160         }
 161
 162         return &rockchip_fb->fb;
 163
 164 err_gem_object_unreference:
 165         for (i--; i >= 0; i--)
 166                 drm_gem_object_put_unlocked(objs[i]);
 167         return ERR_PTR(ret);
 168 }
 169
 170 static const struct drm_mode_config_helper_funcs rockchip_mode_config_helpers = {
 171         .atomic_commit_tail = drm_atomic_helper_commit_tail_rpm,
 172 };
 173
 174 static const struct drm_mode_config_funcs rockchip_drm_mode_config_funcs = {
 175         .fb_create = rockchip_user_fb_create,
 176         .output_poll_changed = drm_fb_helper_output_poll_changed,
 177         .atomic_check = drm_atomic_helper_check,
 178         .atomic_commit = drm_atomic_helper_commit,
 179 };
 180
 181 struct drm_framebuffer *
 182 rockchip_drm_framebuffer_init(struct drm_device *dev,
 183                               const struct drm_mode_fb_cmd2 *mode_cmd,
 184                               struct drm_gem_object *obj)
 185 {
 186         struct rockchip_drm_fb *rockchip_fb;
 187
 188         rockchip_fb = rockchip_fb_alloc(dev, mode_cmd, &obj, 1);
 189         if (IS_ERR(rockchip_fb))
 190                 return ERR_CAST(rockchip_fb);
 191
 192         return &rockchip_fb->fb;
 193 }
 194
 195 void rockchip_drm_mode_config_init(struct drm_device *dev)
 196 {
 197         dev->mode_config.min_width = 0;
 198         dev->mode_config.min_height = 0;
 199
 200         /*
 201          * set max width and height as default value(4096x4096).
 202          * this value would be used to check framebuffer size limitation
 203          * at drm_mode_addfb().
 204          */
 205         dev->mode_config.max_width = 4096;
 206         dev->mode_config.max_height = 4096;
 207
 208         dev->mode_config.funcs = &rockchip_drm_mode_config_funcs;
 209         dev->mode_config.helper_private = &rockchip_mode_config_helpers;
 210 }