1 /*****************************************************************************
2 * araw.c: Pseudo audio decoder; for raw pcm data
3 *****************************************************************************
4 * Copyright (C) 2001, 2003 VLC authors and VideoLAN
7 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
9 * This program is free software; you can redistribute it and/or modify it
10 * under the terms of the GNU Lesser General Public License as published by
11 * the Free Software Foundation; either version 2.1 of the License, or
12 * (at your option) any later version.
14 * This program is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 * GNU Lesser General Public License for more details.
19 * You should have received a copy of the GNU Lesser General Public License
20 * along with this program; if not, write to the Free Software Foundation,
21 * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
22 *****************************************************************************/
24 /*****************************************************************************
26 *****************************************************************************/
33 #include <vlc_common.h>
34 #include <vlc_plugin.h>
35 #include <vlc_codec.h>
38 /*****************************************************************************
40 *****************************************************************************/
41 static int DecoderOpen ( vlc_object_t * );
42 static void DecoderClose( vlc_object_t * );
45 static int EncoderOpen ( vlc_object_t * );
49 /* audio decoder module */
50 set_description( N_("Raw/Log Audio decoder") )
51 set_capability( "decoder", 100 )
52 set_category( CAT_INPUT )
53 set_subcategory( SUBCAT_INPUT_ACODEC )
54 set_callbacks( DecoderOpen, DecoderClose )
57 /* audio encoder submodule */
59 set_description( N_("Raw audio encoder") )
60 set_capability( "encoder", 150 )
61 set_callbacks( EncoderOpen, NULL )
65 /*****************************************************************************
67 *****************************************************************************/
68 static block_t *DecodeBlock( decoder_t *, block_t ** );
72 void (*decode) (void *, const uint8_t *, unsigned);
77 static const uint16_t pi_channels_maps[] =
80 AOUT_CHAN_CENTER, AOUT_CHANS_2_0, AOUT_CHANS_3_0,
81 AOUT_CHANS_4_0, AOUT_CHANS_5_0, AOUT_CHANS_5_1,
82 AOUT_CHANS_7_0, AOUT_CHANS_7_1, AOUT_CHANS_8_1,
85 static void S8Decode( void *, const uint8_t *, unsigned );
86 static void U16BDecode( void *, const uint8_t *, unsigned );
87 static void U16LDecode( void *, const uint8_t *, unsigned );
88 static void S16IDecode( void *, const uint8_t *, unsigned );
89 static void S20BDecode( void *, const uint8_t *, unsigned );
90 static void U32BDecode( void *, const uint8_t *, unsigned );
91 static void U32LDecode( void *, const uint8_t *, unsigned );
92 static void Swap32Decode( void *, const uint8_t *, unsigned );
93 static void Swap64Decode( void *, const uint8_t *, unsigned );
94 static void DAT12Decode( void *, const uint8_t *, unsigned );
96 /*****************************************************************************
97 * DecoderOpen: probe the decoder and return score
98 *****************************************************************************/
99 static int DecoderOpen( vlc_object_t *p_this )
101 decoder_t *p_dec = (decoder_t*)p_this;
102 vlc_fourcc_t format = p_dec->fmt_in.i_codec;
104 switch( p_dec->fmt_in.i_codec )
106 case VLC_FOURCC('a','r','a','w'):
107 case VLC_FOURCC('p','c','m',' '):
108 case VLC_FOURCC('a','f','l','t'):
109 /* _signed_ big endian samples (mov) */
110 case VLC_FOURCC('t','w','o','s'):
111 /* _signed_ little endian samples (mov) */
112 case VLC_FOURCC('s','o','w','t'):
114 vlc_fourcc_GetCodecAudio( p_dec->fmt_in.i_codec,
115 p_dec->fmt_in.audio.i_bitspersample );
118 msg_Err( p_dec, "bad parameters(bits/sample)" );
124 void (*decode) (void *, const uint8_t *, unsigned) = NULL;
129 #ifdef WORDS_BIGENDIAN
134 format = VLC_CODEC_FL64;
135 decode = Swap64Decode;
139 #ifdef WORDS_BIGENDIAN
144 format = VLC_CODEC_FL32;
145 decode = Swap32Decode;
150 format = VLC_CODEC_S32N;
155 format = VLC_CODEC_S32N;
160 format = VLC_CODEC_S32N;
161 decode = Swap32Decode;
170 format = VLC_CODEC_S32N;
175 format = VLC_CODEC_S16N;
180 format = VLC_CODEC_S16N;
185 format = VLC_CODEC_S16N;
190 case VLC_CODEC_DAT12:
191 format = VLC_CODEC_S16N;
192 decode = DAT12Decode;
197 format = VLC_CODEC_U8;
205 if( p_dec->fmt_in.audio.i_channels <= 0 ||
206 p_dec->fmt_in.audio.i_channels > AOUT_CHAN_MAX )
208 msg_Err( p_dec, "bad channels count (1-9): %i",
209 p_dec->fmt_in.audio.i_channels );
213 if( p_dec->fmt_in.audio.i_rate <= 0 )
215 msg_Err( p_dec, "bad samplerate: %d Hz", p_dec->fmt_in.audio.i_rate );
219 msg_Dbg( p_dec, "samplerate:%dHz channels:%d bits/sample:%d",
220 p_dec->fmt_in.audio.i_rate, p_dec->fmt_in.audio.i_channels,
221 p_dec->fmt_in.audio.i_bitspersample );
223 /* Allocate the memory needed to store the decoder's structure */
224 decoder_sys_t *p_sys = malloc(sizeof(*p_sys));
225 if( unlikely(p_sys == NULL) )
228 /* Set output properties */
229 p_dec->fmt_out.i_cat = AUDIO_ES;
230 p_dec->fmt_out.i_codec = format;
231 p_dec->fmt_out.audio.i_format = format;
232 p_dec->fmt_out.audio.i_rate = p_dec->fmt_in.audio.i_rate;
233 if( p_dec->fmt_in.audio.i_physical_channels )
234 p_dec->fmt_out.audio.i_physical_channels =
235 p_dec->fmt_in.audio.i_physical_channels;
237 p_dec->fmt_out.audio.i_physical_channels =
238 pi_channels_maps[p_dec->fmt_in.audio.i_channels];
239 if( p_dec->fmt_in.audio.i_original_channels )
240 p_dec->fmt_out.audio.i_original_channels =
241 p_dec->fmt_in.audio.i_original_channels;
243 p_dec->fmt_out.audio.i_original_channels =
244 p_dec->fmt_out.audio.i_physical_channels;
245 aout_FormatPrepare( &p_dec->fmt_out.audio );
247 p_sys->decode = decode;
248 p_sys->framebits = bits * p_dec->fmt_out.audio.i_channels;
249 assert( p_sys->framebits );
251 date_Init( &p_sys->end_date, p_dec->fmt_out.audio.i_rate, 1 );
252 date_Set( &p_sys->end_date, 0 );
254 p_dec->pf_decode_audio = DecodeBlock;
255 p_dec->p_sys = p_sys;
260 /****************************************************************************
261 * DecodeBlock: the whole thing
262 ****************************************************************************
263 * This function must be fed with whole samples (see nBlockAlign).
264 ****************************************************************************/
265 static block_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
267 decoder_sys_t *p_sys = p_dec->p_sys;
269 if( !pp_block || !*pp_block ) return NULL;
271 block_t *p_block = *pp_block;
273 if( p_block->i_pts > VLC_TS_INVALID &&
274 p_block->i_pts != date_Get( &p_sys->end_date ) )
276 date_Set( &p_sys->end_date, p_block->i_pts );
278 else if( !date_Get( &p_sys->end_date ) )
280 /* We've just started the stream, wait for the first PTS. */
281 block_Release( p_block );
285 /* Don't re-use the same pts twice */
286 p_block->i_pts = VLC_TS_INVALID;
288 unsigned samples = (8 * p_block->i_buffer) / p_sys->framebits;
291 block_Release( p_block );
295 /* Create chunks of max 1024 samples */
296 if( samples > 1024 ) samples = 1024;
298 block_t *p_out = decoder_NewAudioBuffer( p_dec, samples );
301 block_Release( p_block );
305 p_out->i_pts = date_Get( &p_sys->end_date );
306 p_out->i_length = date_Increment( &p_sys->end_date, samples )
309 if( p_sys->decode != NULL )
310 p_sys->decode( p_out->p_buffer, p_block->p_buffer,
311 samples * p_dec->fmt_in.audio.i_channels );
313 memcpy( p_out->p_buffer, p_block->p_buffer, p_out->i_buffer );
315 samples = (samples * p_sys->framebits) / 8;
316 p_block->p_buffer += samples;
317 p_block->i_buffer -= samples;
322 static void S8Decode( void *outp, const uint8_t *in, unsigned samples )
326 for( size_t i = 0; i < samples; i++ )
327 out[i] = in[i] ^ 0x80;
330 static void U16BDecode( void *outp, const uint8_t *in, unsigned samples )
332 uint16_t *out = outp;
334 for( size_t i = 0; i < samples; i++ )
336 *(out++) = GetWBE( in ) - 0x8000;
341 static void U16LDecode( void *outp, const uint8_t *in, unsigned samples )
343 uint16_t *out = outp;
345 for( size_t i = 0; i < samples; i++ )
347 *(out++) = GetWLE( in ) - 0x8000;
352 static void S16IDecode( void *out, const uint8_t *in, unsigned samples )
354 swab( in, out, samples * 2 );
357 static void S20BDecode( void *outp, const uint8_t *in, unsigned samples )
361 while( samples >= 2 )
363 uint32_t dw = U32_AT(in);
365 *(out++) = dw & ~0xFFF;
366 *(out++) = (dw << 20) | (*in << 12);
371 /* No U32_AT() for the last odd sample: avoid off-by-one overflow! */
373 *(out++) = (U16_AT(in) << 16) | ((in[2] & 0xF0) << 8);
376 static void U32BDecode( void *outp, const uint8_t *in, unsigned samples )
378 uint32_t *out = outp;
380 for( size_t i = 0; i < samples; i++ )
382 *(out++) = GetDWBE( in ) - 0x80000000;
387 static void U32LDecode( void *outp, const uint8_t *in, unsigned samples )
389 uint32_t *out = outp;
391 for( size_t i = 0; i < samples; i++ )
393 *(out++) = GetDWLE( in ) - 0x80000000;
398 static void Swap32Decode( void *outp, const uint8_t *in, unsigned samples )
402 for( size_t i = 0; i < samples; i++ )
404 #ifdef WORDS_BIGENDIAN
405 *(out++) = GetDWLE( in );
407 *(out++) = GetDWBE( in );
413 static void Swap64Decode( void *outp, const uint8_t *in, unsigned samples )
417 for( size_t i = 0; i < samples; i++ )
419 #ifdef WORDS_BIGENDIAN
420 *(out++) = GetQWLE( in );
422 *(out++) = GetQWBE( in );
428 static int16_t dat12tos16( uint16_t y )
430 static const uint16_t diff[16] = {
431 0x0000, 0x0000, 0x0100, 0x0200, 0x0300, 0x0400, 0x0500, 0x0600,
432 0x0A00, 0x0B00, 0x0C00, 0x0D00, 0x0E00, 0x0F00, 0x1000, 0x1000 };
433 static const uint8_t shift[16] = {
434 0, 0, 1, 2, 3, 4, 5, 6, 6, 5, 4, 3, 2, 1, 0, 0 };
437 return (y - diff[d]) << shift[d];
440 static void DAT12Decode( void *outp, const uint8_t *in, unsigned samples )
444 while( samples >= 2 )
446 *(out++) = dat12tos16(U16_AT(in) >> 4);
447 *(out++) = dat12tos16(U16_AT(in + 1) & ~0xF000);
453 *(out++) = dat12tos16(U16_AT(in) >> 4);
456 /*****************************************************************************
457 * DecoderClose: decoder destruction
458 *****************************************************************************/
459 static void DecoderClose( vlc_object_t *p_this )
461 decoder_t *p_dec = (decoder_t *)p_this;
463 free( p_dec->p_sys );
467 /* NOTE: Output buffers are always aligned since they are allocated by the araw plugin.
468 * Contrary to the decoder, the encoder can also assume that input buffers are aligned,
469 * since decoded audio blocks must always be aligned. */
471 static void U16IEncode( void *outp, const uint8_t *inp, unsigned samples )
473 const uint16_t *in = (const uint16_t *)inp;
474 uint16_t *out = outp;
476 for( size_t i = 0; i < samples; i++ )
477 *(out++) = bswap16( *(in++) + 0x8000 );
480 static void U16NEncode( void *outp, const uint8_t *inp, unsigned samples )
482 const uint16_t *in = (const uint16_t *)inp;
483 uint16_t *out = outp;
485 for( size_t i = 0; i < samples; i++ )
486 *(out++) = *(in++) + 0x8000;
489 static void U32IEncode( void *outp, const uint8_t *inp, unsigned samples )
491 const uint32_t *in = (const uint32_t *)inp;
492 uint32_t *out = outp;
494 for( size_t i = 0; i < samples; i++ )
495 *(out++) = bswap32( *(in++) + 0x80000000 );
498 static void U32NEncode( void *outp, const uint8_t *inp, unsigned samples )
500 const uint32_t *in = (const uint32_t *)inp;
501 uint32_t *out = outp;
503 for( size_t i = 0; i < samples; i++ )
504 *(out++) = *(in++) + 0x80000000;
507 static void Swap32Encode( void *outp, const uint8_t *inp, unsigned samples )
509 const int32_t *in = (const int32_t *)inp;
512 for( size_t i = 0; i < samples; i++ )
513 *(out++) = bswap32( *(in++) );
516 static void Swap64Encode( void *outp, const uint8_t *inp, unsigned samples )
518 const int64_t *in = (const int64_t *)inp;
521 for( size_t i = 0; i < samples; i++ )
522 *(out++) = bswap64( *(in++) );
525 static block_t *Encode( encoder_t *enc, block_t *in )
530 block_t *out = block_Alloc( in->i_nb_samples
531 * enc->fmt_out.audio.i_bytes_per_frame );
532 if( unlikely(out == NULL) )
535 out->i_flags = in->i_flags;
536 out->i_nb_samples = in->i_nb_samples;
537 out->i_dts = in->i_dts;
538 out->i_pts = in->i_pts;
539 out->i_length = in->i_length;
540 out->i_nb_samples = in->i_nb_samples;
542 void (*encode)(void *, const uint8_t *, unsigned) = (void *)enc->p_sys;
544 encode( out->p_buffer, in->p_buffer, in->i_nb_samples
545 * enc->fmt_out.audio.i_channels );
547 memcpy( out->p_buffer, in->p_buffer, in->i_buffer );
552 * Probes the PCM audio encoder.
554 static int EncoderOpen( vlc_object_t *p_this )
556 encoder_t *p_enc = (encoder_t *)p_this;
557 void (*encode)(void *, const uint8_t *, unsigned) = NULL;
559 switch( p_enc->fmt_out.i_codec )
564 p_enc->fmt_in.i_codec = VLC_CODEC_U8;
565 p_enc->fmt_out.audio.i_bitspersample = 8;
569 p_enc->fmt_in.i_codec = VLC_CODEC_S16N;
570 p_enc->fmt_out.audio.i_bitspersample = 16;
574 p_enc->fmt_in.i_codec = VLC_CODEC_S16N;
575 p_enc->fmt_out.audio.i_bitspersample = 16;
580 p_enc->fmt_in.i_codec = VLC_CODEC_S16N;
581 p_enc->fmt_out.audio.i_bitspersample = 16;
587 p_enc->fmt_out.audio.i_bitspersample = 24;
591 p_enc->fmt_in.i_codec = VLC_CODEC_S32N;
592 p_enc->fmt_out.audio.i_bitspersample = 32;
596 p_enc->fmt_in.i_codec = VLC_CODEC_S32N;
597 p_enc->fmt_out.audio.i_bitspersample = 32;
600 encode = Swap32Encode;
602 p_enc->fmt_in.i_codec = VLC_CODEC_S32N;
603 p_enc->fmt_out.audio.i_bitspersample = 32;
605 #ifdef WORDS_BIGENDIAN
610 encode = Swap32Encode;
612 p_enc->fmt_in.i_codec = VLC_CODEC_FL32;
613 p_enc->fmt_out.audio.i_bitspersample = 32;
615 #ifdef WORDS_BIGENDIAN
620 encode = Swap64Encode;
622 p_enc->fmt_in.i_codec = VLC_CODEC_FL64;
623 p_enc->fmt_out.audio.i_bitspersample = 64;
629 p_enc->p_sys = (void *)encode;
630 p_enc->pf_encode_audio = Encode;
631 p_enc->fmt_out.audio.i_bytes_per_frame =
632 (p_enc->fmt_out.audio.i_bitspersample / 8) *
633 p_enc->fmt_in.audio.i_channels;
634 p_enc->fmt_out.i_bitrate =
635 p_enc->fmt_in.audio.i_channels *
636 p_enc->fmt_in.audio.i_rate *
637 p_enc->fmt_out.audio.i_bitspersample;
639 msg_Dbg( p_enc, "samplerate:%dHz channels:%d bits/sample:%d",
640 p_enc->fmt_out.audio.i_rate, p_enc->fmt_out.audio.i_channels,
641 p_enc->fmt_out.audio.i_bitspersample );
645 #endif /* ENABLE_SOUT */