Message ID | 20190227070757.25092-4-dafna3@gmail.com (mailing list archive) |
---|---|
State | New, archived |
Headers | show |
Series | stateless decoder implementation | expand |
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 <dafna3@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
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); + 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) { + 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();
Add code to support the stateless decoder and the function 'stateless_m2m' that implements it. Signed-off-by: Dafna Hirschfeld <dafna3@gmail.com> --- utils/v4l2-ctl/v4l2-ctl-streaming.cpp | 234 +++++++++++++++++++++++++- 1 file changed, 231 insertions(+), 3 deletions(-)