modules/demux/au.c

   1 /*****************************************************************************
   2  * au.c : au file input module for vlc
   3  *****************************************************************************
   4  * Copyright (C) 2001-2007 VLC authors and VideoLAN
   5  * $Id$
   6  *
   7  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
   8  *
   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.
  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 Lesser General Public License for more details.
  18  *
  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  *****************************************************************************/
  23
  24 /*****************************************************************************
  25  * Preamble
  26  *****************************************************************************/
  27
  28 #ifdef HAVE_CONFIG_H
  29 # include "config.h"
  30 #endif
  31
  32 #include <limits.h>
  33 #include <vlc_common.h>
  34 #include <vlc_plugin.h>
  35 #include <vlc_demux.h>
  36
  37 /* TODO:
  38  *  - all adpcm things (I _NEED_ samples)
  39  *  - ...
  40  */
  41
  42 /*****************************************************************************
  43  * Module descriptor
  44  *****************************************************************************/
  45 static int  Open ( vlc_object_t * );
  46 static void Close( vlc_object_t * );
  47
  48 vlc_module_begin ()
  49     set_category( CAT_INPUT )
  50     set_subcategory( SUBCAT_INPUT_DEMUX )
  51     set_description( N_("AU demuxer") )
  52     set_capability( "demux", 10 )
  53     set_callbacks( Open, Close )
  54     add_shortcut( "au" )
  55 vlc_module_end ()
  56
  57 /*****************************************************************************
  58  * Local prototypes
  59  *****************************************************************************/
  60 enum AuType_e
  61 {
  62     AU_UNKNOWN      =  0,
  63     AU_MULAW_8      =  1,  /* 8-bit ISDN u-law */
  64     AU_LINEAR_8     =  2,  /* 8-bit linear PCM */
  65     AU_LINEAR_16    =  3,  /* 16-bit linear PCM */
  66     AU_LINEAR_24    =  4,  /* 24-bit linear PCM */
  67     AU_LINEAR_32    =  5,  /* 32-bit linear PCM */
  68     AU_FLOAT        =  6,  /* 32-bit IEEE floating point */
  69     AU_DOUBLE       =  7,  /* 64-bit IEEE floating point */
  70     AU_ADPCM_G721   =  23, /* 4-bit CCITT g.721 ADPCM */
  71     AU_ADPCM_G722   =  24, /* CCITT g.722 ADPCM */
  72     AU_ADPCM_G723_3 =  25, /* CCITT g.723 3-bit ADPCM */
  73     AU_ADPCM_G723_5 =  26, /* CCITT g.723 5-bit ADPCM */
  74     AU_ALAW_8       =  27  /* 8-bit ISDN A-law */
  75 };
  76
  77 enum AuCat_e
  78 {
  79     AU_CAT_UNKNOWN  = 0,
  80     AU_CAT_PCM      = 1,
  81     AU_CAT_ADPCM    = 2
  82 };
  83
  84 struct demux_sys_t
  85 {
  86     es_format_t     fmt;
  87     es_out_id_t     *es;
  88
  89     mtime_t         i_time;
  90
  91     int             i_frame_size;
  92     mtime_t         i_frame_length;
  93
  94     uint32_t        i_header_size;
  95 };
  96
  97 static int Demux( demux_t * );
  98 static int Control ( demux_t *, int i_query, va_list args );
  99
 100 /*****************************************************************************
 101  * Open: check file and initializes structures
 102  *****************************************************************************/
 103 static int Open( vlc_object_t *p_this )
 104 {
 105     demux_t     *p_demux = (demux_t*)p_this;
 106     demux_sys_t *p_sys;
 107
 108     uint8_t      hdr[20];
 109     const uint8_t *p_peek;
 110     int          i_cat;
 111
 112     if( vlc_stream_Peek( p_demux->s , &p_peek, 4 ) < 4 )
 113         return VLC_EGENERIC;
 114
 115     if( memcmp( p_peek, ".snd", 4 ) )
 116         return VLC_EGENERIC;
 117
 118     /* skip signature */
 119     if( vlc_stream_Read( p_demux->s, NULL, 4 ) < 4 )
 120         return VLC_EGENERIC;
 121
 122     /* read header */
 123     if( vlc_stream_Read( p_demux->s, hdr, 20 ) < 20 )
 124     {
 125         msg_Err( p_demux, "cannot read" );
 126         return VLC_EGENERIC;
 127     }
 128
 129     if( GetDWBE( &hdr[0]  ) < 24 )
 130     {
 131         msg_Err( p_demux, "invalid file" );
 132         return VLC_EGENERIC;
 133     }
 134
 135     p_sys = malloc( sizeof (*p_sys) );
 136     if( unlikely(p_sys == NULL) )
 137         return VLC_ENOMEM;
 138
 139     p_sys->i_time = 0;
 140     p_sys->i_header_size = GetDWBE( &hdr[0] );
 141
 142     /* skip extra header data */
 143     if( p_sys->i_header_size > 24 )
 144     {
 145 #if (SSIZE_MAX <= INT32_MAX)
 146         if( p_sys->i_header_size > SSIZE_MAX )
 147             goto error;
 148 #endif
 149         size_t skip = p_sys->i_header_size - 24;
 150         if( vlc_stream_Read( p_demux->s, NULL, skip ) < (ssize_t)skip )
 151             goto error;
 152     }
 153
 154     /* init fmt */
 155     es_format_Init( &p_sys->fmt, AUDIO_ES, 0 );
 156     p_sys->fmt.audio.i_rate     = GetDWBE( &hdr[12] );
 157     p_sys->fmt.audio.i_channels = GetDWBE( &hdr[16] );
 158
 159 #if 0
 160     p_sys->au.i_header_size   = GetDWBE( &p_sys->au.i_header_size );
 161     p_sys->au.i_data_size     = GetDWBE( &p_sys->au.i_data_size );
 162     p_sys->au.i_encoding      = GetDWBE( &p_sys->au.i_encoding );
 163     p_sys->au.i_sample_rate   = GetDWBE( &p_sys->au.i_sample_rate );
 164     p_sys->au.i_channels      = GetDWBE( &p_sys->au.i_channels );
 165 #endif
 166     switch( GetDWBE( &hdr[8] ) )
 167     {
 168         case AU_ALAW_8:        /* 8-bit ISDN A-law */
 169             p_sys->fmt.i_codec               = VLC_CODEC_ALAW;
 170             p_sys->fmt.audio.i_bitspersample = 8;
 171             p_sys->fmt.audio.i_blockalign    = 1 * p_sys->fmt.audio.i_channels;
 172             i_cat                    = AU_CAT_PCM;
 173             break;
 174
 175         case AU_MULAW_8:       /* 8-bit ISDN u-law */
 176             p_sys->fmt.i_codec               = VLC_CODEC_MULAW;
 177             p_sys->fmt.audio.i_bitspersample = 8;
 178             p_sys->fmt.audio.i_blockalign    = 1 * p_sys->fmt.audio.i_channels;
 179             i_cat                    = AU_CAT_PCM;
 180             break;
 181
 182         case AU_LINEAR_8:      /* 8-bit linear PCM */
 183             p_sys->fmt.i_codec               = VLC_CODEC_S8;
 184             p_sys->fmt.audio.i_bitspersample = 8;
 185             p_sys->fmt.audio.i_blockalign    = 1 * p_sys->fmt.audio.i_channels;
 186             i_cat                    = AU_CAT_PCM;
 187             break;
 188
 189         case AU_LINEAR_16:     /* 16-bit linear PCM */
 190             p_sys->fmt.i_codec               = VLC_CODEC_S16B;
 191             p_sys->fmt.audio.i_bitspersample = 16;
 192             p_sys->fmt.audio.i_blockalign    = 2 * p_sys->fmt.audio.i_channels;
 193             i_cat                    = AU_CAT_PCM;
 194             break;
 195
 196         case AU_LINEAR_24:     /* 24-bit linear PCM */
 197             p_sys->fmt.i_codec               = VLC_CODEC_S24B;
 198             p_sys->fmt.audio.i_bitspersample = 24;
 199             p_sys->fmt.audio.i_blockalign    = 3 * p_sys->fmt.audio.i_channels;
 200             i_cat                    = AU_CAT_PCM;
 201             break;
 202
 203         case AU_LINEAR_32:     /* 32-bit linear PCM */
 204             p_sys->fmt.i_codec               = VLC_CODEC_S32B;
 205             p_sys->fmt.audio.i_bitspersample = 32;
 206             p_sys->fmt.audio.i_blockalign    = 4 * p_sys->fmt.audio.i_channels;
 207             i_cat                    = AU_CAT_PCM;
 208             break;
 209
 210         case AU_FLOAT:         /* 32-bit IEEE floating point */
 211             p_sys->fmt.i_codec               = VLC_FOURCC( 'a', 'u', 0, AU_FLOAT );
 212             p_sys->fmt.audio.i_bitspersample = 32;
 213             p_sys->fmt.audio.i_blockalign    = 4 * p_sys->fmt.audio.i_channels;
 214             i_cat                    = AU_CAT_PCM;
 215             break;
 216
 217         case AU_DOUBLE:        /* 64-bit IEEE floating point */
 218             p_sys->fmt.i_codec               = VLC_FOURCC( 'a', 'u', 0, AU_DOUBLE );
 219             p_sys->fmt.audio.i_bitspersample = 64;
 220             p_sys->fmt.audio.i_blockalign    = 8 * p_sys->fmt.audio.i_channels;
 221             i_cat                    = AU_CAT_PCM;
 222             break;
 223
 224         case AU_ADPCM_G721:    /* 4-bit CCITT g.721 ADPCM */
 225             p_sys->fmt.i_codec               = VLC_FOURCC( 'a', 'u', 0, AU_ADPCM_G721 );
 226             p_sys->fmt.audio.i_bitspersample = 0;
 227             p_sys->fmt.audio.i_blockalign    = 0 * p_sys->fmt.audio.i_channels;
 228             i_cat                    = AU_CAT_ADPCM;
 229             break;
 230
 231         case AU_ADPCM_G722:    /* CCITT g.722 ADPCM */
 232             p_sys->fmt.i_codec               = VLC_FOURCC( 'a', 'u', 0, AU_ADPCM_G722 );
 233             p_sys->fmt.audio.i_bitspersample = 0;
 234             p_sys->fmt.audio.i_blockalign    = 0 * p_sys->fmt.audio.i_channels;
 235             i_cat                    = AU_CAT_ADPCM;
 236             break;
 237
 238         case AU_ADPCM_G723_3:  /* CCITT g.723 3-bit ADPCM */
 239             p_sys->fmt.i_codec               = VLC_FOURCC( 'a', 'u', 0, AU_ADPCM_G723_3 );
 240             p_sys->fmt.audio.i_bitspersample = 0;
 241             p_sys->fmt.audio.i_blockalign    = 0 * p_sys->fmt.audio.i_channels;
 242             i_cat                    = AU_CAT_ADPCM;
 243             break;
 244
 245         case AU_ADPCM_G723_5:  /* CCITT g.723 5-bit ADPCM */
 246             p_sys->fmt.i_codec               = VLC_FOURCC( 'a', 'u', 0, AU_ADPCM_G723_5 );
 247             p_sys->fmt.audio.i_bitspersample = 0;
 248             p_sys->fmt.audio.i_blockalign    = 0 * p_sys->fmt.audio.i_channels;
 249             i_cat                    = AU_CAT_ADPCM;
 250             break;
 251
 252         default:
 253             msg_Warn( p_demux, "unknown encoding=0x%x", GetDWBE( &hdr[8] ) );
 254             p_sys->fmt.audio.i_bitspersample = 0;
 255             p_sys->fmt.audio.i_blockalign    = 0;
 256             i_cat                    = AU_CAT_UNKNOWN;
 257             break;
 258     }
 259
 260     p_sys->fmt.i_bitrate = p_sys->fmt.audio.i_rate *
 261                            p_sys->fmt.audio.i_channels *
 262                            p_sys->fmt.audio.i_bitspersample;
 263
 264     if( i_cat == AU_CAT_UNKNOWN || i_cat == AU_CAT_ADPCM )
 265     {
 266         msg_Err( p_demux, "unsupported codec/type (Please report it)" );
 267         goto error;
 268     }
 269
 270     if( p_sys->fmt.audio.i_rate == 0 )
 271     {
 272         msg_Err( p_demux, "invalid samplerate: 0" );
 273         goto error;
 274     }
 275
 276     /* add the es */
 277     p_sys->es = es_out_Add( p_demux->out, &p_sys->fmt );
 278
 279     /* calculate 50ms frame size/time */
 280     unsigned i_samples = __MAX( p_sys->fmt.audio.i_rate / 20, 1 );
 281     p_sys->i_frame_size = i_samples * p_sys->fmt.audio.i_channels *
 282                           ( (p_sys->fmt.audio.i_bitspersample + 7) / 8 );
 283     if( p_sys->fmt.audio.i_blockalign > 0 )
 284     {
 285         unsigned mod = p_sys->i_frame_size % p_sys->fmt.audio.i_blockalign;
 286         if( mod != 0 )
 287         {
 288             p_sys->i_frame_size += p_sys->fmt.audio.i_blockalign - mod;
 289         }
 290     }
 291     p_sys->i_frame_length = (mtime_t)1000000 *
 292                             (mtime_t)i_samples /
 293                             (mtime_t)p_sys->fmt.audio.i_rate;
 294
 295     p_demux->p_sys = p_sys;
 296     p_demux->pf_demux = Demux;
 297     p_demux->pf_control = Control;
 298     return VLC_SUCCESS;
 299 error:
 300     free( p_sys );
 301     return VLC_EGENERIC;
 302 }
 303
 304 /*****************************************************************************
 305  * Demux: read packet and send them to decoders
 306  *****************************************************************************
 307  * Returns -1 in case of error, 0 in case of EOF, 1 otherwise
 308  *****************************************************************************/
 309 static int Demux( demux_t *p_demux )
 310 {
 311     demux_sys_t *p_sys = p_demux->p_sys;
 312     block_t     *p_block;
 313
 314     /* set PCR */
 315     es_out_SetPCR( p_demux->out, VLC_TS_0 + p_sys->i_time );
 316
 317     p_block = vlc_stream_Block( p_demux->s, p_sys->i_frame_size );
 318     if( p_block == NULL )
 319     {
 320         msg_Warn( p_demux, "cannot read data" );
 321         return 0;
 322     }
 323
 324     p_block->i_dts =
 325     p_block->i_pts = VLC_TS_0 + p_sys->i_time;
 326
 327     es_out_Send( p_demux->out, p_sys->es, p_block );
 328
 329     p_sys->i_time += p_sys->i_frame_length;
 330
 331     return 1;
 332 }
 333
 334 /*****************************************************************************
 335  * Close: frees unused data
 336  *****************************************************************************/
 337 static void Close( vlc_object_t * p_this )
 338 {
 339     demux_t     *p_demux = (demux_t*)p_this;
 340     demux_sys_t *p_sys = p_demux->p_sys;
 341
 342     free( p_sys );
 343 }
 344
 345 /*****************************************************************************
 346  * Control:
 347  *****************************************************************************/
 348 static int Control( demux_t *p_demux, int i_query, va_list args )
 349 {
 350     demux_sys_t *p_sys = p_demux->p_sys;
 351
 352     return demux_vaControlHelper( p_demux->s, p_sys->i_header_size, -1,
 353                                    p_sys->fmt.i_bitrate, p_sys->fmt.audio.i_blockalign,
 354                                    i_query, args );
 355 }
 356