git.sesse.net Git - vlc/blob - modules/demux/aiff.c

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