mirrors/gstreamer
mirrors/gstreamer
1
1
Fork 0
mirror of https://gitlab.freedesktop.org/gstreamer/gstreamer.git synced 2025-04-26 06:54:49 +00:00
Code Issues 1 Projects Releases Wiki Activity

vc1: handle encapsulated bitstreams.

Browse source
This commit is contained in:
Gwenole Beauchesne 2011-10-05 15:56:36 +02:00
parent 5aea02a66f
commit 032486912f
1 changed files with 119 additions and 34 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

153
gst-libs/gst/vaapi/gstvaapidecoder_vc1.c
View file

@ -60,6 +60,8 @@ struct _GstVaapiDecoderVC1Private {
GstVaapiPicture *prev_picture; GstVaapiPicture *prev_picture;
GstVaapiTSB *tsb; GstVaapiTSB *tsb;
GstBuffer *sub_buffer; GstBuffer *sub_buffer;
guint8 *rbdu_buffer;
guint rbdu_buffer_size;
guint is_constructed : 1; guint is_constructed : 1;
guint is_opened : 1; guint is_opened : 1;
guint is_first_field : 1; guint is_first_field : 1;
@ -152,6 +154,12 @@ gst_vaapi_decoder_vc1_destroy(GstVaapiDecoderVC1 *decoder)
GstVaapiDecoderVC1Private * const priv = decoder->priv; GstVaapiDecoderVC1Private * const priv = decoder->priv;
gst_vaapi_decoder_vc1_close(decoder); gst_vaapi_decoder_vc1_close(decoder);
if (priv->rbdu_buffer) {
g_free(priv->rbdu_buffer);
priv->rbdu_buffer = NULL;
priv->rbdu_buffer_size = 0;
}
} }
static gboolean static gboolean
@ -243,7 +251,7 @@ decode_current_picture(GstVaapiDecoderVC1 *decoder)
} }
static GstVaapiDecoderStatus static GstVaapiDecoderStatus
decode_sequence(GstVaapiDecoderVC1 *decoder, guchar *buf, guint buf_size) decode_sequence(GstVaapiDecoderVC1 *decoder, GstVC1BDU *rbdu, GstVC1BDU *ebdu)
{ {
GstVaapiDecoder * const base_decoder = GST_VAAPI_DECODER(decoder); GstVaapiDecoder * const base_decoder = GST_VAAPI_DECODER(decoder);
GstVaapiDecoderVC1Private * const priv = decoder->priv; GstVaapiDecoderVC1Private * const priv = decoder->priv;
@ -254,7 +262,11 @@ decode_sequence(GstVaapiDecoderVC1 *decoder, guchar *buf, guint buf_size)
GstVaapiProfile profile; GstVaapiProfile profile;
guint width, height, fps_n, fps_d; guint width, height, fps_n, fps_d;
result = gst_vc1_parse_sequence_header(buf, buf_size, seq_hdr); result = gst_vc1_parse_sequence_header(
rbdu->data + rbdu->offset,
rbdu->size,
seq_hdr
);
if (result != GST_VC1_PARSER_OK) { if (result != GST_VC1_PARSER_OK) {
GST_DEBUG("failed to parse sequence layer"); GST_DEBUG("failed to parse sequence layer");
return get_status(result); return get_status(result);
@ -402,7 +414,7 @@ decode_sequence_end(GstVaapiDecoderVC1 *decoder)
} }
static GstVaapiDecoderStatus static GstVaapiDecoderStatus
decode_entry_point(GstVaapiDecoderVC1 *decoder, guchar *buf, guint buf_size) decode_entry_point(GstVaapiDecoderVC1 *decoder, GstVC1BDU *rbdu, GstVC1BDU *ebdu)
{ {
GstVaapiDecoderVC1Private * const priv = decoder->priv; GstVaapiDecoderVC1Private * const priv = decoder->priv;
GstVC1SeqHdr * const seq_hdr = &priv->seq_hdr; GstVC1SeqHdr * const seq_hdr = &priv->seq_hdr;
@ -410,7 +422,8 @@ decode_entry_point(GstVaapiDecoderVC1 *decoder, guchar *buf, guint buf_size)
GstVC1ParserResult result; GstVC1ParserResult result;
result = gst_vc1_parse_entry_point_header( result = gst_vc1_parse_entry_point_header(
buf, buf_size, rbdu->data + rbdu->offset,
rbdu->size,
entrypoint_hdr, entrypoint_hdr,
seq_hdr seq_hdr
); );
@ -833,7 +846,7 @@ fill_picture(GstVaapiDecoderVC1 *decoder, GstVaapiPicture *picture)
} }
static GstVaapiDecoderStatus static GstVaapiDecoderStatus
decode_frame(GstVaapiDecoderVC1 *decoder, guchar *buf, guint buf_size) decode_frame(GstVaapiDecoderVC1 *decoder, GstVC1BDU *rbdu, GstVC1BDU *ebdu)
{ {
GstVaapiDecoder * const base_decoder = GST_VAAPI_DECODER(decoder); GstVaapiDecoder * const base_decoder = GST_VAAPI_DECODER(decoder);
GstVaapiDecoderVC1Private * const priv = decoder->priv; GstVaapiDecoderVC1Private * const priv = decoder->priv;
@ -871,7 +884,8 @@ decode_frame(GstVaapiDecoderVC1 *decoder, guchar *buf, guint buf_size)
memset(frame_hdr, 0, sizeof(*frame_hdr)); memset(frame_hdr, 0, sizeof(*frame_hdr));
result = gst_vc1_parse_frame_header( result = gst_vc1_parse_frame_header(
buf, buf_size, rbdu->data + rbdu->offset,
rbdu->size,
frame_hdr, frame_hdr,
seq_hdr, seq_hdr,
priv->bitplanes priv->bitplanes
@ -921,7 +935,12 @@ decode_frame(GstVaapiDecoderVC1 *decoder, guchar *buf, guint buf_size)
if (!fill_picture(decoder, picture)) if (!fill_picture(decoder, picture))
return GST_VAAPI_DECODER_STATUS_ERROR_UNKNOWN; return GST_VAAPI_DECODER_STATUS_ERROR_UNKNOWN;
slice = gst_vaapi_decoder_new_slice(base_decoder, picture, buf, buf_size); slice = gst_vaapi_decoder_new_slice(
base_decoder,
picture,
ebdu->data + ebdu->sc_offset,
ebdu->size + ebdu->offset - ebdu->sc_offset
);
if (!slice) { if (!slice) {
GST_DEBUG("failed to allocate slice"); GST_DEBUG("failed to allocate slice");
return GST_VAAPI_DECODER_STATUS_ERROR_ALLOCATION_FAILED; return GST_VAAPI_DECODER_STATUS_ERROR_ALLOCATION_FAILED;
@ -929,22 +948,107 @@ decode_frame(GstVaapiDecoderVC1 *decoder, guchar *buf, guint buf_size)
/* Fill in VASliceParameterBufferVC1 */ /* Fill in VASliceParameterBufferVC1 */
slice_param = slice->param; slice_param = slice->param;
slice_param->macroblock_offset = frame_hdr->header_size; slice_param->macroblock_offset = 8 * (ebdu->offset - ebdu->sc_offset) + frame_hdr->header_size;
slice_param->slice_vertical_position = 0; slice_param->slice_vertical_position = 0;
/* Decode picture right away, we got the full frame */ /* Decode picture right away, we got the full frame */
return decode_current_picture(decoder); return decode_current_picture(decoder);
} }
static gboolean
decode_rbdu(GstVaapiDecoderVC1 *decoder, GstVC1BDU *rbdu, GstVC1BDU *ebdu)
{
GstVaapiDecoderVC1Private * const priv = decoder->priv;
guint8 *rbdu_buffer;
guint i, j, rbdu_buffer_size;
/* BDU are encapsulated in advanced profile mode only */
if (priv->profile != GST_VAAPI_PROFILE_VC1_ADVANCED) {
memcpy(rbdu, ebdu, sizeof(*rbdu));
return TRUE;
}
/* Reallocate unescaped bitstream buffer */
rbdu_buffer = priv->rbdu_buffer;
if (!rbdu_buffer || ebdu->size > priv->rbdu_buffer_size) {
rbdu_buffer = g_realloc(priv->rbdu_buffer, ebdu->size);
if (!rbdu_buffer)
return FALSE;
priv->rbdu_buffer = rbdu_buffer;
priv->rbdu_buffer_size = ebdu->size;
}
/* Unescape bitstream buffer */
if (ebdu->size < 4) {
memcpy(rbdu_buffer, ebdu->data + ebdu->offset, ebdu->size);
rbdu_buffer_size = ebdu->size;
}
else {
guint8 * const bdu_buffer = ebdu->data + ebdu->offset;
for (i = 0, j = 0; i < ebdu->size; i++) {
if (i >= 2 && i < ebdu->size - 1 &&
bdu_buffer[i - 1] == 0x00 &&
bdu_buffer[i - 2] == 0x00 &&
bdu_buffer[i ] == 0x03 &&
bdu_buffer[i + 1] <= 0x03)
i++;
rbdu_buffer[j++] = bdu_buffer[i];
}
rbdu_buffer_size = j;
}
/* Reconstruct RBDU */
rbdu->type = ebdu->type;
rbdu->size = rbdu_buffer_size;
rbdu->sc_offset = 0;
rbdu->offset = 0;
rbdu->data = rbdu_buffer;
return TRUE;
}
static GstVaapiDecoderStatus
decode_ebdu(GstVaapiDecoderVC1 *decoder, GstVC1BDU *ebdu)
{
GstVaapiDecoderStatus status;
GstVC1BDU rbdu;
if (!decode_rbdu(decoder, &rbdu, ebdu))
return GST_VAAPI_DECODER_STATUS_ERROR_ALLOCATION_FAILED;
switch (ebdu->type) {
case GST_VC1_SEQUENCE:
status = decode_sequence(decoder, &rbdu, ebdu);
break;
case GST_VC1_ENTRYPOINT:
status = decode_entry_point(decoder, &rbdu, ebdu);
break;
case GST_VC1_FRAME:
status = decode_frame(decoder, &rbdu, ebdu);
break;
case GST_VC1_SLICE:
GST_DEBUG("decode slice");
status = GST_VAAPI_DECODER_STATUS_ERROR_BITSTREAM_PARSER;
break;
case GST_VC1_END_OF_SEQ:
status = decode_sequence_end(decoder);
break;
default:
GST_DEBUG("unsupported BDU type %d", ebdu->type);
status = GST_VAAPI_DECODER_STATUS_ERROR_BITSTREAM_PARSER;
break;
}
return status;
}
static GstVaapiDecoderStatus static GstVaapiDecoderStatus
decode_buffer(GstVaapiDecoderVC1 *decoder, GstBuffer *buffer) decode_buffer(GstVaapiDecoderVC1 *decoder, GstBuffer *buffer)
{ {
GstVaapiDecoderVC1Private * const priv = decoder->priv; GstVaapiDecoderVC1Private * const priv = decoder->priv;
GstVaapiDecoderStatus status; GstVaapiDecoderStatus status;
GstVC1ParserResult result; GstVC1ParserResult result;
GstVC1BDU bdu; GstVC1BDU ebdu;
guchar *buf; guchar *buf;
guint buf_size, bdu_size, ofs; guint buf_size, ofs;
buf = GST_BUFFER_DATA(buffer); buf = GST_BUFFER_DATA(buffer);
buf_size = GST_BUFFER_SIZE(buffer); buf_size = GST_BUFFER_SIZE(buffer);
@ -968,7 +1072,7 @@ decode_buffer(GstVaapiDecoderVC1 *decoder, GstBuffer *buffer)
result = gst_vc1_identify_next_bdu( result = gst_vc1_identify_next_bdu(
buf + ofs, buf + ofs,
buf_size - ofs, buf_size - ofs,
&bdu &ebdu
); );
status = get_status(result); status = get_status(result);
@ -979,29 +1083,8 @@ decode_buffer(GstVaapiDecoderVC1 *decoder, GstBuffer *buffer)
if (status != GST_VAAPI_DECODER_STATUS_SUCCESS) if (status != GST_VAAPI_DECODER_STATUS_SUCCESS)
break; break;
ofs += bdu.offset; ofs += ebdu.offset + ebdu.size;
switch (bdu.type) { status = decode_ebdu(decoder, &ebdu);
case GST_VC1_SEQUENCE:
status = decode_sequence(decoder, buf + ofs, bdu.size);
break;
case GST_VC1_ENTRYPOINT:
status = decode_entry_point(decoder, buf + ofs, bdu.size);
break;
case GST_VC1_FRAME:
status = decode_frame(decoder, buf + ofs, bdu.size);
break;
case GST_VC1_SLICE:
GST_DEBUG("decode slice");
break;
case GST_VC1_END_OF_SEQ:
status = decode_sequence_end(decoder);
break;
default:
GST_DEBUG("unsupported BDU type %d", bdu.type);
status = GST_VAAPI_DECODER_STATUS_ERROR_BITSTREAM_PARSER;
break;
}
ofs += bdu.size;
} while (status == GST_VAAPI_DECODER_STATUS_SUCCESS); } while (status == GST_VAAPI_DECODER_STATUS_SUCCESS);
return status; return status;
} }
@ -1078,6 +1161,8 @@ gst_vaapi_decoder_vc1_init(GstVaapiDecoderVC1 *decoder)
priv->prev_picture = NULL; priv->prev_picture = NULL;
priv->tsb = NULL; priv->tsb = NULL;
priv->sub_buffer = NULL; priv->sub_buffer = NULL;
priv->rbdu_buffer = NULL;
priv->rbdu_buffer_size = 0;
priv->is_constructed = FALSE; priv->is_constructed = FALSE;
priv->is_opened = FALSE; priv->is_opened = FALSE;
priv->is_first_field = FALSE; priv->is_first_field = FALSE;