[FFmpeg-devel] [PATCH 1/2] lavc/h265_profile_level: Expand profile compatibility checking

Xiang, Haihao haihao.xiang at intel.com
Wed Apr 24 16:48:27 EEST 2024


On Ma, 2024-04-22 at 22:22 +0100, Mark Thompson wrote:
> Replace existing get_profile() with find_profile(), which finds the
> lowest compatible profile rather than requiring an exact match.
> ---
>  libavcodec/h265_profile_level.c | 73 +++++++++++++++++++++------------
>  libavcodec/h265_profile_level.h | 70 ++++++++++++++++++++++++++++++-
>  libavcodec/vaapi_hevc.c         |  2 +-
>  libavcodec/vdpau_hevc.c         |  2 +-
>  4 files changed, 117 insertions(+), 30 deletions(-)
> 
> diff --git a/libavcodec/h265_profile_level.c b/libavcodec/h265_profile_level.c
> index 7ff9681f65..4bc72414cb 100644
> --- a/libavcodec/h265_profile_level.c
> +++ b/libavcodec/h265_profile_level.c
> @@ -119,41 +119,60 @@ static const H265ProfileDescriptor h265_profiles[] = {
>        5, 1, 0, 0, 0, 0, 0, 0, 0, 1, 0, 2, 4000, 4400, 6.000, 0.5, 6 },
>  };
> 
> +_Static_assert(H265_PROFILE_COUNT == FF_ARRAY_ELEMS(h265_profiles),
> +               "Incorrect H.265 profiles table.");
> 
> -const H265ProfileDescriptor *ff_h265_get_profile(const
> H265RawProfileTierLevel *ptl)
> +
> +const int ff_h265_profile_compatible(const H265RawProfileTierLevel *ptl,
> +                                     int profile)
>  {
> -    int i;
> +    const H265ProfileDescriptor *desc;
> +
> +    av_assert0(profile >= 0 && profile < H265_PROFILE_COUNT);
> 
>      if (ptl->general_profile_space)
> -        return NULL;
> +        return 0;
> 
> -    for (i = 0; i < FF_ARRAY_ELEMS(h265_profiles); i++) {
> -        const H265ProfileDescriptor *profile = &h265_profiles[i];
> +    desc = &h265_profiles[profile];
> 
> -        if (ptl->general_profile_idc &&
> -            ptl->general_profile_idc != profile->profile_idc)
> -            continue;
> -        if (!ptl->general_profile_compatibility_flag[profile->profile_idc])
> -            continue;
> +    if (ptl->general_profile_idc &&
> +        ptl->general_profile_idc != desc->profile_idc)
> +        return 0;
> +    if (!ptl->general_profile_compatibility_flag[desc->profile_idc])
> +        return 0;
> 
> -#define check_flag(name) \
> -        if (profile->name < 2) { \
> -            if (profile->name != ptl->general_ ## name ## _constraint_flag) \
> -                continue; \
> -        }
> -        check_flag(max_14bit);
> -        check_flag(max_12bit);
> -        check_flag(max_10bit);
> -        check_flag(max_8bit);
> -        check_flag(max_422chroma);
> -        check_flag(max_420chroma);
> -        check_flag(max_monochrome);
> -        check_flag(intra);
> -        check_flag(one_picture_only);
> -        check_flag(lower_bit_rate);
> +#define check_flag(flag) \
> +    if (desc->flag < 2 && \
> +        desc->flag > ptl->general_ ## flag ## _constraint_flag) \
> +        return 0;
> +    check_flag(max_14bit);
> +    check_flag(max_12bit);
> +    check_flag(max_10bit);
> +    check_flag(max_8bit);
> +    check_flag(max_422chroma);
> +    check_flag(max_420chroma);
> +    check_flag(max_monochrome);
> +    check_flag(intra);
> +    check_flag(one_picture_only);
> +    check_flag(lower_bit_rate);
>  #undef check_flag
> 
> -        return profile;
> +    return 1;
> +}
> +
> +
> +const H265ProfileDescriptor *ff_h265_get_profile(int profile)
> +{
> +    av_assert0(profile >= 0 && profile < H265_PROFILE_COUNT);
> +
> +    return &h265_profiles[profile];
> +}
> +
> +const H265ProfileDescriptor *ff_h265_find_profile(const
> H265RawProfileTierLevel *ptl)
> +{
> +    for (int p = 0; p < H265_PROFILE_COUNT; p++) {
> +        if (ff_h265_profile_compatible(ptl, p))
> +            return &h265_profiles[p];
>      }
> 
>      return NULL;
> @@ -171,7 +190,7 @@ const H265LevelDescriptor *ff_h265_guess_level(const
> H265RawProfileTierLevel *pt
>      int i;
> 
>      if (ptl)
> -        profile = ff_h265_get_profile(ptl);
> +        profile = ff_h265_find_profile(ptl);
>      else
>          profile = NULL;
>      if (!profile) {
> diff --git a/libavcodec/h265_profile_level.h b/libavcodec/h265_profile_level.h
> index cd30ac5c50..f403f63211 100644
> --- a/libavcodec/h265_profile_level.h
> +++ b/libavcodec/h265_profile_level.h
> @@ -24,6 +24,49 @@
>  #include "cbs_h265.h"
> 
> 
> +// Enumeration of all H.265 profiles.
> +// The list is ordered to match table A.10; numeric values are an index
> +// into the latest version of this table and have no codec meaning.
> +enum {
> +    H265_PROFILE_MONOCHROME,
> +    H265_PROFILE_MONOCHROME_10,
> +    H265_PROFILE_MONOCHROME_12,
> +    H265_PROFILE_MONOCHROME_16,
> +    H265_PROFILE_MAIN,
> +    H265_PROFILE_SCREEN_EXTENDED_MAIN,
> +    H265_PROFILE_MAIN_10,
> +    H265_PROFILE_SCREEN_EXTENDED_MAIN_10,
> +    H265_PROFILE_MAIN_12,
> +    H265_PROFILE_MAIN_STILL_PICTURE,
> +    H265_PROFILE_MAIN_10_STILL_PICTURE,
> +    H265_PROFILE_MAIN_422_10,
> +    H265_PROFILE_MAIN_422_12,
> +    H265_PROFILE_MAIN_444,
> +    H265_PROFILE_HIGH_THROUGHPUT_444,
> +    H265_PROFILE_SCREEN_EXTENDED_MAIN_444,
> +    H265_PROFILE_SCREEN_EXTENDED_HIGH_THROUGHPUT_444,
> +    H265_PROFILE_MAIN_444_10,
> +    H265_PROFILE_HIGH_THROUGHPUT_444_10,
> +    H265_PROFILE_SCREEN_EXTENDED_MAIN_444_10,
> +    H265_PROFILE_SCREEN_EXTENDED_HIGH_THROUGHPUT_444_10,
> +    H265_PROFILE_MAIN_444_12,
> +    H265_PROFILE_HIGH_THROUGHPUT_444_14,
> +    H265_PROFILE_SCREEN_EXTENDED_HIGH_THROUGHPUT_444_14,
> +    H265_PROFILE_MAIN_INTRA,
> +    H265_PROFILE_MAIN_10_INTRA,
> +    H265_PROFILE_MAIN_12_INTRA,
> +    H265_PROFILE_MAIN_422_10_INTRA,
> +    H265_PROFILE_MAIN_422_12_INTRA,
> +    H265_PROFILE_MAIN_444_INTRA,
> +    H265_PROFILE_MAIN_444_10_INTRA,
> +    H265_PROFILE_MAIN_444_12_INTRA,
> +    H265_PROFILE_MAIN_444_16_INTRA,
> +    H265_PROFILE_MAIN_444_STILL_PICTURE,
> +    H265_PROFILE_MAIN_444_16_STILL_PICTURE,
> +    H265_PROFILE_HIGH_THROUGHPUT_444_16_INTRA,
> +    H265_PROFILE_COUNT,
> +};
> +
>  typedef struct H265LevelDescriptor {
>      char        name[4];    // Large enough for all current levels like "4.1"
>      uint8_t     level_idc;
> @@ -70,7 +113,32 @@ typedef struct H265ProfileDescriptor {
>  } H265ProfileDescriptor;
> 
> 
> -const H265ProfileDescriptor *ff_h265_get_profile(const
> H265RawProfileTierLevel *ptl);
> +/**
> + * Test whether the PTL structure is compatible with the given profile.
> + *
> + * @param ptl      PTL structure to check.
> + * @param profile  Profile to test compatibility with.  Must be a valid
> + *                 H265_PROFILE_* value.
> + * @return  Nonzero if compatible, zero if incompatible.
> + */
> +const int ff_h265_profile_compatible(const H265RawProfileTierLevel *ptl,
> +                                     int profile);
> +
> +/**
> + * Return profile descriptor for the given profile.
> + *
> + * @param profile  Profile number; must be a valid H265_PROFILE_* value.
> + * @return  Profile descriptor.
> + */
> +const H265ProfileDescriptor *ff_h265_get_profile(int profile);

ff_h265_get_profile is called in libavcodec/tests/h265_levels.c which should be
changed too.

libavcodec/tests/h265_levels.c:        profile = ff_h265_get_profile(test_bitrate[i].ptl);

Thanks
Haihao

> +
> +/**
> + * Find the first profile compatible with the given PTL structure.
> + *
> + * @param ptl  PTL structure to search for.
> + * @return  First compatible profile, or NULL if no compatible profiles.
> + */
> +const H265ProfileDescriptor *ff_h265_find_profile(const
> H265RawProfileTierLevel *ptl);
> 
> 
>  /**
> diff --git a/libavcodec/vaapi_hevc.c b/libavcodec/vaapi_hevc.c
> index 3bdd2dd1b8..77f55ff8b1 100644
> --- a/libavcodec/vaapi_hevc.c
> +++ b/libavcodec/vaapi_hevc.c
> @@ -604,7 +604,7 @@ VAProfile
> ff_vaapi_parse_hevc_rext_scc_profile(AVCodecContext *avctx)
>      /* convert PTLCommon to H265RawProfileTierLevel */
>      ptl_convert(general_ptl, &h265_raw_ptl);
> 
> -    profile = ff_h265_get_profile(&h265_raw_ptl);
> +    profile = ff_h265_find_profile(&h265_raw_ptl);
>      if (!profile) {
>          av_log(avctx, AV_LOG_WARNING, "HEVC profile is not found.\n");
>          goto end;
> diff --git a/libavcodec/vdpau_hevc.c b/libavcodec/vdpau_hevc.c
> index 4cd7ce5621..a59a030b1c 100644
> --- a/libavcodec/vdpau_hevc.c
> +++ b/libavcodec/vdpau_hevc.c
> @@ -478,7 +478,7 @@ static int vdpau_hevc_parse_rext_profile(AVCodecContext
> *avctx, VdpDecoderProfil
>      /* convert PTLCommon to H265RawProfileTierLevel */
>      ptl_convert(general_ptl, &h265_raw_ptl);
> 
> -    profile = ff_h265_get_profile(&h265_raw_ptl);
> +    profile = ff_h265_find_profile(&h265_raw_ptl);
>      if (!profile) {
>          av_log(avctx, AV_LOG_WARNING, "HEVC profile is not found.\n");
>          if (avctx->hwaccel_flags & AV_HWACCEL_FLAG_ALLOW_PROFILE_MISMATCH) {



More information about the ffmpeg-devel mailing list