static inline void vc1_importer_check_eof( importer_t *importer, vc1_access_unit_t *access_unit )
{
if( lsmash_bs_is_end( importer->bs, 0 ) && access_unit->incomplete_data_length == 0 )
importer->status = IMPORTER_EOF;
else
importer->status = IMPORTER_OK;
}
static int vc1_importer_get_access_unit_internal( importer_t *importer, int probe )
{
vc1_importer_t *vc1_imp = (vc1_importer_t *)importer->info;
vc1_info_t *info = &vc1_imp->info;
vc1_stream_buffer_t *sb = &info->buffer;
vc1_access_unit_t *access_unit = &info->access_unit;
lsmash_bs_t *bs = vc1_imp->bs;
int complete_au = 0;
access_unit->data_length = 0;
while( 1 )
{
uint8_t bdu_type;
uint64_t trailing_zero_bytes;
uint64_t ebdu_length = vc1_find_next_start_code_prefix( bs, &bdu_type, &trailing_zero_bytes );
if( ebdu_length <= VC1_START_CODE_LENGTH && lsmash_bs_is_end( bs, ebdu_length ) )
{
/* For the last EBDU.
* This EBDU already has been appended into the latest access unit and parsed. */
vc1_complete_au( access_unit, &info->picture, probe );
return vc1_get_au_internal_succeeded( vc1_imp );
}
else if( bdu_type == 0xFF )
{
lsmash_log( importer, LSMASH_LOG_ERROR, "a forbidden BDU type is detected.\n" );
return vc1_get_au_internal_failed( vc1_imp, complete_au );
}
uint64_t next_ebdu_head_pos = info->ebdu_head_pos
+ ebdu_length
+ trailing_zero_bytes;
#if 0
if( probe )
{
fprintf( stderr, "BDU type: %"PRIu8" \n", bdu_type );
fprintf( stderr, " EBDU position: %"PRIx64" \n", info->ebdu_head_pos );
fprintf( stderr, " EBDU length: %"PRIx64" (%"PRIu64")\n", ebdu_length, ebdu_length );
fprintf( stderr, " trailing_zero_bytes: %"PRIx64" \n", trailing_zero_bytes );
fprintf( stderr, " Next EBDU position: %"PRIx64" \n", next_ebdu_head_pos );
}
#endif
if( bdu_type >= 0x0A && bdu_type <= 0x0F )
{
/* Complete the current access unit if encountered delimiter of current access unit. */
if( vc1_find_au_delimit_by_bdu_type( bdu_type, info->prev_bdu_type ) )
/* The last video coded EBDU belongs to the access unit you want at this time. */
complete_au = vc1_complete_au( access_unit, &info->picture, probe );
/* Increase the buffer if needed. */
uint64_t possible_au_length = access_unit->incomplete_data_length + ebdu_length;
if( sb->bank->buffer_size < possible_au_length
&& vc1_supplement_buffer( sb, access_unit, 2 * possible_au_length ) < 0 )
{
lsmash_log( importer, LSMASH_LOG_ERROR, "failed to increase the buffer size.\n" );
return vc1_get_au_internal_failed( vc1_imp, complete_au );
}
/* Process EBDU by its BDU type and append it to access unit. */
uint8_t *ebdu = lsmash_bs_get_buffer_data( bs );
switch( bdu_type )
{
/* FRM_SC: Frame start code
* FLD_SC: Field start code
* SLC_SC: Slice start code
* SEQ_SC: Sequence header start code
* EP_SC: Entry-point start code
* PIC_L: Picture layer
* SLC_L: Slice layer
* SEQ_L: Sequence layer
* EP_L: Entry-point layer */
case 0x0D : /* Frame
* For the Progressive or Frame Interlace mode, shall signal the beginning of a new video frame.
* For the Field Interlace mode, shall signal the beginning of a sequence of two independently coded video fields.
* [FRM_SC][PIC_L][[FLD_SC][PIC_L] (optional)][[SLC_SC][SLC_L] (optional)] ... */
if( vc1_parse_advanced_picture( info->bits, &info->sequence, &info->picture, sb->rbdu, ebdu, ebdu_length ) < 0 )
{
lsmash_log( importer, LSMASH_LOG_ERROR, "failed to parse a frame.\n" );
return vc1_get_au_internal_failed( vc1_imp, complete_au );
}
case 0x0C : /* Field
* Shall only be used for Field Interlaced frames
* and shall only be used to signal the beginning of the second field of the frame.
* [FRM_SC][PIC_L][FLD_SC][PIC_L][[SLC_SC][SLC_L] (optional)] ...
* Field start code is followed by INTERLACE_FIELD_PICTURE_FIELD2() which doesn't have info of its field picture type.*/
break;
case 0x0B : /* Slice
* Shall not be used for start code of the first slice of a frame.
* Shall not be used for start code of the first slice of an interlace field coded picture.
* [FRM_SC][PIC_L][[FLD_SC][PIC_L] (optional)][SLC_SC][SLC_L][[SLC_SC][SLC_L] (optional)] ...
* Slice layer may repeat frame header. We just ignore it. */
info->dvc1_param.slice_present = 1;
break;
case 0x0E : /* Entry-point header
* Entry-point indicates the direct followed frame is a start of group of frames.
* Entry-point doesn't indicates the frame is a random access point when multiple sequence headers are present,
* since it is necessary to decode sequence header which subsequent frames belong to for decoding them.
* Entry point shall be followed by
* 1. I-picture - progressive or frame interlace
* 2. I/I-picture, I/P-picture, or P/I-picture - field interlace
* [[SEQ_SC][SEQ_L] (optional)][EP_SC][EP_L][FRM_SC][PIC_L] ... */
if( vc1_parse_entry_point_header( info, ebdu, ebdu_length, probe ) < 0 )
{
lsmash_log( importer, LSMASH_LOG_ERROR, "failed to parse an entry point.\n" );
return vc1_get_au_internal_failed( vc1_imp, complete_au );
}
/* Signal random access type of the frame that follows this entry-point header. */
info->picture.closed_gop = info->entry_point.closed_entry_point;
info->picture.random_accessible = info->dvc1_param.multiple_sequence ? info->picture.start_of_sequence : 1;
break;
case 0x0F : /* Sequence header
* [SEQ_SC][SEQ_L][EP_SC][EP_L][FRM_SC][PIC_L] ... */
if( vc1_parse_sequence_header( info, ebdu, ebdu_length, probe ) < 0 )
{
lsmash_log( importer, LSMASH_LOG_ERROR, "failed to parse a sequence header.\n" );
return vc1_get_au_internal_failed( vc1_imp, complete_au );
}
/* The frame that is the first frame after this sequence header shall be a random accessible point. */
info->picture.start_of_sequence = 1;
if( probe && !vc1_imp->first_sequence.present )
vc1_imp->first_sequence = info->sequence;
break;
default : /* End-of-sequence (0x0A) */
break;
}
/* Append the current EBDU into the end of an incomplete access unit. */
vc1_append_ebdu_to_au( access_unit, ebdu, ebdu_length, probe );
}
else /* We don't support other BDU types such as user data yet. */
return vc1_get_au_internal_failed( vc1_imp, complete_au );
/* Move to the first byte of the next EBDU. */
info->prev_bdu_type = bdu_type;
if( lsmash_bs_read_seek( bs, next_ebdu_head_pos, SEEK_SET ) != next_ebdu_head_pos )
{
lsmash_log( importer, LSMASH_LOG_ERROR, "failed to seek the next start code suffix.\n" );
return vc1_get_au_internal_failed( vc1_imp, complete_au );
}
/* Check if no more data to read from the stream. */
if( !lsmash_bs_is_end( bs, VC1_START_CODE_PREFIX_LENGTH ) )
info->ebdu_head_pos = next_ebdu_head_pos;
/* If there is no more data in the stream, and flushed chunk of EBDUs, flush it as complete AU here. */
else if( access_unit->incomplete_data_length && access_unit->data_length == 0 )
{
vc1_complete_au( access_unit, &info->picture, probe );
return vc1_get_au_internal_succeeded( vc1_imp );
}
if( complete_au )
return vc1_get_au_internal_succeeded( vc1_imp );
}
}
static inline void vc1_get_au_internal_end( vc1_importer_t *vc1_imp, vc1_access_unit_t *access_unit )
{
vc1_imp->status = lsmash_bs_is_end( vc1_imp->bs, 0 ) && (access_unit->incomplete_data_length == 0)
? IMPORTER_EOF
: IMPORTER_OK;
}