mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-12-22 16:26:39 +00:00
25422b6766
Expose the main-444 profile to h265enc caps, when the upstream chooses to use VUYA as input, we choose main 4:4:4 profile to encode the frames.
419 lines
14 KiB
C
419 lines
14 KiB
C
/*
|
|
* gstvaapiutils_h265.c - H.265 related utilities
|
|
*
|
|
* Copyright (C) 2015 Intel Corporation
|
|
* Author: Sreerenj Balachandran <sreerenj.balachandran@intel.com>
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public License
|
|
* as published by the Free Software Foundation; either version 2.1
|
|
* of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free
|
|
* Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
|
|
* Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
#include "sysdeps.h"
|
|
#include <gst/codecparsers/gsth265parser.h>
|
|
#include "gstvaapicompat.h"
|
|
#include "gstvaapiutils_h265_priv.h"
|
|
|
|
#define DEBUG 1
|
|
#include "gstvaapidebug.h"
|
|
|
|
struct map
|
|
{
|
|
guint value;
|
|
const gchar *name;
|
|
};
|
|
|
|
/* Profile string map */
|
|
static const struct map gst_vaapi_h265_profile_map[] = {
|
|
/* *INDENT-OFF* */
|
|
{ GST_VAAPI_PROFILE_H265_MAIN, "main" },
|
|
{ GST_VAAPI_PROFILE_H265_MAIN10, "main-10" },
|
|
{ GST_VAAPI_PROFILE_H265_MAIN_STILL_PICTURE, "main-still-picture" },
|
|
{ GST_VAAPI_PROFILE_H265_MAIN_444, "main-444" },
|
|
{ 0, NULL }
|
|
/* *INDENT-ON* */
|
|
};
|
|
|
|
/* Tier string map */
|
|
static const struct map gst_vaapi_h265_tier_map[] = {
|
|
/* *INDENT-OFF* */
|
|
{ GST_VAAPI_TIER_H265_MAIN, "main" },
|
|
{ GST_VAAPI_TIER_H265_HIGH, "high"},
|
|
{ GST_VAAPI_TIER_H265_UNKNOWN, "unknown"}
|
|
/* *INDENT-ON* */
|
|
};
|
|
|
|
/* Level string map */
|
|
static const struct map gst_vaapi_h265_level_map[] = {
|
|
/* *INDENT-OFF* */
|
|
{ GST_VAAPI_LEVEL_H265_L1, "1" },
|
|
{ GST_VAAPI_LEVEL_H265_L2, "2" },
|
|
{ GST_VAAPI_LEVEL_H265_L2_1, "2.1" },
|
|
{ GST_VAAPI_LEVEL_H265_L3, "3" },
|
|
{ GST_VAAPI_LEVEL_H265_L3_1, "3.1" },
|
|
{ GST_VAAPI_LEVEL_H265_L4, "4" },
|
|
{ GST_VAAPI_LEVEL_H265_L4_1, "4.1" },
|
|
{ GST_VAAPI_LEVEL_H265_L5, "5" },
|
|
{ GST_VAAPI_LEVEL_H265_L5_1, "5.1" },
|
|
{ GST_VAAPI_LEVEL_H265_L5_2, "5.2" },
|
|
{ GST_VAAPI_LEVEL_H265_L6, "6" },
|
|
{ GST_VAAPI_LEVEL_H265_L6_1, "6.1" },
|
|
{ GST_VAAPI_LEVEL_H265_L6_2, "6.2" },
|
|
{ 0, NULL }
|
|
/* *INDENT-ON* */
|
|
};
|
|
|
|
/* Table A-1 - Level limits */
|
|
/* *INDENT-OFF* */
|
|
static const GstVaapiH265LevelLimits gst_vaapi_h265_level_limits[] = {
|
|
/* level idc MaxLumaPs MCPBMt MCPBHt MSlSeg MTR MTC MaxLumaSr MBRMt MBRHt MinCr*/
|
|
{ GST_VAAPI_LEVEL_H265_L1, 30, 36864, 350, 0, 16, 1, 1, 552960, 128, 0, 2},
|
|
{ GST_VAAPI_LEVEL_H265_L2, 60, 122880, 1500, 0, 16, 1, 1, 3686400, 1500, 0, 2},
|
|
{ GST_VAAPI_LEVEL_H265_L2_1, 63, 245760, 3000, 0, 20, 1, 1, 7372800, 3000, 0, 2},
|
|
{ GST_VAAPI_LEVEL_H265_L3, 90, 552960, 6000, 0, 30, 2, 2, 16588800, 6000, 0, 2},
|
|
{ GST_VAAPI_LEVEL_H265_L3_1, 93, 983040, 10000, 0, 40, 3, 3, 33177600, 10000, 0, 2},
|
|
{ GST_VAAPI_LEVEL_H265_L4, 120, 2228224, 12000, 30000, 75, 5, 5, 66846720, 12000, 30000, 4},
|
|
{ GST_VAAPI_LEVEL_H265_L4_1, 123, 2228224, 20000, 50000, 75, 5, 5, 133693440, 20000, 50000, 4},
|
|
{ GST_VAAPI_LEVEL_H265_L5, 150, 8912896, 25000, 100000, 200, 11, 10, 267386880, 25000, 100000, 6},
|
|
{ GST_VAAPI_LEVEL_H265_L5_1, 153, 8912896, 40000, 160000, 200, 11, 10, 534773760, 40000, 160000, 8},
|
|
{ GST_VAAPI_LEVEL_H265_L5_2, 156, 8912896, 60000, 240000, 200, 11, 10, 1069547520, 60000, 240000, 8},
|
|
{ GST_VAAPI_LEVEL_H265_L6, 180, 35651584, 60000, 240000, 600, 22, 20, 1069547520, 60000, 240000, 8},
|
|
{ GST_VAAPI_LEVEL_H265_L6_1, 183, 35651584, 120000, 480000, 600, 22, 20, 2139095040, 120000, 480000, 8},
|
|
{ GST_VAAPI_LEVEL_H265_L6_2, 186, 35651584, 240000, 800000, 600, 22, 20, 4278190080, 240000, 800000, 6},
|
|
{ 0, }
|
|
};
|
|
/* *INDENT-ON* */
|
|
|
|
/* Lookup value in map */
|
|
static const struct map *
|
|
map_lookup_value (const struct map *m, guint value)
|
|
{
|
|
g_return_val_if_fail (m != NULL, NULL);
|
|
|
|
for (; m->name != NULL; m++) {
|
|
if (m->value == value)
|
|
return m;
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
/* Lookup name in map */
|
|
static const struct map *
|
|
map_lookup_name (const struct map *m, const gchar * name)
|
|
{
|
|
g_return_val_if_fail (m != NULL, NULL);
|
|
|
|
if (!name)
|
|
return NULL;
|
|
|
|
for (; m->name != NULL; m++) {
|
|
if (strcmp (m->name, name) == 0)
|
|
return m;
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
/** Returns a relative score for the supplied GstVaapiProfile */
|
|
guint
|
|
gst_vaapi_utils_h265_get_profile_score (GstVaapiProfile profile)
|
|
{
|
|
const struct map *const m =
|
|
map_lookup_value (gst_vaapi_h265_profile_map, profile);
|
|
|
|
return m ? 1 + (m - gst_vaapi_h265_profile_map) : 0;
|
|
}
|
|
|
|
/** Returns GstVaapiProfile from H.265 profile_idc value */
|
|
GstVaapiProfile
|
|
gst_vaapi_utils_h265_get_profile (GstH265SPS * sps)
|
|
{
|
|
GstVaapiProfile profile;
|
|
|
|
switch (sps->profile_tier_level.profile_idc) {
|
|
case GST_H265_PROFILE_IDC_MAIN:
|
|
profile = GST_VAAPI_PROFILE_H265_MAIN;
|
|
break;
|
|
case GST_H265_PROFILE_IDC_MAIN_10:
|
|
profile = GST_VAAPI_PROFILE_H265_MAIN10;
|
|
break;
|
|
case GST_H265_PROFILE_IDC_MAIN_STILL_PICTURE:
|
|
profile = GST_VAAPI_PROFILE_H265_MAIN_STILL_PICTURE;
|
|
break;
|
|
case GST_H265_PROFILE_IDC_FORMAT_RANGE_EXTENSION:
|
|
if (sps->profile_tier_level.max_12bit_constraint_flag == 1
|
|
&& sps->profile_tier_level.max_10bit_constraint_flag == 1
|
|
&& sps->profile_tier_level.max_8bit_constraint_flag == 0
|
|
&& sps->profile_tier_level.max_422chroma_constraint_flag == 1
|
|
&& sps->profile_tier_level.max_420chroma_constraint_flag == 0
|
|
&& sps->profile_tier_level.max_monochrome_constraint_flag == 0
|
|
&& sps->profile_tier_level.intra_constraint_flag == 0
|
|
&& sps->profile_tier_level.one_picture_only_constraint_flag == 0
|
|
&& sps->profile_tier_level.lower_bit_rate_constraint_flag == 1) {
|
|
profile = GST_VAAPI_PROFILE_H265_MAIN_422_10;
|
|
break;
|
|
} else if (sps->profile_tier_level.max_12bit_constraint_flag == 1
|
|
&& sps->profile_tier_level.max_10bit_constraint_flag == 1
|
|
&& sps->profile_tier_level.max_8bit_constraint_flag == 1
|
|
&& sps->profile_tier_level.max_422chroma_constraint_flag == 0
|
|
&& sps->profile_tier_level.max_420chroma_constraint_flag == 0
|
|
&& sps->profile_tier_level.max_monochrome_constraint_flag == 0
|
|
&& sps->profile_tier_level.intra_constraint_flag == 0
|
|
&& sps->profile_tier_level.one_picture_only_constraint_flag == 0
|
|
&& sps->profile_tier_level.lower_bit_rate_constraint_flag == 1) {
|
|
profile = GST_VAAPI_PROFILE_H265_MAIN_444;
|
|
break;
|
|
} else if (sps->profile_tier_level.max_12bit_constraint_flag == 1
|
|
&& sps->profile_tier_level.max_10bit_constraint_flag == 1
|
|
&& sps->profile_tier_level.max_8bit_constraint_flag == 0
|
|
&& sps->profile_tier_level.max_422chroma_constraint_flag == 0
|
|
&& sps->profile_tier_level.max_420chroma_constraint_flag == 0
|
|
&& sps->profile_tier_level.max_monochrome_constraint_flag == 0
|
|
&& sps->profile_tier_level.intra_constraint_flag == 0
|
|
&& sps->profile_tier_level.one_picture_only_constraint_flag == 0
|
|
&& sps->profile_tier_level.lower_bit_rate_constraint_flag == 1) {
|
|
profile = GST_VAAPI_PROFILE_H265_MAIN_444_10;
|
|
break;
|
|
}
|
|
default:
|
|
GST_DEBUG ("unsupported profile_idc value");
|
|
profile = GST_VAAPI_PROFILE_UNKNOWN;
|
|
break;
|
|
}
|
|
return profile;
|
|
}
|
|
|
|
/** Returns H.265 profile_idc value from GstVaapiProfile */
|
|
guint8
|
|
gst_vaapi_utils_h265_get_profile_idc (GstVaapiProfile profile)
|
|
{
|
|
guint8 profile_idc;
|
|
|
|
switch (profile) {
|
|
case GST_VAAPI_PROFILE_H265_MAIN:
|
|
profile_idc = GST_H265_PROFILE_IDC_MAIN;
|
|
break;
|
|
case GST_VAAPI_PROFILE_H265_MAIN10:
|
|
profile_idc = GST_H265_PROFILE_IDC_MAIN_10;
|
|
break;
|
|
case GST_VAAPI_PROFILE_H265_MAIN_STILL_PICTURE:
|
|
profile_idc = GST_H265_PROFILE_IDC_MAIN_STILL_PICTURE;
|
|
break;
|
|
case GST_VAAPI_PROFILE_H265_MAIN_422_10:
|
|
/* Fall through */
|
|
case GST_VAAPI_PROFILE_H265_MAIN_444:
|
|
profile_idc = GST_H265_PROFILE_IDC_FORMAT_RANGE_EXTENSION;
|
|
break;
|
|
default:
|
|
GST_DEBUG ("unsupported GstVaapiProfile value");
|
|
profile_idc = 0;
|
|
break;
|
|
}
|
|
return profile_idc;
|
|
}
|
|
|
|
/** Returns GstVaapiProfile from a string representation */
|
|
GstVaapiProfile
|
|
gst_vaapi_utils_h265_get_profile_from_string (const gchar * str)
|
|
{
|
|
const struct map *const m = map_lookup_name (gst_vaapi_h265_profile_map, str);
|
|
|
|
return m ? (GstVaapiProfile) m->value : GST_VAAPI_PROFILE_UNKNOWN;
|
|
}
|
|
|
|
/** Returns a string representation for the supplied H.265 profile */
|
|
const gchar *
|
|
gst_vaapi_utils_h265_get_profile_string (GstVaapiProfile profile)
|
|
{
|
|
const struct map *const m =
|
|
map_lookup_value (gst_vaapi_h265_profile_map, profile);
|
|
|
|
return m ? m->name : NULL;
|
|
}
|
|
|
|
/** Returns GstVaapiLevelH265 from H.265 level_idc value */
|
|
GstVaapiLevelH265
|
|
gst_vaapi_utils_h265_get_level (guint8 level_idc)
|
|
{
|
|
const GstVaapiH265LevelLimits *llp;
|
|
|
|
for (llp = gst_vaapi_h265_level_limits; llp->level != 0; llp++) {
|
|
if (llp->level_idc == level_idc)
|
|
return llp->level;
|
|
}
|
|
GST_DEBUG ("unsupported level_idc value");
|
|
return (GstVaapiLevelH265) 0;
|
|
}
|
|
|
|
/** Returns H.265 level_idc value from GstVaapiLevelH265 */
|
|
guint8
|
|
gst_vaapi_utils_h265_get_level_idc (GstVaapiLevelH265 level)
|
|
{
|
|
const GstVaapiH265LevelLimits *const llp =
|
|
gst_vaapi_utils_h265_get_level_limits (level);
|
|
|
|
return llp ? llp->level_idc : 0;
|
|
}
|
|
|
|
/** Returns GstVaapiLevelH265 from a string representation */
|
|
GstVaapiLevelH265
|
|
gst_vaapi_utils_h265_get_level_from_string (const gchar * str)
|
|
{
|
|
gint v, level_idc = 0;
|
|
|
|
if (!str || !str[0])
|
|
goto not_found;
|
|
|
|
v = g_ascii_digit_value (str[0]);
|
|
if (v < 0)
|
|
goto not_found;
|
|
level_idc = v * 30;
|
|
|
|
switch (str[1]) {
|
|
case '\0':
|
|
break;
|
|
case '.':
|
|
v = g_ascii_digit_value (str[2]);
|
|
if (v < 0 || str[3] != '\0')
|
|
goto not_found;
|
|
level_idc += v;
|
|
break;
|
|
default:
|
|
goto not_found;
|
|
}
|
|
return gst_vaapi_utils_h265_get_level (level_idc);
|
|
|
|
not_found:
|
|
return (GstVaapiLevelH265) 0;
|
|
}
|
|
|
|
/** Returns a string representation for the supplied H.265 level */
|
|
const gchar *
|
|
gst_vaapi_utils_h265_get_level_string (GstVaapiLevelH265 level)
|
|
{
|
|
if (level < GST_VAAPI_LEVEL_H265_L1 || level > GST_VAAPI_LEVEL_H265_L6_2)
|
|
return NULL;
|
|
return gst_vaapi_h265_level_map[level - GST_VAAPI_LEVEL_H265_L1].name;
|
|
}
|
|
|
|
/** Returns level limits as specified in Table A-1 of the H.265 standard */
|
|
const GstVaapiH265LevelLimits *
|
|
gst_vaapi_utils_h265_get_level_limits (GstVaapiLevelH265 level)
|
|
{
|
|
if (level < GST_VAAPI_LEVEL_H265_L1 || level > GST_VAAPI_LEVEL_H265_L6_2)
|
|
return NULL;
|
|
return &gst_vaapi_h265_level_limits[level - GST_VAAPI_LEVEL_H265_L1];
|
|
}
|
|
|
|
/** Returns the Table A-1 & A-2 specification */
|
|
const GstVaapiH265LevelLimits *
|
|
gst_vaapi_utils_h265_get_level_limits_table (guint * out_length_ptr)
|
|
{
|
|
if (out_length_ptr)
|
|
*out_length_ptr = G_N_ELEMENTS (gst_vaapi_h265_level_limits) - 1;
|
|
return gst_vaapi_h265_level_limits;
|
|
}
|
|
|
|
/** Returns GstVaapiChromaType from H.265 chroma_format_idc value */
|
|
GstVaapiChromaType
|
|
gst_vaapi_utils_h265_get_chroma_type (guint chroma_format_idc,
|
|
guint luma_bit_depth, guint chroma_bit_depth)
|
|
{
|
|
GstVaapiChromaType chroma_type = (GstVaapiChromaType) 0;
|
|
guint depth = 0;
|
|
|
|
if (luma_bit_depth < 8 || chroma_bit_depth < 8 ||
|
|
luma_bit_depth > 16 || chroma_bit_depth > 16) {
|
|
GST_WARNING ("invalid luma_bit_depth or chroma_bit_depth value");
|
|
return chroma_type;
|
|
}
|
|
|
|
depth = MAX (luma_bit_depth, chroma_bit_depth);
|
|
|
|
switch (chroma_format_idc) {
|
|
case 0:
|
|
chroma_type = GST_VAAPI_CHROMA_TYPE_YUV400;
|
|
break;
|
|
case 1:
|
|
if (depth == 8)
|
|
chroma_type = GST_VAAPI_CHROMA_TYPE_YUV420;
|
|
else if (depth > 8 && depth <= 10)
|
|
chroma_type = GST_VAAPI_CHROMA_TYPE_YUV420_10BPP;
|
|
break;
|
|
case 2:
|
|
if (depth == 8)
|
|
chroma_type = GST_VAAPI_CHROMA_TYPE_YUV422;
|
|
else if (depth > 8 && depth <= 10)
|
|
chroma_type = GST_VAAPI_CHROMA_TYPE_YUV422_10BPP;
|
|
break;
|
|
case 3:
|
|
if (depth == 8)
|
|
chroma_type = GST_VAAPI_CHROMA_TYPE_YUV444;
|
|
else if (depth > 8 && depth <= 10)
|
|
chroma_type = GST_VAAPI_CHROMA_TYPE_YUV444_10BPP;
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
|
|
if (chroma_type == (GstVaapiChromaType) 0)
|
|
GST_DEBUG ("unsupported chroma_format_idc value");
|
|
|
|
return chroma_type;
|
|
}
|
|
|
|
/** Returns H.265 chroma_format_idc value from GstVaapiChromaType */
|
|
guint
|
|
gst_vaapi_utils_h265_get_chroma_format_idc (GstVaapiChromaType chroma_type)
|
|
{
|
|
guint chroma_format_idc;
|
|
|
|
switch (chroma_type) {
|
|
case GST_VAAPI_CHROMA_TYPE_YUV400:
|
|
chroma_format_idc = 0;
|
|
break;
|
|
case GST_VAAPI_CHROMA_TYPE_YUV420:
|
|
case GST_VAAPI_CHROMA_TYPE_YUV420_10BPP:
|
|
chroma_format_idc = 1;
|
|
break;
|
|
case GST_VAAPI_CHROMA_TYPE_YUV422:
|
|
chroma_format_idc = 2;
|
|
break;
|
|
case GST_VAAPI_CHROMA_TYPE_YUV444:
|
|
chroma_format_idc = 3;
|
|
break;
|
|
default:
|
|
GST_DEBUG ("unsupported GstVaapiChromaType value");
|
|
chroma_format_idc = 1;
|
|
break;
|
|
}
|
|
return chroma_format_idc;
|
|
}
|
|
|
|
/** Returns GstVaapiTierH265 from a string representation */
|
|
GstVaapiTierH265
|
|
gst_vaapi_utils_h265_get_tier_from_string (const gchar * str)
|
|
{
|
|
const struct map *const m = map_lookup_name (gst_vaapi_h265_tier_map, str);
|
|
|
|
return m ? (GstVaapiTierH265) m->value : GST_VAAPI_TIER_H265_UNKNOWN;
|
|
}
|
|
|
|
/** Returns a string representation for the supplied H.265 tier */
|
|
const gchar *
|
|
gst_vaapi_utils_h265_get_tier_string (GstVaapiTierH265 tier)
|
|
{
|
|
const struct map *const m = map_lookup_value (gst_vaapi_h265_tier_map, tier);
|
|
|
|
return m ? m->name : NULL;
|
|
}
|