1 /*****************************************************************************
2 * lpcm.c: lpcm decoder/packetizer module
3 *****************************************************************************
4 * Copyright (C) 1999-2008 the VideoLAN team
7 * Authors: Samuel Hocevar <sam@zoy.org>
8 * Henri Fallon <henri@videolan.org>
9 * Christophe Massiot <massiot@via.ecp.fr>
10 * Gildas Bazin <gbazin@videolan.org>
11 * Lauren Aimar <fenrir _AT_ videolan _DOT_ org >
13 * This program is free software; you can redistribute it and/or modify
14 * it under the terms of the GNU General Public License as published by
15 * the Free Software Foundation; either version 2 of the License, or
16 * (at your option) any later version.
18 * This program is distributed in the hope that it will be useful,
19 * but WITHOUT ANY WARRANTY; without even the implied warranty of
20 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
21 * GNU General Public License for more details.
23 * You should have received a copy of the GNU General Public License
24 * along with this program; if not, write to the Free Software
25 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
26 *****************************************************************************/
28 /*****************************************************************************
30 *****************************************************************************/
35 #include <vlc_common.h>
36 #include <vlc_plugin.h>
37 #include <vlc_codec.h>
41 /*****************************************************************************
43 *****************************************************************************/
44 static int OpenDecoder ( vlc_object_t * );
45 static int OpenPacketizer( vlc_object_t * );
46 static void CloseCommon ( vlc_object_t * );
50 set_category( CAT_INPUT )
51 set_subcategory( SUBCAT_INPUT_ACODEC )
52 set_description( N_("Linear PCM audio decoder") )
53 set_capability( "decoder", 100 )
54 set_callbacks( OpenDecoder, CloseCommon )
57 set_description( N_("Linear PCM audio packetizer") )
58 set_capability( "packetizer", 100 )
59 set_callbacks( OpenPacketizer, CloseCommon )
64 /*****************************************************************************
65 * decoder_sys_t : lpcm decoder descriptor
66 *****************************************************************************/
75 audio_date_t end_date;
78 unsigned i_header_size;
84 * - frame number (8 bits)
85 * - unknown (16 bits) == 0x0003 ?
87 * - current frame (4 bits)
89 * - frequency (2 bits) 0 == 48 kHz, 1 == 32 kHz, 2 == ?, 3 == ?
91 * - number of channels - 1 (3 bits) 1 == 2 channels
92 * - start code (8 bits) == 0x80
96 * - number of channels (4 bits)
97 * - frequency (4 bits)
98 * - bits per sample (2 bits)
102 #define LPCM_DVD_HEADER_LEN (6)
103 #define LPCM_BD_HEADER_LEN (4)
105 /*****************************************************************************
107 *****************************************************************************/
108 static void *DecodeFrame ( decoder_t *, block_t ** );
109 static int DvdHeader( unsigned *pi_rate,
110 unsigned *pi_channels, unsigned *pi_original_channels,
112 const uint8_t *p_header );
113 static void DvdExtract( aout_buffer_t *, block_t *, unsigned i_bits );
115 static int BdHeader( unsigned *pi_rate,
116 unsigned *pi_channels, unsigned *pi_original_channels,
118 const uint8_t *p_header );
119 static void BdExtract( aout_buffer_t *, block_t * );
122 /*****************************************************************************
124 *****************************************************************************/
125 static int OpenCommon( vlc_object_t *p_this, bool b_packetizer )
127 decoder_t *p_dec = (decoder_t*)p_this;
128 decoder_sys_t *p_sys;
131 switch( p_dec->fmt_in.i_codec )
134 case VLC_CODEC_DVD_LPCM:
138 case VLC_CODEC_BD_LPCM:
145 /* Allocate the memory needed to store the decoder's structure */
146 if( ( p_dec->p_sys = p_sys = malloc(sizeof(decoder_sys_t)) ) == NULL )
150 p_sys->b_packetizer = b_packetizer;
151 aout_DateSet( &p_sys->end_date, 0 );
152 p_sys->i_header_size = b_dvd ? LPCM_DVD_HEADER_LEN : LPCM_BD_HEADER_LEN;
153 p_sys->b_dvd = b_dvd;
155 /* Set output properties */
156 p_dec->fmt_out.i_cat = AUDIO_ES;
160 p_dec->fmt_out.i_codec = b_dvd ? VLC_CODEC_DVD_LPCM : VLC_CODEC_BD_LPCM;
164 switch( p_dec->fmt_out.audio.i_bitspersample )
168 p_dec->fmt_out.i_codec = VLC_CODEC_S24B;
169 p_dec->fmt_out.audio.i_bitspersample = 24;
172 p_dec->fmt_out.i_codec = VLC_CODEC_S16B;
173 p_dec->fmt_out.audio.i_bitspersample = 16;
179 p_dec->pf_decode_audio = (aout_buffer_t *(*)(decoder_t *, block_t **))
181 p_dec->pf_packetize = (block_t *(*)(decoder_t *, block_t **))
186 static int OpenDecoder( vlc_object_t *p_this )
188 return OpenCommon( p_this, false );
190 static int OpenPacketizer( vlc_object_t *p_this )
192 return OpenCommon( p_this, true );
195 /*****************************************************************************
196 * DecodeFrame: decodes an lpcm frame.
197 ****************************************************************************
198 * Beware, this function must be fed with complete frames (PES packet).
199 *****************************************************************************/
200 static void *DecodeFrame( decoder_t *p_dec, block_t **pp_block )
202 decoder_sys_t *p_sys = p_dec->p_sys;
204 unsigned int i_rate = 0, i_original_channels = 0, i_channels = 0, i_bits = 0;
207 if( !pp_block || !*pp_block ) return NULL;
210 *pp_block = NULL; /* So the packet doesn't get re-sent */
212 /* Date management */
213 if( p_block->i_pts > 0 &&
214 p_block->i_pts != aout_DateGet( &p_sys->end_date ) )
216 aout_DateSet( &p_sys->end_date, p_block->i_pts );
219 if( !aout_DateGet( &p_sys->end_date ) )
221 /* We've just started the stream, wait for the first PTS. */
222 block_Release( p_block );
226 if( p_block->i_buffer <= p_sys->i_header_size )
228 msg_Err(p_dec, "frame is too short");
229 block_Release( p_block );
235 i_ret = DvdHeader( &i_rate, &i_channels, &i_original_channels, &i_bits,
238 i_ret = BdHeader( &i_rate, &i_channels, &i_original_channels, &i_bits,
243 msg_Warn( p_dec, "no frame sync" );
244 block_Release( p_block );
248 /* Set output properties */
249 if( p_dec->fmt_out.audio.i_rate != i_rate )
251 aout_DateInit( &p_sys->end_date, i_rate );
252 aout_DateSet( &p_sys->end_date, p_block->i_pts );
254 p_dec->fmt_out.audio.i_rate = i_rate;
255 p_dec->fmt_out.audio.i_channels = i_channels;
256 p_dec->fmt_out.audio.i_original_channels = i_original_channels;
257 p_dec->fmt_out.audio.i_physical_channels = i_original_channels & AOUT_CHAN_PHYSMASK;
259 i_frame_length = (p_block->i_buffer - p_sys->i_header_size) / i_channels * 8 / i_bits;
261 if( p_sys->b_packetizer )
263 p_block->i_pts = p_block->i_dts = aout_DateGet( &p_sys->end_date );
265 aout_DateIncrement( &p_sys->end_date, i_frame_length ) -
268 /* Just pass on the incoming frame */
276 p_dec->fmt_out.i_codec = VLC_CODEC_S16B;
277 p_dec->fmt_out.audio.i_bitspersample = 16;
281 p_dec->fmt_out.i_codec = VLC_CODEC_S24B;
282 p_dec->fmt_out.audio.i_bitspersample = 24;
286 aout_buffer_t *p_aout_buffer;
287 p_aout_buffer = decoder_NewAudioBuffer( p_dec, i_frame_length );
291 p_aout_buffer->start_date = aout_DateGet( &p_sys->end_date );
292 p_aout_buffer->end_date =
293 aout_DateIncrement( &p_sys->end_date, i_frame_length );
295 p_block->p_buffer += p_sys->i_header_size;
296 p_block->i_buffer -= p_sys->i_header_size;
299 DvdExtract( p_aout_buffer, p_block, i_bits );
301 BdExtract( p_aout_buffer, p_block );
303 block_Release( p_block );
304 return p_aout_buffer;
308 /*****************************************************************************
309 * CloseCommon : lpcm decoder destruction
310 *****************************************************************************/
311 static void CloseCommon( vlc_object_t *p_this )
313 decoder_t *p_dec = (decoder_t*)p_this;
314 free( p_dec->p_sys );
317 /*****************************************************************************
319 *****************************************************************************/
320 static int DvdHeader( unsigned *pi_rate,
321 unsigned *pi_channels, unsigned *pi_original_channels,
323 const uint8_t *p_header )
325 const uint8_t i_header = p_header[4];
327 switch( (i_header >> 4) & 0x3 )
343 *pi_channels = (i_header & 0x7) + 1;
344 switch( *pi_channels - 1 )
347 *pi_original_channels = AOUT_CHAN_CENTER;
350 *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT;
353 /* This is unsure. */
354 *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_LFE;
357 *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT
358 | AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT;
361 /* This is unsure. */
362 *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT
363 | AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT
367 *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT
368 | AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT
369 | AOUT_CHAN_CENTER | AOUT_CHAN_LFE;
372 *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT
373 | AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT
374 | AOUT_CHAN_CENTER | AOUT_CHAN_MIDDLELEFT
375 | AOUT_CHAN_MIDDLERIGHT;
378 *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT
379 | AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT
380 | AOUT_CHAN_CENTER | AOUT_CHAN_MIDDLELEFT
381 | AOUT_CHAN_MIDDLERIGHT | AOUT_CHAN_LFE;
385 switch( (i_header >> 6) & 0x3 )
399 /* Check frame sync and drop it. */
400 if( p_header[5] != 0x80 )
404 static int BdHeader( unsigned *pi_rate,
405 unsigned *pi_channels, unsigned *pi_original_channels,
407 const uint8_t *p_header )
409 const uint32_t h = GetDWBE( p_header );
410 switch( ( h & 0xf000) >> 12 )
414 *pi_original_channels = AOUT_CHAN_CENTER;
418 *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT;
422 *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER;
426 *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_REARCENTER;
430 *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER |
431 AOUT_CHAN_REARCENTER;
435 *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT |
436 AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT;
440 *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER |
441 AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT;
445 *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER |
446 AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT |
451 *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER |
452 AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT |
453 AOUT_CHAN_MIDDLELEFT | AOUT_CHAN_MIDDLERIGHT;
457 *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER |
458 AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT |
459 AOUT_CHAN_MIDDLELEFT | AOUT_CHAN_MIDDLERIGHT |
466 switch( (h >> 6) & 0x03 )
471 case 2: /* 20 bits but samples are stored on 24 bits */
472 case 3: /* 24 bits */
478 switch( (h >> 8) & 0x0f )
495 static void DvdExtract( aout_buffer_t *p_aout_buffer, block_t *p_block,
498 uint8_t *p_out = p_aout_buffer->p_buffer;
500 /* 20/24 bits LPCM use special packing */
503 while( p_block->i_buffer / 12 )
506 p_out[0] = p_block->p_buffer[0];
507 p_out[1] = p_block->p_buffer[1];
508 p_out[2] = p_block->p_buffer[8];
510 p_out[3] = p_block->p_buffer[2];
511 p_out[4] = p_block->p_buffer[3];
512 p_out[5] = p_block->p_buffer[9];
514 p_out[6] = p_block->p_buffer[4];
515 p_out[7] = p_block->p_buffer[5];
516 p_out[8] = p_block->p_buffer[10];
518 p_out[9] = p_block->p_buffer[6];
519 p_out[10] = p_block->p_buffer[7];
520 p_out[11] = p_block->p_buffer[11];
522 p_block->i_buffer -= 12;
523 p_block->p_buffer += 12;
527 else if( i_bits == 20 )
529 while( p_block->i_buffer / 10 )
532 p_out[0] = p_block->p_buffer[0];
533 p_out[1] = p_block->p_buffer[1];
534 p_out[2] = p_block->p_buffer[8] & 0xF0;
536 p_out[3] = p_block->p_buffer[2];
537 p_out[4] = p_block->p_buffer[3];
538 p_out[5] = p_block->p_buffer[8] << 4;
540 p_out[6] = p_block->p_buffer[4];
541 p_out[7] = p_block->p_buffer[5];
542 p_out[8] = p_block->p_buffer[9] & 0xF0;
544 p_out[9] = p_block->p_buffer[6];
545 p_out[10] = p_block->p_buffer[7];
546 p_out[11] = p_block->p_buffer[9] << 4;
548 p_block->i_buffer -= 10;
549 p_block->p_buffer += 10;
555 assert( i_bits == 16 );
556 memcpy( p_out, p_block->p_buffer, p_block->i_buffer );
559 static void BdExtract( aout_buffer_t *p_aout_buffer, block_t *p_block )
561 memcpy( p_aout_buffer->p_buffer, p_block->p_buffer, p_block->i_buffer );