On 2/27/19 8:07 AM, Dafna Hirschfeld wrote:
> Add code to support the stateless decoder
> and the function 'stateless_m2m' that implements it.
> 
> Signed-off-by: Dafna Hirschfeld <daf...@gmail.com>
> ---
>  utils/v4l2-ctl/v4l2-ctl-streaming.cpp | 234 +++++++++++++++++++++++++-
>  1 file changed, 231 insertions(+), 3 deletions(-)
> 
> diff --git a/utils/v4l2-ctl/v4l2-ctl-streaming.cpp 
> b/utils/v4l2-ctl/v4l2-ctl-streaming.cpp
> index dd0eeef6..e279b0b5 100644
> --- a/utils/v4l2-ctl/v4l2-ctl-streaming.cpp
> +++ b/utils/v4l2-ctl/v4l2-ctl-streaming.cpp
> @@ -1071,8 +1071,9 @@ restart:
>                               return false;
>                       }
>               }
> -
> -             if (support_out_crop && 
> v4l2_fwht_find_pixfmt(fmt.g_pixelformat()))
> +             if (q.g_capabilities() & V4L2_BUF_CAP_SUPPORTS_REQUESTS)
> +                     read_fwht_frame(fmt, (unsigned char *)buf, fin, sz, 
> len);

This looks wrong. This should (I think) test if fmt.g_pixelformat() == 
V4L2_PIX_FMT_FWHT_STATELESS
and only then call read_fwht_frame().

Right now it would also call this function for e.g. the cedrus driver, which is 
obviously
wrong.

What is also inefficient is that read_fwht_frame first reads the header in the 
buffer,
then reads the compressed data after the header. This means that you need to do 
a
memmove as you do below.

Why not read the header into a global struct fwht_cframe_hdr? Then you don't 
need a
split_fwht_frame, which is very inefficient.

