git.sesse.net Git - vlc/blob - modules/codec/wmafixed/wma.c

   1 /*****************************************************************************
   2  * wma.c: wma decoder using integer decoder from Rockbox, based on FFmpeg
   3  *****************************************************************************
   4  * Copyright (C) 2008-2009 M2X
   5  *
   6  * Authors: Rafaël Carré <rcarre@m2x.nl>
   7  *
   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.
  12  *
  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.
  17  *
  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  *****************************************************************************/
  22
  23 /*****************************************************************************
  24  * Preamble
  25  *****************************************************************************/
  26 #ifdef HAVE_CONFIG_H
  27 # include "config.h"
  28 #endif
  29
  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>
  36
  37 #include <assert.h>
  38
  39 #include "wmadec.h"
  40
  41 /*****************************************************************************
  42  * decoder_sys_t : wma decoder descriptor
  43  *****************************************************************************/
  44 struct decoder_sys_t
  45 {
  46     audio_date_t end_date; /* To set the PTS */
  47     WMADecodeContext wmadec; /* name is self explanative */
  48
  49     int32_t *p_output; /* buffer where the frames are rendered */
  50
  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 */
  54 };
  55
  56 /* FIXME : check supported configurations */
  57 /* channel configuration */
  58 static unsigned int pi_channels_maps[7] =
  59 {
  60     0,
  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
  67 };
  68
  69 /*****************************************************************************
  70  * Local prototypes
  71  *****************************************************************************/
  72 static int  OpenDecoder   ( vlc_object_t * );
  73 static void CloseDecoder  ( vlc_object_t * );
  74
  75 static aout_buffer_t *DecodeFrame  ( decoder_t *, block_t ** );
  76
  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", 50 );
  85     add_shortcut( "wmafixed" )
  86     set_callbacks( OpenDecoder, CloseDecoder );
  87 vlc_module_end();
  88
  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 )
  94 {
  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;
  98
  99     if( i_samples == 0 ) return NULL;
 100
 101     if( !( p_buffer = p_dec->pf_aout_buffer_new( p_dec, i_samples ) ) )
 102         return NULL;
 103
 104     p_buffer->start_date = aout_DateGet( &p_sys->end_date );
 105     p_buffer->end_date = aout_DateIncrement( &p_sys->end_date, i_samples );
 106
 107     memcpy( p_buffer->p_buffer, p_sys->p_samples, p_buffer->i_nb_bytes );
 108     p_sys->p_samples += p_buffer->i_nb_bytes;
 109     p_sys->i_samples -= i_samples;
 110
 111     return p_buffer;
 112 }
 113
 114 /*****************************************************************************
 115  * OpenDecoder: probe the decoder and return score
 116  *****************************************************************************/
 117 static int OpenDecoder( vlc_object_t *p_this )
 118 {
 119     decoder_t *p_dec = (decoder_t*)p_this;
 120     decoder_sys_t *p_sys;
 121
 122     if( p_dec->fmt_in.i_codec != VLC_FOURCC('w','m','a','1') &&
 123         p_dec->fmt_in.i_codec != VLC_FOURCC('W','M','A','1') &&
 124         p_dec->fmt_in.i_codec != VLC_FOURCC('w','m','a','2') &&
 125         p_dec->fmt_in.i_codec != VLC_FOURCC('W','M','A','2') )
 126     {
 127         return VLC_EGENERIC;
 128     }
 129
 130     /* Allocate the memory needed to store the decoder's structure */
 131     p_dec->p_sys = p_sys = (decoder_sys_t *)malloc(sizeof(decoder_sys_t));
 132     if( !p_sys )
 133         return VLC_ENOMEM;
 134
 135     memset( p_sys, 0, sizeof( decoder_sys_t ) );
 136
 137     /* Date */
 138     aout_DateInit( &p_sys->end_date, p_dec->fmt_in.audio.i_rate );
 139
 140     /* Set output properties */
 141     p_dec->fmt_out.i_cat = AUDIO_ES;
 142     p_dec->fmt_out.i_codec = VLC_FOURCC('f','i','3','2');
 143     p_dec->fmt_out.audio.i_bitspersample = p_dec->fmt_in.audio.i_bitspersample;
 144     p_dec->fmt_out.audio.i_rate = p_dec->fmt_in.audio.i_rate;
 145
 146     p_dec->fmt_out.audio.i_channels = p_dec->fmt_in.audio.i_channels;
 147
 148     assert( p_dec->fmt_out.audio.i_channels <
 149         ( sizeof( pi_channels_maps ) / sizeof( pi_channels_maps[0] ) ) );
 150
 151     p_dec->fmt_out.audio.i_original_channels =
 152         p_dec->fmt_out.audio.i_physical_channels =
 153             pi_channels_maps[p_dec->fmt_out.audio.i_channels];
 154
 155     /* aout core assumes this number is not 0 and uses it in divisions */
 156     assert( p_dec->fmt_out.audio.i_physical_channels != 0 );
 157
 158     asf_waveformatex_t wfx;
 159     wfx.rate = p_dec->fmt_in.audio.i_rate;
 160     wfx.bitrate = p_dec->fmt_in.i_bitrate;
 161     wfx.channels = p_dec->fmt_in.audio.i_channels;
 162     wfx.blockalign = p_dec->fmt_in.audio.i_blockalign;
 163     wfx.bitspersample = p_dec->fmt_in.audio.i_bitspersample;
 164
 165     msg_Dbg( p_dec, "samplerate %d bitrate %d channels %d align %d bps %d",
 166         wfx.rate, wfx.bitrate, wfx.channels, wfx.blockalign,
 167         wfx.bitspersample );
 168
 169     if( p_dec->fmt_in.i_codec == VLC_FOURCC('w','m','a','1')
 170         || p_dec->fmt_in.i_codec == VLC_FOURCC('W','M','A','1') )
 171         wfx.codec_id = ASF_CODEC_ID_WMAV1;
 172     else if( p_dec->fmt_in.i_codec == VLC_FOURCC('W','M','A','2')
 173         || p_dec->fmt_in.i_codec == VLC_FOURCC('w','m','a','2') )
 174         wfx.codec_id = ASF_CODEC_ID_WMAV2;
 175
 176     wfx.datalen = p_dec->fmt_in.i_extra;
 177     if( wfx.datalen > 6 ) wfx.datalen = 6;
 178     if( wfx.datalen > 0 )
 179         memcpy( wfx.data, p_dec->fmt_in.p_extra, wfx.datalen );
 180
 181     /* Init codec */
 182     if( wma_decode_init(&p_sys->wmadec, &wfx ) < 0 )
 183     {
 184         msg_Err( p_dec, "codec init failed" );
 185         free( p_sys );
 186         return VLC_EGENERIC;
 187     }
 188
 189     /* Set callback */
 190     p_dec->pf_decode_audio = DecodeFrame;
 191
 192     return VLC_SUCCESS;
 193 }
 194
 195 /*****************************************************************************
 196  * DecodeFrame: decodes a wma frame.
 197  *****************************************************************************/
 198 static aout_buffer_t *DecodeFrame( decoder_t *p_dec, block_t **pp_block )
 199 {
 200     decoder_sys_t *p_sys = p_dec->p_sys;
 201     block_t       *p_block;
 202     aout_buffer_t *p_aout_buffer = NULL;
 203 #ifdef NDEBUG
 204     mtime_t start = mdate(); /* for statistics */
 205 #endif
 206
 207     if( !pp_block || !*pp_block ) return NULL;
 208
 209     p_block = *pp_block;
 210
 211     if( p_block->i_flags&(BLOCK_FLAG_DISCONTINUITY|BLOCK_FLAG_CORRUPTED) )
 212     {
 213         aout_DateSet( &p_sys->end_date, 0 );
 214         block_Release( p_block );
 215         *pp_block = NULL;
 216         return NULL;
 217     }
 218
 219     if( p_block->i_buffer <= 0 )
 220     {
 221         /* we already decoded the samples, just feed a few to aout */
 222         if( p_sys->i_samples )
 223             p_aout_buffer = SplitBuffer( p_dec );
 224         if( !p_sys->i_samples )
 225         {   /* we need to decode new samples now */
 226             free( p_sys->p_output );
 227             p_sys->p_output = NULL;
 228             block_Release( p_block );
 229             *pp_block = NULL;
 230         }
 231         return p_aout_buffer;
 232     }
 233
 234     /* Date management */
 235     if( p_block->i_pts > 0 &&
 236         p_block->i_pts != aout_DateGet( &p_sys->end_date ) )
 237     {
 238         aout_DateSet( &p_sys->end_date, p_block->i_pts );
 239         /* don't reuse the same pts */
 240         p_block->i_pts = 0;
 241     }
 242     else if( !aout_DateGet( &p_sys->end_date ) )
 243     {
 244         /* We've just started the stream, wait for the first PTS. */
 245         block_Release( p_block );
 246         return NULL;
 247     }
 248
 249     if( wma_decode_superframe_init( &p_sys->wmadec, p_block->p_buffer,
 250             p_block->i_buffer ) == 0 )
 251     {
 252         msg_Err( p_dec, "failed initializing wmafixed decoder" );
 253         block_Release( p_block );
 254         *pp_block = NULL;
 255         return NULL;
 256     }
 257
 258     if( p_sys->wmadec.nb_frames <= 0 )
 259     {
 260         msg_Err( p_dec, "can not decode, invalid ASF packet ?" );
 261         block_Release( p_block );
 262         *pp_block = NULL;
 263         return NULL;
 264     }
 265
 266     /* worst case */
 267     size_t i_buffer = BLOCK_MAX_SIZE * MAX_CHANNELS * p_sys->wmadec.nb_frames;
 268     if( p_sys->p_output )
 269         free( p_sys->p_output );
 270     p_sys->p_output = malloc(i_buffer * sizeof(int32_t) );
 271     p_sys->p_samples = (int8_t*)p_sys->p_output;
 272
 273     if( !p_sys->p_output )
 274     {
 275         /* OOM, will try a bit later if VLC hasn't been killed */
 276         block_Release( p_block );
 277         return NULL;
 278     }
 279
 280     p_sys->i_samples = 0;
 281
 282     for( int i = 0 ; i < p_sys->wmadec.nb_frames; i++ )
 283     {
 284         int i_samples = 0;
 285
 286         i_samples = wma_decode_superframe_frame( &p_sys->wmadec,
 287                  p_sys->p_output + p_sys->i_samples * p_sys->wmadec.nb_channels,
 288                  p_block->p_buffer, p_block->i_buffer );
 289
 290         if( i_samples < 0 )
 291         {
 292             msg_Warn( p_dec,
 293                 "wma_decode_superframe_frame() failed for frame %d", i );
 294             free( p_sys->p_output );
 295             p_sys->p_output = NULL;
 296             return NULL;
 297         }
 298         p_sys->i_samples += i_samples; /* advance in the samples buffer */
 299     }
 300
 301     p_block->i_buffer = 0; /* this block has been decoded */
 302
 303     for( size_t s = 0 ; s < i_buffer; s++ )
 304         p_sys->p_output[s] >>= 2; /* Q30 -> Q28 translation */
 305
 306     p_aout_buffer = SplitBuffer( p_dec );
 307     assert( p_aout_buffer );
 308
 309 #ifdef NDEBUG
 310     msg_Dbg( p_dec, "%s took %"PRIi64" us",__func__,mdate()-start);
 311 #endif
 312     return p_aout_buffer;
 313 }
 314
 315 /*****************************************************************************
 316  * CloseDecoder : wma decoder destruction
 317  *****************************************************************************/
 318 static void CloseDecoder( vlc_object_t *p_this )
 319 {
 320     decoder_sys_t *p_sys = ((decoder_t*)p_this)->p_sys;
 321
 322     free( p_sys->p_output );
 323     free( p_sys );
 324 }