Fix plugin strings translation
[vlc.git] / modules / codec / aes3.c
1 /*****************************************************************************
2  * aes3.c: aes3 decoder/packetizer module
3  *****************************************************************************
4  * Copyright (C) 2008 the VideoLAN team
5  * $Id$
6  *
7  * Authors: Laurent Aimar <fenrir@videolan.org>
8  *
9  * This program is free software; you can redistribute it and/or modify
10  * it under the terms of the GNU General Public License as published by
11  * the Free Software Foundation; either version 2 of the License, or
12  * (at your option) any later version.
13  *
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 General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write to the Free Software
21  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
22  *****************************************************************************/
23
24 /*****************************************************************************
25  * Preamble
26  *****************************************************************************/
27 #ifdef HAVE_CONFIG_H
28 # include "config.h"
29 #endif
30
31 #include <vlc_common.h>
32 #include <vlc_plugin.h>
33 #include <vlc_codec.h>
34 #include <vlc_aout.h>
35 #include <assert.h>
36
37 /*****************************************************************************
38  * Module descriptor
39  *****************************************************************************/
40 static int  OpenDecoder   ( vlc_object_t * );
41 static int  OpenPacketizer( vlc_object_t * );
42 static void Close         ( vlc_object_t * );
43
44 vlc_module_begin ()
45
46     set_category( CAT_INPUT )
47     set_subcategory( SUBCAT_INPUT_ACODEC )
48     set_description( N_("AES3/SMPTE 302M audio decoder") )
49     set_capability( "decoder", 100 )
50     set_callbacks( OpenDecoder, Close )
51
52     add_submodule ()
53     set_description( N_("AES3/SMPTE 302M audio packetizer") )
54     set_capability( "packetizer", 100 )
55     set_callbacks( OpenPacketizer, Close )
56
57 vlc_module_end ()
58
59 /*****************************************************************************
60  * decoder_sys_t : aes3 decoder descriptor
61  *****************************************************************************/
62 struct decoder_sys_t
63 {
64     /*
65      * Output properties
66      */
67     audio_date_t end_date;
68 };
69
70 #define AES3_HEADER_LEN 4
71
72 /*****************************************************************************
73  * Local prototypes
74  *****************************************************************************/
75 static int Open( decoder_t *p_dec, bool b_packetizer );
76
77 static block_t *Parse( decoder_t *p_dec, int *pi_frame_length, int *pi_bits,
78                        block_t **pp_block, bool b_packetizer );
79
80 static inline uint8_t Reverse8( int n )
81 {
82     n = ((n >> 1) & 0x55) | ((n << 1) & 0xaa);
83     n = ((n >> 2) & 0x33) | ((n << 2) & 0xcc);
84     n = ((n >> 4) & 0x0f) | ((n << 4) & 0xf0);
85     return n;
86 }
87
88 /*****************************************************************************
89  * OpenDecoder:
90  *****************************************************************************/
91 static int OpenDecoder( vlc_object_t *p_this )
92 {
93     decoder_t *p_dec = (decoder_t*)p_this;
94
95     return Open( p_dec, false );
96 }
97
98 /*****************************************************************************
99  * OpenPacketizer:
100  *****************************************************************************/
101 static int OpenPacketizer( vlc_object_t *p_this )
102 {
103     decoder_t *p_dec = (decoder_t*)p_this;
104
105     return Open( p_dec, true );
106 }
107
108 /*****************************************************************************
109  * Close : aes3 decoder destruction
110  *****************************************************************************/
111 static void Close( vlc_object_t *p_this )
112 {
113     decoder_t *p_dec = (decoder_t*)p_this;
114     free( p_dec->p_sys );
115 }
116
117 /*****************************************************************************
118  * Decode: decodes an aes3 frame.
119  ****************************************************************************
120  * Beware, this function must be fed with complete frames (PES packet).
121  *****************************************************************************/
122 static aout_buffer_t *Decode( decoder_t *p_dec, block_t **pp_block )
123 {
124     decoder_sys_t *p_sys = p_dec->p_sys;
125     block_t       *p_block;
126     aout_buffer_t *p_aout_buffer;
127     int            i_frame_length, i_bits;
128
129     p_block = Parse( p_dec, &i_frame_length, &i_bits, pp_block, false );
130     if( !p_block )
131         return NULL;
132
133     p_aout_buffer = decoder_NewAudioBuffer( p_dec, i_frame_length );
134     if( p_aout_buffer == NULL )
135         goto exit;
136
137     p_aout_buffer->start_date = aout_DateGet( &p_sys->end_date );
138     p_aout_buffer->end_date = aout_DateIncrement( &p_sys->end_date, i_frame_length );
139
140     p_block->i_buffer -= AES3_HEADER_LEN;
141     p_block->p_buffer += AES3_HEADER_LEN;
142
143     if( i_bits == 24 )
144     {
145         uint8_t *p_out = p_aout_buffer->p_buffer;
146
147         while( p_block->i_buffer / 7 )
148         {
149             p_out[0] = Reverse8( p_block->p_buffer[0] );
150             p_out[1] = Reverse8( p_block->p_buffer[1] );
151             p_out[2] = Reverse8( p_block->p_buffer[2] );
152
153             p_out[3] = (Reverse8( p_block->p_buffer[3] ) >> 4) | ( (Reverse8( p_block->p_buffer[4] ) << 4) & 0xf0 );
154             p_out[4] = (Reverse8( p_block->p_buffer[4] ) >> 4) | ( (Reverse8( p_block->p_buffer[5] ) << 4) & 0xf0 );
155             p_out[5] = (Reverse8( p_block->p_buffer[5] ) >> 4) | ( (Reverse8( p_block->p_buffer[6] ) << 4) & 0xf0 );
156
157             p_block->i_buffer -= 7;
158             p_block->p_buffer += 7;
159             p_out += 6;
160         }
161
162     }
163     else if( i_bits == 20 )
164     {
165         uint8_t *p_out = p_aout_buffer->p_buffer;
166
167         while( p_block->i_buffer / 6 )
168         {
169             p_out[0] =                                            ( (Reverse8( p_block->p_buffer[0] ) << 4) & 0xf0 );
170             p_out[1] = ( Reverse8( p_block->p_buffer[0]) >> 4 ) | ( (Reverse8( p_block->p_buffer[1]) << 4 ) & 0xf0 );
171             p_out[2] = ( Reverse8( p_block->p_buffer[1]) >> 4 ) | ( (Reverse8( p_block->p_buffer[2]) << 4 ) & 0xf0 );
172
173             p_out[3] =                                            ( (Reverse8( p_block->p_buffer[3] ) << 4) & 0xf0 );
174             p_out[4] = ( Reverse8( p_block->p_buffer[3]) >> 4 ) | ( (Reverse8( p_block->p_buffer[4]) << 4 ) & 0xf0 );
175             p_out[5] = ( Reverse8( p_block->p_buffer[4]) >> 4 ) | ( (Reverse8( p_block->p_buffer[5]) << 4 ) & 0xf0 );
176
177             p_block->i_buffer -= 6;
178             p_block->p_buffer += 6;
179             p_out += 6;
180         }
181     }
182     else
183     {
184         uint8_t *p_out = p_aout_buffer->p_buffer;
185
186         assert( i_bits == 16 );
187
188         while( p_block->i_buffer / 5 )
189         {
190             p_out[0] = Reverse8( p_block->p_buffer[0] );
191             p_out[1] = Reverse8( p_block->p_buffer[1] );
192
193             p_out[2] = (Reverse8( p_block->p_buffer[2] ) >> 4) | ( (Reverse8( p_block->p_buffer[3] ) << 4) & 0xf0 );
194             p_out[3] = (Reverse8( p_block->p_buffer[3] ) >> 4) | ( (Reverse8( p_block->p_buffer[4] ) << 4) & 0xf0 );
195
196             p_block->i_buffer -= 5;
197             p_block->p_buffer += 5;
198             p_out += 4;
199         }
200     }
201
202 exit:
203     block_Release( p_block );
204     return p_aout_buffer;
205 }
206
207 /*****************************************************************************
208  * Packetize: packetizes an aes3 frame.
209  ****************************************************************************
210  * Beware, this function must be fed with complete frames (PES packet).
211  *****************************************************************************/
212 static block_t *Packetize( decoder_t *p_dec, block_t **pp_block )
213 {
214     decoder_sys_t *p_sys = p_dec->p_sys;
215     block_t       *p_block;
216     int           i_frame_length, i_bits;
217
218     p_block = Parse( p_dec, &i_frame_length, &i_bits, pp_block, true );
219     if( !p_block )
220         return NULL;
221
222     p_block->i_pts = p_block->i_dts = aout_DateGet( &p_sys->end_date );
223     p_block->i_length = aout_DateIncrement( &p_sys->end_date, i_frame_length ) - p_block->i_pts;
224
225     /* Just pass on the incoming frame */
226     return p_block;
227 }
228
229 /*****************************************************************************
230  *
231  ****************************************************************************/
232 static int Open( decoder_t *p_dec, bool b_packetizer )
233 {
234     decoder_sys_t *p_sys;
235
236     if( p_dec->fmt_in.i_codec != VLC_FOURCC('a','e','s','3') )
237         return VLC_EGENERIC;
238
239     /* Allocate the memory needed to store the decoder's structure */
240     p_dec->p_sys = p_sys = malloc( sizeof(decoder_sys_t) );
241
242     if( !p_sys )
243         return VLC_EGENERIC;
244
245     /* Misc init */
246     aout_DateInit( &p_sys->end_date, 48000 );
247     aout_DateSet( &p_sys->end_date, 0 );
248
249     /* Set output properties */
250     p_dec->fmt_out.i_cat = AUDIO_ES;
251     p_dec->fmt_out.audio.i_rate = 48000;
252
253     /* Set callback */
254     if( b_packetizer )
255     {
256         p_dec->fmt_out.i_codec = VLC_FOURCC('a','e','s','3');
257
258         p_dec->pf_decode_audio = NULL;
259         p_dec->pf_packetize    = Packetize;
260     }
261     else
262     {
263         p_dec->pf_decode_audio = Decode;
264         p_dec->pf_packetize    = NULL;
265     }
266     return VLC_SUCCESS;
267 }
268
269 static const unsigned int pi_original_channels[4] = {
270     AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT,
271     AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT |
272         AOUT_CHAN_CENTER | AOUT_CHAN_LFE,
273     AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT |
274         AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT |
275         AOUT_CHAN_CENTER | AOUT_CHAN_LFE,
276     AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT |
277         AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT |
278         AOUT_CHAN_MIDDLELEFT | AOUT_CHAN_MIDDLERIGHT |
279         AOUT_CHAN_CENTER | AOUT_CHAN_LFE,
280 };
281
282 static block_t *Parse( decoder_t *p_dec, int *pi_frame_length, int *pi_bits,
283                        block_t **pp_block, bool b_packetizer )
284 {
285     decoder_sys_t *p_sys = p_dec->p_sys;
286     block_t       *p_block;
287     uint32_t h;
288     unsigned int i_size;
289     int i_channels;
290     int i_id;
291     int i_bits;
292
293     if( !pp_block || !*pp_block ) return NULL;
294
295     p_block = *pp_block;
296     *pp_block = NULL; /* So the packet doesn't get re-sent */
297
298     /* Date management */
299     if( p_block->i_pts > 0 &&
300         p_block->i_pts != aout_DateGet( &p_sys->end_date ) )
301     {
302         aout_DateSet( &p_sys->end_date, p_block->i_pts );
303     }
304
305     if( !aout_DateGet( &p_sys->end_date ) )
306     {
307         /* We've just started the stream, wait for the first PTS. */
308         block_Release( p_block );
309         return NULL;
310     }
311
312     if( p_block->i_buffer <= AES3_HEADER_LEN )
313     {
314         msg_Err(p_dec, "frame is too short");
315         block_Release( p_block );
316         return NULL;
317     }
318
319     /*
320      * AES3 header :
321      * size:            16
322      * number channels   2
323      * channel_id        8
324      * bits per samples  2
325      * alignments        4
326      */
327
328     h = GetDWBE( p_block->p_buffer );
329     i_size = (h >> 16) & 0xffff;
330     i_channels = 2 + 2*( (h >> 14) & 0x03 );
331     i_id = (h >> 6) & 0xff;
332     i_bits = 16 + 4*( (h >> 4)&0x03 );
333
334     if( AES3_HEADER_LEN + i_size != p_block->i_buffer || i_bits > 24 )
335     {
336         msg_Err(p_dec, "frame has invalid header");
337         block_Release( p_block );
338         return NULL;
339     }
340
341     /* Set output properties */
342     if( b_packetizer )
343     {
344         p_dec->fmt_out.audio.i_bitspersample = i_bits;
345     }
346     else
347     {
348         p_dec->fmt_out.i_codec = i_bits == 16 ? VLC_FOURCC('s','1','6','l') : VLC_FOURCC('s','2','4','l');
349         p_dec->fmt_out.audio.i_bitspersample = i_bits == 16 ? 16 : 24;
350     }
351
352     p_dec->fmt_out.audio.i_channels = i_channels;
353     p_dec->fmt_out.audio.i_original_channels = pi_original_channels[i_channels/2-1];
354     p_dec->fmt_out.audio.i_physical_channels = pi_original_channels[i_channels/2-1] & AOUT_CHAN_PHYSMASK;
355
356     *pi_frame_length = (p_block->i_buffer - AES3_HEADER_LEN) / ( (4+i_bits) * i_channels / 8 );
357     *pi_bits = i_bits;
358     return p_block;
359 }
360