// SPDX-License-Identifier: GPL-2.0-only
/*
 * vivid-touch-cap.c - touch support functions.
 */

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

static int touch_cap_queue_setup(struct vb2_queue *vq, unsigned int *nbuffers,
                                 unsigned int *nplanes, unsigned int sizes[],
                                 struct device *alloc_devs[])
{
        struct vivid_dev *dev = vb2_get_drv_priv(vq);
        struct v4l2_pix_format *f = &dev->tch_format;
        unsigned int size = f->sizeimage;

        if (*nplanes) {
                if (sizes[0] < size)
                        return -EINVAL;
        } else {
                sizes[0] = size;
        }

        if (vq->num_buffers + *nbuffers < 2)
                *nbuffers = 2 - vq->num_buffers;

        *nplanes = 1;
        return 0;
}

static int touch_cap_buf_prepare(struct vb2_buffer *vb)
{
        struct vivid_dev *dev = vb2_get_drv_priv(vb->vb2_queue);
        struct v4l2_pix_format *f = &dev->tch_format;
        unsigned int size = f->sizeimage;

        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 touch_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);

        vbuf->field = V4L2_FIELD_NONE;
        spin_lock(&dev->slock);
        list_add_tail(&buf->list, &dev->touch_cap_active);
        spin_unlock(&dev->slock);
}

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

        dev->touch_cap_seq_count = 0;
        if (dev->start_streaming_error) {
                dev->start_streaming_error = false;
                err = -EINVAL;
        } else {
                err = vivid_start_generating_touch_cap(dev);
        }
        if (err) {
                struct vivid_buffer *buf, *tmp;

                list_for_each_entry_safe(buf, tmp,
                                         &dev->touch_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 touch_cap_stop_streaming(struct vb2_queue *vq)
{
        struct vivid_dev *dev = vb2_get_drv_priv(vq);

        vivid_stop_generating_touch_cap(dev);
}

static void touch_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_touch_cap);
}

const struct vb2_ops vivid_touch_cap_qops = {
        .queue_setup            = touch_cap_queue_setup,
        .buf_prepare            = touch_cap_buf_prepare,
        .buf_queue              = touch_cap_buf_queue,
        .start_streaming        = touch_cap_start_streaming,
        .stop_streaming         = touch_cap_stop_streaming,
        .buf_request_complete   = touch_cap_buf_request_complete,
        .wait_prepare           = vb2_ops_wait_prepare,
        .wait_finish            = vb2_ops_wait_finish,
};

int vivid_enum_fmt_tch(struct file *file, void  *priv, struct v4l2_fmtdesc *f)
{
        if (f->index)
                return -EINVAL;

        f->pixelformat = V4L2_TCH_FMT_DELTA_TD16;
        return 0;
}

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

        if (dev->multiplanar)
                return -ENOTTY;
        f->fmt.pix = dev->tch_format;
        return 0;
}

int vivid_g_fmt_tch_mplane(struct file *file, void *priv, struct v4l2_format *f)
{
        struct vivid_dev *dev = video_drvdata(file);
        struct v4l2_format sp_fmt;

        if (!dev->multiplanar)
                return -ENOTTY;
        sp_fmt.type = V4L2_BUF_TYPE_VIDEO_CAPTURE;
        sp_fmt.fmt.pix = dev->tch_format;
        fmt_sp2mp(&sp_fmt, f);
        return 0;
}

int vivid_g_parm_tch(struct file *file, void *priv,
                     struct v4l2_streamparm *parm)
{
        struct vivid_dev *dev = video_drvdata(file);

        if (parm->type != (dev->multiplanar ?
                           V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE :
                           V4L2_BUF_TYPE_VIDEO_CAPTURE))
                return -EINVAL;

        parm->parm.capture.capability   = V4L2_CAP_TIMEPERFRAME;
        parm->parm.capture.timeperframe = dev->timeperframe_tch_cap;
        parm->parm.capture.readbuffers  = 1;
        return 0;
}

int vivid_enum_input_tch(struct file *file, void *priv, struct v4l2_input *inp)
{
        if (inp->index)
                return -EINVAL;

        inp->type = V4L2_INPUT_TYPE_TOUCH;
        strscpy(inp->name, "Vivid Touch", sizeof(inp->name));
        inp->capabilities = 0;
        return 0;
}

int vivid_g_input_tch(struct file *file, void *priv, unsigned int *i)
{
        *i = 0;
        return 0;
}

int vivid_set_touch(struct vivid_dev *dev, unsigned int i)
{
        struct v4l2_pix_format *f = &dev->tch_format;

        if (i)
                return -EINVAL;

        f->pixelformat = V4L2_TCH_FMT_DELTA_TD16;
        f->width =  VIVID_TCH_WIDTH;
        f->height = VIVID_TCH_HEIGHT;
        f->field = V4L2_FIELD_NONE;
        f->colorspace = V4L2_COLORSPACE_RAW;
        f->bytesperline = f->width * sizeof(s16);
        f->sizeimage = f->width * f->height * sizeof(s16);
        return 0;
}

int vivid_s_input_tch(struct file *file, void *priv, unsigned int i)
{
        return vivid_set_touch(video_drvdata(file), i);
}

