root/drivers/media/test-drivers/vivid/vivid-vbi-cap.c
// SPDX-License-Identifier: GPL-2.0-only
/*
 * vivid-vbi-cap.c - vbi capture support functions.
 *
 * Copyright 2014 Cisco Systems, Inc. and/or its affiliates. All rights reserved.
 */

#include <linux/errno.h>
#include <linux/kernel.h>
#include <linux/videodev2.h>
#include <media/v4l2-common.h>

#include "vivid-core.h"
#include "vivid-kthread-cap.h"
#include "vivid-vbi-cap.h"
#include "vivid-vbi-gen.h"
#include "vivid-vid-common.h"

static void vivid_sliced_vbi_cap_fill(struct vivid_dev *dev, unsigned seqnr)
{
        struct vivid_vbi_gen_data *vbi_gen = &dev->vbi_gen;
        bool is_60hz = dev->std_cap[dev->input] & V4L2_STD_525_60;

        vivid_vbi_gen_sliced(vbi_gen, is_60hz, seqnr);

        if (!is_60hz) {
                if (vivid_vid_can_loop(dev)) {
                        if (dev->vbi_out_have_wss) {
                                vbi_gen->data[12].data[0] = dev->vbi_out_wss[0];
                                vbi_gen->data[12].data[1] = dev->vbi_out_wss[1];
                        } else {
                                vbi_gen->data[12].id = 0;
                        }
                } else {
                        switch (tpg_g_video_aspect(&dev->tpg)) {
                        case TPG_VIDEO_ASPECT_14X9_CENTRE:
                                vbi_gen->data[12].data[0] = 0x01;
                                break;
                        case TPG_VIDEO_ASPECT_16X9_CENTRE:
                                vbi_gen->data[12].data[0] = 0x0b;
                                break;
                        case TPG_VIDEO_ASPECT_16X9_ANAMORPHIC:
                                vbi_gen->data[12].data[0] = 0x07;
                                break;
                        case TPG_VIDEO_ASPECT_4X3:
                        default:
                                vbi_gen->data[12].data[0] = 0x08;
                                break;
                        }
                }
        } else if (vivid_vid_can_loop(dev) && is_60hz) {
                if (dev->vbi_out_have_cc[0]) {
                        vbi_gen->data[0].data[0] = dev->vbi_out_cc[0][0];
                        vbi_gen->data[0].data[1] = dev->vbi_out_cc[0][1];
                } else {
                        vbi_gen->data[0].id = 0;
                }
                if (dev->vbi_out_have_cc[1]) {
                        vbi_gen->data[1].data[0] = dev->vbi_out_cc[1][0];
                        vbi_gen->data[1].data[1] = dev->vbi_out_cc[1][1];
                } else {
                        vbi_gen->data[1].id = 0;
                }
        }
}

static void vivid_g_fmt_vbi_cap(struct vivid_dev *dev, struct v4l2_vbi_format *vbi)
{
        bool is_60hz = dev->std_cap[dev->input] & V4L2_STD_525_60;

        vbi->sampling_rate = 27000000;
        vbi->offset = 24;
        vbi->samples_per_line = 1440;
        vbi->sample_format = V4L2_PIX_FMT_GREY;
        vbi->start[0] = is_60hz ? V4L2_VBI_ITU_525_F1_START + 9 : V4L2_VBI_ITU_625_F1_START + 5;
        vbi->start[1] = is_60hz ? V4L2_VBI_ITU_525_F2_START + 9 : V4L2_VBI_ITU_625_F2_START + 5;
        vbi->count[0] = vbi->count[1] = is_60hz ? 12 : 18;
        vbi->flags = dev->vbi_cap_interlaced ? V4L2_VBI_INTERLACED : 0;
        vbi->reserved[0] = 0;
        vbi->reserved[1] = 0;
}

