[FFmpeg-devel] [PATCH] avcodec/mediacodecenc: Support config qp range

[Zhao Zhili]

From: Zhao Zhili <zhilizhao at tencent.com>

---
 libavcodec/mediacodecenc.c | 64 +++++++++++++++++++++++++++++++++++++-
 libavcodec/version.h       |  2 +-
 2 files changed, 64 insertions(+), 2 deletions(-)

diff --git a/libavcodec/mediacodecenc.c b/libavcodec/mediacodecenc.c
index 464903d628..d1a0388d7e 100644
--- a/libavcodec/mediacodecenc.c
+++ b/libavcodec/mediacodecenc.c
@@ -95,6 +95,13 @@ typedef struct MediaCodecEncContext {
     AVCond output_cond;
     int encode_status;
     AVFifo *async_output;
+
+    int qp_i_min;
+    int qp_p_min;
+    int qp_b_min;
+    int qp_i_max;
+    int qp_p_max;
+    int qp_b_max;
 } MediaCodecEncContext;
 
 enum {
@@ -351,6 +358,39 @@ static void mediacodec_uninit_async_state(AVCodecContext *avctx)
 
 static int mediacodec_generate_extradata(AVCodecContext *avctx);
 
+static void mediacodec_set_qp_range(AVCodecContext *avctx,
+                                    FFAMediaFormat *format)
+{
+    MediaCodecEncContext *s = avctx->priv_data;
+
+    // Handle common options in AVCodecContext first.
+    if (avctx->qmin >= 0) {
+        ff_AMediaFormat_setInt32(format, "video-qp-i-min", avctx->qmin);
+        ff_AMediaFormat_setInt32(format, "video-qp-p-min", avctx->qmin);
+        ff_AMediaFormat_setInt32(format, "video-qp-b-min", avctx->qmin);
+    }
+
+    if (avctx->qmax >= 0) {
+        ff_AMediaFormat_setInt32(format, "video-qp-i-max", avctx->qmax);
+        ff_AMediaFormat_setInt32(format, "video-qp-p-max", avctx->qmax);
+        ff_AMediaFormat_setInt32(format, "video-qp-b-max", avctx->qmax);
+    }
+
+    if (s->qp_i_min >= 0)
+        ff_AMediaFormat_setInt32(format, "video-qp-i-min", s->qp_i_min);
+    if (s->qp_p_min >= 0)
+        ff_AMediaFormat_setInt32(format, "video-qp-p-min", s->qp_p_min);
+    if (s->qp_b_min >= 0)
+        ff_AMediaFormat_setInt32(format, "video-qp-b-min", s->qp_b_min);
+
+    if (s->qp_i_max >= 0)
+        ff_AMediaFormat_setInt32(format, "video-qp-i-max", s->qp_i_max);
+    if (s->qp_p_max >= 0)
+        ff_AMediaFormat_setInt32(format, "video-qp-p-max", s->qp_p_max);
+    if (s->qp_b_max >= 0)
+        ff_AMediaFormat_setInt32(format, "video-qp-b-max", s->qp_b_max);
+}
+
 static av_cold int mediacodec_init(AVCodecContext *avctx)
 {
     const char *codec_mime = NULL;
@@ -484,6 +524,8 @@ static av_cold int mediacodec_init(AVCodecContext *avctx)
         if (s->bitrate_mode == BITRATE_MODE_CQ && avctx->global_quality > 0)
             ff_AMediaFormat_setInt32(format, "quality", avctx->global_quality);
     }
+    mediacodec_set_qp_range(avctx, format);
+
     // frame-rate and i-frame-interval are required to configure codec
     if (avctx->framerate.num >= avctx->framerate.den && avctx->framerate.den > 0) {
         s->fps = avctx->framerate.num / avctx->framerate.den;
@@ -989,6 +1031,12 @@ static const AVCodecHWConfigInternal *const mediacodec_hw_configs[] = {
     NULL
 };
 
+static const FFCodecDefault mediacodec_defaults[] = {
+    {"qmin", "-1"},
+    {"qmax", "-1"},
+    {NULL},
+};
+
 #define OFFSET(x) offsetof(MediaCodecEncContext, x)
 #define VE AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM
 #define COMMON_OPTION                                                                                       \
@@ -1012,7 +1060,20 @@ static const AVCodecHWConfigInternal *const mediacodec_hw_configs[] = {
                     "since most of Android devices don't output B frames by default.",                      \
                     OFFSET(pts_as_dts), AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1, VE },                         \
     { "operating_rate", "The desired operating rate that the codec will need to operate at, zero for unspecified",    \
-            OFFSET(operating_rate), AV_OPT_TYPE_INT, {.i64 = 0}, 0, INT_MAX, VE },                                          \
+                    OFFSET(operating_rate), AV_OPT_TYPE_INT, {.i64 = 0}, 0, INT_MAX, VE },                  \
+    { "qp_i_min", "minimum quantization parameter for I frame",                                             \
+                    OFFSET(qp_i_min), AV_OPT_TYPE_INT, {.i64 = -1}, 0, INT_MAX, VE },                       \
+    { "qp_p_min", "minimum quantization parameter for P frame",                                             \
+                    OFFSET(qp_p_min), AV_OPT_TYPE_INT, {.i64 = -1}, 0, INT_MAX, VE },                       \
+    { "qp_b_min", "minimum quantization parameter for B frame",                                             \
+                    OFFSET(qp_b_min), AV_OPT_TYPE_INT, {.i64 = -1}, 0, INT_MAX, VE },                       \
+    { "qp_i_max", "maximum quantization parameter for I frame",                                             \
+                    OFFSET(qp_i_max), AV_OPT_TYPE_INT, {.i64 = -1}, 0, INT_MAX, VE },                       \
+    { "qp_p_max", "maximum quantization parameter for P frame",                                             \
+                    OFFSET(qp_p_max), AV_OPT_TYPE_INT, {.i64 = -1}, 0, INT_MAX, VE },                       \
+    { "qp_b_max", "maximum quantization parameter for B frame",                                             \
+                    OFFSET(qp_b_max), AV_OPT_TYPE_INT, {.i64 = -1}, 0, INT_MAX, VE },                       \
+
 
 #define MEDIACODEC_ENCODER_CLASS(name)              \
 static const AVClass name ## _mediacodec_class = {  \
@@ -1035,6 +1096,7 @@ const FFCodec ff_ ## short_name ## _mediacodec_encoder = {              \
     .priv_data_size   = sizeof(MediaCodecEncContext),                   \
     .p.pix_fmts       = avc_pix_fmts,                                   \
     .color_ranges   = AVCOL_RANGE_MPEG | AVCOL_RANGE_JPEG,              \
+    .defaults         = mediacodec_defaults,                            \
     .init             = mediacodec_init,                                \
     FF_CODEC_RECEIVE_PACKET_CB(mediacodec_encode),                      \
     .close            = mediacodec_close,                               \
diff --git a/libavcodec/version.h b/libavcodec/version.h
index 497389d3f3..1030154c0e 100644
--- a/libavcodec/version.h
+++ b/libavcodec/version.h
@@ -30,7 +30,7 @@
 #include "version_major.h"
 
 #define LIBAVCODEC_VERSION_MINOR  31
-#define LIBAVCODEC_VERSION_MICRO 100
+#define LIBAVCODEC_VERSION_MICRO 101
 
 #define LIBAVCODEC_VERSION_INT  AV_VERSION_INT(LIBAVCODEC_VERSION_MAJOR, \
                                                LIBAVCODEC_VERSION_MINOR, \
-- 
2.46.0