diff options
| -rw-r--r-- | yavta.c | 270 |
1 files changed, 235 insertions, 35 deletions
@@ -1,20 +1,8 @@ +// SPDX-License-Identifier: GPL-2.0-or-later /* * yavta -- Yet Another V4L2 Test Application * - * Copyright (C) 2005-2010 Laurent Pinchart <laurent.pinchart@ideasonboard.com> - * - * This program is free software; you can redistribute it and/or modify - * it under the terms of the GNU General Public License as published by - * the Free Software Foundation; either version 2 of the License, or - * (at your option) any later version. - * - * This program is distributed in the hope that it will be useful, - * but WITHOUT ANY WARRANTY; without even the implied warranty of - * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - * GNU General Public License for more details. - * - * You should have received a copy of the GNU General Public License along - * with this program; if not, write to the Free Software Foundation, Inc., + * Copyright (C) 2005-2025 Laurent Pinchart <laurent.pinchart@ideasonboard.com> */ #define __STDC_FORMAT_MACROS @@ -197,7 +185,64 @@ static int pause_init(void) } /* ----------------------------------------------------------------------------- - * Pause Handling + * Key-value pairs handling + */ + +struct key_value { + const char *name; + unsigned int value; +}; + +static int __key_value_get(const struct key_value *values, + unsigned int count, const char *name) +{ + unsigned int i; + + for (i = 0; i < count; ++i) { + if (!strcmp(values[i].name, name)) + return values[i].value; + } + + return -EINVAL; +} + +static void __key_value_list(const struct key_value *values, + unsigned int count, const char *type) +{ + unsigned int chars; + unsigned int i; + bool first = true; + + chars = printf("%s: ", type); + + for (i = 0; i < count; ++i) { + unsigned int len = strlen(values[i].name); + + if (chars + len >= 80) { + printf(",\n\t"); + chars = 8; + first = true; + } + + if (first) + first = false; + else + chars += printf(", "); + + chars += printf("%s", values[i].name); + } + + printf("\n"); +} + +#define key_value_get(values, name) \ + __key_value_get(values, ARRAY_SIZE(values), name) + +#define key_value_list(values, type) \ + __key_value_list(values, ARRAY_SIZE(values), type) + +/* ----------------------------------------------------------------------------- + * Format handling */ static bool video_is_mplane(struct device *dev) @@ -226,7 +271,7 @@ static bool video_is_output(struct device *dev) dev->type == V4L2_BUF_TYPE_META_OUTPUT; } -static struct { +static const struct { enum v4l2_buf_type type; bool supported; const char *name; @@ -273,7 +318,7 @@ static const char *v4l2_buf_type_name(enum v4l2_buf_type type) return "Unknown"; } -static struct v4l2_format_info { +static const struct v4l2_format_info { const char *name; unsigned int fourcc; unsigned char n_planes; @@ -358,6 +403,7 @@ static struct v4l2_format_info { { "IPU3_SGBRG10", V4L2_PIX_FMT_IPU3_SGBRG10, 1 }, { "IPU3_SGRBG10", V4L2_PIX_FMT_IPU3_SGRBG10, 1 }, { "IPU3_SRGGB10", V4L2_PIX_FMT_IPU3_SRGGB10, 1 }, + { "IPU3_Y10", V4L2_PIX_FMT_IPU3_Y10, 1 }, { "DV", V4L2_PIX_FMT_DV, 1 }, { "MJPEG", V4L2_PIX_FMT_MJPEG, 1 }, { "MPEG", V4L2_PIX_FMT_MPEG, 1 }, @@ -460,6 +506,77 @@ static const char *v4l2_field_name(enum v4l2_field field) return "unknown"; } +static const struct key_value v4l2_colorspaces[] = { + { "DEFAULT", V4L2_COLORSPACE_DEFAULT }, + { "SMPTE170M", V4L2_COLORSPACE_SMPTE170M }, + { "SMPTE240M", V4L2_COLORSPACE_SMPTE240M }, + { "REC709", V4L2_COLORSPACE_REC709 }, + { "BT878", V4L2_COLORSPACE_BT878 }, + { "470_SYSTEM_M", V4L2_COLORSPACE_470_SYSTEM_M }, + { "470_SYSTEM_BG", V4L2_COLORSPACE_470_SYSTEM_BG }, + { "JPEG", V4L2_COLORSPACE_JPEG }, + { "SRGB", V4L2_COLORSPACE_SRGB }, + { "OPRGB", V4L2_COLORSPACE_OPRGB }, + { "BT2020", V4L2_COLORSPACE_BT2020 }, + { "RAW", V4L2_COLORSPACE_RAW }, + { "DCI_P3", V4L2_COLORSPACE_DCI_P3 }, +}; + +static const struct key_value v4l2_xfer_funcs[] = { + { "DEFAULT", V4L2_COLORSPACE_DEFAULT }, + { "709", V4L2_XFER_FUNC_709 }, + { "SRGB", V4L2_XFER_FUNC_SRGB }, + { "OPRGB", V4L2_XFER_FUNC_OPRGB }, + { "SMPTE240M", V4L2_XFER_FUNC_SMPTE240M }, + { "NONE", V4L2_XFER_FUNC_NONE }, + { "DCI_P3", V4L2_XFER_FUNC_DCI_P3 }, + { "SMPTE2084", V4L2_XFER_FUNC_SMPTE2084 }, +}; + +static const struct key_value v4l2_encodings[] = { + /* enum v4l2_ycbcr_encoding */ + { "DEFAULT", V4L2_YCBCR_ENC_DEFAULT }, + { "601", V4L2_YCBCR_ENC_601 }, + { "709", V4L2_YCBCR_ENC_709 }, + { "XV601", V4L2_YCBCR_ENC_XV601 }, + { "XV709", V4L2_YCBCR_ENC_XV709 }, + { "SYCC", V4L2_YCBCR_ENC_SYCC }, + { "BT2020", V4L2_YCBCR_ENC_BT2020 }, + { "BT2020_CONST_LUM", V4L2_YCBCR_ENC_BT2020_CONST_LUM }, + { "SMPTE240M", V4L2_YCBCR_ENC_SMPTE240M }, + /* enum v4l2_hsv_encoding */ + { "HSV180", V4L2_HSV_ENC_180 }, + { "HSV256", V4L2_HSV_ENC_256 }, +}; + +static const struct key_value v4l2_quantizations[] = { + { "DEFAULT", V4L2_QUANTIZATION_DEFAULT }, + { "FULL_RANGE", V4L2_QUANTIZATION_FULL_RANGE }, + { "LIM_RANGE", V4L2_QUANTIZATION_LIM_RANGE }, +}; + +#define v4l2_colorspace_from_string(name) \ + key_value_get(v4l2_colorspaces, name) +#define v4l2_xfer_func_from_string(name) \ + key_value_get(v4l2_xfer_funcs, name) +#define v4l2_encoding_from_string(name) \ + key_value_get(v4l2_encodings, name) +#define v4l2_quantization_from_string(name) \ + key_value_get(v4l2_quantizations, name) + +#define list_colorspaces() \ + key_value_list(v4l2_colorspaces, "colorspace") +#define list_xfer_funcs() \ + key_value_list(v4l2_xfer_funcs, "xfer-func") +#define list_encodings() \ + key_value_list(v4l2_encodings, "encoding") +#define list_quantizations() \ + key_value_list(v4l2_quantizations, "quantization") + +/* ----------------------------------------------------------------------------- + * + */ + static void video_set_buf_type(struct device *dev, enum v4l2_buf_type type) { dev->type = type; @@ -804,6 +921,10 @@ static int video_get_format(struct device *dev) static int video_set_format(struct device *dev, unsigned int w, unsigned int h, unsigned int format, unsigned int stride, unsigned int buffer_size, enum v4l2_field field, + enum v4l2_colorspace colorspace, + enum v4l2_xfer_func xfer_func, + enum v4l2_ycbcr_encoding encoding, + enum v4l2_quantization quantization, unsigned int flags) { struct v4l2_format fmt; @@ -821,7 +942,11 @@ static int video_set_format(struct device *dev, unsigned int w, unsigned int h, fmt.fmt.pix_mp.pixelformat = format; fmt.fmt.pix_mp.field = field; fmt.fmt.pix_mp.num_planes = info->n_planes; + fmt.fmt.pix_mp.colorspace = colorspace; fmt.fmt.pix_mp.flags = flags; + fmt.fmt.pix_mp.ycbcr_enc = encoding; + fmt.fmt.pix_mp.quantization = quantization; + fmt.fmt.pix_mp.xfer_func = xfer_func; for (i = 0; i < fmt.fmt.pix_mp.num_planes; i++) { fmt.fmt.pix_mp.plane_fmt[i].bytesperline = stride; @@ -837,8 +962,12 @@ static int video_set_format(struct device *dev, unsigned int w, unsigned int h, fmt.fmt.pix.field = field; fmt.fmt.pix.bytesperline = stride; fmt.fmt.pix.sizeimage = buffer_size; + fmt.fmt.pix.colorspace = colorspace; fmt.fmt.pix.priv = V4L2_PIX_FMT_PRIV_MAGIC; fmt.fmt.pix.flags = flags; + fmt.fmt.pix.ycbcr_enc = encoding; + fmt.fmt.pix.quantization = quantization; + fmt.fmt.pix.xfer_func = xfer_func; } ret = ioctl(dev->fd, VIDIOC_S_FMT, &fmt); @@ -1315,7 +1444,8 @@ static void video_query_menu(struct device *dev, printf(" %u: %.32s%s\n", menu.index, menu.name, menu.index == value ? " (*)" : ""); else - printf(" %u: %lld%s\n", menu.index, menu.value, + printf(" %u: %" PRId64 "%s\n", menu.index, + (int64_t)menu.value, menu.index == value ? " (*)" : ""); }; } @@ -1362,7 +1492,7 @@ static void video_print_control_value(const struct v4l2_query_ext_ctrl *query, printf("0x%08x", ctrl->value); break; case V4L2_CTRL_TYPE_INTEGER64: - printf("%lld", ctrl->value64); + printf("%" PRId64, (int64_t)ctrl->value64); break; case V4L2_CTRL_TYPE_STRING: printf("%s", ctrl->string); @@ -1401,9 +1531,10 @@ static int video_get_control(struct device *dev, } if (full) { - printf("control 0x%08x `%s' min %lld max %lld step %lld default %lld ", - query->id, query->name, query->minimum, query->maximum, - query->step, query->default_value); + printf("control 0x%08x `%s' min %" PRId64 " max %" PRId64 " step %" PRIu64 " default %" PRId64 " ", + query->id, query->name, (int64_t)query->minimum, + (int64_t)query->maximum, (uint64_t)query->step, + (int64_t)query->default_value); if (query->nr_of_dims) { for (i = 0; i < query->nr_of_dims; ++i) printf("[%u]", query->dims[i]); @@ -1941,6 +2072,8 @@ static int video_load_test_pattern(struct device *dev, const char *filename) ret = -EINVAL; goto done; } + + size = ret; } else { uint8_t *data = dev->pattern[plane]; unsigned int i; @@ -2192,13 +2325,16 @@ static int video_do_capture(struct device *dev, unsigned int nframes, clock_gettime(CLOCK_MONOTONIC, &ts); get_ts_flags(buf.flags, &ts_type, &ts_source); - printf("%u (%u) [%c] %s %u %u B %ld.%06ld %ld.%06ld %.3f fps ts %s/%s\n", i, buf.index, - (buf.flags & V4L2_BUF_FLAG_ERROR) ? 'E' : '-', - v4l2_field_name(buf.field), - buf.sequence, video_buffer_bytes_used(dev, &buf), - buf.timestamp.tv_sec, buf.timestamp.tv_usec, - ts.tv_sec, ts.tv_nsec/1000, fps, - ts_type, ts_source); + printf("%u (%u) [%c] %s %u %u B %" PRId64 ".%06" PRId64 " %" PRId64 ".%06" PRId64 " %.3f fps ts %s/%s\n", + i, buf.index, + (buf.flags & V4L2_BUF_FLAG_ERROR) ? 'E' : '-', + v4l2_field_name(buf.field), + buf.sequence, video_buffer_bytes_used(dev, &buf), + (int64_t)buf.timestamp.tv_sec, + (int64_t)buf.timestamp.tv_usec, + (int64_t)ts.tv_sec, + (int64_t)(ts.tv_nsec / 1000), fps, + ts_type, ts_source); last = buf.timestamp; @@ -2254,8 +2390,9 @@ static int video_do_capture(struct device *dev, unsigned int nframes, bps = size/(ts.tv_nsec/1000.0+1000000.0*ts.tv_sec)*1000000.0; fps = i/(ts.tv_nsec/1000.0+1000000.0*ts.tv_sec)*1000000.0; - printf("Captured %u frames in %lu.%06lu seconds (%f fps, %f B/s).\n", - i, ts.tv_sec, ts.tv_nsec/1000, fps, bps); + printf("Captured %u frames in %" PRId64 ".%06" PRId64 " seconds (%f fps, %f B/s).\n", + i, (int64_t)ts.tv_sec, (int64_t)(ts.tv_nsec / 1000), fps, + bps); done: video_free_buffers(dev); @@ -2297,6 +2434,8 @@ static void usage(const char *argv0) printf(" --buffer-size Buffer size in bytes\n"); printf(" --enum-formats Enumerate formats\n"); printf(" --enum-inputs Enumerate inputs\n"); + printf(" --colorspace colorspace Set the colorspace\n"); + printf(" --encoding encoding Set the YCbCr encoding\n"); printf(" --fd Use a numeric file descriptor insted of a device\n"); printf(" --field field Set the format field order\n"); printf("\tValid values for field are none, top, bottom, interlaced, seq-tb, seq-bt,\n"); @@ -2305,6 +2444,7 @@ static void usage(const char *argv0) printf(" --no-query Don't query capabilities on open\n"); printf(" --offset User pointer buffer offset from page start\n"); printf(" --premultiplied Color components are premultiplied by alpha value\n"); + printf(" --quantization quantization Set the quantization\n"); printf(" --queue-late Queue buffers after streamon, not before\n"); printf(" --requeue-last Requeue the last buffers before streamoff\n"); printf(" --reset-controls Reset all available controls to their default value\n"); @@ -2312,6 +2452,13 @@ static void usage(const char *argv0) printf(" --skip n Skip the first n frames\n"); printf(" --sleep-forever Sleep forever after configuring the device\n"); printf(" --stride value Line stride in bytes\n"); + printf(" --xfer-func xfer-func Set the transfer function\n"); + + printf("\nValid fields values:\n"); + list_colorspaces(); + list_encodings(); + list_quantizations(); + list_xfer_funcs(); } #define OPT_ENUM_FORMATS 256 @@ -2331,14 +2478,20 @@ static void usage(const char *argv0) #define OPT_QUEUE_LATE 270 #define OPT_DATA_PREFIX 271 #define OPT_RESET_CONTROLS 272 +#define OPT_COLORSPACE 273 +#define OPT_XFER_FUNC 274 +#define OPT_ENCODING 275 +#define OPT_QUANTIZATION 276 -static struct option opts[] = { +static const struct option opts[] = { {"buffer-size", 1, 0, OPT_BUFFER_SIZE}, {"buffer-type", 1, 0, 'B'}, {"capture", 2, 0, 'c'}, {"check-overrun", 0, 0, 'C'}, + {"colorspace", 1, 0, OPT_COLORSPACE}, {"data-prefix", 0, 0, OPT_DATA_PREFIX}, {"delay", 1, 0, 'd'}, + {"encoding", 1, 0, OPT_ENCODING}, {"enum-formats", 0, 0, OPT_ENUM_FORMATS}, {"enum-inputs", 0, 0, OPT_ENUM_INPUTS}, {"fd", 1, 0, OPT_FD}, @@ -2356,6 +2509,7 @@ static struct option opts[] = { {"pause", 2, 0, 'p'}, {"premultiplied", 0, 0, OPT_PREMULTIPLIED}, {"quality", 1, 0, 'q'}, + {"quantization", 1, 0, OPT_QUANTIZATION}, {"queue-late", 0, 0, OPT_QUEUE_LATE}, {"get-control", 1, 0, 'r'}, {"requeue-last", 0, 0, OPT_REQUEUE_LAST}, @@ -2369,6 +2523,7 @@ static struct option opts[] = { {"time-per-frame", 1, 0, 't'}, {"timestamp-source", 1, 0, OPT_TSTAMP_SRC}, {"userptr", 0, 0, 'u'}, + {"xfer-func", 1, 0, OPT_XFER_FUNC}, {0, 0, 0, 0} }; @@ -2391,6 +2546,7 @@ int main(int argc, char *argv[]) int do_sleep_forever = 0, do_requeue_last = 0; int do_rt = 0, do_log_status = 0; int no_query = 0, do_queue_late = 0; + int do_csc = 0; char *endptr; int c; @@ -2414,6 +2570,10 @@ int main(int argc, char *argv[]) unsigned int pause_count = (unsigned int)-1; struct v4l2_fract time_per_frame = {1, 25}; enum v4l2_field field = V4L2_FIELD_ANY; + enum v4l2_colorspace colorspace = V4L2_COLORSPACE_DEFAULT; + enum v4l2_xfer_func xfer_func = V4L2_XFER_FUNC_DEFAULT; + enum v4l2_ycbcr_encoding encoding = V4L2_YCBCR_ENC_DEFAULT; + enum v4l2_quantization quantization = V4L2_QUANTIZATION_DEFAULT; /* Capture loop */ enum buffer_fill_mode fill_mode = BUFFER_FILL_NONE; @@ -2546,6 +2706,27 @@ int main(int argc, char *argv[]) case OPT_BUFFER_SIZE: buffer_size = atoi(optarg); break; + case OPT_COLORSPACE: + ret = v4l2_colorspace_from_string(optarg); + if (ret < 0) { + printf("Invalid colorspace value '%s'\n", optarg); + return 1; + } + colorspace = ret; + do_csc = 1; + break; + case OPT_DATA_PREFIX: + dev.write_data_prefix = true; + break; + case OPT_ENCODING: + ret = v4l2_encoding_from_string(optarg); + if (ret < 0) { + printf("Invalid encoding value '%s'\n", optarg); + return 1; + } + encoding = ret; + do_csc = 1; + break; case OPT_ENUM_FORMATS: do_enum_formats = 1; break; @@ -2577,6 +2758,15 @@ int main(int argc, char *argv[]) case OPT_PREMULTIPLIED: fmt_flags |= V4L2_PIX_FMT_FLAG_PREMUL_ALPHA; break; + case OPT_QUANTIZATION: + ret = v4l2_quantization_from_string(optarg); + if (ret < 0) { + printf("Invalid quantization value '%s'\n", optarg); + return 1; + } + quantization = ret; + do_csc = 1; + break; case OPT_QUEUE_LATE: do_queue_late = 1; break; @@ -2608,8 +2798,14 @@ int main(int argc, char *argv[]) case OPT_USERPTR_OFFSET: userptr_offset = atoi(optarg); break; - case OPT_DATA_PREFIX: - dev.write_data_prefix = true; + case OPT_XFER_FUNC: + ret = v4l2_xfer_func_from_string(optarg); + if (ret < 0) { + printf("Invalid xfer-func value '%s'\n", optarg); + return 1; + } + xfer_func = ret; + do_csc = 1; break; default: printf("Invalid option -%c\n", c); @@ -2701,8 +2897,12 @@ int main(int argc, char *argv[]) /* Set the video format. */ if (do_set_format) { + if (do_csc && video_is_capture(&dev)) + fmt_flags |= V4L2_PIX_FMT_FLAG_SET_CSC; + if (video_set_format(&dev, width, height, pixelformat, stride, - buffer_size, field, fmt_flags) < 0) { + buffer_size, field, colorspace, xfer_func, + encoding, quantization, fmt_flags) < 0) { video_close(&dev); return 1; } |