void vivid_raw_vbi_cap_process(struct vivid_dev *dev, struct vivid_buffer *buf)
{
        struct v4l2_vbi_format vbi;
        u8 *vbuf = vb2_plane_vaddr(&buf->vb.vb2_buf, 0);

        vivid_g_fmt_vbi_cap(dev, &vbi);
        buf->vb.sequence = dev->vbi_cap_seq_count;
        if (dev->field_cap == V4L2_FIELD_ALTERNATE)
                buf->vb.sequence /= 2;

        vivid_sliced_vbi_cap_fill(dev, buf->vb.sequence);

        memset(vbuf, 0x10, vb2_plane_size(&buf->vb.vb2_buf, 0));

        if (!VIVID_INVALID_SIGNAL(dev->std_signal_mode[dev->input]))
                vivid_vbi_gen_raw(&dev->vbi_gen, &vbi, vbuf);
}


void vivid_sliced_vbi_cap_process(struct vivid_dev *dev,
                        struct vivid_buffer *buf)
{
        struct v4l2_sliced_vbi_data *vbuf =
                        vb2_plane_vaddr(&buf->vb.vb2_buf, 0);

        buf->vb.sequence = dev->vbi_cap_seq_count;
        if (dev->field_cap == V4L2_FIELD_ALTERNATE)
                buf->vb.sequence /= 2;

        vivid_sliced_vbi_cap_fill(dev, buf->vb.sequence);

        memset(vbuf, 0, vb2_plane_size(&buf->vb.vb2_buf, 0));
        if (!VIVID_INVALID_SIGNAL(dev->std_signal_mode[dev->input])) {
                unsigned i;

                for (i = 0; i < 25; i++)
                        vbuf[i] = dev->vbi_gen.data[i];
        }
}

static int vbi_cap_queue_setup(struct vb2_queue *vq,
                       unsigned *nbuffers, unsigned *nplanes,
                       unsigned sizes[], struct device *alloc_devs[])
{
        struct vivid_dev *dev = vb2_get_drv_priv(vq);
        bool is_60hz = dev->std_cap[dev->input] & V4L2_STD_525_60;
        unsigned size = vq->type == V4L2_BUF_TYPE_SLICED_VBI_CAPTURE ?
                36 * sizeof(struct v4l2_sliced_vbi_data) :
                1440 * 2 * (is_60hz ? 12 : 18);

        if (!vivid_is_sdtv_cap(dev))
                return -EINVAL;

        if (*nplanes)
                return sizes[0] < size ? -EINVAL : 0;
        sizes[0] = size;

        *nplanes = 1;
        return 0;
}

static int vbi_cap_buf_prepare(struct vb2_buffer *vb)
{
        struct vivid_dev *dev = vb2_get_drv_priv(vb->vb2_queue);
        bool is_60hz = dev->std_cap[dev->input] & V4L2_STD_525_60;
        unsigned size = vb->vb2_queue->type == V4L2_BUF_TYPE_SLICED_VBI_CAPTURE ?
                36 * sizeof(struct v4l2_sliced_vbi_data) :
                1440 * 2 * (is_60hz ? 12 : 18);

        dprintk(dev, 1, "%s\n", __func__);

        if (dev->buf_prepare_error) {
                /*
                 * Error injection: test what happens if buf_prepare() returns
                 * an error.
                 */
                dev->buf_prepare_error = false;
                return -EINVAL;
        }
        if (vb2_plane_size(vb, 0) < size) {
                dprintk(dev, 1, "%s data will not fit into plane (%lu < %u)\n",
                                __func__, vb2_plane_size(vb, 0), size);
                return -EINVAL;
        }
        vb2_set_plane_payload(vb, 0, size);

        return 0;
}

static void vbi_cap_buf_queue(struct vb2_buffer *vb)
{
        struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb);
        struct vivid_dev *dev = vb2_get_drv_priv(vb->vb2_queue);
        struct vivid_buffer *buf = container_of(vbuf, struct vivid_buffer, vb);

        dprintk(dev, 1, "%s\n", __func__);

        spin_lock(&dev->slock);
        list_add_tail(&buf->list, &dev->vbi_cap_active);
        spin_unlock(&dev->slock);
}