static void vivid_fill_buff_noise(__s16 *tch_buf, int size)
{
        int i;

        /* Fill 10% of the values within range -3 and 3, zero the others */
        for (i = 0; i < size; i++) {
                unsigned int rand = get_random_int();

                if (rand % 10)
                        tch_buf[i] = 0;
                else
                        tch_buf[i] = (rand / 10) % 7 - 3;
        }
}

static inline int get_random_pressure(void)
{
        return get_random_int() % VIVID_PRESSURE_LIMIT;
}

static void vivid_tch_buf_set(struct v4l2_pix_format *f,
                              __s16 *tch_buf,
                              int index)
{
        unsigned int x = index % f->width;
        unsigned int y = index / f->width;
        unsigned int offset = VIVID_MIN_PRESSURE;

        tch_buf[index] = offset + get_random_pressure();
        offset /= 2;
        if (x)
                tch_buf[index - 1] = offset + get_random_pressure();
        if (x < f->width - 1)
                tch_buf[index + 1] = offset + get_random_pressure();
        if (y)
                tch_buf[index - f->width] = offset + get_random_pressure();
        if (y < f->height - 1)
                tch_buf[index + f->width] = offset + get_random_pressure();
        offset /= 2;
        if (x && y)
                tch_buf[index - 1 - f->width] = offset + get_random_pressure();
        if (x < f->width - 1 && y)
                tch_buf[index + 1 - f->width] = offset + get_random_pressure();
        if (x && y < f->height - 1)
                tch_buf[index - 1 + f->width] = offset + get_random_pressure();
        if (x < f->width - 1 && y < f->height - 1)
                tch_buf[index + 1 + f->width] = offset + get_random_pressure();
}

void vivid_fillbuff_tch(struct vivid_dev *dev, struct vivid_buffer *buf)
{
        struct v4l2_pix_format *f = &dev->tch_format;
        int size = f->width * f->height;
        int x, y, xstart, ystart, offset_x, offset_y;
        unsigned int test_pattern, test_pat_idx, rand;

        __s16 *tch_buf = vb2_plane_vaddr(&buf->vb.vb2_buf, 0);

        buf->vb.sequence = dev->touch_cap_seq_count;
        test_pattern = (buf->vb.sequence / TCH_SEQ_COUNT) % TEST_CASE_MAX;
        test_pat_idx = buf->vb.sequence % TCH_SEQ_COUNT;

        vivid_fill_buff_noise(tch_buf, size);

        if (test_pat_idx >= TCH_PATTERN_COUNT)
                return;

        if (test_pat_idx == 0)
                dev->tch_pat_random = get_random_int();
        rand = dev->tch_pat_random;

        switch (test_pattern) {
        case SINGLE_TAP:
                if (test_pat_idx == 2)
                        vivid_tch_buf_set(f, tch_buf, rand % size);
                break;
        case DOUBLE_TAP:
                if (test_pat_idx == 2 || test_pat_idx == 4)
                        vivid_tch_buf_set(f, tch_buf, rand % size);
                break;
        case TRIPLE_TAP:
                if (test_pat_idx == 2 || test_pat_idx == 4 || test_pat_idx == 6)
                        vivid_tch_buf_set(f, tch_buf, rand % size);
                break;
        case MOVE_LEFT_TO_RIGHT:
                vivid_tch_buf_set(f, tch_buf,
                                  (rand % f->height) * f->width +
                                  test_pat_idx *
                                  (f->width / TCH_PATTERN_COUNT));
                break;
        case ZOOM_IN:
                x = f->width / 2;
                y = f->height / 2;
                offset_x = ((TCH_PATTERN_COUNT - 1 - test_pat_idx) * x) /
                                TCH_PATTERN_COUNT;
                offset_y = ((TCH_PATTERN_COUNT - 1 - test_pat_idx) * y) /
                                TCH_PATTERN_COUNT;
                vivid_tch_buf_set(f, tch_buf,
                                  (x - offset_x) + f->width * (y - offset_y));
                vivid_tch_buf_set(f, tch_buf,
                                  (x + offset_x) + f->width * (y + offset_y));
                break;
        case ZOOM_OUT:
                x = f->width / 2;
                y = f->height / 2;
                offset_x = (test_pat_idx * x) / TCH_PATTERN_COUNT;
                offset_y = (test_pat_idx * y) / TCH_PATTERN_COUNT;
                vivid_tch_buf_set(f, tch_buf,
                                  (x - offset_x) + f->width * (y - offset_y));
                vivid_tch_buf_set(f, tch_buf,
                                  (x + offset_x) + f->width * (y + offset_y));
                break;
        case PALM_PRESS:
                for (x = 0; x < f->width; x++)
                        for (y = f->height / 2; y < f->height; y++)
                                tch_buf[x + f->width * y] = VIVID_MIN_PRESSURE +
                                                        get_random_pressure();
                break;
        case MULTIPLE_PRESS:
                /* 16 pressure points */
                for (y = 0; y < 4; y++) {
                        for (x = 0; x < 4; x++) {
                                ystart = (y * f->height) / 4 + f->height / 8;
                                xstart = (x * f->width) / 4 + f->width / 8;
                                vivid_tch_buf_set(f, tch_buf,
                                                  ystart * f->width + xstart);
                        }
                }
                break;
        }
#ifdef __BIG_ENDIAN__
        for (x = 0; x < size; x++)
                tch_buf[x] = (__force s16)__cpu_to_le16((u16)tch_buf[x]);
#endif
}