From ea8689452d7bad7b64c96119620598e6aabc669b Mon Sep 17 00:00:00 2001
From: Steven Toth <stoth@kernellabs.com>
Date: Sat, 31 Jul 2010 14:51:30 -0300
Subject: [PATCH] --- yaml --- r: 219364 b: refs/heads/master c:
 2600d71cc535907e5d95cd31751c587afc370065 h: refs/heads/master v: v3

---
 [refs]                                          |  2 +-
 trunk/drivers/media/video/saa7164/saa7164-api.c |  5 ++++-
 .../media/video/saa7164/saa7164-encoder.c       | 17 +++++++++++++++++
 .../drivers/media/video/saa7164/saa7164-types.h |  3 +++
 trunk/drivers/media/video/saa7164/saa7164.h     |  1 +
 5 files changed, 26 insertions(+), 2 deletions(-)

diff --git a/[refs] b/[refs]
index 3d2731d82aa2..61cc9111a6dc 100644
--- a/[refs]
+++ b/[refs]
@@ -1,2 +1,2 @@
 ---
-refs/heads/master: 076031310b182ce89cea861982ff49ed6799767a
+refs/heads/master: 2600d71cc535907e5d95cd31751c587afc370065
diff --git a/trunk/drivers/media/video/saa7164/saa7164-api.c b/trunk/drivers/media/video/saa7164/saa7164-api.c
index 5dcdf7bb12cd..fec5138737e0 100644
--- a/trunk/drivers/media/video/saa7164/saa7164-api.c
+++ b/trunk/drivers/media/video/saa7164/saa7164-api.c
@@ -59,7 +59,10 @@ int saa7164_api_set_encoder(struct saa7164_port *port)
 		printk(KERN_ERR "%s() error, ret = 0x%x\n", __func__, ret);
 
 	/* Establish video bitrates */
-	vb.ucVideoBitRateMode = 0;
+	if (port->encoder_params.bitrate_mode == V4L2_MPEG_VIDEO_BITRATE_MODE_CBR)
+		vb.ucVideoBitRateMode = EU_VIDEO_BIT_RATE_MODE_CONSTANT;
+	else
+		vb.ucVideoBitRateMode = EU_VIDEO_BIT_RATE_MODE_VARIABLE_PEAK;
 	vb.dwVideoBitRate = port->encoder_params.bitrate;
 	vb.dwVideoBitRatePeak = vb.dwVideoBitRate;
 	ret = saa7164_cmd_send(port->dev, port->hwcfg.sourceid, SET_CUR,
diff --git a/trunk/drivers/media/video/saa7164/saa7164-encoder.c b/trunk/drivers/media/video/saa7164/saa7164-encoder.c
index 74aec26dbb25..d7d2fe009ff9 100644
--- a/trunk/drivers/media/video/saa7164/saa7164-encoder.c
+++ b/trunk/drivers/media/video/saa7164/saa7164-encoder.c
@@ -45,6 +45,7 @@ static const u32 saa7164_v4l2_ctrls[] = {
 	V4L2_CID_MPEG_VIDEO_ASPECT,
 	V4L2_CID_MPEG_STREAM_TYPE,
 	V4L2_CID_MPEG_AUDIO_MUTE,
+	V4L2_CID_MPEG_VIDEO_BITRATE_MODE,
 	V4L2_CID_MPEG_VIDEO_BITRATE,
 	0
 };
@@ -379,6 +380,9 @@ static int saa7164_get_ctrl(struct saa7164_port *port,
 	case V4L2_CID_MPEG_VIDEO_ASPECT:
 		ctrl->value = params->ctl_aspect;
 		break;
+	case V4L2_CID_MPEG_VIDEO_BITRATE_MODE:
+		ctrl->value = params->bitrate_mode;
+		break;
 	default:
 		return -EINVAL;
 	}
@@ -438,6 +442,11 @@ static int saa7164_try_ctrl(struct v4l2_ext_control *ctrl, int ac3)
 			(ctrl->value <= 255))
 			ret = 0;
 		break;
