1// SPDX-License-Identifier: GPL-2.0-only
2/*
3 * vivid-meta-cap.c - meta capture support functions.
4 */
5
6#include <linux/errno.h>
7#include <linux/kernel.h>
8#include <linux/videodev2.h>
9#include <media/v4l2-common.h>
10#include <linux/usb/video.h>
11
12#include "vivid-core.h"
13#include "vivid-kthread-cap.h"
14#include "vivid-meta-cap.h"
15
16static int meta_cap_queue_setup(struct vb2_queue *vq, unsigned int *nbuffers,
17				unsigned int *nplanes, unsigned int sizes[],
18				struct device *alloc_devs[])
19{
20	struct vivid_dev *dev = vb2_get_drv_priv(vq);
21	unsigned int size =  sizeof(struct vivid_uvc_meta_buf);
22
23	if (!vivid_is_webcam(dev))
24		return -EINVAL;
25
26	if (*nplanes) {
27		if (sizes[0] < size)
28			return -EINVAL;
29	} else {
30		sizes[0] = size;
31	}
32
33	*nplanes = 1;
34	return 0;
35}
36
37static int meta_cap_buf_prepare(struct vb2_buffer *vb)
38{
39	struct vivid_dev *dev = vb2_get_drv_priv(vb->vb2_queue);
40	unsigned int size = sizeof(struct vivid_uvc_meta_buf);
41
42	dprintk(dev, 1, "%s\n", __func__);
43
44	if (dev->buf_prepare_error) {
45		/*
46		 * Error injection: test what happens if buf_prepare() returns
47		 * an error.
48		 */
49		dev->buf_prepare_error = false;
50		return -EINVAL;
51	}
52	if (vb2_plane_size(vb, 0) < size) {
53		dprintk(dev, 1, "%s data will not fit into plane (%lu < %u)\n",
54			__func__, vb2_plane_size(vb, 0), size);
55		return -EINVAL;
56	}
57	vb2_set_plane_payload(vb, 0, size);
58
59	return 0;
60}
61
62static void meta_cap_buf_queue(struct vb2_buffer *vb)
63{
64	struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb);
65	struct vivid_dev *dev = vb2_get_drv_priv(vb->vb2_queue);
66	struct vivid_buffer *buf = container_of(vbuf, struct vivid_buffer, vb);
67
68	dprintk(dev, 1, "%s\n", __func__);
69
70	spin_lock(&dev->slock);
71	list_add_tail(&buf->list, &dev->meta_cap_active);
72	spin_unlock(&dev->slock);
73}
74
75static int meta_cap_start_streaming(struct vb2_queue *vq, unsigned int count)
76{
77	struct vivid_dev *dev = vb2_get_drv_priv(vq);
78	int err;
79
80	dprintk(dev, 1, "%s\n", __func__);
81	dev->meta_cap_seq_count = 0;
82	if (dev->start_streaming_error) {
83		dev->start_streaming_error = false;
84		err = -EINVAL;
85	} else {
86		err = vivid_start_generating_vid_cap(dev,
87						     &dev->meta_cap_streaming);
88	}
89	if (err) {
90		struct vivid_buffer *buf, *tmp;
91
92		list_for_each_entry_safe(buf, tmp,
93					 &dev->meta_cap_active, list) {
94			list_del(&buf->list);
95			vb2_buffer_done(&buf->vb.vb2_buf,
96					VB2_BUF_STATE_QUEUED);
97		}
98	}
99	return err;
100}
101
102/* abort streaming and wait for last buffer */
103static void meta_cap_stop_streaming(struct vb2_queue *vq)
104{
105	struct vivid_dev *dev = vb2_get_drv_priv(vq);
106
107	dprintk(dev, 1, "%s\n", __func__);
108	vivid_stop_generating_vid_cap(dev, &dev->meta_cap_streaming);
109}
110
111static void meta_cap_buf_request_complete(struct vb2_buffer *vb)
112{
113	struct vivid_dev *dev = vb2_get_drv_priv(vb->vb2_queue);
114
115	v4l2_ctrl_request_complete(vb->req_obj.req, &dev->ctrl_hdl_meta_cap);
116}
117
118const struct vb2_ops vivid_meta_cap_qops = {
119	.queue_setup		= meta_cap_queue_setup,
120	.buf_prepare		= meta_cap_buf_prepare,
121	.buf_queue		= meta_cap_buf_queue,
122	.start_streaming	= meta_cap_start_streaming,
123	.stop_streaming		= meta_cap_stop_streaming,
124	.buf_request_complete	= meta_cap_buf_request_complete,
125	.wait_prepare		= vb2_ops_wait_prepare,
126	.wait_finish		= vb2_ops_wait_finish,
127};
128
129int vidioc_enum_fmt_meta_cap(struct file *file, void  *priv,
130			     struct v4l2_fmtdesc *f)
131{
132	struct vivid_dev *dev = video_drvdata(file);
133
134	if (!vivid_is_webcam(dev))
135		return -EINVAL;
136
137	if (f->index > 0)
138		return -EINVAL;
139
140	f->type = V4L2_BUF_TYPE_META_CAPTURE;
141	f->pixelformat = V4L2_META_FMT_UVC;
142	return 0;
143}
144
145int vidioc_g_fmt_meta_cap(struct file *file, void *priv,
146			  struct v4l2_format *f)
147{
148	struct vivid_dev *dev = video_drvdata(file);
149	struct v4l2_meta_format *meta = &f->fmt.meta;
150
151	if (!vivid_is_webcam(dev) || !dev->has_meta_cap)
152		return -EINVAL;
153
154	meta->dataformat = V4L2_META_FMT_UVC;
155	meta->buffersize = sizeof(struct vivid_uvc_meta_buf);
156	return 0;
157}
158
159void vivid_meta_cap_fillbuff(struct vivid_dev *dev,
160			     struct vivid_buffer *buf, u64 soe)
161{
162	struct vivid_uvc_meta_buf *meta = vb2_plane_vaddr(&buf->vb.vb2_buf, 0);
163	int buf_off = 0;
164
165	buf->vb.sequence = dev->meta_cap_seq_count;
166	if (dev->field_cap == V4L2_FIELD_ALTERNATE)
167		buf->vb.sequence /= 2;
168	memset(meta, 1, vb2_plane_size(&buf->vb.vb2_buf, 0));
169
170	meta->ns = ktime_get_ns();
171	meta->sof = buf->vb.sequence * 30;
172	meta->length = sizeof(*meta) - offsetof(struct vivid_uvc_meta_buf, length);
173	meta->flags = UVC_STREAM_EOH | UVC_STREAM_EOF;
174
175	if ((buf->vb.sequence % 2) == 0)
176		meta->flags |= UVC_STREAM_FID;
177
178	dprintk(dev, 2, "%s ns:%llu sof:%4d len:%u flags: 0x%02x",
179		__func__, meta->ns, meta->sof, meta->length, meta->flags);
180	if (dev->meta_pts) {
181		meta->flags |= UVC_STREAM_PTS;
182		meta->buf[0] = div_u64(soe, VIVID_META_CLOCK_UNIT);
183		buf_off = 4;
184		dprintk(dev, 2, " pts: %u\n", *(__u32 *)(meta->buf));
185	}
186
187	if (dev->meta_scr) {
188		meta->flags |= UVC_STREAM_SCR;
189		meta->buf[buf_off] = div_u64((soe + dev->cap_frame_eof_offset),
190					     VIVID_META_CLOCK_UNIT);
191
192		meta->buf[buf_off + 4] = (buf->vb.sequence * 30) % 1000;
193		dprintk(dev, 2, " stc: %u, sof counter: %u\n",
194			*(__u32 *)(meta->buf + buf_off),
195			*(__u16 *)(meta->buf + buf_off + 4));
196	}
197	dprintk(dev, 2, "\n");
198}
199