Old RC: "rate" is a float nowadays (fix #4088)
[vlc/asuraparaju-public.git] / modules / codec / wmafixed / wma.c
blobb36186137393cf305239f2c36156a7ff0c6ad6a4
1 /*****************************************************************************
2 * wma.c: wma decoder using integer decoder from Rockbox, based on FFmpeg
3 *****************************************************************************
4 * Copyright (C) 2008-2009 M2X
6 * Authors: Rafaël Carré <rcarre@m2x.nl>
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; either version 2 of the License, or
11 * (at your option) any later version.
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
18 * You should have received a copy of the GNU General Public License
19 * along with this program; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
21 *****************************************************************************/
23 /*****************************************************************************
24 * Preamble
25 *****************************************************************************/
26 #ifdef HAVE_CONFIG_H
27 # include "config.h"
28 #endif
30 #include <vlc_common.h>
31 #include <vlc_plugin.h>
32 #include <vlc_codec.h>
33 #include <vlc_aout.h>
34 #include <vlc_block_helper.h>
35 #include <vlc_bits.h>
37 #include <assert.h>
39 #include "wmadec.h"
41 /*****************************************************************************
42 * decoder_sys_t : wma decoder descriptor
43 *****************************************************************************/
44 struct decoder_sys_t
46 date_t end_date; /* To set the PTS */
47 WMADecodeContext wmadec; /* name is self explanative */
49 int32_t *p_output; /* buffer where the frames are rendered */
51 /* to not give too much samples at once to the audio output */
52 int8_t *p_samples; /* point into p_output */
53 unsigned int i_samples; /* number of buffered samples available */
56 /* FIXME : check supported configurations */
57 /* channel configuration */
58 static unsigned int pi_channels_maps[7] =
61 AOUT_CHAN_CENTER,
62 AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT,
63 AOUT_CHAN_CENTER | AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT,
64 AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT,
65 AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER | AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT,
66 AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER | AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT | AOUT_CHAN_LFE
69 /*****************************************************************************
70 * Local prototypes
71 *****************************************************************************/
72 static int OpenDecoder ( vlc_object_t * );
73 static void CloseDecoder ( vlc_object_t * );
75 static aout_buffer_t *DecodeFrame ( decoder_t *, block_t ** );
77 /*****************************************************************************
78 * Module descriptor
79 *****************************************************************************/
80 vlc_module_begin();
81 set_category( CAT_INPUT );
82 set_subcategory( SUBCAT_INPUT_ACODEC );
83 set_description( _("WMA v1/v2 fixed point audio decoder") );
84 set_capability( "decoder", 80 );
85 add_shortcut( "wmafixed" )
86 set_callbacks( OpenDecoder, CloseDecoder );
87 vlc_module_end();
89 /*****************************************************************************
90 * SplitBuffer: Needed because aout really doesn't like big audio chunk and
91 * wma produces easily > 30000 samples...
92 *****************************************************************************/
93 static aout_buffer_t *SplitBuffer( decoder_t *p_dec )
95 decoder_sys_t *p_sys = p_dec->p_sys;
96 unsigned int i_samples = __MIN( p_sys->i_samples, 2048 );
97 aout_buffer_t *p_buffer;
99 if( i_samples == 0 ) return NULL;
101 if( !( p_buffer = p_dec->pf_aout_buffer_new( p_dec, i_samples ) ) )
102 return NULL;
104 p_buffer->i_pts = date_Get( &p_sys->end_date );
105 p_buffer->i_length = date_Increment( &p_sys->end_date, i_samples )
106 - p_buffer->i_pts;
108 memcpy( p_buffer->p_buffer, p_sys->p_samples, p_buffer->i_buffer );
109 p_sys->p_samples += p_buffer->i_buffer;
110 p_sys->i_samples -= i_samples;
112 return p_buffer;
115 /*****************************************************************************
116 * OpenDecoder: probe the decoder and return score
117 *****************************************************************************/
118 static int OpenDecoder( vlc_object_t *p_this )
120 decoder_t *p_dec = (decoder_t*)p_this;
121 decoder_sys_t *p_sys;
123 if( p_dec->fmt_in.i_codec != VLC_CODEC_WMA1 &&
124 p_dec->fmt_in.i_codec != VLC_CODEC_WMA2 )
126 return VLC_EGENERIC;
129 /* Allocate the memory needed to store the decoder's structure */
130 p_dec->p_sys = p_sys = (decoder_sys_t *)malloc(sizeof(decoder_sys_t));
131 if( !p_sys )
132 return VLC_ENOMEM;
134 memset( p_sys, 0, sizeof( decoder_sys_t ) );
136 /* Date */
137 date_Init( &p_sys->end_date, p_dec->fmt_in.audio.i_rate, 1 );
139 /* Set output properties */
140 p_dec->fmt_out.i_cat = AUDIO_ES;
141 p_dec->fmt_out.i_codec = VLC_CODEC_FI32;
142 p_dec->fmt_out.audio.i_bitspersample = p_dec->fmt_in.audio.i_bitspersample;
143 p_dec->fmt_out.audio.i_rate = p_dec->fmt_in.audio.i_rate;
145 p_dec->fmt_out.audio.i_channels = p_dec->fmt_in.audio.i_channels;
147 assert( p_dec->fmt_out.audio.i_channels <
148 ( sizeof( pi_channels_maps ) / sizeof( pi_channels_maps[0] ) ) );
150 p_dec->fmt_out.audio.i_original_channels =
151 p_dec->fmt_out.audio.i_physical_channels =
152 pi_channels_maps[p_dec->fmt_out.audio.i_channels];
154 /* aout core assumes this number is not 0 and uses it in divisions */
155 assert( p_dec->fmt_out.audio.i_physical_channels != 0 );
157 asf_waveformatex_t wfx;
158 wfx.rate = p_dec->fmt_in.audio.i_rate;
159 wfx.bitrate = p_dec->fmt_in.i_bitrate;
160 wfx.channels = p_dec->fmt_in.audio.i_channels;
161 wfx.blockalign = p_dec->fmt_in.audio.i_blockalign;
162 wfx.bitspersample = p_dec->fmt_in.audio.i_bitspersample;
164 msg_Dbg( p_dec, "samplerate %d bitrate %d channels %d align %d bps %d",
165 wfx.rate, wfx.bitrate, wfx.channels, wfx.blockalign,
166 wfx.bitspersample );
168 if( p_dec->fmt_in.i_codec == VLC_CODEC_WMA1 )
169 wfx.codec_id = ASF_CODEC_ID_WMAV1;
170 else if( p_dec->fmt_in.i_codec == VLC_CODEC_WMA2 )
171 wfx.codec_id = ASF_CODEC_ID_WMAV2;
173 wfx.datalen = p_dec->fmt_in.i_extra;
174 if( wfx.datalen > 6 ) wfx.datalen = 6;
175 if( wfx.datalen > 0 )
176 memcpy( wfx.data, p_dec->fmt_in.p_extra, wfx.datalen );
178 /* Init codec */
179 if( wma_decode_init(&p_sys->wmadec, &wfx ) < 0 )
181 msg_Err( p_dec, "codec init failed" );
182 free( p_sys );
183 return VLC_EGENERIC;
186 /* Set callback */
187 p_dec->pf_decode_audio = DecodeFrame;
189 return VLC_SUCCESS;
192 /*****************************************************************************
193 * DecodeFrame: decodes a wma frame.
194 *****************************************************************************/
195 static aout_buffer_t *DecodeFrame( decoder_t *p_dec, block_t **pp_block )
197 decoder_sys_t *p_sys = p_dec->p_sys;
198 block_t *p_block;
199 aout_buffer_t *p_aout_buffer = NULL;
200 #ifdef NDEBUG
201 mtime_t start = mdate(); /* for statistics */
202 #endif
204 if( !pp_block || !*pp_block ) return NULL;
206 p_block = *pp_block;
208 if( p_block->i_flags&(BLOCK_FLAG_DISCONTINUITY|BLOCK_FLAG_CORRUPTED) )
210 date_Set( &p_sys->end_date, 0 );
211 block_Release( p_block );
212 *pp_block = NULL;
213 return NULL;
216 if( p_block->i_buffer <= 0 )
218 /* we already decoded the samples, just feed a few to aout */
219 if( p_sys->i_samples )
220 p_aout_buffer = SplitBuffer( p_dec );
221 if( !p_sys->i_samples )
222 { /* we need to decode new samples now */
223 free( p_sys->p_output );
224 p_sys->p_output = NULL;
225 block_Release( p_block );
226 *pp_block = NULL;
228 return p_aout_buffer;
231 /* Date management */
232 if( p_block->i_pts > VLC_TS_INVALID &&
233 p_block->i_pts != date_Get( &p_sys->end_date ) )
235 date_Set( &p_sys->end_date, p_block->i_pts );
236 /* don't reuse the same pts */
237 p_block->i_pts = VLC_TS_INVALID;
239 else if( !date_Get( &p_sys->end_date ) )
241 /* We've just started the stream, wait for the first PTS. */
242 block_Release( p_block );
243 return NULL;
246 if( wma_decode_superframe_init( &p_sys->wmadec, p_block->p_buffer,
247 p_block->i_buffer ) == 0 )
249 msg_Err( p_dec, "failed initializing wmafixed decoder" );
250 block_Release( p_block );
251 *pp_block = NULL;
252 return NULL;
255 if( p_sys->wmadec.nb_frames <= 0 )
257 msg_Err( p_dec, "can not decode, invalid ASF packet ?" );
258 block_Release( p_block );
259 *pp_block = NULL;
260 return NULL;
263 /* worst case */
264 size_t i_buffer = BLOCK_MAX_SIZE * MAX_CHANNELS * p_sys->wmadec.nb_frames;
265 free( p_sys->p_output );
266 p_sys->p_output = malloc(i_buffer * sizeof(int32_t) );
267 p_sys->p_samples = (int8_t*)p_sys->p_output;
269 if( !p_sys->p_output )
271 /* OOM, will try a bit later if VLC hasn't been killed */
272 block_Release( p_block );
273 return NULL;
276 p_sys->i_samples = 0;
278 for( int i = 0 ; i < p_sys->wmadec.nb_frames; i++ )
280 int i_samples = 0;
282 i_samples = wma_decode_superframe_frame( &p_sys->wmadec,
283 p_sys->p_output + p_sys->i_samples * p_sys->wmadec.nb_channels,
284 p_block->p_buffer, p_block->i_buffer );
286 if( i_samples < 0 )
288 msg_Warn( p_dec,
289 "wma_decode_superframe_frame() failed for frame %d", i );
290 free( p_sys->p_output );
291 p_sys->p_output = NULL;
292 return NULL;
294 p_sys->i_samples += i_samples; /* advance in the samples buffer */
297 p_block->i_buffer = 0; /* this block has been decoded */
299 for( size_t s = 0 ; s < i_buffer; s++ )
300 p_sys->p_output[s] >>= 2; /* Q30 -> Q28 translation */
302 p_aout_buffer = SplitBuffer( p_dec );
303 assert( p_aout_buffer );
305 #ifdef NDEBUG
306 msg_Dbg( p_dec, "%s took %"PRIi64" us",__func__,mdate()-start);
307 #endif
308 return p_aout_buffer;
311 /*****************************************************************************
312 * CloseDecoder : wma decoder destruction
313 *****************************************************************************/
314 static void CloseDecoder( vlc_object_t *p_this )
316 decoder_sys_t *p_sys = ((decoder_t*)p_this)->p_sys;
318 free( p_sys->p_output );
319 free( p_sys );