1 /*****************************************************************************
2 * a52.c: parse A/52 audio sync info and packetize the stream
3 *****************************************************************************
4 * Copyright (C) 2001-2016 VLC authors and VideoLAN
7 * Authors: Stéphane Borel <stef@via.ecp.fr>
8 * Christophe Massiot <massiot@via.ecp.fr>
9 * Gildas Bazin <gbazin@videolan.org>
11 * This program is free software; you can redistribute it and/or modify it
12 * under the terms of the GNU Lesser General Public License as published by
13 * the Free Software Foundation; either version 2.1 of the License, or
14 * (at your option) any later version.
16 * This program is distributed in the hope that it will be useful,
17 * but WITHOUT ANY WARRANTY; without even the implied warranty of
18 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
19 * GNU Lesser General Public License for more details.
21 * You should have received a copy of the GNU Lesser General Public License
22 * along with this program; if not, write to the Free Software Foundation,
23 * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
24 *****************************************************************************/
26 /*****************************************************************************
28 *****************************************************************************/
35 #include <vlc_common.h>
36 #include <vlc_plugin.h>
37 #include <vlc_codec.h>
38 #include <vlc_block_helper.h>
39 #include <vlc_modules.h>
43 #include "packetizer_helper.h"
45 static int Open( vlc_object_t
* );
46 static void Close( vlc_object_t
* );
49 set_category(CAT_SOUT
)
50 set_subcategory(SUBCAT_SOUT_PACKETIZER
)
51 set_description( N_("A/52 audio packetizer") )
52 set_capability( "packetizer", 10 )
53 set_callbacks( Open
, Close
)
63 block_bytestream_t bytestream
;
69 mtime_t i_prev_bytestream_pts
;
72 vlc_a52_header_t frame
;
75 static void PacketizeFlush( decoder_t
*p_dec
)
77 decoder_sys_t
*p_sys
= p_dec
->p_sys
;
79 p_sys
->b_discontuinity
= true;
80 date_Set( &p_sys
->end_date
, VLC_TS_INVALID
);
81 p_sys
->i_state
= STATE_NOSYNC
;
82 p_sys
->i_prev_bytestream_pts
= VLC_TS_INVALID
;
83 block_BytestreamEmpty( &p_sys
->bytestream
);
86 static block_t
*GetOutBuffer( decoder_t
*p_dec
)
88 decoder_sys_t
*p_sys
= p_dec
->p_sys
;
90 assert( p_sys
->frame
.i_rate
> 0 );
92 block_t
*p_block
= block_Alloc( p_sys
->frame
.i_size
);
96 if( p_dec
->fmt_out
.audio
.i_rate
!= p_sys
->frame
.i_rate
)
98 msg_Dbg( p_dec
, "A/52 channels:%d samplerate:%d bitrate:%d",
99 p_sys
->frame
.i_channels
, p_sys
->frame
.i_rate
, p_sys
->frame
.i_bitrate
);
100 if( p_sys
->end_date
.i_divider_num
)
101 date_Change( &p_sys
->end_date
, p_sys
->frame
.i_rate
, 1 );
103 date_Init( &p_sys
->end_date
, p_sys
->frame
.i_rate
, 1 );
106 if( p_sys
->bytestream
.p_block
->i_pts
!= date_Get( &p_sys
->end_date
) &&
107 p_sys
->bytestream
.p_block
->i_pts
!= VLC_TS_INVALID
)
109 date_Set( &p_sys
->end_date
, p_sys
->bytestream
.p_block
->i_pts
);
110 p_sys
->bytestream
.p_block
->i_pts
= VLC_TS_INVALID
;
113 p_dec
->fmt_out
.audio
.i_rate
= p_sys
->frame
.i_rate
;
114 p_dec
->fmt_out
.audio
.i_channels
= p_sys
->frame
.i_channels
;
115 if( p_dec
->fmt_out
.audio
.i_bytes_per_frame
< p_sys
->frame
.i_size
)
116 p_dec
->fmt_out
.audio
.i_bytes_per_frame
= p_sys
->frame
.i_size
;
117 p_dec
->fmt_out
.audio
.i_frame_length
= p_sys
->frame
.i_samples
;
119 p_dec
->fmt_out
.audio
.i_chan_mode
= p_sys
->frame
.i_chan_mode
;
120 p_dec
->fmt_out
.audio
.i_physical_channels
= p_sys
->frame
.i_channels_conf
;
122 p_dec
->fmt_out
.i_bitrate
= p_sys
->frame
.i_bitrate
;
124 p_block
->i_nb_samples
= p_sys
->frame
.i_samples
;
125 p_block
->i_pts
= p_block
->i_dts
= date_Get( &p_sys
->end_date
);
126 if( p_block
->i_pts
!= VLC_TS_INVALID
)
127 p_block
->i_length
= date_Increment( &p_sys
->end_date
,
128 p_block
->i_nb_samples
) - p_block
->i_pts
;
133 static block_t
*PacketizeBlock( decoder_t
*p_dec
, block_t
**pp_block
)
135 decoder_sys_t
*p_sys
= p_dec
->p_sys
;
136 uint8_t p_header
[VLC_A52_HEADER_SIZE
];
137 block_t
*p_out_buffer
;
139 block_t
*p_block
= pp_block
? *pp_block
: NULL
;
143 if( p_block
->i_flags
& (BLOCK_FLAG_DISCONTINUITY
| BLOCK_FLAG_CORRUPTED
) )
145 /* First always drain complete blocks before discontinuity */
146 block_t
*p_drain
= PacketizeBlock( p_dec
, NULL
);
150 PacketizeFlush( p_dec
);
152 if( p_block
->i_flags
& BLOCK_FLAG_CORRUPTED
)
154 block_Release( p_block
);
159 /* Make sure we don't reuse the same pts twice */
160 if( p_block
->i_pts
> VLC_TS_INVALID
)
162 if( p_block
->i_pts
== p_sys
->i_prev_bytestream_pts
)
163 p_block
->i_pts
= VLC_TS_INVALID
;
165 p_sys
->i_prev_bytestream_pts
= p_block
->i_pts
;
168 block_BytestreamPush( &p_sys
->bytestream
, p_block
);
173 switch( p_sys
->i_state
)
176 while( block_PeekBytes( &p_sys
->bytestream
, p_header
, 2 )
179 if( p_header
[0] == 0x0b && p_header
[1] == 0x77 )
181 p_sys
->i_state
= STATE_SYNC
;
184 block_SkipByte( &p_sys
->bytestream
);
186 if( p_sys
->i_state
!= STATE_SYNC
)
188 block_BytestreamFlush( &p_sys
->bytestream
);
196 p_sys
->i_state
= STATE_HEADER
;
200 /* Get A/52 frame header (VLC_A52_HEADER_SIZE bytes) */
201 if( block_PeekBytes( &p_sys
->bytestream
, p_header
,
202 VLC_A52_HEADER_SIZE
) != VLC_SUCCESS
)
208 /* Check if frame is valid and get frame info */
209 vlc_a52_header_t a52
;
210 if( vlc_a52_header_Parse( &a52
, p_header
, VLC_A52_HEADER_SIZE
) )
212 msg_Dbg( p_dec
, "emulated sync word" );
213 block_SkipByte( &p_sys
->bytestream
);
214 p_sys
->i_state
= STATE_NOSYNC
;
218 if( a52
.b_eac3
&& a52
.eac3
.strmtyp
!= EAC3_STRMTYP_INDEPENDENT
)
220 /* Use the channel configuration of the independent stream */
221 if( !p_sys
->frame
.i_blocks_per_sync_frame
)
223 /* Not synced on main stream yet */
224 block_SkipByte( &p_sys
->bytestream
);
225 p_sys
->i_state
= STATE_NOSYNC
;
228 p_sys
->frame
.i_samples
= a52
.i_samples
;
229 p_sys
->frame
.i_size
= a52
.i_size
;
234 p_sys
->i_state
= STATE_NEXT_SYNC
;
237 case STATE_NEXT_SYNC
:
238 /* Check if next expected frame contains the sync word */
239 if( block_PeekOffsetBytes( &p_sys
->bytestream
,
240 p_sys
->frame
.i_size
, p_header
, 2 )
243 if( p_block
== NULL
) /* drain */
245 p_sys
->i_state
= STATE_GET_DATA
;
252 if( p_header
[0] == 0 && p_header
[1] == 0 )
254 /* A52 wav files and audio CD's use stuffing */
255 p_sys
->i_state
= STATE_GET_DATA
;
259 if( p_header
[0] != 0x0b || p_header
[1] != 0x77 )
261 msg_Dbg( p_dec
, "emulated sync word "
262 "(no sync on following frame)" );
263 p_sys
->i_state
= STATE_NOSYNC
;
264 block_SkipByte( &p_sys
->bytestream
);
267 p_sys
->i_state
= STATE_GET_DATA
;
271 /* Make sure we have enough data. */
272 if( block_WaitBytes( &p_sys
->bytestream
,
273 p_sys
->frame
.i_size
) != VLC_SUCCESS
)
278 p_sys
->i_state
= STATE_SEND_DATA
;
281 case STATE_SEND_DATA
:
282 if( !(p_out_buffer
= GetOutBuffer( p_dec
)) )
287 /* Copy the whole frame into the buffer. When we reach this point
288 * we already know we have enough data available. */
289 block_GetBytes( &p_sys
->bytestream
, p_out_buffer
->p_buffer
,
290 __MIN( p_sys
->frame
.i_size
, p_out_buffer
->i_buffer
) );
292 p_sys
->i_state
= STATE_NOSYNC
;
294 if( p_out_buffer
->i_dts
== VLC_TS_INVALID
)
296 block_Release( p_out_buffer
);
300 if( p_sys
->b_discontuinity
)
302 p_out_buffer
->i_flags
|= BLOCK_FLAG_DISCONTINUITY
;
303 p_sys
->b_discontuinity
= false;
306 /* So p_block doesn't get re-added several times */
308 *pp_block
= block_BytestreamPop( &p_sys
->bytestream
);
315 static void Close( vlc_object_t
*p_this
)
317 decoder_t
*p_dec
= (decoder_t
*)p_this
;
318 decoder_sys_t
*p_sys
= p_dec
->p_sys
;
320 block_BytestreamRelease( &p_sys
->bytestream
);
325 static int Open( vlc_object_t
*p_this
)
327 decoder_t
*p_dec
= (decoder_t
*)p_this
;
328 decoder_sys_t
*p_sys
;
330 switch( p_dec
->fmt_in
.i_codec
)
339 /* Allocate the memory needed to store the decoder's structure */
340 if( ( p_dec
->p_sys
= p_sys
=
341 (decoder_sys_t
*)malloc(sizeof(decoder_sys_t
)) ) == NULL
)
345 p_sys
->i_state
= STATE_NOSYNC
;
346 date_Set( &p_sys
->end_date
, VLC_TS_INVALID
);
347 p_sys
->b_discontuinity
= false;
348 p_sys
->i_prev_bytestream_pts
= VLC_TS_INVALID
;
349 memset(&p_sys
->frame
, 0, sizeof(vlc_a52_header_t
));
351 block_BytestreamInit( &p_sys
->bytestream
);
353 /* Set output properties (Passthrough ONLY) */
354 p_dec
->fmt_out
.i_codec
= p_dec
->fmt_in
.i_codec
;
355 p_dec
->fmt_out
.audio
= p_dec
->fmt_in
.audio
;
356 p_dec
->fmt_out
.audio
.i_rate
= 0;
359 p_dec
->pf_packetize
= PacketizeBlock
;
360 p_dec
->pf_flush
= PacketizeFlush
;