Add support for intra-refresh period v4l2 control and drop
cyclic intra-refresh macroblock control in the same time.

Signed-off-by: Stanimir Varbanov <[email protected]>
---
 drivers/media/platform/qcom/venus/core.h      |  1 +
 drivers/media/platform/qcom/venus/venc.c      | 28 +++++++++++++++++++
 .../media/platform/qcom/venus/venc_ctrls.c    | 13 ++++-----
 3 files changed, 35 insertions(+), 7 deletions(-)

diff --git a/drivers/media/platform/qcom/venus/core.h 
b/drivers/media/platform/qcom/venus/core.h
index a252ed32cc14..8602b7af6a6f 100644
--- a/drivers/media/platform/qcom/venus/core.h
+++ b/drivers/media/platform/qcom/venus/core.h
@@ -222,6 +222,7 @@ struct venc_controls {
        u32 multi_slice_max_mb;
 
        u32 header_mode;
+       u32 intra_refresh_period;
 
        struct {
                u32 h264;
diff --git a/drivers/media/platform/qcom/venus/venc.c 
b/drivers/media/platform/qcom/venus/venc.c
index 6976ed553647..dcc9bd3475bf 100644
--- a/drivers/media/platform/qcom/venus/venc.c
+++ b/drivers/media/platform/qcom/venus/venc.c
@@ -546,6 +546,7 @@ static int venc_set_properties(struct venus_inst *inst)
        struct hfi_quantization quant;
        struct hfi_quantization_range quant_range;
        struct hfi_enable en;
+       struct hfi_intra_refresh intra_refresh = {};
        u32 ptype, rate_control, bitrate;
        u32 profile, level;
        int ret;
@@ -754,6 +755,33 @@ static int venc_set_properties(struct venus_inst *inst)
        if (ret)
                return ret;
 
+       if ((inst->fmt_cap->pixfmt == V4L2_PIX_FMT_H264 ||
+            inst->fmt_cap->pixfmt == V4L2_PIX_FMT_HEVC) &&
+           (rate_control == HFI_RATE_CONTROL_CBR_VFR ||
+            rate_control == HFI_RATE_CONTROL_CBR_CFR)) {
+               intra_refresh.mode = HFI_INTRA_REFRESH_NONE;
+               intra_refresh.cir_mbs = 0;
+
+               if (ctr->intra_refresh_period) {
+                       u32 mbs;
+
+                       mbs = ALIGN(inst->width, 16) * ALIGN(inst->height, 16);
+                       mbs /= 16 * 16;
+                       if (mbs % ctr->intra_refresh_period)
+                               mbs++;
+                       mbs /= ctr->intra_refresh_period;
+
+                       intra_refresh.mode = HFI_INTRA_REFRESH_RANDOM;
+                       intra_refresh.cir_mbs = mbs;
+               }
+
+               ptype = HFI_PROPERTY_PARAM_VENC_INTRA_REFRESH;
+
+               ret = hfi_session_set_property(inst, ptype, &intra_refresh);
+               if (ret)
+                       return ret;
+       }
+
        return 0;
 }
 
diff --git a/drivers/media/platform/qcom/venus/venc_ctrls.c 
b/drivers/media/platform/qcom/venus/venc_ctrls.c
index a52b80055173..6909426882ac 100644
--- a/drivers/media/platform/qcom/venus/venc_ctrls.c
+++ b/drivers/media/platform/qcom/venus/venc_ctrls.c
@@ -17,7 +17,6 @@
 #define SLICE_BYTE_SIZE_MAX    1024
 #define SLICE_BYTE_SIZE_MIN    1024
 #define SLICE_MB_SIZE_MAX      300
-#define INTRA_REFRESH_MBS_MAX  300
 #define AT_SLICE_BOUNDARY      \
        V4L2_MPEG_VIDEO_H264_LOOP_FILTER_MODE_DISABLED_AT_SLICE_BOUNDARY
 
@@ -224,8 +223,6 @@ static int venc_op_s_ctrl(struct v4l2_ctrl *ctrl)
                }
                mutex_unlock(&inst->lock);
                break;
-       case V4L2_CID_MPEG_VIDEO_CYCLIC_INTRA_REFRESH_MB:
-               break;
        case V4L2_CID_MPEG_VIDEO_GOP_SIZE:
                ret = venc_calc_bpframes(ctrl->val, ctr->num_b_frames, &bframes,
                                         &ctr->num_p_frames);
@@ -276,6 +273,9 @@ static int venc_op_s_ctrl(struct v4l2_ctrl *ctrl)
        case V4L2_CID_MPEG_VIDEO_BASELAYER_PRIORITY_ID:
                ctr->base_priority_id = ctrl->val;
                break;
+       case V4L2_CID_MPEG_VIDEO_INTRA_REFRESH_PERIOD:
+               ctr->intra_refresh_period = ctrl->val;
+               break;
        default:
                return -EINVAL;
        }
@@ -459,10 +459,6 @@ int venc_ctrl_init(struct venus_inst *inst)
        v4l2_ctrl_new_std(&inst->ctrl_handler, &venc_ctrl_ops,
                V4L2_CID_MPEG_VIDEO_H264_LOOP_FILTER_BETA, -6, 6, 1, 0);
 
-       v4l2_ctrl_new_std(&inst->ctrl_handler, &venc_ctrl_ops,
-               V4L2_CID_MPEG_VIDEO_CYCLIC_INTRA_REFRESH_MB,
-               0, INTRA_REFRESH_MBS_MAX, 1, 0);
-
        v4l2_ctrl_new_std(&inst->ctrl_handler, &venc_ctrl_ops,
                V4L2_CID_MPEG_VIDEO_GOP_SIZE, 0, (1 << 16) - 1, 1, 30);
 
@@ -497,6 +493,9 @@ int venc_ctrl_init(struct venus_inst *inst)
        v4l2_ctrl_new_std(&inst->ctrl_handler, &venc_ctrl_ops,
                          V4L2_CID_MPEG_VIDEO_BASELAYER_PRIORITY_ID, 0,
                          6, 1, 0);
+       v4l2_ctrl_new_std(&inst->ctrl_handler, &venc_ctrl_ops,
+                         V4L2_CID_MPEG_VIDEO_INTRA_REFRESH_PERIOD, 0,
+                         ((4096 * 2304) >> 8), 1, 0);
 
        ret = inst->ctrl_handler.error;
        if (ret)
-- 
2.25.1

Reply via email to