vivid-meta-out.c

  1// SPDX-License-Identifier: GPL-2.0-only
  2/*
  3 * vivid-meta-out.c - meta output 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-out.h"
 14#include "vivid-meta-out.h"
 15
 16static int meta_out_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_meta_out_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_out_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_meta_out_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_out_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_out_active);
 72	spin_unlock(&dev->slock);
 73}
 74
 75static int meta_out_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_out_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_out(dev,
 87						     &dev->meta_out_streaming);
 88	}
 89	if (err) {
 90		struct vivid_buffer *buf, *tmp;
 91
 92		list_for_each_entry_safe(buf, tmp,
 93					 &dev->meta_out_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_out_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_out(dev, &dev->meta_out_streaming);
109}
110
111static void meta_out_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_out);
116}
117
118const struct vb2_ops vivid_meta_out_qops = {
119	.queue_setup            = meta_out_queue_setup,
120	.buf_prepare            = meta_out_buf_prepare,
121	.buf_queue              = meta_out_buf_queue,
122	.start_streaming        = meta_out_start_streaming,
123	.stop_streaming         = meta_out_stop_streaming,
124	.buf_request_complete   = meta_out_buf_request_complete,
125};
126
127int vidioc_enum_fmt_meta_out(struct file *file, void  *priv,
128			     struct v4l2_fmtdesc *f)
129{
130	struct vivid_dev *dev = video_drvdata(file);
131
132	if (!vivid_is_webcam(dev))
133		return -EINVAL;
134
135	if (f->index > 0)
136		return -EINVAL;
137
138	f->type = V4L2_BUF_TYPE_META_OUTPUT;
139	f->pixelformat = V4L2_META_FMT_VIVID;
140	return 0;
141}
142
143int vidioc_g_fmt_meta_out(struct file *file, void *priv,
144			  struct v4l2_format *f)
145{
146	struct vivid_dev *dev = video_drvdata(file);
147	struct v4l2_meta_format *meta = &f->fmt.meta;
148
149	if (!vivid_is_webcam(dev) || !dev->has_meta_out)
150		return -EINVAL;
151
152	meta->dataformat = V4L2_META_FMT_VIVID;
153	meta->buffersize = sizeof(struct vivid_meta_out_buf);
154	return 0;
155}
156
157void vivid_meta_out_process(struct vivid_dev *dev,
158			    struct vivid_buffer *buf)
159{
160	struct vivid_meta_out_buf *meta = vb2_plane_vaddr(&buf->vb.vb2_buf, 0);
161
162	v4l2_ctrl_s_ctrl(dev->brightness, meta->brightness);
163	v4l2_ctrl_s_ctrl(dev->contrast, meta->contrast);
164	v4l2_ctrl_s_ctrl(dev->saturation, meta->saturation);
165	v4l2_ctrl_s_ctrl(dev->hue, meta->hue);
166
167	dprintk(dev, 2, " %s brightness %u contrast %u saturation %u hue %d\n",
168		__func__, meta->brightness, meta->contrast,
169		meta->saturation, meta->hue);
170}