1 /*
2  * V4L2 deinterlacing support.
3  *
4  * Copyright (c) 2012 Vista Silicon S.L.
5  * Javier Martin <javier.martin@vista-silicon.com>
6  *
7  * This program is free software; you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License as published by the
9  * Free Software Foundation; either version 2 of the
10  * License, or (at your option) any later version
11  */
12 
13 #include <linux/module.h>
14 #include <linux/slab.h>
15 #include <linux/interrupt.h>
16 #include <linux/dmaengine.h>
17 #include <linux/platform_device.h>
18 
19 #include <media/v4l2-mem2mem.h>
20 #include <media/v4l2-device.h>
21 #include <media/v4l2-ioctl.h>
22 #include <media/videobuf2-dma-contig.h>
23 
24 #define MEM2MEM_TEST_MODULE_NAME "mem2mem-deinterlace"
25 
26 MODULE_DESCRIPTION("mem2mem device which supports deinterlacing using dmaengine");
27 MODULE_AUTHOR("Javier Martin <javier.martin@vista-silicon.com");
28 MODULE_LICENSE("GPL");
29 MODULE_VERSION("0.0.1");
30 
31 static bool debug;
32 module_param(debug, bool, 0644);
33 
34 /* Flags that indicate a format can be used for capture/output */
35 #define MEM2MEM_CAPTURE	(1 << 0)
36 #define MEM2MEM_OUTPUT	(1 << 1)
37 
38 #define MEM2MEM_NAME		"m2m-deinterlace"
39 
40 #define dprintk(dev, fmt, arg...) \
41 	v4l2_dbg(1, debug, &dev->v4l2_dev, "%s: " fmt, __func__, ## arg)
42 
43 struct deinterlace_fmt {
44 	char	*name;
45 	u32	fourcc;
46 	/* Types the format can be used for */
47 	u32	types;
48 };
49 
50 static struct deinterlace_fmt formats[] = {
51 	{
52 		.name	= "YUV 4:2:0 Planar",
53 		.fourcc	= V4L2_PIX_FMT_YUV420,
54 		.types	= MEM2MEM_CAPTURE | MEM2MEM_OUTPUT,
55 	},
56 	{
57 		.name	= "YUYV 4:2:2",
58 		.fourcc	= V4L2_PIX_FMT_YUYV,
59 		.types	= MEM2MEM_CAPTURE | MEM2MEM_OUTPUT,
60 	},
61 };
62 
63 #define NUM_FORMATS ARRAY_SIZE(formats)
64 
65 /* Per-queue, driver-specific private data */
66 struct deinterlace_q_data {
67 	unsigned int		width;
68 	unsigned int		height;
69 	unsigned int		sizeimage;
70 	struct deinterlace_fmt	*fmt;
71 	enum v4l2_field		field;
72 };
73 
74 enum {
75 	V4L2_M2M_SRC = 0,
76 	V4L2_M2M_DST = 1,
77 };
78 
79 enum {
80 	YUV420_DMA_Y_ODD,
81 	YUV420_DMA_Y_EVEN,
82 	YUV420_DMA_U_ODD,
83 	YUV420_DMA_U_EVEN,
84 	YUV420_DMA_V_ODD,
85 	YUV420_DMA_V_EVEN,
86 	YUV420_DMA_Y_ODD_DOUBLING,
87 	YUV420_DMA_U_ODD_DOUBLING,
88 	YUV420_DMA_V_ODD_DOUBLING,
89 	YUYV_DMA_ODD,
90 	YUYV_DMA_EVEN,
91 	YUYV_DMA_EVEN_DOUBLING,
92 };
93 
94 /* Source and destination queue data */
95 static struct deinterlace_q_data q_data[2];
96 
97 static struct deinterlace_q_data *get_q_data(enum v4l2_buf_type type)
98 {
99 	switch (type) {
100 	case V4L2_BUF_TYPE_VIDEO_OUTPUT:
101 		return &q_data[V4L2_M2M_SRC];
102 	case V4L2_BUF_TYPE_VIDEO_CAPTURE:
103 		return &q_data[V4L2_M2M_DST];
104 	default:
105 		BUG();
106 	}
107 	return NULL;
108 }
109 
110 static struct deinterlace_fmt *find_format(struct v4l2_format *f)
111 {
112 	struct deinterlace_fmt *fmt;
113 	unsigned int k;
114 
115 	for (k = 0; k < NUM_FORMATS; k++) {
116 		fmt = &formats[k];
117 		if ((fmt->types & f->type) &&
118 			(fmt->fourcc == f->fmt.pix.pixelformat))
119 			break;
120 	}
121 
122 	if (k == NUM_FORMATS)
123 		return NULL;
124 
125 	return &formats[k];
126 }
127 
128 struct deinterlace_dev {
129 	struct v4l2_device	v4l2_dev;
130 	struct video_device	vfd;
131 
132 	atomic_t		busy;
133 	struct mutex		dev_mutex;
134 	spinlock_t		irqlock;
135 
136 	struct dma_chan		*dma_chan;
137 
138 	struct v4l2_m2m_dev	*m2m_dev;
139 	struct vb2_alloc_ctx	*alloc_ctx;
140 };
141 
142 struct deinterlace_ctx {
143 	struct deinterlace_dev	*dev;
144 
145 	/* Abort requested by m2m */
146 	int			aborting;
147 	enum v4l2_colorspace	colorspace;
148 	dma_cookie_t		cookie;
149 	struct v4l2_m2m_ctx	*m2m_ctx;
150 	struct dma_interleaved_template *xt;
151 };
152 
153 /*
154  * mem2mem callbacks
155  */
156 static int deinterlace_job_ready(void *priv)
157 {
158 	struct deinterlace_ctx *ctx = priv;
159 	struct deinterlace_dev *pcdev = ctx->dev;
160 
161 	if ((v4l2_m2m_num_src_bufs_ready(ctx->m2m_ctx) > 0)
162 	    && (v4l2_m2m_num_dst_bufs_ready(ctx->m2m_ctx) > 0)
163 	    && (atomic_read(&ctx->dev->busy) == 0)) {
164 		dprintk(pcdev, "Task ready\n");
165 		return 1;
166 	}
167 
168 	dprintk(pcdev, "Task not ready to run\n");
169 
170 	return 0;
171 }
172 
173 static void deinterlace_job_abort(void *priv)
174 {
175 	struct deinterlace_ctx *ctx = priv;
176 	struct deinterlace_dev *pcdev = ctx->dev;
177 
178 	ctx->aborting = 1;
179 
180 	dprintk(pcdev, "Aborting task\n");
181 
182 	v4l2_m2m_job_finish(pcdev->m2m_dev, ctx->m2m_ctx);
183 }
184 
185 static void deinterlace_lock(void *priv)
186 {
187 	struct deinterlace_ctx *ctx = priv;
188 	struct deinterlace_dev *pcdev = ctx->dev;
189 	mutex_lock(&pcdev->dev_mutex);
190 }
191 
192 static void deinterlace_unlock(void *priv)
193 {
194 	struct deinterlace_ctx *ctx = priv;
195 	struct deinterlace_dev *pcdev = ctx->dev;
196 	mutex_unlock(&pcdev->dev_mutex);
197 }
198 
199 static void dma_callback(void *data)
200 {
201 	struct deinterlace_ctx *curr_ctx = data;
202 	struct deinterlace_dev *pcdev = curr_ctx->dev;
203 	struct vb2_buffer *src_vb, *dst_vb;
204 
205 	atomic_set(&pcdev->busy, 0);
206 
207 	src_vb = v4l2_m2m_src_buf_remove(curr_ctx->m2m_ctx);
208 	dst_vb = v4l2_m2m_dst_buf_remove(curr_ctx->m2m_ctx);
209 
210 	dst_vb->v4l2_buf.timestamp = src_vb->v4l2_buf.timestamp;
211 	dst_vb->v4l2_buf.flags &= ~V4L2_BUF_FLAG_TSTAMP_SRC_MASK;
212 	dst_vb->v4l2_buf.flags |=
213 		src_vb->v4l2_buf.flags & V4L2_BUF_FLAG_TSTAMP_SRC_MASK;
214 	dst_vb->v4l2_buf.timecode = src_vb->v4l2_buf.timecode;
215 
216 	v4l2_m2m_buf_done(src_vb, VB2_BUF_STATE_DONE);
217 	v4l2_m2m_buf_done(dst_vb, VB2_BUF_STATE_DONE);
218 
219 	v4l2_m2m_job_finish(pcdev->m2m_dev, curr_ctx->m2m_ctx);
220 
221 	dprintk(pcdev, "dma transfers completed.\n");
222 }
223 
224 static void deinterlace_issue_dma(struct deinterlace_ctx *ctx, int op,
225 				  int do_callback)
226 {
227 	struct deinterlace_q_data *s_q_data;
228 	struct vb2_buffer *src_buf, *dst_buf;
229 	struct deinterlace_dev *pcdev = ctx->dev;
230 	struct dma_chan *chan = pcdev->dma_chan;
231 	struct dma_device *dmadev = chan->device;
232 	struct dma_async_tx_descriptor *tx;
233 	unsigned int s_width, s_height;
234 	unsigned int s_size;
235 	dma_addr_t p_in, p_out;
236 	enum dma_ctrl_flags flags;
237 
238 	src_buf = v4l2_m2m_next_src_buf(ctx->m2m_ctx);
239 	dst_buf = v4l2_m2m_next_dst_buf(ctx->m2m_ctx);
240 
241 	s_q_data = get_q_data(V4L2_BUF_TYPE_VIDEO_OUTPUT);
242 	s_width	= s_q_data->width;
243 	s_height = s_q_data->height;
244 	s_size = s_width * s_height;
245 
246 	p_in = (dma_addr_t)vb2_dma_contig_plane_dma_addr(src_buf, 0);
247 	p_out = (dma_addr_t)vb2_dma_contig_plane_dma_addr(dst_buf, 0);
248 	if (!p_in || !p_out) {
249 		v4l2_err(&pcdev->v4l2_dev,
250 			 "Acquiring kernel pointers to buffers failed\n");
251 		return;
252 	}
253 
254 	switch (op) {
255 	case YUV420_DMA_Y_ODD:
256 		ctx->xt->numf = s_height / 2;
257 		ctx->xt->sgl[0].size = s_width;
258 		ctx->xt->sgl[0].icg = s_width;
259 		ctx->xt->src_start = p_in;
260 		ctx->xt->dst_start = p_out;
261 		break;
262 	case YUV420_DMA_Y_EVEN:
263 		ctx->xt->numf = s_height / 2;
264 		ctx->xt->sgl[0].size = s_width;
265 		ctx->xt->sgl[0].icg = s_width;
266 		ctx->xt->src_start = p_in + s_size / 2;
267 		ctx->xt->dst_start = p_out + s_width;
268 		break;
269 	case YUV420_DMA_U_ODD:
270 		ctx->xt->numf = s_height / 4;
271 		ctx->xt->sgl[0].size = s_width / 2;
272 		ctx->xt->sgl[0].icg = s_width / 2;
273 		ctx->xt->src_start = p_in + s_size;
274 		ctx->xt->dst_start = p_out + s_size;
275 		break;
276 	case YUV420_DMA_U_EVEN:
277 		ctx->xt->numf = s_height / 4;
278 		ctx->xt->sgl[0].size = s_width / 2;
279 		ctx->xt->sgl[0].icg = s_width / 2;
280 		ctx->xt->src_start = p_in + (9 * s_size) / 8;
281 		ctx->xt->dst_start = p_out + s_size + s_width / 2;
282 		break;
283 	case YUV420_DMA_V_ODD:
284 		ctx->xt->numf = s_height / 4;
285 		ctx->xt->sgl[0].size = s_width / 2;
286 		ctx->xt->sgl[0].icg = s_width / 2;
287 		ctx->xt->src_start = p_in + (5 * s_size) / 4;
288 		ctx->xt->dst_start = p_out + (5 * s_size) / 4;
289 		break;
290 	case YUV420_DMA_V_EVEN:
291 		ctx->xt->numf = s_height / 4;
292 		ctx->xt->sgl[0].size = s_width / 2;
293 		ctx->xt->sgl[0].icg = s_width / 2;
294 		ctx->xt->src_start = p_in + (11 * s_size) / 8;
295 		ctx->xt->dst_start = p_out + (5 * s_size) / 4 + s_width / 2;
296 		break;
297 	case YUV420_DMA_Y_ODD_DOUBLING:
298 		ctx->xt->numf = s_height / 2;
299 		ctx->xt->sgl[0].size = s_width;
300 		ctx->xt->sgl[0].icg = s_width;
301 		ctx->xt->src_start = p_in;
302 		ctx->xt->dst_start = p_out + s_width;
303 		break;
304 	case YUV420_DMA_U_ODD_DOUBLING:
305 		ctx->xt->numf = s_height / 4;
306 		ctx->xt->sgl[0].size = s_width / 2;
307 		ctx->xt->sgl[0].icg = s_width / 2;
308 		ctx->xt->src_start = p_in + s_size;
309 		ctx->xt->dst_start = p_out + s_size + s_width / 2;
310 		break;
311 	case YUV420_DMA_V_ODD_DOUBLING:
312 		ctx->xt->numf = s_height / 4;
313 		ctx->xt->sgl[0].size = s_width / 2;
314 		ctx->xt->sgl[0].icg = s_width / 2;
315 		ctx->xt->src_start = p_in + (5 * s_size) / 4;
316 		ctx->xt->dst_start = p_out + (5 * s_size) / 4 + s_width / 2;
317 		break;
318 	case YUYV_DMA_ODD:
319 		ctx->xt->numf = s_height / 2;
320 		ctx->xt->sgl[0].size = s_width * 2;
321 		ctx->xt->sgl[0].icg = s_width * 2;
322 		ctx->xt->src_start = p_in;
323 		ctx->xt->dst_start = p_out;
324 		break;
325 	case YUYV_DMA_EVEN:
326 		ctx->xt->numf = s_height / 2;
327 		ctx->xt->sgl[0].size = s_width * 2;
328 		ctx->xt->sgl[0].icg = s_width * 2;
329 		ctx->xt->src_start = p_in + s_size;
330 		ctx->xt->dst_start = p_out + s_width * 2;
331 		break;
332 	case YUYV_DMA_EVEN_DOUBLING:
333 	default:
334 		ctx->xt->numf = s_height / 2;
335 		ctx->xt->sgl[0].size = s_width * 2;
336 		ctx->xt->sgl[0].icg = s_width * 2;
337 		ctx->xt->src_start = p_in;
338 		ctx->xt->dst_start = p_out + s_width * 2;
339 		break;
340 	}
341 
342 	/* Common parameters for al transfers */
343 	ctx->xt->frame_size = 1;
344 	ctx->xt->dir = DMA_MEM_TO_MEM;
345 	ctx->xt->src_sgl = false;
346 	ctx->xt->dst_sgl = true;
347 	flags = DMA_CTRL_ACK | DMA_PREP_INTERRUPT;
348 
349 	tx = dmadev->device_prep_interleaved_dma(chan, ctx->xt, flags);
350 	if (tx == NULL) {
351 		v4l2_warn(&pcdev->v4l2_dev, "DMA interleaved prep error\n");
352 		return;
353 	}
354 
355 	if (do_callback) {
356 		tx->callback = dma_callback;
357 		tx->callback_param = ctx;
358 	}
359 
360 	ctx->cookie = dmaengine_submit(tx);
361 	if (dma_submit_error(ctx->cookie)) {
362 		v4l2_warn(&pcdev->v4l2_dev,
363 			  "DMA submit error %d with src=0x%x dst=0x%x len=0x%x\n",
364 			  ctx->cookie, (unsigned)p_in, (unsigned)p_out,
365 			  s_size * 3/2);
366 		return;
367 	}
368 
369 	dma_async_issue_pending(chan);
370 }
371 
372 static void deinterlace_device_run(void *priv)
373 {
374 	struct deinterlace_ctx *ctx = priv;
375 	struct deinterlace_q_data *dst_q_data;
376 
377 	atomic_set(&ctx->dev->busy, 1);
378 
379 	dprintk(ctx->dev, "%s: DMA try issue.\n", __func__);
380 
381 	dst_q_data = get_q_data(V4L2_BUF_TYPE_VIDEO_CAPTURE);
382 
383 	/*
384 	 * 4 possible field conversions are possible at the moment:
385 	 *  V4L2_FIELD_SEQ_TB --> V4L2_FIELD_INTERLACED_TB:
386 	 *	two separate fields in the same input buffer are interlaced
387 	 * 	in the output buffer using weaving. Top field comes first.
388 	 *  V4L2_FIELD_SEQ_TB --> V4L2_FIELD_NONE:
389 	 * 	top field from the input buffer is copied to the output buffer
390 	 * 	using line doubling. Bottom field from the input buffer is discarded.
391 	 * V4L2_FIELD_SEQ_BT --> V4L2_FIELD_INTERLACED_BT:
392 	 *	two separate fields in the same input buffer are interlaced
393 	 * 	in the output buffer using weaving. Bottom field comes first.
394 	 * V4L2_FIELD_SEQ_BT --> V4L2_FIELD_NONE:
395 	 * 	bottom field from the input buffer is copied to the output buffer
396 	 * 	using line doubling. Top field from the input buffer is discarded.
397 	 */
398 	switch (dst_q_data->fmt->fourcc) {
399 	case V4L2_PIX_FMT_YUV420:
400 		switch (dst_q_data->field) {
401 		case V4L2_FIELD_INTERLACED_TB:
402 		case V4L2_FIELD_INTERLACED_BT:
403 			dprintk(ctx->dev, "%s: yuv420 interlaced tb.\n",
404 				__func__);
405 			deinterlace_issue_dma(ctx, YUV420_DMA_Y_ODD, 0);
406 			deinterlace_issue_dma(ctx, YUV420_DMA_Y_EVEN, 0);
407 			deinterlace_issue_dma(ctx, YUV420_DMA_U_ODD, 0);
408 			deinterlace_issue_dma(ctx, YUV420_DMA_U_EVEN, 0);
409 			deinterlace_issue_dma(ctx, YUV420_DMA_V_ODD, 0);
410 			deinterlace_issue_dma(ctx, YUV420_DMA_V_EVEN, 1);
411 			break;
412 		case V4L2_FIELD_NONE:
413 		default:
414 			dprintk(ctx->dev, "%s: yuv420 interlaced line doubling.\n",
415 				__func__);
416 			deinterlace_issue_dma(ctx, YUV420_DMA_Y_ODD, 0);
417 			deinterlace_issue_dma(ctx, YUV420_DMA_Y_ODD_DOUBLING, 0);
418 			deinterlace_issue_dma(ctx, YUV420_DMA_U_ODD, 0);
419 			deinterlace_issue_dma(ctx, YUV420_DMA_U_ODD_DOUBLING, 0);
420 			deinterlace_issue_dma(ctx, YUV420_DMA_V_ODD, 0);
421 			deinterlace_issue_dma(ctx, YUV420_DMA_V_ODD_DOUBLING, 1);
422 			break;
423 		}
424 		break;
425 	case V4L2_PIX_FMT_YUYV:
426 	default:
427 		switch (dst_q_data->field) {
428 		case V4L2_FIELD_INTERLACED_TB:
429 		case V4L2_FIELD_INTERLACED_BT:
430 			dprintk(ctx->dev, "%s: yuyv interlaced_tb.\n",
431 				__func__);
432 			deinterlace_issue_dma(ctx, YUYV_DMA_ODD, 0);
433 			deinterlace_issue_dma(ctx, YUYV_DMA_EVEN, 1);
434 			break;
435 		case V4L2_FIELD_NONE:
436 		default:
437 			dprintk(ctx->dev, "%s: yuyv interlaced line doubling.\n",
438 				__func__);
439 			deinterlace_issue_dma(ctx, YUYV_DMA_ODD, 0);
440 			deinterlace_issue_dma(ctx, YUYV_DMA_EVEN_DOUBLING, 1);
441 			break;
442 		}
443 		break;
444 	}
445 
446 	dprintk(ctx->dev, "%s: DMA issue done.\n", __func__);
447 }
448 
449 /*
450  * video ioctls
451  */
452 static int vidioc_querycap(struct file *file, void *priv,
453 			   struct v4l2_capability *cap)
454 {
455 	strlcpy(cap->driver, MEM2MEM_NAME, sizeof(cap->driver));
456 	strlcpy(cap->card, MEM2MEM_NAME, sizeof(cap->card));
457 	strlcpy(cap->bus_info, MEM2MEM_NAME, sizeof(cap->card));
458 	/*
459 	 * This is only a mem-to-mem video device. The capture and output
460 	 * device capability flags are left only for backward compatibility
461 	 * and are scheduled for removal.
462 	 */
463 	cap->device_caps = V4L2_CAP_VIDEO_CAPTURE | V4L2_CAP_VIDEO_OUTPUT |
464 			   V4L2_CAP_VIDEO_M2M | V4L2_CAP_STREAMING;
465 	cap->capabilities = cap->device_caps | V4L2_CAP_DEVICE_CAPS;
466 
467 	return 0;
468 }
469 
470 static int enum_fmt(struct v4l2_fmtdesc *f, u32 type)
471 {
472 	int i, num;
473 	struct deinterlace_fmt *fmt;
474 
475 	num = 0;
476 
477 	for (i = 0; i < NUM_FORMATS; ++i) {
478 		if (formats[i].types & type) {
479 			/* index-th format of type type found ? */
480 			if (num == f->index)
481 				break;
482 			/* Correct type but haven't reached our index yet,
483 			 * just increment per-type index */
484 			++num;
485 		}
486 	}
487 
488 	if (i < NUM_FORMATS) {
489 		/* Format found */
490 		fmt = &formats[i];
491 		strlcpy(f->description, fmt->name, sizeof(f->description));
492 		f->pixelformat = fmt->fourcc;
493 		return 0;
494 	}
495 
496 	/* Format not found */
497 	return -EINVAL;
498 }
499 
500 static int vidioc_enum_fmt_vid_cap(struct file *file, void *priv,
501 				   struct v4l2_fmtdesc *f)
502 {
503 	return enum_fmt(f, MEM2MEM_CAPTURE);
504 }
505 
506 static int vidioc_enum_fmt_vid_out(struct file *file, void *priv,
507 				   struct v4l2_fmtdesc *f)
508 {
509 	return enum_fmt(f, MEM2MEM_OUTPUT);
510 }
511 
512 static int vidioc_g_fmt(struct deinterlace_ctx *ctx, struct v4l2_format *f)
513 {
514 	struct vb2_queue *vq;
515 	struct deinterlace_q_data *q_data;
516 
517 	vq = v4l2_m2m_get_vq(ctx->m2m_ctx, f->type);
518 	if (!vq)
519 		return -EINVAL;
520 
521 	q_data = get_q_data(f->type);
522 
523 	f->fmt.pix.width	= q_data->width;
524 	f->fmt.pix.height	= q_data->height;
525 	f->fmt.pix.field	= q_data->field;
526 	f->fmt.pix.pixelformat	= q_data->fmt->fourcc;
527 
528 	switch (q_data->fmt->fourcc) {
529 	case V4L2_PIX_FMT_YUV420:
530 		f->fmt.pix.bytesperline = q_data->width * 3 / 2;
531 		break;
532 	case V4L2_PIX_FMT_YUYV:
533 	default:
534 		f->fmt.pix.bytesperline = q_data->width * 2;
535 	}
536 
537 	f->fmt.pix.sizeimage	= q_data->sizeimage;
538 	f->fmt.pix.colorspace	= ctx->colorspace;
539 
540 	return 0;
541 }
542 
543 static int vidioc_g_fmt_vid_out(struct file *file, void *priv,
544 				struct v4l2_format *f)
545 {
546 	return vidioc_g_fmt(priv, f);
547 }
548 
549 static int vidioc_g_fmt_vid_cap(struct file *file, void *priv,
550 				struct v4l2_format *f)
551 {
552 	return vidioc_g_fmt(priv, f);
553 }
554 
555 static int vidioc_try_fmt(struct v4l2_format *f, struct deinterlace_fmt *fmt)
556 {
557 	switch (f->fmt.pix.pixelformat) {
558 	case V4L2_PIX_FMT_YUV420:
559 		f->fmt.pix.bytesperline = f->fmt.pix.width * 3 / 2;
560 		break;
561 	case V4L2_PIX_FMT_YUYV:
562 	default:
563 		f->fmt.pix.bytesperline = f->fmt.pix.width * 2;
564 	}
565 	f->fmt.pix.sizeimage = f->fmt.pix.height * f->fmt.pix.bytesperline;
566 
567 	return 0;
568 }
569 
570 static int vidioc_try_fmt_vid_cap(struct file *file, void *priv,
571 				  struct v4l2_format *f)
572 {
573 	struct deinterlace_fmt *fmt;
574 	struct deinterlace_ctx *ctx = priv;
575 
576 	fmt = find_format(f);
577 	if (!fmt || !(fmt->types & MEM2MEM_CAPTURE))
578 		f->fmt.pix.pixelformat = V4L2_PIX_FMT_YUV420;
579 
580 	f->fmt.pix.colorspace = ctx->colorspace;
581 
582 	if (f->fmt.pix.field != V4L2_FIELD_INTERLACED_TB &&
583 	    f->fmt.pix.field != V4L2_FIELD_INTERLACED_BT &&
584 	    f->fmt.pix.field != V4L2_FIELD_NONE)
585 		f->fmt.pix.field = V4L2_FIELD_INTERLACED_TB;
586 
587 	return vidioc_try_fmt(f, fmt);
588 }
589 
590 static int vidioc_try_fmt_vid_out(struct file *file, void *priv,
591 				  struct v4l2_format *f)
592 {
593 	struct deinterlace_fmt *fmt;
594 
595 	fmt = find_format(f);
596 	if (!fmt || !(fmt->types & MEM2MEM_OUTPUT))
597 		f->fmt.pix.pixelformat = V4L2_PIX_FMT_YUV420;
598 
599 	if (!f->fmt.pix.colorspace)
600 		f->fmt.pix.colorspace = V4L2_COLORSPACE_REC709;
601 
602 	if (f->fmt.pix.field != V4L2_FIELD_SEQ_TB &&
603 	    f->fmt.pix.field != V4L2_FIELD_SEQ_BT)
604 		f->fmt.pix.field = V4L2_FIELD_SEQ_TB;
605 
606 	return vidioc_try_fmt(f, fmt);
607 }
608 
609 static int vidioc_s_fmt(struct deinterlace_ctx *ctx, struct v4l2_format *f)
610 {
611 	struct deinterlace_q_data *q_data;
612 	struct vb2_queue *vq;
613 
614 	vq = v4l2_m2m_get_vq(ctx->m2m_ctx, f->type);
615 	if (!vq)
616 		return -EINVAL;
617 
618 	q_data = get_q_data(f->type);
619 	if (!q_data)
620 		return -EINVAL;
621 
622 	if (vb2_is_busy(vq)) {
623 		v4l2_err(&ctx->dev->v4l2_dev, "%s queue busy\n", __func__);
624 		return -EBUSY;
625 	}
626 
627 	q_data->fmt = find_format(f);
628 	if (!q_data->fmt) {
629 		v4l2_err(&ctx->dev->v4l2_dev,
630 			 "Couldn't set format type %d, wxh: %dx%d. fmt: %d, field: %d\n",
631 			f->type, f->fmt.pix.width, f->fmt.pix.height,
632 			f->fmt.pix.pixelformat, f->fmt.pix.field);
633 		return -EINVAL;
634 	}
635 
636 	q_data->width		= f->fmt.pix.width;
637 	q_data->height		= f->fmt.pix.height;
638 	q_data->field		= f->fmt.pix.field;
639 
640 	switch (f->fmt.pix.pixelformat) {
641 	case V4L2_PIX_FMT_YUV420:
642 		f->fmt.pix.bytesperline = f->fmt.pix.width * 3 / 2;
643 		q_data->sizeimage = (q_data->width * q_data->height * 3) / 2;
644 		break;
645 	case V4L2_PIX_FMT_YUYV:
646 	default:
647 		f->fmt.pix.bytesperline = f->fmt.pix.width * 2;
648 		q_data->sizeimage = q_data->width * q_data->height * 2;
649 	}
650 
651 	dprintk(ctx->dev,
652 		"Setting format for type %d, wxh: %dx%d, fmt: %d, field: %d\n",
653 		f->type, q_data->width, q_data->height, q_data->fmt->fourcc,
654 		q_data->field);
655 
656 	return 0;
657 }
658 
659 static int vidioc_s_fmt_vid_cap(struct file *file, void *priv,
660 				struct v4l2_format *f)
661 {
662 	int ret;
663 
664 	ret = vidioc_try_fmt_vid_cap(file, priv, f);
665 	if (ret)
666 		return ret;
667 	return vidioc_s_fmt(priv, f);
668 }
669 
670 static int vidioc_s_fmt_vid_out(struct file *file, void *priv,
671 				struct v4l2_format *f)
672 {
673 	struct deinterlace_ctx *ctx = priv;
674 	int ret;
675 
676 	ret = vidioc_try_fmt_vid_out(file, priv, f);
677 	if (ret)
678 		return ret;
679 
680 	ret = vidioc_s_fmt(priv, f);
681 	if (!ret)
682 		ctx->colorspace = f->fmt.pix.colorspace;
683 
684 	return ret;
685 }
686 
687 static int vidioc_reqbufs(struct file *file, void *priv,
688 			  struct v4l2_requestbuffers *reqbufs)
689 {
690 	struct deinterlace_ctx *ctx = priv;
691 
692 	return v4l2_m2m_reqbufs(file, ctx->m2m_ctx, reqbufs);
693 }
694 
695 static int vidioc_querybuf(struct file *file, void *priv,
696 			   struct v4l2_buffer *buf)
697 {
698 	struct deinterlace_ctx *ctx = priv;
699 
700 	return v4l2_m2m_querybuf(file, ctx->m2m_ctx, buf);
701 }
702 
703 static int vidioc_qbuf(struct file *file, void *priv, struct v4l2_buffer *buf)
704 {
705 	struct deinterlace_ctx *ctx = priv;
706 
707 	return v4l2_m2m_qbuf(file, ctx->m2m_ctx, buf);
708 }
709 
710 static int vidioc_dqbuf(struct file *file, void *priv, struct v4l2_buffer *buf)
711 {
712 	struct deinterlace_ctx *ctx = priv;
713 
714 	return v4l2_m2m_dqbuf(file, ctx->m2m_ctx, buf);
715 }
716 
717 static int vidioc_streamon(struct file *file, void *priv,
718 			   enum v4l2_buf_type type)
719 {
720 	struct deinterlace_q_data *s_q_data, *d_q_data;
721 	struct deinterlace_ctx *ctx = priv;
722 
723 	s_q_data = get_q_data(V4L2_BUF_TYPE_VIDEO_OUTPUT);
724 	d_q_data = get_q_data(V4L2_BUF_TYPE_VIDEO_CAPTURE);
725 
726 	/* Check that src and dst queues have the same pix format */
727 	if (s_q_data->fmt->fourcc != d_q_data->fmt->fourcc) {
728 		v4l2_err(&ctx->dev->v4l2_dev,
729 			 "src and dst formats don't match.\n");
730 		return -EINVAL;
731 	}
732 
733 	/* Check that input and output deinterlacing types are compatible */
734 	switch (s_q_data->field) {
735 	case V4L2_FIELD_SEQ_BT:
736 		if (d_q_data->field != V4L2_FIELD_NONE &&
737 			d_q_data->field != V4L2_FIELD_INTERLACED_BT) {
738 			v4l2_err(&ctx->dev->v4l2_dev,
739 			 "src and dst field conversion [(%d)->(%d)] not supported.\n",
740 				s_q_data->field, d_q_data->field);
741 			return -EINVAL;
742 		}
743 		break;
744 	case V4L2_FIELD_SEQ_TB:
745 		if (d_q_data->field != V4L2_FIELD_NONE &&
746 			d_q_data->field != V4L2_FIELD_INTERLACED_TB) {
747 			v4l2_err(&ctx->dev->v4l2_dev,
748 			 "src and dst field conversion [(%d)->(%d)] not supported.\n",
749 				s_q_data->field, d_q_data->field);
750 			return -EINVAL;
751 		}
752 		break;
753 	default:
754 		return -EINVAL;
755 	}
756 
757 	return v4l2_m2m_streamon(file, ctx->m2m_ctx, type);
758 }
759 
760 static int vidioc_streamoff(struct file *file, void *priv,
761 			    enum v4l2_buf_type type)
762 {
763 	struct deinterlace_ctx *ctx = priv;
764 
765 	return v4l2_m2m_streamoff(file, ctx->m2m_ctx, type);
766 }
767 
768 static const struct v4l2_ioctl_ops deinterlace_ioctl_ops = {
769 	.vidioc_querycap	= vidioc_querycap,
770 
771 	.vidioc_enum_fmt_vid_cap = vidioc_enum_fmt_vid_cap,
772 	.vidioc_g_fmt_vid_cap	= vidioc_g_fmt_vid_cap,
773 	.vidioc_try_fmt_vid_cap	= vidioc_try_fmt_vid_cap,
774 	.vidioc_s_fmt_vid_cap	= vidioc_s_fmt_vid_cap,
775 
776 	.vidioc_enum_fmt_vid_out = vidioc_enum_fmt_vid_out,
777 	.vidioc_g_fmt_vid_out	= vidioc_g_fmt_vid_out,
778 	.vidioc_try_fmt_vid_out	= vidioc_try_fmt_vid_out,
779 	.vidioc_s_fmt_vid_out	= vidioc_s_fmt_vid_out,
780 
781 	.vidioc_reqbufs		= vidioc_reqbufs,
782 	.vidioc_querybuf	= vidioc_querybuf,
783 
784 	.vidioc_qbuf		= vidioc_qbuf,
785 	.vidioc_dqbuf		= vidioc_dqbuf,
786 
787 	.vidioc_streamon	= vidioc_streamon,
788 	.vidioc_streamoff	= vidioc_streamoff,
789 };
790 
791 
792 /*
793  * Queue operations
794  */
795 struct vb2_dc_conf {
796 	struct device           *dev;
797 };
798 
799 static int deinterlace_queue_setup(struct vb2_queue *vq,
800 				const struct v4l2_format *fmt,
801 				unsigned int *nbuffers, unsigned int *nplanes,
802 				unsigned int sizes[], void *alloc_ctxs[])
803 {
804 	struct deinterlace_ctx *ctx = vb2_get_drv_priv(vq);
805 	struct deinterlace_q_data *q_data;
806 	unsigned int size, count = *nbuffers;
807 
808 	q_data = get_q_data(vq->type);
809 
810 	switch (q_data->fmt->fourcc) {
811 	case V4L2_PIX_FMT_YUV420:
812 		size = q_data->width * q_data->height * 3 / 2;
813 		break;
814 	case V4L2_PIX_FMT_YUYV:
815 	default:
816 		size = q_data->width * q_data->height * 2;
817 	}
818 
819 	*nplanes = 1;
820 	*nbuffers = count;
821 	sizes[0] = size;
822 
823 	alloc_ctxs[0] = ctx->dev->alloc_ctx;
824 
825 	dprintk(ctx->dev, "get %d buffer(s) of size %d each.\n", count, size);
826 
827 	return 0;
828 }
829 
830 static int deinterlace_buf_prepare(struct vb2_buffer *vb)
831 {
832 	struct deinterlace_ctx *ctx = vb2_get_drv_priv(vb->vb2_queue);
833 	struct deinterlace_q_data *q_data;
834 
835 	dprintk(ctx->dev, "type: %d\n", vb->vb2_queue->type);
836 
837 	q_data = get_q_data(vb->vb2_queue->type);
838 
839 	if (vb2_plane_size(vb, 0) < q_data->sizeimage) {
840 		dprintk(ctx->dev, "%s data will not fit into plane (%lu < %lu)\n",
841 			__func__, vb2_plane_size(vb, 0), (long)q_data->sizeimage);
842 		return -EINVAL;
843 	}
844 
845 	vb2_set_plane_payload(vb, 0, q_data->sizeimage);
846 
847 	return 0;
848 }
849 
850 static void deinterlace_buf_queue(struct vb2_buffer *vb)
851 {
852 	struct deinterlace_ctx *ctx = vb2_get_drv_priv(vb->vb2_queue);
853 	v4l2_m2m_buf_queue(ctx->m2m_ctx, vb);
854 }
855 
856 static struct vb2_ops deinterlace_qops = {
857 	.queue_setup	 = deinterlace_queue_setup,
858 	.buf_prepare	 = deinterlace_buf_prepare,
859 	.buf_queue	 = deinterlace_buf_queue,
860 };
861 
862 static int queue_init(void *priv, struct vb2_queue *src_vq,
863 		      struct vb2_queue *dst_vq)
864 {
865 	struct deinterlace_ctx *ctx = priv;
866 	int ret;
867 
868 	src_vq->type = V4L2_BUF_TYPE_VIDEO_OUTPUT;
869 	src_vq->io_modes = VB2_MMAP | VB2_USERPTR;
870 	src_vq->drv_priv = ctx;
871 	src_vq->buf_struct_size = sizeof(struct v4l2_m2m_buffer);
872 	src_vq->ops = &deinterlace_qops;
873 	src_vq->mem_ops = &vb2_dma_contig_memops;
874 	src_vq->timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_COPY;
875 	q_data[V4L2_M2M_SRC].fmt = &formats[0];
876 	q_data[V4L2_M2M_SRC].width = 640;
877 	q_data[V4L2_M2M_SRC].height = 480;
878 	q_data[V4L2_M2M_SRC].sizeimage = (640 * 480 * 3) / 2;
879 	q_data[V4L2_M2M_SRC].field = V4L2_FIELD_SEQ_TB;
880 
881 	ret = vb2_queue_init(src_vq);
882 	if (ret)
883 		return ret;
884 
885 	dst_vq->type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
886 	dst_vq->io_modes = VB2_MMAP | VB2_USERPTR;
887 	dst_vq->drv_priv = ctx;
888 	dst_vq->buf_struct_size = sizeof(struct v4l2_m2m_buffer);
889 	dst_vq->ops = &deinterlace_qops;
890 	dst_vq->mem_ops = &vb2_dma_contig_memops;
891 	dst_vq->timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_COPY;
892 	q_data[V4L2_M2M_DST].fmt = &formats[0];
893 	q_data[V4L2_M2M_DST].width = 640;
894 	q_data[V4L2_M2M_DST].height = 480;
895 	q_data[V4L2_M2M_DST].sizeimage = (640 * 480 * 3) / 2;
896 	q_data[V4L2_M2M_SRC].field = V4L2_FIELD_INTERLACED_TB;
897 
898 	return vb2_queue_init(dst_vq);
899 }
900 
901 /*
902  * File operations
903  */
904 static int deinterlace_open(struct file *file)
905 {
906 	struct deinterlace_dev *pcdev = video_drvdata(file);
907 	struct deinterlace_ctx *ctx = NULL;
908 
909 	ctx = kzalloc(sizeof *ctx, GFP_KERNEL);
910 	if (!ctx)
911 		return -ENOMEM;
912 
913 	file->private_data = ctx;
914 	ctx->dev = pcdev;
915 
916 	ctx->m2m_ctx = v4l2_m2m_ctx_init(pcdev->m2m_dev, ctx, &queue_init);
917 	if (IS_ERR(ctx->m2m_ctx)) {
918 		int ret = PTR_ERR(ctx->m2m_ctx);
919 
920 		kfree(ctx);
921 		return ret;
922 	}
923 
924 	ctx->xt = kzalloc(sizeof(struct dma_interleaved_template) +
925 				sizeof(struct data_chunk), GFP_KERNEL);
926 	if (!ctx->xt) {
927 		kfree(ctx);
928 		return -ENOMEM;
929 	}
930 
931 	ctx->colorspace = V4L2_COLORSPACE_REC709;
932 
933 	dprintk(pcdev, "Created instance %p, m2m_ctx: %p\n", ctx, ctx->m2m_ctx);
934 
935 	return 0;
936 }
937 
938 static int deinterlace_release(struct file *file)
939 {
940 	struct deinterlace_dev *pcdev = video_drvdata(file);
941 	struct deinterlace_ctx *ctx = file->private_data;
942 
943 	dprintk(pcdev, "Releasing instance %p\n", ctx);
944 
945 	v4l2_m2m_ctx_release(ctx->m2m_ctx);
946 	kfree(ctx->xt);
947 	kfree(ctx);
948 
949 	return 0;
950 }
951 
952 static unsigned int deinterlace_poll(struct file *file,
953 				 struct poll_table_struct *wait)
954 {
955 	struct deinterlace_ctx *ctx = file->private_data;
956 	int ret;
957 
958 	deinterlace_lock(ctx);
959 	ret = v4l2_m2m_poll(file, ctx->m2m_ctx, wait);
960 	deinterlace_unlock(ctx);
961 
962 	return ret;
963 }
964 
965 static int deinterlace_mmap(struct file *file, struct vm_area_struct *vma)
966 {
967 	struct deinterlace_ctx *ctx = file->private_data;
968 
969 	return v4l2_m2m_mmap(file, ctx->m2m_ctx, vma);
970 }
971 
972 static const struct v4l2_file_operations deinterlace_fops = {
973 	.owner		= THIS_MODULE,
974 	.open		= deinterlace_open,
975 	.release	= deinterlace_release,
976 	.poll		= deinterlace_poll,
977 	.unlocked_ioctl	= video_ioctl2,
978 	.mmap		= deinterlace_mmap,
979 };
980 
981 static struct video_device deinterlace_videodev = {
982 	.name		= MEM2MEM_NAME,
983 	.fops		= &deinterlace_fops,
984 	.ioctl_ops	= &deinterlace_ioctl_ops,
985 	.minor		= -1,
986 	.release	= video_device_release_empty,
987 	.vfl_dir	= VFL_DIR_M2M,
988 };
989 
990 static struct v4l2_m2m_ops m2m_ops = {
991 	.device_run	= deinterlace_device_run,
992 	.job_ready	= deinterlace_job_ready,
993 	.job_abort	= deinterlace_job_abort,
994 	.lock		= deinterlace_lock,
995 	.unlock		= deinterlace_unlock,
996 };
997 
998 static int deinterlace_probe(struct platform_device *pdev)
999 {
1000 	struct deinterlace_dev *pcdev;
1001 	struct video_device *vfd;
1002 	dma_cap_mask_t mask;
1003 	int ret = 0;
1004 
1005 	pcdev = devm_kzalloc(&pdev->dev, sizeof(*pcdev), GFP_KERNEL);
1006 	if (!pcdev)
1007 		return -ENOMEM;
1008 
1009 	spin_lock_init(&pcdev->irqlock);
1010 
1011 	dma_cap_zero(mask);
1012 	dma_cap_set(DMA_INTERLEAVE, mask);
1013 	pcdev->dma_chan = dma_request_channel(mask, NULL, pcdev);
1014 	if (!pcdev->dma_chan)
1015 		return -ENODEV;
1016 
1017 	if (!dma_has_cap(DMA_INTERLEAVE, pcdev->dma_chan->device->cap_mask)) {
1018 		v4l2_err(&pcdev->v4l2_dev, "DMA does not support INTERLEAVE\n");
1019 		goto rel_dma;
1020 	}
1021 
1022 	ret = v4l2_device_register(&pdev->dev, &pcdev->v4l2_dev);
1023 	if (ret)
1024 		goto rel_dma;
1025 
1026 	atomic_set(&pcdev->busy, 0);
1027 	mutex_init(&pcdev->dev_mutex);
1028 
1029 	vfd = &pcdev->vfd;
1030 	*vfd = deinterlace_videodev;
1031 	vfd->lock = &pcdev->dev_mutex;
1032 	vfd->v4l2_dev = &pcdev->v4l2_dev;
1033 
1034 	ret = video_register_device(vfd, VFL_TYPE_GRABBER, 0);
1035 	if (ret) {
1036 		v4l2_err(&pcdev->v4l2_dev, "Failed to register video device\n");
1037 		goto unreg_dev;
1038 	}
1039 
1040 	video_set_drvdata(vfd, pcdev);
1041 	snprintf(vfd->name, sizeof(vfd->name), "%s", deinterlace_videodev.name);
1042 	v4l2_info(&pcdev->v4l2_dev, MEM2MEM_TEST_MODULE_NAME
1043 			" Device registered as /dev/video%d\n", vfd->num);
1044 
1045 	platform_set_drvdata(pdev, pcdev);
1046 
1047 	pcdev->alloc_ctx = vb2_dma_contig_init_ctx(&pdev->dev);
1048 	if (IS_ERR(pcdev->alloc_ctx)) {
1049 		v4l2_err(&pcdev->v4l2_dev, "Failed to alloc vb2 context\n");
1050 		ret = PTR_ERR(pcdev->alloc_ctx);
1051 		goto err_ctx;
1052 	}
1053 
1054 	pcdev->m2m_dev = v4l2_m2m_init(&m2m_ops);
1055 	if (IS_ERR(pcdev->m2m_dev)) {
1056 		v4l2_err(&pcdev->v4l2_dev, "Failed to init mem2mem device\n");
1057 		ret = PTR_ERR(pcdev->m2m_dev);
1058 		goto err_m2m;
1059 	}
1060 
1061 	return 0;
1062 
1063 err_m2m:
1064 	video_unregister_device(&pcdev->vfd);
1065 err_ctx:
1066 	vb2_dma_contig_cleanup_ctx(pcdev->alloc_ctx);
1067 unreg_dev:
1068 	v4l2_device_unregister(&pcdev->v4l2_dev);
1069 rel_dma:
1070 	dma_release_channel(pcdev->dma_chan);
1071 
1072 	return ret;
1073 }
1074 
1075 static int deinterlace_remove(struct platform_device *pdev)
1076 {
1077 	struct deinterlace_dev *pcdev = platform_get_drvdata(pdev);
1078 
1079 	v4l2_info(&pcdev->v4l2_dev, "Removing " MEM2MEM_TEST_MODULE_NAME);
1080 	v4l2_m2m_release(pcdev->m2m_dev);
1081 	video_unregister_device(&pcdev->vfd);
1082 	v4l2_device_unregister(&pcdev->v4l2_dev);
1083 	vb2_dma_contig_cleanup_ctx(pcdev->alloc_ctx);
1084 	dma_release_channel(pcdev->dma_chan);
1085 
1086 	return 0;
1087 }
1088 
1089 static struct platform_driver deinterlace_pdrv = {
1090 	.probe		= deinterlace_probe,
1091 	.remove		= deinterlace_remove,
1092 	.driver		= {
1093 		.name	= MEM2MEM_NAME,
1094 	},
1095 };
1096 module_platform_driver(deinterlace_pdrv);
1097 
1098