modules/demux/aiff.c

   1 /*****************************************************************************
   2  * aiff.c: Audio Interchange File Format demuxer
   3  *****************************************************************************
   4  * Copyright (C) 2004-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 <vlc_common.h>
  33 #include <vlc_plugin.h>
  34 #include <vlc_demux.h>
  35 #include <limits.h>
  36
  37 /* TODO:
  38  *  - ...
  39  */
  40
  41 /*****************************************************************************
  42  * Module descriptor
  43  *****************************************************************************/
  44 static int  Open    ( vlc_object_t * );
  45 static void Close  ( vlc_object_t * );
  46
  47 vlc_module_begin ()
  48     set_category( CAT_INPUT )
  49     set_subcategory( SUBCAT_INPUT_DEMUX )
  50     set_description( N_("AIFF demuxer" ) )
  51     set_capability( "demux", 10 )
  52     set_callbacks( Open, Close )
  53     add_shortcut( "aiff" )
  54 vlc_module_end ()
  55
  56 /*****************************************************************************
  57  * Local prototypes
  58  *****************************************************************************/
  59
  60 struct demux_sys_t
  61 {
  62     es_format_t  fmt;
  63     es_out_id_t *es;
  64
  65     int64_t     i_ssnd_pos;
  66     int64_t     i_ssnd_size;
  67     int         i_ssnd_offset;
  68     int         i_ssnd_blocksize;
  69
  70     /* real data start */
  71     int64_t     i_ssnd_start;
  72     int64_t     i_ssnd_end;
  73
  74     int         i_ssnd_fsize;
  75
  76     int64_t     i_time;
  77 };
  78
  79 static int Demux  ( demux_t *p_demux );
  80 static int Control( demux_t *p_demux, int i_query, va_list args );
  81
  82 /* GetF80BE: read a 80 bits float in big endian */
  83 static unsigned int GetF80BE( const uint8_t p[10] )
  84 {
  85     unsigned int i_mantissa = GetDWBE( &p[2] );
  86     int          i_exp = 30 - p[1];
  87     unsigned int i_last = 0;
  88
  89     while( i_exp-- > 0 )
  90     {
  91         i_last = i_mantissa;
  92         i_mantissa >>= 1;
  93     }
  94     if( i_last&0x01 )
  95     {
  96         i_mantissa++;
  97     }
  98     return i_mantissa;
  99 }
 100
 101 /*****************************************************************************
 102  * Open
 103  *****************************************************************************/
 104 static int Open( vlc_object_t *p_this )
 105 {
 106     demux_t     *p_demux = (demux_t*)p_this;
 107     demux_sys_t *p_sys;
 108
 109     const uint8_t *p_peek;
 110
 111     if( vlc_stream_Peek( p_demux->s, &p_peek, 12 ) < 12 )
 112         return VLC_EGENERIC;
 113     if( memcmp( p_peek, "FORM", 4 ) || memcmp( &p_peek[8], "AIFF", 4 ) )
 114         return VLC_EGENERIC;
 115
 116     /* skip aiff header */
 117     if( vlc_stream_Read( p_demux->s, NULL, 12 ) < 12 )
 118         return VLC_EGENERIC;
 119
 120     /* Fill p_demux field */
 121     DEMUX_INIT_COMMON(); p_sys = p_demux->p_sys;
 122     es_format_Init( &p_sys->fmt, AUDIO_ES, VLC_FOURCC( 't', 'w', 'o', 's' ) );
 123     p_sys->i_time = 0;
 124     p_sys->i_ssnd_pos = -1;
 125
 126     for( ;; )
 127     {
 128         if( vlc_stream_Peek( p_demux->s, &p_peek, 8 ) < 8 )
 129             goto error;
 130
 131         uint32_t i_data_size = GetDWBE( &p_peek[4] );
 132         uint64_t i_chunk_size = UINT64_C( 8 ) + i_data_size + ( i_data_size & 1 );
 133
 134         msg_Dbg( p_demux, "chunk fcc=%4.4s size=%" PRIu64 " data_size=%" PRIu32,
 135             p_peek, i_chunk_size, i_data_size );
 136
 137         if( !memcmp( p_peek, "COMM", 4 ) )
 138         {
 139             if( vlc_stream_Peek( p_demux->s, &p_peek, 18+8 ) < 18+8 )
 140                 goto error;
 141
 142             p_sys->fmt.audio.i_channels = GetWBE( &p_peek[8] );
 143             p_sys->fmt.audio.i_bitspersample = GetWBE( &p_peek[14] );
 144             p_sys->fmt.audio.i_rate     = GetF80BE( &p_peek[16] );
 145
 146             msg_Dbg( p_demux, "COMM: channels=%d samples_frames=%d bits=%d rate=%d",
 147                      GetWBE( &p_peek[8] ), GetDWBE( &p_peek[10] ), GetWBE( &p_peek[14] ),
 148                      GetF80BE( &p_peek[16] ) );
 149         }
 150         else if( !memcmp( p_peek, "SSND", 4 ) )
 151         {
 152             if( vlc_stream_Peek( p_demux->s, &p_peek, 8+8 ) < 8+8 )
 153                 goto error;
 154
 155             p_sys->i_ssnd_pos = vlc_stream_Tell( p_demux->s );
 156             p_sys->i_ssnd_size = i_data_size;
 157             p_sys->i_ssnd_offset = GetDWBE( &p_peek[8] );
 158             p_sys->i_ssnd_blocksize = GetDWBE( &p_peek[12] );
 159
 160             msg_Dbg( p_demux, "SSND: (offset=%d blocksize=%d)",
 161                      p_sys->i_ssnd_offset, p_sys->i_ssnd_blocksize );
 162         }
 163         if( p_sys->i_ssnd_pos >= 12 && p_sys->fmt.audio.i_channels != 0 )
 164         {
 165             /* We have found the 2 needed chunks */
 166             break;
 167         }
 168
 169         /* consume chunk data */
 170         for( ssize_t i_req; i_chunk_size; i_chunk_size -= i_req )
 171         {
 172 #if SSIZE_MAX < UINT64_MAX
 173             i_req = __MIN( SSIZE_MAX, i_chunk_size );
 174 #else
 175             i_req = i_chunk_size;
 176 #endif
 177             if( vlc_stream_Read( p_demux->s, NULL, i_req ) != i_req )
 178             {
 179                 msg_Warn( p_demux, "incomplete file" );
 180                 goto error;
 181             }
 182         }
 183     }
 184
 185     p_sys->i_ssnd_start = p_sys->i_ssnd_pos + 16 + p_sys->i_ssnd_offset;
 186     p_sys->i_ssnd_end   = p_sys->i_ssnd_start + p_sys->i_ssnd_size;
 187
 188     p_sys->i_ssnd_fsize = p_sys->fmt.audio.i_channels *
 189                           ((p_sys->fmt.audio.i_bitspersample + 7) / 8);
 190
 191     if( p_sys->i_ssnd_fsize <= 0 || p_sys->fmt.audio.i_rate == 0 )
 192     {
 193         msg_Err( p_demux, "invalid audio parameters" );
 194         goto error;
 195     }
 196
 197     if( p_sys->i_ssnd_size <= 0 )
 198     {
 199         /* unknown */
 200         p_sys->i_ssnd_end = 0;
 201     }
 202
 203     /* seek into SSND chunk */
 204     if( vlc_stream_Seek( p_demux->s, p_sys->i_ssnd_start ) )
 205     {
 206         msg_Err( p_demux, "cannot seek to data chunk" );
 207         goto error;
 208     }
 209
 210     /* */
 211     p_sys->es = es_out_Add( p_demux->out, &p_sys->fmt );
 212
 213     return VLC_SUCCESS;
 214
 215 error:
 216     free( p_sys );
 217     return VLC_EGENERIC;
 218 }
 219
 220 /*****************************************************************************
 221  * Close
 222  *****************************************************************************/
 223 static void Close( vlc_object_t *p_this )
 224 {
 225     demux_t     *p_demux = (demux_t*)p_this;
 226     demux_sys_t *p_sys = p_demux->p_sys;
 227
 228     free( p_sys );
 229 }
 230
 231
 232 /*****************************************************************************
 233  * Demux:
 234  *****************************************************************************/
 235 static int Demux( demux_t *p_demux )
 236 {
 237     demux_sys_t *p_sys = p_demux->p_sys;
 238     int64_t     i_tell = vlc_stream_Tell( p_demux->s );
 239
 240     block_t     *p_block;
 241     int         i_read;
 242
 243     if( p_sys->i_ssnd_end > 0 && i_tell >= p_sys->i_ssnd_end )
 244     {
 245         /* EOF */
 246         return 0;
 247     }
 248
 249     /* Set PCR */
 250     es_out_SetPCR( p_demux->out, VLC_TS_0 + p_sys->i_time);
 251
 252     /* we will read 100ms at once */
 253     i_read = p_sys->i_ssnd_fsize * ( p_sys->fmt.audio.i_rate / 10 );
 254     if( p_sys->i_ssnd_end > 0 && p_sys->i_ssnd_end - i_tell < i_read )
 255     {
 256         i_read = p_sys->i_ssnd_end - i_tell;
 257     }
 258     if( ( p_block = vlc_stream_Block( p_demux->s, i_read ) ) == NULL )
 259     {
 260         return 0;
 261     }
 262
 263     p_block->i_dts =
 264     p_block->i_pts = VLC_TS_0 + p_sys->i_time;
 265
 266     p_sys->i_time += (int64_t)1000000 *
 267                      p_block->i_buffer /
 268                      p_sys->i_ssnd_fsize /
 269                      p_sys->fmt.audio.i_rate;
 270
 271     /* */
 272     es_out_Send( p_demux->out, p_sys->es, p_block );
 273     return 1;
 274 }
 275
 276 /*****************************************************************************
 277  * Control:
 278  *****************************************************************************/
 279 static int Control( demux_t *p_demux, int i_query, va_list args )
 280 {
 281     demux_sys_t *p_sys = p_demux->p_sys;
 282     double f, *pf;
 283     int64_t *pi64;
 284
 285     switch( i_query )
 286     {
 287         case DEMUX_CAN_SEEK:
 288             return vlc_stream_vaControl( p_demux->s, i_query, args );
 289
 290         case DEMUX_GET_POSITION:
 291         {
 292             int64_t i_start = p_sys->i_ssnd_start;
 293             int64_t i_end   = p_sys->i_ssnd_end > 0 ? p_sys->i_ssnd_end : stream_Size( p_demux->s );
 294             int64_t i_tell  = vlc_stream_Tell( p_demux->s );
 295
 296             pf = va_arg( args, double * );
 297
 298             if( i_start < i_end )
 299             {
 300                 *pf = (double)(i_tell - i_start)/(double)(i_end - i_start);
 301                 return VLC_SUCCESS;
 302             }
 303             return VLC_EGENERIC;
 304         }
 305
 306         case DEMUX_SET_POSITION:
 307         {
 308             int64_t i_start = p_sys->i_ssnd_start;
 309             int64_t i_end  = p_sys->i_ssnd_end > 0 ? p_sys->i_ssnd_end : stream_Size( p_demux->s );
 310
 311             f = va_arg( args, double );
 312
 313             if( i_start < i_end )
 314             {
 315                 int     i_frame = (f * ( i_end - i_start )) / p_sys->i_ssnd_fsize;
 316                 int64_t i_new   = i_start + i_frame * p_sys->i_ssnd_fsize;
 317
 318                 if( vlc_stream_Seek( p_demux->s, i_new ) )
 319                 {
 320                     return VLC_EGENERIC;
 321                 }
 322                 p_sys->i_time = (int64_t)1000000 * i_frame / p_sys->fmt.audio.i_rate;
 323                 return VLC_SUCCESS;
 324             }
 325             return VLC_EGENERIC;
 326         }
 327
 328         case DEMUX_GET_TIME:
 329             pi64 = va_arg( args, int64_t * );
 330             *pi64 = p_sys->i_time;
 331             return VLC_SUCCESS;
 332
 333         case DEMUX_GET_LENGTH:
 334         {
 335             int64_t i_end  = p_sys->i_ssnd_end > 0 ? p_sys->i_ssnd_end : stream_Size( p_demux->s );
 336
 337             pi64 = va_arg( args, int64_t * );
 338             if( p_sys->i_ssnd_start < i_end )
 339             {
 340                 *pi64 = (int64_t)1000000 * ( i_end - p_sys->i_ssnd_start ) / p_sys->i_ssnd_fsize / p_sys->fmt.audio.i_rate;
 341                 return VLC_SUCCESS;
 342             }
 343             return VLC_EGENERIC;
 344         }
 345         case DEMUX_SET_TIME:
 346         case DEMUX_GET_FPS:
 347         default:
 348             return VLC_EGENERIC;
 349     }
 350 }