afbb4636e714d5b5d064cb1f0b4eb53d3571febe
[linux-2.6-microblaze.git] / drivers / staging / media / ipu3 / ipu3-v4l2.c
1 // SPDX-License-Identifier: GPL-2.0
2 // Copyright (C) 2018 Intel Corporation
3
4 #include <linux/module.h>
5 #include <linux/pm_runtime.h>
6
7 #include <media/v4l2-event.h>
8 #include <media/v4l2-ioctl.h>
9
10 #include "ipu3.h"
11 #include "ipu3-dmamap.h"
12
13 /******************** v4l2_subdev_ops ********************/
14
15 #define IPU3_RUNNING_MODE_VIDEO         0
16 #define IPU3_RUNNING_MODE_STILL         1
17
18 static int imgu_subdev_open(struct v4l2_subdev *sd, struct v4l2_subdev_fh *fh)
19 {
20         struct imgu_v4l2_subdev *imgu_sd = container_of(sd,
21                                                         struct imgu_v4l2_subdev,
22                                                         subdev);
23         struct imgu_device *imgu = v4l2_get_subdevdata(sd);
24         struct imgu_media_pipe *imgu_pipe = &imgu->imgu_pipe[imgu_sd->pipe];
25         struct v4l2_rect try_crop = {
26                 .top = 0,
27                 .left = 0,
28         };
29         unsigned int i;
30
31         try_crop.width =
32                 imgu_pipe->nodes[IMGU_NODE_IN].vdev_fmt.fmt.pix_mp.width;
33         try_crop.height =
34                 imgu_pipe->nodes[IMGU_NODE_IN].vdev_fmt.fmt.pix_mp.height;
35
36         /* Initialize try_fmt */
37         for (i = 0; i < IMGU_NODE_NUM; i++) {
38                 struct v4l2_mbus_framefmt *try_fmt =
39                         v4l2_subdev_get_try_format(sd, fh->pad, i);
40
41                 try_fmt->width = try_crop.width;
42                 try_fmt->height = try_crop.height;
43                 try_fmt->code = imgu_pipe->nodes[i].pad_fmt.code;
44                 try_fmt->field = V4L2_FIELD_NONE;
45         }
46
47         *v4l2_subdev_get_try_crop(sd, fh->pad, IMGU_NODE_IN) = try_crop;
48         *v4l2_subdev_get_try_compose(sd, fh->pad, IMGU_NODE_IN) = try_crop;
49
50         return 0;
51 }
52
53 static int imgu_subdev_s_stream(struct v4l2_subdev *sd, int enable)
54 {
55         int i;
56         unsigned int node;
57         int r = 0;
58         struct imgu_device *imgu = v4l2_get_subdevdata(sd);
59         struct imgu_v4l2_subdev *imgu_sd = container_of(sd,
60                                                         struct imgu_v4l2_subdev,
61                                                         subdev);
62         unsigned int pipe = imgu_sd->pipe;
63         struct device *dev = &imgu->pci_dev->dev;
64         struct v4l2_pix_format_mplane *fmts[IPU3_CSS_QUEUES] = { NULL };
65         struct v4l2_rect *rects[IPU3_CSS_RECTS] = { NULL };
66         struct imgu_css_pipe *css_pipe = &imgu->css.pipes[pipe];
67         struct imgu_media_pipe *imgu_pipe = &imgu->imgu_pipe[pipe];
68
69         dev_dbg(dev, "%s %d for pipe %u", __func__, enable, pipe);
70         /* grab ctrl after streamon and return after off */
71         v4l2_ctrl_grab(imgu_sd->ctrl, enable);
72
73         if (!enable) {
74                 imgu_sd->active = false;
75                 return 0;
76         }
77
78         for (i = 0; i < IMGU_NODE_NUM; i++)
79                 imgu_pipe->queue_enabled[i] = imgu_pipe->nodes[i].enabled;
80
81         /* This is handled specially */
82         imgu_pipe->queue_enabled[IPU3_CSS_QUEUE_PARAMS] = false;
83
84         /* Initialize CSS formats */
85         for (i = 0; i < IPU3_CSS_QUEUES; i++) {
86                 node = imgu_map_node(imgu, i);
87                 /* No need to reconfig meta nodes */
88                 if (node == IMGU_NODE_STAT_3A || node == IMGU_NODE_PARAMS)
89                         continue;
90                 fmts[i] = imgu_pipe->queue_enabled[node] ?
91                         &imgu_pipe->nodes[node].vdev_fmt.fmt.pix_mp : NULL;
92         }
93
94         /* Enable VF output only when VF queue requested by user */
95         css_pipe->vf_output_en = false;
96         if (imgu_pipe->nodes[IMGU_NODE_VF].enabled)
97                 css_pipe->vf_output_en = true;
98
99         if (atomic_read(&imgu_sd->running_mode) == IPU3_RUNNING_MODE_VIDEO)
100                 css_pipe->pipe_id = IPU3_CSS_PIPE_ID_VIDEO;
101         else
102                 css_pipe->pipe_id = IPU3_CSS_PIPE_ID_CAPTURE;
103
104         dev_dbg(dev, "IPU3 pipe %u pipe_id %u", pipe, css_pipe->pipe_id);
105
106         rects[IPU3_CSS_RECT_EFFECTIVE] = &imgu_sd->rect.eff;
107         rects[IPU3_CSS_RECT_BDS] = &imgu_sd->rect.bds;
108         rects[IPU3_CSS_RECT_GDC] = &imgu_sd->rect.gdc;
109
110         r = imgu_css_fmt_set(&imgu->css, fmts, rects, pipe);
111         if (r) {
112                 dev_err(dev, "failed to set initial formats pipe %u with (%d)",
113                         pipe, r);
114                 return r;
115         }
116
117         imgu_sd->active = true;
118
119         return 0;
120 }
121
122 static int imgu_subdev_get_fmt(struct v4l2_subdev *sd,
123                                struct v4l2_subdev_pad_config *cfg,
124                                struct v4l2_subdev_format *fmt)
125 {
126         struct imgu_device *imgu = v4l2_get_subdevdata(sd);
127         struct v4l2_mbus_framefmt *mf;
128         struct imgu_media_pipe *imgu_pipe;
129         u32 pad = fmt->pad;
130         struct imgu_v4l2_subdev *imgu_sd = container_of(sd,
131                                                         struct imgu_v4l2_subdev,
132                                                         subdev);
133         unsigned int pipe = imgu_sd->pipe;
134
135         imgu_pipe = &imgu->imgu_pipe[pipe];
136         if (fmt->which == V4L2_SUBDEV_FORMAT_ACTIVE) {
137                 fmt->format = imgu_pipe->nodes[pad].pad_fmt;
138         } else {
139                 mf = v4l2_subdev_get_try_format(sd, cfg, pad);
140                 fmt->format = *mf;
141         }
142
143         return 0;
144 }
145
146 static int imgu_subdev_set_fmt(struct v4l2_subdev *sd,
147                                struct v4l2_subdev_pad_config *cfg,
148                                struct v4l2_subdev_format *fmt)
149 {
150         struct imgu_media_pipe *imgu_pipe;
151         struct imgu_device *imgu = v4l2_get_subdevdata(sd);
152         struct imgu_v4l2_subdev *imgu_sd = container_of(sd,
153                                                         struct imgu_v4l2_subdev,
154                                                         subdev);
155         struct v4l2_mbus_framefmt *mf;
156         u32 pad = fmt->pad;
157         unsigned int pipe = imgu_sd->pipe;
158
159         dev_dbg(&imgu->pci_dev->dev, "set subdev %u pad %u fmt to [%ux%u]",
160                 pipe, pad, fmt->format.width, fmt->format.height);
161
162         imgu_pipe = &imgu->imgu_pipe[pipe];
163         if (fmt->which == V4L2_SUBDEV_FORMAT_TRY)
164                 mf = v4l2_subdev_get_try_format(sd, cfg, pad);
165         else
166                 mf = &imgu_pipe->nodes[pad].pad_fmt;
167
168         fmt->format.code = mf->code;
169         /* Clamp the w and h based on the hardware capabilities */
170         if (imgu_sd->subdev_pads[pad].flags & MEDIA_PAD_FL_SOURCE) {
171                 fmt->format.width = clamp(fmt->format.width,
172                                           IPU3_OUTPUT_MIN_WIDTH,
173                                           IPU3_OUTPUT_MAX_WIDTH);
174                 fmt->format.height = clamp(fmt->format.height,
175                                            IPU3_OUTPUT_MIN_HEIGHT,
176                                            IPU3_OUTPUT_MAX_HEIGHT);
177         } else {
178                 fmt->format.width = clamp(fmt->format.width,
179                                           IPU3_INPUT_MIN_WIDTH,
180                                           IPU3_INPUT_MAX_WIDTH);
181                 fmt->format.height = clamp(fmt->format.height,
182                                            IPU3_INPUT_MIN_HEIGHT,
183                                            IPU3_INPUT_MAX_HEIGHT);
184         }
185
186         *mf = fmt->format;
187
188         return 0;
189 }
190
191 static int imgu_subdev_get_selection(struct v4l2_subdev *sd,
192                                      struct v4l2_subdev_pad_config *cfg,
193                                      struct v4l2_subdev_selection *sel)
194 {
195         struct v4l2_rect *try_sel, *r;
196         struct imgu_v4l2_subdev *imgu_sd = container_of(sd,
197                                                         struct imgu_v4l2_subdev,
198                                                         subdev);
199
200         if (sel->pad != IMGU_NODE_IN)
201                 return -EINVAL;
202
203         switch (sel->target) {
204         case V4L2_SEL_TGT_CROP:
205                 try_sel = v4l2_subdev_get_try_crop(sd, cfg, sel->pad);
206                 r = &imgu_sd->rect.eff;
207                 break;
208         case V4L2_SEL_TGT_COMPOSE:
209                 try_sel = v4l2_subdev_get_try_compose(sd, cfg, sel->pad);
210                 r = &imgu_sd->rect.bds;
211                 break;
212         default:
213                 return -EINVAL;
214         }
215
216         if (sel->which == V4L2_SUBDEV_FORMAT_TRY)
217                 sel->r = *try_sel;
218         else
219                 sel->r = *r;
220
221         return 0;
222 }
223
224 static int imgu_subdev_set_selection(struct v4l2_subdev *sd,
225                                      struct v4l2_subdev_pad_config *cfg,
226                                      struct v4l2_subdev_selection *sel)
227 {
228         struct imgu_device *imgu = v4l2_get_subdevdata(sd);
229         struct imgu_v4l2_subdev *imgu_sd = container_of(sd,
230                                                         struct imgu_v4l2_subdev,
231                                                         subdev);
232         struct v4l2_rect *rect, *try_sel;
233
234         dev_dbg(&imgu->pci_dev->dev,
235                  "set subdev %u sel which %u target 0x%4x rect [%ux%u]",
236                  imgu_sd->pipe, sel->which, sel->target,
237                  sel->r.width, sel->r.height);
238
239         if (sel->pad != IMGU_NODE_IN)
240                 return -EINVAL;
241
242         switch (sel->target) {
243         case V4L2_SEL_TGT_CROP:
244                 try_sel = v4l2_subdev_get_try_crop(sd, cfg, sel->pad);
245                 rect = &imgu_sd->rect.eff;
246                 break;
247         case V4L2_SEL_TGT_COMPOSE:
248                 try_sel = v4l2_subdev_get_try_compose(sd, cfg, sel->pad);
249                 rect = &imgu_sd->rect.bds;
250                 break;
251         default:
252                 return -EINVAL;
253         }
254
255         if (sel->which == V4L2_SUBDEV_FORMAT_TRY)
256                 *try_sel = sel->r;
257         else
258                 *rect = sel->r;
259
260         return 0;
261 }
262
263 /******************** media_entity_operations ********************/
264
265 static int imgu_link_setup(struct media_entity *entity,
266                            const struct media_pad *local,
267                            const struct media_pad *remote, u32 flags)
268 {
269         struct imgu_media_pipe *imgu_pipe;
270         struct v4l2_subdev *sd = container_of(entity, struct v4l2_subdev,
271                                               entity);
272         struct imgu_device *imgu = v4l2_get_subdevdata(sd);
273         struct imgu_v4l2_subdev *imgu_sd = container_of(sd,
274                                                         struct imgu_v4l2_subdev,
275                                                         subdev);
276         unsigned int pipe = imgu_sd->pipe;
277         u32 pad = local->index;
278
279         WARN_ON(pad >= IMGU_NODE_NUM);
280
281         dev_dbg(&imgu->pci_dev->dev, "pipe %u pad %u is %s", pipe, pad,
282                  flags & MEDIA_LNK_FL_ENABLED ? "enabled" : "disabled");
283
284         imgu_pipe = &imgu->imgu_pipe[pipe];
285         imgu_pipe->nodes[pad].enabled = flags & MEDIA_LNK_FL_ENABLED;
286
287         /* enable input node to enable the pipe */
288         if (pad != IMGU_NODE_IN)
289                 return 0;
290
291         if (flags & MEDIA_LNK_FL_ENABLED)
292                 __set_bit(pipe, imgu->css.enabled_pipes);
293         else
294                 __clear_bit(pipe, imgu->css.enabled_pipes);
295
296         dev_dbg(&imgu->pci_dev->dev, "pipe %u is %s", pipe,
297                  flags & MEDIA_LNK_FL_ENABLED ? "enabled" : "disabled");
298
299         return 0;
300 }
301
302 /******************** vb2_ops ********************/
303
304 static int imgu_vb2_buf_init(struct vb2_buffer *vb)
305 {
306         struct sg_table *sg = vb2_dma_sg_plane_desc(vb, 0);
307         struct imgu_device *imgu = vb2_get_drv_priv(vb->vb2_queue);
308         struct imgu_buffer *buf = container_of(vb,
309                 struct imgu_buffer, vid_buf.vbb.vb2_buf);
310         struct imgu_video_device *node =
311                 container_of(vb->vb2_queue, struct imgu_video_device, vbq);
312         unsigned int queue = imgu_node_to_queue(node->id);
313
314         if (queue == IPU3_CSS_QUEUE_PARAMS)
315                 return 0;
316
317         return imgu_dmamap_map_sg(imgu, sg->sgl, sg->nents, &buf->map);
318 }
319
320 /* Called when each buffer is freed */
321 static void imgu_vb2_buf_cleanup(struct vb2_buffer *vb)
322 {
323         struct imgu_device *imgu = vb2_get_drv_priv(vb->vb2_queue);
324         struct imgu_buffer *buf = container_of(vb,
325                 struct imgu_buffer, vid_buf.vbb.vb2_buf);
326         struct imgu_video_device *node =
327                 container_of(vb->vb2_queue, struct imgu_video_device, vbq);
328         unsigned int queue = imgu_node_to_queue(node->id);
329
330         if (queue == IPU3_CSS_QUEUE_PARAMS)
331                 return;
332
333         imgu_dmamap_unmap(imgu, &buf->map);
334 }
335
336 /* Transfer buffer ownership to me */
337 static void imgu_vb2_buf_queue(struct vb2_buffer *vb)
338 {
339         struct imgu_device *imgu = vb2_get_drv_priv(vb->vb2_queue);
340         struct imgu_video_device *node =
341                 container_of(vb->vb2_queue, struct imgu_video_device, vbq);
342         unsigned int queue = imgu_node_to_queue(node->id);
343         struct imgu_buffer *buf = container_of(vb, struct imgu_buffer,
344                                                vid_buf.vbb.vb2_buf);
345         unsigned long need_bytes;
346         unsigned long payload = vb2_get_plane_payload(vb, 0);
347
348         if (vb->vb2_queue->type == V4L2_BUF_TYPE_META_CAPTURE ||
349             vb->vb2_queue->type == V4L2_BUF_TYPE_META_OUTPUT)
350                 need_bytes = node->vdev_fmt.fmt.meta.buffersize;
351         else
352                 need_bytes = node->vdev_fmt.fmt.pix_mp.plane_fmt[0].sizeimage;
353
354         if (queue == IPU3_CSS_QUEUE_PARAMS && payload && payload < need_bytes) {
355                 dev_err(&imgu->pci_dev->dev, "invalid data size for params.");
356                 vb2_buffer_done(vb, VB2_BUF_STATE_ERROR);
357                 return;
358         }
359
360         mutex_lock(&imgu->lock);
361         if (queue != IPU3_CSS_QUEUE_PARAMS)
362                 imgu_css_buf_init(&buf->css_buf, queue, buf->map.daddr);
363
364         list_add_tail(&buf->vid_buf.list, &node->buffers);
365         mutex_unlock(&imgu->lock);
366
367         vb2_set_plane_payload(vb, 0, need_bytes);
368
369         mutex_lock(&imgu->streaming_lock);
370         if (imgu->streaming)
371                 imgu_queue_buffers(imgu, false, node->pipe);
372         mutex_unlock(&imgu->streaming_lock);
373
374         dev_dbg(&imgu->pci_dev->dev, "%s for pipe %u node %u", __func__,
375                 node->pipe, node->id);
376 }
377
378 static int imgu_vb2_queue_setup(struct vb2_queue *vq,
379                                 unsigned int *num_buffers,
380                                 unsigned int *num_planes,
381                                 unsigned int sizes[],
382                                 struct device *alloc_devs[])
383 {
384         struct imgu_device *imgu = vb2_get_drv_priv(vq);
385         struct imgu_video_device *node =
386                 container_of(vq, struct imgu_video_device, vbq);
387         const struct v4l2_format *fmt = &node->vdev_fmt;
388         unsigned int size;
389
390         *num_buffers = clamp_val(*num_buffers, 1, VB2_MAX_FRAME);
391         alloc_devs[0] = &imgu->pci_dev->dev;
392
393         if (vq->type == V4L2_BUF_TYPE_META_CAPTURE ||
394             vq->type == V4L2_BUF_TYPE_META_OUTPUT)
395                 size = fmt->fmt.meta.buffersize;
396         else
397                 size = fmt->fmt.pix_mp.plane_fmt[0].sizeimage;
398
399         if (*num_planes) {
400                 if (sizes[0] < size)
401                         return -EINVAL;
402                 size = sizes[0];
403         }
404
405         *num_planes = 1;
406         sizes[0] = size;
407
408         /* Initialize buffer queue */
409         INIT_LIST_HEAD(&node->buffers);
410
411         return 0;
412 }
413
414 /* Check if all enabled video nodes are streaming, exception ignored */
415 static bool imgu_all_nodes_streaming(struct imgu_device *imgu,
416                                      struct imgu_video_device *except)
417 {
418         unsigned int i, pipe, p;
419         struct imgu_video_device *node;
420         struct device *dev = &imgu->pci_dev->dev;
421
422         pipe = except->pipe;
423         if (!test_bit(pipe, imgu->css.enabled_pipes)) {
424                 dev_warn(&imgu->pci_dev->dev,
425                          "pipe %u link is not ready yet", pipe);
426                 return false;
427         }
428
429         for_each_set_bit(p, imgu->css.enabled_pipes, IMGU_MAX_PIPE_NUM) {
430                 for (i = 0; i < IMGU_NODE_NUM; i++) {
431                         node = &imgu->imgu_pipe[p].nodes[i];
432                         dev_dbg(dev, "%s pipe %u queue %u name %s enabled = %u",
433                                 __func__, p, i, node->name, node->enabled);
434                         if (node == except)
435                                 continue;
436                         if (node->enabled && !vb2_start_streaming_called(&node->vbq))
437                                 return false;
438                 }
439         }
440
441         return true;
442 }
443
444 static void imgu_return_all_buffers(struct imgu_device *imgu,
445                                     struct imgu_video_device *node,
446                                     enum vb2_buffer_state state)
447 {
448         struct imgu_vb2_buffer *b, *b0;
449
450         /* Return all buffers */
451         mutex_lock(&imgu->lock);
452         list_for_each_entry_safe(b, b0, &node->buffers, list) {
453                 list_del(&b->list);
454                 vb2_buffer_done(&b->vbb.vb2_buf, state);
455         }
456         mutex_unlock(&imgu->lock);
457 }
458
459 static int imgu_vb2_start_streaming(struct vb2_queue *vq, unsigned int count)
460 {
461         struct imgu_media_pipe *imgu_pipe;
462         struct imgu_device *imgu = vb2_get_drv_priv(vq);
463         struct device *dev = &imgu->pci_dev->dev;
464         struct imgu_video_device *node =
465                 container_of(vq, struct imgu_video_device, vbq);
466         int r;
467         unsigned int pipe;
468
469         dev_dbg(dev, "%s node name %s pipe %u id %u", __func__,
470                 node->name, node->pipe, node->id);
471
472         mutex_lock(&imgu->streaming_lock);
473         if (imgu->streaming) {
474                 r = -EBUSY;
475                 mutex_unlock(&imgu->streaming_lock);
476                 goto fail_return_bufs;
477         }
478         mutex_unlock(&imgu->streaming_lock);
479
480         if (!node->enabled) {
481                 dev_err(dev, "IMGU node is not enabled");
482                 r = -EINVAL;
483                 goto fail_return_bufs;
484         }
485
486         pipe = node->pipe;
487         imgu_pipe = &imgu->imgu_pipe[pipe];
488         r = media_pipeline_start(&node->vdev.entity, &imgu_pipe->pipeline);
489         if (r < 0)
490                 goto fail_return_bufs;
491
492         if (!imgu_all_nodes_streaming(imgu, node))
493                 return 0;
494
495         for_each_set_bit(pipe, imgu->css.enabled_pipes, IMGU_MAX_PIPE_NUM) {
496                 r = v4l2_subdev_call(&imgu->imgu_pipe[pipe].imgu_sd.subdev,
497                                      video, s_stream, 1);
498                 if (r < 0)
499                         goto fail_stop_pipeline;
500         }
501
502         /* Start streaming of the whole pipeline now */
503         dev_dbg(dev, "IMGU streaming is ready to start");
504         mutex_lock(&imgu->streaming_lock);
505         r = imgu_s_stream(imgu, true);
506         if (!r)
507                 imgu->streaming = true;
508         mutex_unlock(&imgu->streaming_lock);
509
510         return 0;
511
512 fail_stop_pipeline:
513         media_pipeline_stop(&node->vdev.entity);
514 fail_return_bufs:
515         imgu_return_all_buffers(imgu, node, VB2_BUF_STATE_QUEUED);
516
517         return r;
518 }
519
520 static void imgu_vb2_stop_streaming(struct vb2_queue *vq)
521 {
522         struct imgu_media_pipe *imgu_pipe;
523         struct imgu_device *imgu = vb2_get_drv_priv(vq);
524         struct device *dev = &imgu->pci_dev->dev;
525         struct imgu_video_device *node =
526                 container_of(vq, struct imgu_video_device, vbq);
527         int r;
528         unsigned int pipe;
529
530         WARN_ON(!node->enabled);
531
532         pipe = node->pipe;
533         dev_dbg(dev, "Try to stream off node [%u][%u]", pipe, node->id);
534         imgu_pipe = &imgu->imgu_pipe[pipe];
535         r = v4l2_subdev_call(&imgu_pipe->imgu_sd.subdev, video, s_stream, 0);
536         if (r)
537                 dev_err(&imgu->pci_dev->dev,
538                         "failed to stop subdev streaming\n");
539
540         mutex_lock(&imgu->streaming_lock);
541         /* Was this the first node with streaming disabled? */
542         if (imgu->streaming && imgu_all_nodes_streaming(imgu, node)) {
543                 /* Yes, really stop streaming now */
544                 dev_dbg(dev, "IMGU streaming is ready to stop");
545                 r = imgu_s_stream(imgu, false);
546                 if (!r)
547                         imgu->streaming = false;
548         }
549
550         imgu_return_all_buffers(imgu, node, VB2_BUF_STATE_ERROR);
551         mutex_unlock(&imgu->streaming_lock);
552
553         media_pipeline_stop(&node->vdev.entity);
554 }
555
556 /******************** v4l2_ioctl_ops ********************/
557
558 #define VID_CAPTURE     0
559 #define VID_OUTPUT      1
560 #define DEF_VID_CAPTURE 0
561 #define DEF_VID_OUTPUT  1
562
563 struct imgu_fmt {
564         u32     fourcc;
565         u16     type; /* VID_CAPTURE or VID_OUTPUT not both */
566 };
567
568 /* format descriptions for capture and preview */
569 static const struct imgu_fmt formats[] = {
570         { V4L2_PIX_FMT_NV12, VID_CAPTURE },
571         { V4L2_PIX_FMT_IPU3_SGRBG10, VID_OUTPUT },
572         { V4L2_PIX_FMT_IPU3_SBGGR10, VID_OUTPUT },
573         { V4L2_PIX_FMT_IPU3_SGBRG10, VID_OUTPUT },
574         { V4L2_PIX_FMT_IPU3_SRGGB10, VID_OUTPUT },
575 };
576
577 /* Find the first matched format, return default if not found */
578 static const struct imgu_fmt *find_format(struct v4l2_format *f, u32 type)
579 {
580         unsigned int i;
581
582         for (i = 0; i < ARRAY_SIZE(formats); i++) {
583                 if (formats[i].fourcc == f->fmt.pix_mp.pixelformat &&
584                     formats[i].type == type)
585                         return &formats[i];
586         }
587
588         return type == VID_CAPTURE ? &formats[DEF_VID_CAPTURE] :
589                                      &formats[DEF_VID_OUTPUT];
590 }
591
592 static int imgu_vidioc_querycap(struct file *file, void *fh,
593                                 struct v4l2_capability *cap)
594 {
595         struct imgu_video_device *node = file_to_intel_imgu_node(file);
596
597         strscpy(cap->driver, IMGU_NAME, sizeof(cap->driver));
598         strscpy(cap->card, IMGU_NAME, sizeof(cap->card));
599         snprintf(cap->bus_info, sizeof(cap->bus_info), "PCI:%s", node->name);
600
601         return 0;
602 }
603
604 static int enum_fmts(struct v4l2_fmtdesc *f, u32 type)
605 {
606         unsigned int i, j;
607
608         for (i = j = 0; i < ARRAY_SIZE(formats); ++i) {
609                 if (formats[i].type == type) {
610                         if (j == f->index)
611                                 break;
612                         ++j;
613                 }
614         }
615
616         if (i < ARRAY_SIZE(formats)) {
617                 f->pixelformat = formats[i].fourcc;
618                 return 0;
619         }
620
621         return -EINVAL;
622 }
623
624 static int vidioc_enum_fmt_vid_cap(struct file *file, void *priv,
625                                    struct v4l2_fmtdesc *f)
626 {
627         if (f->type != V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE)
628                 return -EINVAL;
629
630         return enum_fmts(f, VID_CAPTURE);
631 }
632
633 static int vidioc_enum_fmt_vid_out(struct file *file, void *priv,
634                                    struct v4l2_fmtdesc *f)
635 {
636         if (f->type != V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE)
637                 return -EINVAL;
638
639         return enum_fmts(f, VID_OUTPUT);
640 }
641
642 /* Propagate forward always the format from the CIO2 subdev */
643 static int imgu_vidioc_g_fmt(struct file *file, void *fh,
644                              struct v4l2_format *f)
645 {
646         struct imgu_video_device *node = file_to_intel_imgu_node(file);
647
648         f->fmt = node->vdev_fmt.fmt;
649
650         return 0;
651 }
652
653 /*
654  * Set input/output format. Unless it is just a try, this also resets
655  * selections (ie. effective and BDS resolutions) to defaults.
656  */
657 static int imgu_fmt(struct imgu_device *imgu, unsigned int pipe, int node,
658                     struct v4l2_format *f, bool try)
659 {
660         struct device *dev = &imgu->pci_dev->dev;
661         struct v4l2_pix_format_mplane *fmts[IPU3_CSS_QUEUES] = { NULL };
662         struct v4l2_rect *rects[IPU3_CSS_RECTS] = { NULL };
663         struct v4l2_mbus_framefmt pad_fmt;
664         unsigned int i, css_q;
665         int ret;
666         struct imgu_css_pipe *css_pipe = &imgu->css.pipes[pipe];
667         struct imgu_media_pipe *imgu_pipe = &imgu->imgu_pipe[pipe];
668         struct imgu_v4l2_subdev *imgu_sd = &imgu_pipe->imgu_sd;
669
670         dev_dbg(dev, "set fmt node [%u][%u](try = %u)", pipe, node, try);
671
672         for (i = 0; i < IMGU_NODE_NUM; i++)
673                 dev_dbg(dev, "IMGU pipe %u node %u enabled = %u",
674                         pipe, i, imgu_pipe->nodes[i].enabled);
675
676         if (imgu_pipe->nodes[IMGU_NODE_VF].enabled)
677                 css_pipe->vf_output_en = true;
678
679         if (atomic_read(&imgu_sd->running_mode) == IPU3_RUNNING_MODE_VIDEO)
680                 css_pipe->pipe_id = IPU3_CSS_PIPE_ID_VIDEO;
681         else
682                 css_pipe->pipe_id = IPU3_CSS_PIPE_ID_CAPTURE;
683
684         dev_dbg(dev, "IPU3 pipe %u pipe_id = %u", pipe, css_pipe->pipe_id);
685
686         for (i = 0; i < IPU3_CSS_QUEUES; i++) {
687                 unsigned int inode = imgu_map_node(imgu, i);
688
689                 /* Skip the meta node */
690                 if (inode == IMGU_NODE_STAT_3A || inode == IMGU_NODE_PARAMS)
691                         continue;
692
693                 if (try) {
694                         fmts[i] = kmemdup(&imgu_pipe->nodes[inode].vdev_fmt.fmt.pix_mp,
695                                           sizeof(struct v4l2_pix_format_mplane),
696                                           GFP_KERNEL);
697                         if (!fmts[i]) {
698                                 ret = -ENOMEM;
699                                 goto out;
700                         }
701                 } else {
702                         fmts[i] = &imgu_pipe->nodes[inode].vdev_fmt.fmt.pix_mp;
703                 }
704
705                 /* CSS expects some format on OUT queue */
706                 if (i != IPU3_CSS_QUEUE_OUT &&
707                     !imgu_pipe->nodes[inode].enabled)
708                         fmts[i] = NULL;
709         }
710
711         if (!try) {
712                 /* eff and bds res got by imgu_s_sel */
713                 struct imgu_v4l2_subdev *imgu_sd = &imgu_pipe->imgu_sd;
714
715                 rects[IPU3_CSS_RECT_EFFECTIVE] = &imgu_sd->rect.eff;
716                 rects[IPU3_CSS_RECT_BDS] = &imgu_sd->rect.bds;
717                 rects[IPU3_CSS_RECT_GDC] = &imgu_sd->rect.gdc;
718
719                 /* suppose that pad fmt was set by subdev s_fmt before */
720                 pad_fmt = imgu_pipe->nodes[IMGU_NODE_IN].pad_fmt;
721                 rects[IPU3_CSS_RECT_GDC]->width = pad_fmt.width;
722                 rects[IPU3_CSS_RECT_GDC]->height = pad_fmt.height;
723         }
724
725         /*
726          * imgu doesn't set the node to the value given by user
727          * before we return success from this function, so set it here.
728          */
729         css_q = imgu_node_to_queue(node);
730         if (!fmts[css_q]) {
731                 ret = -EINVAL;
732                 goto out;
733         }
734         *fmts[css_q] = f->fmt.pix_mp;
735
736         if (try)
737                 ret = imgu_css_fmt_try(&imgu->css, fmts, rects, pipe);
738         else
739                 ret = imgu_css_fmt_set(&imgu->css, fmts, rects, pipe);
740
741         /* ret is the binary number in the firmware blob */
742         if (ret < 0)
743                 goto out;
744
745         if (try)
746                 f->fmt.pix_mp = *fmts[css_q];
747         else
748                 f->fmt = imgu_pipe->nodes[node].vdev_fmt.fmt;
749
750 out:
751         if (try) {
752                 for (i = 0; i < IPU3_CSS_QUEUES; i++)
753                         kfree(fmts[i]);
754         }
755
756         return ret;
757 }
758
759 static int imgu_try_fmt(struct file *file, void *fh, struct v4l2_format *f)
760 {
761         struct v4l2_pix_format_mplane *pixm = &f->fmt.pix_mp;
762         const struct imgu_fmt *fmt;
763
764         if (f->type == V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE)
765                 fmt = find_format(f, VID_CAPTURE);
766         else if (f->type == V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE)
767                 fmt = find_format(f, VID_OUTPUT);
768         else
769                 return -EINVAL;
770
771         pixm->pixelformat = fmt->fourcc;
772
773         memset(pixm->plane_fmt[0].reserved, 0,
774                sizeof(pixm->plane_fmt[0].reserved));
775
776         return 0;
777 }
778
779 static int imgu_vidioc_try_fmt(struct file *file, void *fh,
780                                struct v4l2_format *f)
781 {
782         struct imgu_device *imgu = video_drvdata(file);
783         struct device *dev = &imgu->pci_dev->dev;
784         struct imgu_video_device *node = file_to_intel_imgu_node(file);
785         struct v4l2_pix_format_mplane *pix_mp = &f->fmt.pix_mp;
786         int r;
787
788         dev_dbg(dev, "%s [%ux%u] for node %u\n", __func__,
789                 pix_mp->width, pix_mp->height, node->id);
790
791         r = imgu_try_fmt(file, fh, f);
792         if (r)
793                 return r;
794
795         return imgu_fmt(imgu, node->pipe, node->id, f, true);
796 }
797
798 static int imgu_vidioc_s_fmt(struct file *file, void *fh, struct v4l2_format *f)
799 {
800         struct imgu_device *imgu = video_drvdata(file);
801         struct device *dev = &imgu->pci_dev->dev;
802         struct imgu_video_device *node = file_to_intel_imgu_node(file);
803         struct v4l2_pix_format_mplane *pix_mp = &f->fmt.pix_mp;
804         int r;
805
806         dev_dbg(dev, "%s [%ux%u] for node %u\n", __func__,
807                 pix_mp->width, pix_mp->height, node->id);
808
809         r = imgu_try_fmt(file, fh, f);
810         if (r)
811                 return r;
812
813         return imgu_fmt(imgu, node->pipe, node->id, f, false);
814 }
815
816 struct imgu_meta_fmt {
817         __u32 fourcc;
818         char *name;
819 };
820
821 /* From drivers/media/v4l2-core/v4l2-ioctl.c */
822 static const struct imgu_meta_fmt meta_fmts[] = {
823         { V4L2_META_FMT_IPU3_PARAMS, "IPU3 processing parameters" },
824         { V4L2_META_FMT_IPU3_STAT_3A, "IPU3 3A statistics" },
825 };
826
827 static int imgu_meta_enum_format(struct file *file, void *fh,
828                                  struct v4l2_fmtdesc *fmt)
829 {
830         struct imgu_video_device *node = file_to_intel_imgu_node(file);
831         unsigned int i = fmt->type == V4L2_BUF_TYPE_META_OUTPUT ? 0 : 1;
832
833         /* Each node is dedicated to only one meta format */
834         if (fmt->index > 0 || fmt->type != node->vbq.type)
835                 return -EINVAL;
836
837         strscpy(fmt->description, meta_fmts[i].name, sizeof(fmt->description));
838         fmt->pixelformat = meta_fmts[i].fourcc;
839
840         return 0;
841 }
842
843 static int imgu_vidioc_g_meta_fmt(struct file *file, void *fh,
844                                   struct v4l2_format *f)
845 {
846         struct imgu_video_device *node = file_to_intel_imgu_node(file);
847
848         if (f->type != node->vbq.type)
849                 return -EINVAL;
850
851         f->fmt = node->vdev_fmt.fmt;
852
853         return 0;
854 }
855
856 static int imgu_vidioc_enum_input(struct file *file, void *fh,
857                                   struct v4l2_input *input)
858 {
859         if (input->index > 0)
860                 return -EINVAL;
861         strscpy(input->name, "camera", sizeof(input->name));
862         input->type = V4L2_INPUT_TYPE_CAMERA;
863
864         return 0;
865 }
866
867 static int imgu_vidioc_g_input(struct file *file, void *fh, unsigned int *input)
868 {
869         *input = 0;
870
871         return 0;
872 }
873
874 static int imgu_vidioc_s_input(struct file *file, void *fh, unsigned int input)
875 {
876         return input == 0 ? 0 : -EINVAL;
877 }
878
879 static int imgu_vidioc_enum_output(struct file *file, void *fh,
880                                    struct v4l2_output *output)
881 {
882         if (output->index > 0)
883                 return -EINVAL;
884         strscpy(output->name, "camera", sizeof(output->name));
885         output->type = V4L2_INPUT_TYPE_CAMERA;
886
887         return 0;
888 }
889
890 static int imgu_vidioc_g_output(struct file *file, void *fh,
891                                 unsigned int *output)
892 {
893         *output = 0;
894
895         return 0;
896 }
897
898 static int imgu_vidioc_s_output(struct file *file, void *fh,
899                                 unsigned int output)
900 {
901         return output == 0 ? 0 : -EINVAL;
902 }
903
904 /******************** function pointers ********************/
905
906 static struct v4l2_subdev_internal_ops imgu_subdev_internal_ops = {
907         .open = imgu_subdev_open,
908 };
909
910 static const struct v4l2_subdev_core_ops imgu_subdev_core_ops = {
911         .subscribe_event = v4l2_ctrl_subdev_subscribe_event,
912         .unsubscribe_event = v4l2_event_subdev_unsubscribe,
913 };
914
915 static const struct v4l2_subdev_video_ops imgu_subdev_video_ops = {
916         .s_stream = imgu_subdev_s_stream,
917 };
918
919 static const struct v4l2_subdev_pad_ops imgu_subdev_pad_ops = {
920         .link_validate = v4l2_subdev_link_validate_default,
921         .get_fmt = imgu_subdev_get_fmt,
922         .set_fmt = imgu_subdev_set_fmt,
923         .get_selection = imgu_subdev_get_selection,
924         .set_selection = imgu_subdev_set_selection,
925 };
926
927 static const struct v4l2_subdev_ops imgu_subdev_ops = {
928         .core = &imgu_subdev_core_ops,
929         .video = &imgu_subdev_video_ops,
930         .pad = &imgu_subdev_pad_ops,
931 };
932
933 static const struct media_entity_operations imgu_media_ops = {
934         .link_setup = imgu_link_setup,
935         .link_validate = v4l2_subdev_link_validate,
936 };
937
938 /****************** vb2_ops of the Q ********************/
939
940 static const struct vb2_ops imgu_vb2_ops = {
941         .buf_init = imgu_vb2_buf_init,
942         .buf_cleanup = imgu_vb2_buf_cleanup,
943         .buf_queue = imgu_vb2_buf_queue,
944         .queue_setup = imgu_vb2_queue_setup,
945         .start_streaming = imgu_vb2_start_streaming,
946         .stop_streaming = imgu_vb2_stop_streaming,
947         .wait_prepare = vb2_ops_wait_prepare,
948         .wait_finish = vb2_ops_wait_finish,
949 };
950
951 /****************** v4l2_file_operations *****************/
952
953 static const struct v4l2_file_operations imgu_v4l2_fops = {
954         .unlocked_ioctl = video_ioctl2,
955         .open = v4l2_fh_open,
956         .release = vb2_fop_release,
957         .poll = vb2_fop_poll,
958         .mmap = vb2_fop_mmap,
959 };
960
961 /******************** v4l2_ioctl_ops ********************/
962
963 static const struct v4l2_ioctl_ops imgu_v4l2_ioctl_ops = {
964         .vidioc_querycap = imgu_vidioc_querycap,
965
966         .vidioc_enum_fmt_vid_cap = vidioc_enum_fmt_vid_cap,
967         .vidioc_g_fmt_vid_cap_mplane = imgu_vidioc_g_fmt,
968         .vidioc_s_fmt_vid_cap_mplane = imgu_vidioc_s_fmt,
969         .vidioc_try_fmt_vid_cap_mplane = imgu_vidioc_try_fmt,
970
971         .vidioc_enum_fmt_vid_out = vidioc_enum_fmt_vid_out,
972         .vidioc_g_fmt_vid_out_mplane = imgu_vidioc_g_fmt,
973         .vidioc_s_fmt_vid_out_mplane = imgu_vidioc_s_fmt,
974         .vidioc_try_fmt_vid_out_mplane = imgu_vidioc_try_fmt,
975
976         .vidioc_enum_output = imgu_vidioc_enum_output,
977         .vidioc_g_output = imgu_vidioc_g_output,
978         .vidioc_s_output = imgu_vidioc_s_output,
979
980         .vidioc_enum_input = imgu_vidioc_enum_input,
981         .vidioc_g_input = imgu_vidioc_g_input,
982         .vidioc_s_input = imgu_vidioc_s_input,
983
984         /* buffer queue management */
985         .vidioc_reqbufs = vb2_ioctl_reqbufs,
986         .vidioc_create_bufs = vb2_ioctl_create_bufs,
987         .vidioc_prepare_buf = vb2_ioctl_prepare_buf,
988         .vidioc_querybuf = vb2_ioctl_querybuf,
989         .vidioc_qbuf = vb2_ioctl_qbuf,
990         .vidioc_dqbuf = vb2_ioctl_dqbuf,
991         .vidioc_streamon = vb2_ioctl_streamon,
992         .vidioc_streamoff = vb2_ioctl_streamoff,
993         .vidioc_expbuf = vb2_ioctl_expbuf,
994 };
995
996 static const struct v4l2_ioctl_ops imgu_v4l2_meta_ioctl_ops = {
997         .vidioc_querycap = imgu_vidioc_querycap,
998
999         /* meta capture */
1000         .vidioc_enum_fmt_meta_cap = imgu_meta_enum_format,
1001         .vidioc_g_fmt_meta_cap = imgu_vidioc_g_meta_fmt,
1002         .vidioc_s_fmt_meta_cap = imgu_vidioc_g_meta_fmt,
1003         .vidioc_try_fmt_meta_cap = imgu_vidioc_g_meta_fmt,
1004
1005         /* meta output */
1006         .vidioc_enum_fmt_meta_out = imgu_meta_enum_format,
1007         .vidioc_g_fmt_meta_out = imgu_vidioc_g_meta_fmt,
1008         .vidioc_s_fmt_meta_out = imgu_vidioc_g_meta_fmt,
1009         .vidioc_try_fmt_meta_out = imgu_vidioc_g_meta_fmt,
1010
1011         .vidioc_reqbufs = vb2_ioctl_reqbufs,
1012         .vidioc_create_bufs = vb2_ioctl_create_bufs,
1013         .vidioc_prepare_buf = vb2_ioctl_prepare_buf,
1014         .vidioc_querybuf = vb2_ioctl_querybuf,
1015         .vidioc_qbuf = vb2_ioctl_qbuf,
1016         .vidioc_dqbuf = vb2_ioctl_dqbuf,
1017         .vidioc_streamon = vb2_ioctl_streamon,
1018         .vidioc_streamoff = vb2_ioctl_streamoff,
1019         .vidioc_expbuf = vb2_ioctl_expbuf,
1020 };
1021
1022 static int imgu_sd_s_ctrl(struct v4l2_ctrl *ctrl)
1023 {
1024         struct imgu_v4l2_subdev *imgu_sd =
1025                 container_of(ctrl->handler, struct imgu_v4l2_subdev, ctrl_handler);
1026         struct imgu_device *imgu = v4l2_get_subdevdata(&imgu_sd->subdev);
1027         struct device *dev = &imgu->pci_dev->dev;
1028
1029         dev_dbg(dev, "set val %d to ctrl 0x%8x for subdev %u",
1030                 ctrl->val, ctrl->id, imgu_sd->pipe);
1031
1032         switch (ctrl->id) {
1033         case V4L2_CID_INTEL_IPU3_MODE:
1034                 atomic_set(&imgu_sd->running_mode, ctrl->val);
1035                 return 0;
1036         default:
1037                 return -EINVAL;
1038         }
1039 }
1040
1041 static const struct v4l2_ctrl_ops imgu_subdev_ctrl_ops = {
1042         .s_ctrl = imgu_sd_s_ctrl,
1043 };
1044
1045 static const char * const imgu_ctrl_mode_strings[] = {
1046         "Video mode",
1047         "Still mode",
1048 };
1049
1050 static const struct v4l2_ctrl_config imgu_subdev_ctrl_mode = {
1051         .ops = &imgu_subdev_ctrl_ops,
1052         .id = V4L2_CID_INTEL_IPU3_MODE,
1053         .name = "IPU3 Pipe Mode",
1054         .type = V4L2_CTRL_TYPE_MENU,
1055         .max = ARRAY_SIZE(imgu_ctrl_mode_strings) - 1,
1056         .def = IPU3_RUNNING_MODE_VIDEO,
1057         .qmenu = imgu_ctrl_mode_strings,
1058 };
1059
1060 /******************** Framework registration ********************/
1061
1062 /* helper function to config node's video properties */
1063 static void imgu_node_to_v4l2(u32 node, struct video_device *vdev,
1064                               struct v4l2_format *f)
1065 {
1066         u32 cap;
1067
1068         /* Should not happen */
1069         WARN_ON(node >= IMGU_NODE_NUM);
1070
1071         switch (node) {
1072         case IMGU_NODE_IN:
1073                 cap = V4L2_CAP_VIDEO_OUTPUT_MPLANE;
1074                 f->type = V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE;
1075                 vdev->ioctl_ops = &imgu_v4l2_ioctl_ops;
1076                 break;
1077         case IMGU_NODE_PARAMS:
1078                 cap = V4L2_CAP_META_OUTPUT;
1079                 f->type = V4L2_BUF_TYPE_META_OUTPUT;
1080                 f->fmt.meta.dataformat = V4L2_META_FMT_IPU3_PARAMS;
1081                 vdev->ioctl_ops = &imgu_v4l2_meta_ioctl_ops;
1082                 imgu_css_meta_fmt_set(&f->fmt.meta);
1083                 break;
1084         case IMGU_NODE_STAT_3A:
1085                 cap = V4L2_CAP_META_CAPTURE;
1086                 f->type = V4L2_BUF_TYPE_META_CAPTURE;
1087                 f->fmt.meta.dataformat = V4L2_META_FMT_IPU3_STAT_3A;
1088                 vdev->ioctl_ops = &imgu_v4l2_meta_ioctl_ops;
1089                 imgu_css_meta_fmt_set(&f->fmt.meta);
1090                 break;
1091         default:
1092                 cap = V4L2_CAP_VIDEO_CAPTURE_MPLANE;
1093                 f->type = V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE;
1094                 vdev->ioctl_ops = &imgu_v4l2_ioctl_ops;
1095         }
1096
1097         vdev->device_caps = V4L2_CAP_STREAMING | cap;
1098 }
1099
1100 static int imgu_v4l2_subdev_register(struct imgu_device *imgu,
1101                                      struct imgu_v4l2_subdev *imgu_sd,
1102                                      unsigned int pipe)
1103 {
1104         int i, r;
1105         struct v4l2_ctrl_handler *hdl = &imgu_sd->ctrl_handler;
1106         struct imgu_media_pipe *imgu_pipe = &imgu->imgu_pipe[pipe];
1107
1108         /* Initialize subdev media entity */
1109         r = media_entity_pads_init(&imgu_sd->subdev.entity, IMGU_NODE_NUM,
1110                                    imgu_sd->subdev_pads);
1111         if (r) {
1112                 dev_err(&imgu->pci_dev->dev,
1113                         "failed initialize subdev media entity (%d)\n", r);
1114                 return r;
1115         }
1116         imgu_sd->subdev.entity.ops = &imgu_media_ops;
1117         for (i = 0; i < IMGU_NODE_NUM; i++) {
1118                 imgu_sd->subdev_pads[i].flags = imgu_pipe->nodes[i].output ?
1119                         MEDIA_PAD_FL_SINK : MEDIA_PAD_FL_SOURCE;
1120         }
1121
1122         /* Initialize subdev */
1123         v4l2_subdev_init(&imgu_sd->subdev, &imgu_subdev_ops);
1124         imgu_sd->subdev.entity.function = MEDIA_ENT_F_PROC_VIDEO_STATISTICS;
1125         imgu_sd->subdev.internal_ops = &imgu_subdev_internal_ops;
1126         imgu_sd->subdev.flags = V4L2_SUBDEV_FL_HAS_DEVNODE |
1127                                 V4L2_SUBDEV_FL_HAS_EVENTS;
1128         snprintf(imgu_sd->subdev.name, sizeof(imgu_sd->subdev.name),
1129                  "%s %u", IMGU_NAME, pipe);
1130         v4l2_set_subdevdata(&imgu_sd->subdev, imgu);
1131         atomic_set(&imgu_sd->running_mode, IPU3_RUNNING_MODE_VIDEO);
1132         v4l2_ctrl_handler_init(hdl, 1);
1133         imgu_sd->subdev.ctrl_handler = hdl;
1134         imgu_sd->ctrl = v4l2_ctrl_new_custom(hdl, &imgu_subdev_ctrl_mode, NULL);
1135         if (hdl->error) {
1136                 r = hdl->error;
1137                 dev_err(&imgu->pci_dev->dev,
1138                         "failed to create subdev v4l2 ctrl with err %d", r);
1139                 goto fail_subdev;
1140         }
1141         r = v4l2_device_register_subdev(&imgu->v4l2_dev, &imgu_sd->subdev);
1142         if (r) {
1143                 dev_err(&imgu->pci_dev->dev,
1144                         "failed initialize subdev (%d)\n", r);
1145                 goto fail_subdev;
1146         }
1147
1148         imgu_sd->pipe = pipe;
1149         return 0;
1150
1151 fail_subdev:
1152         v4l2_ctrl_handler_free(imgu_sd->subdev.ctrl_handler);
1153         media_entity_cleanup(&imgu_sd->subdev.entity);
1154
1155         return r;
1156 }
1157
1158 static int imgu_v4l2_node_setup(struct imgu_device *imgu, unsigned int pipe,
1159                                 int node_num)
1160 {
1161         int r;
1162         u32 flags;
1163         struct v4l2_mbus_framefmt def_bus_fmt = { 0 };
1164         struct v4l2_pix_format_mplane def_pix_fmt = { 0 };
1165         struct device *dev = &imgu->pci_dev->dev;
1166         struct imgu_media_pipe *imgu_pipe = &imgu->imgu_pipe[pipe];
1167         struct v4l2_subdev *sd = &imgu_pipe->imgu_sd.subdev;
1168         struct imgu_video_device *node = &imgu_pipe->nodes[node_num];
1169         struct video_device *vdev = &node->vdev;
1170         struct vb2_queue *vbq = &node->vbq;
1171
1172         /* Initialize formats to default values */
1173         def_bus_fmt.width = 1920;
1174         def_bus_fmt.height = 1080;
1175         def_bus_fmt.code = MEDIA_BUS_FMT_FIXED;
1176         def_bus_fmt.field = V4L2_FIELD_NONE;
1177         def_bus_fmt.colorspace = V4L2_COLORSPACE_RAW;
1178         def_bus_fmt.ycbcr_enc = V4L2_YCBCR_ENC_DEFAULT;
1179         def_bus_fmt.quantization = V4L2_QUANTIZATION_DEFAULT;
1180         def_bus_fmt.xfer_func = V4L2_XFER_FUNC_DEFAULT;
1181
1182         def_pix_fmt.width = def_bus_fmt.width;
1183         def_pix_fmt.height = def_bus_fmt.height;
1184         def_pix_fmt.field = def_bus_fmt.field;
1185         def_pix_fmt.num_planes = 1;
1186         def_pix_fmt.plane_fmt[0].bytesperline = def_pix_fmt.width * 2;
1187         def_pix_fmt.plane_fmt[0].sizeimage =
1188                 def_pix_fmt.height * def_pix_fmt.plane_fmt[0].bytesperline;
1189         def_pix_fmt.flags = 0;
1190         def_pix_fmt.colorspace = def_bus_fmt.colorspace;
1191         def_pix_fmt.ycbcr_enc = def_bus_fmt.ycbcr_enc;
1192         def_pix_fmt.quantization = def_bus_fmt.quantization;
1193         def_pix_fmt.xfer_func = def_bus_fmt.xfer_func;
1194
1195         /* Initialize miscellaneous variables */
1196         mutex_init(&node->lock);
1197         INIT_LIST_HEAD(&node->buffers);
1198
1199         /* Initialize formats to default values */
1200         node->pad_fmt = def_bus_fmt;
1201         node->id = node_num;
1202         node->pipe = pipe;
1203         imgu_node_to_v4l2(node_num, vdev, &node->vdev_fmt);
1204         if (node->vdev_fmt.type ==
1205             V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE ||
1206             node->vdev_fmt.type ==
1207             V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE) {
1208                 def_pix_fmt.pixelformat = node->output ?
1209                         V4L2_PIX_FMT_IPU3_SGRBG10 :
1210                         V4L2_PIX_FMT_NV12;
1211                 node->vdev_fmt.fmt.pix_mp = def_pix_fmt;
1212         }
1213
1214         /* Initialize media entities */
1215         r = media_entity_pads_init(&vdev->entity, 1, &node->vdev_pad);
1216         if (r) {
1217                 dev_err(dev, "failed initialize media entity (%d)\n", r);
1218                 mutex_destroy(&node->lock);
1219                 return r;
1220         }
1221         node->vdev_pad.flags = node->output ?
1222                 MEDIA_PAD_FL_SOURCE : MEDIA_PAD_FL_SINK;
1223         vdev->entity.ops = NULL;
1224
1225         /* Initialize vbq */
1226         vbq->type = node->vdev_fmt.type;
1227         vbq->io_modes = VB2_USERPTR | VB2_MMAP | VB2_DMABUF;
1228         vbq->ops = &imgu_vb2_ops;
1229         vbq->mem_ops = &vb2_dma_sg_memops;
1230         if (imgu->buf_struct_size <= 0)
1231                 imgu->buf_struct_size =
1232                         sizeof(struct imgu_vb2_buffer);
1233         vbq->buf_struct_size = imgu->buf_struct_size;
1234         vbq->timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_MONOTONIC;
1235         /* can streamon w/o buffers */
1236         vbq->min_buffers_needed = 0;
1237         vbq->drv_priv = imgu;
1238         vbq->lock = &node->lock;
1239         r = vb2_queue_init(vbq);
1240         if (r) {
1241                 dev_err(dev, "failed to initialize video queue (%d)", r);
1242                 media_entity_cleanup(&vdev->entity);
1243                 return r;
1244         }
1245
1246         /* Initialize vdev */
1247         snprintf(vdev->name, sizeof(vdev->name), "%s %u %s",
1248                  IMGU_NAME, pipe, node->name);
1249         vdev->release = video_device_release_empty;
1250         vdev->fops = &imgu_v4l2_fops;
1251         vdev->lock = &node->lock;
1252         vdev->v4l2_dev = &imgu->v4l2_dev;
1253         vdev->queue = &node->vbq;
1254         vdev->vfl_dir = node->output ? VFL_DIR_TX : VFL_DIR_RX;
1255         video_set_drvdata(vdev, imgu);
1256         r = video_register_device(vdev, VFL_TYPE_VIDEO, -1);
1257         if (r) {
1258                 dev_err(dev, "failed to register video device (%d)", r);
1259                 media_entity_cleanup(&vdev->entity);
1260                 return r;
1261         }
1262
1263         /* Create link between video node and the subdev pad */
1264         flags = 0;
1265         if (node->enabled)
1266                 flags |= MEDIA_LNK_FL_ENABLED;
1267         if (node->output) {
1268                 r = media_create_pad_link(&vdev->entity, 0, &sd->entity,
1269                                           node_num, flags);
1270         } else {
1271                 if (node->id == IMGU_NODE_OUT) {
1272                         flags |= MEDIA_LNK_FL_ENABLED | MEDIA_LNK_FL_IMMUTABLE;
1273                         node->enabled = true;
1274                 }
1275
1276                 r = media_create_pad_link(&sd->entity, node_num, &vdev->entity,
1277                                           0, flags);
1278         }
1279         if (r) {
1280                 dev_err(dev, "failed to create pad link (%d)", r);
1281                 video_unregister_device(vdev);
1282                 return r;
1283         }
1284
1285         return 0;
1286 }
1287
1288 static void imgu_v4l2_nodes_cleanup_pipe(struct imgu_device *imgu,
1289                                          unsigned int pipe, int node)
1290 {
1291         int i;
1292         struct imgu_media_pipe *imgu_pipe = &imgu->imgu_pipe[pipe];
1293
1294         for (i = 0; i < node; i++) {
1295                 video_unregister_device(&imgu_pipe->nodes[i].vdev);
1296                 media_entity_cleanup(&imgu_pipe->nodes[i].vdev.entity);
1297                 mutex_destroy(&imgu_pipe->nodes[i].lock);
1298         }
1299 }
1300
1301 static int imgu_v4l2_nodes_setup_pipe(struct imgu_device *imgu, int pipe)
1302 {
1303         int i;
1304
1305         for (i = 0; i < IMGU_NODE_NUM; i++) {
1306                 int r = imgu_v4l2_node_setup(imgu, pipe, i);
1307
1308                 if (r) {
1309                         imgu_v4l2_nodes_cleanup_pipe(imgu, pipe, i);
1310                         return r;
1311                 }
1312         }
1313         return 0;
1314 }
1315
1316 static void imgu_v4l2_subdev_cleanup(struct imgu_device *imgu, unsigned int i)
1317 {
1318         struct imgu_media_pipe *imgu_pipe = &imgu->imgu_pipe[i];
1319
1320         v4l2_device_unregister_subdev(&imgu_pipe->imgu_sd.subdev);
1321         v4l2_ctrl_handler_free(imgu_pipe->imgu_sd.subdev.ctrl_handler);
1322         media_entity_cleanup(&imgu_pipe->imgu_sd.subdev.entity);
1323 }
1324
1325 static void imgu_v4l2_cleanup_pipes(struct imgu_device *imgu, unsigned int pipe)
1326 {
1327         int i;
1328
1329         for (i = 0; i < pipe; i++) {
1330                 imgu_v4l2_nodes_cleanup_pipe(imgu, i, IMGU_NODE_NUM);
1331                 imgu_v4l2_subdev_cleanup(imgu, i);
1332         }
1333 }
1334
1335 static int imgu_v4l2_register_pipes(struct imgu_device *imgu)
1336 {
1337         struct imgu_media_pipe *imgu_pipe;
1338         int i, r;
1339
1340         for (i = 0; i < IMGU_MAX_PIPE_NUM; i++) {
1341                 imgu_pipe = &imgu->imgu_pipe[i];
1342                 r = imgu_v4l2_subdev_register(imgu, &imgu_pipe->imgu_sd, i);
1343                 if (r) {
1344                         dev_err(&imgu->pci_dev->dev,
1345                                 "failed to register subdev%u ret (%d)\n", i, r);
1346                         goto pipes_cleanup;
1347                 }
1348                 r = imgu_v4l2_nodes_setup_pipe(imgu, i);
1349                 if (r) {
1350                         imgu_v4l2_subdev_cleanup(imgu, i);
1351                         goto pipes_cleanup;
1352                 }
1353         }
1354
1355         return 0;
1356
1357 pipes_cleanup:
1358         imgu_v4l2_cleanup_pipes(imgu, i);
1359         return r;
1360 }
1361
1362 int imgu_v4l2_register(struct imgu_device *imgu)
1363 {
1364         int r;
1365
1366         /* Initialize miscellaneous variables */
1367         imgu->streaming = false;
1368
1369         /* Set up media device */
1370         media_device_pci_init(&imgu->media_dev, imgu->pci_dev, IMGU_NAME);
1371
1372         /* Set up v4l2 device */
1373         imgu->v4l2_dev.mdev = &imgu->media_dev;
1374         imgu->v4l2_dev.ctrl_handler = NULL;
1375         r = v4l2_device_register(&imgu->pci_dev->dev, &imgu->v4l2_dev);
1376         if (r) {
1377                 dev_err(&imgu->pci_dev->dev,
1378                         "failed to register V4L2 device (%d)\n", r);
1379                 goto fail_v4l2_dev;
1380         }
1381
1382         r = imgu_v4l2_register_pipes(imgu);
1383         if (r) {
1384                 dev_err(&imgu->pci_dev->dev,
1385                         "failed to register pipes (%d)\n", r);
1386                 goto fail_v4l2_pipes;
1387         }
1388
1389         r = v4l2_device_register_subdev_nodes(&imgu->v4l2_dev);
1390         if (r) {
1391                 dev_err(&imgu->pci_dev->dev,
1392                         "failed to register subdevs (%d)\n", r);
1393                 goto fail_subdevs;
1394         }
1395
1396         r = media_device_register(&imgu->media_dev);
1397         if (r) {
1398                 dev_err(&imgu->pci_dev->dev,
1399                         "failed to register media device (%d)\n", r);
1400                 goto fail_subdevs;
1401         }
1402
1403         return 0;
1404
1405 fail_subdevs:
1406         imgu_v4l2_cleanup_pipes(imgu, IMGU_MAX_PIPE_NUM);
1407 fail_v4l2_pipes:
1408         v4l2_device_unregister(&imgu->v4l2_dev);
1409 fail_v4l2_dev:
1410         media_device_cleanup(&imgu->media_dev);
1411
1412         return r;
1413 }
1414
1415 int imgu_v4l2_unregister(struct imgu_device *imgu)
1416 {
1417         media_device_unregister(&imgu->media_dev);
1418         imgu_v4l2_cleanup_pipes(imgu, IMGU_MAX_PIPE_NUM);
1419         v4l2_device_unregister(&imgu->v4l2_dev);
1420         media_device_cleanup(&imgu->media_dev);
1421
1422         return 0;
1423 }
1424
1425 void imgu_v4l2_buffer_done(struct vb2_buffer *vb,
1426                            enum vb2_buffer_state state)
1427 {
1428         struct imgu_vb2_buffer *b =
1429                 container_of(vb, struct imgu_vb2_buffer, vbb.vb2_buf);
1430
1431         list_del(&b->list);
1432         vb2_buffer_done(&b->vbb.vb2_buf, state);
1433 }