+	case V4L2_CID_MPEG_VIDEO_BITRATE_MODE:
+		if ((ctrl->value == V4L2_MPEG_VIDEO_BITRATE_MODE_VBR) ||
+			(ctrl->value == V4L2_MPEG_VIDEO_BITRATE_MODE_CBR))
+			ret = 0;
+		break;
 	default:
 		ret = -EINVAL;
 	}
@@ -497,6 +506,9 @@ static int saa7164_set_ctrl(struct saa7164_port *port,
 			ret = -EIO;
 		}
 		break;
+	case V4L2_CID_MPEG_VIDEO_BITRATE_MODE:
+		params->bitrate_mode = ctrl->value;
+		break;
 	default:
 		return -EINVAL;
 	}
@@ -667,6 +679,10 @@ static int fill_queryctrl(struct saa7164_encoder_params *params,
 			1, V4L2_MPEG_VIDEO_ASPECT_4x3);
 	case V4L2_CID_MPEG_VIDEO_GOP_SIZE:
 		return v4l2_ctrl_query_fill(c, 1, 255, 1, 15);
+	case V4L2_CID_MPEG_VIDEO_BITRATE_MODE:
+		return v4l2_ctrl_query_fill(c,
+			V4L2_MPEG_VIDEO_BITRATE_MODE_VBR, V4L2_MPEG_VIDEO_BITRATE_MODE_CBR,
+			1, V4L2_MPEG_VIDEO_BITRATE_MODE_VBR);
 	default:
 		return -EINVAL;
 	}
@@ -1287,6 +1303,7 @@ int saa7164_encoder_register(struct saa7164_port *port)
 	port->ctl_saturation = 62;
 	port->ctl_sharpness = 8;
 	port->encoder_params.bitrate = ENCODER_DEF_BITRATE;
+	port->encoder_params.bitrate_mode = V4L2_MPEG_VIDEO_BITRATE_MODE_VBR;
 	port->encoder_params.stream_type = V4L2_MPEG_STREAM_TYPE_MPEG2_PS;
 	port->encoder_params.ctl_mute = 0;
 	port->encoder_params.ctl_aspect = V4L2_MPEG_VIDEO_ASPECT_4x3;
diff --git a/trunk/drivers/media/video/saa7164/saa7164-types.h b/trunk/drivers/media/video/saa7164/saa7164-types.h
index 7c2788d9a647..dacd9ce1d419 100644
--- a/trunk/drivers/media/video/saa7164/saa7164-types.h
+++ b/trunk/drivers/media/video/saa7164/saa7164-types.h
@@ -311,6 +311,9 @@ typedef struct
 } __attribute__((packed)) tmComResProcDescrHeader_t;
 
 /* Video bitrate control message */
+#define EU_VIDEO_BIT_RATE_MODE_CONSTANT		(0)
+#define EU_VIDEO_BIT_RATE_MODE_VARIABLE_AVERAGE (1)
+#define EU_VIDEO_BIT_RATE_MODE_VARIABLE_PEAK	(2)
 typedef struct
 {
 	u8	ucVideoBitRateMode;
diff --git a/trunk/drivers/media/video/saa7164/saa7164.h b/trunk/drivers/media/video/saa7164/saa7164.h
index a262875a39e3..712f762d9f73 100644
--- a/trunk/drivers/media/video/saa7164/saa7164.h
+++ b/trunk/drivers/media/video/saa7164/saa7164.h
@@ -255,6 +255,7 @@ struct saa7164_encoder_params {
 	u32 width;
 	u32 is_50hz;
 	u32 bitrate; /* bps */
+	u32 bitrate_mode;
 	u32 stream_type; /* V4L2_MPEG_STREAM_TYPE_MPEG2_TS */
 
 	u32 audio_sampling_freq;