1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  * Copyright (C) 2020 BayLibre, SAS
4  * Author: Neil Armstrong <narmstrong@baylibre.com>
5  */
6 
7 #include <linux/clk.h>
8 #include <linux/delay.h>
9 #include <linux/bitfield.h>
10 #include <linux/interrupt.h>
11 #include <linux/module.h>
12 #include <linux/of.h>
13 #include <linux/reset.h>
14 #include <linux/sched.h>
15 #include <linux/slab.h>
16 #include <linux/timer.h>
17 #include <linux/regmap.h>
18 
19 #include <linux/platform_device.h>
20 #include <media/v4l2-device.h>
21 #include <media/v4l2-event.h>
22 #include <media/v4l2-ioctl.h>
23 #include <media/v4l2-mem2mem.h>
24 #include <media/v4l2-ctrls.h>
25 #include <media/videobuf2-v4l2.h>
26 #include <media/videobuf2-dma-contig.h>
27 
28 #include "ge2d-regs.h"
29 
30 #define GE2D_NAME	"meson-ge2d"
31 
32 #define DEFAULT_WIDTH	128
33 #define DEFAULT_HEIGHT	128
34 #define DEFAULT_STRIDE	512
35 
36 #define MAX_WIDTH	8191
37 #define MAX_HEIGHT	8191
38 
39 /*
40  * Missing features:
41  * - Scaling
42  * - Simple 1/2 vertical scaling
43  * - YUV input support
44  * - Source global alpha
45  * - Colorspace conversion
46  */
47 
48 struct ge2d_fmt {
49 	u32 fourcc;
50 	bool alpha;
51 	bool le;
52 	unsigned int depth;
53 	unsigned int hw_fmt;
54 	unsigned int hw_map;
55 };
56 
57 struct ge2d_frame {
58 	struct vb2_v4l2_buffer *buf;
59 
60 	/* Image Format */
61 	struct v4l2_pix_format pix_fmt;
62 
63 	/* Crop */
64 	struct v4l2_rect crop;
65 
66 	/* Image format */
67 	const struct ge2d_fmt *fmt;
68 };
69 
70 struct ge2d_ctx {
71 	struct v4l2_fh fh;
72 	struct meson_ge2d *ge2d;
73 	struct ge2d_frame in;
74 	struct ge2d_frame out;
75 	struct v4l2_ctrl_handler ctrl_handler;
76 
77 	unsigned long sequence_out, sequence_cap;
78 
79 	/* Control values */
80 	u32 hflip;
81 	u32 vflip;
82 	u32 xy_swap;
83 };
84 
85 struct meson_ge2d {
86 	struct v4l2_device v4l2_dev;
87 	struct v4l2_m2m_dev *m2m_dev;
88 	struct video_device *vfd;
89 
90 	struct device *dev;
91 	struct regmap *map;
92 	struct clk *clk;
93 
94 	/* vb2 queue lock */
95 	struct mutex mutex;
96 
97 	struct ge2d_ctx *curr;
98 };
99 
100 #define FMT(_fourcc, _alpha, _depth, _map)		\
101 {							\
102 	.fourcc = _fourcc,				\
103 	.alpha = (_alpha),				\
104 	.depth = (_depth),				\
105 	.hw_fmt = GE2D_FORMAT_ ## _depth ## BIT,	\
106 	.hw_map = GE2D_COLOR_MAP_ ## _map,		\
107 }
108 
109 /* TOFIX Handle the YUV input formats */
110 static const struct ge2d_fmt formats[] = {
111 	/*  FOURCC Alpha  HW FMT  HW MAP */
112 	FMT(V4L2_PIX_FMT_XRGB32, false, 32, BGRA8888),
113 	FMT(V4L2_PIX_FMT_RGB32, true, 32, BGRA8888),
114 	FMT(V4L2_PIX_FMT_ARGB32, true, 32, BGRA8888),
115 	FMT(V4L2_PIX_FMT_RGBX32, false, 32, ABGR8888),
116 	FMT(V4L2_PIX_FMT_RGBA32, true, 32, ABGR8888),
117 	FMT(V4L2_PIX_FMT_BGRX32, false, 32, RGBA8888),
118 	FMT(V4L2_PIX_FMT_BGRA32, true, 32, RGBA8888),
119 	FMT(V4L2_PIX_FMT_BGR32, true, 32, ARGB8888),
120 	FMT(V4L2_PIX_FMT_ABGR32, true, 32, ARGB8888),
121 	FMT(V4L2_PIX_FMT_XBGR32, false, 32, ARGB8888),
122 
123 	FMT(V4L2_PIX_FMT_RGB24, false, 24, BGR888),
124 	FMT(V4L2_PIX_FMT_BGR24, false, 24, RGB888),
125 
126 	FMT(V4L2_PIX_FMT_XRGB555X, false, 16, ARGB1555),
127 	FMT(V4L2_PIX_FMT_ARGB555X, true, 16, ARGB1555),
128 	FMT(V4L2_PIX_FMT_RGB565, false, 16, RGB565),
129 	FMT(V4L2_PIX_FMT_RGBX444, false, 16, RGBA4444),
130 	FMT(V4L2_PIX_FMT_RGBA444, true, 16, RGBA4444),
131 	FMT(V4L2_PIX_FMT_XRGB444, false, 16, ARGB4444),
132 	FMT(V4L2_PIX_FMT_ARGB444, true, 16, ARGB4444),
133 };
134 
135 #define NUM_FORMATS ARRAY_SIZE(formats)
136 
137 static const struct ge2d_fmt *find_fmt(struct v4l2_format *f)
138 {
139 	unsigned int i;
140 
141 	for (i = 0; i < NUM_FORMATS; i++) {
142 		if (formats[i].fourcc == f->fmt.pix.pixelformat)
143 			return &formats[i];
144 	}
145 
146 	/*
147 	 * TRY_FMT/S_FMT should never return an error when the requested format
148 	 * is not supported. Drivers should always return a valid format,
149 	 * preferably a format that is as widely supported by applications as
150 	 * possible.
151 	 */
152 	return &formats[0];
153 }
154 
155 static struct ge2d_frame *get_frame(struct ge2d_ctx *ctx,
156 				    enum v4l2_buf_type type)
157 {
158 	switch (type) {
159 	case V4L2_BUF_TYPE_VIDEO_OUTPUT:
160 		return &ctx->in;
161 	case V4L2_BUF_TYPE_VIDEO_CAPTURE:
162 		return &ctx->out;
163 	default:
164 		/* This should never happen, warn and return OUTPUT frame */
165 		dev_warn(ctx->ge2d->dev, "%s: invalid buffer type\n", __func__);
166 		return &ctx->in;
167 	}
168 }
169 
170 static void ge2d_hw_start(struct meson_ge2d *ge2d)
171 {
172 	struct ge2d_ctx *ctx = ge2d->curr;
173 	u32 reg;
174 
175 	/* Reset */
176 	regmap_update_bits(ge2d->map, GE2D_GEN_CTRL1,
177 			   GE2D_SOFT_RST, GE2D_SOFT_RST);
178 	regmap_update_bits(ge2d->map, GE2D_GEN_CTRL1,
179 			   GE2D_SOFT_RST, 0);
180 
181 	usleep_range(100, 200);
182 
183 	/* Implement CANVAS for non-AXG */
184 	regmap_write(ge2d->map, GE2D_SRC1_BADDR_CTRL,
185 		     (vb2_dma_contig_plane_dma_addr(&ctx->in.buf->vb2_buf, 0) + 7) >> 3);
186 	regmap_write(ge2d->map, GE2D_SRC1_STRIDE_CTRL,
187 		     (ctx->in.pix_fmt.bytesperline + 7) >> 3);
188 	regmap_write(ge2d->map, GE2D_SRC2_BADDR_CTRL,
189 		     (vb2_dma_contig_plane_dma_addr(&ctx->out.buf->vb2_buf, 0) + 7) >> 3);
190 	regmap_write(ge2d->map, GE2D_SRC2_STRIDE_CTRL,
191 		     (ctx->out.pix_fmt.bytesperline + 7) >> 3);
192 	regmap_write(ge2d->map, GE2D_DST1_BADDR_CTRL,
193 		     (vb2_dma_contig_plane_dma_addr(&ctx->out.buf->vb2_buf, 0) + 7) >> 3);
194 	regmap_write(ge2d->map, GE2D_DST1_STRIDE_CTRL,
195 		     (ctx->out.pix_fmt.bytesperline + 7) >> 3);
196 
197 	regmap_write(ge2d->map, GE2D_GEN_CTRL0, 0);
198 	regmap_write(ge2d->map, GE2D_GEN_CTRL1,
199 		     FIELD_PREP(GE2D_INTERRUPT_CTRL, 2) |
200 		     FIELD_PREP(GE2D_SRC2_BURST_SIZE_CTRL, 3) |
201 		     FIELD_PREP(GE2D_SRC1_BURST_SIZE_CTRL, 0x3f));
202 
203 	regmap_write(ge2d->map, GE2D_GEN_CTRL2,
204 		     GE2D_SRC1_LITTLE_ENDIAN |
205 		     GE2D_SRC2_LITTLE_ENDIAN |
206 		     GE2D_DST_LITTLE_ENDIAN |
207 		     FIELD_PREP(GE2D_DST1_COLOR_MAP, ctx->out.fmt->hw_map) |
208 		     FIELD_PREP(GE2D_DST1_FORMAT, ctx->out.fmt->hw_fmt) |
209 		     FIELD_PREP(GE2D_SRC2_COLOR_MAP, ctx->out.fmt->hw_map) |
210 		     FIELD_PREP(GE2D_SRC2_FORMAT, ctx->out.fmt->hw_fmt) |
211 		     FIELD_PREP(GE2D_SRC1_COLOR_MAP, ctx->in.fmt->hw_map) |
212 		     FIELD_PREP(GE2D_SRC1_FORMAT, ctx->in.fmt->hw_fmt));
213 	regmap_write(ge2d->map, GE2D_GEN_CTRL3,
214 		     GE2D_DST1_ENABLE);
215 
216 	regmap_write(ge2d->map, GE2D_SRC1_CLIPY_START_END,
217 		     FIELD_PREP(GE2D_START, ctx->in.crop.top) |
218 		     FIELD_PREP(GE2D_END, ctx->in.crop.top + ctx->in.crop.height - 1));
219 	regmap_write(ge2d->map, GE2D_SRC1_CLIPX_START_END,
220 		     FIELD_PREP(GE2D_START, ctx->in.crop.left) |
221 		     FIELD_PREP(GE2D_END, ctx->in.crop.left + ctx->in.crop.width - 1));
222 	regmap_write(ge2d->map, GE2D_SRC2_CLIPY_START_END,
223 		     FIELD_PREP(GE2D_START, ctx->out.crop.top) |
224 		     FIELD_PREP(GE2D_END, ctx->out.crop.top + ctx->out.crop.height - 1));
225 	regmap_write(ge2d->map, GE2D_SRC2_CLIPX_START_END,
226 		     FIELD_PREP(GE2D_START, ctx->out.crop.left) |
227 		     FIELD_PREP(GE2D_END, ctx->out.crop.left + ctx->out.crop.width - 1));
228 	regmap_write(ge2d->map, GE2D_DST_CLIPY_START_END,
229 		     FIELD_PREP(GE2D_START, ctx->out.crop.top) |
230 		     FIELD_PREP(GE2D_END, ctx->out.crop.top + ctx->out.crop.height - 1));
231 	regmap_write(ge2d->map, GE2D_DST_CLIPX_START_END,
232 		     FIELD_PREP(GE2D_START, ctx->out.crop.left) |
233 		     FIELD_PREP(GE2D_END, ctx->out.crop.left + ctx->out.crop.width - 1));
234 
235 	regmap_write(ge2d->map, GE2D_SRC1_Y_START_END,
236 		     FIELD_PREP(GE2D_END, ctx->in.pix_fmt.height - 1));
237 	regmap_write(ge2d->map, GE2D_SRC1_X_START_END,
238 		     FIELD_PREP(GE2D_END, ctx->in.pix_fmt.width - 1));
239 	regmap_write(ge2d->map, GE2D_SRC2_Y_START_END,
240 		     FIELD_PREP(GE2D_END, ctx->out.pix_fmt.height - 1));
241 	regmap_write(ge2d->map, GE2D_SRC2_X_START_END,
242 		     FIELD_PREP(GE2D_END, ctx->out.pix_fmt.width - 1));
243 	regmap_write(ge2d->map, GE2D_DST_Y_START_END,
244 		     FIELD_PREP(GE2D_END, ctx->out.pix_fmt.height - 1));
245 	regmap_write(ge2d->map, GE2D_DST_X_START_END,
246 		     FIELD_PREP(GE2D_END, ctx->out.pix_fmt.width - 1));
247 
248 	/* Color, no blend, use source color */
249 	reg = GE2D_ALU_DO_COLOR_OPERATION_LOGIC(LOGIC_OPERATION_COPY,
250 						COLOR_FACTOR_SRC_COLOR);
251 
252 	if (ctx->in.fmt->alpha && ctx->out.fmt->alpha)
253 		/* Take source alpha */
254 		reg |= GE2D_ALU_DO_ALPHA_OPERATION_LOGIC(LOGIC_OPERATION_COPY,
255 							 COLOR_FACTOR_SRC_ALPHA);
256 	else if (!ctx->out.fmt->alpha)
257 		/* Set alpha to 0 */
258 		reg |= GE2D_ALU_DO_ALPHA_OPERATION_LOGIC(LOGIC_OPERATION_SET,
259 							 COLOR_FACTOR_ZERO);
260 	else
261 		/* Keep original alpha */
262 		reg |= GE2D_ALU_DO_ALPHA_OPERATION_LOGIC(LOGIC_OPERATION_COPY,
263 							 COLOR_FACTOR_DST_ALPHA);
264 
265 	regmap_write(ge2d->map, GE2D_ALU_OP_CTRL, reg);
266 
267 	/* Start */
268 	regmap_write(ge2d->map, GE2D_CMD_CTRL,
269 		     (ctx->xy_swap ? GE2D_DST_XY_SWAP : 0) |
270 		     (ctx->hflip ? GE2D_SRC1_Y_REV : 0) |
271 		     (ctx->vflip ? GE2D_SRC1_X_REV : 0) |
272 		     GE2D_CBUS_CMD_WR);
273 }
274 
275 static void device_run(void *priv)
276 {
277 	struct ge2d_ctx *ctx = priv;
278 	struct meson_ge2d *ge2d = ctx->ge2d;
279 
280 	ge2d->curr = ctx;
281 
282 	ctx->in.buf = v4l2_m2m_next_src_buf(ctx->fh.m2m_ctx);
283 	ctx->out.buf = v4l2_m2m_next_dst_buf(ctx->fh.m2m_ctx);
284 
285 	ge2d_hw_start(ge2d);
286 }
287 
288 static irqreturn_t ge2d_isr(int irq, void *priv)
289 {
290 	struct meson_ge2d *ge2d = priv;
291 	u32 intr;
292 
293 	regmap_read(ge2d->map, GE2D_STATUS0, &intr);
294 
295 	if (!(intr & GE2D_GE2D_BUSY)) {
296 		struct vb2_v4l2_buffer *src, *dst;
297 		struct ge2d_ctx *ctx = ge2d->curr;
298 
299 		ge2d->curr = NULL;
300 
301 		src = v4l2_m2m_src_buf_remove(ctx->fh.m2m_ctx);
302 		dst = v4l2_m2m_dst_buf_remove(ctx->fh.m2m_ctx);
303 
304 		src->sequence = ctx->sequence_out++;
305 		dst->sequence = ctx->sequence_cap++;
306 
307 		dst->timecode = src->timecode;
308 		dst->vb2_buf.timestamp = src->vb2_buf.timestamp;
309 		dst->flags = src->flags;
310 
311 		v4l2_m2m_buf_done(src, VB2_BUF_STATE_DONE);
312 		v4l2_m2m_buf_done(dst, VB2_BUF_STATE_DONE);
313 		v4l2_m2m_job_finish(ge2d->m2m_dev, ctx->fh.m2m_ctx);
314 	}
315 
316 	return IRQ_HANDLED;
317 }
318 
319 static const struct v4l2_m2m_ops ge2d_m2m_ops = {
320 	.device_run = device_run,
321 };
322 
323 static int ge2d_queue_setup(struct vb2_queue *vq,
324 			    unsigned int *nbuffers, unsigned int *nplanes,
325 			    unsigned int sizes[], struct device *alloc_devs[])
326 {
327 	struct ge2d_ctx *ctx = vb2_get_drv_priv(vq);
328 	struct ge2d_frame *f = get_frame(ctx, vq->type);
329 
330 	if (*nplanes)
331 		return sizes[0] < f->pix_fmt.sizeimage ? -EINVAL : 0;
332 
333 	sizes[0] = f->pix_fmt.sizeimage;
334 	*nplanes = 1;
335 
336 	return 0;
337 }
338 
339 static int ge2d_buf_prepare(struct vb2_buffer *vb)
340 {
341 	struct ge2d_ctx *ctx = vb2_get_drv_priv(vb->vb2_queue);
342 	struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb);
343 	struct ge2d_frame *f = get_frame(ctx, vb->vb2_queue->type);
344 
345 	vbuf->field = V4L2_FIELD_NONE;
346 
347 	vb2_set_plane_payload(vb, 0, f->pix_fmt.sizeimage);
348 
349 	return 0;
350 }
351 
352 static void ge2d_buf_queue(struct vb2_buffer *vb)
353 {
354 	struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb);
355 	struct ge2d_ctx *ctx = vb2_get_drv_priv(vb->vb2_queue);
356 
357 	v4l2_m2m_buf_queue(ctx->fh.m2m_ctx, vbuf);
358 }
359 
360 static int ge2d_start_streaming(struct vb2_queue *vq, unsigned int count)
361 {
362 	struct ge2d_ctx *ctx = vb2_get_drv_priv(vq);
363 
364 	if (V4L2_TYPE_IS_OUTPUT(vq->type))
365 		ctx->sequence_out = 0;
366 	else
367 		ctx->sequence_cap = 0;
368 
369 	return 0;
370 }
371 
372 static void ge2d_stop_streaming(struct vb2_queue *vq)
373 {
374 	struct ge2d_ctx *ctx = vb2_get_drv_priv(vq);
375 	struct vb2_v4l2_buffer *vbuf;
376 
377 	for (;;) {
378 		if (V4L2_TYPE_IS_OUTPUT(vq->type))
379 			vbuf = v4l2_m2m_src_buf_remove(ctx->fh.m2m_ctx);
380 		else
381 			vbuf = v4l2_m2m_dst_buf_remove(ctx->fh.m2m_ctx);
382 		if (!vbuf)
383 			break;
384 		v4l2_m2m_buf_done(vbuf, VB2_BUF_STATE_ERROR);
385 	}
386 }
387 
388 static const struct vb2_ops ge2d_qops = {
389 	.queue_setup = ge2d_queue_setup,
390 	.buf_prepare = ge2d_buf_prepare,
391 	.buf_queue = ge2d_buf_queue,
392 	.start_streaming = ge2d_start_streaming,
393 	.stop_streaming = ge2d_stop_streaming,
394 	.wait_prepare = vb2_ops_wait_prepare,
395 	.wait_finish = vb2_ops_wait_finish,
396 };
397 
398 static int
399 queue_init(void *priv, struct vb2_queue *src_vq, struct vb2_queue *dst_vq)
400 {
401 	struct ge2d_ctx *ctx = priv;
402 	int ret;
403 
404 	src_vq->type = V4L2_BUF_TYPE_VIDEO_OUTPUT;
405 	src_vq->io_modes = VB2_MMAP | VB2_DMABUF;
406 	src_vq->drv_priv = ctx;
407 	src_vq->ops = &ge2d_qops;
408 	src_vq->mem_ops = &vb2_dma_contig_memops;
409 	src_vq->buf_struct_size = sizeof(struct v4l2_m2m_buffer);
410 	src_vq->timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_COPY;
411 	src_vq->lock = &ctx->ge2d->mutex;
412 	src_vq->dev = ctx->ge2d->v4l2_dev.dev;
413 
414 	ret = vb2_queue_init(src_vq);
415 	if (ret)
416 		return ret;
417 
418 	dst_vq->type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
419 	dst_vq->io_modes = VB2_MMAP | VB2_DMABUF;
420 	dst_vq->drv_priv = ctx;
421 	dst_vq->ops = &ge2d_qops;
422 	dst_vq->mem_ops = &vb2_dma_contig_memops;
423 	dst_vq->buf_struct_size = sizeof(struct v4l2_m2m_buffer);
424 	dst_vq->timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_COPY;
425 	dst_vq->lock = &ctx->ge2d->mutex;
426 	dst_vq->dev = ctx->ge2d->v4l2_dev.dev;
427 
428 	return vb2_queue_init(dst_vq);
429 }
430 
431 static int
432 vidioc_querycap(struct file *file, void *priv, struct v4l2_capability *cap)
433 {
434 	strscpy(cap->driver, GE2D_NAME, sizeof(cap->driver));
435 	strscpy(cap->card, GE2D_NAME, sizeof(cap->card));
436 	strscpy(cap->bus_info, "platform:" GE2D_NAME, sizeof(cap->bus_info));
437 
438 	return 0;
439 }
440 
441 static int vidioc_enum_fmt(struct file *file, void *priv, struct v4l2_fmtdesc *f)
442 {
443 	const struct ge2d_fmt *fmt;
444 
445 	if (f->index >= NUM_FORMATS)
446 		return -EINVAL;
447 
448 	fmt = &formats[f->index];
449 	f->pixelformat = fmt->fourcc;
450 
451 	return 0;
452 }
453 
454 static int vidioc_g_selection(struct file *file, void *priv,
455 			      struct v4l2_selection *s)
456 {
457 	struct ge2d_ctx *ctx = priv;
458 	struct ge2d_frame *f;
459 	bool use_frame = false;
460 
461 	if (s->type != V4L2_BUF_TYPE_VIDEO_OUTPUT &&
462 	    s->type != V4L2_BUF_TYPE_VIDEO_CAPTURE)
463 		return -EINVAL;
464 
465 	f = get_frame(ctx, s->type);
466 
467 	switch (s->target) {
468 	case V4L2_SEL_TGT_COMPOSE_DEFAULT:
469 	case V4L2_SEL_TGT_COMPOSE_BOUNDS:
470 		if (s->type != V4L2_BUF_TYPE_VIDEO_CAPTURE)
471 			return -EINVAL;
472 		break;
473 	case V4L2_SEL_TGT_CROP_DEFAULT:
474 	case V4L2_SEL_TGT_CROP_BOUNDS:
475 		if (s->type != V4L2_BUF_TYPE_VIDEO_OUTPUT)
476 			return -EINVAL;
477 		break;
478 	case V4L2_SEL_TGT_COMPOSE:
479 		if (s->type != V4L2_BUF_TYPE_VIDEO_CAPTURE)
480 			return -EINVAL;
481 		use_frame = true;
482 		break;
483 	case V4L2_SEL_TGT_CROP:
484 		if (s->type != V4L2_BUF_TYPE_VIDEO_OUTPUT)
485 			return -EINVAL;
486 		use_frame = true;
487 		break;
488 	default:
489 		return -EINVAL;
490 	}
491 
492 	if (use_frame) {
493 		s->r = f->crop;
494 	} else {
495 		s->r.left = 0;
496 		s->r.top = 0;
497 		s->r.width = f->pix_fmt.width;
498 		s->r.height = f->pix_fmt.height;
499 	}
500 
501 	return 0;
502 }
503 
504 static int vidioc_s_selection(struct file *file, void *priv,
505 			      struct v4l2_selection *s)
506 {
507 	struct ge2d_ctx *ctx = priv;
508 	struct meson_ge2d *ge2d = ctx->ge2d;
509 	struct ge2d_frame *f;
510 	int ret = 0;
511 
512 	if (s->type != V4L2_BUF_TYPE_VIDEO_OUTPUT &&
513 	    s->type != V4L2_BUF_TYPE_VIDEO_CAPTURE)
514 		return -EINVAL;
515 
516 	f = get_frame(ctx, s->type);
517 
518 	switch (s->target) {
519 	case V4L2_SEL_TGT_COMPOSE:
520 		/*
521 		 * COMPOSE target is only valid for capture buffer type, return
522 		 * error for output buffer type
523 		 */
524 		if (s->type != V4L2_BUF_TYPE_VIDEO_CAPTURE)
525 			return -EINVAL;
526 		break;
527 	case V4L2_SEL_TGT_CROP:
528 		/*
529 		 * CROP target is only valid for output buffer type, return
530 		 * error for capture buffer type
531 		 */
532 		if (s->type != V4L2_BUF_TYPE_VIDEO_OUTPUT)
533 			return -EINVAL;
534 		break;
535 	/*
536 	 * bound and default crop/compose targets are invalid targets to
537 	 * try/set
538 	 */
539 	default:
540 		return -EINVAL;
541 	}
542 
543 	if (s->r.top < 0 || s->r.left < 0) {
544 		v4l2_err(&ge2d->v4l2_dev,
545 			 "doesn't support negative values for top & left.\n");
546 		return -EINVAL;
547 	}
548 
549 	if (s->r.left + s->r.width > f->pix_fmt.width ||
550 	    s->r.top + s->r.height > f->pix_fmt.height) {
551 		v4l2_err(&ge2d->v4l2_dev, "unsupported rectangle value.\n");
552 		return -EINVAL;
553 	}
554 
555 	f->crop = s->r;
556 
557 	return ret;
558 }
559 
560 static void vidioc_setup_cap_fmt(struct ge2d_ctx *ctx, struct v4l2_pix_format *f)
561 {
562 	struct ge2d_frame *frm_out = get_frame(ctx, V4L2_BUF_TYPE_VIDEO_OUTPUT);
563 
564 	*f = frm_out->pix_fmt;
565 
566 	if (ctx->xy_swap) {
567 		f->width = frm_out->pix_fmt.height;
568 		f->height = frm_out->pix_fmt.width;
569 	}
570 }
571 
572 static int vidioc_try_fmt_cap(struct file *file, void *priv, struct v4l2_format *f)
573 {
574 	const struct ge2d_fmt *fmt = find_fmt(f);
575 	struct ge2d_ctx *ctx = priv;
576 	struct v4l2_pix_format fmt_cap;
577 
578 	vidioc_setup_cap_fmt(ctx, &fmt_cap);
579 
580 	fmt_cap.pixelformat = fmt->fourcc;
581 
582 	fmt_cap.bytesperline = max(f->fmt.pix.bytesperline,
583 				   ALIGN((fmt_cap.width * fmt->depth) >> 3, 8));
584 
585 	fmt_cap.sizeimage = max(f->fmt.pix.sizeimage,
586 				fmt_cap.height * fmt_cap.bytesperline);
587 
588 	f->fmt.pix = fmt_cap;
589 
590 	return 0;
591 }
592 
593 static int vidioc_s_fmt_cap(struct file *file, void *priv, struct v4l2_format *f)
594 {
595 	struct ge2d_ctx *ctx = priv;
596 	struct meson_ge2d *ge2d = ctx->ge2d;
597 	struct vb2_queue *vq;
598 	struct ge2d_frame *frm;
599 	int ret = 0;
600 
601 	/* Adjust all values accordingly to the hardware capabilities
602 	 * and chosen format.
603 	 */
604 	ret = vidioc_try_fmt_cap(file, priv, f);
605 	if (ret)
606 		return ret;
607 
608 	vq = v4l2_m2m_get_vq(ctx->fh.m2m_ctx, f->type);
609 	if (vb2_is_busy(vq)) {
610 		v4l2_err(&ge2d->v4l2_dev, "queue (%d) bust\n", f->type);
611 		return -EBUSY;
612 	}
613 
614 	frm = get_frame(ctx, V4L2_BUF_TYPE_VIDEO_CAPTURE);
615 
616 	frm->pix_fmt = f->fmt.pix;
617 	frm->fmt = find_fmt(f);
618 	f->fmt.pix.pixelformat = frm->fmt->fourcc;
619 
620 	/* Reset crop settings */
621 	frm->crop.left = 0;
622 	frm->crop.top = 0;
623 	frm->crop.width = frm->pix_fmt.width;
624 	frm->crop.height = frm->pix_fmt.height;
625 
626 	return 0;
627 }
628 
629 static int vidioc_g_fmt(struct file *file, void *priv, struct v4l2_format *f)
630 {
631 	struct ge2d_ctx *ctx = priv;
632 	struct vb2_queue *vq;
633 	struct ge2d_frame *frm;
634 
635 	vq = v4l2_m2m_get_vq(ctx->fh.m2m_ctx, f->type);
636 	if (!vq)
637 		return -EINVAL;
638 
639 	frm = get_frame(ctx, f->type);
640 
641 	f->fmt.pix = frm->pix_fmt;
642 	f->fmt.pix.pixelformat = frm->fmt->fourcc;
643 
644 	return 0;
645 }
646 
647 static int vidioc_try_fmt_out(struct file *file, void *priv, struct v4l2_format *f)
648 {
649 	const struct ge2d_fmt *fmt = find_fmt(f);
650 
651 	f->fmt.pix.field = V4L2_FIELD_NONE;
652 	f->fmt.pix.pixelformat = fmt->fourcc;
653 
654 	if (f->fmt.pix.width > MAX_WIDTH)
655 		f->fmt.pix.width = MAX_WIDTH;
656 	if (f->fmt.pix.height > MAX_HEIGHT)
657 		f->fmt.pix.height = MAX_HEIGHT;
658 
659 	f->fmt.pix.bytesperline = max(f->fmt.pix.bytesperline,
660 				      ALIGN((f->fmt.pix.width * fmt->depth) >> 3, 8));
661 
662 	f->fmt.pix.sizeimage = max(f->fmt.pix.sizeimage,
663 				   f->fmt.pix.height * f->fmt.pix.bytesperline);
664 
665 	return 0;
666 }
667 
668 static int vidioc_s_fmt_out(struct file *file, void *priv, struct v4l2_format *f)
669 {
670 	struct ge2d_ctx *ctx = priv;
671 	struct meson_ge2d *ge2d = ctx->ge2d;
672 	struct vb2_queue *vq;
673 	struct ge2d_frame *frm, *frm_cap;
674 	int ret = 0;
675 
676 	/* Adjust all values accordingly to the hardware capabilities
677 	 * and chosen format.
678 	 */
679 	ret = vidioc_try_fmt_out(file, priv, f);
680 	if (ret)
681 		return ret;
682 
683 	vq = v4l2_m2m_get_vq(ctx->fh.m2m_ctx, f->type);
684 	if (vb2_is_busy(vq)) {
685 		v4l2_err(&ge2d->v4l2_dev, "queue (%d) bust\n", f->type);
686 		return -EBUSY;
687 	}
688 
689 	frm = get_frame(ctx, V4L2_BUF_TYPE_VIDEO_OUTPUT);
690 	frm_cap = get_frame(ctx, V4L2_BUF_TYPE_VIDEO_CAPTURE);
691 
692 	frm->pix_fmt = f->fmt.pix;
693 	frm->fmt = find_fmt(f);
694 	f->fmt.pix.pixelformat = frm->fmt->fourcc;
695 
696 	/* Reset crop settings */
697 	frm->crop.left = 0;
698 	frm->crop.top = 0;
699 	frm->crop.width = frm->pix_fmt.width;
700 	frm->crop.height = frm->pix_fmt.height;
701 
702 	/* Propagate settings to capture */
703 	vidioc_setup_cap_fmt(ctx, &frm_cap->pix_fmt);
704 
705 	return 0;
706 }
707 
708 static const struct v4l2_ioctl_ops ge2d_ioctl_ops = {
709 	.vidioc_querycap = vidioc_querycap,
710 
711 	.vidioc_enum_fmt_vid_cap = vidioc_enum_fmt,
712 	.vidioc_g_fmt_vid_cap = vidioc_g_fmt,
713 	.vidioc_try_fmt_vid_cap = vidioc_try_fmt_cap,
714 	.vidioc_s_fmt_vid_cap = vidioc_s_fmt_cap,
715 
716 	.vidioc_enum_fmt_vid_out = vidioc_enum_fmt,
717 	.vidioc_g_fmt_vid_out = vidioc_g_fmt,
718 	.vidioc_try_fmt_vid_out = vidioc_try_fmt_out,
719 	.vidioc_s_fmt_vid_out = vidioc_s_fmt_out,
720 
721 	.vidioc_reqbufs = v4l2_m2m_ioctl_reqbufs,
722 	.vidioc_querybuf = v4l2_m2m_ioctl_querybuf,
723 	.vidioc_qbuf = v4l2_m2m_ioctl_qbuf,
724 	.vidioc_dqbuf = v4l2_m2m_ioctl_dqbuf,
725 	.vidioc_prepare_buf = v4l2_m2m_ioctl_prepare_buf,
726 	.vidioc_create_bufs = v4l2_m2m_ioctl_create_bufs,
727 	.vidioc_expbuf = v4l2_m2m_ioctl_expbuf,
728 
729 	.vidioc_subscribe_event = v4l2_ctrl_subscribe_event,
730 	.vidioc_unsubscribe_event = v4l2_event_unsubscribe,
731 
732 	.vidioc_streamon = v4l2_m2m_ioctl_streamon,
733 	.vidioc_streamoff = v4l2_m2m_ioctl_streamoff,
734 
735 	.vidioc_g_selection = vidioc_g_selection,
736 	.vidioc_s_selection = vidioc_s_selection,
737 };
738 
739 static int ge2d_s_ctrl(struct v4l2_ctrl *ctrl)
740 {
741 	struct ge2d_ctx *ctx = container_of(ctrl->handler, struct ge2d_ctx,
742 					   ctrl_handler);
743 	struct v4l2_pix_format fmt;
744 	struct vb2_queue *vq;
745 
746 	switch (ctrl->id) {
747 	case V4L2_CID_HFLIP:
748 		ctx->hflip = ctrl->val;
749 		break;
750 	case V4L2_CID_VFLIP:
751 		ctx->vflip = ctrl->val;
752 		break;
753 	case V4L2_CID_ROTATE:
754 		vq = v4l2_m2m_get_vq(ctx->fh.m2m_ctx, V4L2_BUF_TYPE_VIDEO_CAPTURE);
755 		if (vb2_is_busy(vq))
756 			return -EBUSY;
757 
758 		if (ctrl->val == 90) {
759 			ctx->hflip = 0;
760 			ctx->vflip = 1;
761 			ctx->xy_swap = 1;
762 		} else if (ctrl->val == 180) {
763 			ctx->hflip = 1;
764 			ctx->vflip = 1;
765 			ctx->xy_swap = 0;
766 		} else if (ctrl->val == 270) {
767 			ctx->hflip = 1;
768 			ctx->vflip = 0;
769 			ctx->xy_swap = 1;
770 		} else {
771 			ctx->hflip = 0;
772 			ctx->vflip = 0;
773 			ctx->xy_swap = 0;
774 		}
775 
776 		vidioc_setup_cap_fmt(ctx, &fmt);
777 
778 		/*
779 		 * If the rotation parameter changes the OUTPUT frames
780 		 * parameters, take them in account
781 		 */
782 		ctx->out.pix_fmt = fmt;
783 
784 		break;
785 	}
786 
787 	return 0;
788 }
789 
790 static const struct v4l2_ctrl_ops ge2d_ctrl_ops = {
791 	.s_ctrl = ge2d_s_ctrl,
792 };
793 
794 static int ge2d_setup_ctrls(struct ge2d_ctx *ctx)
795 {
796 	struct meson_ge2d *ge2d = ctx->ge2d;
797 
798 	v4l2_ctrl_handler_init(&ctx->ctrl_handler, 4);
799 
800 	v4l2_ctrl_new_std(&ctx->ctrl_handler, &ge2d_ctrl_ops,
801 			  V4L2_CID_HFLIP, 0, 1, 1, 0);
802 
803 	v4l2_ctrl_new_std(&ctx->ctrl_handler, &ge2d_ctrl_ops,
804 			  V4L2_CID_VFLIP, 0, 1, 1, 0);
805 
806 	v4l2_ctrl_new_std(&ctx->ctrl_handler, &ge2d_ctrl_ops,
807 			  V4L2_CID_ROTATE, 0, 270, 90, 0);
808 
809 	if (ctx->ctrl_handler.error) {
810 		int err = ctx->ctrl_handler.error;
811 
812 		v4l2_err(&ge2d->v4l2_dev, "%s failed\n", __func__);
813 		v4l2_ctrl_handler_free(&ctx->ctrl_handler);
814 		return err;
815 	}
816 
817 	return 0;
818 }
819 
820 static const struct ge2d_frame def_frame = {
821 	.pix_fmt = {
822 		.width = DEFAULT_WIDTH,
823 		.height = DEFAULT_HEIGHT,
824 		.bytesperline = DEFAULT_STRIDE,
825 		.sizeimage = DEFAULT_STRIDE * DEFAULT_HEIGHT,
826 		.field = V4L2_FIELD_NONE,
827 	},
828 	.crop.width = DEFAULT_WIDTH,
829 	.crop.height = DEFAULT_HEIGHT,
830 	.fmt = &formats[0],
831 };
832 
833 static int ge2d_open(struct file *file)
834 {
835 	struct meson_ge2d *ge2d = video_drvdata(file);
836 	struct ge2d_ctx *ctx = NULL;
837 	int ret = 0;
838 
839 	ctx = kzalloc(sizeof(*ctx), GFP_KERNEL);
840 	if (!ctx)
841 		return -ENOMEM;
842 	ctx->ge2d = ge2d;
843 
844 	/* Set default formats */
845 	ctx->in = def_frame;
846 	ctx->out = def_frame;
847 
848 	if (mutex_lock_interruptible(&ge2d->mutex)) {
849 		kfree(ctx);
850 		return -ERESTARTSYS;
851 	}
852 	ctx->fh.m2m_ctx = v4l2_m2m_ctx_init(ge2d->m2m_dev, ctx, &queue_init);
853 	if (IS_ERR(ctx->fh.m2m_ctx)) {
854 		ret = PTR_ERR(ctx->fh.m2m_ctx);
855 		mutex_unlock(&ge2d->mutex);
856 		kfree(ctx);
857 		return ret;
858 	}
859 	v4l2_fh_init(&ctx->fh, video_devdata(file));
860 	file->private_data = &ctx->fh;
861 	v4l2_fh_add(&ctx->fh);
862 
863 	ge2d_setup_ctrls(ctx);
864 
865 	/* Write the default values to the ctx struct */
866 	v4l2_ctrl_handler_setup(&ctx->ctrl_handler);
867 
868 	ctx->fh.ctrl_handler = &ctx->ctrl_handler;
869 	mutex_unlock(&ge2d->mutex);
870 
871 	return 0;
872 }
873 
874 static int ge2d_release(struct file *file)
875 {
876 	struct ge2d_ctx *ctx =
877 		container_of(file->private_data, struct ge2d_ctx, fh);
878 	struct meson_ge2d *ge2d = ctx->ge2d;
879 
880 	mutex_lock(&ge2d->mutex);
881 
882 	v4l2_m2m_ctx_release(ctx->fh.m2m_ctx);
883 
884 	v4l2_ctrl_handler_free(&ctx->ctrl_handler);
885 	v4l2_fh_del(&ctx->fh);
886 	v4l2_fh_exit(&ctx->fh);
887 	kfree(ctx);
888 
889 	mutex_unlock(&ge2d->mutex);
890 
891 	return 0;
892 }
893 
894 static const struct v4l2_file_operations ge2d_fops = {
895 	.owner = THIS_MODULE,
896 	.open = ge2d_open,
897 	.release = ge2d_release,
898 	.poll = v4l2_m2m_fop_poll,
899 	.unlocked_ioctl = video_ioctl2,
900 	.mmap = v4l2_m2m_fop_mmap,
901 };
902 
903 static const struct video_device ge2d_videodev = {
904 	.name = "meson-ge2d",
905 	.fops = &ge2d_fops,
906 	.ioctl_ops = &ge2d_ioctl_ops,
907 	.minor = -1,
908 	.release = video_device_release,
909 	.vfl_dir = VFL_DIR_M2M,
910 	.device_caps = V4L2_CAP_VIDEO_M2M | V4L2_CAP_STREAMING,
911 };
912 
913 static const struct regmap_config meson_ge2d_regmap_conf = {
914 	.reg_bits = 8,
915 	.val_bits = 32,
916 	.reg_stride = 4,
917 	.max_register = GE2D_SRC2_STRIDE_CTRL,
918 };
919 
920 static int ge2d_probe(struct platform_device *pdev)
921 {
922 	struct reset_control *rst;
923 	struct video_device *vfd;
924 	struct meson_ge2d *ge2d;
925 	void __iomem *regs;
926 	int ret = 0;
927 	int irq;
928 
929 	if (!pdev->dev.of_node)
930 		return -ENODEV;
931 
932 	ge2d = devm_kzalloc(&pdev->dev, sizeof(*ge2d), GFP_KERNEL);
933 	if (!ge2d)
934 		return -ENOMEM;
935 
936 	ge2d->dev = &pdev->dev;
937 	mutex_init(&ge2d->mutex);
938 
939 	regs = devm_platform_ioremap_resource(pdev, 0);
940 	if (IS_ERR(regs))
941 		return PTR_ERR(regs);
942 
943 	ge2d->map = devm_regmap_init_mmio(ge2d->dev, regs,
944 					  &meson_ge2d_regmap_conf);
945 	if (IS_ERR(ge2d->map))
946 		return PTR_ERR(ge2d->map);
947 
948 	irq = platform_get_irq(pdev, 0);
949 	ret = devm_request_irq(ge2d->dev, irq, ge2d_isr, 0,
950 			       dev_name(ge2d->dev), ge2d);
951 	if (ret < 0) {
952 		dev_err(ge2d->dev, "failed to request irq\n");
953 		return ret;
954 	}
955 
956 	rst = devm_reset_control_get(ge2d->dev, NULL);
957 	if (IS_ERR(rst)) {
958 		dev_err(ge2d->dev, "failed to get core reset controller\n");
959 		return PTR_ERR(rst);
960 	}
961 
962 	ge2d->clk = devm_clk_get(ge2d->dev, NULL);
963 	if (IS_ERR(ge2d->clk)) {
964 		dev_err(ge2d->dev, "failed to get clock\n");
965 		return PTR_ERR(ge2d->clk);
966 	}
967 
968 	reset_control_assert(rst);
969 	udelay(1);
970 	reset_control_deassert(rst);
971 
972 	ret = clk_prepare_enable(ge2d->clk);
973 	if (ret) {
974 		dev_err(ge2d->dev, "Cannot enable ge2d sclk: %d\n", ret);
975 		return ret;
976 	}
977 
978 	ret = v4l2_device_register(&pdev->dev, &ge2d->v4l2_dev);
979 	if (ret)
980 		goto disable_clks;
981 
982 	vfd = video_device_alloc();
983 	if (!vfd) {
984 		v4l2_err(&ge2d->v4l2_dev, "Failed to allocate video device\n");
985 		ret = -ENOMEM;
986 		goto unreg_v4l2_dev;
987 	}
988 
989 	*vfd = ge2d_videodev;
990 	vfd->lock = &ge2d->mutex;
991 	vfd->v4l2_dev = &ge2d->v4l2_dev;
992 
993 	video_set_drvdata(vfd, ge2d);
994 	ge2d->vfd = vfd;
995 
996 	platform_set_drvdata(pdev, ge2d);
997 	ge2d->m2m_dev = v4l2_m2m_init(&ge2d_m2m_ops);
998 	if (IS_ERR(ge2d->m2m_dev)) {
999 		v4l2_err(&ge2d->v4l2_dev, "Failed to init mem2mem device\n");
1000 		ret = PTR_ERR(ge2d->m2m_dev);
1001 		goto rel_vdev;
1002 	}
1003 
1004 	ret = video_register_device(vfd, VFL_TYPE_VIDEO, -1);
1005 	if (ret) {
1006 		v4l2_err(&ge2d->v4l2_dev, "Failed to register video device\n");
1007 		goto rel_m2m;
1008 	}
1009 
1010 	v4l2_info(&ge2d->v4l2_dev, "Registered %s as /dev/%s\n",
1011 		  vfd->name, video_device_node_name(vfd));
1012 
1013 	return 0;
1014 
1015 rel_m2m:
1016 	v4l2_m2m_release(ge2d->m2m_dev);
1017 rel_vdev:
1018 	video_device_release(ge2d->vfd);
1019 unreg_v4l2_dev:
1020 	v4l2_device_unregister(&ge2d->v4l2_dev);
1021 disable_clks:
1022 	clk_disable_unprepare(ge2d->clk);
1023 
1024 	return ret;
1025 }
1026 
1027 static int ge2d_remove(struct platform_device *pdev)
1028 {
1029 	struct meson_ge2d *ge2d = platform_get_drvdata(pdev);
1030 
1031 	video_unregister_device(ge2d->vfd);
1032 	v4l2_m2m_release(ge2d->m2m_dev);
1033 	video_device_release(ge2d->vfd);
1034 	v4l2_device_unregister(&ge2d->v4l2_dev);
1035 	clk_disable_unprepare(ge2d->clk);
1036 
1037 	return 0;
1038 }
1039 
1040 static const struct of_device_id meson_ge2d_match[] = {
1041 	{
1042 		.compatible = "amlogic,axg-ge2d",
1043 	},
1044 	{},
1045 };
1046 
1047 MODULE_DEVICE_TABLE(of, meson_ge2d_match);
1048 
1049 static struct platform_driver ge2d_drv = {
1050 	.probe = ge2d_probe,
1051 	.remove = ge2d_remove,
1052 	.driver = {
1053 		.name = "meson-ge2d",
1054 		.of_match_table = meson_ge2d_match,
1055 	},
1056 };
1057 
1058 module_platform_driver(ge2d_drv);
1059 
1060 MODULE_AUTHOR("Neil Armstrong <narmstrong@baylibre.com>");
1061 MODULE_DESCRIPTION("Amlogic 2D Graphic Acceleration Unit");
1062 MODULE_LICENSE("GPL");
1063