static int vbi_cap_start_streaming(struct vb2_queue *vq, unsigned count)
{
        struct vivid_dev *dev = vb2_get_drv_priv(vq);
        int err;

        dprintk(dev, 1, "%s\n", __func__);
        dev->vbi_cap_seq_count = 0;
        if (dev->start_streaming_error) {
                dev->start_streaming_error = false;
                err = -EINVAL;
        } else {
                err = vivid_start_generating_vid_cap(dev, &dev->vbi_cap_streaming);
        }
        if (err) {
                struct vivid_buffer *buf, *tmp;

                list_for_each_entry_safe(buf, tmp, &dev->vbi_cap_active, list) {
                        list_del(&buf->list);
                        vb2_buffer_done(&buf->vb.vb2_buf,
                                        VB2_BUF_STATE_QUEUED);
                }
        }
        return err;
}

/* abort streaming and wait for last buffer */
static void vbi_cap_stop_streaming(struct vb2_queue *vq)
{
        struct vivid_dev *dev = vb2_get_drv_priv(vq);

        dprintk(dev, 1, "%s\n", __func__);
        vivid_stop_generating_vid_cap(dev, &dev->vbi_cap_streaming);
}

static void vbi_cap_buf_request_complete(struct vb2_buffer *vb)
{
        struct vivid_dev *dev = vb2_get_drv_priv(vb->vb2_queue);

        v4l2_ctrl_request_complete(vb->req_obj.req, &dev->ctrl_hdl_vbi_cap);
}

const struct vb2_ops vivid_vbi_cap_qops = {
        .queue_setup            = vbi_cap_queue_setup,
        .buf_prepare            = vbi_cap_buf_prepare,
        .buf_queue              = vbi_cap_buf_queue,
        .start_streaming        = vbi_cap_start_streaming,
        .stop_streaming         = vbi_cap_stop_streaming,
        .buf_request_complete   = vbi_cap_buf_request_complete,
};

int vidioc_g_fmt_vbi_cap(struct file *file, void *priv,
                                        struct v4l2_format *f)
{
        struct vivid_dev *dev = video_drvdata(file);
        struct v4l2_vbi_format *vbi = &f->fmt.vbi;

        if (!vivid_is_sdtv_cap(dev) || !dev->has_raw_vbi_cap)
                return -EINVAL;

        vivid_g_fmt_vbi_cap(dev, vbi);
        return 0;
}

int vidioc_s_fmt_vbi_cap(struct file *file, void *priv,
                                        struct v4l2_format *f)
{
        struct vivid_dev *dev = video_drvdata(file);
        int ret = vidioc_g_fmt_vbi_cap(file, priv, f);

        if (ret)
                return ret;
        if (f->type != V4L2_BUF_TYPE_VBI_CAPTURE && vb2_is_busy(&dev->vb_vbi_cap_q))
                return -EBUSY;
        return 0;
}

void vivid_fill_service_lines(struct v4l2_sliced_vbi_format *vbi, u32 service_set)
{
        vbi->io_size = sizeof(struct v4l2_sliced_vbi_data) * 36;
        vbi->service_set = service_set;
        memset(vbi->service_lines, 0, sizeof(vbi->service_lines));
        memset(vbi->reserved, 0, sizeof(vbi->reserved));

        if (vbi->service_set == 0)
                return;

        if (vbi->service_set & V4L2_SLICED_CAPTION_525) {
                vbi->service_lines[0][21] = V4L2_SLICED_CAPTION_525;
                vbi->service_lines[1][21] = V4L2_SLICED_CAPTION_525;
        }
        if (vbi->service_set & V4L2_SLICED_WSS_625) {
                unsigned i;

                for (i = 7; i <= 18; i++)
                        vbi->service_lines[0][i] =
                        vbi->service_lines[1][i] = V4L2_SLICED_TELETEXT_B;
                vbi->service_lines[0][23] = V4L2_SLICED_WSS_625;
        }
}

