xref: /linux/drivers/media/platform/nxp/mx2_emmaprp.c (revision 55a42f78ffd386e01a5404419f8c5ded7db70a21)
1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  * Support eMMa-PrP through mem2mem framework.
4  *
5  * eMMa-PrP is a piece of HW that allows fetching buffers
6  * from one memory location and do several operations on
7  * them such as scaling or format conversion giving, as a result
8  * a new processed buffer in another memory location.
9  *
10  * Based on mem2mem_testdev.c by Pawel Osciak.
11  *
12  * Copyright (c) 2011 Vista Silicon S.L.
13  * Javier Martin <javier.martin@vista-silicon.com>
14  */
15 #include <linux/module.h>
16 #include <linux/clk.h>
17 #include <linux/slab.h>
18 #include <linux/interrupt.h>
19 #include <linux/io.h>
20 
21 #include <linux/platform_device.h>
22 #include <media/v4l2-mem2mem.h>
23 #include <media/v4l2-device.h>
24 #include <media/v4l2-ioctl.h>
25 #include <media/videobuf2-dma-contig.h>
26 #include <linux/sizes.h>
27 
28 #define EMMAPRP_MODULE_NAME "mem2mem-emmaprp"
29 
30 MODULE_DESCRIPTION("Mem-to-mem device which supports eMMa-PrP present in mx2 SoCs");
31 MODULE_AUTHOR("Javier Martin <javier.martin@vista-silicon.com");
32 MODULE_LICENSE("GPL");
33 MODULE_VERSION("0.0.1");
34 
35 static bool debug;
36 module_param(debug, bool, 0644);
37 
38 #define MIN_W 32
39 #define MIN_H 32
40 #define MAX_W 2040
41 #define MAX_H 2046
42 
43 #define S_ALIGN		1 /* multiple of 2 */
44 #define W_ALIGN_YUV420	3 /* multiple of 8 */
45 #define W_ALIGN_OTHERS	2 /* multiple of 4 */
46 #define H_ALIGN		1 /* multiple of 2 */
47 
48 /* Flags that indicate a format can be used for capture/output */
49 #define MEM2MEM_CAPTURE	(1 << 0)
50 #define MEM2MEM_OUTPUT	(1 << 1)
51 
52 #define MEM2MEM_NAME		"m2m-emmaprp"
53 
54 /* In bytes, per queue */
55 #define MEM2MEM_VID_MEM_LIMIT	SZ_16M
56 
57 #define dprintk(dev, fmt, arg...) \
58 	v4l2_dbg(1, debug, &dev->v4l2_dev, "%s: " fmt, __func__, ## arg)
59 
60 /* EMMA PrP */
61 #define PRP_CNTL                        0x00
62 #define PRP_INTR_CNTL                   0x04
63 #define PRP_INTRSTATUS                  0x08
64 #define PRP_SOURCE_Y_PTR                0x0c
65 #define PRP_SOURCE_CB_PTR               0x10
66 #define PRP_SOURCE_CR_PTR               0x14
67 #define PRP_DEST_RGB1_PTR               0x18
68 #define PRP_DEST_RGB2_PTR               0x1c
69 #define PRP_DEST_Y_PTR                  0x20
70 #define PRP_DEST_CB_PTR                 0x24
71 #define PRP_DEST_CR_PTR                 0x28
72 #define PRP_SRC_FRAME_SIZE              0x2c
73 #define PRP_DEST_CH1_LINE_STRIDE        0x30
74 #define PRP_SRC_PIXEL_FORMAT_CNTL       0x34
75 #define PRP_CH1_PIXEL_FORMAT_CNTL       0x38
76 #define PRP_CH1_OUT_IMAGE_SIZE          0x3c
77 #define PRP_CH2_OUT_IMAGE_SIZE          0x40
78 #define PRP_SRC_LINE_STRIDE             0x44
79 #define PRP_CSC_COEF_012                0x48
80 #define PRP_CSC_COEF_345                0x4c
81 #define PRP_CSC_COEF_678                0x50
82 #define PRP_CH1_RZ_HORI_COEF1           0x54
83 #define PRP_CH1_RZ_HORI_COEF2           0x58
84 #define PRP_CH1_RZ_HORI_VALID           0x5c
85 #define PRP_CH1_RZ_VERT_COEF1           0x60
86 #define PRP_CH1_RZ_VERT_COEF2           0x64
87 #define PRP_CH1_RZ_VERT_VALID           0x68
88 #define PRP_CH2_RZ_HORI_COEF1           0x6c
89 #define PRP_CH2_RZ_HORI_COEF2           0x70
90 #define PRP_CH2_RZ_HORI_VALID           0x74
91 #define PRP_CH2_RZ_VERT_COEF1           0x78
92 #define PRP_CH2_RZ_VERT_COEF2           0x7c
93 #define PRP_CH2_RZ_VERT_VALID           0x80
94 
95 #define PRP_CNTL_CH1EN          (1 << 0)
96 #define PRP_CNTL_CH2EN          (1 << 1)
97 #define PRP_CNTL_CSIEN          (1 << 2)
98 #define PRP_CNTL_DATA_IN_YUV420 (0 << 3)
99 #define PRP_CNTL_DATA_IN_YUV422 (1 << 3)
100 #define PRP_CNTL_DATA_IN_RGB16  (2 << 3)
101 #define PRP_CNTL_DATA_IN_RGB32  (3 << 3)
102 #define PRP_CNTL_CH1_OUT_RGB8   (0 << 5)
103 #define PRP_CNTL_CH1_OUT_RGB16  (1 << 5)
104 #define PRP_CNTL_CH1_OUT_RGB32  (2 << 5)
105 #define PRP_CNTL_CH1_OUT_YUV422 (3 << 5)
106 #define PRP_CNTL_CH2_OUT_YUV420 (0 << 7)
107 #define PRP_CNTL_CH2_OUT_YUV422 (1 << 7)
108 #define PRP_CNTL_CH2_OUT_YUV444 (2 << 7)
109 #define PRP_CNTL_CH1_LEN        (1 << 9)
110 #define PRP_CNTL_CH2_LEN        (1 << 10)
111 #define PRP_CNTL_SKIP_FRAME     (1 << 11)
112 #define PRP_CNTL_SWRST          (1 << 12)
113 #define PRP_CNTL_CLKEN          (1 << 13)
114 #define PRP_CNTL_WEN            (1 << 14)
115 #define PRP_CNTL_CH1BYP         (1 << 15)
116 #define PRP_CNTL_IN_TSKIP(x)    ((x) << 16)
117 #define PRP_CNTL_CH1_TSKIP(x)   ((x) << 19)
118 #define PRP_CNTL_CH2_TSKIP(x)   ((x) << 22)
119 #define PRP_CNTL_INPUT_FIFO_LEVEL(x)    ((x) << 25)
120 #define PRP_CNTL_RZ_FIFO_LEVEL(x)       ((x) << 27)
121 #define PRP_CNTL_CH2B1EN        (1 << 29)
122 #define PRP_CNTL_CH2B2EN        (1 << 30)
123 #define PRP_CNTL_CH2FEN         (1UL << 31)
124 
125 #define PRP_SIZE_HEIGHT(x)	(x)
126 #define PRP_SIZE_WIDTH(x)	((x) << 16)
127 
128 /* IRQ Enable and status register */
129 #define PRP_INTR_RDERR          (1 << 0)
130 #define PRP_INTR_CH1WERR        (1 << 1)
131 #define PRP_INTR_CH2WERR        (1 << 2)
132 #define PRP_INTR_CH1FC          (1 << 3)
133 #define PRP_INTR_CH2FC          (1 << 5)
134 #define PRP_INTR_LBOVF          (1 << 7)
135 #define PRP_INTR_CH2OVF         (1 << 8)
136 
137 #define PRP_INTR_ST_RDERR	(1 << 0)
138 #define PRP_INTR_ST_CH1WERR	(1 << 1)
139 #define PRP_INTR_ST_CH2WERR	(1 << 2)
140 #define PRP_INTR_ST_CH2B2CI	(1 << 3)
141 #define PRP_INTR_ST_CH2B1CI	(1 << 4)
142 #define PRP_INTR_ST_CH1B2CI	(1 << 5)
143 #define PRP_INTR_ST_CH1B1CI	(1 << 6)
144 #define PRP_INTR_ST_LBOVF	(1 << 7)
145 #define PRP_INTR_ST_CH2OVF	(1 << 8)
146 
147 struct emmaprp_fmt {
148 	u32	fourcc;
149 	/* Types the format can be used for */
150 	u32	types;
151 };
152 
153 static struct emmaprp_fmt formats[] = {
154 	{
155 		.fourcc	= V4L2_PIX_FMT_YUV420,
156 		.types	= MEM2MEM_CAPTURE,
157 	},
158 	{
159 		.fourcc	= V4L2_PIX_FMT_YUYV,
160 		.types	= MEM2MEM_OUTPUT,
161 	},
162 };
163 
164 /* Per-queue, driver-specific private data */
165 struct emmaprp_q_data {
166 	unsigned int		width;
167 	unsigned int		height;
168 	unsigned int		sizeimage;
169 	struct emmaprp_fmt	*fmt;
170 };
171 
172 enum {
173 	V4L2_M2M_SRC = 0,
174 	V4L2_M2M_DST = 1,
175 };
176 
177 #define NUM_FORMATS ARRAY_SIZE(formats)
178 
179 static struct emmaprp_fmt *find_format(struct v4l2_format *f)
180 {
181 	struct emmaprp_fmt *fmt;
182 	unsigned int k;
183 
184 	for (k = 0; k < NUM_FORMATS; k++) {
185 		fmt = &formats[k];
186 		if (fmt->fourcc == f->fmt.pix.pixelformat)
187 			break;
188 	}
189 
190 	if (k == NUM_FORMATS)
191 		return NULL;
192 
193 	return &formats[k];
194 }
195 
196 struct emmaprp_dev {
197 	struct v4l2_device	v4l2_dev;
198 	struct video_device	*vfd;
199 
200 	struct mutex		dev_mutex;
201 	spinlock_t		irqlock;
202 
203 	void __iomem		*base_emma;
204 	struct clk		*clk_emma_ahb, *clk_emma_ipg;
205 
206 	struct v4l2_m2m_dev	*m2m_dev;
207 };
208 
209 struct emmaprp_ctx {
210 	struct v4l2_fh		fh;
211 	struct emmaprp_dev	*dev;
212 	/* Abort requested by m2m */
213 	int			aborting;
214 	struct emmaprp_q_data	q_data[2];
215 };
216 
217 static inline struct emmaprp_ctx *file_to_emmaprp_ctx(struct file *filp)
218 {
219 	return container_of(file_to_v4l2_fh(filp), struct emmaprp_ctx, fh);
220 }
221 
222 static struct emmaprp_q_data *get_q_data(struct emmaprp_ctx *ctx,
223 					 enum v4l2_buf_type type)
224 {
225 	switch (type) {
226 	case V4L2_BUF_TYPE_VIDEO_OUTPUT:
227 		return &(ctx->q_data[V4L2_M2M_SRC]);
228 	case V4L2_BUF_TYPE_VIDEO_CAPTURE:
229 		return &(ctx->q_data[V4L2_M2M_DST]);
230 	default:
231 		BUG();
232 	}
233 	return NULL;
234 }
235 
236 /*
237  * mem2mem callbacks
238  */
239 static void emmaprp_job_abort(void *priv)
240 {
241 	struct emmaprp_ctx *ctx = priv;
242 	struct emmaprp_dev *pcdev = ctx->dev;
243 
244 	ctx->aborting = 1;
245 
246 	dprintk(pcdev, "Aborting task\n");
247 
248 	v4l2_m2m_job_finish(pcdev->m2m_dev, ctx->fh.m2m_ctx);
249 }
250 
251 static inline void emmaprp_dump_regs(struct emmaprp_dev *pcdev)
252 {
253 	dprintk(pcdev,
254 		"eMMa-PrP Registers:\n"
255 		"  SOURCE_Y_PTR = 0x%08X\n"
256 		"  SRC_FRAME_SIZE = 0x%08X\n"
257 		"  DEST_Y_PTR = 0x%08X\n"
258 		"  DEST_CR_PTR = 0x%08X\n"
259 		"  DEST_CB_PTR = 0x%08X\n"
260 		"  CH2_OUT_IMAGE_SIZE = 0x%08X\n"
261 		"  CNTL = 0x%08X\n",
262 		readl(pcdev->base_emma + PRP_SOURCE_Y_PTR),
263 		readl(pcdev->base_emma + PRP_SRC_FRAME_SIZE),
264 		readl(pcdev->base_emma + PRP_DEST_Y_PTR),
265 		readl(pcdev->base_emma + PRP_DEST_CR_PTR),
266 		readl(pcdev->base_emma + PRP_DEST_CB_PTR),
267 		readl(pcdev->base_emma + PRP_CH2_OUT_IMAGE_SIZE),
268 		readl(pcdev->base_emma + PRP_CNTL));
269 }
270 
271 static void emmaprp_device_run(void *priv)
272 {
273 	struct emmaprp_ctx *ctx = priv;
274 	struct emmaprp_q_data *s_q_data, *d_q_data;
275 	struct vb2_v4l2_buffer *src_buf, *dst_buf;
276 	struct emmaprp_dev *pcdev = ctx->dev;
277 	unsigned int s_width, s_height;
278 	unsigned int d_width, d_height;
279 	unsigned int d_size;
280 	dma_addr_t p_in, p_out;
281 	u32 tmp;
282 
283 	src_buf = v4l2_m2m_next_src_buf(ctx->fh.m2m_ctx);
284 	dst_buf = v4l2_m2m_next_dst_buf(ctx->fh.m2m_ctx);
285 
286 	s_q_data = get_q_data(ctx, V4L2_BUF_TYPE_VIDEO_OUTPUT);
287 	s_width	= s_q_data->width;
288 	s_height = s_q_data->height;
289 
290 	d_q_data = get_q_data(ctx, V4L2_BUF_TYPE_VIDEO_CAPTURE);
291 	d_width = d_q_data->width;
292 	d_height = d_q_data->height;
293 	d_size = d_width * d_height;
294 
295 	p_in = vb2_dma_contig_plane_dma_addr(&src_buf->vb2_buf, 0);
296 	p_out = vb2_dma_contig_plane_dma_addr(&dst_buf->vb2_buf, 0);
297 	if (!p_in || !p_out) {
298 		v4l2_err(&pcdev->v4l2_dev,
299 			 "Acquiring kernel pointers to buffers failed\n");
300 		return;
301 	}
302 
303 	/* Input frame parameters */
304 	writel(p_in, pcdev->base_emma + PRP_SOURCE_Y_PTR);
305 	writel(PRP_SIZE_WIDTH(s_width) | PRP_SIZE_HEIGHT(s_height),
306 	       pcdev->base_emma + PRP_SRC_FRAME_SIZE);
307 
308 	/* Output frame parameters */
309 	writel(p_out, pcdev->base_emma + PRP_DEST_Y_PTR);
310 	writel(p_out + d_size, pcdev->base_emma + PRP_DEST_CB_PTR);
311 	writel(p_out + d_size + (d_size >> 2),
312 	       pcdev->base_emma + PRP_DEST_CR_PTR);
313 	writel(PRP_SIZE_WIDTH(d_width) | PRP_SIZE_HEIGHT(d_height),
314 	       pcdev->base_emma + PRP_CH2_OUT_IMAGE_SIZE);
315 
316 	/* IRQ configuration */
317 	tmp = readl(pcdev->base_emma + PRP_INTR_CNTL);
318 	writel(tmp | PRP_INTR_RDERR |
319 		PRP_INTR_CH2WERR |
320 		PRP_INTR_CH2FC,
321 		pcdev->base_emma + PRP_INTR_CNTL);
322 
323 	emmaprp_dump_regs(pcdev);
324 
325 	/* Enable transfer */
326 	tmp = readl(pcdev->base_emma + PRP_CNTL);
327 	writel(tmp | PRP_CNTL_CH2_OUT_YUV420 |
328 		PRP_CNTL_DATA_IN_YUV422 |
329 		PRP_CNTL_CH2EN,
330 		pcdev->base_emma + PRP_CNTL);
331 }
332 
333 static irqreturn_t emmaprp_irq(int irq_emma, void *data)
334 {
335 	struct emmaprp_dev *pcdev = data;
336 	struct emmaprp_ctx *curr_ctx;
337 	struct vb2_v4l2_buffer *src_vb, *dst_vb;
338 	unsigned long flags;
339 	u32 irqst;
340 
341 	/* Check irq flags and clear irq */
342 	irqst = readl(pcdev->base_emma + PRP_INTRSTATUS);
343 	writel(irqst, pcdev->base_emma + PRP_INTRSTATUS);
344 	dprintk(pcdev, "irqst = 0x%08x\n", irqst);
345 
346 	curr_ctx = v4l2_m2m_get_curr_priv(pcdev->m2m_dev);
347 	if (curr_ctx == NULL) {
348 		pr_err("Instance released before the end of transaction\n");
349 		return IRQ_HANDLED;
350 	}
351 
352 	if (!curr_ctx->aborting) {
353 		if ((irqst & PRP_INTR_ST_RDERR) ||
354 		(irqst & PRP_INTR_ST_CH2WERR)) {
355 			pr_err("PrP bus error occurred, this transfer is probably corrupted\n");
356 			writel(PRP_CNTL_SWRST, pcdev->base_emma + PRP_CNTL);
357 		} else if (irqst & PRP_INTR_ST_CH2B1CI) { /* buffer ready */
358 			src_vb = v4l2_m2m_src_buf_remove(curr_ctx->fh.m2m_ctx);
359 			dst_vb = v4l2_m2m_dst_buf_remove(curr_ctx->fh.m2m_ctx);
360 
361 			dst_vb->vb2_buf.timestamp = src_vb->vb2_buf.timestamp;
362 			dst_vb->flags &=
363 				~V4L2_BUF_FLAG_TSTAMP_SRC_MASK;
364 			dst_vb->flags |=
365 				src_vb->flags
366 				& V4L2_BUF_FLAG_TSTAMP_SRC_MASK;
367 			dst_vb->timecode = src_vb->timecode;
368 
369 			spin_lock_irqsave(&pcdev->irqlock, flags);
370 			v4l2_m2m_buf_done(src_vb, VB2_BUF_STATE_DONE);
371 			v4l2_m2m_buf_done(dst_vb, VB2_BUF_STATE_DONE);
372 			spin_unlock_irqrestore(&pcdev->irqlock, flags);
373 		}
374 	}
375 
376 	v4l2_m2m_job_finish(pcdev->m2m_dev, curr_ctx->fh.m2m_ctx);
377 	return IRQ_HANDLED;
378 }
379 
380 /*
381  * video ioctls
382  */
383 static int vidioc_querycap(struct file *file, void *priv,
384 			   struct v4l2_capability *cap)
385 {
386 	strscpy(cap->driver, MEM2MEM_NAME, sizeof(cap->driver));
387 	strscpy(cap->card, MEM2MEM_NAME, sizeof(cap->card));
388 	return 0;
389 }
390 
391 static int enum_fmt(struct v4l2_fmtdesc *f, u32 type)
392 {
393 	int i, num;
394 	struct emmaprp_fmt *fmt;
395 
396 	num = 0;
397 
398 	for (i = 0; i < NUM_FORMATS; ++i) {
399 		if (formats[i].types & type) {
400 			/* index-th format of type type found ? */
401 			if (num == f->index)
402 				break;
403 			/* Correct type but haven't reached our index yet,
404 			 * just increment per-type index */
405 			++num;
406 		}
407 	}
408 
409 	if (i < NUM_FORMATS) {
410 		/* Format found */
411 		fmt = &formats[i];
412 		f->pixelformat = fmt->fourcc;
413 		return 0;
414 	}
415 
416 	/* Format not found */
417 	return -EINVAL;
418 }
419 
420 static int vidioc_enum_fmt_vid_cap(struct file *file, void *priv,
421 				   struct v4l2_fmtdesc *f)
422 {
423 	return enum_fmt(f, MEM2MEM_CAPTURE);
424 }
425 
426 static int vidioc_enum_fmt_vid_out(struct file *file, void *priv,
427 				   struct v4l2_fmtdesc *f)
428 {
429 	return enum_fmt(f, MEM2MEM_OUTPUT);
430 }
431 
432 static int vidioc_g_fmt(struct emmaprp_ctx *ctx, struct v4l2_format *f)
433 {
434 	struct vb2_queue *vq;
435 	struct emmaprp_q_data *q_data;
436 
437 	vq = v4l2_m2m_get_vq(ctx->fh.m2m_ctx, f->type);
438 	if (!vq)
439 		return -EINVAL;
440 
441 	q_data = get_q_data(ctx, f->type);
442 
443 	f->fmt.pix.width	= q_data->width;
444 	f->fmt.pix.height	= q_data->height;
445 	f->fmt.pix.field	= V4L2_FIELD_NONE;
446 	f->fmt.pix.pixelformat	= q_data->fmt->fourcc;
447 	if (f->fmt.pix.pixelformat == V4L2_PIX_FMT_YUV420)
448 		f->fmt.pix.bytesperline = q_data->width * 3 / 2;
449 	else /* YUYV */
450 		f->fmt.pix.bytesperline = q_data->width * 2;
451 	f->fmt.pix.sizeimage	= q_data->sizeimage;
452 
453 	return 0;
454 }
455 
456 static int vidioc_g_fmt_vid_out(struct file *file, void *priv,
457 				struct v4l2_format *f)
458 {
459 	return vidioc_g_fmt(file_to_emmaprp_ctx(file), f);
460 }
461 
462 static int vidioc_g_fmt_vid_cap(struct file *file, void *priv,
463 				struct v4l2_format *f)
464 {
465 	return vidioc_g_fmt(file_to_emmaprp_ctx(file), f);
466 }
467 
468 static int vidioc_try_fmt(struct v4l2_format *f)
469 {
470 	enum v4l2_field field;
471 
472 
473 	if (!find_format(f))
474 		return -EINVAL;
475 
476 	field = f->fmt.pix.field;
477 	if (field == V4L2_FIELD_ANY)
478 		field = V4L2_FIELD_NONE;
479 	else if (V4L2_FIELD_NONE != field)
480 		return -EINVAL;
481 
482 	/* V4L2 specification suggests the driver corrects the format struct
483 	 * if any of the dimensions is unsupported */
484 	f->fmt.pix.field = field;
485 
486 	if (f->fmt.pix.pixelformat == V4L2_PIX_FMT_YUV420) {
487 		v4l_bound_align_image(&f->fmt.pix.width, MIN_W, MAX_W,
488 				      W_ALIGN_YUV420, &f->fmt.pix.height,
489 				      MIN_H, MAX_H, H_ALIGN, S_ALIGN);
490 		f->fmt.pix.bytesperline = f->fmt.pix.width * 3 / 2;
491 	} else {
492 		v4l_bound_align_image(&f->fmt.pix.width, MIN_W, MAX_W,
493 				      W_ALIGN_OTHERS, &f->fmt.pix.height,
494 				      MIN_H, MAX_H, H_ALIGN, S_ALIGN);
495 		f->fmt.pix.bytesperline = f->fmt.pix.width * 2;
496 	}
497 	f->fmt.pix.sizeimage = f->fmt.pix.height * f->fmt.pix.bytesperline;
498 
499 	return 0;
500 }
501 
502 static int vidioc_try_fmt_vid_cap(struct file *file, void *priv,
503 				  struct v4l2_format *f)
504 {
505 	struct emmaprp_ctx *ctx = file_to_emmaprp_ctx(file);
506 	struct emmaprp_fmt *fmt;
507 
508 	fmt = find_format(f);
509 	if (!fmt || !(fmt->types & MEM2MEM_CAPTURE)) {
510 		v4l2_err(&ctx->dev->v4l2_dev,
511 			 "Fourcc format (0x%08x) invalid.\n",
512 			 f->fmt.pix.pixelformat);
513 		return -EINVAL;
514 	}
515 
516 	return vidioc_try_fmt(f);
517 }
518 
519 static int vidioc_try_fmt_vid_out(struct file *file, void *priv,
520 				  struct v4l2_format *f)
521 {
522 	struct emmaprp_ctx *ctx = file_to_emmaprp_ctx(file);
523 	struct emmaprp_fmt *fmt;
524 
525 	fmt = find_format(f);
526 	if (!fmt || !(fmt->types & MEM2MEM_OUTPUT)) {
527 		v4l2_err(&ctx->dev->v4l2_dev,
528 			 "Fourcc format (0x%08x) invalid.\n",
529 			 f->fmt.pix.pixelformat);
530 		return -EINVAL;
531 	}
532 
533 	return vidioc_try_fmt(f);
534 }
535 
536 static int vidioc_s_fmt(struct emmaprp_ctx *ctx, struct v4l2_format *f)
537 {
538 	struct emmaprp_q_data *q_data;
539 	struct vb2_queue *vq;
540 	int ret;
541 
542 	vq = v4l2_m2m_get_vq(ctx->fh.m2m_ctx, f->type);
543 	if (!vq)
544 		return -EINVAL;
545 
546 	q_data = get_q_data(ctx, f->type);
547 	if (!q_data)
548 		return -EINVAL;
549 
550 	if (vb2_is_busy(vq)) {
551 		v4l2_err(&ctx->dev->v4l2_dev, "%s queue busy\n", __func__);
552 		return -EBUSY;
553 	}
554 
555 	ret = vidioc_try_fmt(f);
556 	if (ret)
557 		return ret;
558 
559 	q_data->fmt		= find_format(f);
560 	q_data->width		= f->fmt.pix.width;
561 	q_data->height		= f->fmt.pix.height;
562 	if (q_data->fmt->fourcc == V4L2_PIX_FMT_YUV420)
563 		q_data->sizeimage = q_data->width * q_data->height * 3 / 2;
564 	else /* YUYV */
565 		q_data->sizeimage = q_data->width * q_data->height * 2;
566 
567 	dprintk(ctx->dev,
568 		"Setting format for type %d, wxh: %dx%d, fmt: %d\n",
569 		f->type, q_data->width, q_data->height, q_data->fmt->fourcc);
570 
571 	return 0;
572 }
573 
574 static int vidioc_s_fmt_vid_cap(struct file *file, void *priv,
575 				struct v4l2_format *f)
576 {
577 	int ret;
578 
579 	ret = vidioc_try_fmt_vid_cap(file, priv, f);
580 	if (ret)
581 		return ret;
582 
583 	return vidioc_s_fmt(file_to_emmaprp_ctx(file), f);
584 }
585 
586 static int vidioc_s_fmt_vid_out(struct file *file, void *priv,
587 				struct v4l2_format *f)
588 {
589 	int ret;
590 
591 	ret = vidioc_try_fmt_vid_out(file, priv, f);
592 	if (ret)
593 		return ret;
594 
595 	return vidioc_s_fmt(file_to_emmaprp_ctx(file), f);
596 }
597 
598 static const struct v4l2_ioctl_ops emmaprp_ioctl_ops = {
599 	.vidioc_querycap	= vidioc_querycap,
600 
601 	.vidioc_enum_fmt_vid_cap = vidioc_enum_fmt_vid_cap,
602 	.vidioc_g_fmt_vid_cap	= vidioc_g_fmt_vid_cap,
603 	.vidioc_try_fmt_vid_cap	= vidioc_try_fmt_vid_cap,
604 	.vidioc_s_fmt_vid_cap	= vidioc_s_fmt_vid_cap,
605 
606 	.vidioc_enum_fmt_vid_out = vidioc_enum_fmt_vid_out,
607 	.vidioc_g_fmt_vid_out	= vidioc_g_fmt_vid_out,
608 	.vidioc_try_fmt_vid_out	= vidioc_try_fmt_vid_out,
609 	.vidioc_s_fmt_vid_out	= vidioc_s_fmt_vid_out,
610 
611 	.vidioc_reqbufs		= v4l2_m2m_ioctl_reqbufs,
612 	.vidioc_querybuf	= v4l2_m2m_ioctl_querybuf,
613 	.vidioc_qbuf		= v4l2_m2m_ioctl_qbuf,
614 	.vidioc_dqbuf		= v4l2_m2m_ioctl_dqbuf,
615 	.vidioc_prepare_buf	= v4l2_m2m_ioctl_prepare_buf,
616 	.vidioc_expbuf		= v4l2_m2m_ioctl_expbuf,
617 	.vidioc_streamon	= v4l2_m2m_ioctl_streamon,
618 	.vidioc_streamoff	= v4l2_m2m_ioctl_streamoff,
619 };
620 
621 
622 /*
623  * Queue operations
624  */
625 static int emmaprp_queue_setup(struct vb2_queue *vq,
626 				unsigned int *nbuffers, unsigned int *nplanes,
627 				unsigned int sizes[], struct device *alloc_devs[])
628 {
629 	struct emmaprp_ctx *ctx = vb2_get_drv_priv(vq);
630 	struct emmaprp_q_data *q_data;
631 	unsigned int size, count = *nbuffers;
632 
633 	q_data = get_q_data(ctx, vq->type);
634 
635 	if (q_data->fmt->fourcc == V4L2_PIX_FMT_YUV420)
636 		size = q_data->width * q_data->height * 3 / 2;
637 	else
638 		size = q_data->width * q_data->height * 2;
639 
640 	while (size * count > MEM2MEM_VID_MEM_LIMIT)
641 		(count)--;
642 
643 	*nplanes = 1;
644 	*nbuffers = count;
645 	sizes[0] = size;
646 
647 	dprintk(ctx->dev, "get %d buffer(s) of size %d each.\n", count, size);
648 
649 	return 0;
650 }
651 
652 static int emmaprp_buf_prepare(struct vb2_buffer *vb)
653 {
654 	struct emmaprp_ctx *ctx = vb2_get_drv_priv(vb->vb2_queue);
655 	struct emmaprp_q_data *q_data;
656 
657 	dprintk(ctx->dev, "type: %d\n", vb->vb2_queue->type);
658 
659 	q_data = get_q_data(ctx, vb->vb2_queue->type);
660 
661 	if (vb2_plane_size(vb, 0) < q_data->sizeimage) {
662 		dprintk(ctx->dev,
663 			"%s data will not fit into plane(%lu < %lu)\n",
664 			__func__, vb2_plane_size(vb, 0),
665 			(long)q_data->sizeimage);
666 		return -EINVAL;
667 	}
668 
669 	vb2_set_plane_payload(vb, 0, q_data->sizeimage);
670 
671 	return 0;
672 }
673 
674 static void emmaprp_buf_queue(struct vb2_buffer *vb)
675 {
676 	struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb);
677 	struct emmaprp_ctx *ctx = vb2_get_drv_priv(vb->vb2_queue);
678 	v4l2_m2m_buf_queue(ctx->fh.m2m_ctx, vbuf);
679 }
680 
681 static const struct vb2_ops emmaprp_qops = {
682 	.queue_setup	 = emmaprp_queue_setup,
683 	.buf_prepare	 = emmaprp_buf_prepare,
684 	.buf_queue	 = emmaprp_buf_queue,
685 };
686 
687 static int queue_init(void *priv, struct vb2_queue *src_vq,
688 		      struct vb2_queue *dst_vq)
689 {
690 	struct emmaprp_ctx *ctx = priv;
691 	int ret;
692 
693 	src_vq->type = V4L2_BUF_TYPE_VIDEO_OUTPUT;
694 	src_vq->io_modes = VB2_MMAP | VB2_USERPTR | VB2_DMABUF;
695 	src_vq->drv_priv = ctx;
696 	src_vq->buf_struct_size = sizeof(struct v4l2_m2m_buffer);
697 	src_vq->ops = &emmaprp_qops;
698 	src_vq->mem_ops = &vb2_dma_contig_memops;
699 	src_vq->timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_COPY;
700 	src_vq->dev = ctx->dev->v4l2_dev.dev;
701 	src_vq->lock = &ctx->dev->dev_mutex;
702 
703 	ret = vb2_queue_init(src_vq);
704 	if (ret)
705 		return ret;
706 
707 	dst_vq->type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
708 	dst_vq->io_modes = VB2_MMAP | VB2_USERPTR | VB2_DMABUF;
709 	dst_vq->drv_priv = ctx;
710 	dst_vq->buf_struct_size = sizeof(struct v4l2_m2m_buffer);
711 	dst_vq->ops = &emmaprp_qops;
712 	dst_vq->mem_ops = &vb2_dma_contig_memops;
713 	dst_vq->timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_COPY;
714 	dst_vq->dev = ctx->dev->v4l2_dev.dev;
715 	dst_vq->lock = &ctx->dev->dev_mutex;
716 
717 	return vb2_queue_init(dst_vq);
718 }
719 
720 /*
721  * File operations
722  */
723 static int emmaprp_open(struct file *file)
724 {
725 	struct emmaprp_dev *pcdev = video_drvdata(file);
726 	struct emmaprp_ctx *ctx;
727 
728 	ctx = kzalloc(sizeof *ctx, GFP_KERNEL);
729 	if (!ctx)
730 		return -ENOMEM;
731 
732 	v4l2_fh_init(&ctx->fh, video_devdata(file));
733 	ctx->dev = pcdev;
734 
735 	if (mutex_lock_interruptible(&pcdev->dev_mutex)) {
736 		kfree(ctx);
737 		return -ERESTARTSYS;
738 	}
739 
740 	ctx->fh.m2m_ctx = v4l2_m2m_ctx_init(pcdev->m2m_dev, ctx, &queue_init);
741 
742 	if (IS_ERR(ctx->fh.m2m_ctx)) {
743 		int ret = PTR_ERR(ctx->fh.m2m_ctx);
744 
745 		mutex_unlock(&pcdev->dev_mutex);
746 		kfree(ctx);
747 		return ret;
748 	}
749 
750 	clk_prepare_enable(pcdev->clk_emma_ipg);
751 	clk_prepare_enable(pcdev->clk_emma_ahb);
752 	ctx->q_data[V4L2_M2M_SRC].fmt = &formats[1];
753 	ctx->q_data[V4L2_M2M_DST].fmt = &formats[0];
754 	v4l2_fh_add(&ctx->fh, file);
755 	mutex_unlock(&pcdev->dev_mutex);
756 
757 	dprintk(pcdev, "Created instance %p, m2m_ctx: %p\n", ctx, ctx->fh.m2m_ctx);
758 
759 	return 0;
760 }
761 
762 static int emmaprp_release(struct file *file)
763 {
764 	struct emmaprp_dev *pcdev = video_drvdata(file);
765 	struct emmaprp_ctx *ctx = file_to_emmaprp_ctx(file);
766 
767 	dprintk(pcdev, "Releasing instance %p\n", ctx);
768 
769 	mutex_lock(&pcdev->dev_mutex);
770 	clk_disable_unprepare(pcdev->clk_emma_ahb);
771 	clk_disable_unprepare(pcdev->clk_emma_ipg);
772 	v4l2_fh_del(&ctx->fh, file);
773 	v4l2_fh_exit(&ctx->fh);
774 	v4l2_m2m_ctx_release(ctx->fh.m2m_ctx);
775 	mutex_unlock(&pcdev->dev_mutex);
776 	kfree(ctx);
777 
778 	return 0;
779 }
780 
781 static const struct v4l2_file_operations emmaprp_fops = {
782 	.owner		= THIS_MODULE,
783 	.open		= emmaprp_open,
784 	.release	= emmaprp_release,
785 	.poll		= v4l2_m2m_fop_poll,
786 	.unlocked_ioctl	= video_ioctl2,
787 	.mmap		= v4l2_m2m_fop_mmap,
788 };
789 
790 static const struct video_device emmaprp_videodev = {
791 	.name		= MEM2MEM_NAME,
792 	.fops		= &emmaprp_fops,
793 	.ioctl_ops	= &emmaprp_ioctl_ops,
794 	.minor		= -1,
795 	.release	= video_device_release,
796 	.vfl_dir	= VFL_DIR_M2M,
797 	.device_caps	= V4L2_CAP_VIDEO_M2M | V4L2_CAP_STREAMING,
798 };
799 
800 static const struct v4l2_m2m_ops m2m_ops = {
801 	.device_run	= emmaprp_device_run,
802 	.job_abort	= emmaprp_job_abort,
803 };
804 
805 static int emmaprp_probe(struct platform_device *pdev)
806 {
807 	struct emmaprp_dev *pcdev;
808 	struct video_device *vfd;
809 	int irq, ret;
810 
811 	pcdev = devm_kzalloc(&pdev->dev, sizeof(*pcdev), GFP_KERNEL);
812 	if (!pcdev)
813 		return -ENOMEM;
814 
815 	spin_lock_init(&pcdev->irqlock);
816 
817 	pcdev->clk_emma_ipg = devm_clk_get(&pdev->dev, "ipg");
818 	if (IS_ERR(pcdev->clk_emma_ipg)) {
819 		return PTR_ERR(pcdev->clk_emma_ipg);
820 	}
821 
822 	pcdev->clk_emma_ahb = devm_clk_get(&pdev->dev, "ahb");
823 	if (IS_ERR(pcdev->clk_emma_ahb))
824 		return PTR_ERR(pcdev->clk_emma_ahb);
825 
826 	pcdev->base_emma = devm_platform_ioremap_resource(pdev, 0);
827 	if (IS_ERR(pcdev->base_emma))
828 		return PTR_ERR(pcdev->base_emma);
829 
830 	ret = v4l2_device_register(&pdev->dev, &pcdev->v4l2_dev);
831 	if (ret)
832 		return ret;
833 
834 	mutex_init(&pcdev->dev_mutex);
835 
836 	vfd = video_device_alloc();
837 	if (!vfd) {
838 		v4l2_err(&pcdev->v4l2_dev, "Failed to allocate video device\n");
839 		ret = -ENOMEM;
840 		goto unreg_dev;
841 	}
842 
843 	*vfd = emmaprp_videodev;
844 	vfd->lock = &pcdev->dev_mutex;
845 	vfd->v4l2_dev = &pcdev->v4l2_dev;
846 
847 	video_set_drvdata(vfd, pcdev);
848 	pcdev->vfd = vfd;
849 	v4l2_info(&pcdev->v4l2_dev, EMMAPRP_MODULE_NAME
850 		  " Device registered as /dev/video%d\n", vfd->num);
851 
852 	platform_set_drvdata(pdev, pcdev);
853 
854 	irq = platform_get_irq(pdev, 0);
855 	if (irq < 0) {
856 		ret = irq;
857 		goto rel_vdev;
858 	}
859 
860 	ret = devm_request_irq(&pdev->dev, irq, emmaprp_irq, 0,
861 			       dev_name(&pdev->dev), pcdev);
862 	if (ret)
863 		goto rel_vdev;
864 
865 	pcdev->m2m_dev = v4l2_m2m_init(&m2m_ops);
866 	if (IS_ERR(pcdev->m2m_dev)) {
867 		v4l2_err(&pcdev->v4l2_dev, "Failed to init mem2mem device\n");
868 		ret = PTR_ERR(pcdev->m2m_dev);
869 		goto rel_vdev;
870 	}
871 
872 	ret = video_register_device(vfd, VFL_TYPE_VIDEO, 0);
873 	if (ret) {
874 		v4l2_err(&pcdev->v4l2_dev, "Failed to register video device\n");
875 		goto rel_m2m;
876 	}
877 
878 	return 0;
879 
880 
881 rel_m2m:
882 	v4l2_m2m_release(pcdev->m2m_dev);
883 rel_vdev:
884 	video_device_release(vfd);
885 unreg_dev:
886 	v4l2_device_unregister(&pcdev->v4l2_dev);
887 
888 	mutex_destroy(&pcdev->dev_mutex);
889 
890 	return ret;
891 }
892 
893 static void emmaprp_remove(struct platform_device *pdev)
894 {
895 	struct emmaprp_dev *pcdev = platform_get_drvdata(pdev);
896 
897 	v4l2_info(&pcdev->v4l2_dev, "Removing " EMMAPRP_MODULE_NAME);
898 
899 	video_unregister_device(pcdev->vfd);
900 	v4l2_m2m_release(pcdev->m2m_dev);
901 	v4l2_device_unregister(&pcdev->v4l2_dev);
902 	mutex_destroy(&pcdev->dev_mutex);
903 }
904 
905 static struct platform_driver emmaprp_pdrv = {
906 	.probe		= emmaprp_probe,
907 	.remove		= emmaprp_remove,
908 	.driver		= {
909 		.name	= MEM2MEM_NAME,
910 	},
911 };
912 module_platform_driver(emmaprp_pdrv);
913