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 | |
16 | static 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(q: 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 | |
37 | static int meta_cap_buf_prepare(struct vb2_buffer *vb) |
38 | { |
39 | struct vivid_dev *dev = vb2_get_drv_priv(q: 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, plane_no: 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, plane_no: 0, size); |
58 | |
59 | return 0; |
60 | } |
61 | |
62 | static 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(q: 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(lock: &dev->slock); |
71 | list_add_tail(new: &buf->list, head: &dev->meta_cap_active); |
72 | spin_unlock(lock: &dev->slock); |
73 | } |
74 | |
75 | static int meta_cap_start_streaming(struct vb2_queue *vq, unsigned int count) |
76 | { |
77 | struct vivid_dev *dev = vb2_get_drv_priv(q: 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 | pstreaming: &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(entry: &buf->list); |
95 | vb2_buffer_done(vb: &buf->vb.vb2_buf, |
96 | state: VB2_BUF_STATE_QUEUED); |
97 | } |
98 | } |
99 | return err; |
100 | } |
101 | |
102 | /* abort streaming and wait for last buffer */ |
103 | static void meta_cap_stop_streaming(struct vb2_queue *vq) |
104 | { |
105 | struct vivid_dev *dev = vb2_get_drv_priv(q: vq); |
106 | |
107 | dprintk(dev, 1, "%s\n" , __func__); |
108 | vivid_stop_generating_vid_cap(dev, pstreaming: &dev->meta_cap_streaming); |
109 | } |
110 | |
111 | static void meta_cap_buf_request_complete(struct vb2_buffer *vb) |
112 | { |
113 | struct vivid_dev *dev = vb2_get_drv_priv(q: vb->vb2_queue); |
114 | |
115 | v4l2_ctrl_request_complete(req: vb->req_obj.req, parent: &dev->ctrl_hdl_meta_cap); |
116 | } |
117 | |
118 | const 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 | |
129 | int 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 | |
145 | int 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 | |
159 | void 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(vb: &buf->vb.vb2_buf, plane_no: 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(dividend: 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(dividend: (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 | |