Add colorspace supportHEADmaster

Add four command line options to support setting the colorspace,
transfer function, encoding and quantization.

Signed-off-by: Laurent Pinchart <laurent.pinchart+renesas@ideasonboard.com>
Reviewed-by: Sakari Ailus <sakari.ailus@linux.intel.com>

1 files changed, 202 insertions, 3 deletions

diff --git a/yavta.c b/yavta.c
index 4f7306d..b463f58 100644
--- a/yavta.c
+++ b/yavta.c
@@ -185,6 +185,63 @@ static int pause_init(void)
 }
 
 /* -----------------------------------------------------------------------------
+ * 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
  */
 
@@ -449,6 +506,73 @@ 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")
+
 /* -----------------------------------------------------------------------------
  *
  */
@@ -797,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;
@@ -814,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;
@@ -830,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);
@@ -2298,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");
@@ -2306,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");
@@ -2313,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
@@ -2332,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 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},
@@ -2357,6 +2509,7 @@ static const 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},
@@ -2370,6 +2523,7 @@ static const 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}
 };
 
@@ -2392,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;
 
@@ -2415,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;
@@ -2547,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;
@@ -2578,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;
@@ -2609,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);
@@ -2702,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;
 		}