int vidioc_g_fmt_sliced_vbi_cap(struct file *file, void *priv, struct v4l2_format *fmt)
{
        struct vivid_dev *dev = video_drvdata(file);
        struct v4l2_sliced_vbi_format *vbi = &fmt->fmt.sliced;

        if (!vivid_is_sdtv_cap(dev) || !dev->has_sliced_vbi_cap)
                return -EINVAL;

        vivid_fill_service_lines(vbi, dev->service_set_cap);
        return 0;
}

int vidioc_try_fmt_sliced_vbi_cap(struct file *file, void *priv, struct v4l2_format *fmt)
{
        struct vivid_dev *dev = video_drvdata(file);
        struct v4l2_sliced_vbi_format *vbi = &fmt->fmt.sliced;
        bool is_60hz = dev->std_cap[dev->input] & V4L2_STD_525_60;
        u32 service_set = vbi->service_set;

        if (!vivid_is_sdtv_cap(dev) || !dev->has_sliced_vbi_cap)
                return -EINVAL;

        service_set &= is_60hz ? V4L2_SLICED_CAPTION_525 :
                                 V4L2_SLICED_WSS_625 | V4L2_SLICED_TELETEXT_B;
        vivid_fill_service_lines(vbi, service_set);
        return 0;
}

int vidioc_s_fmt_sliced_vbi_cap(struct file *file, void *priv, struct v4l2_format *fmt)
{
        struct vivid_dev *dev = video_drvdata(file);
        struct v4l2_sliced_vbi_format *vbi = &fmt->fmt.sliced;
        int ret = vidioc_try_fmt_sliced_vbi_cap(file, priv, fmt);

        if (ret)
                return ret;
        if (fmt->type != V4L2_BUF_TYPE_SLICED_VBI_CAPTURE && vb2_is_busy(&dev->vb_vbi_cap_q))
                return -EBUSY;
        dev->service_set_cap = vbi->service_set;
        return 0;
}

int vidioc_g_sliced_vbi_cap(struct file *file, void *priv, struct v4l2_sliced_vbi_cap *cap)
{
        struct vivid_dev *dev = video_drvdata(file);
        struct video_device *vdev = video_devdata(file);
        bool is_60hz;

        if (vdev->vfl_dir == VFL_DIR_RX) {
                is_60hz = dev->std_cap[dev->input] & V4L2_STD_525_60;
                if (!vivid_is_sdtv_cap(dev) || !dev->has_sliced_vbi_cap ||
                    cap->type != V4L2_BUF_TYPE_SLICED_VBI_CAPTURE)
                        return -EINVAL;
        } else {
                is_60hz = dev->std_out & V4L2_STD_525_60;
                if (!vivid_is_svid_out(dev) || !dev->has_sliced_vbi_out ||
                    cap->type != V4L2_BUF_TYPE_SLICED_VBI_OUTPUT)
                        return -EINVAL;
        }

        cap->service_set = is_60hz ? V4L2_SLICED_CAPTION_525 :
                                     V4L2_SLICED_WSS_625 | V4L2_SLICED_TELETEXT_B;
        if (is_60hz) {
                cap->service_lines[0][21] = V4L2_SLICED_CAPTION_525;
                cap->service_lines[1][21] = V4L2_SLICED_CAPTION_525;
        } else {
                unsigned i;

                for (i = 7; i <= 18; i++)
                        cap->service_lines[0][i] =
                        cap->service_lines[1][i] = V4L2_SLICED_TELETEXT_B;
                cap->service_lines[0][23] = V4L2_SLICED_WSS_625;
        }
        return 0;
}