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 U24BDecode( void *, const uint8_t *, unsigned );
91 static void U24LDecode( void *, const uint8_t *, unsigned );
92 static void S24BDecode( void *, const uint8_t *, unsigned );
93 static void S24LDecode( void *, const uint8_t *, unsigned );
94 static void S24B32Decode( void *, const uint8_t *, unsigned );
95 static void S24L32Decode( void *, const uint8_t *, unsigned );
96 static void U32BDecode( void *, const uint8_t *, unsigned );
97 static void U32LDecode( void *, const uint8_t *, unsigned );
98 static void S32IDecode( void *, const uint8_t *, unsigned );
99 static void F32IDecode( void *, const uint8_t *, unsigned );
100 static void F64IDecode( void *, const uint8_t *, unsigned );
101 static void DAT12Decode( void *, const uint8_t *, unsigned );
103 /*****************************************************************************
104 * DecoderOpen: probe the decoder and return score
105 *****************************************************************************/
106 static int DecoderOpen( vlc_object_t *p_this )
108 decoder_t *p_dec = (decoder_t*)p_this;
109 vlc_fourcc_t format = p_dec->fmt_in.i_codec;
111 switch( p_dec->fmt_in.i_codec )
113 case VLC_FOURCC('a','r','a','w'):
114 case VLC_FOURCC('p','c','m',' '):
115 case VLC_FOURCC('a','f','l','t'):
116 /* _signed_ big endian samples (mov) */
117 case VLC_FOURCC('t','w','o','s'):
118 /* _signed_ little endian samples (mov) */
119 case VLC_FOURCC('s','o','w','t'):
121 vlc_fourcc_GetCodecAudio( p_dec->fmt_in.i_codec,
122 p_dec->fmt_in.audio.i_bitspersample );
125 msg_Err( p_dec, "bad parameters(bits/sample)" );
131 void (*decode) (void *, const uint8_t *, unsigned) = NULL;
136 #ifdef WORDS_BIGENDIAN
141 format = VLC_CODEC_FL64;
146 #ifdef WORDS_BIGENDIAN
151 format = VLC_CODEC_FL32;
157 format = VLC_CODEC_S32N;
162 format = VLC_CODEC_S32N;
167 format = VLC_CODEC_S32N;
172 case VLC_CODEC_S24B32:
173 format = VLC_CODEC_S32N;
174 decode = S24B32Decode;
177 case VLC_CODEC_S24L32:
178 format = VLC_CODEC_S32N;
179 decode = S24L32Decode;
183 format = VLC_CODEC_S32N;
188 format = VLC_CODEC_S32N;
193 format = VLC_CODEC_S32N;
198 format = VLC_CODEC_S32N;
203 format = VLC_CODEC_S32N;
208 format = VLC_CODEC_S16N;
213 format = VLC_CODEC_S16N;
218 format = VLC_CODEC_S16N;
223 case VLC_CODEC_DAT12:
224 format = VLC_CODEC_S16N;
225 decode = DAT12Decode;
230 format = VLC_CODEC_U8;
238 if( p_dec->fmt_in.audio.i_channels <= 0 ||
239 p_dec->fmt_in.audio.i_channels > AOUT_CHAN_MAX )
241 msg_Err( p_dec, "bad channels count (1-9): %i",
242 p_dec->fmt_in.audio.i_channels );
246 if( p_dec->fmt_in.audio.i_rate <= 0 )
248 msg_Err( p_dec, "bad samplerate: %d Hz", p_dec->fmt_in.audio.i_rate );
252 msg_Dbg( p_dec, "samplerate:%dHz channels:%d bits/sample:%d",
253 p_dec->fmt_in.audio.i_rate, p_dec->fmt_in.audio.i_channels,
254 p_dec->fmt_in.audio.i_bitspersample );
256 /* Allocate the memory needed to store the decoder's structure */
257 decoder_sys_t *p_sys = malloc(sizeof(*p_sys));
258 if( unlikely(p_sys == NULL) )
261 /* Set output properties */
262 p_dec->fmt_out.i_cat = AUDIO_ES;
263 p_dec->fmt_out.i_codec = format;
264 p_dec->fmt_out.audio.i_format = format;
265 p_dec->fmt_out.audio.i_rate = p_dec->fmt_in.audio.i_rate;
266 if( p_dec->fmt_in.audio.i_physical_channels )
267 p_dec->fmt_out.audio.i_physical_channels =
268 p_dec->fmt_in.audio.i_physical_channels;
270 p_dec->fmt_out.audio.i_physical_channels =
271 pi_channels_maps[p_dec->fmt_in.audio.i_channels];
272 if( p_dec->fmt_in.audio.i_original_channels )
273 p_dec->fmt_out.audio.i_original_channels =
274 p_dec->fmt_in.audio.i_original_channels;
276 p_dec->fmt_out.audio.i_original_channels =
277 p_dec->fmt_out.audio.i_physical_channels;
278 aout_FormatPrepare( &p_dec->fmt_out.audio );
280 p_sys->decode = decode;
281 p_sys->framebits = bits * p_dec->fmt_out.audio.i_channels;
282 assert( p_sys->framebits );
284 date_Init( &p_sys->end_date, p_dec->fmt_out.audio.i_rate, 1 );
285 date_Set( &p_sys->end_date, 0 );
287 p_dec->pf_decode_audio = DecodeBlock;
288 p_dec->p_sys = p_sys;
293 /****************************************************************************
294 * DecodeBlock: the whole thing
295 ****************************************************************************
296 * This function must be fed with whole samples (see nBlockAlign).
297 ****************************************************************************/
298 static block_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
300 decoder_sys_t *p_sys = p_dec->p_sys;
302 if( !pp_block || !*pp_block ) return NULL;
304 block_t *p_block = *pp_block;
306 if( p_block->i_pts > VLC_TS_INVALID &&
307 p_block->i_pts != date_Get( &p_sys->end_date ) )
309 date_Set( &p_sys->end_date, p_block->i_pts );
311 else if( !date_Get( &p_sys->end_date ) )
313 /* We've just started the stream, wait for the first PTS. */
314 block_Release( p_block );
318 /* Don't re-use the same pts twice */
319 p_block->i_pts = VLC_TS_INVALID;
321 unsigned samples = (8 * p_block->i_buffer) / p_sys->framebits;
324 block_Release( p_block );
328 /* Create chunks of max 1024 samples */
329 if( samples > 1024 ) samples = 1024;
331 block_t *p_out = decoder_NewAudioBuffer( p_dec, samples );
334 block_Release( p_block );
338 p_out->i_pts = date_Get( &p_sys->end_date );
339 p_out->i_length = date_Increment( &p_sys->end_date, samples )
342 if( p_sys->decode != NULL )
343 p_sys->decode( p_out->p_buffer, p_block->p_buffer,
344 samples * p_dec->fmt_in.audio.i_channels );
346 memcpy( p_out->p_buffer, p_block->p_buffer, p_out->i_buffer );
348 samples = (samples * p_sys->framebits) / 8;
349 p_block->p_buffer += samples;
350 p_block->i_buffer -= samples;
355 static void S8Decode( void *outp, const uint8_t *in, unsigned samples )
359 for( size_t i = 0; i < samples; i++ )
360 out[i] = in[i] ^ 0x80;
363 static void U16BDecode( void *outp, const uint8_t *in, unsigned samples )
365 uint16_t *out = outp;
367 for( size_t i = 0; i < samples; i++ )
369 *(out++) = GetWBE( in ) - 0x8000;
374 static void U16LDecode( void *outp, const uint8_t *in, unsigned samples )
376 uint16_t *out = outp;
378 for( size_t i = 0; i < samples; i++ )
380 *(out++) = GetWLE( in ) - 0x8000;
385 static void S16IDecode( void *out, const uint8_t *in, unsigned samples )
387 swab( in, out, samples * 2 );
390 static void S20BDecode( void *outp, const uint8_t *in, unsigned samples )
394 while( samples >= 2 )
396 uint32_t dw = U32_AT(in);
398 *(out++) = dw & ~0xFFF;
399 *(out++) = (dw << 20) | (*in << 12);
404 /* No U32_AT() for the last odd sample: avoid off-by-one overflow! */
406 *(out++) = (U16_AT(in) << 16) | ((in[2] & 0xF0) << 8);
409 static void U24BDecode( void *outp, const uint8_t *in, unsigned samples )
411 uint32_t *out = outp;
413 for( size_t i = 0; i < samples; i++ )
415 uint32_t s = ((in[0] << 24) | (in[1] << 16) | (in[2] << 8)) - 0x80000000;
421 static void U24LDecode( void *outp, const uint8_t *in, unsigned samples )
423 uint32_t *out = outp;
425 for( size_t i = 0; i < samples; i++ )
427 uint32_t s = ((in[2] << 24) | (in[1] << 16) | (in[0] << 8)) - 0x80000000;
433 static void S24BDecode( void *outp, const uint8_t *in, unsigned samples )
435 uint32_t *out = outp;
437 for( size_t i = 0; i < samples; i++ )
439 uint32_t s = ((in[0] << 24) | (in[1] << 16) | (in[2] << 8));
445 static void S24LDecode( void *outp, const uint8_t *in, unsigned samples )
447 uint32_t *out = outp;
449 for( size_t i = 0; i < samples; i++ )
451 uint32_t s = ((in[2] << 24) | (in[1] << 16) | (in[0] << 8));
457 static void S24B32Decode( void *outp, const uint8_t *in, unsigned samples )
459 uint32_t *out = outp;
461 for( size_t i = 0; i < samples; i++ )
463 *(out++) = GetDWBE( in ) << 8;
468 static void S24L32Decode( void *outp, const uint8_t *in, unsigned samples )
470 uint32_t *out = outp;
472 for( size_t i = 0; i < samples; i++ )
474 *(out++) = GetDWLE( in ) << 8;
479 static void U32BDecode( void *outp, const uint8_t *in, unsigned samples )
481 uint32_t *out = outp;
483 for( size_t i = 0; i < samples; i++ )
485 *(out++) = GetDWBE( in ) - 0x80000000;
490 static void U32LDecode( void *outp, const uint8_t *in, unsigned samples )
492 uint32_t *out = outp;
494 for( size_t i = 0; i < samples; i++ )
496 *(out++) = GetDWLE( in ) - 0x80000000;
501 static void S32IDecode( void *outp, const uint8_t *in, unsigned samples )
505 for( size_t i = 0; i < samples; i++ )
507 #ifdef WORDS_BIGENDIAN
508 *(out++) = GetDWLE( in );
510 *(out++) = GetDWBE( in );
516 static void F32IDecode( void *outp, const uint8_t *in, unsigned samples )
520 for( size_t i = 0; i < samples; i++ )
522 union { float f; uint32_t u; } s;
524 #ifdef WORDS_BIGENDIAN
534 static void F64IDecode( void *outp, const uint8_t *in, unsigned samples )
538 for( size_t i = 0; i < samples; i++ )
540 union { double d; uint64_t u; } s;
542 #ifdef WORDS_BIGENDIAN
552 static int16_t dat12tos16( uint16_t y )
554 static const uint16_t diff[16] = {
555 0x0000, 0x0000, 0x0100, 0x0200, 0x0300, 0x0400, 0x0500, 0x0600,
556 0x0A00, 0x0B00, 0x0C00, 0x0D00, 0x0E00, 0x0F00, 0x1000, 0x1000 };
557 static const uint8_t shift[16] = {
558 0, 0, 1, 2, 3, 4, 5, 6, 6, 5, 4, 3, 2, 1, 0, 0 };
561 return (y - diff[d]) << shift[d];
564 static void DAT12Decode( void *outp, const uint8_t *in, unsigned samples )
568 while( samples >= 2 )
570 *(out++) = dat12tos16(U16_AT(in) >> 4);
571 *(out++) = dat12tos16(U16_AT(in + 1) & ~0xF000);
577 *(out++) = dat12tos16(U16_AT(in) >> 4);
580 /*****************************************************************************
581 * DecoderClose: decoder destruction
582 *****************************************************************************/
583 static void DecoderClose( vlc_object_t *p_this )
585 decoder_t *p_dec = (decoder_t *)p_this;
587 free( p_dec->p_sys );
591 /* NOTE: Output buffers are always aligned since they are allocated by the araw plugin.
592 * Contrary to the decoder, the encoder can also assume that input buffers are aligned,
593 * since decoded audio blocks must always be aligned. */
595 static void U16IEncode( void *outp, const uint8_t *inp, unsigned samples )
597 const uint16_t *in = (const uint16_t *)inp;
598 uint16_t *out = outp;
600 for( size_t i = 0; i < samples; i++ )
601 *(out++) = bswap16( *(in++) + 0x8000 );
604 static void U16NEncode( void *outp, const uint8_t *inp, unsigned samples )
606 const uint16_t *in = (const uint16_t *)inp;
607 uint16_t *out = outp;
609 for( size_t i = 0; i < samples; i++ )
610 *(out++) = *(in++) + 0x8000;
613 static void U24BEncode( void *outp, const uint8_t *inp, unsigned samples )
615 const uint32_t *in = (const uint32_t *)inp;
618 for( size_t i = 0; i < samples; i++ )
620 uint32_t s = *(in++);
621 *(out++) = (s >> 24) + 0x80;
622 *(out++) = (s >> 16);
627 static void U24LEncode( void *outp, const uint8_t *inp, unsigned samples )
629 const uint32_t *in = (const uint32_t *)inp;
632 for( size_t i = 0; i < samples; i++ )
634 uint32_t s = *(in++);
636 *(out++) = (s >> 16);
637 *(out++) = (s >> 24) + 0x80;
641 static void S24BEncode( void *outp, const uint8_t *inp, unsigned samples )
643 const uint32_t *in = (const uint32_t *)inp;
646 for( size_t i = 0; i < samples; i++ )
648 uint32_t s = *(in++);
649 *(out++) = (s >> 24);
650 *(out++) = (s >> 16);
655 static void S24LEncode( void *outp, const uint8_t *inp, unsigned samples )
657 const uint32_t *in = (const uint32_t *)inp;
660 for( size_t i = 0; i < samples; i++ )
662 uint32_t s = *(in++);
664 *(out++) = (s >> 16);
665 *(out++) = (s >> 24);
669 static void U32IEncode( void *outp, const uint8_t *inp, unsigned samples )
671 const uint32_t *in = (const uint32_t *)inp;
672 uint32_t *out = outp;
674 for( size_t i = 0; i < samples; i++ )
675 *(out++) = bswap32( *(in++) + 0x80000000 );
678 static void U32NEncode( void *outp, const uint8_t *inp, unsigned samples )
680 const uint32_t *in = (const uint32_t *)inp;
681 uint32_t *out = outp;
683 for( size_t i = 0; i < samples; i++ )
684 *(out++) = *(in++) + 0x80000000;
687 static void S32IEncode( void *outp, const uint8_t *inp, unsigned samples )
689 const int32_t *in = (const int32_t *)inp;
692 for( size_t i = 0; i < samples; i++ )
693 *(out++) = bswap32( *(in++) );
696 static void F32IEncode( void *outp, const uint8_t *inp, unsigned samples )
698 const float *in = (const float *)inp;
701 for( size_t i = 0; i < samples; i++ )
703 union { float f; uint32_t u; char b[4]; } s;
706 s.u = bswap32( s.u );
707 memcpy( out, s.b, 4 );
712 static void F64IEncode( void *outp, const uint8_t *inp, unsigned samples )
714 const double *in = (const double *)inp;
717 for( size_t i = 0; i < samples; i++ )
719 union { double d; uint64_t u; char b[8]; } s;
722 s.u = bswap64( s.u );
723 memcpy( out, s.b, 8 );
728 static block_t *Encode( encoder_t *enc, block_t *in )
733 block_t *out = block_Alloc( in->i_nb_samples
734 * enc->fmt_out.audio.i_bytes_per_frame );
735 if( unlikely(out == NULL) )
738 out->i_flags = in->i_flags;
739 out->i_nb_samples = in->i_nb_samples;
740 out->i_dts = in->i_dts;
741 out->i_pts = in->i_pts;
742 out->i_length = in->i_length;
743 out->i_nb_samples = in->i_nb_samples;
745 void (*encode)(void *, const uint8_t *, unsigned) = (void *)enc->p_sys;
747 encode( out->p_buffer, in->p_buffer, in->i_nb_samples
748 * enc->fmt_out.audio.i_channels );
750 memcpy( out->p_buffer, in->p_buffer, in->i_buffer );
755 * Probes the PCM audio encoder.
757 static int EncoderOpen( vlc_object_t *p_this )
759 encoder_t *p_enc = (encoder_t *)p_this;
760 void (*encode)(void *, const uint8_t *, unsigned) = NULL;
762 switch( p_enc->fmt_out.i_codec )
767 p_enc->fmt_in.i_codec = VLC_CODEC_U8;
768 p_enc->fmt_out.audio.i_bitspersample = 8;
772 p_enc->fmt_in.i_codec = VLC_CODEC_S16N;
773 p_enc->fmt_out.audio.i_bitspersample = 16;
777 p_enc->fmt_in.i_codec = VLC_CODEC_S16N;
778 p_enc->fmt_out.audio.i_bitspersample = 16;
783 p_enc->fmt_in.i_codec = VLC_CODEC_S16N;
784 p_enc->fmt_out.audio.i_bitspersample = 16;
788 p_enc->fmt_in.i_codec = VLC_CODEC_S32N;
789 p_enc->fmt_out.audio.i_bitspersample = 24;
793 p_enc->fmt_in.i_codec = VLC_CODEC_S32N;
794 p_enc->fmt_out.audio.i_bitspersample = 24;
798 p_enc->fmt_in.i_codec = VLC_CODEC_S32N;
799 p_enc->fmt_out.audio.i_bitspersample = 24;
803 p_enc->fmt_in.i_codec = VLC_CODEC_S32N;
804 p_enc->fmt_out.audio.i_bitspersample = 24;
808 p_enc->fmt_in.i_codec = VLC_CODEC_S32N;
809 p_enc->fmt_out.audio.i_bitspersample = 32;
813 p_enc->fmt_in.i_codec = VLC_CODEC_S32N;
814 p_enc->fmt_out.audio.i_bitspersample = 32;
819 p_enc->fmt_in.i_codec = VLC_CODEC_S32N;
820 p_enc->fmt_out.audio.i_bitspersample = 32;
822 #ifdef WORDS_BIGENDIAN
829 p_enc->fmt_in.i_codec = VLC_CODEC_FL32;
830 p_enc->fmt_out.audio.i_bitspersample = 32;
832 #ifdef WORDS_BIGENDIAN
839 p_enc->fmt_in.i_codec = VLC_CODEC_FL64;
840 p_enc->fmt_out.audio.i_bitspersample = 64;
846 p_enc->p_sys = (void *)encode;
847 p_enc->pf_encode_audio = Encode;
848 p_enc->fmt_out.audio.i_bytes_per_frame =
849 (p_enc->fmt_out.audio.i_bitspersample / 8) *
850 p_enc->fmt_in.audio.i_channels;
851 p_enc->fmt_out.i_bitrate =
852 p_enc->fmt_in.audio.i_channels *
853 p_enc->fmt_in.audio.i_rate *
854 p_enc->fmt_out.audio.i_bitspersample;
856 msg_Dbg( p_enc, "samplerate:%dHz channels:%d bits/sample:%d",
857 p_enc->fmt_out.audio.i_rate, p_enc->fmt_out.audio.i_channels,
858 p_enc->fmt_out.audio.i_bitspersample );
862 #endif /* ENABLE_SOUT */