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 #include <drm/drm_gem_framebuffer_helper.h>
22 
23 #include "rockchip_drm_drv.h"
24 #include "rockchip_drm_fb.h"
25 #include "rockchip_drm_gem.h"
26 #include "rockchip_drm_psr.h"
27 
28 static int rockchip_drm_fb_dirty(struct drm_framebuffer *fb,
29 				 struct drm_file *file,
30 				 unsigned int flags, unsigned int color,
31 				 struct drm_clip_rect *clips,
32 				 unsigned int num_clips)
33 {
34 	rockchip_drm_psr_flush_all(fb->dev);
35 	return 0;
36 }
37 
38 static const struct drm_framebuffer_funcs rockchip_drm_fb_funcs = {
39 	.destroy       = drm_gem_fb_destroy,
40 	.create_handle = drm_gem_fb_create_handle,
41 	.dirty	       = rockchip_drm_fb_dirty,
42 };
43 
44 static struct drm_framebuffer *
45 rockchip_fb_alloc(struct drm_device *dev, const struct drm_mode_fb_cmd2 *mode_cmd,
46 		  struct drm_gem_object **obj, unsigned int num_planes)
47 {
48 	struct drm_framebuffer *fb;
49 	int ret;
50 	int i;
51 
52 	fb = kzalloc(sizeof(*fb), GFP_KERNEL);
53 	if (!fb)
54 		return ERR_PTR(-ENOMEM);
55 
56 	drm_helper_mode_fill_fb_struct(dev, fb, mode_cmd);
57 
58 	for (i = 0; i < num_planes; i++)
59 		fb->obj[i] = obj[i];
60 
61 	ret = drm_framebuffer_init(dev, fb, &rockchip_drm_fb_funcs);
62 	if (ret) {
63 		DRM_DEV_ERROR(dev->dev,
64 			      "Failed to initialize framebuffer: %d\n",
65 			      ret);
66 		kfree(fb);
67 		return ERR_PTR(ret);
68 	}
69 
70 	return fb;
71 }
72 
73 static struct drm_framebuffer *
74 rockchip_user_fb_create(struct drm_device *dev, struct drm_file *file_priv,
75 			const struct drm_mode_fb_cmd2 *mode_cmd)
76 {
77 	struct drm_framebuffer *fb;
78 	struct drm_gem_object *objs[ROCKCHIP_MAX_FB_BUFFER];
79 	struct drm_gem_object *obj;
80 	unsigned int hsub;
81 	unsigned int vsub;
82 	int num_planes;
83 	int ret;
84 	int i;
85 
86 	hsub = drm_format_horz_chroma_subsampling(mode_cmd->pixel_format);
87 	vsub = drm_format_vert_chroma_subsampling(mode_cmd->pixel_format);
88 	num_planes = min(drm_format_num_planes(mode_cmd->pixel_format),
89 			 ROCKCHIP_MAX_FB_BUFFER);
90 
91 	for (i = 0; i < num_planes; i++) {
92 		unsigned int width = mode_cmd->width / (i ? hsub : 1);
93 		unsigned int height = mode_cmd->height / (i ? vsub : 1);
94 		unsigned int min_size;
95 
96 		obj = drm_gem_object_lookup(file_priv, mode_cmd->handles[i]);
97 		if (!obj) {
98 			DRM_DEV_ERROR(dev->dev,
99 				      "Failed to lookup GEM object\n");
100 			ret = -ENXIO;
101 			goto err_gem_object_unreference;
102 		}
103 
104 		min_size = (height - 1) * mode_cmd->pitches[i] +
105 			mode_cmd->offsets[i] +
106 			width * drm_format_plane_cpp(mode_cmd->pixel_format, i);
107 
108 		if (obj->size < min_size) {
109 			drm_gem_object_put_unlocked(obj);
110 			ret = -EINVAL;
111 			goto err_gem_object_unreference;
112 		}
113 		objs[i] = obj;
114 	}
115 
116 	fb = rockchip_fb_alloc(dev, mode_cmd, objs, i);
117 	if (IS_ERR(fb)) {
118 		ret = PTR_ERR(fb);
119 		goto err_gem_object_unreference;
120 	}
121 
122 	return fb;
123 
124 err_gem_object_unreference:
125 	for (i--; i >= 0; i--)
126 		drm_gem_object_put_unlocked(objs[i]);
127 	return ERR_PTR(ret);
128 }
129 
130 static void
131 rockchip_drm_psr_inhibit_get_state(struct drm_atomic_state *state)
132 {
133 	struct drm_crtc *crtc;
134 	struct drm_crtc_state *crtc_state;
135 	struct drm_encoder *encoder;
136 	u32 encoder_mask = 0;
137 	int i;
138 
139 	for_each_old_crtc_in_state(state, crtc, crtc_state, i) {
140 		encoder_mask |= crtc_state->encoder_mask;
141 		encoder_mask |= crtc->state->encoder_mask;
142 	}
143 
144 	drm_for_each_encoder_mask(encoder, state->dev, encoder_mask)
145 		rockchip_drm_psr_inhibit_get(encoder);
146 }
147 
148 static void
149 rockchip_drm_psr_inhibit_put_state(struct drm_atomic_state *state)
150 {
151 	struct drm_crtc *crtc;
152 	struct drm_crtc_state *crtc_state;
153 	struct drm_encoder *encoder;
154 	u32 encoder_mask = 0;
155 	int i;
156 
157 	for_each_old_crtc_in_state(state, crtc, crtc_state, i) {
158 		encoder_mask |= crtc_state->encoder_mask;
159 		encoder_mask |= crtc->state->encoder_mask;
160 	}
161 
162 	drm_for_each_encoder_mask(encoder, state->dev, encoder_mask)
163 		rockchip_drm_psr_inhibit_put(encoder);
164 }
165 
166 static void
167 rockchip_atomic_helper_commit_tail_rpm(struct drm_atomic_state *old_state)
168 {
169 	struct drm_device *dev = old_state->dev;
170 
171 	rockchip_drm_psr_inhibit_get_state(old_state);
172 
173 	drm_atomic_helper_commit_modeset_disables(dev, old_state);
174 
175 	drm_atomic_helper_commit_modeset_enables(dev, old_state);
176 
177 	drm_atomic_helper_commit_planes(dev, old_state,
178 					DRM_PLANE_COMMIT_ACTIVE_ONLY);
179 
180 	rockchip_drm_psr_inhibit_put_state(old_state);
181 
182 	drm_atomic_helper_commit_hw_done(old_state);
183 
184 	drm_atomic_helper_wait_for_vblanks(dev, old_state);
185 
186 	drm_atomic_helper_cleanup_planes(dev, old_state);
187 }
188 
189 static const struct drm_mode_config_helper_funcs rockchip_mode_config_helpers = {
190 	.atomic_commit_tail = rockchip_atomic_helper_commit_tail_rpm,
191 };
192 
193 static const struct drm_mode_config_funcs rockchip_drm_mode_config_funcs = {
194 	.fb_create = rockchip_user_fb_create,
195 	.output_poll_changed = drm_fb_helper_output_poll_changed,
196 	.atomic_check = drm_atomic_helper_check,
197 	.atomic_commit = drm_atomic_helper_commit,
198 };
199 
200 struct drm_framebuffer *
201 rockchip_drm_framebuffer_init(struct drm_device *dev,
202 			      const struct drm_mode_fb_cmd2 *mode_cmd,
203 			      struct drm_gem_object *obj)
204 {
205 	struct drm_framebuffer *fb;
206 
207 	fb = rockchip_fb_alloc(dev, mode_cmd, &obj, 1);
208 	if (IS_ERR(fb))
209 		return ERR_CAST(fb);
210 
211 	return fb;
212 }
213 
214 void rockchip_drm_mode_config_init(struct drm_device *dev)
215 {
216 	dev->mode_config.min_width = 0;
217 	dev->mode_config.min_height = 0;
218 
219 	/*
220 	 * set max width and height as default value(4096x4096).
221 	 * this value would be used to check framebuffer size limitation
222 	 * at drm_mode_addfb().
223 	 */
224 	dev->mode_config.max_width = 4096;
225 	dev->mode_config.max_height = 4096;
226 
227 	dev->mode_config.funcs = &rockchip_drm_mode_config_funcs;
228 	dev->mode_config.helper_private = &rockchip_mode_config_helpers;
229 }
230