[FFmpeg-devel] [PATCH v2] avcodec/libaomenc: Add unmet target level warning

[Bohan Li]

Another ping :)

On Fri, Apr 29, 2022 at 2:46 PM Bohan Li <bohanli at google.com> wrote:

> Gentle ping on this :)
>
> On Tue, Apr 19, 2022 at 11:20 AM Bohan Li <bohanli at google.com> wrote:
>
>> When target levels are set, this patch checks whether they are
>> satisfied by libaom. If not, a warning is shown. Otherwise the output
>> levels are also logged.
>>
>> This patch applies basically the same approach used for libvpx.
>>
>> Signed-off-by: Bohan Li <bohanli at google.com>
>> ---
>>  libavcodec/libaomenc.c | 64 ++++++++++++++++++++++++++++++++++++++++++
>>  1 file changed, 64 insertions(+)
>>
>> diff --git a/libavcodec/libaomenc.c b/libavcodec/libaomenc.c
>> index 054903e6e2..77be56fa51 100644
>> --- a/libavcodec/libaomenc.c
>> +++ b/libavcodec/libaomenc.c
>> @@ -198,6 +198,12 @@ static const char *const ctlidstr[] = {
>>      [AV1E_SET_ENABLE_SMOOTH_INTERINTRA] =
>> "AV1E_SET_ENABLE_SMOOTH_INTERINTRA",
>>      [AV1E_SET_ENABLE_REF_FRAME_MVS]     =
>> "AV1E_SET_ENABLE_REF_FRAME_MVS",
>>  #endif
>> +#ifdef AOM_CTRL_AV1E_GET_SEQ_LEVEL_IDX
>> +    [AV1E_GET_SEQ_LEVEL_IDX]            = "AV1E_GET_SEQ_LEVEL_IDX",
>> +#endif
>> +#ifdef AOM_CTRL_AV1E_GET_TARGET_SEQ_LEVEL_IDX
>> +    [AV1E_GET_TARGET_SEQ_LEVEL_IDX]     =
>> "AV1E_GET_TARGET_SEQ_LEVEL_IDX",
>> +#endif
>>  };
>>
>>  static av_cold void log_encoder_error(AVCodecContext *avctx, const char
>> *desc)
>> @@ -323,10 +329,68 @@ static av_cold int codecctl_int(AVCodecContext
>> *avctx,
>>      return 0;
>>  }
>>
>> +#if defined(AOM_CTRL_AV1E_GET_SEQ_LEVEL_IDX) && \
>> +    defined(AOM_CTRL_AV1E_GET_TARGET_SEQ_LEVEL_IDX)
>> +static av_cold int codecctl_intp(AVCodecContext *avctx,
>> +#ifdef UENUM1BYTE
>> +                                 aome_enc_control_id id,
>> +#else
>> +                                 enum aome_enc_control_id id,
>> +#endif
>> +                                 int* ptr)
>> +{
>> +    AOMContext *ctx = avctx->priv_data;
>> +    char buf[80];
>> +    int width = -30;
>> +    int res;
>> +
>> +    snprintf(buf, sizeof(buf), "%s:", ctlidstr[id]);
>> +    av_log(avctx, AV_LOG_DEBUG, "  %*s%d\n", width, buf, *ptr);
>> +
>> +    res = aom_codec_control(&ctx->encoder, id, ptr);
>> +    if (res != AOM_CODEC_OK) {
>> +        snprintf(buf, sizeof(buf), "Failed to set %s codec control",
>> +                 ctlidstr[id]);
>> +        log_encoder_error(avctx, buf);
>> +        return AVERROR(EINVAL);
>> +    }
>> +
>> +    return 0;
>> +}
>> +#endif
>> +
>>  static av_cold int aom_free(AVCodecContext *avctx)
>>  {
>>      AOMContext *ctx = avctx->priv_data;
>>
>> +#if defined(AOM_CTRL_AV1E_GET_SEQ_LEVEL_IDX) && \
>> +    defined(AOM_CTRL_AV1E_GET_TARGET_SEQ_LEVEL_IDX)
>> +    if (!(avctx->flags & AV_CODEC_FLAG_PASS1)) {
>> +        int levels[32] = { 0 };
>> +        int target_levels[32] = { 0 };
>> +
>> +        if (!codecctl_intp(avctx, AV1E_GET_SEQ_LEVEL_IDX, levels) &&
>> +            !codecctl_intp(avctx, AV1E_GET_TARGET_SEQ_LEVEL_IDX,
>> +                           target_levels)) {
>> +            for (int i = 0; i < 32; i++) {
>> +                if (levels[i] > target_levels[i]) {
>> +                    // Warn when the target level was not met
>> +                    av_log(avctx, AV_LOG_WARNING,
>> +                           "Could not encode to target level %d.%d for "
>> +                           "operating point %d. The output level is
>> %d.%d.\n",
>> +                           2 + (target_levels[i] >> 2), target_levels[i]
>> & 3,
>> +                           i, 2 + (levels[i] >> 2), levels[i] & 3);
>> +                } else if (target_levels[i] < 31) {
>> +                    // Log the encoded level if a target level was given
>> +                    av_log(avctx, AV_LOG_INFO,
>> +                           "Output level for operating point %d is
>> %d.%d.\n",
>> +                           i, 2 + (levels[i] >> 2), levels[i] & 3);
>> +                }
>> +            }
>> +        }
>> +    }
>> +#endif
>> +
>>      aom_codec_destroy(&ctx->encoder);
>>      av_freep(&ctx->twopass_stats.buf);
>>      av_freep(&avctx->stats_out);
>> --
>> 2.36.0.rc0.470.gd361397f0d-goog
>>
>>