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
 29struct rockchip_drm_fb {
 30	struct drm_framebuffer fb;
 31	struct drm_gem_object *obj[ROCKCHIP_MAX_FB_BUFFER];
 32};
 33
 34struct 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
 45static 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
 57static 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
 67static 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
 77static 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
 83static struct rockchip_drm_fb *
 84rockchip_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
113static struct drm_framebuffer *
114rockchip_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
164err_gem_object_unreference:
165	for (i--; i >= 0; i--)
166		drm_gem_object_put_unlocked(objs[i]);
167	return ERR_PTR(ret);
168}
169
170static const struct drm_mode_config_helper_funcs rockchip_mode_config_helpers = {
171	.atomic_commit_tail = drm_atomic_helper_commit_tail_rpm,
172};
173
174static 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
181struct drm_framebuffer *
182rockchip_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
195void 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}