1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  * vimc-capture.c Virtual Media Controller Driver
4  *
5  * Copyright (C) 2015-2017 Helen Koike <helen.fornazier@gmail.com>
6  */
7 
8 #include <media/v4l2-ioctl.h>
9 #include <media/videobuf2-core.h>
10 #include <media/videobuf2-dma-contig.h>
11 #include <media/videobuf2-vmalloc.h>
12 
13 #include "vimc-common.h"
14 #include "vimc-streamer.h"
15 
16 struct vimc_capture_device {
17 	struct vimc_ent_device ved;
18 	struct video_device vdev;
19 	struct v4l2_pix_format format;
20 	struct vb2_queue queue;
21 	struct list_head buf_list;
22 	/*
23 	 * NOTE: in a real driver, a spin lock must be used to access the
24 	 * queue because the frames are generated from a hardware interruption
25 	 * and the isr is not allowed to sleep.
26 	 * Even if it is not necessary a spinlock in the vimc driver, we
27 	 * use it here as a code reference
28 	 */
29 	spinlock_t qlock;
30 	struct mutex lock;
31 	u32 sequence;
32 	struct vimc_stream stream;
33 	struct media_pad pad;
34 };
35 
36 static const struct v4l2_pix_format fmt_default = {
37 	.width = 640,
38 	.height = 480,
39 	.pixelformat = V4L2_PIX_FMT_RGB24,
40 	.field = V4L2_FIELD_NONE,
41 	.colorspace = V4L2_COLORSPACE_SRGB,
42 };
43 
44 struct vimc_capture_buffer {
45 	/*
46 	 * struct vb2_v4l2_buffer must be the first element
47 	 * the videobuf2 framework will allocate this struct based on
48 	 * buf_struct_size and use the first sizeof(struct vb2_buffer) bytes of
49 	 * memory as a vb2_buffer
50 	 */
51 	struct vb2_v4l2_buffer vb2;
52 	struct list_head list;
53 };
54 
55 static int vimc_capture_querycap(struct file *file, void *priv,
56 			     struct v4l2_capability *cap)
57 {
58 	strscpy(cap->driver, VIMC_PDEV_NAME, sizeof(cap->driver));
59 	strscpy(cap->card, KBUILD_MODNAME, sizeof(cap->card));
60 	snprintf(cap->bus_info, sizeof(cap->bus_info),
61 		 "platform:%s", VIMC_PDEV_NAME);
62 
63 	return 0;
64 }
65 
66 static void vimc_capture_get_format(struct vimc_ent_device *ved,
67 				struct v4l2_pix_format *fmt)
68 {
69 	struct vimc_capture_device *vcapture = container_of(ved, struct vimc_capture_device,
70 						    ved);
71 
72 	*fmt = vcapture->format;
73 }
74 
75 static int vimc_capture_g_fmt_vid_cap(struct file *file, void *priv,
76 				  struct v4l2_format *f)
77 {
78 	struct vimc_capture_device *vcapture = video_drvdata(file);
79 
80 	f->fmt.pix = vcapture->format;
81 
82 	return 0;
83 }
84 
85 static int vimc_capture_try_fmt_vid_cap(struct file *file, void *priv,
86 				    struct v4l2_format *f)
87 {
88 	struct v4l2_pix_format *format = &f->fmt.pix;
89 	const struct vimc_pix_map *vpix;
90 
91 	format->width = clamp_t(u32, format->width, VIMC_FRAME_MIN_WIDTH,
92 				VIMC_FRAME_MAX_WIDTH) & ~1;
93 	format->height = clamp_t(u32, format->height, VIMC_FRAME_MIN_HEIGHT,
94 				 VIMC_FRAME_MAX_HEIGHT) & ~1;
95 
96 	/* Don't accept a pixelformat that is not on the table */
97 	vpix = vimc_pix_map_by_pixelformat(format->pixelformat);
98 	if (!vpix) {
99 		format->pixelformat = fmt_default.pixelformat;
100 		vpix = vimc_pix_map_by_pixelformat(format->pixelformat);
101 	}
102 	/* TODO: Add support for custom bytesperline values */
103 	format->bytesperline = format->width * vpix->bpp;
104 	format->sizeimage = format->bytesperline * format->height;
105 
106 	if (format->field == V4L2_FIELD_ANY)
107 		format->field = fmt_default.field;
108 
109 	vimc_colorimetry_clamp(format);
110 
111 	if (format->colorspace == V4L2_COLORSPACE_DEFAULT)
112 		format->colorspace = fmt_default.colorspace;
113 
114 	return 0;
115 }
116 
117 static int vimc_capture_s_fmt_vid_cap(struct file *file, void *priv,
118 				  struct v4l2_format *f)
119 {
120 	struct vimc_capture_device *vcapture = video_drvdata(file);
121 	int ret;
122 
123 	/* Do not change the format while stream is on */
124 	if (vb2_is_busy(&vcapture->queue))
125 		return -EBUSY;
126 
127 	ret = vimc_capture_try_fmt_vid_cap(file, priv, f);
128 	if (ret)
129 		return ret;
130 
131 	dev_dbg(vcapture->ved.dev, "%s: format update: "
132 		"old:%dx%d (0x%x, %d, %d, %d, %d) "
133 		"new:%dx%d (0x%x, %d, %d, %d, %d)\n", vcapture->vdev.name,
134 		/* old */
135 		vcapture->format.width, vcapture->format.height,
136 		vcapture->format.pixelformat, vcapture->format.colorspace,
137 		vcapture->format.quantization, vcapture->format.xfer_func,
138 		vcapture->format.ycbcr_enc,
139 		/* new */
140 		f->fmt.pix.width, f->fmt.pix.height,
141 		f->fmt.pix.pixelformat,	f->fmt.pix.colorspace,
142 		f->fmt.pix.quantization, f->fmt.pix.xfer_func,
143 		f->fmt.pix.ycbcr_enc);
144 
145 	vcapture->format = f->fmt.pix;
146 
147 	return 0;
148 }
149 
150 static int vimc_capture_enum_fmt_vid_cap(struct file *file, void *priv,
151 				     struct v4l2_fmtdesc *f)
152 {
153 	const struct vimc_pix_map *vpix;
154 
155 	if (f->mbus_code) {
156 		if (f->index > 0)
157 			return -EINVAL;
158 
159 		vpix = vimc_pix_map_by_code(f->mbus_code);
160 	} else {
161 		vpix = vimc_pix_map_by_index(f->index);
162 	}
163 
164 	if (!vpix)
165 		return -EINVAL;
166 
167 	f->pixelformat = vpix->pixelformat;
168 
169 	return 0;
170 }
171 
172 static int vimc_capture_enum_framesizes(struct file *file, void *fh,
173 				    struct v4l2_frmsizeenum *fsize)
174 {
175 	const struct vimc_pix_map *vpix;
176 
177 	if (fsize->index)
178 		return -EINVAL;
179 
180 	/* Only accept code in the pix map table */
181 	vpix = vimc_pix_map_by_code(fsize->pixel_format);
182 	if (!vpix)
183 		return -EINVAL;
184 
185 	fsize->type = V4L2_FRMSIZE_TYPE_CONTINUOUS;
186 	fsize->stepwise.min_width = VIMC_FRAME_MIN_WIDTH;
187 	fsize->stepwise.max_width = VIMC_FRAME_MAX_WIDTH;
188 	fsize->stepwise.min_height = VIMC_FRAME_MIN_HEIGHT;
189 	fsize->stepwise.max_height = VIMC_FRAME_MAX_HEIGHT;
190 	fsize->stepwise.step_width = 1;
191 	fsize->stepwise.step_height = 1;
192 
193 	return 0;
194 }
195 
196 static const struct v4l2_file_operations vimc_capture_fops = {
197 	.owner		= THIS_MODULE,
198 	.open		= v4l2_fh_open,
199 	.release	= vb2_fop_release,
200 	.read           = vb2_fop_read,
201 	.poll		= vb2_fop_poll,
202 	.unlocked_ioctl = video_ioctl2,
203 	.mmap           = vb2_fop_mmap,
204 };
205 
206 static const struct v4l2_ioctl_ops vimc_capture_ioctl_ops = {
207 	.vidioc_querycap = vimc_capture_querycap,
208 
209 	.vidioc_g_fmt_vid_cap = vimc_capture_g_fmt_vid_cap,
210 	.vidioc_s_fmt_vid_cap = vimc_capture_s_fmt_vid_cap,
211 	.vidioc_try_fmt_vid_cap = vimc_capture_try_fmt_vid_cap,
212 	.vidioc_enum_fmt_vid_cap = vimc_capture_enum_fmt_vid_cap,
213 	.vidioc_enum_framesizes = vimc_capture_enum_framesizes,
214 
215 	.vidioc_reqbufs = vb2_ioctl_reqbufs,
216 	.vidioc_create_bufs = vb2_ioctl_create_bufs,
217 	.vidioc_prepare_buf = vb2_ioctl_prepare_buf,
218 	.vidioc_querybuf = vb2_ioctl_querybuf,
219 	.vidioc_qbuf = vb2_ioctl_qbuf,
220 	.vidioc_dqbuf = vb2_ioctl_dqbuf,
221 	.vidioc_expbuf = vb2_ioctl_expbuf,
222 	.vidioc_streamon = vb2_ioctl_streamon,
223 	.vidioc_streamoff = vb2_ioctl_streamoff,
224 };
225 
226 static void vimc_capture_return_all_buffers(struct vimc_capture_device *vcapture,
227 					enum vb2_buffer_state state)
228 {
229 	struct vimc_capture_buffer *vbuf, *node;
230 
231 	spin_lock(&vcapture->qlock);
232 
233 	list_for_each_entry_safe(vbuf, node, &vcapture->buf_list, list) {
234 		list_del(&vbuf->list);
235 		vb2_buffer_done(&vbuf->vb2.vb2_buf, state);
236 	}
237 
238 	spin_unlock(&vcapture->qlock);
239 }
240 
241 static int vimc_capture_start_streaming(struct vb2_queue *vq, unsigned int count)
242 {
243 	struct vimc_capture_device *vcapture = vb2_get_drv_priv(vq);
244 	struct media_entity *entity = &vcapture->vdev.entity;
245 	int ret;
246 
247 	vcapture->sequence = 0;
248 
249 	/* Start the media pipeline */
250 	ret = media_pipeline_start(entity, &vcapture->stream.pipe);
251 	if (ret) {
252 		vimc_capture_return_all_buffers(vcapture, VB2_BUF_STATE_QUEUED);
253 		return ret;
254 	}
255 
256 	ret = vimc_streamer_s_stream(&vcapture->stream, &vcapture->ved, 1);
257 	if (ret) {
258 		media_pipeline_stop(entity);
259 		vimc_capture_return_all_buffers(vcapture, VB2_BUF_STATE_QUEUED);
260 		return ret;
261 	}
262 
263 	return 0;
264 }
265 
266 /*
267  * Stop the stream engine. Any remaining buffers in the stream queue are
268  * dequeued and passed on to the vb2 framework marked as STATE_ERROR.
269  */
270 static void vimc_capture_stop_streaming(struct vb2_queue *vq)
271 {
272 	struct vimc_capture_device *vcapture = vb2_get_drv_priv(vq);
273 
274 	vimc_streamer_s_stream(&vcapture->stream, &vcapture->ved, 0);
275 
276 	/* Stop the media pipeline */
277 	media_pipeline_stop(&vcapture->vdev.entity);
278 
279 	/* Release all active buffers */
280 	vimc_capture_return_all_buffers(vcapture, VB2_BUF_STATE_ERROR);
281 }
282 
283 static void vimc_capture_buf_queue(struct vb2_buffer *vb2_buf)
284 {
285 	struct vimc_capture_device *vcapture = vb2_get_drv_priv(vb2_buf->vb2_queue);
286 	struct vimc_capture_buffer *buf = container_of(vb2_buf,
287 						   struct vimc_capture_buffer,
288 						   vb2.vb2_buf);
289 
290 	spin_lock(&vcapture->qlock);
291 	list_add_tail(&buf->list, &vcapture->buf_list);
292 	spin_unlock(&vcapture->qlock);
293 }
294 
295 static int vimc_capture_queue_setup(struct vb2_queue *vq, unsigned int *nbuffers,
296 				unsigned int *nplanes, unsigned int sizes[],
297 				struct device *alloc_devs[])
298 {
299 	struct vimc_capture_device *vcapture = vb2_get_drv_priv(vq);
300 
301 	if (*nplanes)
302 		return sizes[0] < vcapture->format.sizeimage ? -EINVAL : 0;
303 	/* We don't support multiplanes for now */
304 	*nplanes = 1;
305 	sizes[0] = vcapture->format.sizeimage;
306 
307 	return 0;
308 }
309 
310 static int vimc_capture_buffer_prepare(struct vb2_buffer *vb)
311 {
312 	struct vimc_capture_device *vcapture = vb2_get_drv_priv(vb->vb2_queue);
313 	unsigned long size = vcapture->format.sizeimage;
314 
315 	if (vb2_plane_size(vb, 0) < size) {
316 		dev_err(vcapture->ved.dev, "%s: buffer too small (%lu < %lu)\n",
317 			vcapture->vdev.name, vb2_plane_size(vb, 0), size);
318 		return -EINVAL;
319 	}
320 	return 0;
321 }
322 
323 static const struct vb2_ops vimc_capture_qops = {
324 	.start_streaming	= vimc_capture_start_streaming,
325 	.stop_streaming		= vimc_capture_stop_streaming,
326 	.buf_queue		= vimc_capture_buf_queue,
327 	.queue_setup		= vimc_capture_queue_setup,
328 	.buf_prepare		= vimc_capture_buffer_prepare,
329 	/*
330 	 * Since q->lock is set we can use the standard
331 	 * vb2_ops_wait_prepare/finish helper functions.
332 	 */
333 	.wait_prepare		= vb2_ops_wait_prepare,
334 	.wait_finish		= vb2_ops_wait_finish,
335 };
336 
337 static const struct media_entity_operations vimc_capture_mops = {
338 	.link_validate		= vimc_vdev_link_validate,
339 };
340 
341 static void vimc_capture_release(struct vimc_ent_device *ved)
342 {
343 	struct vimc_capture_device *vcapture =
344 		container_of(ved, struct vimc_capture_device, ved);
345 
346 	media_entity_cleanup(vcapture->ved.ent);
347 	kfree(vcapture);
348 }
349 
350 static void vimc_capture_unregister(struct vimc_ent_device *ved)
351 {
352 	struct vimc_capture_device *vcapture =
353 		container_of(ved, struct vimc_capture_device, ved);
354 
355 	vb2_video_unregister_device(&vcapture->vdev);
356 }
357 
358 static void *vimc_capture_process_frame(struct vimc_ent_device *ved,
359 				    const void *frame)
360 {
361 	struct vimc_capture_device *vcapture = container_of(ved, struct vimc_capture_device,
362 						    ved);
363 	struct vimc_capture_buffer *vimc_buf;
364 	void *vbuf;
365 
366 	spin_lock(&vcapture->qlock);
367 
368 	/* Get the first entry of the list */
369 	vimc_buf = list_first_entry_or_null(&vcapture->buf_list,
370 					    typeof(*vimc_buf), list);
371 	if (!vimc_buf) {
372 		spin_unlock(&vcapture->qlock);
373 		return ERR_PTR(-EAGAIN);
374 	}
375 
376 	/* Remove this entry from the list */
377 	list_del(&vimc_buf->list);
378 
379 	spin_unlock(&vcapture->qlock);
380 
381 	/* Fill the buffer */
382 	vimc_buf->vb2.vb2_buf.timestamp = ktime_get_ns();
383 	vimc_buf->vb2.sequence = vcapture->sequence++;
384 	vimc_buf->vb2.field = vcapture->format.field;
385 
386 	vbuf = vb2_plane_vaddr(&vimc_buf->vb2.vb2_buf, 0);
387 
388 	memcpy(vbuf, frame, vcapture->format.sizeimage);
389 
390 	/* Set it as ready */
391 	vb2_set_plane_payload(&vimc_buf->vb2.vb2_buf, 0,
392 			      vcapture->format.sizeimage);
393 	vb2_buffer_done(&vimc_buf->vb2.vb2_buf, VB2_BUF_STATE_DONE);
394 	return NULL;
395 }
396 
397 static struct vimc_ent_device *vimc_capture_add(struct vimc_device *vimc,
398 					    const char *vcfg_name)
399 {
400 	struct v4l2_device *v4l2_dev = &vimc->v4l2_dev;
401 	const struct vimc_pix_map *vpix;
402 	struct vimc_capture_device *vcapture;
403 	struct video_device *vdev;
404 	struct vb2_queue *q;
405 	int ret;
406 
407 	/* Allocate the vimc_capture_device struct */
408 	vcapture = kzalloc(sizeof(*vcapture), GFP_KERNEL);
409 	if (!vcapture)
410 		return ERR_PTR(-ENOMEM);
411 
412 	/* Initialize the media entity */
413 	vcapture->vdev.entity.name = vcfg_name;
414 	vcapture->vdev.entity.function = MEDIA_ENT_F_IO_V4L;
415 	vcapture->pad.flags = MEDIA_PAD_FL_SINK;
416 	ret = media_entity_pads_init(&vcapture->vdev.entity,
417 				     1, &vcapture->pad);
418 	if (ret)
419 		goto err_free_vcapture;
420 
421 	/* Initialize the lock */
422 	mutex_init(&vcapture->lock);
423 
424 	/* Initialize the vb2 queue */
425 	q = &vcapture->queue;
426 	q->type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
427 	q->io_modes = VB2_MMAP | VB2_DMABUF;
428 	if (vimc_allocator == VIMC_ALLOCATOR_VMALLOC)
429 		q->io_modes |= VB2_USERPTR;
430 	q->drv_priv = vcapture;
431 	q->buf_struct_size = sizeof(struct vimc_capture_buffer);
432 	q->ops = &vimc_capture_qops;
433 	q->mem_ops = vimc_allocator == VIMC_ALLOCATOR_DMA_CONTIG
434 		   ? &vb2_dma_contig_memops : &vb2_vmalloc_memops;
435 	q->timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_MONOTONIC;
436 	q->min_buffers_needed = 2;
437 	q->lock = &vcapture->lock;
438 	q->dev = v4l2_dev->dev;
439 
440 	ret = vb2_queue_init(q);
441 	if (ret) {
442 		dev_err(vimc->mdev.dev, "%s: vb2 queue init failed (err=%d)\n",
443 			vcfg_name, ret);
444 		goto err_clean_m_ent;
445 	}
446 
447 	/* Initialize buffer list and its lock */
448 	INIT_LIST_HEAD(&vcapture->buf_list);
449 	spin_lock_init(&vcapture->qlock);
450 
451 	/* Set default frame format */
452 	vcapture->format = fmt_default;
453 	vpix = vimc_pix_map_by_pixelformat(vcapture->format.pixelformat);
454 	vcapture->format.bytesperline = vcapture->format.width * vpix->bpp;
455 	vcapture->format.sizeimage = vcapture->format.bytesperline *
456 				 vcapture->format.height;
457 
458 	/* Fill the vimc_ent_device struct */
459 	vcapture->ved.ent = &vcapture->vdev.entity;
460 	vcapture->ved.process_frame = vimc_capture_process_frame;
461 	vcapture->ved.vdev_get_format = vimc_capture_get_format;
462 	vcapture->ved.dev = vimc->mdev.dev;
463 
464 	/* Initialize the video_device struct */
465 	vdev = &vcapture->vdev;
466 	vdev->device_caps = V4L2_CAP_VIDEO_CAPTURE | V4L2_CAP_STREAMING
467 			  | V4L2_CAP_IO_MC;
468 	vdev->entity.ops = &vimc_capture_mops;
469 	vdev->release = video_device_release_empty;
470 	vdev->fops = &vimc_capture_fops;
471 	vdev->ioctl_ops = &vimc_capture_ioctl_ops;
472 	vdev->lock = &vcapture->lock;
473 	vdev->queue = q;
474 	vdev->v4l2_dev = v4l2_dev;
475 	vdev->vfl_dir = VFL_DIR_RX;
476 	strscpy(vdev->name, vcfg_name, sizeof(vdev->name));
477 	video_set_drvdata(vdev, &vcapture->ved);
478 
479 	/* Register the video_device with the v4l2 and the media framework */
480 	ret = video_register_device(vdev, VFL_TYPE_VIDEO, -1);
481 	if (ret) {
482 		dev_err(vimc->mdev.dev, "%s: video register failed (err=%d)\n",
483 			vcapture->vdev.name, ret);
484 		goto err_clean_m_ent;
485 	}
486 
487 	return &vcapture->ved;
488 
489 err_clean_m_ent:
490 	media_entity_cleanup(&vcapture->vdev.entity);
491 err_free_vcapture:
492 	kfree(vcapture);
493 
494 	return ERR_PTR(ret);
495 }
496 
497 struct vimc_ent_type vimc_capture_type = {
498 	.add = vimc_capture_add,
499 	.unregister = vimc_capture_unregister,
500 	.release = vimc_capture_release
501 };
502