FaSTTUBe/GST-Tensordecoder-ov_ep
61
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

h264decoder: Use newly added avcC data parsing API

Browse Source 
Use gst_h264_parser_parse_decoder_config_record() method to parse
codec_data.

Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/2449>
This commit is contained in:
Seungha Yang 2022-05-19 04:49:36 +09:00 committed by GStreamer Marge Bot
parent 72975fbd6d
commit b48c8eaee6
1 changed files with 24 additions and 58 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

82
subprojects/gst-plugins-bad/gst-libs/gst/codecs/gsth264decoder.c
View File

@ -649,82 +649,48 @@ gst_h264_decoder_parse_codec_data (GstH264Decoder * self, const guint8 * data,
gsize size) gsize size)
{ {
GstH264DecoderPrivate *priv = self->priv; GstH264DecoderPrivate *priv = self->priv;
guint num_sps, num_pps; GstH264DecoderConfigRecord *config = NULL;
guint off;
gint i;
GstH264ParserResult pres;
GstH264NalUnit nalu;
GstFlowReturn ret = GST_FLOW_OK; GstFlowReturn ret = GST_FLOW_OK;
#ifndef GST_DISABLE_GST_DEBUG GstH264NalUnit *nalu;
guint profile; guint i;
#endif
/* parse the avcC data */ if (gst_h264_parser_parse_decoder_config_record (priv->parser, data, size,
if (size < 7) { /* when numSPS==0 and numPPS==0, length is 7 bytes */ &config) != GST_H264_PARSER_OK) {
GST_WARNING_OBJECT (self, "Failed to parse codec-data");
return GST_FLOW_ERROR; return GST_FLOW_ERROR;
} }
/* parse the version, this must be 1 */ priv->nal_length_size = config->length_size_minus_one + 1;
if (data[0] != 1) { for (i = 0; i < config->sps->len; i++) {
return GST_FLOW_ERROR; nalu = &g_array_index (config->sps, GstH264NalUnit, i);
}
#ifndef GST_DISABLE_GST_DEBUG
/* AVCProfileIndication */
/* profile_compat */
/* AVCLevelIndication */
profile = (data[1] << 16) | (data[2] << 8) | data[3];
GST_DEBUG_OBJECT (self, "profile %06x", profile);
#endif
/* 6 bits reserved | 2 bits lengthSizeMinusOne */ /* TODO: handle subset sps for SVC/MVC. That would need to be stored in
/* this is the number of bytes in front of the NAL units to mark their * separate array instead of putting SPS/subset-SPS into a single array */
* length */ if (nalu->type != GST_H264_NAL_SPS)
priv->nal_length_size = (data[4] & 0x03) + 1; continue;
GST_DEBUG_OBJECT (self, "nal length size %u", priv->nal_length_size);
num_sps = data[5] & 0x1f; ret = gst_h264_decoder_parse_sps (self, nalu);
off = 6;
for (i = 0; i < num_sps; i++) {
pres = gst_h264_parser_identify_nalu_avc (priv->parser,
data, off, size, 2, &nalu);
if (pres != GST_H264_PARSER_OK) {
GST_WARNING_OBJECT (self, "Failed to identify SPS nalu");
return GST_FLOW_ERROR;
}
ret = gst_h264_decoder_parse_sps (self, &nalu);
if (ret != GST_FLOW_OK) { if (ret != GST_FLOW_OK) {
GST_WARNING_OBJECT (self, "Failed to parse SPS"); GST_WARNING_OBJECT (self, "Failed to parse SPS");
return ret; goto out;
} }
off = nalu.offset + nalu.size;
} }
if (off >= size) { for (i = 0; i < config->pps->len; i++) {
GST_WARNING_OBJECT (self, "Too small avcC"); nalu = &g_array_index (config->pps, GstH264NalUnit, i);
return GST_FLOW_ERROR; if (nalu->type != GST_H264_NAL_PPS)
} continue;
num_pps = data[off]; ret = gst_h264_decoder_parse_pps (self, nalu);
off++;
for (i = 0; i < num_pps; i++) {
pres = gst_h264_parser_identify_nalu_avc (priv->parser,
data, off, size, 2, &nalu);
if (pres != GST_H264_PARSER_OK) {
GST_WARNING_OBJECT (self, "Failed to identify PPS nalu");
return GST_FLOW_ERROR;
}
ret = gst_h264_decoder_parse_pps (self, &nalu);
if (ret != GST_FLOW_OK) { if (ret != GST_FLOW_OK) {
GST_WARNING_OBJECT (self, "Failed to parse PPS"); GST_WARNING_OBJECT (self, "Failed to parse PPS");
return ret; goto out;
} }
off = nalu.offset + nalu.size;
} }
return GST_FLOW_OK; out:
gst_h264_decoder_config_record_free (config);
return ret;
} }
static gboolean static gboolean