> +             else if (support_out_crop && 
> v4l2_fwht_find_pixfmt(fmt.g_pixelformat()))
>                       read_write_padded_frame(fmt, (unsigned char *)buf, fin, 
> sz, len, true);
>               else
>                       sz = fread(buf, 1, len, fin);
> @@ -1099,6 +1100,21 @@ restart:
>       return true;
>  }
>  
> +static bool split_fwht_frame(u8 *frame, struct fwht_cframe_hdr *hdr, 
> unsigned max_len)
> +{
> +     unsigned int len;
> +     bool ret = true;
> +
> +     memcpy(hdr, frame, sizeof(struct fwht_cframe_hdr));
> +     len = ntohl(hdr->size);
> +     if (len > max_len) {
> +             len = max_len;
> +             ret = false;
> +     }
> +     memmove(frame, frame + sizeof(struct fwht_cframe_hdr), len);
> +     return ret;
> +}
> +
>  static int do_setup_out_buffers(cv4l_fd &fd, cv4l_queue &q, FILE *fin, bool 
> qbuf)
>  {
>       tpg_pixel_aspect aspect = TPG_PIXEL_ASPECT_SQUARE;
> @@ -1203,6 +1219,33 @@ static int do_setup_out_buffers(cv4l_fd &fd, 
> cv4l_queue &q, FILE *fin, bool qbuf
>               if (fin && !fill_buffer_from_file(fd, q, buf, fmt, fin))
>                       return -2;
>  
> +             struct fwht_cframe_hdr hdr;
> +
> +             if (q.g_capabilities() & V4L2_BUF_CAP_SUPPORTS_REQUESTS) {

Same thing here, check against V4L2_PIX_FMT_FWHT_STATELESS instead. I won't 
repeat myself,
so check and replace other occurrences of this.

> +                     int media_fd = mi_get_media_fd(fd.g_fd());
> +
> +                     if (!split_fwht_frame((u8 *)q.g_dataptr(buf.g_index(), 
> 0),
> +                                           &hdr, buf.g_length(0)))
> +                             fprintf(stderr, "%s: warning: size field in 
> fwht header is larger than buf size\n",
> +                                     __func__);
> +
> +                     if (media_fd < 0) {
> +                             fprintf(stderr, "%s: mi_get_media_fd failed\n", 
> __func__);
> +                             return media_fd;
> +                     }
> +
> +                     if (alloc_fwht_req(media_fd, i))
> +                             return -1;
> +                     buf.s_request_fd(fwht_reqs[i].fd);
> +                     buf.or_flags(V4L2_BUF_FLAG_REQUEST_FD);
> +
> +                     if (set_fwht_ext_ctrl(fd, &hdr, last_fwht_bf_ts,
> +                                           buf.g_request_fd())) {
> +                             fprintf(stderr, "%s: set_fwht_ext_ctrls failed 
> on %dth buf: %s\n",
> +                                     __func__, i, strerror(errno));
> +                             return -1;
> +                     }
> +             }
>               if (qbuf) {
>                       set_time_stamp(buf);
>                       if (fd.qbuf(buf))
> @@ -1212,6 +1255,16 @@ static int do_setup_out_buffers(cv4l_fd &fd, 
> cv4l_queue &q, FILE *fin, bool qbuf
>                               fprintf(stderr, ">");
>                       fflush(stderr);
>               }
> +             if (q.g_capabilities() & V4L2_BUF_CAP_SUPPORTS_REQUESTS) {
> +                     set_fwht_req_by_idx(i, &hdr,
> +                                         last_fwht_bf_ts, 
> get_ns_timestamp(buf));
> +                     last_fwht_bf_ts = get_ns_timestamp(buf);
> +                     if (ioctl(buf.g_request_fd(), MEDIA_REQUEST_IOC_QUEUE) 
> < 0) {
> +                             fprintf(stderr, "Unable to queue media request: 
> %s\n",
> +                                     strerror(errno));
> +                             return -1;
> +                     }
> +             }
>       }
>       if (qbuf)
>               output_field = field;
> @@ -1450,12 +1503,48 @@ static int do_handle_out(cv4l_fd &fd, cv4l_queue &q, 
> FILE *fin, cv4l_buffer *cap
>                                      (u8 *)q.g_dataptr(buf.g_index(), j));
>       }
>  
> +     struct fwht_cframe_hdr hdr;
> +
> +     if (q.g_capabilities() & V4L2_BUF_CAP_SUPPORTS_REQUESTS) {
> +             if (ioctl(buf.g_request_fd(), MEDIA_REQUEST_IOC_REINIT, NULL)) {
> +                     fprintf(stderr, "Unable to reinit media request: %s\n",
> +                             strerror(errno));
> +                     return -1;
> +             }
> +             if (!split_fwht_frame((u8 *)q.g_dataptr(buf.g_index(), 0), 
> &hdr, buf.g_length(0)))
> +                     fprintf(stderr, "%s: warning: size field in fwht header 
> is larger than buf size\n",
> +                                     __func__);
> +
> +             if (set_fwht_ext_ctrl(fd, &hdr, last_fwht_bf_ts,
> +                                   buf.g_request_fd())) {
> +                     fprintf(stderr, "%s: set_fwht_ext_ctrls failed: %s\n",
> +                             __func__, strerror(errno));
> +                     return -1;
> +             }
> +     }
> +
>       set_time_stamp(buf);
>  
>       if (fd.qbuf(buf)) {
>               fprintf(stderr, "%s: failed: %s\n", "VIDIOC_QBUF", 
> strerror(errno));
>               return -1;
>       }
> +     if (q.g_capabilities() & V4L2_BUF_CAP_SUPPORTS_REQUESTS) {
> +             if (!set_fwht_req_by_fd(&hdr, buf.g_request_fd(), 
> last_fwht_bf_ts,
> +                                     get_ns_timestamp(buf))) {
> +                     fprintf(stderr, "%s: request for fd %d does not 
> exist\n",
> +                             __func__, buf.g_request_fd());
> +                     return -1;
> +             }
> +
> +             last_fwht_bf_ts = get_ns_timestamp(buf);
> +             if (ioctl(buf.g_request_fd(), MEDIA_REQUEST_IOC_QUEUE) < 0) {
> +                     fprintf(stderr, "Unable to queue media request: %s\n",
> +                             strerror(errno));
> +                     return -1;
> +             }
> +     }
> +
>       tpg_update_mv_count(&tpg, V4L2_FIELD_HAS_T_OR_B(output_field));
>  
>       if (!verbose)
> @@ -2244,6 +2333,140 @@ static void stateful_m2m(cv4l_fd &fd, cv4l_queue &in, 
> cv4l_queue &out,
>       tpg_free(&tpg);
>  }
>  
> +static void stateless_m2m(cv4l_fd &fd, cv4l_queue &in, cv4l_queue &out,
> +                       FILE *fin, FILE *fout, cv4l_fd *exp_fd_p)
> +{
> +     fps_timestamps fps_ts[2];
> +     unsigned count[2] = { 0, 0 };
> +     cv4l_fmt fmt[2];
> +     int fd_flags = fcntl(fd.g_fd(), F_GETFL);
> +
> +     fd.g_fmt(fmt[OUT], out.g_type());
> +     fd.g_fmt(fmt[CAP], in.g_type());
> +
> +     if (out.reqbufs(&fd, reqbufs_count_out)) {
> +             fprintf(stderr, "%s: out.reqbufs failed\n", __func__);
> +             return;
> +     }
> +
> +     if (in.reqbufs(&fd, reqbufs_count_cap)) {
> +             fprintf(stderr, "%s: in.reqbufs failed\n", __func__);
> +             return;
> +     }
> +
> +     if (exp_fd_p && in.export_bufs(exp_fd_p, exp_fd_p->g_type()))
> +             return;
> +
> +     if (in.obtain_bufs(&fd)) {
> +             fprintf(stderr, "%s: in.obtain_bufs error\n", __func__);
> +             return;
> +     }
> +
> +     if (do_setup_out_buffers(fd, out, fout, true) == -1) {
> +             fprintf(stderr, "%s: do_setup_out_buffers failed\n", __func__);
> +             return;
> +     }
> +
> +     if (in.queue_all(&fd)) {
> +             fprintf(stderr, "%s: in.queue_all failed\n", __func__);
> +             return;
> +     }
> +
> +     if (fd.streamon(out.g_type())) {
> +             fprintf(stderr, "%s: streamon for out failed\n", __func__);
> +             return;
> +     }
> +
> +     if (fd.streamon(in.g_type())) {
> +             fprintf(stderr, "%s: streamon for in failed\n", __func__);
> +             return;
> +     }
> +     int index = 0;
> +     bool queue_lst_buf = false;
> +     cv4l_buffer last_in_buf;
> +
> +     fcntl(fd.g_fd(), F_SETFL, fd_flags | O_NONBLOCK);
> +
> +     while (true) {
> +             fd_set except_fds;
> +             int req_fd =fwht_reqs[index].fd;
> +             struct timeval tv = { 2, 0 };
> +
> +             FD_ZERO(&except_fds);
> +             FD_SET(req_fd, &except_fds);
> +
> +             int rc = select(req_fd + 1, NULL, NULL, &except_fds, &tv);
> +
> +             if (rc == 0) {
> +                     fprintf(stderr, "Timeout when waiting for media 
> request\n");
> +                     return;
> +             } else if (rc < 0) {
> +                     fprintf(stderr, "Unable to select media request: %s\n",
> +                             strerror(errno));
> +                     return;
> +             }
> +             /*
> +              * it is safe to queue back last cap buffer only after
> +              * the following request is done so that the buffer
> +              * is not needed anymore as a reference frame
> +              */
> +             if (queue_lst_buf) {
> +                     if (fd.qbuf(last_in_buf) < 0) {
> +                             fprintf(stderr, "%s: qbuf failed\n", __func__);
> +                             return;
> +                     }
> +             }
> +             int buf_idx = -1;
> +               /*
> +                * fin is not sent to do_handle_cap since the capture buf is
> +                * written to the file in current function
> +                */
> +             rc = do_handle_cap(fd, in, NULL, &buf_idx, count[CAP],
> +                                fps_ts[CAP], fmt[CAP]);
> +             if (rc) {
> +                     fprintf(stderr, "%s: do_handle_cap err\n", __func__);
> +                     return;
> +             }
> +             /*
> +              * in case of an error in the frame, set last ts to 0 as a mean
> +              * to recover so that next request will not use a
> +              * reference buffer. Otherwise the error flag will be set to
> +              * all the future capture buffers.
> +              */
> +             if (buf_idx == -1) {
> +                     fprintf(stderr, "%s: frame returned with error\n", 
> __func__);
> +                     last_fwht_bf_ts = 0;
> +             } else {
> +                     cv4l_buffer cap_buf(in, index);
> +                     if (fd.querybuf(cap_buf))
> +                             return;
> +                     last_in_buf = cap_buf;
> +                     queue_lst_buf = true;
> +                     if (fin && cap_buf.g_bytesused(0) &&
> +                         !(cap_buf.g_flags() & V4L2_BUF_FLAG_ERROR)) {
> +                             int idx = 
> get_fwht_req_by_ts(get_ns_timestamp(cap_buf));
> +
> +                             if (idx < 0) {
> +                                     fprintf(stderr, "%s: could not find 
> request from buffer\n", __func__);
> +                                     fprintf(stderr, "%s: ts = %llu\n", 
> __func__, get_ns_timestamp(cap_buf));
> +                                     return;
> +                             }
> +                             composed_width = fwht_reqs[idx].params.width;
> +                             composed_height = fwht_reqs[idx].params.height;
> +                             write_buffer_to_file(fd, in, cap_buf,
> +                                                  fmt[CAP], fin);
> +                     }
> +             }
> +             rc = do_handle_out(fd, out, fout, NULL, count[OUT],
> +                                fps_ts[OUT], fmt[OUT]);
> +             if (rc) {
> +                     fprintf(stderr, "%s: output stream ended\n", __func__);
> +                     close(req_fd);
> +             }
> +             index = (index + 1) % out.g_buffers();
> +     }
> +}
> +
>  static void streaming_set_m2m(cv4l_fd &fd, cv4l_fd &exp_fd)
>  {
>       cv4l_queue in(fd.g_type(), memory);
> @@ -2280,7 +2503,12 @@ static void streaming_set_m2m(cv4l_fd &fd, cv4l_fd 
> &exp_fd)
>               if (out.export_bufs(&exp_fd, exp_fd.g_type()))
>                       return;
>       }
> -     stateful_m2m(fd, in, out, file[CAP], file[OUT], exp_fd_p);
> +     if (out.reqbufs(&fd, 0))
> +             goto done;
> +     if (out.g_capabilities() & V4L2_BUF_CAP_SUPPORTS_REQUESTS)
> +             stateless_m2m(fd, in, out, file[CAP], file[OUT], exp_fd_p);
> +     else
> +             stateful_m2m(fd, in, out, file[CAP], file[OUT], exp_fd_p);
>  done:
>       if (options[OptStreamDmaBuf] || options[OptStreamOutDmaBuf])
>               exp_q.close_exported_fds();
> 

Regards,

        Hans

Reply via email to